OpenSRF/src/srfsh/srfsh.c

   1 #include "srfsh.h"
   2
   3 int caught_sigint = 0;
   4
   5 int main( int argc, char* argv[] ) {
   6
   7
   8
   9         /* --------------------------------------------- */
  10         if( argc < 2 ) {
  11
  12                 /* see if they have a .srfsh.xml in their home directory */
  13                 char* home = getenv("HOME");
  14                 int l = strlen(home) + 36;
  15                 char fbuf[l];
  16                 memset(fbuf, 0, l);
  17                 sprintf(fbuf,"%s/.srfsh.xml",home);
  18
  19                 if(!access(fbuf, R_OK)) {
  20                         if( ! osrf_system_bootstrap_client(fbuf) )
  21                                 fatal_handler( "Unable to bootstrap client for requests");
  22
  23                 } else {
  24                         fatal_handler( "No Config file found at %s and none specified. "
  25                                         "\nusage: %s <config_file>", fbuf, argv[0] );
  26                 }
  27
  28         } else {
  29                 if( ! osrf_system_bootstrap_client(argv[1]) )
  30                         fatal_handler( "Unable to bootstrap client for requests");
  31         }
  32         /* --------------------------------------------- */
  33         load_history();
  34
  35
  36         client = osrf_system_get_transport_client();
  37
  38
  39         /* main process loop */
  40         char* request;
  41         signal(SIGINT,sig_int_handler);
  42         while((request=readline(prompt))) {
  43
  44                 if( !strcmp(request, "exit") || !strcmp(request,"quit"))
  45                         break;
  46
  47                 char* req_copy = strdup(request);
  48
  49                 parse_request( req_copy );
  50                 if( request && strlen(request) > 1 ) {
  51                         add_history(request);
  52                 }
  53
  54                 free(request);
  55                 free(req_copy);
  56         }
  57
  58         if(history_file != NULL )
  59                 write_history(history_file);
  60         free(request);
  61         client_disconnect( client );
  62         client_free( client );
  63         config_reader_free();
  64         log_free();
  65
  66         return 0;
  67 }
  68
  69 void sig_child_handler( int s ) {
  70         child_dead = 1;
  71 }
  72
  73 void sig_int_handler( int s ) {
  74         printf("\n");
  75         caught_sigint = 1;
  76         signal(SIGINT,sig_int_handler);
  77 }
  78
  79 int load_history() {
  80
  81         char* home = getenv("HOME");
  82         int l = strlen(home) + 24;
  83         char fbuf[l];
  84
  85         memset(fbuf, 0, l);
  86         sprintf(fbuf,"%s/.srfsh_history",home);
  87         history_file = strdup(fbuf);
  88
  89         if(!access(history_file, W_OK | R_OK )) {
  90                 //set_history_length(999);
  91                 history_length = 999;
  92                 read_history(history_file);
  93         }
  94         return 1;
  95 }
  96
  97
  98 int parse_error( char* words[] ) {
  99
 100         if( ! words )
 101                 return 0;
 102
 103         int i = 0;
 104         char* current;
 105         char buffer[256];
 106         memset(buffer, 0, 256);
 107         while( (current=words[i++]) ) {
 108                 strcat(buffer, current);
 109                 strcat(buffer, " ");
 110         }
 111         if( ! buffer || strlen(buffer) < 1 )
 112                 printf("\n");
 113
 114         fprintf( stderr, "???: %s\n", buffer );
 115         return 0;
 116
 117 }
 118
 119
 120 int parse_request( char* request ) {
 121
 122         if( request == NULL )
 123                 return 0;
 124
 125         int ret_val = 0;
 126         int i = 0;
 127         char* words[COMMAND_BUFSIZE];
 128         memset(words,0,COMMAND_BUFSIZE);
 129         char* req = request;
 130
 131         char* cur_tok = strtok( req, " " );
 132
 133         if( cur_tok == NULL )
 134                 return 0;
 135
 136         while(cur_tok != NULL) {
 137                 words[i++] = cur_tok;
 138                 cur_tok = strtok( NULL, " " );
 139         }
 140
 141
 142         // not sure why (strtok?), but this is necessary
 143         memset( words + i, 0, COMMAND_BUFSIZE - i );
 144
 145         /* pass off to the top level command */
 146         if( !strcmp(words[0],"router") )
 147                 ret_val = handle_router( words );
 148
 149         else if( !strcmp(words[0],"time") )
 150                 ret_val = handle_time( words );
 151
 152         else if (!strcmp(words[0],"request"))
 153                 ret_val = handle_request( words, 0 );
 154
 155         else if (!strcmp(words[0],"relay"))
 156                 ret_val = handle_request( words, 1 );
 157
 158         else if (!strcmp(words[0],"help"))
 159                 ret_val = print_help();
 160
 161         else if (!strcmp(words[0],"set"))
 162                 ret_val = handle_set(words);
 163
 164         else if (!strcmp(words[0],"print"))
 165                 ret_val = handle_print(words);
 166
 167         else if (!strcmp(words[0],"math_bench"))
 168                 ret_val = handle_math(words);
 169
 170         else if (words[0][0] == '!')
 171                 ret_val = handle_exec( words );
 172
 173         if(!ret_val)
 174                 return parse_error( words );
 175
 176         return 1;
 177
 178 }
 179
 180 int handle_set( char* words[]) {
 181
 182         char* variable;
 183         if( (variable=words[1]) ) {
 184
 185                 char* val;
 186                 if( (val=words[2]) ) {
 187
 188                         if(!strcmp(variable,"pretty_print")) {
 189                                 if(!strcmp(val,"true")) {
 190                                         pretty_print = 1;
 191                                         printf("pretty_print = true\n");
 192                                         return 1;
 193                                 }
 194                                 if(!strcmp(val,"false")) {
 195                                         pretty_print = 0;
 196                                         printf("pretty_print = false\n");
 197                                         return 1;
 198                                 }
 199                         }
 200                 }
 201         }
 202
 203         return 0;
 204 }
 205
 206
 207 int handle_print( char* words[]) {
 208
 209         char* variable;
 210         if( (variable=words[1]) ) {
 211                 if(!strcmp(variable,"pretty_print")) {
 212                         if(pretty_print) {
 213                                 printf("pretty_print = true\n");
 214                                 return 1;
 215                         } else {
 216                                 printf("pretty_print = false\n");
 217                                 return 1;
 218                         }
 219                 }
 220         }
 221         return 0;
 222 }
 223
 224 int handle_router( char* words[] ) {
 225
 226         if(!client)
 227                 return 1;
 228
 229         int i;
 230
 231         if( words[1] ) {
 232                 if( !strcmp(words[1],"query") ) {
 233
 234                         if( words[2] && !strcmp(words[2],"servers") ) {
 235                                 for(i=3; i < COMMAND_BUFSIZE - 3 && words[i]; i++ ) {
 236                                         router_query_servers( words[i] );
 237                                 }
 238                                 return 1;
 239                         }
 240                         return 0;
 241                 }
 242                 return 0;
 243         }
 244         return 0;
 245 }
 246
 247
 248 int handle_exec(char* words[]) {
 249
 250         int len = strlen(words[0]);
 251         char command[len];
 252         memset(command,0,len);
 253
 254         int i; /* chop out the ! */
 255         for( i=1; i!= len; i++) {
 256                 command[i-1] = words[0][i];
 257         }
 258
 259         free(words[0]);
 260         words[0] = strdup(command);
 261         signal(SIGCHLD,sig_child_handler);
 262         if(fork()) {
 263                 while(1) {
 264                         sleep(100);
 265                         if(child_dead) {
 266                                 signal(SIGCHLD,sig_child_handler);
 267                                 child_dead = 0;
 268                                 break;
 269                         }
 270                 }
 271         } else {
 272                 execvp( words[0], words );
 273                 exit(0);
 274         }
 275         return 1;
 276 }
 277
 278
 279 int handle_request( char* words[], int relay ) {
 280
 281         if(!client)
 282                 return 1;
 283
 284         if(words[1]) {
 285                 char* server = words[1];
 286                 char* method = words[2];
 287                 int i;
 288                 growing_buffer* buffer = NULL;
 289                 if(!relay) {
 290                         buffer = buffer_init(128);
 291                         buffer_add(buffer, "[");
 292                         for(i = 3; words[i] != NULL; i++ ) {
 293                                 /* removes trailing semicolon if user accidentally enters it */
 294                                 if( words[i][strlen(words[i])-1] == ';' )
 295                                         words[i][strlen(words[i])-1] = '\0';
 296                                 buffer_add( buffer, words[i] );
 297                                 buffer_add(buffer, " ");
 298                         }
 299                         buffer_add(buffer, "]");
 300                 }
 301
 302                 return send_request( server, method, buffer, relay );
 303         }
 304
 305         return 0;
 306 }
 307
 308 int send_request( char* server,
 309                 char* method, growing_buffer* buffer, int relay ) {
 310         if( server == NULL || method == NULL )
 311                 return 0;
 312
 313         json* params = NULL;
 314         if( !relay ) {
 315                 if( buffer != NULL && buffer->n_used > 0 )
 316                         params = json_tokener_parse(buffer->buf);
 317         } else {
 318                 if(!last_result || ! last_result->result_content) {
 319                         printf("We're not going to call 'relay' with no result params\n");
 320                         return 1;
 321                 }
 322                 else {
 323                         json* arr = json_object_new_array();
 324                         json_object_array_add( arr, last_result->result_content );
 325                         params = arr;
 326                 }
 327         }
 328
 329         osrf_app_session* session = osrf_app_client_session_init(server);
 330
 331         if(!osrf_app_session_connect(session)) {
 332                 warning_handler( "Unable to connect to remote service %s\n", server );
 333                 return 1;
 334         }
 335
 336         double start = get_timestamp_millis();
 337         int req_id = osrf_app_session_make_request( session, params, method, 1 );
 338
 339         if( caught_sigint )
 340                 caught_sigint = 0;
 341
 342         osrf_message* omsg = osrf_app_session_request_recv( session, req_id, 8 );
 343
 344         if( caught_sigint ) {
 345                 caught_sigint = 0;
 346                 return 1;
 347         }
 348
 349         if(!omsg)
 350                 printf("\nReceived no data from server\n");
 351
 352
 353         signal(SIGPIPE, SIG_IGN);
 354
 355         FILE* less = popen( "less -EX", "w");
 356         if( less == NULL ) { less = stdout; }
 357
 358         growing_buffer* resp_buffer = buffer_init(4096);
 359
 360         while(omsg) {
 361
 362                 if(omsg->result_content) {
 363
 364                         osrf_message_free(last_result);
 365                         last_result = omsg;
 366
 367                         if( pretty_print ) {
 368                                 char* content = json_printer( omsg->result_content );
 369                                 buffer_add( resp_buffer, "\nReceived Data:" );
 370                                 buffer_add( resp_buffer, content );
 371                                 buffer_add( resp_buffer, "\n" );
 372                                 free(content);
 373                         } else {
 374                                 char* content = json_object_get_string(omsg->result_content);
 375                                         buffer_add( resp_buffer, "\nReceived Data:" );
 376                                         buffer_add( resp_buffer, content );
 377                                         buffer_add( resp_buffer, "\n" );
 378                         }
 379
 380                 } else {
 381
 382                         buffer_add( resp_buffer, "\nReceived Exception:\nName: " );
 383                         buffer_add( resp_buffer, omsg->status_name );
 384                         buffer_add( resp_buffer, "\nStatus: " );
 385                         buffer_add( resp_buffer, omsg->status_text );
 386                         buffer_add( resp_buffer, "\nStatus: " );
 387                         char code[16];
 388                         memset(code, 0, 16);
 389                         sprintf( code, "%d", omsg->status_code );
 390                         buffer_add( resp_buffer, code );
 391                 }
 392
 393                 if( caught_sigint )
 394                         caught_sigint = 0;
 395
 396                 omsg = osrf_app_session_request_recv( session, req_id, 5 );
 397
 398                 if( caught_sigint ) {
 399                         caught_sigint = 0;
 400                         return 1;
 401                 }
 402         }
 403
 404
 405         double end = get_timestamp_millis();
 406
 407         fprintf( less, resp_buffer->buf );
 408         buffer_free( resp_buffer );
 409         fprintf( less, "\n------------------------------------\n");
 410         if( osrf_app_session_request_complete( session, req_id ))
 411                 fprintf(less, "Request Completed Successfully\n");
 412
 413
 414         fprintf(less, "Request Time in seconds: %.3f\n", end - start );
 415         fprintf(less, "------------------------------------\n");
 416
 417         pclose(less);
 418
 419         osrf_app_session_request_finish( session, req_id );
 420         osrf_app_session_disconnect( session );
 421         osrf_app_session_destroy( session );
 422
 423
 424         return 1;
 425
 426
 427 }
 428
 429 int handle_time( char* words[] ) {
 430
 431         if( ! words[1] ) {
 432
 433                 char buf[36];
 434                 memset(buf,0,36);
 435                 get_timestamp(buf);
 436                 printf( "%s\n", buf );
 437                 return 1;
 438         }
 439
 440         if( words[1] ) {
 441                 time_t epoch = (time_t)atoi( words[1] );
 442                 char* localtime = strdup( ctime( &epoch ) );
 443                 printf( "%s => %s", words[1], localtime );
 444                 free(localtime);
 445                 return 1;
 446         }
 447
 448         return 0;
 449
 450 }
 451
 452
 453
 454 int router_query_servers( char* router_server ) {
 455
 456         if( ! router_server || strlen(router_server) == 0 )
 457                 return 0;
 458
 459         char rbuf[256];
 460         memset(rbuf,0,256);
 461         sprintf(rbuf,"router@%s/router", router_server );
 462
 463         transport_message* send =
 464                 message_init( "servers", NULL, NULL, rbuf, NULL );
 465         message_set_router_info( send, NULL, NULL, NULL, "query", 0 );
 466
 467         client_send_message( client, send );
 468         message_free( send );
 469
 470         transport_message* recv = client_recv( client, -1 );
 471         if( recv == NULL ) {
 472                 fprintf(stderr, "NULL message received from router\n");
 473                 return 1;
 474         }
 475
 476         printf(
 477                         "---------------------------------------------------------------------------------\n"
 478                         "Received from 'server' query on %s\n"
 479                         "---------------------------------------------------------------------------------\n"
 480                         "original reg time | latest reg time | last used time | class | server\n"
 481                         "---------------------------------------------------------------------------------\n"
 482                         "%s"
 483                         "---------------------------------------------------------------------------------\n"
 484                         , router_server, recv->body );
 485
 486         message_free( recv );
 487
 488         return 1;
 489 }
 490
 491 int print_help() {
 492
 493         printf(
 494                         "---------------------------------------------------------------------------------\n"
 495                         "Commands:\n"
 496                         "---------------------------------------------------------------------------------\n"
 497                         "help                   - Display this message\n"
 498                         "!<command> [args] - Forks and runs the given command in the shell\n"
 499                         "time                   - Prints the current time\n"
 500                         "time <timestamp>       - Formats seconds since epoch into readable format\n"
 501                         "set <variable> <value> - set a srfsh variable (e.g. set pretty_print true )\n"
 502                         "print <variable>               - Displays the value of a srfsh variable\n"
 503                         "---------------------------------------------------------------------------------\n"
 504                         "router query servers <server1 [, server2, ...]>\n"
 505                         "       - Returns stats on connected services\n"
 506                         "\n"
 507                         "request <service> <method> [ <json formatted string of params> ]\n"
 508                         "       - Anything passed in will be wrapped in a json array,\n"
 509                         "               so add commas if there is more than one param\n"
 510                         "\n"
 511                         "relay <service> <method>\n"
 512                         "       - Performs the requested query using the last received result as the param\n"
 513                         "\n"
 514                         "math_bench <num_batches> [0|1|2]\n"
 515                         "       - 0 means don't reconnect, 1 means reconnect after each batch of 4, and\n"
 516                         "                2 means reconnect after every request\n"
 517                         "---------------------------------------------------------------------------------\n"
 518                         );
 519
 520         return 1;
 521 }
 522
 523
 524
 525 char* tabs(int count) {
 526         growing_buffer* buf = buffer_init(24);
 527         int i;
 528         for(i=0;i!=count;i++)
 529                 buffer_add(buf, "   ");
 530
 531         char* final = buffer_data( buf );
 532         buffer_free( buf );
 533         return final;
 534 }
 535
 536 char* json_printer( json* object ) {
 537
 538         if(object == NULL)
 539                 return NULL;
 540         char* string = json_object_get_string(object);
 541
 542         growing_buffer* buf = buffer_init(64);
 543         int i;
 544         int tab_var = 0;
 545         for(i=0; i!= strlen(string); i++) {
 546
 547                 if( string[i] == '{' ) {
 548
 549                         buffer_add(buf, "\n");
 550                         char* tab = tabs(tab_var);
 551                         buffer_add(buf, tab);
 552                         free(tab);
 553                         buffer_add( buf, "{");
 554                         tab_var++;
 555                         buffer_add( buf, "\n" );
 556                         tab = tabs(tab_var);
 557                         buffer_add( buf, tab );
 558                         free(tab);
 559
 560                 } else if( string[i] == '[' ) {
 561
 562                         buffer_add(buf, "\n");
 563                         char* tab = tabs(tab_var);
 564                         buffer_add(buf, tab);
 565                         free(tab);
 566                         buffer_add( buf, "[");
 567                         tab_var++;
 568                         buffer_add( buf, "\n" );
 569                         tab = tabs(tab_var);
 570                         buffer_add( buf, tab );
 571                         free(tab);
 572
 573                 } else if( string[i] == '}' ) {
 574
 575                         tab_var--;
 576                         buffer_add(buf, "\n");
 577                         char* tab = tabs(tab_var);
 578                         buffer_add(buf, tab);
 579                         free(tab);
 580                         buffer_add( buf, "}");
 581                         buffer_add( buf, "\n" );
 582                         tab = tabs(tab_var);
 583                         buffer_add( buf, tab );
 584                         free(tab);
 585
 586                 } else if( string[i] == ']' ) {
 587
 588                         tab_var--;
 589                         buffer_add(buf, "\n");
 590                         char* tab = tabs(tab_var);
 591                         buffer_add(buf, tab);
 592                         free(tab);
 593                         buffer_add( buf, "]");
 594                         buffer_add( buf, "\n" );
 595                         tab = tabs(tab_var);
 596                         buffer_add( buf, tab );
 597                         free(tab);
 598
 599                 } else if( string[i] == ',' ) {
 600
 601                         buffer_add( buf, ",");
 602                         buffer_add( buf, "\n" );
 603                         char* tab = tabs(tab_var);
 604                         buffer_add(buf, tab);
 605                         free(tab);
 606
 607                 } else {
 608
 609                         char b[2];
 610                         b[0] = string[i];
 611                         b[1] = '\0';
 612                         buffer_add( buf, b );
 613                 }
 614
 615         }
 616
 617         char* result = buffer_data(buf);
 618         buffer_free(buf);
 619         return result;
 620
 621 }
 622
 623 int handle_math( char* words[] ) {
 624         if( words[1] && words[2] )
 625                 return do_math( atoi(words[1]), atoi(words[2]) );
 626         return 0;
 627 }
 628
 629
 630 int do_math( int count, int style ) {
 631
 632         osrf_app_session* session = osrf_app_client_session_init(  "opensrf.math" );
 633
 634         json* params = json_object_new_array();
 635         json_object_array_add(params, json_object_new_string("1"));
 636         json_object_array_add(params, json_object_new_string("2"));
 637
 638         char* methods[] = { "add", "sub", "mult", "div" };
 639         char* answers[] = { "3", "-1", "2", "0.500000" };
 640
 641         float times[ count * 4 ];
 642         memset(times,0,count*4);
 643
 644         int k;
 645         for(k=0;k!=100;k++) {
 646                 if(!(k%10))
 647                         fprintf(stderr,"|");
 648                 else
 649                         fprintf(stderr,".");
 650         }
 651
 652         fprintf(stderr,"\n\n");
 653
 654         int running = 0;
 655         int i;
 656         for(i=0; i!= count; i++) {
 657
 658                 int j;
 659                 for(j=0; j != 4; j++) {
 660
 661                         ++running;
 662                         struct timeb t1;
 663                         struct timeb t2;
 664
 665                         ftime(&t1);
 666                         int req_id = osrf_app_session_make_request( session, params, methods[j], 1 );
 667
 668                         if( caught_sigint )
 669                                 caught_sigint = 0;
 670
 671                         osrf_message* omsg = osrf_app_session_request_recv( session, req_id, 5 );
 672
 673                         if( caught_sigint )  {
 674                                 caught_sigint = 0;
 675                                 return 1;
 676                         }
 677
 678                         ftime(&t2);
 679
 680                         double start    = ( (int)t1.time        + ( ((float)t1.millitm) / 1000 ) );
 681                         double end              = ( (int)t2.time        + ( ((float)t2.millitm) / 1000 ) );
 682
 683                         times[(4*i) + j] = end - start;
 684
 685                         if(omsg) {
 686
 687                                 if(omsg->result_content) {
 688                                         char* jsn = json_object_get_string( omsg->result_content );
 689                                         if(!strcmp(jsn, answers[j]))
 690                                                 fprintf(stderr, "+");
 691                                         else
 692                                                 fprintf(stderr, "\n![%s] - should be %s\n", jsn, answers[j] );
 693                                 }
 694
 695                                 osrf_message_free(omsg);
 696
 697                         } else { fprintf( stderr, "\nempty message for tt: %d\n", req_id ); }
 698
 699                         osrf_app_session_request_finish( session, req_id );
 700
 701                         if(style == 2)
 702                                 osrf_app_session_disconnect( session );
 703
 704                         if(!(running%100))
 705                                 fprintf(stderr,"\n");
 706                 }
 707
 708                 if(style==1)
 709                         osrf_app_session_disconnect( session );
 710         }
 711
 712         osrf_app_session_destroy( session );
 713         json_object_put( params );
 714
 715         int c;
 716         float total = 0;
 717         for(c=0; c!= count*4; c++)
 718                 total += times[c];
 719
 720         float avg = total / (count*4);
 721         fprintf(stderr, "\n      Average round trip time: %f\n", avg );
 722
 723         return 1;
 724 }