12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222 |
- /*
- * 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:
- * [email protected]
- *
- * 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 */
- #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 16384 /* for non readline mode */
- #define MAX_REPLY_SIZE 65536
- #define MAX_BODY_SIZE 65536
- #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:2048 , 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:
- 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);
- }
|