static int add_fifo_socket(modparam_t type, void * val)
{
char *s;
struct id_list* id;
if ((type & PARAM_STRING)==0){
LOG(L_CRIT, "BUG: ctl: add_fifo: bad parameter type %d\n",
type);
goto error;
}
s=(char*)val;
id=parse_listen_id(s, strlen(s), FIFO_SOCK); /* default udp proto */
if (id==0){
LOG(L_ERR, "ERROR: ctl: bad fifo: \"%s\"\n", s);
goto error;
}
id->data_proto=P_FIFO;
id->next=listen_lst;
listen_lst=id;
return 0;
error:
return -1;
}
int main(int argc, char** argv)
{
int c;
char* sock_name;
int sock_type;
int s;
struct binrpc_cmd cmd;
struct id_list* sock_id;
char* format;
char* line;
char* l;
quit=0;
format=0;
line=0;
s=-1;
sock_name=0;
sock_type=UNIXS_SOCK;
opterr=0;
while((c=getopt(argc, argv, "UVhs:D:R:vf:"))!=-1){
switch(c){
case 'V':
printf("version: %s\n", version);
printf("%s compiled on %s \n", __FILE__,
compiled);
exit(0);
break;
case 'h':
printf("version: %s\n", version);
printf("%s", help_msg);
exit(0);
break;
case 's':
sock_name=optarg;
break;
case 'R':
reply_socket=optarg;
break;
case 'D':
sock_dir=optarg;
break;
case 'U':
sock_type=UDP_SOCK;
break;
case 'v':
verbose++;
break;
case 'f':
format=str_escape(optarg);
if (format==0){
fprintf(stderr, "ERROR: memory allocation failure\n");
goto error;
}
break;
case '?':
if (isprint(optopt))
fprintf(stderr, "Unknown option `-%c'.\n", optopt);
else
fprintf(stderr,
"Unknown option character `\\x%x'.\n",
optopt);
goto error;
case ':':
fprintf(stderr,
"Option `-%c' requires an argument.\n",
optopt);
goto error;
default:
abort();
}
}
if (sock_name==0){
sock_name=DEFAULT_CTL_SOCKET;
}
/* init the random number generator */
srand(getpid()+time(0)); /* we don't need very strong random numbers */
if (sock_name==0){
fprintf(stderr, "ERROR: no server socket address specified\n");
goto error;
}
sock_id=parse_listen_id(sock_name, strlen(sock_name), sock_type);
if (sock_id==0){
fprintf(stderr, "ERROR: error parsing server socket address %s\n", sock_name);
goto error;
}
switch(sock_id->proto){
case UDP_SOCK:
case TCP_SOCK:
if (sock_id->port==0){
sock_id->port=DEFAULT_CTL_PORT;
/*
fprintf(stderr, "ERROR: no port specified: %s:<port>\n",
sock_name);
goto error;
*/
}
if ((s=connect_tcpudp_socket(sock_id->name, sock_id->port,
(sock_id->proto==UDP_SOCK)?SOCK_DGRAM:
SOCK_STREAM))<0){
goto error;
}
break;
case UNIXS_SOCK:
case UNIXD_SOCK:
if ((s=connect_unix_sock(sock_id->name,
(sock_id->proto==UNIXD_SOCK)?SOCK_DGRAM:
SOCK_STREAM))<0)
goto error;
break;
case UNKNOWN_SOCK:
fprintf(stderr, "ERROR: Bad socket type for %s\n", sock_name);
goto error;
}
free(sock_id); /* not needed anymore */
sock_id=0;
if (optind>=argc){
/*fprintf(stderr, "ERROR: no command specified\n");
goto error; */
}else{
if (parse_cmd(&cmd, &argv[optind], argc-optind)<0)
goto error;
if (run_cmd(s, &cmd, format)<0)
goto error;
goto end;
}
/* interactive mode */
if (get_kamcmd_list(s)==0){
#ifdef USE_CFG_VARS
get_cfgvars_list(s);
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
get_counters_list(s);
#endif /* USE_COUNTERS */
}
/* banners */
printf("%s %s\n", NAME, VERSION);
printf("%s\n", COPYRIGHT);
printf("%s\n", DISCLAIMER);
#ifdef USE_READLINE
/* initialize readline */
/* allow conditional parsing of the ~/.inputrc file*/
rl_readline_name=NAME;
rl_completion_entry_function=kamcmd_generator;
rl_attempted_completion_function=kamcmd_completion;
while(!quit){
line=readline(NAME "> ");
if (line==0) /* EOF */
break;
l=trim_ws(line); /* trim whitespace */
if (*l){
add_history(l);
run_line(s, l, format);
}
free(line);
line=0;
}
#else
line=malloc(MAX_LINE_SIZE);
if (line==0){
fprintf(stderr, "memory allocation error\n");
goto error;
}
printf(NAME "> "); fflush(stdout); /* prompt */
while(!quit && fgets(line, MAX_LINE_SIZE, stdin)){
l=trim_ws(line);
if (*l){
run_line(s, l, format);
}
printf(NAME "> "); fflush(stdout); /* prompt */
};
free(line);
line=0;
#endif /* USE_READLINE */
end:
/* normal exit */
if (line)
free(line);
if (format)
free(format);
if (rpc_array)
free_rpc_array(rpc_array, rpc_no);
#ifdef USE_CFG_VARS
if (cfg_grp_lst)
free_cfg_grp_lst();
if (cfg_vars_array){
free_rpc_array(cfg_vars_array, cfg_vars_no);
cfg_vars_array=0;
cfg_vars_no=0;
}
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
if (cnt_grp_lst)
free_cnt_grp_lst();
if (cnt_grps_array){
free_rpc_array(cnt_grps_array, cnt_grps_no);
cnt_grps_array=0;
cnt_grps_no=0;
}
#endif /* USE_COUNTERS */
cleanup();
exit(0);
error:
if (line)
free(line);
if (format)
free(format);
if (rpc_array)
free_rpc_array(rpc_array, rpc_no);
#ifdef USE_CFG_VARS
if (cfg_grp_lst)
free_cfg_grp_lst();
if (cfg_vars_array){
free_rpc_array(cfg_vars_array, cfg_vars_no);
cfg_vars_array=0;
cfg_vars_no=0;
}
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
if (cnt_grp_lst)
free_cnt_grp_lst();
if (cnt_grps_array){
free_rpc_array(cnt_grps_array, cnt_grps_no);
cnt_grps_array=0;
cnt_grps_no=0;
}
#endif /* USE_COUNTERS */
cleanup();
exit(-1);
}
