diff options
Diffstat (limited to 'mcast/tool/.svn/text-base/netcvupdate.c.svn-base')
-rw-r--r-- | mcast/tool/.svn/text-base/netcvupdate.c.svn-base | 773 |
1 files changed, 0 insertions, 773 deletions
diff --git a/mcast/tool/.svn/text-base/netcvupdate.c.svn-base b/mcast/tool/.svn/text-base/netcvupdate.c.svn-base deleted file mode 100644 index e311470..0000000 --- a/mcast/tool/.svn/text-base/netcvupdate.c.svn-base +++ /dev/null @@ -1,773 +0,0 @@ -/*------------------------------------------------------------------------ - * netcvupdate - NetCeiver update tool - * - * Principle for firmware update - * - Unpack given .tgz into host /tmp/mkdtemp() - * - "md5sum -c md5sums.txt" - * - read script file update.scr - * - feed commands into tnftp - * - * - *------------------------------------------------------------------------*/ -#define USE_MCLI_API - -#ifdef USE_MCLI_API -#include "headers.h" -#else -#include <stdio.h> -#include <stdlib.h> -#include <string.h> -#include <errno.h> -#include <unistd.h> -#include <signal.h> -#include <ctype.h> -#endif - -#define STATE_FILE "update.log" -#define FTP_CMD "ftp" -#define NC_CONFPATH "/mmc/etc/" -#define NC_CONFFILE "netceiver.conf" - -char ftp_cmd[512]=FTP_CMD; -int verbose=0; -int no_reboot=0; -char username[256]="root"; -char password[256]="root"; -char device[256]="eth0"; -char *uuids[256]={0}; -char *versions[256]={0}; -int num_uuids=0; -char socket_path[256]=API_SOCK_NAMESPACE; - -#ifdef USE_MCLI_API -/*------------------------------------------------------------------------*/ -#define API_WAIT_RESPONSE(cmd) { cmd->state=API_REQUEST; \ - send (sock_comm, &sock_cmd, sizeof(api_cmd_t), 0); \ - recv (sock_comm, &sock_cmd, sizeof(api_cmd_t), 0); \ - if (cmd->state == API_ERROR) {warn ( "SHM parameter error, incompatible versions?\n"); exit(-1);}} - -int sock_comm; - -int api_init(char *path) -{ - int sock_name_len = 0; - struct sockaddr_un sock_name; - sock_name.sun_family = AF_UNIX; - - strcpy(sock_name.sun_path, path); - sock_name_len = sizeof(struct sockaddr_un); - - if((sock_comm = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) - { - warn ("socket create failure %d\n", errno); - return -1; - } - - if (connect(sock_comm, (struct sockaddr*)&sock_name, sock_name_len) < 0) { - warn ("connect failure to %s: %s (are you root?)\n",path, strerror(errno)); - return -1; - } - return 0; -} -#endif -/*------------------------------------------------------------------------*/ -void add_upload(char* cmd, char* localfile, char *remotepath, char *remotefile) -{ - char tmp[1024]; - sprintf(tmp, - "cd %s\n" - "site exec rm -f /tmp/update/%s\n" - "put %s\n", - remotepath, remotefile, localfile); - strcat(cmd,tmp); -} -/*------------------------------------------------------------------------*/ -void add_download(char* cmd, char *remotepath, char *remotefile) -{ - char tmp[1024]; - sprintf(tmp, - "cd %s\n" - "get %s\n", - remotepath,remotefile); - strcat(cmd,tmp); -} -/*------------------------------------------------------------------------*/ -int script_interpreter(char *script, char *line, - char *ip, char *iface, char *user, char *pwd) -{ - char cmd[256],p1[256],p2[256]; - int end=0; - - *cmd=0; - *p1=0; - *p2=0; - sscanf(line,"%s %s %s\n",cmd,p1,p2); - - if (cmd[0]=='#') - return 0; - if (!strcmp(cmd,"connect")) { - char tmp[1024]; - sprintf(tmp, - "open %s%%%s\n" - "user %s %s\n", - ip,iface,user,pwd); - strcat(script,tmp); - } - else if (!strcmp(cmd,"upload")) { - add_upload(script,p2,p1,p2); - } - else if (!strcmp(cmd,"download")) { - add_download(script,p1,p2); - } - else if (!strcmp(cmd,"exit")) { - strcat(script,"quit\n"); - end=1; - } - else { - strcat(script,line); - } - return end; -} -/*------------------------------------------------------------------------*/ -char script[128*1024]; -int generate_script(char *filename, char *tmpname, char *ip, char *iface, char *user, char *pwd) -{ - FILE *f; - f=fopen(filename,"r"); - if (!f) { - fprintf(stderr,"Can't open script file <%s>: %s\n",filename,strerror(errno)); - return -1; - } - script[0]=0; - - while(!feof(f)) { - char line[256]; - fgets(line,255,f); - if (script_interpreter(script,line,ip,iface,user,pwd)) - break; - } - fclose(f); - return 0; -} -/*------------------------------------------------------------------------*/ -int check_state_file(char *tmpname) -{ - char cmd[512]; - int ret; - printf("\nUPDATE RESULT:\n"); - snprintf(cmd,512,"cat %s/update.log",tmpname); - ret=system(cmd); - printf("\n"); - return ret; -} -/*------------------------------------------------------------------------*/ -void sigground(int x) -{ -} -/*------------------------------------------------------------------------*/ -int run_ftp(char *tmpdir,char *script, int timeout, char *pipeout) -{ - FILE *f; - char cmd[512]; - int ret; - if (!strlen(ftp_cmd)) - return -1; - signal(SIGPIPE,sigground); - if (strlen(tmpdir)) -// snprintf(cmd,511,"cd %s; %s -q %i -n %s",tmpdir,ftp_cmd,timeout,verbose?"":"-V"); - snprintf(cmd,511,"cd %s; %s -n %s %s",tmpdir,ftp_cmd,verbose?"":"-V",pipeout); - else - snprintf(cmd,511,"%s -q %i -n %s %s",ftp_cmd,timeout,verbose?"":"-V",pipeout); - - f=popen(cmd,"w"); - if (!f) - return -1; - fputs(script,f); - ret=pclose(f); - return ret; -} -/*------------------------------------------------------------------------*/ -int do_reboot(char *tmpdir, char *ip, char* iface, char *user, char* pwd) -{ - sprintf(script, - "open %s%%%s\n" - "user %s %s\n" - "site exec reboot -d 5\n" - "quit\n" - , - ip,iface,user,pwd); - return run_ftp(tmpdir, script, 15,""); -} -/*------------------------------------------------------------------------*/ - -int do_list_fw(char *tmpdir, char *ip, char* iface, char *user, char* pwd, int maxf, int *found, char **versions) -{ - char tmpfile[256]="/tmp/ncvup.XXXXXX"; - char pipeout[256]; - int n=0; - int ret=0; - FILE *file; - - *found=0; - - if (!mkstemp(tmpfile)) { - fprintf(stderr,"Can't make temporary directory %s!\n",tmpfile); - return -2; - } - - sprintf(script, - "open %s%%%s\n" - "user %s %s\n" - "ls /mmc/\n" - "quit\n" - , - ip,iface,user,pwd); - sprintf(pipeout," > %s",tmpfile); - ret=run_ftp(tmpdir, script, 15, pipeout); - if (ret) { - unlink(tmpfile); - return ret; - } - - file=fopen(tmpfile,"r"); - if (!file) { - unlink(tmpfile); // ? - perror("Can't read temp file"); - return ret; - } - - while(!feof(file)) { - char line[1024]; - char *p; - *line=0; - fgets(line, 1023,file); - line[1023]=0; - p=strstr(line,"etceivr."); - if (p) { - - char *pp=strchr(p,'\n'); - if (pp) - *pp=0; - - if (n < maxf) { - n++; - *versions++=strdup(p-1); - } - } - } - *found=n; - fclose(file); - unlink(tmpfile); - return 0; -} -/*------------------------------------------------------------------------*/ -int do_kill(char *tmpdir, char *ip, char* iface, char *user, char* pwd) -{ - sprintf(script, - "open %s%%%s\n" - "user %s %s\n" - "site exec killall -9 mserv\n" - "quit\n" - , - ip,iface,user,pwd); - return run_ftp(tmpdir, script, 15,""); -} -/*------------------------------------------------------------------------*/ -int do_single_update(char *tmpdir, char *uuid, char *device) -{ - char path[256]; - int ret; - - snprintf(path,255,"%s/%s",tmpdir,"update.scr"); - if (generate_script(path, tmpdir, uuid, device, username, password)) - return -1; -// puts(script); - - printf("Upload update... "); - fflush(stdout); - - ret=run_ftp(tmpdir, script, 600,""); - if (ret) - return ret; - - printf("check result... \n"); - fflush(stdout); - - if (check_state_file(tmpdir)) - return -1; - -#if 1 - if (!no_reboot) { - printf("Issue Reboot... "); - fflush(stdout); - ret=do_reboot(tmpdir, uuid, device, username, password); - if (!ret) - return ret; - } -#endif - return 0; -} -/*------------------------------------------------------------------------*/ -int do_single_upload( char *uuid, char *device, char *remote_path, char *fname, char *remote_file) -{ - int ret; - sprintf(script, - "open %s%%%s\n" - "user %s %s\n" - "cd %s\n" - "put %s %s\n" -// "site exec killall -HUP mserv\n" - "quit", - uuid,device,username,password,remote_path,fname,remote_file); - ret=run_ftp("", script, 120,""); - return ret; -} -/*------------------------------------------------------------------------*/ -int do_single_download( char *uuid, char *device, char *remote_path, char *fname) -{ - int ret; - sprintf(script, - "open %s%%%s\n" - "user %s %s\n" - "cd %s\n" - "get %s\n" - "quit", - uuid,device,username,password,remote_path,fname); - ret=run_ftp("", script, 120,""); - return ret; -} -/*------------------------------------------------------------------------*/ -int fw_action(char *uuid, char* iface, char *user, char* pwd, int mode, char *version) -{ - int ret; - if (mode==0) { // inactivate - printf("Inactivating version %s\n",version); - sprintf(script, - "open %s%%%s\n" - "user %s %s\n" - "cd /mmc\n" - "rename netceivr.%s xetceivr.%s\n" - "quit", - uuid,device,username,password,version,version); - ret=run_ftp("", script, 120,""); - return ret; - } - else if (mode==1) { // enable - printf("Enabling version %s\n",version); - sprintf(script, - "open %s%%%s\n" - "user %s %s\n" - "cd /mmc\n" - "rename xetceivr.%s netceivr.%s\n" - "quit", - uuid,device,username,password,version,version); - ret=run_ftp("", script, 120,""); - return ret; - } - else if (mode==2) { // delete - printf("Removing version %s\n",version); - sprintf(script, - "open %s%%%s\n" - "user %s %s\n" - "site exec rm -rf /mmc/netceivr.%s\n" - "site exec rm -rf /mmc/xetceivr.%s\n" - "quit", - uuid,device,username,password,version,version); - ret=run_ftp("", script, 120,""); - return ret; - } - return 0; -} -/*------------------------------------------------------------------------*/ -int cleanup(char *tmpdir) -{ - int ret; - char cmd[1024]; - snprintf(cmd,1024,"rm -rf '%s'",tmpdir); -// puts(cmd); - ret=system(cmd); - return ret; -} -/*------------------------------------------------------------------------*/ -int unpack(char *tmpdir, char *file) -{ - int ret; - char cmd[1024]; - snprintf(cmd,1024,"tar xfz '%s' --directory %s",file,tmpdir); -// puts(cmd); - ret=system(cmd); - return ret; -} -/*------------------------------------------------------------------------*/ -int check_xml(char *file) -{ - int ret; - char cmd[1024]; - snprintf(cmd,1024,"xmllint --noout '%s'\n",file); -// puts(cmd); - ret=system(cmd); - return ret; -} -/*------------------------------------------------------------------------*/ -int check_integrity(char *tmpdir) -{ - int ret; - char cmd[1024]; - snprintf(cmd,1024,"cd %s; md5sum -c --status md5sums.txt \n",tmpdir); -// puts(cmd); - ret=system(cmd); - return ret; -} -/*------------------------------------------------------------------------*/ -int do_update(char **uuids, int num_uuids, char *device, char *optarg) -{ - char tmpdir[256]="/tmp/ncvupXXXXXX"; - int n; - int ret=0; - if (!mkdtemp(tmpdir)) { - fprintf(stderr,"Can't make temporary directory %s!\n",tmpdir); - return -2; - } -// printf("TEMP DIR %s\n",tmpdir); - if (unpack(tmpdir,optarg)) { - fprintf(stderr,"Update file <%s> cannot be unpacked!\n",optarg); - cleanup(tmpdir); - return -2; - } - if (check_integrity(tmpdir)) { - fprintf(stderr,"Update file <%s> corrupted!\n",optarg); - cleanup(tmpdir); - return -2; - } - printf("Update file integrity OK\n"); - printf("NUM uuids %i\n",num_uuids); - for(n=0;n<num_uuids;n++) { - if (!uuids[n]) - continue; - - printf("UUID %s: ",uuids[n]); - fflush(stdout); - ret=do_single_update(tmpdir, uuids[n], device); - if (!ret) - printf("-> Update done <-\n"); - else { - printf("-> Update failed (ret=%i) <-\n",ret); - uuids[n]=NULL; - } - } - - cleanup(tmpdir); - return ret; -} -/*------------------------------------------------------------------------*/ -int do_upload(char **uuids, int num_uuids, char *device, char *optarg) -{ - int n; - int ret=0; - if (check_xml(optarg)) { - fprintf(stderr,"Configuration file <%s> not valid XML\n",optarg); - return -2; - } - for(n=0;n<num_uuids;n++) { - if (!uuids[n]) - continue; - - printf("UUID %s: Uploading %s ... ",uuids[n], optarg); - fflush(stdout); - ret=do_single_upload(uuids[n], device, "/mmc/etc/", optarg, NC_CONFFILE); - if (!ret) - printf("Upload done\n"); - else { - printf("Upload failed (ret=%i)\n",ret); - uuids[n]=NULL; - } - } - return ret; -} -/*------------------------------------------------------------------------*/ -int do_download(char **uuids, int num_uuids, char *device, char *remotepath, char *file) -{ - int n,ret=0; - - for(n=0;n<num_uuids;n++) { - char newfile[1024]; - if (!uuids[n]) - continue; - - if (num_uuids!=1) - snprintf(newfile,1024,"%s-%s",file,uuids[n]); - else - strncpy(newfile,file,1024); - - printf("UUID %s: Downloading %s ... ",uuids[n], newfile); - fflush(stdout); - ret=do_single_download(uuids[n], device, remotepath, file); - if (!ret) { - printf("Done\n"); - if (num_uuids!=1) - rename(file,newfile); - } - else { - printf("Download failed (ret=%i)\n",ret); - uuids[n]=NULL; - } - } - return ret; -} -/*------------------------------------------------------------------------*/ -int do_all_reboot(char **uuids, int num_uuids, char *device) -{ - int n,ret=0; - - for(n=0;n<num_uuids;n++) { - if (!uuids[n]) - continue; - printf("UUID %s: Issue Reboot... ",uuids[n]); - fflush(stdout); - ret=do_reboot("/tmp", uuids[n], device, username, password); - if (!ret) - printf("Reboot done\n"); - else - printf("Reboot failed (ret=%i)\n",ret); - } - return ret; -} -/*------------------------------------------------------------------------*/ -int do_all_kill(char **uuids, int num_uuids, char *device) -{ - int n,ret=0; - - for(n=0;n<num_uuids;n++) { - if (!uuids[n]) - continue; - printf("UUID %s: Issue Kill... ",uuids[n]); - fflush(stdout); - ret=do_kill("/tmp", uuids[n], device, username, password); - if (!ret) - printf("Kill done\n"); - else - printf("Kill failed (ret=%i)\n",ret); - } - return ret; -} -/*------------------------------------------------------------------------*/ -int get_uuids(char **uuids, int max) -{ - int count,n; - api_cmd_t sock_cmd; - api_cmd_t *api_cmd=&sock_cmd; - - if (api_init(socket_path)==-1) { - exit(-1); - } - api_cmd->cmd=API_GET_NC_NUM; - api_cmd->magic = MCLI_MAGIC; - api_cmd->version = MCLI_VERSION; - API_WAIT_RESPONSE(api_cmd); - if(api_cmd->magic != MCLI_MAGIC || api_cmd->version != MCLI_VERSION) { - err("API version mismatch!\n"); - } - count=api_cmd->parm[API_PARM_NC_NUM]; - - for(n=0;n<max && n<count;n++) { - api_cmd->cmd=API_GET_NC_INFO; - api_cmd->parm[API_PARM_NC_NUM]=n; - API_WAIT_RESPONSE(api_cmd); - if(api_cmd->u.nc_info.magic != MCLI_MAGIC || api_cmd->u.nc_info.version != MCLI_VERSION) { - err("API version mismatch!\n"); - } - - uuids[n]=strdup(api_cmd->u.nc_info.uuid); - versions[n]=strdup(api_cmd->u.nc_info.FirmwareVersion); - } - return count; -} -/*------------------------------------------------------------------------*/ -int show_uuids(void) -{ - char *uuids[256]={0}; - int num_uuids,n; - num_uuids=get_uuids(uuids,256); - for(n=0;n<num_uuids;n++) { - printf("%s %s\n",uuids[n],versions[n]); - } - return 0; -} -/*------------------------------------------------------------------------*/ -#define MAX_FWS 64 -void show_firmwares(char *uuid, char *device, char *username, char *passwd) -{ - char *fwversions[MAX_FWS]; - int found,m; - found=0; - - do_list_fw("/tmp", uuid, device, username, password, MAX_FWS, &found, fwversions); - - printf("Firmware versions found: %i\n Versions: ", found); - for(m=0;m<found;m++) { - if (m!=0) printf(", "); - if (fwversions[m][0]!='n') - printf("%s (disabled)",fwversions[m]+9); - else - printf("%s",fwversions[m]+9); - - free(versions[m]); - } - puts(""); -} -/*------------------------------------------------------------------------*/ - -int show_all_firmwares(char **uuids, int num_uuids) -{ - int n; - - for(n=0;n<num_uuids;n++) { - printf("%s: ",uuids[n]); - fflush(stdout); - show_firmwares(uuids[n],device,username,password); - } - return 0; -} -/*------------------------------------------------------------------------*/ -void do_fw_actions(char **uuids, int max, int mode, char *version) -{ - int n; - - if (strlen(version)!=3 || !strcmp(version,"000")) { - fprintf(stderr,"Invalid version number\n"); - return; - } - - for(n=0;n<3;n++) - version[n]=toupper(version[n]); - - for(n=0;n<max;n++) { - printf("UUID %s\n",uuids[n]); - if (fw_action(uuids[n], device, username, password, mode, version)) { - fprintf(stderr,"Failed\n"); - return; - } - show_firmwares(uuids[n],device,username,password); - } -} -/*------------------------------------------------------------------------*/ -void usage(void) -{ - fprintf(stderr, - "netcvupdate - NetCeiver update tool, version " MCLI_VERSION_STR "\n" - "(c) BayCom GmbH\n" - "Usage: netcvupdate <options> <actions> \n" - "Actions: \n" - " -l List all seen NetCeivers and their UUID\n" - " -L List available FWs\n" - " -X <Update.tgz> Update with given file\n" - " -U <configfile> Upload configfile\n" - " -D Download configfile netceiver.conf\n" - " -I <version> Inactivate FW version\n" - " -E <version> Enable FW version\n" - " -Z <version> Remove FW version\n" - " -K Restart streaming server\n" - " -R Issue reboot\n" - "Options:\n" - " -A Use all found NetCeivers (mcli must be running)\n" - " -i <uuid> Use specific UUID (can be used multiple times)\n" - " *** Either -A or -i must be given for most actions! ***\n" - "Rare options:\n" - " -d <device> Set network device (default: eth0)\n" - " -F <ftp-command> Set ftp command/path\n" - " *** ftp command must understand the -q (timeout) option! ***\n" - " -P <path> Set API socket\n" - " -u <user> Set username\n" - " -p <password> Set password\n" - " -r No reboot after update\n" - " -q Be more quiet\n" - ); - exit(0); -} -/*------------------------------------------------------------------------*/ -int main(int argc, char **argv) -{ - int ret=0; - - while(1) { - int ret = getopt(argc,argv, "U:X:Di:AlLI:E:Z:d:F:P:u:p:rRqK"); - if (ret==-1) - break; - - char c=(char)ret; - - switch(c) { - case 'F': - strncpy(ftp_cmd,optarg,512); - ftp_cmd[511]=0; - break; - case 'X': - ret=do_update(uuids, num_uuids, device, optarg); - if (ret==-2) - exit(ret); - break; - case 'U': - ret=do_upload(uuids, num_uuids, device, optarg); - if (ret==-2) - exit(ret); - break; - case 'D': - ret|=do_download(uuids, num_uuids, device, NC_CONFPATH, NC_CONFFILE); - break; - case 'i': - uuids[num_uuids]=strdup(optarg); - num_uuids++; - break; - case 'A': - num_uuids=get_uuids(uuids,255); - break; - case 'l': - show_uuids(); - break; - case 'd': - strncpy(device,optarg,255); - device[255]=0; - break; - case 'P': - strncpy(socket_path,optarg,255); - socket_path[255]=0; - break; - case 'p': - strncpy(password,optarg,255); - password[255]=0; - break; - case 'u': - strncpy(username,optarg,255); - username[255]=0; - break; - case 'r': - no_reboot=1; - break; - case 'K': - ret|=do_all_kill(uuids,num_uuids,device); - break; - case 'R': - ret|=do_all_reboot(uuids, num_uuids, device); - break; - case 'L': - show_all_firmwares(uuids, num_uuids); - break; - case 'I': - do_fw_actions(uuids, num_uuids, 0, optarg); - break; - case 'E': - do_fw_actions(uuids, num_uuids, 1, optarg); - break; - case 'Z': - do_fw_actions(uuids, num_uuids, 2, optarg); - break; - case 'q': - verbose=0; - break; - default: - usage(); - break; - } - } - exit(ret); -} |