src/yaz-my-client.cpp

   1 /*
   2  * Copyright (c) 1998-2000, Index Data.
   3  * See the file LICENSE for details.
   4  *
   5  * $Log: yaz-my-client.cpp,v $
   6  * Revision 1.3  2001-04-05 13:09:44  adam
   7  * Removed ursula dependancy.
   8  *
   9  * Revision 1.2  2001/04/04 14:02:49  adam
  10  * URSULA / Z-ruth service.
  11  *
  12  * Revision 1.1  2001/03/27 14:47:45  adam
  13  * New server facility scheme.
  14  *
  15  * Revision 1.17  2001/03/26 14:43:49  adam
  16  * New threaded PDU association.
  17  *
  18  * Revision 1.16  2000/11/01 14:22:59  adam
  19  * Added fd parameter for method IYaz_PDU_Observer::clone.
  20  *
  21  * Revision 1.15  2000/10/11 11:58:16  adam
  22  * Moved header files to include/yaz++. Switched to libtool and automake.
  23  * Configure script creates yaz++-config script.
  24  *
  25  * Revision 1.14  2000/09/08 10:23:42  adam
  26  * Added skeleton of yaz-z-server.
  27  *
  28  * Revision 1.13  2000/09/06 14:23:45  adam
  29  * WIN32 updates.
  30  *
  31  * Revision 1.12  2000/09/04 08:59:16  adam
  32  * Changed call to logging functions (yaz_ added).
  33  *
  34  * Revision 1.11  2000/07/04 13:48:49  adam
  35  * Implemented upper-limit on proxy-to-target sessions.
  36  *
  37  * Revision 1.10  2000/05/30 03:12:27  ian
  38  * minor change to stop g++ 2.95.2 complaining about taking the address
  39  * of a member function.
  40  *
  41  * Revision 1.9  1999/12/06 13:52:45  adam
  42  * Modified for new location of YAZ header files. Experimental threaded
  43  * operation.
  44  *
  45  * Revision 1.8  1999/11/10 10:02:34  adam
  46  * Work on proxy.
  47  *
  48  * Revision 1.7  1999/04/21 12:09:01  adam
  49  * Many improvements. Modified to proxy server to work with "sessions"
  50  * based on cookies.
  51  *
  52  * Revision 1.6  1999/04/20 10:30:05  adam
  53  * Implemented various stuff for client and proxy. Updated calls
  54  * to ODR to reflect new name parameter.
  55  *
  56  * Revision 1.5  1999/04/09 11:46:57  adam
  57  * Added object Yaz_Z_Assoc. Much more functional client.
  58  *
  59  * Revision 1.4  1999/03/23 14:17:57  adam
  60  * More work on timeout handling. Work on yaz-client.
  61  *
  62  * Revision 1.3  1999/02/02 14:01:18  adam
  63  * First WIN32 port of YAZ++.
  64  *
  65  * Revision 1.2  1999/01/28 13:08:42  adam
  66  * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
  67  * yaz-socket-manager.cc.
  68  *
  69  * Revision 1.1.1.1  1999/01/28 09:41:07  adam
  70  * First implementation of YAZ++.
  71  *
  72  */
  73
  74 #include <yaz/log.h>
  75 #include <yaz/options.h>
  76 #include <yaz/diagbib1.h>
  77 #include <yaz/marcdisp.h>
  78 #include <yaz++/yaz-ir-assoc.h>
  79 #include <yaz++/yaz-pdu-assoc.h>
  80 #include <yaz++/yaz-socket-manager.h>
  81
  82 #if HAVE_YAZ_URSULA_H
  83 #include <yaz/zes-ursula.h>
  84 #endif
  85
  86 extern "C" {
  87 #if HAVE_READLINE_READLINE_H
  88 #include <readline/readline.h>
  89 #endif
  90 #if HAVE_READLINE_HISTORY_H
  91 #include <readline/history.h>
  92 #endif
  93 }
  94
  95 class YAZ_EXPORT MyClient : public Yaz_IR_Assoc {
  96 private:
  97     int m_interactive_flag;
  98     char m_thisCommand[1024];
  99     char m_lastCommand[1024];
 100     int m_setOffset;
 101     Yaz_SocketManager *m_socketManager;
 102 public:
 103     MyClient(IYaz_PDU_Observable *the_PDU_Observable,
 104              Yaz_SocketManager *the_SocketManager);
 105     IYaz_PDU_Observer *sessionNotify(
 106         IYaz_PDU_Observable *the_PDU_Observable, int fd);
 107     int args(Yaz_SocketManager *socketManager, int argc, char **argv);
 108     int interactive(Yaz_SocketManager *socketManager);
 109     int wait();
 110     void recv_initResponse(Z_InitResponse *initResponse);
 111     void recv_searchResponse(Z_SearchResponse *searchResponse);
 112     void recv_presentResponse(Z_PresentResponse *presentResponse);
 113     void recv_records (Z_Records *records);
 114     void recv_diagrecs(Z_DiagRec **pp, int num);
 115     void recv_namePlusRecord (Z_NamePlusRecord *zpr, int offset);
 116     void recv_record(Z_DatabaseRecord *record, int offset,
 117                      const char *databaseName);
 118     void recv_textRecord(int type, const char *buf, size_t len);
 119     void recv_genericRecord(Z_GenericRecord *r);
 120     void display_genericRecord(Z_GenericRecord *r, int level);
 121     void display_variant(Z_Variant *v, int level);
 122     void connectNotify();
 123     void failNotify();
 124     void timeoutNotify();
 125     char *get_cookie (Z_OtherInformation **oi);
 126     int processCommand(const char *cmd);
 127     const char *MyClient::getCommand();
 128     int cmd_open(char *host);
 129     int cmd_connect(char *host);
 130     int cmd_quit(char *args);
 131     int cmd_close(char *args);
 132     int cmd_find(char *args);
 133     int cmd_show(char *args);
 134     int cmd_cookie(char *args);
 135     int cmd_init(char *args);
 136     int cmd_format(char *args);
 137     int cmd_proxy(char *args);
 138 #if HAVE_YAZ_URSULA_H
 139     int cmd_ursula(char *args);
 140 #endif
 141 };
 142
 143
 144 void MyClient::connectNotify()
 145 {
 146     printf ("Connection accepted by target\n");
 147     set_lastReceived(-1);
 148 }
 149
 150 void MyClient::timeoutNotify()
 151 {
 152     printf ("Connection timeout\n");
 153     close();
 154 }
 155
 156 void MyClient::failNotify()
 157 {
 158     printf ("Connection closed by target\n");
 159     set_lastReceived(-1);
 160 }
 161
 162 IYaz_PDU_Observer *MyClient::sessionNotify(
 163     IYaz_PDU_Observable *the_PDU_Observable, int fd)
 164 {
 165     return new MyClient(the_PDU_Observable, m_socketManager);
 166 }
 167
 168 MyClient::MyClient(IYaz_PDU_Observable *the_PDU_Observable,
 169                    Yaz_SocketManager *the_socketManager) :
 170     Yaz_IR_Assoc (the_PDU_Observable)
 171 {
 172     m_setOffset = 1;
 173     m_interactive_flag = 1;
 174     m_thisCommand[0] = '\0';
 175     m_lastCommand[0] = '\0';
 176     m_socketManager = the_socketManager;
 177 }
 178
 179 void usage(char *prog)
 180 {
 181     fprintf (stderr, "%s: [-v log] [-c cookie] [-p proxy] [zurl]\n", prog);
 182     exit (1);
 183 }
 184
 185 char *MyClient::get_cookie(Z_OtherInformation **otherInfo)
 186 {
 187     int oid[OID_SIZE];
 188     Z_OtherInformationUnit *oi;
 189     struct oident ent;
 190     ent.proto = PROTO_Z3950;
 191     ent.oclass = CLASS_USERINFO;
 192     ent.value = (oid_value) VAL_COOKIE;
 193
 194     if (oid_ent_to_oid (&ent, oid) &&
 195         (oi = update_otherInformation(otherInfo, 0, oid, 1, 1)) &&
 196         oi->which == Z_OtherInfo_characterInfo)
 197         return oi->information.characterInfo;
 198     return 0;
 199 }
 200
 201 void MyClient::recv_initResponse(Z_InitResponse *initResponse)
 202 {
 203     printf ("Got InitResponse. Status ");
 204     if (*initResponse->result)
 205     {
 206         printf ("Ok\n");
 207
 208         const char *p = get_cookie (&initResponse->otherInfo);
 209         if (p)
 210         {
 211             printf ("cookie = %s\n", p);
 212             set_cookie(p);
 213         }
 214     }
 215     else
 216         printf ("Fail\n");
 217 }
 218
 219 void MyClient::recv_diagrecs(Z_DiagRec **pp, int num)
 220 {
 221     int i;
 222     oident *ent;
 223     Z_DefaultDiagFormat *r;
 224
 225     printf("Diagnostic message(s) from database:\n");
 226     for (i = 0; i<num; i++)
 227     {
 228         Z_DiagRec *p = pp[i];
 229         if (p->which != Z_DiagRec_defaultFormat)
 230         {
 231             printf("Diagnostic record not in default format.\n");
 232             return;
 233         }
 234         else
 235             r = p->u.defaultFormat;
 236         if (!(ent = oid_getentbyoid(r->diagnosticSetId)) ||
 237             ent->oclass != CLASS_DIAGSET || ent->value != VAL_BIB1)
 238             printf("Missing or unknown diagset\n");
 239         printf("    [%d] %s", *r->condition, diagbib1_str(*r->condition));
 240 #ifdef ASN_COMPILED
 241         switch (r->which)
 242         {
 243         case Z_DefaultDiagFormat_v2Addinfo:
 244             printf (" -- v2 addinfo '%s'\n", r->u.v2Addinfo);
 245             break;
 246         case Z_DefaultDiagFormat_v3Addinfo:
 247             printf (" -- v3 addinfo '%s'\n", r->u.v3Addinfo);
 248             break;
 249         }
 250 #else
 251         if (r->addinfo && *r->addinfo)
 252             printf(" -- '%s'\n", r->addinfo);
 253         else
 254             printf("\n");
 255 #endif
 256     }
 257 }
 258
 259 void MyClient::recv_textRecord(int type, const char *buf, size_t len)
 260 {
 261     fwrite (buf, 1, len, stdout);
 262     fputc ('\n', stdout);
 263 }
 264
 265 void MyClient::display_variant(Z_Variant *v, int level)
 266 {
 267     int i;
 268
 269     for (i = 0; i < v->num_triples; i++)
 270     {
 271         printf("%*sclass=%d,type=%d", level * 4, "", *v->triples[i]->zclass,
 272             *v->triples[i]->type);
 273         if (v->triples[i]->which == Z_Triple_internationalString)
 274             printf(",value=%s\n", v->triples[i]->value.internationalString);
 275         else
 276             printf("\n");
 277     }
 278 }
 279
 280 void MyClient::display_genericRecord(Z_GenericRecord *r, int level)
 281 {
 282     int i;
 283
 284     if (!r)
 285         return;
 286     for (i = 0; i < r->num_elements; i++)
 287     {
 288         Z_TaggedElement *t;
 289
 290         printf("%*s", level * 4, "");
 291         t = r->elements[i];
 292         printf("(");
 293         if (t->tagType)
 294             printf("%d,", *t->tagType);
 295         else
 296             printf("?,");
 297         if (t->tagValue->which == Z_StringOrNumeric_numeric)
 298             printf("%d) ", *t->tagValue->u.numeric);
 299         else
 300             printf("%s) ", t->tagValue->u.string);
 301         if (t->content->which == Z_ElementData_subtree)
 302         {
 303             printf("\n");
 304             display_genericRecord(t->content->u.subtree, level+1);
 305         }
 306         else if (t->content->which == Z_ElementData_string)
 307             printf("%s\n", t->content->u.string);
 308         else if (t->content->which == Z_ElementData_numeric)
 309             printf("%d\n", *t->content->u.numeric);
 310         else if (t->content->which == Z_ElementData_oid)
 311         {
 312             int *ip = t->content->u.oid;
 313             oident *oent;
 314
 315             if ((oent = oid_getentbyoid(t->content->u.oid)))
 316                 printf("OID: %s\n", oent->desc);
 317             else
 318             {
 319                 printf("{");
 320                 while (ip && *ip >= 0)
 321                     printf(" %d", *(ip++));
 322                 printf(" }\n");
 323             }
 324         }
 325         else if (t->content->which == Z_ElementData_noDataRequested)
 326             printf("[No data requested]\n");
 327         else if (t->content->which == Z_ElementData_elementEmpty)
 328             printf("[Element empty]\n");
 329         else if (t->content->which == Z_ElementData_elementNotThere)
 330             printf("[Element not there]\n");
 331         else
 332             printf("??????\n");
 333         if (t->appliedVariant)
 334             display_variant(t->appliedVariant, level+1);
 335         if (t->metaData && t->metaData->supportedVariants)
 336         {
 337             int c;
 338
 339             printf("%*s---- variant list\n", (level+1)*4, "");
 340             for (c = 0; c < t->metaData->num_supportedVariants; c++)
 341             {
 342                 printf("%*svariant #%d\n", (level+1)*4, "", c);
 343                 display_variant(t->metaData->supportedVariants[c], level + 2);
 344             }
 345         }
 346     }
 347 }
 348
 349 void MyClient::recv_genericRecord(Z_GenericRecord *r)
 350 {
 351     display_genericRecord(r, 0);
 352 }
 353
 354 void MyClient::recv_record(Z_DatabaseRecord *record, int offset,
 355                            const char *databaseName)
 356 {
 357     Z_External *r = (Z_External*) record;
 358     oident *ent = oid_getentbyoid(r->direct_reference);
 359
 360     /*
 361      * Tell the user what we got.
 362      */
 363     if (r->direct_reference)
 364     {
 365         printf("Record type: ");
 366         if (ent)
 367             printf("%s\n", ent->desc);
 368     }
 369     /* Check if this is a known, ASN.1 type tucked away in an octet string */
 370     Z_ext_typeent *etype = z_ext_getentbyref(ent->value);
 371     if (ent && (r->which == Z_External_octet || r->which == Z_External_single)
 372         && (etype = z_ext_getentbyref(ent->value)))
 373
 374     {
 375         void *rr;
 376         /*
 377          * Call the given decoder to process the record.
 378          */
 379         odr_setbuf(odr_decode(), (char*)record->u.octet_aligned->buf,
 380                    record->u.octet_aligned->len, 0);
 381         if (!(*etype->fun)(odr_decode(), (char **)&rr, 0, 0))
 382         {
 383             odr_perror(odr_decode(), "Decoding constructed record.");
 384             fprintf(stderr, "[Near %d]\n", odr_offset(odr_decode()));
 385             fprintf(stderr, "Packet dump:\n---------\n");
 386             odr_dumpBER(stderr, (char*)record->u.octet_aligned->buf,
 387                         record->u.octet_aligned->len);
 388             fprintf(stderr, "---------\n");
 389         }
 390         if (etype->what == Z_External_sutrs)
 391         {
 392             Z_SUTRS *sutrs = (Z_SUTRS *) rr;
 393             recv_textRecord ((int) VAL_SUTRS, (const char *) sutrs->buf,
 394                              (size_t) sutrs->len);
 395         }
 396         return;
 397     }
 398     if (r->which == Z_External_octet && record->u.octet_aligned->len)
 399     {
 400         switch (ent->value)
 401         {
 402         case VAL_ISO2709:
 403         case VAL_UNIMARC:
 404         case VAL_INTERMARC:
 405         case VAL_USMARC:
 406         case VAL_UKMARC:
 407         case VAL_NORMARC:
 408         case VAL_LIBRISMARC:
 409         case VAL_DANMARC:
 410         case VAL_FINMARC:
 411         case VAL_MAB:
 412         case VAL_CANMARC:
 413         case VAL_SBN:
 414         case VAL_PICAMARC:
 415         case VAL_AUSMARC:
 416         case VAL_IBERMARC:
 417         case VAL_CATMARC:
 418         case VAL_MALMARC:
 419         case VAL_JPMARC:
 420         case VAL_SWEMARC:
 421         case VAL_SIGLEMARC:
 422         case VAL_ISDSMARC:
 423         case VAL_RUSMARC:
 424             marc_display((char*) record->u.octet_aligned->buf,stdout);
 425             break;
 426         default:
 427             recv_textRecord((int) ent->value,
 428                             (const char *) record->u.octet_aligned->buf,
 429                             (size_t) record->u.octet_aligned->len);
 430         }
 431     }
 432     else if (ent && ent->value == VAL_SUTRS && r->which == Z_External_sutrs)
 433         recv_textRecord((int) VAL_SUTRS, (const char *) r->u.sutrs->buf,
 434                         (size_t) r->u.sutrs->len);
 435     else if (ent && ent->value == VAL_GRS1 && r->which == Z_External_grs1)
 436         recv_genericRecord(r->u.grs1);
 437     else
 438     {
 439         printf("Unknown record representation.\n");
 440         if (!z_External(odr_print(), &r, 0, 0))
 441         {
 442             odr_perror(odr_print(), "Printing external");
 443             odr_reset(odr_print());
 444         }
 445     }
 446 }
 447
 448 void MyClient::recv_namePlusRecord (Z_NamePlusRecord *zpr, int offset)
 449 {
 450     if (zpr->databaseName)
 451         printf("[%s]", zpr->databaseName);
 452     if (zpr->which == Z_NamePlusRecord_surrogateDiagnostic)
 453         recv_diagrecs(&zpr->u.surrogateDiagnostic, 1);
 454     else
 455         recv_record(zpr->u.databaseRecord, offset, zpr->databaseName);
 456 }
 457
 458 void MyClient::recv_records (Z_Records *records)
 459 {
 460 #ifdef ASN_COMPILED
 461     Z_DiagRec dr, *dr_p = &dr;
 462 #endif
 463     if (!records)
 464         return;
 465     int i;
 466     switch (records->which)
 467     {
 468     case Z_Records_DBOSD:
 469         for (i = 0; i < records->u.databaseOrSurDiagnostics->num_records; i++)
 470             recv_namePlusRecord(records->u.databaseOrSurDiagnostics->
 471                                 records[i], i + m_setOffset);
 472         m_setOffset += records->u.databaseOrSurDiagnostics->num_records;
 473         break;
 474     case Z_Records_NSD:
 475 #ifdef ASN_COMPILED
 476         dr.which = Z_DiagRec_defaultFormat;
 477         dr.u.defaultFormat = records->u.nonSurrogateDiagnostic;
 478         recv_diagrecs (&dr_p, 1);
 479 #else
 480         recv_diagrecs (&records->u.nonSurrogateDiagnostic, 1);
 481 #endif
 482         break;
 483     case Z_Records_multipleNSD:
 484         recv_diagrecs (records->u.multipleNonSurDiagnostics->diagRecs,
 485                        records->u.multipleNonSurDiagnostics->num_diagRecs);
 486         break;
 487     }
 488 }
 489
 490 void MyClient::recv_searchResponse(Z_SearchResponse *searchResponse)
 491 {
 492     printf ("Got SearchResponse. Status ");
 493     if (!*searchResponse->searchStatus)
 494     {
 495         printf ("Fail\n");
 496         return;
 497     }
 498     printf ("Ok\n");
 499     printf ("Hits: %d\n", *searchResponse->resultCount);
 500     recv_records (searchResponse->records);
 501 }
 502
 503 void MyClient::recv_presentResponse(Z_PresentResponse *presentResponse)
 504 {
 505     printf ("Got PresentResponse\n");
 506     recv_records (presentResponse->records);
 507 }
 508
 509 int MyClient::wait()
 510 {
 511     set_lastReceived(0);
 512     while (m_socketManager->processEvent() > 0)
 513     {
 514         if (get_lastReceived())
 515             return 1;
 516     }
 517     return 0;
 518 }
 519
 520 #define C_PROMPT "Z>"
 521
 522 int MyClient::cmd_connect(char *host)
 523 {
 524     client (host);
 525     timeout (10);
 526     wait ();
 527     timeout (0);
 528     return 1;
 529 }
 530
 531 int MyClient::cmd_open(char *host)
 532 {
 533     client (host);
 534     timeout (10);
 535     wait ();
 536     timeout (0);
 537     send_initRequest();
 538     wait ();
 539     return 1;
 540 }
 541
 542 int MyClient::cmd_init(char *args)
 543 {
 544     if (send_initRequest() >= 0)
 545         wait();
 546     else
 547         close();
 548     return 1;
 549 }
 550
 551 int MyClient::cmd_quit(char *args)
 552 {
 553     return 0;
 554 }
 555
 556 int MyClient::cmd_close(char *args)
 557 {
 558     close();
 559     return 1;
 560 }
 561
 562 int MyClient::cmd_find(char *args)
 563 {
 564     Yaz_Z_Query query;
 565
 566     if (query.set_rpn(args) <= 0)
 567     {
 568         printf ("Bad RPN query\n");
 569         return 1;
 570     }
 571     if (send_searchRequest(&query) >= 0)
 572         wait();
 573     else
 574         printf ("Not connected\n");
 575     return 1;
 576 }
 577
 578 int MyClient::cmd_show(char *args)
 579 {
 580     int start = m_setOffset, number = 1;
 581
 582     sscanf (args, "%d %d", &start, &number);
 583     m_setOffset = start;
 584     if (send_presentRequest(start, number) >= 0)
 585         wait();
 586     else
 587         printf ("Not connected\n");
 588     return 1;
 589 }
 590
 591 int MyClient::cmd_cookie(char *args)
 592 {
 593     set_cookie(*args ? args : 0);
 594     return 1;
 595 }
 596
 597 int MyClient::cmd_format(char *args)
 598 {
 599     set_preferredRecordSyntax(args);
 600     return 1;
 601 }
 602
 603 int MyClient::cmd_proxy(char *args)
 604 {
 605     set_proxy(args);
 606     return 1;
 607 }
 608
 609 #if HAVE_YAZ_URSULA_H
 610 int MyClient::cmd_ursula(char *args)
 611 {
 612     Z_APDU *apdu = create_Z_PDU(Z_APDU_extendedServicesRequest);
 613     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
 614
 615     req->packageType = odr_getoidbystr(odr_encode(), "1.2.840.10003");
 616
 617     Z_External *ext = (Z_External *) odr_malloc(odr_encode(), sizeof(*ext));
 618     req->taskSpecificParameters = ext;
 619     ext->direct_reference = req->packageType;
 620     ext->descriptor = 0;
 621     ext->indirect_reference = 0;
 622
 623     ext->which = Z_External_octet;
 624     ext->u.single_ASN1_type = (Odr_oct *)
 625         odr_malloc (odr_encode(), sizeof(Odr_oct));
 626
 627     Z_UrsPDU *pdu = (Z_UrsPDU *) odr_malloc (odr_encode(), sizeof(*pdu));
 628     pdu->which = Z_UrsPDU_request;
 629     pdu->u.request = (Z_UrsRequest *)
 630         odr_malloc (odr_encode(), sizeof(*pdu->u.request));
 631     pdu->u.request->libraryNo = odr_strdup(odr_encode(), "000200");
 632     pdu->u.request->borrowerTickerNo = 0;
 633     pdu->u.request->disposalType = 0;
 634     pdu->u.request->lastUseDate = 0;
 635     pdu->u.request->num_items = 0;
 636     pdu->u.request->items = (Z_UrsRequestItem **) odr_nullval();
 637     pdu->u.request->counter = 0;
 638     pdu->u.request->priority = 0;
 639     pdu->u.request->disposalNote = 0;
 640     pdu->u.request->overrule = 0;
 641
 642     if (!z_UrsPDU (odr_encode(), &pdu, 0, ""))
 643     {
 644         yaz_log (LOG_LOG, "ursula encoding failed");
 645         return 1;
 646     }
 647     char *buf =
 648         odr_getbuf (odr_encode(), &ext->u.single_ASN1_type->len, 0);
 649
 650     ext->u.single_ASN1_type->buf = (unsigned char*)
 651         odr_malloc (odr_encode(), ext->u.single_ASN1_type->len);
 652     memcpy (ext->u.single_ASN1_type->buf, buf, ext->u.single_ASN1_type->len);
 653     ext->u.single_ASN1_type->size = ext->u.single_ASN1_type->len;
 654
 655     if (send_Z_PDU(apdu) >= 0)
 656         wait();
 657     return 1;
 658 }
 659 #endif
 660
 661 int MyClient::processCommand(const char *commandLine)
 662 {
 663     char cmdStr[1024], cmdArgs[1024];
 664     cmdArgs[0] = '\0';
 665     cmdStr[0] = '\0';
 666     static struct {
 667         char *cmd;
 668         int (MyClient::*fun)(char *arg);
 669         char *ad;
 670     } cmd[] = {
 671         {"open", &MyClient::cmd_open, "<host>[':'<port>][/<database>]"},
 672         {"connect", &MyClient::cmd_connect, "<host>[':'<port>][/<database>]"},
 673         {"quit", &MyClient::cmd_quit, ""},
 674         {"close", &MyClient::cmd_close, ""},
 675         {"find", &MyClient::cmd_find, "<query>"},
 676         {"show", &MyClient::cmd_show, "[<start> [<number>]]"},
 677         {"cookie", &MyClient::cmd_cookie, "<cookie>"},
 678         {"init", &MyClient::cmd_init, ""},
 679         {"format", &MyClient::cmd_format, "<record-syntax>"},
 680         {"proxy", &MyClient::cmd_proxy, "<host>:[':'<port>]"},
 681 #if HAVE_YAZ_URSULA_H
 682         {"ursula", &MyClient::cmd_ursula, ""},
 683 #endif
 684         {0,0,0}
 685     };
 686
 687     if (sscanf(commandLine, "%s %[^;]", cmdStr, cmdArgs) < 1)
 688         return 1;
 689     int i;
 690     for (i = 0; cmd[i].cmd; i++)
 691         if (!strncmp(cmd[i].cmd, cmdStr, strlen(cmdStr)))
 692             break;
 693
 694     int res = 1;
 695     if (cmd[i].cmd) // Invoke command handler
 696         res = (this->*cmd[i].fun)(cmdArgs);
 697     else            // Dump help screen
 698     {
 699         printf("Unknown command: %s.\n", cmdStr);
 700         printf("Currently recognized commands:\n");
 701         for (i = 0; cmd[i].cmd; i++)
 702             printf("   %s %s\n", cmd[i].cmd, cmd[i].ad);
 703     }
 704     return res;
 705 }
 706
 707 const char *MyClient::getCommand()
 708 {
 709 #if HAVE_READLINE_READLINE_H
 710     // Read using GNU readline
 711     char *line_in;
 712     line_in=readline(C_PROMPT);
 713     if (!line_in)
 714         return 0;
 715 #if HAVE_READLINE_HISTORY_H
 716     if (*line_in)
 717         add_history(line_in);
 718 #endif
 719     strncpy(m_thisCommand,line_in, 1023);
 720     m_thisCommand[1023] = '\0';
 721     free (line_in);
 722 #else
 723     // Read using fgets(3)
 724     printf (C_PROMPT);
 725     fflush(stdout);
 726     if (!fgets(m_thisCommand, 1023, stdin))
 727         return 0;
 728 #endif
 729     // Remove trailing whitespace
 730     char *cp = m_thisCommand + strlen(m_thisCommand);
 731     while (cp != m_thisCommand && strchr("\t \n", cp[-1]))
 732         cp--;
 733     *cp = '\0';
 734     cp = m_thisCommand;
 735     // Remove leading spaces...
 736     while (*cp && strchr ("\t \n", *cp))
 737         cp++;
 738     // Save command if non-empty
 739     if (*cp != '\0')
 740         strcpy (m_lastCommand, cp);
 741     return m_lastCommand;
 742 }
 743
 744 int MyClient::interactive(Yaz_SocketManager *socketManager)
 745 {
 746     const char *cmd;
 747     if (!m_interactive_flag)
 748         return 0;
 749     while ((cmd = getCommand()))
 750     {
 751         if (!processCommand(cmd))
 752             break;
 753     }
 754     return 0;
 755 }
 756
 757 int MyClient::args(Yaz_SocketManager *socketManager, int argc, char **argv)
 758 {
 759     char *host = 0;
 760     char *proxy = 0;
 761     char *arg;
 762     char *prog = argv[0];
 763     int ret;
 764
 765     while ((ret = options("c:p:v:q", argv, argc, &arg)) != -2)
 766     {
 767         switch (ret)
 768         {
 769         case 0:
 770             if (host)
 771             {
 772                 usage(prog);
 773                 return 1;
 774             }
 775             host = arg;
 776             break;
 777         case 'p':
 778             if (proxy)
 779             {
 780                 usage(prog);
 781                 return 1;
 782             }
 783             set_proxy(arg);
 784             break;
 785         case 'c':
 786             set_cookie(arg);
 787             break;
 788         case 'v':
 789             yaz_log_init_level (yaz_log_mask_str(arg));
 790             break;
 791         case 'q':
 792             m_interactive_flag = 0;
 793             break;
 794         default:
 795             usage(prog);
 796             return 1;
 797         }
 798     }
 799     if (host)
 800     {
 801         client (host);
 802         timeout (10);
 803         wait ();
 804         timeout (0);
 805         send_initRequest();
 806         wait ();
 807     }
 808     return 0;
 809 }
 810
 811 int main(int argc, char **argv)
 812 {
 813     Yaz_SocketManager mySocketManager;
 814     Yaz_PDU_Assoc *some = new Yaz_PDU_Assoc(&mySocketManager);
 815
 816     MyClient z(some, &mySocketManager);
 817
 818     if (z.args(&mySocketManager, argc, argv))
 819         exit (1);
 820     if (z.interactive(&mySocketManager))
 821         exit (1);
 822     return 0;
 823 }