-/* ------------------------------------------------\r
- Copyright 2014 AT&T Intellectual Property\r
- Licensed under the Apache License, Version 2.0 (the "License");\r
- you may not use this file except in compliance with the License.\r
- You may obtain a copy of the License at\r
- \r
- http://www.apache.org/licenses/LICENSE-2.0\r
- \r
- Unless required by applicable law or agreed to in writing, software\r
- distributed under the License is distributed on an "AS IS" BASIS,\r
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
- See the License for the specific language governing permissions and\r
- limitations under the License.\r
- ------------------------------------------- */\r
-#include <app.h>\r
-#include <stdlib.h>\r
-#include <stdio.h>\r
-#include <unistd.h>\r
-#include <signal.h>\r
-#include <time.h>\r
-#include <string.h>\r
-#include <sys/time.h>\r
-#include <sys/stat.h>\r
-#include <sys/types.h>\r
-#include <sys/socket.h>\r
-#include <netinet/in.h>\r
-\r
-\r
-#include "gsconfig.h"\r
-#include "gstypes.h"\r
-#include "gshub.h"\r
-\r
-#include <schemaparser.h>\r
-\r
-#define MAXLINE 100000\r
-static unsigned tcpport=0;\r
-static char linebuf[MAXLINE];\r
-int listensockfd=0;\r
-int fd=0;\r
-\r
-\r
-// Not all systems have timersub defined so make sure its ther\r
-#ifndef timersub\r
-\r
-#define timersub(tvp, uvp, vvp) \\r
-do { \\r
-(vvp)->tv_sec = (tvp)->tv_sec - (uvp)->tv_sec; \\r
-(vvp)->tv_usec = (tvp)->tv_usec - (uvp)->tv_usec; \\r
-if ((vvp)->tv_usec < 0) { \\r
-(vvp)->tv_sec--; \\r
-(vvp)->tv_usec += 1000000; \\r
-} \\r
-} while (0)\r
-\r
-#endif\r
-\r
-void hand(int iv) {\r
- ftaapp_exit();\r
- fprintf(stderr, "exiting via signal handler %d...\n", iv);\r
- exit(1);\r
-}\r
-\r
-static void wait_for_client() {\r
- struct sockaddr_in serv_addr,cli_addr;\r
- socklen_t clilen;\r
- if (listensockfd==0) {\r
- gs_int32_t on = 1;\r
- listensockfd=socket(AF_INET, SOCK_STREAM, 0);\r
- if (listensockfd < 0) {\r
- gslog(LOG_EMERG,"Error:Could not create socket for tcp data stream");\r
- exit(1);\r
- }\r
- bzero((char *) &serv_addr, sizeof(serv_addr));\r
- serv_addr.sin_family = AF_INET;\r
- serv_addr.sin_addr.s_addr = INADDR_ANY;\r
- serv_addr.sin_port = htons(tcpport);\r
-#ifndef __linux__\r
- /* make sure we can reuse the common port rapidly */\r
- if (setsockopt(listensockfd, SOL_SOCKET, SO_REUSEPORT,\r
- (gs_sp_t )&on, sizeof(on)) != 0) {\r
- gslog(LOG_EMERG,"Error::could not set socket option\n");\r
- exit(1);\r
- }\r
-#endif\r
- if (setsockopt(listensockfd, SOL_SOCKET, SO_REUSEADDR,\r
- (gs_sp_t )&on, sizeof(on)) != 0) {\r
- gslog(LOG_EMERG,"Error::could not set socket option\n");\r
- exit(1);\r
- }\r
- \r
- if (bind(listensockfd, (struct sockaddr *) &serv_addr,\r
- sizeof(serv_addr)) < 0) {\r
- gslog(LOG_EMERG,"Error:Could not bind socket for tcp data stream");\r
- exit(1);\r
- }\r
- }\r
- \r
- do {\r
- listen(listensockfd,5);\r
- clilen = sizeof(cli_addr);\r
- fd=accept(listensockfd, (struct sockaddr *) &cli_addr, &clilen);\r
- if (fd<0) {\r
- gslog(LOG_EMERG,"Error:Could not accept connection on tcp socket");\r
- }\r
- } while (fd==0);\r
-}\r
-\r
-\r
-static void emit_socket() {\r
- unsigned o,w,l;\r
- o=0;\r
- w=0;\r
- l=strlen(linebuf);\r
- do {\r
- if((w=write(fd,&linebuf[o],l))==0) {\r
- close(fd);\r
- wait_for_client();\r
- }\r
- o=o+w;\r
- } while (o<l);\r
-}\r
-\r
-static void emit_line() {\r
- \r
- if (tcpport==0) {\r
- printf("%s",linebuf);\r
- } else {\r
- emit_socket();\r
- }\r
- \r
-}\r
-\r
-int main(int argc, char* argv[]) {\r
- gs_sp_t me = argv[0];\r
- FTAID fta_id;\r
- gs_int32_t schema, ch;\r
- \r
- FTAID rfta_id;\r
- gs_uint32_t rsize;\r
- gs_uint32_t bufsz=8*1024*1024;\r
- gs_int8_t rbuf[2*MAXTUPLESZ];\r
- \r
- gs_int32_t numberoffields;\r
- gs_int32_t verbose=0;\r
- gs_int32_t y, lcv;\r
- \r
- void *pblk;\r
- gs_int32_t pblklen;\r
- gs_int32_t n_actual_param;\r
- gs_int32_t n_expected_param;\r
- gs_int32_t xit = 0;\r
- gs_int32_t dump = 0;\r
- struct timeval tvs, tve, tvd;\r
- gs_retval_t code;\r
- endpoint gshub;\r
- endpoint dummyep;\r
- gs_uint32_t tip1,tip2,tip3,tip4;\r
- gs_sp_t instance_name;\r
-\r
- gs_uint32_t tlimit = 0; // time limit in seconds\r
- time_t start_time, curr_time;\r
- \r
- gsopenlog(argv[0]);\r
- \r
- \r
- while ((ch = getopt(argc, argv, "l:p:r:vXD")) != -1) {\r
- switch (ch) {\r
- case 'r':\r
- bufsz=atoi(optarg);\r
- break;\r
- case 'p':\r
- tcpport=atoi(optarg);\r
- break;\r
- case 'v':\r
- verbose++;\r
- break;\r
- case 'X':\r
- xit++;\r
- break;\r
- case 'D':\r
- dump++;\r
- break;\r
- case 'l':\r
- tlimit = atoi(optarg);\r
- break;\r
- default:\r
- usage:\r
- fprintf(stderr, "usage: %s [-r <bufsz>] [-p <port>] [-l <time_limit>] [-v] [-X] [-D] <gshub-hostname>:<gshub-port> <gsinstance_name> query param1 param2...\n",\r
- *argv);\r
- exit(1);\r
- }\r
- }\r
- argc -= optind;\r
- argv += optind;\r
- if (argc<3) goto usage;\r
- \r
- if (sscanf(argv[0],"%u.%u.%u.%u:%hu",&tip1,&tip2,&tip3,&tip4,&(gshub.port))!= 5 ) {\r
- gslog(LOG_EMERG,"HUB IP NOT DEFINED");\r
- exit(1);\r
- }\r
- gshub.ip=htonl(tip1<<24|tip2<<16|tip3<<8|tip4);\r
- gshub.port=htons(gshub.port);\r
- instance_name=strdup(argv[1]);\r
- if (set_hub(gshub)!=0) {\r
- gslog(LOG_EMERG,"Could not set hub");\r
- exit(1);\r
- }\r
- if (set_instance_name(instance_name)!=0) {\r
- gslog(LOG_EMERG,"Could not set instance name");\r
- exit(1);\r
- }\r
- \r
- if (get_initinstance(gshub,instance_name,&dummyep,1)!=0) {\r
- gslog(LOG_EMERG,"Did not receive signal that GS is initiated\n");\r
- }\r
- \r
- gettimeofday(&tvs, 0);\r
- argc -=2;\r
- argv +=2;\r
- if (argc < 1)\r
- goto usage;\r
- \r
- /* initialize host library and the sgroup */\r
- \r
- if (verbose>=2) fprintf(stderr,"Inializin gscp\n");\r
- \r
- if (ftaapp_init(bufsz)!=0) {\r
- fprintf(stderr,"%s::error:could not initialize gscp\n", me);\r
- exit(1);\r
- }\r
- \r
- signal(SIGTERM, hand);\r
- signal(SIGINT, hand);\r
- \r
- schema = ftaapp_get_fta_schema_by_name(argv[0]);\r
- if (schema < 0) {\r
- fprintf(stderr,"%s::error:could not get fta '%s' schema\n",\r
- me ,argv[0]);\r
- exit(1);\r
- }\r
- n_expected_param = ftaschema_parameter_len(schema);\r
- if (n_expected_param == 0) {\r
- pblk = 0;\r
- pblklen = 0;\r
- } else {\r
- /* parse the params */\r
- n_actual_param = argc-1;\r
- if(n_actual_param < n_expected_param){\r
- fprintf(stderr,"Error, %d query parameters expected, %d provided.\n",n_expected_param, n_actual_param);\r
- exit(1);\r
- }\r
- for (lcv = 1 ; lcv < argc ; lcv++) {\r
- char *k, *e;\r
- int rv;\r
- k = argv[lcv];\r
- e = k;\r
- while (*e && *e != '=') e++;\r
- if (*e == 0) {\r
- fprintf(stderr,"param parse error '%s' (fmt 'key=val')\n",\r
- argv[lcv]);\r
- exit(1);\r
- }\r
- *e = 0;\r
- rv = ftaschema_setparam_by_name(schema, k, e+1, strlen(e+1));\r
- *e = '=';\r
- if (rv < 0) {\r
- fprintf(stderr,"param setparam error '%s' (fmt 'key=val')\n",\r
- argv[lcv]);\r
- exit(1);\r
- }\r
- }\r
- if (ftaschema_create_param_block(schema, &pblk, &pblklen) < 0) {\r
- fprintf(stderr, "ftaschema_create_param_block failed!\n");\r
- exit(1);\r
- }\r
- }\r
- ftaschema_free(schema); /* XXXCDC */\r
- \r
- \r
- if (verbose>=2) fprintf(stderr,"Initalice FTA\n");\r
- \r
- fta_id=ftaapp_add_fta(argv[0],0,0,0,pblklen,pblk);\r
- if (fta_id.streamid==0) {\r
- fprintf(stderr,"%s::error:could not initialize fta %s\n",\r
- me, argv[0]);\r
- exit(1);\r
- }\r
- /* XXXCDC: pblk is malloc'd, should we free it? */\r
- \r
- if (verbose>=2) fprintf(stderr,"Get schema handle\n");\r
- \r
- if ((schema=ftaapp_get_fta_schema(fta_id))<0) {\r
- fprintf(stderr,"%s::error:could not get schema\n", me);\r
- exit(1);\r
- }\r
- \r
- if ((numberoffields=ftaschema_tuple_len(schema))<0) {\r
- fprintf(stderr,"%s::error:could not get number of fields in schema\n",\r
- me);\r
- exit(1);\r
- }\r
- \r
- if (verbose>=1) {\r
- for(y=0; y<numberoffields;y++) {\r
- printf("%s",ftaschema_field_name(schema,y));\r
- if (y<numberoffields-1) printf("|");\r
- }\r
- printf("\n");\r
- }\r
- if (xit) {\r
- gettimeofday(&tve, 0);\r
- timersub(&tve, &tvs, &tvd);\r
- printf("TIME= %ld.%06d sec\n", tvd.tv_sec, tvd.tv_usec);\r
- hand(0);\r
- }\r
- if (tcpport!=0) {\r
- wait_for_client();\r
- }\r
-\r
- start_time = time(NULL);\r
-\r
- while((code=ftaapp_get_tuple(&rfta_id,&rsize,rbuf,2*MAXTUPLESZ,0))>=0) {\r
- if (dump)\r
- continue;\r
- if (ftaschema_is_eof_tuple(schema, rbuf)) {\r
- /* initiate shutdown or something of that nature */\r
- printf("#All data proccessed\n");\r
- exit(0);\r
- }\r
- if (!rsize)\r
- continue;\r
- if (verbose >=2) {\r
- snprintf(linebuf,MAXLINE,"RESULT CODE => %u\n",code);\r
- emit_line();\r
- }\r
- if ((code==0)&&(rfta_id.streamid == fta_id.streamid)) {\r
- for(y=0; y<numberoffields;y++) {\r
- struct access_result ar;\r
- if (verbose>=2)\r
- printf("%s->",ftaschema_field_name(schema,y));\r
- ar=ftaschema_get_field_by_index(schema,y,rbuf,rsize);\r
- switch (ar.field_data_type) {\r
- case INT_TYPE:\r
- snprintf(linebuf,MAXLINE,"%d",ar.r.i);\r
- break;\r
- case UINT_TYPE:\r
- snprintf(linebuf,MAXLINE,"%u",ar.r.ui);\r
- break;\r
- case IP_TYPE:\r
- snprintf(linebuf,MAXLINE,"%u.%u.%u.%u",ar.r.ui>>24&0xff,\r
- ar.r.ui>>16&0xff,\r
- ar.r.ui>>8&0xff,\r
- ar.r.ui&0xff);\r
- break;\r
- case IPV6_TYPE:\r
- {\r
- unsigned x;\r
- unsigned zc=0;\r
- for(x=0;x<4;x++) { if (ar.r.ip6.v[x]==0) zc++;}\r
- if (zc!=4) {\r
- snprintf(linebuf,MAXLINE,"");\r
- for(x=0;x<8;x++) {\r
- unsigned char * a = (unsigned char *) &(ar.r.ip6.v[0]);\r
- unsigned y;\r
- y=((unsigned)a[2*x])<<8|((unsigned) a[2*x+1]);\r
- snprintf(&linebuf[strlen(linebuf)],MAXLINE,"%04x",y);\r
- if (x<7) snprintf(&linebuf[strlen(linebuf)],MAXLINE,":");\r
- }\r
- } else {\r
- snprintf(linebuf,MAXLINE,"::");\r
- }\r
- }\r
- break;\r
- \r
- case USHORT_TYPE:\r
- snprintf(linebuf,MAXLINE,"%u",ar.r.ui);\r
- break;\r
- case BOOL_TYPE:\r
- if (ar.r.ui==0) {\r
- snprintf(linebuf,MAXLINE,"FALSE");\r
- } else {\r
- snprintf(linebuf,MAXLINE,"TRUE");\r
- }\r
- break;\r
- case ULLONG_TYPE:\r
- snprintf(linebuf,MAXLINE,"%llu",ar.r.ul);\r
- break;\r
- case LLONG_TYPE:\r
- snprintf(linebuf,MAXLINE,"%lld",ar.r.l);\r
- break;\r
- case FLOAT_TYPE:\r
- snprintf(linebuf,MAXLINE,"%f",ar.r.f);\r
- break;\r
- case TIMEVAL_TYPE:\r
- {\r
- gs_float_t t;\r
- t= ar.r.t.tv_usec;\r
- t=t/1000000;\r
- t=t+ar.r.t.tv_sec;\r
- snprintf(linebuf,MAXLINE,"%f sec",t);\r
- }\r
- break;\r
- case VSTR_TYPE:\r
- {\r
- int x;\r
- int c;\r
- int d=0;\r
- char * src;\r
- src=(char*)ar.r.vs.offset;\r
- if(d<MAXLINE){\r
- linebuf[d] = '\0';\r
- }\r
- for(x=0;x<ar.r.vs.length;x++) {\r
- c=src[x];\r
- if ((c<='~') && (c>=' ')) {\r
- if (d<MAXLINE-1) {\r
- linebuf[d]=c;\r
- linebuf[d+1]=0;\r
- d++;\r
- }\r
- } else {\r
- if (d<MAXLINE-1) {\r
- linebuf[d]='.';\r
- linebuf[d+1]=0;\r
- d++;\r
- }\r
- }\r
- }\r
- }\r
- break;\r
- default:\r
- linebuf[0]=0;\r
- break;\r
- }\r
- if (y<numberoffields-1) snprintf(&linebuf[strlen(linebuf)],MAXLINE,"|");\r
- emit_line();\r
- }\r
- snprintf(linebuf,MAXLINE,"\n");\r
- emit_line();\r
- if (verbose!=0) fflush(stdout);\r
- } else {\r
- if (rfta_id.streamid != fta_id.streamid)\r
- fprintf(stderr,"Got unkown streamid %llu \n",rfta_id.streamid);\r
- }\r
-\r
- // whenever we receive a temp tuple check if we reached time limit\r
- if ((code==2) && tlimit && (time(NULL)-start_time)>=tlimit) {\r
- fprintf(stderr,"Reached time limit of %d seconds\n",tlimit); \r
- ftaapp_exit();\r
- exit(0);\r
- } \r
- }\r
-}\r
-\r
+/* ------------------------------------------------
+ Copyright 2014 AT&T Intellectual Property
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ ------------------------------------------- */
+#include <app.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <signal.h>
+#include <time.h>
+#include <string.h>
+#include <sys/time.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+
+
+#include "gsconfig.h"
+#include "gstypes.h"
+#include "gshub.h"
+
+#include <schemaparser.h>
+
+#define MAXLINE 100000
+static unsigned tcpport=0;
+static char linebuf[MAXLINE];
+int listensockfd=0;
+int fd=0;
+
+
+// Not all systems have timersub defined so make sure its ther
+#ifndef timersub
+
+#define timersub(tvp, uvp, vvp) \
+do { \
+(vvp)->tv_sec = (tvp)->tv_sec - (uvp)->tv_sec; \
+(vvp)->tv_usec = (tvp)->tv_usec - (uvp)->tv_usec; \
+if ((vvp)->tv_usec < 0) { \
+(vvp)->tv_sec--; \
+(vvp)->tv_usec += 1000000; \
+} \
+} while (0)
+
+#endif
+
+void hand(int iv) {
+ ftaapp_exit();
+ fprintf(stderr, "exiting via signal handler %d...\n", iv);
+ exit(1);
+}
+
+static void wait_for_client() {
+ struct sockaddr_in serv_addr,cli_addr;
+ socklen_t clilen;
+ if (listensockfd==0) {
+ gs_int32_t on = 1;
+ listensockfd=socket(AF_INET, SOCK_STREAM, 0);
+ if (listensockfd < 0) {
+ gslog(LOG_EMERG,"Error:Could not create socket for tcp data stream");
+ exit(1);
+ }
+ bzero((char *) &serv_addr, sizeof(serv_addr));
+ serv_addr.sin_family = AF_INET;
+ serv_addr.sin_addr.s_addr = INADDR_ANY;
+ serv_addr.sin_port = htons(tcpport);
+#ifndef __linux__
+ /* make sure we can reuse the common port rapidly */
+ if (setsockopt(listensockfd, SOL_SOCKET, SO_REUSEPORT,
+ (gs_sp_t )&on, sizeof(on)) != 0) {
+ gslog(LOG_EMERG,"Error::could not set socket option\n");
+ exit(1);
+ }
+#endif
+ if (setsockopt(listensockfd, SOL_SOCKET, SO_REUSEADDR,
+ (gs_sp_t )&on, sizeof(on)) != 0) {
+ gslog(LOG_EMERG,"Error::could not set socket option\n");
+ exit(1);
+ }
+
+ if (bind(listensockfd, (struct sockaddr *) &serv_addr,
+ sizeof(serv_addr)) < 0) {
+ gslog(LOG_EMERG,"Error:Could not bind socket for tcp data stream");
+ exit(1);
+ }
+ }
+
+ do {
+ listen(listensockfd,5);
+ clilen = sizeof(cli_addr);
+ fd=accept(listensockfd, (struct sockaddr *) &cli_addr, &clilen);
+ if (fd<0) {
+ gslog(LOG_EMERG,"Error:Could not accept connection on tcp socket");
+ }
+ } while (fd==0);
+}
+
+
+static void emit_socket() {
+ unsigned o,w,l;
+ o=0;
+ w=0;
+ l=strlen(linebuf);
+ do {
+ if((w=write(fd,&linebuf[o],l))==0) {
+ close(fd);
+ wait_for_client();
+ }
+ o=o+w;
+ } while (o<l);
+}
+
+static void emit_line() {
+
+ if (tcpport==0) {
+ printf("%s",linebuf);
+ } else {
+ emit_socket();
+ }
+
+}
+
+int main(int argc, char* argv[]) {
+ gs_sp_t me = argv[0];
+ FTAID fta_id;
+ gs_int32_t schema, ch;
+
+ FTAID rfta_id;
+ gs_uint32_t rsize;
+ gs_uint32_t bufsz=8*1024*1024;
+ gs_int8_t rbuf[2*MAXTUPLESZ];
+
+ gs_int32_t numberoffields;
+ gs_int32_t verbose=0;
+ gs_int32_t y, lcv;
+
+ void *pblk;
+ gs_int32_t pblklen;
+ gs_int32_t n_actual_param;
+ gs_int32_t n_expected_param;
+ gs_int32_t xit = 0;
+ gs_int32_t dump = 0;
+ struct timeval tvs, tve, tvd;
+ gs_retval_t code;
+ endpoint gshub;
+ endpoint dummyep;
+ gs_uint32_t tip1,tip2,tip3,tip4;
+ gs_sp_t instance_name;
+
+ gs_uint32_t tlimit = 0; // time limit in seconds
+ time_t start_time, curr_time;
+
+ gsopenlog(argv[0]);
+
+
+ while ((ch = getopt(argc, argv, "l:p:r:vXD")) != -1) {
+ switch (ch) {
+ case 'r':
+ bufsz=atoi(optarg);
+ break;
+ case 'p':
+ tcpport=atoi(optarg);
+ break;
+ case 'v':
+ verbose++;
+ break;
+ case 'X':
+ xit++;
+ break;
+ case 'D':
+ dump++;
+ break;
+ case 'l':
+ tlimit = atoi(optarg);
+ break;
+ default:
+ usage:
+ fprintf(stderr, "usage: %s [-r <bufsz>] [-p <port>] [-l <time_limit>] [-v] [-X] [-D] <gshub-hostname>:<gshub-port> <gsinstance_name> query param1 param2...\n",
+ *argv);
+ exit(1);
+ }
+ }
+ argc -= optind;
+ argv += optind;
+ if (argc<3) goto usage;
+
+ if (sscanf(argv[0],"%u.%u.%u.%u:%hu",&tip1,&tip2,&tip3,&tip4,&(gshub.port))!= 5 ) {
+ gslog(LOG_EMERG,"HUB IP NOT DEFINED");
+ exit(1);
+ }
+ gshub.ip=htonl(tip1<<24|tip2<<16|tip3<<8|tip4);
+ gshub.port=htons(gshub.port);
+ instance_name=strdup(argv[1]);
+ if (set_hub(gshub)!=0) {
+ gslog(LOG_EMERG,"Could not set hub");
+ exit(1);
+ }
+ if (set_instance_name(instance_name)!=0) {
+ gslog(LOG_EMERG,"Could not set instance name");
+ exit(1);
+ }
+
+ if (get_initinstance(gshub,instance_name,&dummyep,1)!=0) {
+ gslog(LOG_EMERG,"Did not receive signal that GS is initiated\n");
+ }
+
+ gettimeofday(&tvs, 0);
+ argc -=2;
+ argv +=2;
+ if (argc < 1)
+ goto usage;
+
+ /* initialize host library and the sgroup */
+
+ if (verbose>=2) fprintf(stderr,"Inializin gscp\n");
+
+ if (ftaapp_init(bufsz)!=0) {
+ fprintf(stderr,"%s::error:could not initialize gscp\n", me);
+ exit(1);
+ }
+
+ signal(SIGTERM, hand);
+ signal(SIGINT, hand);
+
+ schema = ftaapp_get_fta_schema_by_name(argv[0]);
+ if (schema < 0) {
+ fprintf(stderr,"%s::error:could not get fta '%s' schema\n",
+ me ,argv[0]);
+ exit(1);
+ }
+ n_expected_param = ftaschema_parameter_len(schema);
+ if (n_expected_param == 0) {
+ pblk = 0;
+ pblklen = 0;
+ } else {
+ /* parse the params */
+ n_actual_param = argc-1;
+ if(n_actual_param < n_expected_param){
+ fprintf(stderr,"Error, %d query parameters expected, %d provided.\n",n_expected_param, n_actual_param);
+ exit(1);
+ }
+ for (lcv = 1 ; lcv < argc ; lcv++) {
+ char *k, *e;
+ int rv;
+ k = argv[lcv];
+ e = k;
+ while (*e && *e != '=') e++;
+ if (*e == 0) {
+ fprintf(stderr,"param parse error '%s' (fmt 'key=val')\n",
+ argv[lcv]);
+ exit(1);
+ }
+ *e = 0;
+ rv = ftaschema_setparam_by_name(schema, k, e+1, strlen(e+1));
+ *e = '=';
+ if (rv < 0) {
+ fprintf(stderr,"param setparam error '%s' (fmt 'key=val')\n",
+ argv[lcv]);
+ exit(1);
+ }
+ }
+ if (ftaschema_create_param_block(schema, &pblk, &pblklen) < 0) {
+ fprintf(stderr, "ftaschema_create_param_block failed!\n");
+ exit(1);
+ }
+ }
+ ftaschema_free(schema); /* XXXCDC */
+
+
+ if (verbose>=2) fprintf(stderr,"Initalice FTA\n");
+
+ fta_id=ftaapp_add_fta(argv[0],0,0,0,pblklen,pblk);
+ if (fta_id.streamid==0) {
+ fprintf(stderr,"%s::error:could not initialize fta %s\n",
+ me, argv[0]);
+ exit(1);
+ }
+ /* XXXCDC: pblk is malloc'd, should we free it? */
+
+ if (verbose>=2) fprintf(stderr,"Get schema handle\n");
+
+ if ((schema=ftaapp_get_fta_schema(fta_id))<0) {
+ fprintf(stderr,"%s::error:could not get schema\n", me);
+ exit(1);
+ }
+
+ if ((numberoffields=ftaschema_tuple_len(schema))<0) {
+ fprintf(stderr,"%s::error:could not get number of fields in schema\n",
+ me);
+ exit(1);
+ }
+
+ if (verbose>=1) {
+ for(y=0; y<numberoffields;y++) {
+ printf("%s",ftaschema_field_name(schema,y));
+ if (y<numberoffields-1) printf("|");
+ }
+ printf("\n");
+ }
+ if (xit) {
+ gettimeofday(&tve, 0);
+ timersub(&tve, &tvs, &tvd);
+ printf("TIME= %ld.%06d sec\n", tvd.tv_sec, tvd.tv_usec);
+ hand(0);
+ }
+ if (tcpport!=0) {
+ wait_for_client();
+ }
+
+ start_time = time(NULL);
+
+ while((code=ftaapp_get_tuple(&rfta_id,&rsize,rbuf,2*MAXTUPLESZ,0))>=0) {
+ if (dump)
+ continue;
+ if (ftaschema_is_eof_tuple(schema, rbuf)) {
+ /* initiate shutdown or something of that nature */
+ printf("#All data proccessed\n");
+ exit(0);
+ }
+ if (!rsize)
+ continue;
+ if (verbose >=2) {
+ snprintf(linebuf,MAXLINE,"RESULT CODE => %u\n",code);
+ emit_line();
+ }
+ if ((code==0)&&(rfta_id.streamid == fta_id.streamid)) {
+ for(y=0; y<numberoffields;y++) {
+ struct access_result ar;
+ if (verbose>=2)
+ printf("%s->",ftaschema_field_name(schema,y));
+ ar=ftaschema_get_field_by_index(schema,y,rbuf,rsize);
+ switch (ar.field_data_type) {
+ case INT_TYPE:
+ snprintf(linebuf,MAXLINE,"%d",ar.r.i);
+ break;
+ case UINT_TYPE:
+ snprintf(linebuf,MAXLINE,"%u",ar.r.ui);
+ break;
+ case IP_TYPE:
+ snprintf(linebuf,MAXLINE,"%u.%u.%u.%u",ar.r.ui>>24&0xff,
+ ar.r.ui>>16&0xff,
+ ar.r.ui>>8&0xff,
+ ar.r.ui&0xff);
+ break;
+ case IPV6_TYPE:
+ {
+ unsigned x;
+ unsigned zc=0;
+ for(x=0;x<4;x++) { if (ar.r.ip6.v[x]==0) zc++;}
+ if (zc!=4) {
+ snprintf(linebuf,MAXLINE,"");
+ for(x=0;x<8;x++) {
+ unsigned char * a = (unsigned char *) &(ar.r.ip6.v[0]);
+ unsigned y;
+ y=((unsigned)a[2*x])<<8|((unsigned) a[2*x+1]);
+ snprintf(&linebuf[strlen(linebuf)],MAXLINE,"%04x",y);
+ if (x<7) snprintf(&linebuf[strlen(linebuf)],MAXLINE,":");
+ }
+ } else {
+ snprintf(linebuf,MAXLINE,"::");
+ }
+ }
+ break;
+
+ case USHORT_TYPE:
+ snprintf(linebuf,MAXLINE,"%u",ar.r.ui);
+ break;
+ case BOOL_TYPE:
+ if (ar.r.ui==0) {
+ snprintf(linebuf,MAXLINE,"FALSE");
+ } else {
+ snprintf(linebuf,MAXLINE,"TRUE");
+ }
+ break;
+ case ULLONG_TYPE:
+ snprintf(linebuf,MAXLINE,"%llu",ar.r.ul);
+ break;
+ case LLONG_TYPE:
+ snprintf(linebuf,MAXLINE,"%lld",ar.r.l);
+ break;
+ case FLOAT_TYPE:
+ snprintf(linebuf,MAXLINE,"%f",ar.r.f);
+ break;
+ case TIMEVAL_TYPE:
+ {
+ gs_float_t t;
+ t= ar.r.t.tv_usec;
+ t=t/1000000;
+ t=t+ar.r.t.tv_sec;
+ snprintf(linebuf,MAXLINE,"%f sec",t);
+ }
+ break;
+ case VSTR_TYPE:
+ {
+ int x;
+ int c;
+ int d=0;
+ char * src;
+ src=(char*)ar.r.vs.offset;
+ if(d<MAXLINE){
+ linebuf[d] = '\0';
+ }
+ for(x=0;x<ar.r.vs.length;x++) {
+ c=src[x];
+ if ((c<='~') && (c>=' ')) {
+ if (d<MAXLINE-1) {
+ linebuf[d]=c;
+ linebuf[d+1]=0;
+ d++;
+ }
+ } else {
+ if (d<MAXLINE-1) {
+ linebuf[d]='.';
+ linebuf[d+1]=0;
+ d++;
+ }
+ }
+ }
+ }
+ break;
+ default:
+ linebuf[0]=0;
+ break;
+ }
+ if (y<numberoffields-1) snprintf(&linebuf[strlen(linebuf)],MAXLINE,"|");
+ emit_line();
+ }
+ snprintf(linebuf,MAXLINE,"\n");
+ emit_line();
+ if (verbose!=0) fflush(stdout);
+ } else {
+ if (rfta_id.streamid != fta_id.streamid)
+ fprintf(stderr,"Got unkown streamid %llu \n",rfta_id.streamid);
+ }
+
+ // whenever we receive a temp tuple check if we reached time limit
+ if ((code==2) && tlimit && (time(NULL)-start_time)>=tlimit) {
+ fprintf(stderr,"Reached time limit of %d seconds\n",tlimit);
+ ftaapp_exit();
+ exit(0);
+ }
+ }
+}
+
Code Review / com / gs-lite.git / blobdiff