/*
* Copyright (C) 2006 iptelorg GmbH
*
* This file is part of kamcmd, a free cli tool for Kamailio SIP server.
*
* kamailio is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version
*
* For a license to use the kamailio software under conditions
* other than those described here, or to purchase support for this
* software, please contact iptel.org by e-mail at the following addresses:
* info@iptel.org
*
* kamailio is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
/*
* send commands using binrpc
*
*/
#include <stdlib.h> /* exit, abort */
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <ctype.h> /* isprint */
#include <sys/socket.h>
#include <sys/un.h> /* unix sock*/
#include <netinet/in.h> /* udp sock */
#include <sys/uio.h> /* writev */
#include <netdb.h> /* gethostbyname */
#include <fcntl.h>
#include <time.h> /* time */
#include <sys/time.h>
#ifdef USE_READLINE
#include <readline/readline.h>
#include <readline/history.h>
#define USE_CFG_VARS /* cfg group and vars completion */
#define USE_COUNTERS /* counters/statistics completion */
#endif
#include "parse_listen_id.h"
#include "license.h"
#include "../../src/modules/ctl/ctl_defaults.h" /* default socket & port */
#include "../../src/modules/ctl/binrpc.h"
#include "../../src/modules/ctl/binrpc.c" /* ugly hack */
#ifndef NAME
#define NAME "kamcmd"
#endif
#ifndef VERSION
#define VERSION "1.5"
#endif
#define IOVEC_CNT 20
#define MAX_LINE_SIZE 16*1024 /* for non readline mode */
#define MAX_REPLY_SIZE 128*1024
#define MAX_BODY_SIZE 128*1024
#define MAX_BINRPC_ARGS 256
#ifndef UNIX_PATH_MAX
#define UNIX_PATH_MAX 104
#endif
static char version[]= NAME " " VERSION;
#ifdef VERSION_NODATE
static char compiled[] = "";
#else
#ifdef VERSION_DATE
static char compiled[]= VERSION_DATE;
#else
static char compiled[]= __TIME__ " " __DATE__;
#endif
#endif
static char help_msg[]="\
Usage: " NAME " [options][-s address] [ cmd ]\n\
Options:\n\
-s address unix socket name or host name to send the commands on\n\
-R name force reply socket name, for the unix datagram socket mode\n\
-D dir create the reply socket in the directory <dir> if no reply \n\
socket is forced (-R) and a unix datagram socket is selected\n\
as the transport\n\
-f format print the result using format. Format is a string containing\n\
%v at the places where values read from the reply should be\n\
substituted. To print '%v', escape it using '%': %%v.\n\
-v Verbose \n\
-V Version number\n\
-h This help message\n\
address:\n\
[proto:]name[:port] where proto is one of tcp, udp, unixs or unixd\n\
e.g.: tcp:localhost:2049 , unixs:/tmp/kamailio_ctl\n\
cmd:\n\
method [arg1 [arg2...]]\n\
arg:\n\
string or number; to force a number to be interpreted as string \n\
prefix it by \"s:\", e.g. s:1\n\
Examples:\n\
" NAME " -s unixs:/tmp/" NAME "_ctl system.listMethods\n\
" NAME " -f \"pid: %v desc: %v\\n\" -s udp:localhost:2047 core.ps \n\
" NAME " ps # uses default ctl socket \n\
" NAME " # enters interactive mode on the default socket \n\
" NAME " -s tcp:localhost # interactive mode, default port \n\
";
int verbose=0;
char* reply_socket=0; /* unix datagram reply socket name */
char* sock_dir=0; /* same as above, but only the directory */
char* unix_socket=0;
struct sockaddr_un mysun;
int quit; /* used only in interactive mode */
struct binrpc_val* rpc_array;
int rpc_no=0;
#ifdef USE_CFG_VARS
struct binrpc_val* cfg_vars_array;
int cfg_vars_no;
struct cfg_var_grp{
struct cfg_var_grp* next;
str grp_name; /**< group name */
str* var_names; /**< str array, null terminated */
int var_no;
};
struct cfg_var_grp* cfg_grp_lst; /** cfg groups list, allong with var names*/
struct cfg_var_grp* crt_cfg_grp;
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
struct binrpc_val* cnt_grps_array; /* response array */
int cnt_grps_no; /* number of response records */
struct cnt_var_grp {
struct cnt_var_grp * next;
str grp_name;
str* var_names; /**< str array (null terminated strings)*/
int var_no;
struct binrpc_val* cnt_vars_array; /* var_name will point here */
int cnt_vars_no; /* cnt_vars_array size (no. of response records) */
};
struct cnt_var_grp* cnt_grp_lst; /* counters groups list, allong with vars */
struct cnt_var_grp* crt_cnt_grp;
#endif /* USE_COUNTERS */
#define IOV_SET(vect, str) \
do{\
(vect).iov_base=(str); \
(vect).iov_len=strlen((str)); \
}while(0)
#define INT2STR_MAX_LEN (19+1+1) /* 2^64~= 16*10^18 => 19+1 digits + \0 */
/* returns a pointer to a static buffer containing l in asciiz & sets len */
static inline char* int2str(unsigned int l, int* len)
{
static char r[INT2STR_MAX_LEN];
int i;
i=INT2STR_MAX_LEN-2;
r[INT2STR_MAX_LEN-1]=0; /* null terminate */
do{
r[i]=l%10+'0';
i--;
l/=10;
}while(l && (i>=0));
if (l && (i<0)){
fprintf(stderr, "BUG: int2str: overflow\n");
}
if (len) *len=(INT2STR_MAX_LEN-2)-i;
return &r[i+1];
}
static char* trim_ws(char* l)
{
char* ret;
for(;*l && ((*l==' ')||(*l=='\t')||(*l=='\n')||(*l=='\r')); l++);
ret=l;
if (*ret==0) return ret;
for(l=l+strlen(l)-1; (l>ret) &&
((*l==' ')||(*l=='\t')||(*l=='\n')||(*l=='\r')); l--);
*(l+1)=0;
return ret;
}
int gen_cookie()
{
return rand();
}
struct binrpc_cmd{
char* method;
int argc;
struct binrpc_val argv[MAX_BINRPC_ARGS];
};
struct cmd_alias{
char* name;
char* method;
char* format; /* reply print format */
};
struct kamcmd_builtin{
char* name;
int (*f)(int, struct binrpc_cmd*);
char* doc;
};
static int kamcmd_help(int s, struct binrpc_cmd* cmd);
static int kamcmd_ver(int s, struct binrpc_cmd* cmd);
static int kamcmd_quit(int s, struct binrpc_cmd* cmd);
static int kamcmd_warranty(int s, struct binrpc_cmd* cmd);
static struct cmd_alias cmd_aliases[]={
{ "ps", "core.ps", "%v\t%v\n" },
{ "psx", "core.psx", 0 },
{ "list", "system.listMethods", 0 },
{ "ls", "system.listMethods", 0 },
{ "ver", "core.version", 0 },
{ "version", "core.version", 0 },
{ "who", "ctl.who", "[%v] %v: %v %v -> %v %v\n"},
{ "listen", "ctl.listen", "[%v] %v: %v %v\n"},
{ "dns_mem_info", "dns.mem_info", "%v / %v\n"},
{ "dns_debug", "dns.debug",
"%v (%v): size=%v ref=%v expire=%vs last=%vs ago f=%v\n"},
{ "dns_debug_all", "dns.debug_all",
"%v (%v) [%v]: size=%v ref=%v expire=%vs last=%vs ago f=%v\n"
"\t\t%v:%v expire=%vs f=%v\n"},
{ "dst_blacklist_mem_info", "dst_blacklist.mem_info", "%v / %v\n"},
{ "dst_blacklist_debug", "dst_blacklist.debug",
"%v:%v:%v expire:%v flags: %v\n"},
{0,0,0}
};
static struct kamcmd_builtin builtins[]={
{ "?", kamcmd_help, "help"},
{ "help", kamcmd_help, "displays help for a command"},
{ "version", kamcmd_ver, "displays " NAME "version"},
{ "quit", kamcmd_quit, "exits " NAME },
{ "exit", kamcmd_quit, "exits " NAME },
{ "warranty", kamcmd_warranty, "displays " NAME "'s warranty info"},
{ "license", kamcmd_warranty, "displays " NAME "'s license"},
{0,0}
};
#ifdef USE_READLINE
enum complete_states {
COMPLETE_INIT,
COMPLETE_CMD_NAME,
#ifdef USE_CFG_VARS
COMPLETE_CFG_GRP,
COMPLETE_CFG_VAR,
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
COMPLETE_CNT_GRP,
COMPLETE_CNT_VAR,
#endif /* USE_COUNTERS */
COMPLETE_NOTHING
};
/* instead of rl_attempted_completion_over which is not present in
some readline emulations, use attempted_completion_state */
static enum complete_states attempted_completion_state;
static int crt_param_no;
/* commands for which we complete the params to other method names */
char* complete_params_methods[]={
"?",
"h",
"help",
"system.methodSignature",
"system.methodHelp",
0
};
#ifdef USE_CFG_VARS
/* commands for which we complete the first param with a cfg var grp*/
char* complete_params_cfg_var[]={
"cfg.get",
"cfg.help",
"cfg.set_delayed_int",
"cfg.set_delayed_string",
"cfg.set_now_int",
"cfg.set_now_string",
0
};
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
/* commands for which we complete the first param with a counter group */
char* complete_param1_counter_grp[] = {
"cnt.get",
"cnt.get_raw",
"cnt.grp_get_all",
"cnt.reset",
"cnt.var_list",
"cnt.help",
0
};
/* commands for which we completed the 2nd param with a counter name */
char* complete_param2_counter_name[] = {
"cnt.get",
"cnt.get_raw",
"cnt.reset",
"cnt.help",
0
};
#endif /* USE_COUNTERS */
#endif /* USE_READLINE */
static int parse_arg(struct binrpc_val* v, char* arg)
{
int i;
double f;
char* tmp;
int len;
i=strtol(arg, &tmp, 10);
if ((tmp==0) || (*tmp)){
f=strtod(arg, &tmp);
if ((tmp==0) || (*tmp)){
/* not an int or a float => string */
len=strlen(arg);
if ((len>=2) && (arg[0]=='s') && (arg[1]==':')){
tmp=&arg[2];
len-=2;
}else{
tmp=arg;
}
v->type=BINRPC_T_STR;
v->u.strval.s=tmp;
v->u.strval.len=len;
}else{ /* float */
v->type=BINRPC_T_DOUBLE;
v->u.fval=f;
}
}else{ /* int */
v->type=BINRPC_T_INT;
v->u.intval=i;
}
return 0;
}
static int parse_cmd(struct binrpc_cmd* cmd, char** argv, int count)
{
int r;
cmd->method=argv[0];
if ((count-1)>MAX_BINRPC_ARGS){
fprintf(stderr, "ERROR: too many args %d, only %d allowed\n",
count-1, MAX_BINRPC_ARGS);
return -1;
}
for (r=1; r<count; r++){
if (parse_arg(&cmd->argv[r-1], argv[r])<0)
return -1;
}
cmd->argc=r-1;
return 0;
}
void print_binrpc_val(struct binrpc_val* v, int ident)
{
int r;
if ((v->type==BINRPC_T_STRUCT) && !v->u.end)
ident--; /* fix to have struct beg. idented differently */
for (r=0; r<ident; r++) putchar(' ');
if (v->name.s){
printf("%.*s: ", v->name.len, v->name.s);
}
switch(v->type){
case BINRPC_T_INT:
printf("%d", v->u.intval);
break;
case BINRPC_T_STR:
case BINRPC_T_BYTES:
printf("%.*s", v->u.strval.len, v->u.strval.s);
break;
case BINRPC_T_ARRAY:
printf("%c", (v->u.end)?']':'[');
break;
case BINRPC_T_STRUCT:
printf("%c", (v->u.end)?'}':'{');
break;
case BINRPC_T_DOUBLE:
if(v->u.fval == (double)((long long int)v->u.fval)) {
printf("%lld", (long long int)v->u.fval);
} else {
printf("%f", v->u.fval);
}
break;
default:
printf("ERROR: unknown type %d\n", v->type);
};
}
/* opens, and connects on a STREAM unix socket
* returns socket fd or -1 on error */
int connect_unix_sock(char* name, int type)
{
struct sockaddr_un ifsun;
int s;
int len;
int ret;
int retries;
retries=0;
s=-1;
memset(&ifsun, 0, sizeof (struct sockaddr_un));
len=strlen(name);
if (len>UNIX_PATH_MAX){
fprintf(stderr, "ERROR: connect_unix_sock: name too long "
"(%d > %d): %s\n", len, UNIX_PATH_MAX, name);
goto error;
}
ifsun.sun_family=AF_UNIX;
memcpy(ifsun.sun_path, name, len);
#ifdef HAVE_SOCKADDR_SA_LEN
ifsun.sun_len=len;
#endif
s=socket(PF_UNIX, type, 0);
if (s==-1){
fprintf(stderr, "ERROR: connect_unix_sock: cannot create unix socket"
" %s: %s [%d]\n", name, strerror(errno), errno);
goto error;
}
if (type==SOCK_DGRAM){
/* we must bind so that we can receive replies */
if (reply_socket==0){
if (sock_dir==0)
sock_dir="/tmp";
retry:
ret=snprintf(mysun.sun_path, UNIX_PATH_MAX, "%s/" NAME "_%d",
sock_dir, rand());
if ((ret<0) ||(ret>=UNIX_PATH_MAX)){
fprintf(stderr, "ERROR: buffer overflow while trying to"
"generate unix datagram socket name");
goto error;
}
}else{
if (strlen(reply_socket)>UNIX_PATH_MAX){
fprintf(stderr, "ERROR: buffer overflow while trying to"
"use the provided unix datagram socket name (%s)",
reply_socket);
goto error;
}
strcpy(mysun.sun_path, reply_socket);
}
mysun.sun_family=AF_UNIX;
if (bind(s, (struct sockaddr*)&mysun, sizeof(mysun))==-1){
if (errno==EADDRINUSE && (reply_socket==0) && (retries < 10)){
retries++;
/* try another one */
goto retry;
}
fprintf(stderr, "ERROR: could not bind the unix socket to"
" %s: %s (%d)\n",
mysun.sun_path, strerror(errno), errno);
goto error;
}
unix_socket=mysun.sun_path;
}
if (connect(s, (struct sockaddr *)&ifsun, sizeof(ifsun))==-1){
fprintf(stderr, "ERROR: connect_unix_sock: connect(%s): %s [%d]\n",
name, strerror(errno), errno);
goto error;
}
return s;
error:
if (s!=-1) close(s);
return -1;
}
int connect_tcpudp_socket(char* address, int port, int type)
{
struct sockaddr_in addr;
struct hostent* he;
int sock;
sock=-1;
/* resolve destination */
he=gethostbyname(address);
if (he==0){
fprintf(stderr, "ERROR: could not resolve %s\n", address);
goto error;
}
/* open socket*/
addr.sin_family=he->h_addrtype;
addr.sin_port=htons(port);
memcpy(&addr.sin_addr.s_addr, he->h_addr_list[0], he->h_length);
sock = socket(he->h_addrtype, type, 0);
if (sock==-1){
fprintf(stderr, "ERROR: socket: %s\n", strerror(errno));
goto error;
}
if (connect(sock, (struct sockaddr*) &addr, sizeof(struct sockaddr))!=0){
fprintf(stderr, "ERROR: connect: %s\n", strerror(errno));
goto error;
}
return sock;
error:
if (sock!=-1) close(sock);
return -1;
}
static void hexdump(unsigned char* buf, int len, int ascii)
{
int r, i;
/* dump it in hex */
for (r=0; r<len; r++){
if ((r) && ((r%16)==0)){
if (ascii){
putchar(' ');
for (i=r-16; i<r; i++){
if (isprint(buf[i]))
putchar(buf[i]);
else
putchar('.');
}
}
putchar('\n');
}
printf("%02x ", buf[r]);
};
if (ascii){
for (i=r;i%16; i++)
printf(" ");
putchar(' ');
for (i=16*(r/16); i<r; i++){
if (isprint(buf[i]))
putchar(buf[i]);
else
putchar('.');
}
}
putchar('\n');
}
/* returns: -1 on error, number of bytes written on success */
static int send_binrpc_cmd(int s, struct binrpc_cmd* cmd, int cookie)
{
struct iovec v[IOVEC_CNT];
int r;
unsigned char msg_body[MAX_BODY_SIZE];
unsigned char msg_hdr[BINRPC_MAX_HDR_SIZE];
struct binrpc_pkt body;
int ret;
int n;
ret=binrpc_init_pkt(&body, msg_body, MAX_BODY_SIZE);
if (ret<0) goto binrpc_err;
ret=binrpc_addstr(&body, cmd->method, strlen(cmd->method));
if (ret<0) goto binrpc_err;
for (r=0; r<cmd->argc; r++){
switch(cmd->argv[r].type){
case BINRPC_T_STR:
ret=binrpc_addstr(&body, cmd->argv[r].u.strval.s,
cmd->argv[r].u.strval.len);
break;
case BINRPC_T_INT:
ret=binrpc_addint(&body, cmd->argv[r].u.intval);
break;
case BINRPC_T_DOUBLE:
ret=binrpc_adddouble(&body, cmd->argv[r].u.fval);
break;
default:
fprintf(stderr, "ERROR: unsupported type %d\n",
cmd->argv[r].type);
}
if (ret<0) goto binrpc_err;
}
ret=binrpc_build_hdr(BINRPC_REQ, binrpc_pkt_len(&body), cookie, msg_hdr,
BINRPC_MAX_HDR_SIZE);
if (ret<0) goto binrpc_err;
v[0].iov_base=msg_hdr;
v[0].iov_len=ret;
v[1].iov_base=msg_body;
v[1].iov_len=binrpc_pkt_len(&body);
write_again:
if ((n=writev(s, v, 2))<0){
if (errno==EINTR)
goto write_again;
goto error_send;
}
return n;
error_send:
return -1;
binrpc_err:
return -2;
}
static int binrpc_errno=0;
/* reads the whole reply
* returns < 0 on error, reply size on success + initializes in_pkt
* if ret==-2 (parse error), sets binrpc_errno to the binrpc error
* error returns: -1 - read error (check errno)
* -2 - binrpc parse error (chekc binrpc_errno)
* -3 - cookie error (the cookied doesn't match)
* -4 - message too big */
static int get_reply(int s, unsigned char* reply_buf, int max_reply_size,
int cookie, struct binrpc_parse_ctx* in_pkt,
unsigned char** body)
{
unsigned char* crt;
unsigned char* hdr_end;
unsigned char* msg_end;
int n;
int ret;
hdr_end=crt=reply_buf;
msg_end=reply_buf+max_reply_size;
binrpc_errno=0;
do{
n=read(s, crt, (int)(msg_end-crt));
if (n<=0){
if (errno==EINTR)
continue;
goto error_read;
}
if (verbose >= 3){
/* dump it in hex */
printf("received %d bytes in reply (@offset %d):\n",
n, (int)(crt-reply_buf));
hexdump(crt, n, 1);
}
crt+=n;
/* parse header if not parsed yet */
if (hdr_end==reply_buf){
hdr_end=binrpc_parse_init(in_pkt, reply_buf, n, &ret);
if (ret<0){
if (ret==E_BINRPC_MORE_DATA)
continue;
goto error_parse;
}
if (verbose>1){
printf("new packet: type %02x, len %d, cookie %02x\n",
in_pkt->type, in_pkt->tlen, in_pkt->cookie);
}
if (in_pkt->cookie!=cookie){
fprintf(stderr, "bad reply, cookie doesn't match: sent %02x "
"and received %02x\n",
cookie, in_pkt->cookie);
goto error;
}
msg_end=hdr_end+in_pkt->tlen;
if ((int)(msg_end-reply_buf)>max_reply_size) {
/* reading the rest from the socket */
struct timeval timeout_save;
unsigned sizeoft = sizeof(timeout_save);
if (getsockopt(s, SOL_SOCKET, SO_RCVTIMEO,
&timeout_save, &sizeoft)==0) {
struct timeval timeout;
timeout.tv_sec = 1;
timeout.tv_usec = 0;
if (setsockopt (s, SOL_SOCKET, SO_RCVTIMEO,
(char*)&timeout,sizeof(timeout))==0) {
while(read(s, reply_buf, max_reply_size)>0);
setsockopt(s, SOL_SOCKET, SO_RCVTIMEO,
(char*)&timeout_save,sizeof(timeout_save));
}
}
goto error_toolong;
}
}
}while(crt<msg_end);
*body=hdr_end;
return (int)(msg_end-reply_buf);
error_read:
return -1;
error_parse:
binrpc_errno=ret;
return -2;
error:
return -3;
error_toolong:
return -4;
}
/* returns a malloced copy of str, with all the escapes ('\') resolved */
static char* str_escape(char* str)
{
char* n;
char* ret;
ret=n=malloc(strlen(str)+1);
if (n==0)
goto end;
for(;*str;str++){
*n=*str;
if (*str=='\\'){
switch(*(str+1)){
case 'n':
*n='\n';
str++;
break;
case 'r':
*n='\r';
str++;
break;
case 't':
*n='\t';
str++;
break;
case '\\':
str++;
break;
}
}
n++;
}
*n=*str; /* terminating 0 */
end:
return ret;
}
/* parses strings like "bla bla %v 10%% %v\n test=%v",
* and stops at each %v, returning a pointer after the %v, setting *size
* to the string length (not including %v) and *type to the corresponding
* BINRPC type (for now only BINRPC_T_ALL).
* To escape a '%', use "%%", and check for type==-1 (which means skip an call
* again parse_fmt).
* Usage:
* n="test: %v,%v,%v\n";
* while(*n){
* s=n;
* n=parse_fmt(n, &type, &size);
* printf("%.*s", size, s);
* if (type==-1)
* continue;
* else
* printf("now we should get & print an object of type %d\n", type)
* }
*/
static char* parse_fmt(char* fmt, int* type, int* size)
{
char* s;
s=fmt;
do{
for(;*fmt && *fmt!='%'; fmt++);
if (*fmt=='%'){
switch(*(fmt+1)){
case 'v':
*type=BINRPC_T_ALL;
*size=(int)(fmt-s);
return (fmt+2);
break;
case '%':
/* escaped % */
*size=(int)(fmt-s)+1;
*type=-1; /* skip */
return (fmt+2);
break;
}
}
}while(*fmt);
*type=-1; /* no value */
*size=(fmt-s);
return fmt;
}
static int print_body(struct binrpc_parse_ctx* in_pkt,
unsigned char* body, int size, char* fmt)
{
unsigned char* p;
unsigned char* end;
struct binrpc_val val;
int ret;
int rec;
char *f;
char* s;
int f_size;
int fmt_has_values;
p=body;
end=p+size;
rec=0;
f=fmt;
fmt_has_values=0;
/* read body */
while(p<end){
if (f){
do{
if (*f==0)
f=fmt; /* reset */
s=f;
f=parse_fmt(f, &val.type, &f_size);
printf("%.*s", f_size, s);
if (val.type!=-1){
fmt_has_values=1;
goto read_value;
}
}while(*f || fmt_has_values);
val.type=BINRPC_T_ALL;
}else{
val.type=BINRPC_T_ALL;
}
read_value:
val.name.s=0;
val.name.len=0;
p=binrpc_read_record(in_pkt, p, end, &val, 1, &ret);
if (ret<0){
if (fmt)
putchar('\n');
/*if (ret==E_BINRPC_MORE_DATA)
goto error_read_again;*/
if (ret==E_BINRPC_EOP){
printf("end of message detected\n");
break;
}
fprintf(stderr, "ERROR:: while parsing the record %d,"
" @%d: %02x : %s\n", rec,
in_pkt->offset, *p, binrpc_error(ret));
goto error;
}
rec++;
if (fmt){
print_binrpc_val(&val, 0);
}else{
print_binrpc_val(&val, in_pkt->in_struct+in_pkt->in_array);
putchar('\n');
}
}
if (fmt && *f){
/* print the rest, with empty values */
while(*f){
s=f;
f=parse_fmt(f, &val.type, &f_size);
printf("%.*s", f_size, s);
}
}
return 0;
error:
return -1;
/*error_read_again:
fprintf(stderr, "ERROR: more data needed\n");
return -2;
*/
}
static int print_fault(struct binrpc_parse_ctx* in_pkt,
unsigned char* body, int size)
{
printf("error: ");
return print_body(in_pkt, body, size, "%v - %v\n");
}
static int run_binrpc_cmd(int s, struct binrpc_cmd * cmd, char* fmt)
{
int cookie;
unsigned char reply_buf[MAX_REPLY_SIZE];
unsigned char* msg_body;
struct binrpc_parse_ctx in_pkt;
int ret;
cookie=gen_cookie();
if ((ret=send_binrpc_cmd(s, cmd, cookie))<0){
if (ret==-1) goto error_send;
else goto binrpc_err;
}
/* read reply */
memset(&in_pkt, 0, sizeof(in_pkt));
if ((ret=get_reply(s, reply_buf, MAX_REPLY_SIZE, cookie, &in_pkt,
&msg_body))<0){
switch(ret){
case -1:
goto error_read;
case -2:
goto error_parse;
case -3:
goto error_cookie;
case -4:
goto error_toobig;
}
goto error;
}
switch(in_pkt.type){
case BINRPC_FAULT:
if (print_fault(&in_pkt, msg_body, in_pkt.tlen)<0){
goto error;
}
break;
case BINRPC_REPL:
if (print_body(&in_pkt, msg_body, in_pkt.tlen, fmt)<0){
goto error;
}
break;
default:
fprintf(stderr, "ERROR: not a reply\n");
goto error;
}
if (verbose) printf(".\n");
/* normal exit */
return 0;
binrpc_err:
fprintf(stderr, "ERROR while building the packet: %s\n",
binrpc_error(ret));
goto error;
error_parse:
fprintf(stderr, "ERROR while parsing the reply: %s\n",
binrpc_error(binrpc_errno));
goto error;
error_cookie:
fprintf(stderr, "ERROR: cookie does not match\n");
goto error;
error_toobig:
fprintf(stderr, "ERROR: reply too big\n");
goto error;
error_send:
fprintf(stderr, "ERROR: send packet failed: %s (%d)\n",
strerror(errno), errno);
goto error;
error_read:
fprintf(stderr, "ERROR: read reply failed: %s (%d)\n",
strerror(errno), errno);
goto error;
error:
return -1;
}
static int parse_line(struct binrpc_cmd* cmd, char* line)
{
char* p;
int count;
cmd->method=strtok(line, " \t");
if (cmd->method==0)
goto error_no_method;
count=0;
for(p=strtok(0, " \t"); p; p=strtok(0, " \t")){
if (count>=MAX_BINRPC_ARGS)
goto error_too_many;
if (parse_arg(&cmd->argv[count], p)<0){
goto error_arg;
}
count++;
}
cmd->argc=count;
return 0;
error_no_method:
printf( "ERROR: no method name\n");
return -1;
error_too_many:
printf("ERROR: too many arguments (%d), no more than %d allowed\n",
count, MAX_BINRPC_ARGS);
return -1;
error_arg:
printf("ERROR: bad argument %d: %s\n", count+1, p);
return -1;
}
/* resolves builtin aliases */
static void fix_cmd(struct binrpc_cmd* cmd, char** format)
{
int r;
for (r=0; cmd_aliases[r].name; r++){
if (strcmp(cmd_aliases[r].name, cmd->method)==0){
cmd->method=cmd_aliases[r].method;
if (*format==0)
*format=cmd_aliases[r].format;
break;
}
}
}
/* intercept builtin commands, returns 1 if intercepted, 0 if not, <0 on error
*/
static int run_builtins(int s, struct binrpc_cmd* cmd)
{
int r;
int ret;
for (r=0; builtins[r].name; r++){
if (strcmp(builtins[r].name, cmd->method)==0){
ret=builtins[r].f(s, cmd);
return (ret<0)?ret:1;
}
}
return 0;
}
/* runs command from cmd */
inline static int run_cmd(int s, struct binrpc_cmd* cmd, char* format)
{
int ret;
char* fmt;
fmt=format;
fix_cmd(cmd, &fmt);
if (!(ret=run_builtins(s, cmd))){
ret=run_binrpc_cmd(s, cmd, fmt);
}
return (ret>0)?0:ret;
}
/* runs a command represented in line */
inline static int run_line(int s, char* l, char* format)
{
struct binrpc_cmd cmd;
int ret;
if ((ret=parse_line(&cmd, l))==0){
return run_cmd(s, &cmd, format);
}
return ret;
}
static void free_rpc_array(struct binrpc_val* a, int size)
{
int r;
for (r=0; r<size; r++){
if (a[r].name.s)
free(a[r].name.s);
if ((a[r].type==BINRPC_T_STR || a[r].type==BINRPC_T_BYTES) &&
a[r].u.strval.s){
free(a[r].u.strval.s);
}
}
free(a);
}
/* parse the body into a malloc allocated, binrpc_val array */
static struct binrpc_val* parse_reply_body(int* records,
struct binrpc_parse_ctx* in_pkt,
unsigned char* body, int size)
{
struct binrpc_val* a;
struct binrpc_val* t;
unsigned char* p;
unsigned char* end;
struct binrpc_val val;
int ret;
int rec;
rec=0;
if (*records==0){
*records=100; /* start with a reasonable size */
};
a=malloc(*records*sizeof(struct binrpc_val));
if (a==0)
goto error_mem;
p=body;
end=p+size;
/* read body */
while(p<end){
val.type=BINRPC_T_ALL;
val.name.s=0;
val.name.len=0;
p=binrpc_read_record(in_pkt, p, end, &val, 1, &ret);
if (ret<0){
if (ret==E_BINRPC_EOP){
printf("end of message detected\n");
break;
}
fprintf(stderr, "ERROR: while parsing the record %d,"
" @%d: %02x : %s\n", rec,
in_pkt->offset, *p, binrpc_error(ret));
goto error;
}
if (rec>=*records){
t=realloc(a, *records*sizeof(struct binrpc_val)*2);
if (t==0)
goto error_mem;
a=t;
*records*=2;
}
a[rec]=val;
if (val.name.s){
if ((a[rec].name.s=malloc(val.name.len+1))==0)
goto error_mem;
memcpy(a[rec].name.s, val.name.s, val.name.len);
a[rec].name.s[val.name.len+1]=0; /* 0-term */
}
if (val.u.strval.s){
if (val.type==BINRPC_T_STR){
if ((a[rec].u.strval.s=malloc(val.u.strval.len+1))==0)
goto error_mem;
memcpy(a[rec].u.strval.s, val.u.strval.s, val.u.strval.len);
a[rec].u.strval.s[val.u.strval.len]=0; /* 0-term */
}else if (val.type==BINRPC_T_BYTES){
if ((a[rec].u.strval.s=malloc(val.u.strval.len))==0)
goto error_mem;
memcpy(a[rec].u.strval.s, val.u.strval.s, val.u.strval.len);
}
}
rec++;
}
if (rec && (rec<*records)){
a=realloc(a, rec*sizeof(struct binrpc_val));
}
*records=rec;
return a;
error_mem:
fprintf(stderr, "ERROR: parse_reply_body: out of memory\n");
error:
if (a){
free_rpc_array(a, rec);
}
*records=0;
return 0;
}
static int get_kamcmd_list(int s)
{
struct binrpc_cmd cmd;
int cookie;
unsigned char reply_buf[MAX_REPLY_SIZE];
unsigned char* msg_body;
struct binrpc_parse_ctx in_pkt;
int ret;
cmd.method="system.listMethods";
cmd.argc=0;
cookie=gen_cookie();
if ((ret=send_binrpc_cmd(s, &cmd, cookie))<0){
if (ret==-1) goto error_send;
else goto binrpc_err;
}
/* read reply */
memset(&in_pkt, 0, sizeof(in_pkt));
if ((ret=get_reply(s, reply_buf, MAX_REPLY_SIZE, cookie, &in_pkt,
&msg_body))<0){
goto error;
}
switch(in_pkt.type){
case BINRPC_FAULT:
if (print_fault(&in_pkt, msg_body, in_pkt.tlen)<0){
goto error;
}
break;
case BINRPC_REPL:
rpc_no=100; /* default cmd list */
if ((rpc_array=parse_reply_body(&rpc_no, &in_pkt, msg_body,
in_pkt.tlen))==0)
goto error;
break;
default:
fprintf(stderr, "ERROR: not a reply\n");
goto error;
}
return 0;
binrpc_err:
error_send:
error:
return -1;
}
#if defined(USE_CFG_VARS) || defined (USE_COUNTERS)
/** check if cmd is a rpc command.
* Quick check (using the internal rpc_array) if cmd is a valid rpc command.
* @param cmd - null terminated ascii string
* @return 1 on success, 0 on failure.
*/
static int is_rpc_cmd(char* cmd)
{
int r;
int cmd_len;
cmd_len=strlen(cmd);
for (r=0; r<rpc_no; r++){
if ((rpc_array[r].type==BINRPC_T_STR) &&
(rpc_array[r].u.strval.len==cmd_len) &&
(strncmp(cmd, rpc_array[r].u.strval.s, cmd_len)==0))
return 1;
}
return 0;
}
#endif /* USE_CFG_VARS */
#ifdef USE_CFG_VARS
/* retrieve the cfg vars and group list */
static int get_cfgvars_list(int s)
{
struct binrpc_cmd cmd;
int cookie;
unsigned char reply_buf[MAX_REPLY_SIZE];
unsigned char* msg_body;
struct binrpc_parse_ctx in_pkt;
struct cfg_var_grp* grp;
struct cfg_var_grp* last_grp;
char* p;
char* end;
str grp_name;
str var_name;
int r;
int ret;
cmd.method="cfg.list";
cmd.argc=0;
if (!is_rpc_cmd(cmd.method)) goto error;
cookie=gen_cookie();
if ((ret=send_binrpc_cmd(s, &cmd, cookie))<0){
if (ret==-1) goto error_send;
else goto binrpc_err;
}
/* read reply */
memset(&in_pkt, 0, sizeof(in_pkt));
if ((ret=get_reply(s, reply_buf, MAX_REPLY_SIZE, cookie, &in_pkt,
&msg_body))<0){
goto error;
}
switch(in_pkt.type){
case BINRPC_FAULT:
if (print_fault(&in_pkt, msg_body, in_pkt.tlen)<0){
goto error;
}
break;
case BINRPC_REPL:
cfg_vars_no=100; /* default cmd list */
if ((cfg_vars_array=parse_reply_body(&cfg_vars_no, &in_pkt,
msg_body, in_pkt.tlen))==0)
goto error;
break;
default:
fprintf(stderr, "ERROR: not a reply\n");
goto error;
}
/* get the config groups */
last_grp=0;
for (r=0; r<cfg_vars_no; r++){
grp_name.s=0; grp_name.len=0;
if (cfg_vars_array[r].type!=BINRPC_T_STR)
continue;
grp_name.s=cfg_vars_array[r].u.strval.s;
end=cfg_vars_array[r].u.strval.len+grp_name.s;
/* parse <grp>: <var_name>*/
for (p=grp_name.s; p<end; p++){
if (*p==':'){
grp_name.len=(int)(long)(p-grp_name.s);
break;
}
}
for (grp=cfg_grp_lst; grp; grp=grp->next){
if (grp->grp_name.len==grp_name.len &&
memcmp(grp->grp_name.s, grp_name.s, grp_name.len)==0){
break; /* found */
}
}
if (grp==0){
/* not found => create a new one */
grp=malloc(sizeof(*grp));
if (grp==0) goto error_mem;
memset(grp, 0, sizeof(*grp));
grp->grp_name=grp_name;
if (last_grp){
last_grp->next=grp;
last_grp=grp;
}else{
cfg_grp_lst=grp;
last_grp=cfg_grp_lst;
}
}
grp->var_no++;
}
/* alloc the var arrays per group */
for (grp=cfg_grp_lst; grp; grp=grp->next){
grp->var_names=malloc(sizeof(str)*grp->var_no);
if (grp->var_names==0) goto error_mem;
memset(grp->var_names, 0, sizeof(str)*grp->var_no);
grp->var_no=0;
}
/* reparse to get the var names per group */
for (r=0; r<cfg_vars_no; r++){
grp_name.s=0; grp_name.len=0;
var_name.s=0; var_name.len=0;
if (cfg_vars_array[r].type!=BINRPC_T_STR)
continue;
grp_name.s=cfg_vars_array[r].u.strval.s;
end=cfg_vars_array[r].u.strval.len+grp_name.s;
/* parse <grp>: <var_name>*/
for (p=grp_name.s; p<end; p++){
if (*p==':'){
grp_name.len=(int)(long)(p-grp_name.s);
p++;
for (; p<end && *p==' '; p++);
var_name.s=p;
var_name.len=(int)(long)(end-p);
if (var_name.len==0) break;
for (grp=cfg_grp_lst; grp; grp=grp->next){
if (grp->grp_name.len==grp_name.len &&
memcmp(grp->grp_name.s, grp_name.s, grp_name.len)==0){
/* add var */
grp->var_names[grp->var_no]=var_name;
grp->var_no++;
}
}
break;
}
}
}
return 0;
binrpc_err:
error_send:
error:
error_mem:
return -1;
}
void free_cfg_grp_lst()
{
struct cfg_var_grp* grp;
struct cfg_var_grp* last;
grp=cfg_grp_lst;
while(grp){
last=grp;
grp=grp->next;
free(last);
}
cfg_grp_lst=0;
}
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
/* retrieve the counters names and group list */
static int get_counters_list(int s)
{
struct binrpc_cmd cmd;
int cookie;
unsigned char reply_buf[MAX_REPLY_SIZE];
unsigned char* msg_body;
struct binrpc_parse_ctx in_pkt;
struct cnt_var_grp* grp;
struct cnt_var_grp* last_grp;
str grp_name;
str var_name;
int r;
int ret;
cmd.method="cnt.grps_list";
cmd.argc=0;
if (!is_rpc_cmd(cmd.method)) goto error;
cookie=gen_cookie();
if ((ret=send_binrpc_cmd(s, &cmd, cookie))<0){
if (ret==-1) goto error_send;
else goto binrpc_err;
}
/* read reply */
memset(&in_pkt, 0, sizeof(in_pkt));
if ((ret=get_reply(s, reply_buf, MAX_REPLY_SIZE, cookie, &in_pkt,
&msg_body))<0){
goto error;
}
switch(in_pkt.type){
case BINRPC_FAULT:
if (print_fault(&in_pkt, msg_body, in_pkt.tlen)<0){
goto error;
}
break;
case BINRPC_REPL:
cnt_grps_no=20; /* default counter list */
if ((cnt_grps_array=parse_reply_body(&cnt_grps_no, &in_pkt,
msg_body, in_pkt.tlen))==0)
goto error;
break;
default:
fprintf(stderr, "ERROR: not a reply\n");
goto error;
}
/* get the config groups */
last_grp=0;
for (r=0; r<cnt_grps_no; r++){
grp_name.s=0; grp_name.len=0;
if (cnt_grps_array[r].type!=BINRPC_T_STR)
continue;
grp_name=cnt_grps_array[r].u.strval;
/* check for duplicates */
for (grp=cnt_grp_lst; grp; grp=grp->next){
if (grp->grp_name.len==grp_name.len &&
memcmp(grp->grp_name.s, grp_name.s, grp_name.len)==0){
break; /* found */
}
}
if (grp==0){
/* not found => create a new one */
grp=malloc(sizeof(*grp));
if (grp==0) goto error_mem;
memset(grp, 0, sizeof(*grp));
grp->grp_name=grp_name;
if (last_grp){
last_grp->next=grp;
last_grp=grp;
}else{
cnt_grp_lst=grp;
last_grp=cnt_grp_lst;
}
}
}
/* gets vars per group */
for (grp=cnt_grp_lst; grp; grp=grp->next){
cmd.method="cnt.var_list";
cmd.argv[0].type=BINRPC_T_STR;
cmd.argv[0].u.strval=grp->grp_name;
cmd.argc=1;
if (!is_rpc_cmd(cmd.method)) goto error;
cookie=gen_cookie();
if ((ret=send_binrpc_cmd(s, &cmd, cookie))<0){
if (ret==-1) goto error_send;
else goto binrpc_err;
}
/* read reply */
memset(&in_pkt, 0, sizeof(in_pkt));
if ((ret=get_reply(s, reply_buf, MAX_REPLY_SIZE, cookie, &in_pkt,
&msg_body))<0){
goto error;
}
switch(in_pkt.type){
case BINRPC_FAULT:
if (print_fault(&in_pkt, msg_body, in_pkt.tlen)<0){
goto error;
}
break;
case BINRPC_REPL:
grp->cnt_vars_no=100; /* default counter list */
if ((grp->cnt_vars_array=parse_reply_body(&grp->cnt_vars_no,
&in_pkt, msg_body,
in_pkt.tlen))==0)
goto error;
break;
default:
fprintf(stderr, "ERROR: not a reply\n");
goto error;
}
grp->var_no = 0;
grp->var_names=malloc(sizeof(str)*grp->cnt_vars_no);
if (grp->var_names==0) goto error_mem;
memset(grp->var_names, 0, sizeof(str)*grp->cnt_vars_no);
for (r=0; r<grp->cnt_vars_no; r++) {
if (grp->cnt_vars_array[r].type!=BINRPC_T_STR)
continue;
var_name=grp->cnt_vars_array[r].u.strval;
grp->var_names[grp->var_no] = var_name;
grp->var_no++;
}
}
return 0;
binrpc_err:
error_send:
error:
error_mem:
return -1;
}
void free_cnt_grp_lst()
{
struct cnt_var_grp* grp;
struct cnt_var_grp* last;
grp=cnt_grp_lst;
while(grp){
last=grp;
grp=grp->next;
if (last->cnt_vars_array)
free_rpc_array(last->cnt_vars_array, last->cnt_vars_no);
free(last);
}
cnt_grp_lst=0;
}
#endif /* USE_COUNTERS */
static void print_formatting(char* prefix, char* format, char* suffix)
{
if (format){
printf("%s", prefix);
for (;*format;format++){
switch(*format){
case '\t':
printf("\\t");
break;
case '\n':
printf("\\n");
break;
case '\r':
printf("\\r");
break;
default:
putchar(*format);
}
}
printf("%s", suffix);
}
}
static int kamcmd_help(int s, struct binrpc_cmd* cmd)
{
int r;
if (cmd->argc && (cmd->argv[0].type==BINRPC_T_STR)){
/* if it has args, try command help */
for (r=0; cmd_aliases[r].name; r++){
if (strcmp(cmd->argv[0].u.strval.s, cmd_aliases[r].name)==0){
printf("%s is an alias for %s", cmd->argv[0].u.strval.s,
cmd_aliases[r].method);
print_formatting(" with reply formatting: \"",
cmd_aliases[r].format, "\"");
putchar('\n');
return 0;
}
}
for(r=0; builtins[r].name; r++){
if (strcmp(cmd->argv[0].u.strval.s, builtins[r].name)==0){
printf("builtin command: %s\n",
builtins[r].doc?builtins[r].doc:"undocumented");
return 0;
}
}
cmd->method="system.methodHelp";
if (run_binrpc_cmd(s, cmd, 0)<0){
printf("error: no such command %s\n", cmd->argv[0].u.strval.s);
}
return 0;
}
if (rpc_no==0){
if (get_kamcmd_list(s)<0)
goto error;
}
for (r=0; r<rpc_no; r++){
if (rpc_array[r].type==BINRPC_T_STR){
printf("%s\n", rpc_array[r].u.strval.s);
}
}
for (r=0; cmd_aliases[r].name; r++){
printf("alias: %s\n", cmd_aliases[r].name);
}
for(r=0; builtins[r].name; r++){
printf("builtin: %s\n", builtins[r].name);
}
return 0;
error:
return -1;
}
static int kamcmd_ver(int s, struct binrpc_cmd* cmd)
{
printf("%s\n", version);
printf("%s compiled on %s \n", __FILE__, compiled);
#ifdef USE_READLINE
printf("interactive mode command completion support\n");
#endif
return 0;
}
static int kamcmd_quit(int s, struct binrpc_cmd* cmd)
{
quit=1;
return 0;
}
static int kamcmd_warranty(int s, struct binrpc_cmd *cmd)
{
printf("%s %s\n", NAME, VERSION);
printf("%s\n", COPYRIGHT);
printf("\n%s\n", LICENSE);
return 0;
}
#ifdef USE_READLINE
/* readline command generator */
static char* kamcmd_generator(const char* text, int state)
{
static int idx;
static int list; /* aliases, builtins, rpc_array */
static int len;
char* name;
#ifdef USE_CFG_VARS
static struct cfg_var_grp* grp;
#endif
#ifdef USE_COUNTERS
static struct cnt_var_grp* cnt_grp;
#endif
switch(attempted_completion_state){
case COMPLETE_INIT:
case COMPLETE_NOTHING:
return 0;
case COMPLETE_CMD_NAME:
if (state==0){
/* init */
idx=list=0;
len=strlen(text);
}
/* return next partial match */
switch(list){
case 0: /* aliases*/
while((name=cmd_aliases[idx].name)){
idx++;
if (strncmp(name, text, len)==0)
return strdup(name);
}
list++;
idx=0;
/* no break */
case 1: /* builtins */
while((name=builtins[idx].name)){
idx++;
if (strncmp(name, text, len)==0)
return strdup(name);
}
list++;
idx=0;
/* no break */
case 2: /* rpc_array */
while(idx < rpc_no){
if (rpc_array[idx].type==BINRPC_T_STR){
name=rpc_array[idx].u.strval.s;
idx++;
if (strncmp(name, text, len)==0)
return strdup(name);
}else{
idx++;
}
}
}
break;
#ifdef USE_CFG_VARS
case COMPLETE_CFG_GRP:
if (state==0){
/* init */
len=strlen(text);
grp=cfg_grp_lst;
}else{
grp=grp->next;
}
for(;grp; grp=grp->next){
if (len<=grp->grp_name.len &&
memcmp(text, grp->grp_name.s, len)==0) {
/* zero-term copy of the grp name */
name=malloc(grp->grp_name.len+1);
if (name){
memcpy(name, grp->grp_name.s, grp->grp_name.len);
name[grp->grp_name.len]=0;
}
return name;
}
}
break;
case COMPLETE_CFG_VAR:
if (state==0){
/* init */
len=strlen(text);
idx=0;
}
while(idx < crt_cfg_grp->var_no){
if (len<=crt_cfg_grp->var_names[idx].len &&
memcmp(text, crt_cfg_grp->var_names[idx].s, len)==0) {
/* zero-term copy of the var name */
name=malloc(crt_cfg_grp->var_names[idx].len+1);
if (name){
memcpy(name, crt_cfg_grp->var_names[idx].s,
crt_cfg_grp->var_names[idx].len);
name[crt_cfg_grp->var_names[idx].len]=0;
}
idx++;
return name;
}
idx++;
}
break;
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
case COMPLETE_CNT_GRP:
if (state==0){
/* init */
len=strlen(text);
cnt_grp=cnt_grp_lst;
}else{
cnt_grp=cnt_grp->next;
}
for(;cnt_grp; cnt_grp=cnt_grp->next){
if (len<=cnt_grp->grp_name.len &&
memcmp(text, cnt_grp->grp_name.s, len)==0) {
/* zero-term copy of the cnt_grp name */
name=malloc(cnt_grp->grp_name.len+1);
if (name){
memcpy(name, cnt_grp->grp_name.s,
cnt_grp->grp_name.len);
name[cnt_grp->grp_name.len]=0;
}
return name;
}
}
break;
case COMPLETE_CNT_VAR:
if (state==0){
/* init */
len=strlen(text);
idx=0;
}
while(idx < crt_cnt_grp->var_no){
if (len<=crt_cnt_grp->var_names[idx].len &&
memcmp(text, crt_cnt_grp->var_names[idx].s, len)==0) {
/* zero-term copy of the var name */
name=malloc(crt_cnt_grp->var_names[idx].len+1);
if (name){
memcpy(name, crt_cnt_grp->var_names[idx].s,
crt_cnt_grp->var_names[idx].len);
name[crt_cnt_grp->var_names[idx].len]=0;
}
idx++;
return name;
}
idx++;
}
break;
#endif /* USE_COUNTERS */
}
/* no matches */
return 0;
}
char** kamcmd_completion(const char* text, int start, int end)
{
int i, j;
int cmd_start, cmd_end, cmd_len;
int whitespace;
#ifdef USE_CFG_VARS
struct cfg_var_grp* grp;
static int grp_start;
int grp_len;
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
struct cnt_var_grp* cnt_grp;
static int cnt_grp_start;
int cnt_grp_len;
#endif /* USE_COUNTERS */
crt_param_no=0;
/* skip over whitespace at the beginning */
for (j=0; (j<start) && (rl_line_buffer[j]==' ' ||
rl_line_buffer[j]=='\t'); j++);
cmd_start=j;
if (start==cmd_start){
/* complete cmd name at beginning */
attempted_completion_state=COMPLETE_CMD_NAME;
#ifdef USE_CFG_VARS
grp_start=0;
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
cnt_grp_start=0;
#endif /* USE_COUNTERS */
}else{ /* or if this is a command for which we complete the parameters */
/* find first whitespace after command name*/
for(; (j<start) && (rl_line_buffer[j]!=' ') &&
(rl_line_buffer[j]!='\t'); j++);
cmd_end=j;
cmd_len=cmd_end-cmd_start;
/* count params before the current one */
whitespace=1;
for (; j<start; j++){
if (rl_line_buffer[j]!=' ' && rl_line_buffer[j]!='\t'){
if (whitespace) crt_param_no++;
whitespace=0;
}else
whitespace=1;
}
crt_param_no++;
if (crt_param_no==1){
for(i=0; complete_params_methods[i]; i++){
if ((cmd_len==strlen(complete_params_methods[i])) &&
(strncmp(&rl_line_buffer[cmd_start],
complete_params_methods[i],
cmd_len)==0)){
attempted_completion_state=COMPLETE_CMD_NAME;
goto end;
}
}
#ifdef USE_CFG_VARS
/* try complete_param*_cfg_grp */
for(i=0; complete_params_cfg_var[i]; i++){
if ((cmd_len==strlen(complete_params_cfg_var[i])) &&
(strncmp(&rl_line_buffer[cmd_start],
complete_params_cfg_var[i],
cmd_len)==0)){
attempted_completion_state=COMPLETE_CFG_GRP;
grp_start=start;
goto end;
}
}
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
/* try complete_param*_cfg_grp */
for(i=0; complete_param1_counter_grp[i]; i++){
if ((cmd_len==strlen(complete_param1_counter_grp[i])) &&
(strncmp(&rl_line_buffer[cmd_start],
complete_param1_counter_grp[i],
cmd_len)==0)){
attempted_completion_state=COMPLETE_CNT_GRP;
cnt_grp_start=start;
goto end;
}
}
#endif /* USE_COUNTERS */
}else if (crt_param_no==2){
#ifdef USE_CFG_VARS
/* see if we complete cfg. var names for this command */
for(i=0; complete_params_cfg_var[i]; i++){
if ((cmd_len==strlen(complete_params_cfg_var[i])) &&
(strncmp(&rl_line_buffer[cmd_start],
complete_params_cfg_var[i],
cmd_len)==0)){
/* get the group name: */
/* find grp_start */
for(j=cmd_end; (j<start) && ((rl_line_buffer[j]==' ') ||
(rl_line_buffer[j]=='\t')); j++);
grp_start=j;
/* find group end / grp_len*/
for(j=grp_start; (j<start) && (rl_line_buffer[j]!=' ') &&
(rl_line_buffer[j]!='\t'); j++);
grp_len=j-grp_start;
for(grp=cfg_grp_lst; grp; grp=grp->next){
if (grp_len==grp->grp_name.len &&
memcmp(&rl_line_buffer[grp_start],
grp->grp_name.s, grp_len)==0) {
attempted_completion_state=COMPLETE_CFG_VAR;
crt_cfg_grp=grp;
goto end;
}
}
}
}
#endif /* USE_CFG_VARS */
#ifdef USE_COUNTERS
/* see if we complete counter names for this command */
for(i=0; complete_param2_counter_name[i]; i++){
if ((cmd_len==strlen(complete_param2_counter_name[i])) &&
(strncmp(&rl_line_buffer[cmd_start],
complete_param2_counter_name[i],
cmd_len)==0)){
/* get the group name: */
/* find grp_start */
for(j=cmd_end; (j<start) && ((rl_line_buffer[j]==' ') ||
(rl_line_buffer[j]=='\t')); j++);
cnt_grp_start=j;
/* find group end / cnt_grp_len*/
for(j=cnt_grp_start; (j<start) &&
(rl_line_buffer[j]!=' ') &&
(rl_line_buffer[j]!='\t'); j++);
cnt_grp_len=j-cnt_grp_start;
for(cnt_grp=cnt_grp_lst; cnt_grp; cnt_grp=cnt_grp->next){
if (cnt_grp_len==cnt_grp->grp_name.len &&
memcmp(&rl_line_buffer[cnt_grp_start],
cnt_grp->grp_name.s, cnt_grp_len)==0) {
attempted_completion_state=COMPLETE_CNT_VAR;
crt_cnt_grp=cnt_grp;
goto end;
}
}
}
}
#endif /* COUNTERS */
}
attempted_completion_state=COMPLETE_NOTHING;
}
end:
return 0; /* let readline call kamcmd_generator */
}
#endif /* USE_READLINE */
/* on exit cleanup */
static void cleanup()
{
if (unix_socket){
if (unlink(unix_socket)<0){
fprintf(stderr, "ERROR: failed to delete %s: %s\n",
unix_socket, strerror(errno));
}
}
}
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);
}