src/srfsh/srfsh.c

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