client/client.c

   1 /*
   2  * Copyright (c) 1995, Index Data.
   3  * See the file LICENSE for details.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * $Log: client.c,v $
   7  * Revision 1.35  1996-06-03 09:45:50  quinn
   8  * Added display of OIDs in the GRS routine.
   9  *
  10  * Revision 1.34  1996/05/09  07:26:49  quinn
  11  * *** empty log message ***
  12  *
  13  * Revision 1.33  1996/05/09  07:25:22  quinn
  14  * Small
  15  *
  16  * Revision 1.32  1996/03/15  11:05:33  adam
  17  * The user can set the preferred query type (prefix, ccl, ..) with the
  18  * querytype command.
  19  *
  20  * Revision 1.31  1996/02/20  12:51:54  quinn
  21  * Fixed problems with EXTERNAL.
  22  *
  23  * Revision 1.30  1996/02/12  18:18:09  quinn
  24  * Fidgeting.
  25  *
  26  * Revision 1.29  1996/01/02  08:57:25  quinn
  27  * Changed enums in the ASN.1 .h files to #defines. Changed oident.class to oclass
  28  *
  29  * Revision 1.28  1995/12/14  11:09:31  quinn
  30  * Added Explain record syntax to the format command.
  31  *
  32  * Revision 1.27  1995/12/12  16:37:02  quinn
  33  * Added destroy element to data1_node.
  34  *
  35  * Revision 1.26  1995/12/12  14:11:00  quinn
  36  * Minimal.
  37  *
  38  * Revision 1.25  1995/11/13  09:27:22  quinn
  39  * Fiddling with the variant stuff.
  40  *
  41  * Revision 1.24  1995/10/30  12:41:13  quinn
  42  * Added hostname lookup for server.
  43  *
  44  * Revision 1.23  1995/10/18  16:12:30  quinn
  45  * Better diagnostics.
  46  *
  47  * Revision 1.22  1995/10/11  14:49:12  quinn
  48  * Smallish.
  49  *
  50  * Revision 1.21  1995/09/29  17:01:47  quinn
  51  * More Windows work
  52  *
  53  * Revision 1.20  1995/08/29  14:24:13  quinn
  54  * Added second half of close-handshake
  55  *
  56  * Revision 1.19  1995/08/29  11:17:28  quinn
  57  * Added code to receive close
  58  *
  59  * Revision 1.18  1995/08/28  12:21:27  quinn
  60  * Client can now ask for simple element set names.
  61  *
  62  * Revision 1.17  1995/08/17  12:45:02  quinn
  63  * Fixed minor problems with GRS-1. Added support in c&s.
  64  *
  65  * Revision 1.16  1995/08/15  12:00:04  quinn
  66  * Updated External
  67  *
  68  * Revision 1.15  1995/06/22  09:28:03  quinn
  69  * Fixed bug in SUTRS processing.
  70  *
  71  * Revision 1.14  1995/06/19  12:37:41  quinn
  72  * Added BER dumper.
  73  *
  74  * Revision 1.13  1995/06/16  10:29:11  quinn
  75  * *** empty log message ***
  76  *
  77  * Revision 1.12  1995/06/15  07:44:57  quinn
  78  * Moving to v3.
  79  *
  80  * Revision 1.11  1995/06/14  15:26:40  quinn
  81  * *** empty log message ***
  82  *
  83  * Revision 1.10  1995/06/06  14:56:58  quinn
  84  * Better diagnostics.
  85  *
  86  * Revision 1.9  1995/06/06  08:15:19  quinn
  87  * Cosmetic.
  88  *
  89  * Revision 1.8  1995/06/05  10:52:22  quinn
  90  * Added SCAN.
  91  *
  92  * Revision 1.7  1995/06/02  09:50:09  quinn
  93  * Smallish.
  94  *
  95  * Revision 1.6  1995/05/31  08:29:21  quinn
  96  * Nothing significant.
  97  *
  98  * Revision 1.5  1995/05/29  08:10:47  quinn
  99  * Moved oid.c to util.
 100  *
 101  * Revision 1.4  1995/05/22  15:30:13  adam
 102  * Client uses prefix query notation.
 103  *
 104  * Revision 1.3  1995/05/22  15:06:53  quinn
 105  * *** empty log message ***
 106  *
 107  * Revision 1.2  1995/05/22  14:56:40  quinn
 108  * *** empty log message ***
 109  *
 110  * Revision 1.1  1995/05/22  11:30:31  quinn
 111  * Added prettier client.
 112  *
 113  *
 114  */
 115
 116 /*
 117  * This is the obligatory little toy client, whose primary purpose is
 118  * to illustrate the use of the YAZ service-level API.
 119  */
 120
 121 #include <stdio.h>
 122 #include <stdlib.h>
 123 #ifdef WINDOWS
 124 #include <time.h>
 125 #else
 126 #include <sys/time.h>
 127 #endif
 128 #include <assert.h>
 129 #ifdef _AIX
 130 #include <sys/select.h>
 131 #endif
 132
 133 #include <comstack.h>
 134 #include <tcpip.h>
 135 #ifdef USE_XTIMOSI
 136 #include <xmosi.h>
 137 #endif
 138
 139 #include <proto.h>
 140 #include <marcdisp.h>
 141 #include <diagbib1.h>
 142
 143 #include <pquery.h>
 144
 145 #if CCL2RPN
 146 #include <yaz-ccl.h>
 147 #endif
 148
 149 #define C_PROMPT "Z> "
 150
 151 static ODR out, in, print;              /* encoding and decoding streams */
 152 static COMSTACK conn = 0;               /* our z-association */
 153 static Z_IdAuthentication *auth = 0;    /* our current auth definition */
 154 static char database[512] = "Default";  /* Database name */
 155 static int setnumber = 0;               /* current result set number */
 156 static int smallSetUpperBound = 0;
 157 static int largeSetLowerBound = 1;
 158 static int mediumSetPresentNumber = 0;
 159 static Z_ElementSetNames *elementSetNames = 0;
 160 static int setno = 1;                   /* current set offset */
 161 static int protocol = PROTO_Z3950;      /* current app protocol */
 162 static int recordsyntax = VAL_USMARC;
 163 static int sent_close = 0;
 164 static ODR_MEM session_mem;             /* memory handle for init-response */
 165 static Z_InitResponse *session = 0;     /* session parameters */
 166 static char last_scan[512] = "0";
 167 static char last_cmd[100] = "?";
 168 static oid_value attributeset = VAL_BIB1;
 169 static FILE *marcdump = 0;
 170 static char marcdump_file[512] = "marc.out";
 171
 172 typedef enum {
 173     QueryType_Prefix,
 174     QueryType_CCL,
 175     QueryType_CCL2RPN
 176 } QueryType;
 177
 178 static QueryType queryType = QueryType_Prefix;
 179
 180 #if CCL2RPN
 181 static CCL_bibset bibset;               /* CCL bibset handle */
 182 #endif
 183
 184 static void send_apdu(Z_APDU *a)
 185 {
 186     char *buf;
 187     int len;
 188
 189     if (!z_APDU(out, &a, 0))
 190     {
 191         odr_perror(out, "Encoding APDU");
 192         exit(1);
 193     }
 194     buf = odr_getbuf(out, &len, 0);
 195     odr_reset(out); /* release the APDU structure  */
 196     if (cs_put(conn, buf, len) < 0)
 197     {
 198         fprintf(stderr, "cs_put: %s", cs_errmsg(cs_errno(conn)));
 199         exit(1);
 200     }
 201 }
 202
 203 /* INIT SERVICE ------------------------------- */
 204
 205 static void send_initRequest()
 206 {
 207     Z_APDU *apdu = zget_APDU(out, Z_APDU_initRequest);
 208     Z_InitRequest *req = apdu->u.initRequest;
 209
 210     ODR_MASK_SET(req->options, Z_Options_search);
 211     ODR_MASK_SET(req->options, Z_Options_present);
 212     ODR_MASK_SET(req->options, Z_Options_namedResultSets);
 213     ODR_MASK_SET(req->options, Z_Options_triggerResourceCtrl);
 214     ODR_MASK_SET(req->options, Z_Options_scan);
 215
 216     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_1);
 217     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_2);
 218     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_3);
 219
 220     *req->maximumRecordSize = 1024*1024;
 221
 222     req->idAuthentication = auth;
 223
 224     send_apdu(apdu);
 225     printf("Sent initrequest.\n");
 226 }
 227
 228 static int process_initResponse(Z_InitResponse *res)
 229 {
 230     /* save session parameters for later use */
 231     session_mem = odr_extract_mem(in);
 232     session = res;
 233
 234     if (!*res->result)
 235         printf("Connection rejected by target.\n");
 236     else
 237         printf("Connection accepted by target.\n");
 238     if (res->implementationId)
 239         printf("ID     : %s\n", res->implementationId);
 240     if (res->implementationName)
 241         printf("Name   : %s\n", res->implementationName);
 242     if (res->implementationVersion)
 243         printf("Version: %s\n", res->implementationVersion);
 244     if (res->userInformationField)
 245     {
 246         printf("UserInformationfield:\n");
 247         if (!z_External(print, (Z_External**)&res-> userInformationField,
 248             0))
 249         {
 250             odr_perror(print, "Printing userinfo\n");
 251             odr_reset(print);
 252         }
 253         if (res->userInformationField->which == Z_External_octet)
 254         {
 255             printf("Guessing visiblestring:\n");
 256             printf("'%s'\n", res->userInformationField->u. octet_aligned->buf);
 257         }
 258     }
 259     return 0;
 260 }
 261
 262 int cmd_open(char *arg)
 263 {
 264     void *add;
 265     char type[100], addr[100];
 266     CS_TYPE t;
 267
 268     if (conn)
 269     {
 270         printf("Already connected.\n");
 271         return 0;
 272     }
 273     if (!*arg || sscanf(arg, "%[^:]:%s", type, addr) < 2)
 274     {
 275         fprintf(stderr, "Usage: open (osi|tcp) ':' [tsel '/']host[':'port]\n");
 276         return 0;
 277     }
 278 #ifdef USE_XTIMOSI
 279     if (!strcmp(type, "osi"))
 280     {
 281         if (!(add = mosi_strtoaddr(addr)))
 282         {
 283             perror(arg);
 284             return 0;
 285         }
 286         t = mosi_type;
 287         protocol = PROTO_SR;
 288     }
 289     else
 290 #endif
 291     if (!strcmp(type, "tcp"))
 292     {
 293         if (!(add = tcpip_strtoaddr(addr)))
 294         {
 295             perror(arg);
 296             return 0;
 297         }
 298         t = tcpip_type;
 299         protocol = PROTO_Z3950;
 300     }
 301     else
 302     {
 303         fprintf(stderr, "Bad type: %s\n", type);
 304         return 0;
 305     }
 306     if (!(conn = cs_create(t, 1, protocol)))
 307     {
 308         perror("cs_create");
 309         return 0;
 310     }
 311     printf("Connecting...");
 312     fflush(stdout);
 313     if (cs_connect(conn, add) < 0)
 314     {
 315         perror("connect");
 316         cs_close(conn);
 317         conn = 0;
 318         return 0;
 319     }
 320     printf("Ok.\n");
 321     send_initRequest();
 322     return 2;
 323 }
 324
 325 int cmd_authentication(char *arg)
 326 {
 327     static Z_IdAuthentication au;
 328     static char open[256];
 329
 330     if (!*arg)
 331     {
 332         printf("Auth field set to null\n");
 333         auth = 0;
 334         return 1;
 335     }
 336     auth = &au;
 337     au.which = Z_IdAuthentication_open;
 338     au.u.open = open;
 339     strcpy(open, arg);
 340     return 1;
 341 }
 342
 343 /* SEARCH SERVICE ------------------------------ */
 344
 345 static void display_variant(Z_Variant *v, int level)
 346 {
 347     int i;
 348
 349     for (i = 0; i < v->num_triples; i++)
 350     {
 351         printf("%*sclass=%d,type=%d", level * 4, "", *v->triples[i]->class,
 352             *v->triples[i]->type);
 353         if (v->triples[i]->which == Z_Triple_internationalString)
 354             printf(",value=%s\n", v->triples[i]->value.internationalString);
 355         else
 356             printf("\n");
 357     }
 358 }
 359
 360 static void display_grs1(Z_GenericRecord *r, int level)
 361 {
 362     int i;
 363
 364     if (!r)
 365         return;
 366     for (i = 0; i < r->num_elements; i++)
 367     {
 368         Z_TaggedElement *t;
 369
 370         printf("%*s", level * 4, "");
 371         t = r->elements[i];
 372         printf("(");
 373         if (t->tagType)
 374             printf("%d,", *t->tagType);
 375         else
 376             printf("?,");
 377         if (t->tagValue->which == Z_StringOrNumeric_numeric)
 378             printf("%d) ", *t->tagValue->u.numeric);
 379         else
 380             printf("%s) ", t->tagValue->u.string);
 381         if (t->content->which == Z_ElementData_subtree)
 382         {
 383             printf("\n");
 384             display_grs1(t->content->u.subtree, level+1);
 385         }
 386         else if (t->content->which == Z_ElementData_string)
 387             printf("%s\n", t->content->u.string);
 388         else if (t->content->which == Z_ElementData_numeric)
 389             printf("%d\n", *t->content->u.numeric);
 390         else if (t->content->which == Z_ElementData_oid)
 391         {
 392             int *ip = t->content->u.oid;
 393             oident *oent;
 394
 395             if ((oent = oid_getentbyoid(t->content->u.oid)))
 396                 printf("OID: %s\n", oent->desc);
 397             else
 398             {
 399                 printf("{");
 400                 while (ip && *ip >= 0)
 401                     printf(" %d", *(ip++));
 402                 printf(" }\n");
 403             }
 404         }
 405         else if (t->content->which == Z_ElementData_noDataRequested)
 406             printf("[No data requested]\n");
 407         else if (t->content->which == Z_ElementData_elementEmpty)
 408             printf("[Element empty]\n");
 409         else if (t->content->which == Z_ElementData_elementNotThere)
 410             printf("[Element not there]\n");
 411         else
 412             printf("??????\n");
 413         if (t->appliedVariant)
 414             display_variant(t->appliedVariant, level+1);
 415         if (t->metaData && t->metaData->supportedVariants)
 416         {
 417             int c;
 418
 419             printf("%*s---- variant list\n", (level+1)*4, "");
 420             for (c = 0; c < t->metaData->num_supportedVariants; c++)
 421             {
 422                 printf("%*svariant #%d\n", (level+1)*4, "", c);
 423                 display_variant(t->metaData->supportedVariants[c], level + 2);
 424             }
 425         }
 426     }
 427 }
 428
 429 static void display_record(Z_DatabaseRecord *p)
 430 {
 431     Z_External *r = (Z_External*) p;
 432     oident *ent = oid_getentbyoid(r->direct_reference);
 433
 434     /*
 435      * Tell the user what we got.
 436      */
 437     if (r->direct_reference)
 438     {
 439         printf("Record type: ");
 440         if (ent)
 441             printf("%s\n", ent->desc);
 442         else if (!odr_oid(print, &r->direct_reference, 0))
 443         {
 444             odr_perror(print, "print oid");
 445             odr_reset(print);
 446         }
 447     }
 448
 449     /* Check if this is a known, ASN.1 type tucked away in an octet string */
 450     if (ent && r->which == Z_External_octet)
 451     {
 452         Z_ext_typeent *type = z_ext_getentbyref(ent->value);
 453         void *rr;
 454
 455         if (type)
 456         {
 457             /*
 458              * Call the given decoder to process the record.
 459              */
 460             odr_setbuf(in, (char*)p->u.octet_aligned->buf,
 461                 p->u.octet_aligned->len, 0);
 462             if (!(*type->fun)(in, &rr, 0))
 463             {
 464                 odr_perror(in, "Decoding constructed record.");
 465                 fprintf(stderr, "[Near %d]\n", odr_offset(in));
 466                 fprintf(stderr, "Packet dump:\n---------\n");
 467                 odr_dumpBER(stderr, (char*)p->u.octet_aligned->buf,
 468                     p->u.octet_aligned->len);
 469                 fprintf(stderr, "---------\n");
 470                 exit(1);
 471             }
 472             /*
 473              * Note: we throw away the original, BER-encoded record here.
 474              * Do something else with it if you want to keep it.
 475              */
 476             r->u.sutrs = rr;    /* we don't actually check the type here. */
 477             r->which = type->what;
 478         }
 479     }
 480     if (r->which == Z_External_octet && p->u.octet_aligned->len)
 481         marc_display ((char*)p->u.octet_aligned->buf, stdout);
 482     else if (ent->value == VAL_SUTRS)
 483     {
 484         if (r->which != Z_External_sutrs)
 485         {
 486             printf("Expecting single SUTRS type for SUTRS.\n");
 487             return;
 488         }
 489         printf("%.*s", r->u.sutrs->len, r->u.sutrs->buf);
 490     }
 491     else if (ent->value == VAL_GRS1)
 492     {
 493         if (r->which != Z_External_grs1)
 494         {
 495             printf("Expecting single GRS type for GRS.\n");
 496             return;
 497         }
 498         display_grs1(r->u.grs1, 0);
 499     }
 500     else
 501     {
 502         printf("Unknown record representation.\n");
 503         if (!z_External(print, &r, 0))
 504         {
 505             odr_perror(print, "Printing external");
 506             odr_reset(print);
 507         }
 508     }
 509 }
 510
 511 static void display_diagrec(Z_DiagRec *p)
 512 {
 513     oident *ent;
 514 #ifdef Z_95
 515     Z_DefaultDiagFormat *r;
 516 #else
 517     Z_DiagRec *r = p;
 518 #endif
 519
 520     printf("Diagnostic message from database:\n");
 521 #ifdef Z_95
 522     if (p->which != Z_DiagRec_defaultFormat)
 523     {
 524         printf("Diagnostic record not in default format.\n");
 525         return;
 526     }
 527     else
 528         r = p->u.defaultFormat;
 529 #endif
 530     if (!(ent = oid_getentbyoid(r->diagnosticSetId)) ||
 531         ent->oclass != CLASS_DIAGSET || ent->value != VAL_BIB1)
 532         printf("Missing or unknown diagset\n");
 533     printf("    [%d] %s", *r->condition, diagbib1_str(*r->condition));
 534     if (r->addinfo && *r->addinfo)
 535         printf(" -- '%s'\n", r->addinfo);
 536     else
 537         printf("\n");
 538 }
 539
 540 static void display_nameplusrecord(Z_NamePlusRecord *p)
 541 {
 542     if (p->databaseName)
 543         printf("[%s]", p->databaseName);
 544     if (p->which == Z_NamePlusRecord_surrogateDiagnostic)
 545         display_diagrec(p->u.surrogateDiagnostic);
 546     else
 547         display_record(p->u.databaseRecord);
 548 }
 549
 550 static void display_records(Z_Records *p)
 551 {
 552     int i;
 553
 554     if (p->which == Z_Records_NSD)
 555         display_diagrec(p->u.nonSurrogateDiagnostic);
 556     else
 557     {
 558         printf("Records: %d\n", p->u.databaseOrSurDiagnostics->num_records);
 559         for (i = 0; i < p->u.databaseOrSurDiagnostics->num_records; i++)
 560             display_nameplusrecord(p->u.databaseOrSurDiagnostics->records[i]);
 561     }
 562 }
 563
 564 static int send_searchRequest(char *arg)
 565 {
 566     Z_APDU *apdu = zget_APDU(out, Z_APDU_searchRequest);
 567     Z_SearchRequest *req = apdu->u.searchRequest;
 568     char *databaseNames = database;
 569     Z_Query query;
 570 #if CCL2RPN
 571     struct ccl_rpn_node *rpn;
 572     int error, pos;
 573     oident bib1;
 574 #endif
 575     char setstring[100];
 576     Z_RPNQuery *RPNquery;
 577     Odr_oct ccl_query;
 578
 579 #if CCL2RPN
 580     if (queryType == QueryType_CCL2RPN)
 581     {
 582         rpn = ccl_find_str(bibset, arg, &error, &pos);
 583         if (error)
 584         {
 585             printf("CCL ERROR: %s\n", ccl_err_msg(error));
 586             return 0;
 587         }
 588     }
 589 #endif
 590
 591     if (!strcmp(arg, "@big")) /* strictly for troublemaking */
 592     {
 593         static unsigned char big[2100];
 594         static Odr_oct bigo;
 595
 596         /* send a very big referenceid to test transport stack etc. */
 597         memset(big, 'A', 2100);
 598         bigo.len = bigo.size = 2100;
 599         bigo.buf = big;
 600         req->referenceId = &bigo;
 601     }
 602
 603     if (setnumber >= 0)
 604     {
 605         sprintf(setstring, "%d", ++setnumber);
 606         req->resultSetName = setstring;
 607     }
 608     *req->smallSetUpperBound = smallSetUpperBound;
 609     *req->largeSetLowerBound = largeSetLowerBound;
 610     *req->mediumSetPresentNumber = mediumSetPresentNumber;
 611     if (smallSetUpperBound > 0 || (largeSetLowerBound > 1 &&
 612         mediumSetPresentNumber > 0))
 613     {
 614         oident prefsyn;
 615
 616         prefsyn.proto = protocol;
 617         prefsyn.oclass = CLASS_RECSYN;
 618         prefsyn.value = recordsyntax;
 619         req->preferredRecordSyntax =
 620             odr_oiddup(out, oid_getoidbyent(&prefsyn));
 621         req->smallSetElementSetNames =
 622             req->mediumSetElementSetNames = elementSetNames;
 623     }
 624     req->num_databaseNames = 1;
 625     req->databaseNames = &databaseNames;
 626
 627     req->query = &query;
 628
 629     switch (queryType)
 630     {
 631     case QueryType_Prefix:
 632         query.which = Z_Query_type_1;
 633         RPNquery = p_query_rpn (out, protocol, arg);
 634         if (!RPNquery)
 635         {
 636             printf("Prefix query error\n");
 637             return 0;
 638         }
 639         query.u.type_1 = RPNquery;
 640         break;
 641     case QueryType_CCL:
 642         query.which = Z_Query_type_2;
 643         query.u.type_2 = &ccl_query;
 644         ccl_query.buf = (unsigned char*) arg;
 645         ccl_query.len = strlen(arg);
 646         break;
 647 #if CCL2RPN
 648     case QueryType_CCL2RPN:
 649         query.which = Z_Query_type_1;
 650         assert((RPNquery = ccl_rpn_query(rpn)));
 651         bib1.proto = protocol;
 652         bib1.oclass = CLASS_ATTSET;
 653         bib1.value = VAL_BIB1;
 654         RPNquery->attributeSetId = oid_getoidbyent(&bib1);
 655         query.u.type_1 = RPNquery;
 656         break;
 657 #endif
 658     default:
 659         printf ("Unsupported query type\n");
 660         return 0;
 661     }
 662     send_apdu(apdu);
 663     setno = 1;
 664     printf("Sent searchRequest.\n");
 665     return 2;
 666 }
 667
 668 static int process_searchResponse(Z_SearchResponse *res)
 669 {
 670     if (*res->searchStatus)
 671         printf("Search was a success.\n");
 672     else
 673         printf("Search was a bloomin' failure.\n");
 674     printf("Number of hits: %d, setno %d\n",
 675         *res->resultCount, setnumber);
 676     printf("records returned: %d\n",
 677         *res->numberOfRecordsReturned);
 678     setno += *res->numberOfRecordsReturned;
 679     if (res->records)
 680         display_records(res->records);
 681     return 0;
 682 }
 683
 684 static int cmd_find(char *arg)
 685 {
 686     if (!*arg)
 687     {
 688         printf("Find what?\n");
 689         return 0;
 690     }
 691     if (!conn)
 692     {
 693         printf("Not connected yet\n");
 694         return 0;
 695     }
 696     if (!send_searchRequest(arg))
 697         return 0;
 698     return 2;
 699 }
 700
 701 static int cmd_ssub(char *arg)
 702 {
 703     if (!(smallSetUpperBound = atoi(arg)))
 704         return 0;
 705     return 1;
 706 }
 707
 708 static int cmd_lslb(char *arg)
 709 {
 710     if (!(largeSetLowerBound = atoi(arg)))
 711         return 0;
 712     return 1;
 713 }
 714
 715 static int cmd_mspn(char *arg)
 716 {
 717     if (!(mediumSetPresentNumber = atoi(arg)))
 718         return 0;
 719     return 1;
 720 }
 721
 722 static int cmd_status(char *arg)
 723 {
 724     printf("smallSetUpperBound: %d\n", smallSetUpperBound);
 725     printf("largeSetLowerBound: %d\n", largeSetLowerBound);
 726     printf("mediumSetPresentNumber: %d\n", mediumSetPresentNumber);
 727     return 1;
 728 }
 729
 730 static int cmd_base(char *arg)
 731 {
 732     if (!*arg)
 733     {
 734         printf("Usage: base <database>\n");
 735         return 0;
 736     }
 737     strcpy(database, arg);
 738     return 1;
 739 }
 740
 741 static int cmd_setnames(char *arg)
 742 {
 743     if (setnumber < 0)
 744     {
 745         printf("Set numbering enabled.\n");
 746         setnumber = 0;
 747     }
 748     else
 749     {
 750         printf("Set numbering disabled.\n");
 751         setnumber = -1;
 752     }
 753     return 1;
 754 }
 755
 756 /* PRESENT SERVICE ----------------------------- */
 757
 758 static int send_presentRequest(char *arg)
 759 {
 760     Z_APDU *apdu = zget_APDU(out, Z_APDU_presentRequest);
 761     Z_PresentRequest *req = apdu->u.presentRequest;
 762     Z_RecordComposition compo;
 763     oident prefsyn;
 764     int nos = 1;
 765     char *p;
 766     char setstring[100];
 767
 768     if ((p = strchr(arg, '+')))
 769     {
 770         nos = atoi(p + 1);
 771         *p = 0;
 772     }
 773     if (*arg)
 774         setno = atoi(arg);
 775
 776     if (setnumber >= 0)
 777     {
 778         sprintf(setstring, "%d", setnumber);
 779         req->resultSetId = setstring;
 780     }
 781     req->resultSetStartPoint = &setno;
 782     req->numberOfRecordsRequested = &nos;
 783     prefsyn.proto = protocol;
 784     prefsyn.oclass = CLASS_RECSYN;
 785     prefsyn.value = recordsyntax;
 786     req->preferredRecordSyntax = oid_getoidbyent(&prefsyn);
 787     if (elementSetNames)
 788     {
 789         req->recordComposition = &compo;
 790         compo.which = Z_RecordComp_simple;
 791         compo.u.simple = elementSetNames;
 792     }
 793     send_apdu(apdu);
 794     printf("Sent presentRequest (%d+%d).\n", setno, nos);
 795     return 2;
 796 }
 797
 798 void process_close(Z_Close *req)
 799 {
 800     Z_APDU *apdu = zget_APDU(out, Z_APDU_close);
 801     Z_Close *res = apdu->u.close;
 802
 803     static char *reasons[] =
 804     {
 805         "finished",
 806         "shutdown",
 807         "system problem",
 808         "cost limit reached",
 809         "resources",
 810         "security violation",
 811         "protocolError",
 812         "lack of activity",
 813         "peer abort",
 814         "unspecified"
 815     };
 816
 817     printf("Reason: %s, message: %s\n", reasons[*req->closeReason],
 818         req->diagnosticInformation ? req->diagnosticInformation : "NULL");
 819     if (sent_close)
 820     {
 821         printf("Goodbye.\n");
 822         exit(0);
 823     }
 824     *res->closeReason = Z_Close_finished;
 825     send_apdu(apdu);
 826     printf("Sent response.\n");
 827     sent_close = 1;
 828 }
 829
 830 static int cmd_show(char *arg)
 831 {
 832     if (!send_presentRequest(arg))
 833         return 0;
 834     return 2;
 835 }
 836
 837 int cmd_quit(char *arg)
 838 {
 839     printf("See you later, alligator.\n");
 840     exit(0);
 841 }
 842
 843 int cmd_cancel(char *arg)
 844 {
 845     Z_APDU *apdu = zget_APDU(out, Z_APDU_triggerResourceControlRequest);
 846     Z_TriggerResourceControlRequest *req =
 847         apdu->u.triggerResourceControlRequest;
 848     bool_t false = 0;
 849
 850     if (!session)
 851     {
 852         printf("Session not initialized yet\n");
 853         return 0;
 854     }
 855     if (!ODR_MASK_GET(session->options, Z_Options_triggerResourceCtrl))
 856     {
 857         printf("Target doesn't support cancel (trigger resource ctrl)\n");
 858         return 0;
 859     }
 860     *req->requestedAction = Z_TriggerResourceCtrl_cancel;
 861     req->resultSetWanted = &false;
 862
 863     send_apdu(apdu);
 864     printf("Sent cancel request\n");
 865     return 2;
 866 }
 867
 868 int send_scanrequest(char *string, int pp, int num)
 869 {
 870     Z_APDU *apdu = zget_APDU(out, Z_APDU_scanRequest);
 871     Z_ScanRequest *req = apdu->u.scanRequest;
 872     char *db = database;
 873
 874     req->num_databaseNames = 1;
 875     req->databaseNames = &db;
 876     req->termListAndStartPoint = p_query_scan(out, protocol,
 877                                               &req->attributeSet, string);
 878     req->numberOfTermsRequested = &num;
 879     req->preferredPositionInResponse = &pp;
 880     send_apdu(apdu);
 881     return 2;
 882 }
 883
 884 void display_term(Z_TermInfo *t)
 885 {
 886     if (t->term->which == Z_Term_general)
 887     {
 888         printf("%.*s (%d)\n", t->term->u.general->len, t->term->u.general->buf,
 889             t->globalOccurrences ? *t->globalOccurrences : -1);
 890         sprintf(last_scan, "%.*s", t->term->u.general->len,
 891             t->term->u.general->buf);
 892     }
 893     else
 894         printf("Term type not general.\n");
 895 }
 896
 897 void process_scanResponse(Z_ScanResponse *res)
 898 {
 899     int i;
 900
 901     printf("SCAN: %d entries, position=%d\n", *res->numberOfEntriesReturned,
 902         *res->positionOfTerm);
 903     if (*res->scanStatus != Z_Scan_success)
 904         printf("Scan returned code %d\n", *res->scanStatus);
 905     if (!res->entries)
 906         return;
 907     if (res->entries->which == Z_ListEntries_entries)
 908     {
 909         Z_Entries *ent = res->entries->u.entries;
 910
 911         for (i = 0; i < ent->num_entries; i++)
 912             if (ent->entries[i]->which == Z_Entry_termInfo)
 913             {
 914                 printf("%c ", i + 1 == *res->positionOfTerm ? '*' : ' ');
 915                 display_term(ent->entries[i]->u.termInfo);
 916             }
 917             else
 918                 display_diagrec(ent->entries[i]->u.surrogateDiagnostic);
 919     }
 920     else
 921         display_diagrec(res->entries->u.nonSurrogateDiagnostics->diagRecs[0]);
 922 }
 923
 924 int cmd_scan(char *arg)
 925 {
 926     if (!session)
 927     {
 928         printf("Session not initialized yet\n");
 929         return 0;
 930     }
 931     if (!ODR_MASK_GET(session->options, Z_Options_scan))
 932     {
 933         printf("Target doesn't support scan\n");
 934         return 0;
 935     }
 936     if (*arg)
 937     {
 938         if (send_scanrequest(arg, 5, 20) < 0)
 939             return 0;
 940     }
 941     else
 942         if (send_scanrequest(last_scan, 1, 20) < 0)
 943             return 0;
 944     return 2;
 945 }
 946
 947 int cmd_format(char *arg)
 948 {
 949     if (!arg || !*arg)
 950     {
 951         printf("Usage: format <recordsyntax>\n");
 952         return 0;
 953     }
 954     if (!strcmp(arg, "sutrs"))
 955     {
 956         printf("Preferred format is SUTRS.\n");
 957         recordsyntax = VAL_SUTRS;
 958         return 1;
 959     }
 960     else if (!strcmp(arg, "usmarc"))
 961     {
 962         printf("Preferred format is USMARC\n");
 963         recordsyntax = VAL_USMARC;
 964         return 1;
 965     }
 966     else if (!strcmp(arg, "danmarc"))
 967     {
 968         printf("Preferred format is DANMARC\n");
 969         recordsyntax = VAL_DANMARC;
 970         return 1;
 971     }
 972     else if (!strcmp(arg, "grs1"))
 973     {
 974         printf("Preferred format is GRS1\n");
 975         recordsyntax = VAL_GRS1;
 976         return 1;
 977     }
 978     else if (!strcmp(arg, "explain"))
 979     {
 980         printf("Preferred format is Explain\n");
 981         recordsyntax = VAL_EXPLAIN;
 982         return 1;
 983     }
 984     else
 985     {
 986         printf("Specify one of {sutrs,usmarc,danmarc,grs1}.\n");
 987         return 0;
 988     }
 989 }
 990
 991 int cmd_elements(char *arg)
 992 {
 993     static Z_ElementSetNames esn;
 994     static char what[100];
 995
 996     if (!arg || !*arg)
 997     {
 998         printf("Usage: elements <esn>\n");
 999         return 0;
1000     }
1001     strcpy(what, arg);
1002     esn.which = Z_ElementSetNames_generic;
1003     esn.u.generic = what;
1004     elementSetNames = &esn;
1005     return 1;
1006 }
1007
1008 int cmd_attributeset(char *arg)
1009 {
1010     char what[100];
1011     oid_value v;
1012
1013     if (!arg || !*arg)
1014     {
1015         printf("Usage: attributeset <setname>\n");
1016         return 0;
1017     }
1018     sscanf(arg, "%s", what);
1019     if ((v = oid_getvalbyname(what)) == VAL_NONE)
1020     {
1021         printf("Unknown attribute set name\n");
1022         return 0;
1023     }
1024     attributeset = v;
1025     return 1;
1026 }
1027
1028 int cmd_querytype (char *arg)
1029 {
1030     if (!strcmp (arg, "ccl"))
1031         queryType = QueryType_CCL;
1032     else if (!strcmp (arg, "prefix"))
1033         queryType = QueryType_Prefix;
1034 #if CCL2RPN
1035     else if (!strcmp (arg, "ccl2rpn") || !strcmp (arg, "cclrpn"))
1036         queryType = QueryType_CCL2RPN;
1037 #endif
1038     else
1039     {
1040         printf ("Querytype must be one of:\n");
1041         printf (" prefix         - Prefix query\n");
1042         printf (" ccl            - CCL query\n");
1043 #if CCL2RPN
1044         printf (" ccl2rpn        - CCL query converted to RPN\n");
1045 #endif
1046         return 0;
1047     }
1048     return 1;
1049 }
1050
1051 int cmd_close(char *arg)
1052 {
1053     Z_APDU *apdu = zget_APDU(out, Z_APDU_close);
1054     Z_Close *req = apdu->u.close;
1055
1056     *req->closeReason = Z_Close_finished;
1057     send_apdu(apdu);
1058     printf("Sent close request.\n");
1059     sent_close = 1;
1060     return 2;
1061 }
1062
1063 static void initialize(void)
1064 {
1065 #if CCL2RPN
1066     FILE *inf;
1067 #endif
1068
1069     if (!(out = odr_createmem(ODR_ENCODE)) ||
1070         !(in = odr_createmem(ODR_DECODE)) ||
1071         !(print = odr_createmem(ODR_PRINT)))
1072     {
1073         fprintf(stderr, "failed to allocate ODR streams\n");
1074         exit(1);
1075     }
1076     setvbuf(stdout, 0, _IONBF, 0);
1077
1078 #if CCL2RPN
1079     bibset = ccl_qual_mk ();
1080     inf = fopen ("default.bib", "r");
1081     if (inf)
1082     {
1083         ccl_qual_file (bibset, inf);
1084         fclose (inf);
1085     }
1086 #endif
1087 }
1088
1089 static int client(void)
1090 {
1091     static struct {
1092         char *cmd;
1093         int (*fun)(char *arg);
1094         char *ad;
1095     } cmd[] = {
1096         {"open", cmd_open, "('tcp'|'osi')':'[<tsel>'/']<host>[':'<port>]"},
1097         {"quit", cmd_quit, ""},
1098         {"find", cmd_find, "<query>"},
1099         {"base", cmd_base, "<base-name>"},
1100         {"show", cmd_show, "<rec#>['+'<#recs>]"},
1101         {"scan", cmd_scan, "<term>"},
1102         {"authentication", cmd_authentication, "<acctstring>"},
1103         {"lslb", cmd_lslb, "<largeSetLowerBound>"},
1104         {"ssub", cmd_ssub, "<smallSetUpperBound>"},
1105         {"mspn", cmd_mspn, "<mediumSetPresentNumber>"},
1106         {"status", cmd_status, ""},
1107         {"setnames", cmd_setnames, ""},
1108         {"cancel", cmd_cancel, ""},
1109         {"format", cmd_format, "<recordsyntax>"},
1110         {"elements", cmd_elements, "<elementSetName>"},
1111         {"close", cmd_close, ""},
1112         {"attributeset", cmd_attributeset, "<attrset>"},
1113         {"querytype", cmd_querytype, "<type>"},
1114         {0,0}
1115     };
1116     char *netbuffer= 0;
1117     int netbufferlen = 0;
1118     int i;
1119     Z_APDU *apdu;
1120
1121     while (1)
1122     {
1123         int res;
1124         fd_set input;
1125         char line[1024], word[1024], arg[1024];
1126
1127         FD_ZERO(&input);
1128         FD_SET(0, &input);
1129         if (conn)
1130             FD_SET(cs_fileno(conn), &input);
1131         if ((res = select(20, &input, 0, 0, 0)) < 0)
1132         {
1133             perror("select");
1134             exit(1);
1135         }
1136         if (!res)
1137             continue;
1138         if (FD_ISSET(0, &input))
1139         {
1140             /* quick & dirty way to get a command line. */
1141             if (!gets(line))
1142                 break;
1143             if ((res = sscanf(line, "%s %[^;]", word, arg)) <= 0)
1144             {
1145                 strcpy(word, last_cmd);
1146                 *arg = '\0';
1147             }
1148             else if (res == 1)
1149                 *arg = 0;
1150             strcpy(last_cmd, word);
1151             for (i = 0; cmd[i].cmd; i++)
1152                 if (!strncmp(cmd[i].cmd, word, strlen(word)))
1153                 {
1154                     res = (*cmd[i].fun)(arg);
1155                     break;
1156                 }
1157             if (!cmd[i].cmd) /* dump our help-screen */
1158             {
1159                 printf("Unknown command: %s.\n", word);
1160                 printf("Currently recognized commands:\n");
1161                 for (i = 0; cmd[i].cmd; i++)
1162                     printf("   %s %s\n", cmd[i].cmd, cmd[i].ad);
1163                 res = 1;
1164             }
1165             if (res < 2)
1166                 printf(C_PROMPT);
1167         }
1168         if (conn && FD_ISSET(cs_fileno(conn), &input))
1169         {
1170             do
1171             {
1172                 if ((res = cs_get(conn, &netbuffer, &netbufferlen)) < 0)
1173                 {
1174                     perror("cs_get");
1175                     exit(1);
1176                 }
1177                 if (!res)
1178                 {
1179                     printf("Target closed connection.\n");
1180                     exit(1);
1181                 }
1182                 odr_reset(in); /* release APDU from last round */
1183                 odr_setbuf(in, netbuffer, res, 0);
1184                 if (!z_APDU(in, &apdu, 0))
1185                 {
1186                     odr_perror(in, "Decoding incoming APDU");
1187                     fprintf(stderr, "[Near %d]\n", odr_offset(in));
1188                     fprintf(stderr, "Packet dump:\n---------\n");
1189                     odr_dumpBER(stderr, netbuffer, res);
1190                     fprintf(stderr, "---------\n");
1191                     exit(1);
1192                 }
1193 #if 0
1194                 if (!z_APDU(print, &apdu, 0))
1195                 {
1196                     odr_perror(print, "Failed to print incoming APDU");
1197                     odr_reset(print);
1198                     continue;
1199                 }
1200 #endif
1201                 switch(apdu->which)
1202                 {
1203                     case Z_APDU_initResponse:
1204                         process_initResponse(apdu->u.initResponse);
1205                         break;
1206                     case Z_APDU_searchResponse:
1207                         process_searchResponse(apdu->u.searchResponse);
1208                         break;
1209                     case Z_APDU_scanResponse:
1210                         process_scanResponse(apdu->u.scanResponse);
1211                         break;
1212                     case Z_APDU_presentResponse:
1213                         printf("Received presentResponse.\n");
1214                         setno +=
1215                             *apdu->u.presentResponse->numberOfRecordsReturned;
1216                         if (apdu->u.presentResponse->records)
1217                             display_records(apdu->u.presentResponse->records);
1218                         else
1219                             printf("No records.\n");
1220                         break;
1221                     case Z_APDU_close:
1222                         printf("Target has closed the association.\n");
1223                         process_close(apdu->u.close);
1224                         break;
1225                     default:
1226                         printf("Received unknown APDU type (%d).\n",
1227                             apdu->which);
1228                         exit(1);
1229                 }
1230                 printf(C_PROMPT);
1231                 fflush(stdout);
1232             }
1233             while (cs_more(conn));
1234         }
1235     }
1236     return 0;
1237 }
1238
1239 int main(int argc, char **argv)
1240 {
1241     initialize();
1242     if (argc > 1)
1243         cmd_open(argv[1]);
1244     else
1245         printf(C_PROMPT);
1246     if (*marcdump_file && !(marcdump = fopen(marcdump_file, "a")))
1247     {
1248         perror(marcdump_file);
1249         exit(1);
1250     }
1251     return client();
1252 }