client/client.c

   1 /*
   2  * Copyright (c) 1995-2002, Index Data
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: client.c,v 1.142 2002-02-12 21:12:39 adam Exp $
   6  */
   7
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <time.h>
  11
  12 #include <yaz/yaz-util.h>
  13
  14 #include <yaz/tcpip.h>
  15 #ifdef USE_XTIMOSI
  16 #include <yaz/xmosi.h>
  17 #endif
  18
  19 #include <yaz/proto.h>
  20 #include <yaz/marcdisp.h>
  21 #include <yaz/diagbib1.h>
  22 #include <yaz/otherinfo.h>
  23
  24 #include <yaz/pquery.h>
  25 #include <yaz/sortspec.h>
  26
  27 #if YAZ_MODULE_ill
  28 #include <yaz/ill.h>
  29 #endif
  30
  31 #if YAZ_MODULE_ccl
  32 #include <yaz/yaz-ccl.h>
  33 #endif
  34
  35 #if HAVE_READLINE_READLINE_H
  36 #include <readline/readline.h>
  37 #endif
  38 #if HAVE_READLINE_HISTORY_H
  39 #include <readline/history.h>
  40 #endif
  41
  42 #include <sys/stat.h>
  43
  44
  45 #include "admin.h"
  46 #include "tabcomplete.h"
  47
  48 #define C_PROMPT "Z> "
  49
  50 static ODR out, in, print;              /* encoding and decoding streams */
  51 static FILE *apdu_file = 0;
  52 static COMSTACK conn = 0;               /* our z-association */
  53 static Z_IdAuthentication *auth = 0;    /* our current auth definition */
  54 char *databaseNames[128];
  55 int num_databaseNames = 0;
  56 static Z_External *record_last = 0;
  57 static int setnumber = -1;               /* current result set number */
  58 static int smallSetUpperBound = 0;
  59 static int largeSetLowerBound = 1;
  60 static int mediumSetPresentNumber = 0;
  61 static Z_ElementSetNames *elementSetNames = 0;
  62 static int setno = 1;                   /* current set offset */
  63 static enum oid_proto protocol = PROTO_Z3950;      /* current app protocol */
  64 static enum oid_value recordsyntax = VAL_USMARC;
  65 static enum oid_value schema = VAL_NONE;
  66 static int sent_close = 0;
  67 static NMEM session_mem = NULL;      /* memory handle for init-response */
  68 static Z_InitResponse *session = 0;     /* session parameters */
  69 static char last_scan_line[512] = "0";
  70 static char last_scan_query[512] = "0";
  71 static char ccl_fields[512] = "default.bib";
  72 static char* esPackageName = 0;
  73 static char* yazProxy = 0;
  74 static int kilobytes = 1024;
  75
  76 static char last_cmd[32] = "?";
  77 static FILE *marcdump = 0;
  78 static char *refid = NULL;
  79
  80 typedef enum {
  81     QueryType_Prefix,
  82     QueryType_CCL,
  83     QueryType_CCL2RPN
  84 } QueryType;
  85
  86 static QueryType queryType = QueryType_Prefix;
  87
  88 #if YAZ_MODULE_ccl
  89 static CCL_bibset bibset;               /* CCL bibset handle */
  90 #endif
  91
  92
  93 /* set this one to 1, to avoid decode of unknown MARCs  */
  94 #define AVOID_MARC_DECODE 1
  95
  96 /* nice helper macro as extensive tabbing gives spaces at the en of the args lines */
  97 #define REMOVE_TAILING_BLANKS(a) {\
  98   char* args_end=(a)+strlen(a)-1; \
  99   while(isspace(*args_end)) {*args_end=0;--args_end;}; \
 100   }
 101
 102
 103 void process_cmd_line(char* line);
 104 char ** readline_completer(char *text, int start, int end);
 105 char *command_generator(const char *text, int state);
 106
 107
 108 ODR getODROutputStream()
 109 {
 110     return out;
 111 }
 112
 113 void send_apdu(Z_APDU *a)
 114 {
 115     char *buf;
 116     int len;
 117
 118     if (apdu_file)
 119     {
 120         z_APDU(print, &a, 0, 0);
 121         odr_reset(print);
 122     }
 123     if (!z_APDU(out, &a, 0, 0))
 124     {
 125         odr_perror(out, "Encoding APDU");
 126         exit(1);
 127     }
 128     buf = odr_getbuf(out, &len, 0);
 129     /* printf ("sending APDU of size %d\n", len); */
 130     if (cs_put(conn, buf, len) < 0)
 131     {
 132         fprintf(stderr, "cs_put: %s", cs_errmsg(cs_errno(conn)));
 133         exit(1);
 134     }
 135     odr_reset(out); /* release the APDU structure  */
 136 }
 137
 138 static void print_stringn(const unsigned char *buf, size_t len)
 139 {
 140    size_t i;
 141    for (i = 0; i<len; i++)
 142        if ((buf[i] <= 126 && buf[i] >= 32) || strchr ("\n\r\t\f", buf[i]))
 143            fputc (buf[i], stdout);
 144        else
 145            printf ("\\X%02X", buf[i]);
 146 }
 147
 148 static void print_refid (Z_ReferenceId *id)
 149 {
 150     if (id)
 151     {
 152         printf ("Reference Id: ");
 153         print_stringn (id->buf, id->len);
 154         printf ("\n");
 155     }
 156 }
 157
 158 static Z_ReferenceId *set_refid (ODR out)
 159 {
 160     Z_ReferenceId *id;
 161     if (!refid)
 162         return 0;
 163     id = (Z_ReferenceId *) odr_malloc (out, sizeof(*id));
 164     id->size = id->len = strlen(refid);
 165     id->buf = (unsigned char *) odr_malloc (out, id->len);
 166     memcpy (id->buf, refid, id->len);
 167     return id;
 168 }
 169
 170 /* INIT SERVICE ------------------------------- */
 171
 172 static void send_initRequest(const char* type_and_host)
 173 {
 174     Z_APDU *apdu = zget_APDU(out, Z_APDU_initRequest);
 175     Z_InitRequest *req = apdu->u.initRequest;
 176
 177     ODR_MASK_SET(req->options, Z_Options_search);
 178     ODR_MASK_SET(req->options, Z_Options_present);
 179     ODR_MASK_SET(req->options, Z_Options_namedResultSets);
 180     ODR_MASK_SET(req->options, Z_Options_triggerResourceCtrl);
 181     ODR_MASK_SET(req->options, Z_Options_scan);
 182     ODR_MASK_SET(req->options, Z_Options_sort);
 183     ODR_MASK_SET(req->options, Z_Options_extendedServices);
 184     ODR_MASK_SET(req->options, Z_Options_delSet);
 185
 186     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_1);
 187     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_2);
 188     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_3);
 189
 190     *req->maximumRecordSize = 1024*kilobytes;
 191     *req->preferredMessageSize = 1024*kilobytes;
 192
 193     req->idAuthentication = auth;
 194
 195     req->referenceId = set_refid (out);
 196
 197     if (yazProxy)
 198         yaz_oi_set_string_oidval(&req->otherInfo, out, VAL_PROXY,
 199         1, type_and_host);
 200
 201     send_apdu(apdu);
 202     printf("Sent initrequest.\n");
 203 }
 204
 205 static int process_initResponse(Z_InitResponse *res)
 206 {
 207     /* save session parameters for later use */
 208     session_mem = odr_extract_mem(in);
 209     session = res;
 210
 211     if (!*res->result)
 212         printf("Connection rejected by target.\n");
 213     else
 214         printf("Connection accepted by target.\n");
 215     if (res->implementationId)
 216         printf("ID     : %s\n", res->implementationId);
 217     if (res->implementationName)
 218         printf("Name   : %s\n", res->implementationName);
 219     if (res->implementationVersion)
 220         printf("Version: %s\n", res->implementationVersion);
 221     if (res->userInformationField)
 222     {
 223         printf("UserInformationfield:\n");
 224         if (!z_External(print, (Z_External**)&res-> userInformationField,
 225             0, 0))
 226         {
 227             odr_perror(print, "Printing userinfo\n");
 228             odr_reset(print);
 229         }
 230         if (res->userInformationField->which == Z_External_octet)
 231         {
 232             printf("Guessing visiblestring:\n");
 233             printf("'%s'\n", res->userInformationField->u. octet_aligned->buf);
 234         }
 235         odr_reset (print);
 236     }
 237     printf ("Options:");
 238     if (ODR_MASK_GET(res->options, Z_Options_search))
 239         printf (" search");
 240     if (ODR_MASK_GET(res->options, Z_Options_present))
 241         printf (" present");
 242     if (ODR_MASK_GET(res->options, Z_Options_delSet))
 243         printf (" delSet");
 244     if (ODR_MASK_GET(res->options, Z_Options_resourceReport))
 245         printf (" resourceReport");
 246     if (ODR_MASK_GET(res->options, Z_Options_resourceCtrl))
 247         printf (" resourceCtrl");
 248     if (ODR_MASK_GET(res->options, Z_Options_accessCtrl))
 249         printf (" accessCtrl");
 250     if (ODR_MASK_GET(res->options, Z_Options_scan))
 251         printf (" scan");
 252     if (ODR_MASK_GET(res->options, Z_Options_sort))
 253         printf (" sort");
 254     if (ODR_MASK_GET(res->options, Z_Options_extendedServices))
 255         printf (" extendedServices");
 256     if (ODR_MASK_GET(res->options, Z_Options_level_1Segmentation))
 257         printf (" level1Segmentation");
 258     if (ODR_MASK_GET(res->options, Z_Options_level_2Segmentation))
 259         printf (" level2Segmentation");
 260     if (ODR_MASK_GET(res->options, Z_Options_concurrentOperations))
 261         printf (" concurrentOperations");
 262     if (ODR_MASK_GET(res->options, Z_Options_namedResultSets))
 263     {
 264         printf (" namedResultSets");
 265         setnumber = 0;
 266     }
 267     if (ODR_MASK_GET(res->options, Z_Options_encapsulation))
 268         printf (" encapsulation");
 269     if (ODR_MASK_GET(res->options, Z_Options_resultCount))
 270         printf (" resultCount");
 271     if (ODR_MASK_GET(res->options, Z_Options_negotiationModel))
 272         printf (" negotiationModel");
 273     if (ODR_MASK_GET(res->options, Z_Options_duplicateDetection))
 274         printf (" duplicateDetection");
 275     if (ODR_MASK_GET(res->options, Z_Options_queryType104))
 276         printf (" queryType104");
 277     printf ("\n");
 278     fflush (stdout);
 279     return 0;
 280 }
 281
 282 static int cmd_base(char *arg)
 283 {
 284     int i;
 285     char *cp;
 286
 287     if (!*arg)
 288     {
 289         printf("Usage: base <database> <database> ...\n");
 290         return 0;
 291     }
 292     for (i = 0; i<num_databaseNames; i++)
 293         xfree (databaseNames[i]);
 294     num_databaseNames = 0;
 295     while (1)
 296     {
 297         if (!(cp = strchr(arg, ' ')))
 298             cp = arg + strlen(arg);
 299         if (cp - arg < 1)
 300             break;
 301         databaseNames[num_databaseNames] = (char *)xmalloc (1 + cp - arg);
 302         memcpy (databaseNames[num_databaseNames], arg, cp - arg);
 303         databaseNames[num_databaseNames++][cp - arg] = '\0';
 304         if (!*cp)
 305             break;
 306         arg = cp+1;
 307     }
 308     return 1;
 309 }
 310
 311
 312 int cmd_open(char *arg)
 313 {
 314     void *add;
 315     char type_and_host[101], base[101];
 316     CS_TYPE t;
 317
 318     if (conn)
 319     {
 320         printf("Already connected.\n");
 321
 322         cs_close (conn);
 323         conn = NULL;
 324         if (session_mem)
 325         {
 326             nmem_destroy (session_mem);
 327             session_mem = NULL;
 328         }
 329     }
 330     t = tcpip_type;
 331     base[0] = '\0';
 332     if (sscanf (arg, "%100[^/]/%100s", type_and_host, base) < 1)
 333         return 0;
 334
 335     if(yazProxy)
 336     conn = cs_create_host(yazProxy, 1, &add);
 337     else
 338     conn = cs_create_host(type_and_host, 1, &add);
 339
 340     if (!conn)
 341     {
 342     printf ("Couldn't create comstack\n");
 343     return 0;
 344     }
 345     printf("Connecting...");
 346     fflush(stdout);
 347     if (cs_connect(conn, add) < 0)
 348     {
 349     printf ("error = %s\n", cs_strerror(conn));
 350     if (conn->cerrno == CSYSERR)
 351         perror("system");
 352         cs_close(conn);
 353         conn = 0;
 354         return 0;
 355     }
 356     printf("Ok.\n");
 357
 358     send_initRequest(type_and_host);
 359     if (*base)
 360         cmd_base (base);
 361     return 2;
 362 }
 363
 364 int cmd_authentication(char *arg)
 365 {
 366     static Z_IdAuthentication au;
 367     static char user[40], group[40], pass[40];
 368     static Z_IdPass idPass;
 369     int r;
 370
 371     if (!*arg)
 372     {
 373         printf("Auth field set to null\n");
 374         auth = 0;
 375         return 1;
 376     }
 377     r = sscanf (arg, "%39s %39s %39s", user, group, pass);
 378     if (r == 0)
 379     {
 380         printf("Auth field set to null\n");
 381         auth = 0;
 382     }
 383     if (r == 1)
 384     {
 385         auth = &au;
 386         au.which = Z_IdAuthentication_open;
 387         au.u.open = user;
 388     }
 389     if (r == 3)
 390     {
 391         auth = &au;
 392         au.which = Z_IdAuthentication_idPass;
 393         au.u.idPass = &idPass;
 394         idPass.groupId = group;
 395         idPass.userId = user;
 396         idPass.password = pass;
 397     }
 398     return 1;
 399 }
 400
 401 /* SEARCH SERVICE ------------------------------ */
 402
 403 static void display_variant(Z_Variant *v, int level)
 404 {
 405     int i;
 406
 407     for (i = 0; i < v->num_triples; i++)
 408     {
 409         printf("%*sclass=%d,type=%d", level * 4, "", *v->triples[i]->zclass,
 410             *v->triples[i]->type);
 411         if (v->triples[i]->which == Z_Triple_internationalString)
 412             printf(",value=%s\n", v->triples[i]->value.internationalString);
 413         else
 414             printf("\n");
 415     }
 416 }
 417
 418 static void display_grs1(Z_GenericRecord *r, int level)
 419 {
 420     int i;
 421
 422     if (!r)
 423         return;
 424     for (i = 0; i < r->num_elements; i++)
 425     {
 426         Z_TaggedElement *t;
 427
 428         printf("%*s", level * 4, "");
 429         t = r->elements[i];
 430         printf("(");
 431         if (t->tagType)
 432             printf("%d,", *t->tagType);
 433         else
 434             printf("?,");
 435         if (t->tagValue->which == Z_StringOrNumeric_numeric)
 436             printf("%d) ", *t->tagValue->u.numeric);
 437         else
 438             printf("%s) ", t->tagValue->u.string);
 439         if (t->content->which == Z_ElementData_subtree)
 440         {
 441             printf("\n");
 442             display_grs1(t->content->u.subtree, level+1);
 443         }
 444         else if (t->content->which == Z_ElementData_string)
 445             printf("%s\n", t->content->u.string);
 446         else if (t->content->which == Z_ElementData_numeric)
 447             printf("%d\n", *t->content->u.numeric);
 448         else if (t->content->which == Z_ElementData_oid)
 449         {
 450             int *ip = t->content->u.oid;
 451             oident *oent;
 452
 453             if ((oent = oid_getentbyoid(t->content->u.oid)))
 454                 printf("OID: %s\n", oent->desc);
 455             else
 456             {
 457                 printf("{");
 458                 while (ip && *ip >= 0)
 459                     printf(" %d", *(ip++));
 460                 printf(" }\n");
 461             }
 462         }
 463         else if (t->content->which == Z_ElementData_noDataRequested)
 464             printf("[No data requested]\n");
 465         else if (t->content->which == Z_ElementData_elementEmpty)
 466             printf("[Element empty]\n");
 467         else if (t->content->which == Z_ElementData_elementNotThere)
 468             printf("[Element not there]\n");
 469         else
 470             printf("??????\n");
 471         if (t->appliedVariant)
 472             display_variant(t->appliedVariant, level+1);
 473         if (t->metaData && t->metaData->supportedVariants)
 474         {
 475             int c;
 476
 477             printf("%*s---- variant list\n", (level+1)*4, "");
 478             for (c = 0; c < t->metaData->num_supportedVariants; c++)
 479             {
 480                 printf("%*svariant #%d\n", (level+1)*4, "", c);
 481                 display_variant(t->metaData->supportedVariants[c], level + 2);
 482             }
 483         }
 484     }
 485 }
 486
 487
 488 static void print_record(const unsigned char *buf, size_t len)
 489 {
 490     size_t i = len;
 491     print_stringn (buf, len);
 492     /* add newline if not already added ... */
 493     if (i <= 0 || buf[i-1] != '\n')
 494        fputc ('\n', stdout);
 495 }
 496
 497 static void display_record(Z_External *r)
 498 {
 499     oident *ent = oid_getentbyoid(r->direct_reference);
 500
 501     record_last = r;
 502     /*
 503      * Tell the user what we got.
 504      */
 505     if (r->direct_reference)
 506     {
 507         printf("Record type: ");
 508         if (ent)
 509             printf("%s\n", ent->desc);
 510         else if (!odr_oid(print, &r->direct_reference, 0, 0))
 511         {
 512             odr_perror(print, "print oid");
 513             odr_reset(print);
 514         }
 515     }
 516     /* Check if this is a known, ASN.1 type tucked away in an octet string */
 517     if (ent && r->which == Z_External_octet)
 518     {
 519         Z_ext_typeent *type = z_ext_getentbyref(ent->value);
 520         void *rr;
 521
 522         if (type)
 523         {
 524             /*
 525              * Call the given decoder to process the record.
 526              */
 527             odr_setbuf(in, (char*)r->u.octet_aligned->buf,
 528                 r->u.octet_aligned->len, 0);
 529             if (!(*type->fun)(in, (char **)&rr, 0, 0))
 530             {
 531                 odr_perror(in, "Decoding constructed record.");
 532                 fprintf(stderr, "[Near %d]\n", odr_offset(in));
 533                 fprintf(stderr, "Packet dump:\n---------\n");
 534                 odr_dumpBER(stderr, (char*)r->u.octet_aligned->buf,
 535                     r->u.octet_aligned->len);
 536                 fprintf(stderr, "---------\n");
 537                 exit(1);
 538             }
 539             /*
 540              * Note: we throw away the original, BER-encoded record here.
 541              * Do something else with it if you want to keep it.
 542              */
 543             r->u.sutrs = (Z_SUTRS *) rr; /* we don't actually check the type here. */
 544             r->which = type->what;
 545         }
 546     }
 547     if (ent && ent->value == VAL_SOIF)
 548         print_record((const unsigned char *) r->u.octet_aligned->buf,
 549                      r->u.octet_aligned->len);
 550     else if (r->which == Z_External_octet && r->u.octet_aligned->len)
 551     {
 552         const char *octet_buf = (char*)r->u.octet_aligned->buf;
 553         if (ent->value == VAL_TEXT_XML || ent->value == VAL_APPLICATION_XML ||
 554             ent->value == VAL_HTML)
 555         {
 556             print_record((const unsigned char *) octet_buf,
 557                          r->u.octet_aligned->len);
 558         }
 559         else if (ent->value == VAL_POSTSCRIPT)
 560         {
 561             int size = r->u.octet_aligned->len;
 562             if (size > 100)
 563                 size = 100;
 564             print_record((const unsigned char *) octet_buf, size);
 565         }
 566         else
 567         {
 568             if (
 569 #if AVOID_MARC_DECODE
 570                 /* primitive check for a marc OID 5.1-29 */
 571                 ent->oidsuffix[0] == 5 && ent->oidsuffix[1] < 30
 572 #else
 573         1
 574 #endif
 575         )
 576         {
 577                 if (marc_display_exl (octet_buf, NULL, 0 /* debug */,
 578                                       r->u.octet_aligned->len) <= 0)
 579                 {
 580                     printf ("bad MARC. Dumping as it is:\n");
 581                     print_record((const unsigned char*) octet_buf,
 582                                  r->u.octet_aligned->len);
 583                 }
 584             }
 585             else
 586             {
 587                 print_record((const unsigned char*) octet_buf,
 588                              r->u.octet_aligned->len);
 589             }
 590         }
 591         if (marcdump)
 592             fwrite (octet_buf, 1, r->u.octet_aligned->len, marcdump);
 593     }
 594     else if (ent && ent->value == VAL_SUTRS)
 595     {
 596         if (r->which != Z_External_sutrs)
 597         {
 598             printf("Expecting single SUTRS type for SUTRS.\n");
 599             return;
 600         }
 601         print_record(r->u.sutrs->buf, r->u.sutrs->len);
 602     }
 603     else if (ent && ent->value == VAL_GRS1)
 604     {
 605         if (r->which != Z_External_grs1)
 606         {
 607             printf("Expecting single GRS type for GRS.\n");
 608             return;
 609         }
 610         display_grs1(r->u.grs1, 0);
 611     }
 612     else
 613     {
 614         printf("Unknown record representation.\n");
 615         if (!z_External(print, &r, 0, 0))
 616         {
 617             odr_perror(print, "Printing external");
 618             odr_reset(print);
 619         }
 620     }
 621 }
 622
 623
 624 static void display_diagrecs(Z_DiagRec **pp, int num)
 625 {
 626     int i;
 627     oident *ent;
 628     Z_DefaultDiagFormat *r;
 629
 630     printf("Diagnostic message(s) from database:\n");
 631     for (i = 0; i<num; i++)
 632     {
 633         Z_DiagRec *p = pp[i];
 634         if (p->which != Z_DiagRec_defaultFormat)
 635         {
 636             printf("Diagnostic record not in default format.\n");
 637             return;
 638         }
 639         else
 640             r = p->u.defaultFormat;
 641         if (!(ent = oid_getentbyoid(r->diagnosticSetId)) ||
 642             ent->oclass != CLASS_DIAGSET || ent->value != VAL_BIB1)
 643             printf("Missing or unknown diagset\n");
 644         printf("    [%d] %s", *r->condition, diagbib1_str(*r->condition));
 645         switch (r->which)
 646         {
 647         case Z_DefaultDiagFormat_v2Addinfo:
 648             printf (" -- v2 addinfo '%s'\n", r->u.v2Addinfo);
 649             break;
 650         case Z_DefaultDiagFormat_v3Addinfo:
 651             printf (" -- v3 addinfo '%s'\n", r->u.v3Addinfo);
 652             break;
 653         }
 654     }
 655 }
 656
 657
 658 static void display_nameplusrecord(Z_NamePlusRecord *p)
 659 {
 660     if (p->databaseName)
 661         printf("[%s]", p->databaseName);
 662     if (p->which == Z_NamePlusRecord_surrogateDiagnostic)
 663         display_diagrecs(&p->u.surrogateDiagnostic, 1);
 664     else if (p->which == Z_NamePlusRecord_databaseRecord)
 665         display_record(p->u.databaseRecord);
 666 }
 667
 668 static void display_records(Z_Records *p)
 669 {
 670     int i;
 671
 672     if (p->which == Z_Records_NSD)
 673     {
 674         Z_DiagRec dr, *dr_p = &dr;
 675         dr.which = Z_DiagRec_defaultFormat;
 676         dr.u.defaultFormat = p->u.nonSurrogateDiagnostic;
 677         display_diagrecs (&dr_p, 1);
 678     }
 679     else if (p->which == Z_Records_multipleNSD)
 680         display_diagrecs (p->u.multipleNonSurDiagnostics->diagRecs,
 681                           p->u.multipleNonSurDiagnostics->num_diagRecs);
 682     else
 683     {
 684         printf("Records: %d\n", p->u.databaseOrSurDiagnostics->num_records);
 685         for (i = 0; i < p->u.databaseOrSurDiagnostics->num_records; i++)
 686             display_nameplusrecord(p->u.databaseOrSurDiagnostics->records[i]);
 687     }
 688 }
 689
 690 static int send_deleteResultSetRequest(char *arg)
 691 {
 692     char names[8][32];
 693     int i;
 694
 695     Z_APDU *apdu = zget_APDU(out, Z_APDU_deleteResultSetRequest);
 696     Z_DeleteResultSetRequest *req = apdu->u.deleteResultSetRequest;
 697
 698     req->referenceId = set_refid (out);
 699
 700     req->num_resultSetList =
 701         sscanf (arg, "%30s %30s %30s %30s %30s %30s %30s %30s",
 702                 names[0], names[1], names[2], names[3],
 703                 names[4], names[5], names[6], names[7]);
 704
 705     req->deleteFunction = (int *)
 706         odr_malloc (out, sizeof(*req->deleteFunction));
 707     if (req->num_resultSetList > 0)
 708     {
 709         *req->deleteFunction = Z_DeleteRequest_list;
 710         req->resultSetList = (char **)
 711             odr_malloc (out, sizeof(*req->resultSetList)*
 712                         req->num_resultSetList);
 713         for (i = 0; i<req->num_resultSetList; i++)
 714             req->resultSetList[i] = names[i];
 715     }
 716     else
 717     {
 718         *req->deleteFunction = Z_DeleteRequest_all;
 719         req->resultSetList = 0;
 720     }
 721
 722     send_apdu(apdu);
 723     printf("Sent deleteResultSetRequest.\n");
 724     return 2;
 725 }
 726
 727 static int send_searchRequest(char *arg)
 728 {
 729     Z_APDU *apdu = zget_APDU(out, Z_APDU_searchRequest);
 730     Z_SearchRequest *req = apdu->u.searchRequest;
 731     Z_Query query;
 732     int oid[OID_SIZE];
 733 #if YAZ_MODULE_ccl
 734     struct ccl_rpn_node *rpn = NULL;
 735     int error, pos;
 736 #endif
 737     char setstring[100];
 738     Z_RPNQuery *RPNquery;
 739     Odr_oct ccl_query;
 740
 741 #if YAZ_MODULE_ccl
 742     if (queryType == QueryType_CCL2RPN)
 743     {
 744         rpn = ccl_find_str(bibset, arg, &error, &pos);
 745         if (error)
 746         {
 747             printf("CCL ERROR: %s\n", ccl_err_msg(error));
 748             return 0;
 749         }
 750     }
 751 #endif
 752     req->referenceId = set_refid (out);
 753     if (!strcmp(arg, "@big")) /* strictly for troublemaking */
 754     {
 755         static unsigned char big[2100];
 756         static Odr_oct bigo;
 757
 758         /* send a very big referenceid to test transport stack etc. */
 759         memset(big, 'A', 2100);
 760         bigo.len = bigo.size = 2100;
 761         bigo.buf = big;
 762         req->referenceId = &bigo;
 763     }
 764
 765     if (setnumber >= 0)
 766     {
 767         sprintf(setstring, "%d", ++setnumber);
 768         req->resultSetName = setstring;
 769     }
 770     *req->smallSetUpperBound = smallSetUpperBound;
 771     *req->largeSetLowerBound = largeSetLowerBound;
 772     *req->mediumSetPresentNumber = mediumSetPresentNumber;
 773     if (smallSetUpperBound > 0 || (largeSetLowerBound > 1 &&
 774         mediumSetPresentNumber > 0))
 775     {
 776         oident prefsyn;
 777
 778         prefsyn.proto = protocol;
 779         prefsyn.oclass = CLASS_RECSYN;
 780         prefsyn.value = recordsyntax;
 781         req->preferredRecordSyntax =
 782             odr_oiddup(out, oid_ent_to_oid(&prefsyn, oid));
 783         req->smallSetElementSetNames =
 784             req->mediumSetElementSetNames = elementSetNames;
 785     }
 786     req->num_databaseNames = num_databaseNames;
 787     req->databaseNames = databaseNames;
 788
 789     req->query = &query;
 790
 791     switch (queryType)
 792     {
 793     case QueryType_Prefix:
 794         query.which = Z_Query_type_1;
 795         RPNquery = p_query_rpn (out, protocol, arg);
 796         if (!RPNquery)
 797         {
 798             printf("Prefix query error\n");
 799             return 0;
 800         }
 801         query.u.type_1 = RPNquery;
 802         break;
 803     case QueryType_CCL:
 804         query.which = Z_Query_type_2;
 805         query.u.type_2 = &ccl_query;
 806         ccl_query.buf = (unsigned char*) arg;
 807         ccl_query.len = strlen(arg);
 808         break;
 809 #if YAZ_MODULE_ccl
 810     case QueryType_CCL2RPN:
 811         query.which = Z_Query_type_1;
 812         RPNquery = ccl_rpn_query(out, rpn);
 813         if (!RPNquery)
 814         {
 815             printf ("Couldn't convert from CCL to RPN\n");
 816             return 0;
 817         }
 818         query.u.type_1 = RPNquery;
 819         ccl_rpn_delete (rpn);
 820         break;
 821 #endif
 822     default:
 823         printf ("Unsupported query type\n");
 824         return 0;
 825     }
 826     send_apdu(apdu);
 827     setno = 1;
 828     printf("Sent searchRequest.\n");
 829     return 2;
 830 }
 831
 832 static int process_searchResponse(Z_SearchResponse *res)
 833 {
 834     printf ("Received SearchResponse.\n");
 835     print_refid (res->referenceId);
 836     if (*res->searchStatus)
 837         printf("Search was a success.\n");
 838     else
 839         printf("Search was a bloomin' failure.\n");
 840     printf("Number of hits: %d", *res->resultCount);
 841     if (setnumber >= 0)
 842         printf (", setno %d", setnumber);
 843     printf("\nrecords returned: %d\n",
 844         *res->numberOfRecordsReturned);
 845     setno += *res->numberOfRecordsReturned;
 846     if (res->records)
 847         display_records(res->records);
 848     return 0;
 849 }
 850
 851 static void print_level(int iLevel)
 852 {
 853     int i;
 854     for (i = 0; i < iLevel * 4; i++)
 855         printf(" ");
 856 }
 857
 858 static void print_int(int iLevel, const char *pTag, int *pInt)
 859 {
 860     if (pInt != NULL)
 861     {
 862         print_level(iLevel);
 863         printf("%s: %d\n", pTag, *pInt);
 864     }
 865 }
 866
 867 static void print_string(int iLevel, const char *pTag, const char *pString)
 868 {
 869     if (pString != NULL)
 870     {
 871         print_level(iLevel);
 872         printf("%s: %s\n", pTag, pString);
 873     }
 874 }
 875
 876 static void print_oid(int iLevel, const char *pTag, Odr_oid *pOid)
 877 {
 878     if (pOid != NULL)
 879     {
 880         int *pInt = pOid;
 881
 882         print_level(iLevel);
 883         printf("%s:", pTag);
 884         for (; *pInt != -1; pInt++)
 885             printf(" %d", *pInt);
 886         printf("\n");
 887     }
 888 }
 889
 890 static void print_referenceId(int iLevel, Z_ReferenceId *referenceId)
 891 {
 892     if (referenceId != NULL)
 893     {
 894         int i;
 895
 896         print_level(iLevel);
 897         printf("Ref Id (%d, %d): ", referenceId->len, referenceId->size);
 898         for (i = 0; i < referenceId->len; i++)
 899             printf("%c", referenceId->buf[i]);
 900         printf("\n");
 901     }
 902 }
 903
 904 static void print_string_or_numeric(int iLevel, const char *pTag, Z_StringOrNumeric *pStringNumeric)
 905 {
 906     if (pStringNumeric != NULL)
 907     {
 908         switch (pStringNumeric->which)
 909         {
 910             case Z_StringOrNumeric_string:
 911                 print_string(iLevel, pTag, pStringNumeric->u.string);
 912                 break;
 913
 914             case Z_StringOrNumeric_numeric:
 915                 print_int(iLevel, pTag, pStringNumeric->u.numeric);
 916                 break;
 917
 918             default:
 919                 print_level(iLevel);
 920                 printf("%s: valid type for Z_StringOrNumeric\n", pTag);
 921                 break;
 922         }
 923     }
 924 }
 925
 926 static void print_universe_report_duplicate(int iLevel, Z_UniverseReportDuplicate *pUniverseReportDuplicate)
 927 {
 928     if (pUniverseReportDuplicate != NULL)
 929     {
 930         print_level(iLevel);
 931         printf("Universe Report Duplicate: \n");
 932         iLevel++;
 933         print_string_or_numeric(iLevel, "Hit No", pUniverseReportDuplicate->hitno);
 934     }
 935 }
 936
 937 static void print_universe_report_hits(int iLevel, Z_UniverseReportHits *pUniverseReportHits)
 938 {
 939     if (pUniverseReportHits != NULL)
 940     {
 941         print_level(iLevel);
 942         printf("Universe Report Hits: \n");
 943         iLevel++;
 944         print_string_or_numeric(iLevel, "Database", pUniverseReportHits->database);
 945         print_string_or_numeric(iLevel, "Hits", pUniverseReportHits->hits);
 946     }
 947 }
 948
 949 static void print_universe_report(int iLevel, Z_UniverseReport *pUniverseReport)
 950 {
 951     if (pUniverseReport != NULL)
 952     {
 953         print_level(iLevel);
 954         printf("Universe Report: \n");
 955         iLevel++;
 956         print_int(iLevel, "Total Hits", pUniverseReport->totalHits);
 957         switch (pUniverseReport->which)
 958         {
 959             case Z_UniverseReport_databaseHits:
 960                 print_universe_report_hits(iLevel, pUniverseReport->u.databaseHits);
 961                 break;
 962
 963             case Z_UniverseReport_duplicate:
 964                 print_universe_report_duplicate(iLevel, pUniverseReport->u.duplicate);
 965                 break;
 966
 967             default:
 968                 print_level(iLevel);
 969                 printf("Type: %d\n", pUniverseReport->which);
 970                 break;
 971         }
 972     }
 973 }
 974
 975 static void print_external(int iLevel, Z_External *pExternal)
 976 {
 977     if (pExternal != NULL)
 978     {
 979         print_level(iLevel);
 980         printf("External: \n");
 981         iLevel++;
 982         print_oid(iLevel, "Direct Reference", pExternal->direct_reference);
 983         print_int(iLevel, "InDirect Reference", pExternal->indirect_reference);
 984         print_string(iLevel, "Descriptor", pExternal->descriptor);
 985         switch (pExternal->which)
 986         {
 987             case Z_External_universeReport:
 988                 print_universe_report(iLevel, pExternal->u.universeReport);
 989                 break;
 990
 991             default:
 992                 print_level(iLevel);
 993                 printf("Type: %d\n", pExternal->which);
 994                 break;
 995         }
 996     }
 997 }
 998
 999 static int process_resourceControlRequest (Z_ResourceControlRequest *req)
1000 {
1001     printf ("Received ResourceControlRequest.\n");
1002     print_referenceId(1, req->referenceId);
1003     print_int(1, "Suspended Flag", req->suspendedFlag);
1004     print_int(1, "Partial Results Available", req->partialResultsAvailable);
1005     print_int(1, "Response Required", req->responseRequired);
1006     print_int(1, "Triggered Request Flag", req->triggeredRequestFlag);
1007     print_external(1, req->resourceReport);
1008     return 0;
1009 }
1010
1011 void process_ESResponse(Z_ExtendedServicesResponse *res)
1012 {
1013     printf("Status: ");
1014     switch (*res->operationStatus)
1015     {
1016     case Z_ExtendedServicesResponse_done:
1017         printf ("done\n");
1018         break;
1019     case Z_ExtendedServicesResponse_accepted:
1020         printf ("accepted\n");
1021         break;
1022     case Z_ExtendedServicesResponse_failure:
1023         printf ("failure\n");
1024         display_diagrecs(res->diagnostics, res->num_diagnostics);
1025         break;
1026     default:
1027         printf ("unknown\n");
1028     }
1029     if ( (*res->operationStatus != Z_ExtendedServicesResponse_failure) &&
1030         (res->num_diagnostics != 0) ) {
1031         display_diagrecs(res->diagnostics, res->num_diagnostics);
1032     }
1033     print_refid (res->referenceId);
1034     if (res->taskPackage &&
1035         res->taskPackage->which == Z_External_extendedService)
1036     {
1037         Z_TaskPackage *taskPackage = res->taskPackage->u.extendedService;
1038         Odr_oct *id = taskPackage->targetReference;
1039         Z_External *ext = taskPackage->taskSpecificParameters;
1040
1041         if (id)
1042         {
1043             printf ("Target Reference: ");
1044             print_stringn (id->buf, id->len);
1045             printf ("\n");
1046         }
1047         if (ext->which == Z_External_update)
1048         {
1049             Z_IUUpdateTaskPackage *utp = ext->u.update->u.taskPackage;
1050             if (utp && utp->targetPart)
1051             {
1052                 Z_IUTargetPart *targetPart = utp->targetPart;
1053                 int i;
1054
1055                 for (i = 0; i<targetPart->num_taskPackageRecords;  i++)
1056                 {
1057
1058                     Z_IUTaskPackageRecordStructure *tpr =
1059                         targetPart->taskPackageRecords[i];
1060                     printf ("task package record %d\n", i+1);
1061                     if (tpr->which == Z_IUTaskPackageRecordStructure_record)
1062                     {
1063                         display_record (tpr->u.record);
1064                     }
1065                     else
1066                     {
1067                         printf ("other type\n");
1068                     }
1069                 }
1070             }
1071         }
1072     }
1073 }
1074
1075 #if YAZ_MODULE_ill
1076
1077 const char *get_ill_element (void *clientData, const char *element)
1078 {
1079     return 0;
1080 }
1081
1082 static Z_External *create_external_itemRequest()
1083 {
1084     struct ill_get_ctl ctl;
1085     ILL_ItemRequest *req;
1086     Z_External *r = 0;
1087     int item_request_size = 0;
1088     char *item_request_buf = 0;
1089
1090     ctl.odr = out;
1091     ctl.clientData = 0;
1092     ctl.f = get_ill_element;
1093
1094     req = ill_get_ItemRequest(&ctl, "ill", 0);
1095     if (!req)
1096         printf ("ill_get_ItemRequest failed\n");
1097
1098     if (!ill_ItemRequest (out, &req, 0, 0))
1099     {
1100         if (apdu_file)
1101         {
1102             ill_ItemRequest(print, &req, 0, 0);
1103             odr_reset(print);
1104         }
1105         item_request_buf = odr_getbuf (out, &item_request_size, 0);
1106         if (item_request_buf)
1107             odr_setbuf (out, item_request_buf, item_request_size, 1);
1108         printf ("Couldn't encode ItemRequest, size %d\n", item_request_size);
1109         return 0;
1110     }
1111     else
1112     {
1113         oident oid;
1114
1115         item_request_buf = odr_getbuf (out, &item_request_size, 0);
1116         oid.proto = PROTO_GENERAL;
1117         oid.oclass = CLASS_GENERAL;
1118         oid.value = VAL_ISO_ILL_1;
1119
1120         r = (Z_External *) odr_malloc (out, sizeof(*r));
1121         r->direct_reference = odr_oiddup(out,oid_getoidbyent(&oid));
1122         r->indirect_reference = 0;
1123         r->descriptor = 0;
1124         r->which = Z_External_single;
1125
1126         r->u.single_ASN1_type = (Odr_oct *)
1127             odr_malloc (out, sizeof(*r->u.single_ASN1_type));
1128         r->u.single_ASN1_type->buf = (unsigned char *)
1129         odr_malloc (out, item_request_size);
1130         r->u.single_ASN1_type->len = item_request_size;
1131         r->u.single_ASN1_type->size = item_request_size;
1132         memcpy (r->u.single_ASN1_type->buf, item_request_buf,
1133                 item_request_size);
1134         printf ("len = %d\n", item_request_size);
1135     }
1136     return r;
1137 }
1138 #endif
1139
1140 #ifdef YAZ_MODULE_ill
1141 static Z_External *create_external_ILL_APDU(int which)
1142 {
1143     struct ill_get_ctl ctl;
1144     ILL_APDU *ill_apdu;
1145     Z_External *r = 0;
1146     int ill_request_size = 0;
1147     char *ill_request_buf = 0;
1148
1149     ctl.odr = out;
1150     ctl.clientData = 0;
1151     ctl.f = get_ill_element;
1152
1153     ill_apdu = ill_get_APDU(&ctl, "ill", 0);
1154
1155     if (!ill_APDU (out, &ill_apdu, 0, 0))
1156     {
1157         if (apdu_file)
1158         {
1159             printf ("-------------------\n");
1160             ill_APDU(print, &ill_apdu, 0, 0);
1161             odr_reset(print);
1162             printf ("-------------------\n");
1163         }
1164         ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
1165         if (ill_request_buf)
1166             odr_setbuf (out, ill_request_buf, ill_request_size, 1);
1167         printf ("Couldn't encode ILL-Request, size %d\n", ill_request_size);
1168         return 0;
1169     }
1170     else
1171     {
1172         oident oid;
1173         ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
1174
1175         oid.proto = PROTO_GENERAL;
1176         oid.oclass = CLASS_GENERAL;
1177         oid.value = VAL_ISO_ILL_1;
1178
1179         r = (Z_External *) odr_malloc (out, sizeof(*r));
1180         r->direct_reference = odr_oiddup(out,oid_getoidbyent(&oid));
1181         r->indirect_reference = 0;
1182         r->descriptor = 0;
1183         r->which = Z_External_single;
1184
1185         r->u.single_ASN1_type = (Odr_oct *)
1186             odr_malloc (out, sizeof(*r->u.single_ASN1_type));
1187         r->u.single_ASN1_type->buf = (unsigned char *)
1188         odr_malloc (out, ill_request_size);
1189         r->u.single_ASN1_type->len = ill_request_size;
1190         r->u.single_ASN1_type->size = ill_request_size;
1191         memcpy (r->u.single_ASN1_type->buf, ill_request_buf, ill_request_size);
1192         printf ("len = %d\n", ill_request_size);
1193     }
1194     return r;
1195 }
1196 #endif
1197
1198
1199 static Z_External *create_ItemOrderExternal(const char *type, int itemno)
1200 {
1201     Z_External *r = (Z_External *) odr_malloc(out, sizeof(Z_External));
1202     oident ItemOrderRequest;
1203
1204     ItemOrderRequest.proto = PROTO_Z3950;
1205     ItemOrderRequest.oclass = CLASS_EXTSERV;
1206     ItemOrderRequest.value = VAL_ITEMORDER;
1207
1208     r->direct_reference = odr_oiddup(out,oid_getoidbyent(&ItemOrderRequest));
1209     r->indirect_reference = 0;
1210     r->descriptor = 0;
1211
1212     r->which = Z_External_itemOrder;
1213
1214     r->u.itemOrder = (Z_ItemOrder *) odr_malloc(out,sizeof(Z_ItemOrder));
1215     memset(r->u.itemOrder, 0, sizeof(Z_ItemOrder));
1216     r->u.itemOrder->which=Z_IOItemOrder_esRequest;
1217
1218     r->u.itemOrder->u.esRequest = (Z_IORequest *)
1219         odr_malloc(out,sizeof(Z_IORequest));
1220     memset(r->u.itemOrder->u.esRequest, 0, sizeof(Z_IORequest));
1221
1222     r->u.itemOrder->u.esRequest->toKeep = (Z_IOOriginPartToKeep *)
1223         odr_malloc(out,sizeof(Z_IOOriginPartToKeep));
1224     memset(r->u.itemOrder->u.esRequest->toKeep, 0, sizeof(Z_IOOriginPartToKeep));
1225     r->u.itemOrder->u.esRequest->notToKeep = (Z_IOOriginPartNotToKeep *)
1226         odr_malloc(out,sizeof(Z_IOOriginPartNotToKeep));
1227     memset(r->u.itemOrder->u.esRequest->notToKeep, 0, sizeof(Z_IOOriginPartNotToKeep));
1228
1229     r->u.itemOrder->u.esRequest->toKeep->supplDescription = NULL;
1230     r->u.itemOrder->u.esRequest->toKeep->contact = NULL;
1231     r->u.itemOrder->u.esRequest->toKeep->addlBilling = NULL;
1232
1233     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem =
1234         (Z_IOResultSetItem *) odr_malloc(out, sizeof(Z_IOResultSetItem));
1235     memset(r->u.itemOrder->u.esRequest->notToKeep->resultSetItem, 0, sizeof(Z_IOResultSetItem));
1236     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->resultSetId = "1";
1237
1238     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->item =
1239         (int *) odr_malloc(out, sizeof(int));
1240     *r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->item = itemno;
1241
1242 #if YAZ_MODULE_ill
1243     if (!strcmp (type, "item") || !strcmp(type, "2"))
1244     {
1245         printf ("using item-request\n");
1246         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
1247             create_external_itemRequest();
1248     }
1249     else if (!strcmp(type, "ill") || !strcmp(type, "1"))
1250     {
1251         printf ("using ILL-request\n");
1252         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
1253             create_external_ILL_APDU(ILL_APDU_ILL_Request);
1254     }
1255     else if (!strcmp(type, "xml") || !strcmp(type, "3"))
1256     {
1257     const char *xml_buf =
1258         "<itemorder>\n"
1259         "  <type>request</type>\n"
1260         "  <libraryNo>000200</libraryNo>\n"
1261         "  <borrowerTicketNo> 1212 </borrowerTicketNo>\n"
1262         "</itemorder>";
1263         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
1264             z_ext_record (out, VAL_TEXT_XML, xml_buf, strlen(xml_buf));
1265     }
1266     else
1267         r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
1268
1269 #else
1270     r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
1271 #endif
1272     return r;
1273 }
1274
1275 static int send_itemorder(const char *type, int itemno)
1276 {
1277     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest);
1278     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
1279     oident ItemOrderRequest;
1280
1281     ItemOrderRequest.proto = PROTO_Z3950;
1282     ItemOrderRequest.oclass = CLASS_EXTSERV;
1283     ItemOrderRequest.value = VAL_ITEMORDER;
1284     req->packageType = odr_oiddup(out,oid_getoidbyent(&ItemOrderRequest));
1285     req->packageName = esPackageName;
1286
1287     req->taskSpecificParameters = create_ItemOrderExternal(type, itemno);
1288
1289     send_apdu(apdu);
1290     return 0;
1291 }
1292
1293 static int cmd_update(char *arg)
1294 {
1295     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest );
1296     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
1297     Z_External *r;
1298     int oid[OID_SIZE];
1299     Z_IUOriginPartToKeep *toKeep;
1300     Z_IUSuppliedRecords *notToKeep;
1301     oident update_oid;
1302     printf ("Update request\n");
1303     fflush(stdout);
1304
1305     if (!record_last)
1306         return 0;
1307     update_oid.proto = PROTO_Z3950;
1308     update_oid.oclass = CLASS_EXTSERV;
1309     update_oid.value = VAL_DBUPDATE;
1310     oid_ent_to_oid (&update_oid, oid);
1311     req->packageType = odr_oiddup(out,oid);
1312     req->packageName = esPackageName;
1313
1314     req->referenceId = set_refid (out);
1315
1316     r = req->taskSpecificParameters = (Z_External *)
1317         odr_malloc (out, sizeof(*r));
1318     r->direct_reference = odr_oiddup(out,oid);
1319     r->indirect_reference = 0;
1320     r->descriptor = 0;
1321     r->which = Z_External_update;
1322     r->u.update = (Z_IUUpdate *) odr_malloc(out, sizeof(*r->u.update));
1323     r->u.update->which = Z_IUUpdate_esRequest;
1324     r->u.update->u.esRequest = (Z_IUUpdateEsRequest *)
1325         odr_malloc(out, sizeof(*r->u.update->u.esRequest));
1326     toKeep = r->u.update->u.esRequest->toKeep = (Z_IUOriginPartToKeep *)
1327         odr_malloc(out, sizeof(*r->u.update->u.esRequest->toKeep));
1328     toKeep->databaseName = databaseNames[0];
1329     toKeep->schema = 0;
1330     toKeep->elementSetName = 0;
1331     toKeep->actionQualifier = 0;
1332     toKeep->action = (int *) odr_malloc(out, sizeof(*toKeep->action));
1333     *toKeep->action = Z_IUOriginPartToKeep_recordInsert;
1334
1335     notToKeep = r->u.update->u.esRequest->notToKeep = (Z_IUSuppliedRecords *)
1336         odr_malloc(out, sizeof(*r->u.update->u.esRequest->notToKeep));
1337     notToKeep->num = 1;
1338     notToKeep->elements = (Z_IUSuppliedRecords_elem **)
1339         odr_malloc(out, sizeof(*notToKeep->elements));
1340     notToKeep->elements[0] = (Z_IUSuppliedRecords_elem *)
1341         odr_malloc(out, sizeof(**notToKeep->elements));
1342     notToKeep->elements[0]->u.number = 0;
1343     notToKeep->elements[0]->supplementalId = 0;
1344     notToKeep->elements[0]->correlationInfo = 0;
1345     notToKeep->elements[0]->record = record_last;
1346
1347     send_apdu(apdu);
1348
1349     return 2;
1350 }
1351
1352 /* II : Added to do DALI Item Order Extended services request */
1353 static int cmd_itemorder(char *arg)
1354 {
1355     char type[12];
1356     int itemno;
1357
1358     if (sscanf (arg, "%10s %d", type, &itemno) != 2)
1359         return 0;
1360
1361     printf("Item order request\n");
1362     fflush(stdout);
1363     send_itemorder(type, itemno);
1364     return(2);
1365 }
1366
1367 static int cmd_find(char *arg)
1368 {
1369     if (!*arg)
1370     {
1371         printf("Find what?\n");
1372         return 0;
1373     }
1374     if (!conn)
1375     {
1376         printf("Not connected yet\n");
1377         return 0;
1378     }
1379     if (!send_searchRequest(arg))
1380         return 0;
1381     return 2;
1382 }
1383
1384 static int cmd_delete(char *arg)
1385 {
1386     if (!conn)
1387     {
1388         printf("Not connected yet\n");
1389         return 0;
1390     }
1391     if (!send_deleteResultSetRequest(arg))
1392         return 0;
1393     return 2;
1394 }
1395
1396 static int cmd_ssub(char *arg)
1397 {
1398     if (!(smallSetUpperBound = atoi(arg)))
1399         return 0;
1400     return 1;
1401 }
1402
1403 static int cmd_lslb(char *arg)
1404 {
1405     if (!(largeSetLowerBound = atoi(arg)))
1406         return 0;
1407     return 1;
1408 }
1409
1410 static int cmd_mspn(char *arg)
1411 {
1412     if (!(mediumSetPresentNumber = atoi(arg)))
1413         return 0;
1414     return 1;
1415 }
1416
1417 static int cmd_status(char *arg)
1418 {
1419     printf("smallSetUpperBound: %d\n", smallSetUpperBound);
1420     printf("largeSetLowerBound: %d\n", largeSetLowerBound);
1421     printf("mediumSetPresentNumber: %d\n", mediumSetPresentNumber);
1422     return 1;
1423 }
1424
1425 static int cmd_setnames(char *arg)
1426 {
1427     if (*arg == '1')         /* enable ? */
1428         setnumber = 0;
1429     else if (*arg == '0')    /* disable ? */
1430         setnumber = -1;
1431     else if (setnumber < 0)  /* no args, toggle .. */
1432         setnumber = 0;
1433     else
1434         setnumber = -1;
1435
1436     if (setnumber >= 0)
1437         printf("Set numbering enabled.\n");
1438     else
1439         printf("Set numbering disabled.\n");
1440     return 1;
1441 }
1442
1443 /* PRESENT SERVICE ----------------------------- */
1444
1445 static int send_presentRequest(char *arg)
1446 {
1447     Z_APDU *apdu = zget_APDU(out, Z_APDU_presentRequest);
1448     Z_PresentRequest *req = apdu->u.presentRequest;
1449     Z_RecordComposition compo;
1450     oident prefsyn;
1451     int nos = 1;
1452     int oid[OID_SIZE];
1453     char *p;
1454     char setstring[100];
1455
1456     req->referenceId = set_refid (out);
1457     if ((p = strchr(arg, '+')))
1458     {
1459         nos = atoi(p + 1);
1460         *p = 0;
1461     }
1462     if (*arg)
1463         setno = atoi(arg);
1464     if (p && (p=strchr(p+1, '+')))
1465     {
1466         strcpy (setstring, p+1);
1467         req->resultSetId = setstring;
1468     }
1469     else if (setnumber >= 0)
1470     {
1471         sprintf(setstring, "%d", setnumber);
1472         req->resultSetId = setstring;
1473     }
1474 #if 0
1475     if (1)
1476     {
1477         static Z_Range range;
1478         static Z_Range *rangep = &range;
1479     req->num_ranges = 1;
1480 #endif
1481     req->resultSetStartPoint = &setno;
1482     req->numberOfRecordsRequested = &nos;
1483     prefsyn.proto = protocol;
1484     prefsyn.oclass = CLASS_RECSYN;
1485     prefsyn.value = recordsyntax;
1486     req->preferredRecordSyntax =
1487         odr_oiddup (out, oid_ent_to_oid(&prefsyn, oid));
1488
1489     if (schema != VAL_NONE)
1490     {
1491         oident prefschema;
1492
1493         prefschema.proto = protocol;
1494         prefschema.oclass = CLASS_SCHEMA;
1495         prefschema.value = schema;
1496
1497         req->recordComposition = &compo;
1498         compo.which = Z_RecordComp_complex;
1499         compo.u.complex = (Z_CompSpec *)
1500             odr_malloc(out, sizeof(*compo.u.complex));
1501         compo.u.complex->selectAlternativeSyntax = (bool_t *)
1502             odr_malloc(out, sizeof(bool_t));
1503         *compo.u.complex->selectAlternativeSyntax = 0;
1504
1505         compo.u.complex->generic = (Z_Specification *)
1506             odr_malloc(out, sizeof(*compo.u.complex->generic));
1507         compo.u.complex->generic->schema = (Odr_oid *)
1508             odr_oiddup(out, oid_ent_to_oid(&prefschema, oid));
1509         if (!compo.u.complex->generic->schema)
1510         {
1511             /* OID wasn't a schema! Try record syntax instead. */
1512             prefschema.oclass = CLASS_RECSYN;
1513             compo.u.complex->generic->schema = (Odr_oid *)
1514                 odr_oiddup(out, oid_ent_to_oid(&prefschema, oid));
1515         }
1516         if (!elementSetNames)
1517             compo.u.complex->generic->elementSpec = 0;
1518         else
1519         {
1520             compo.u.complex->generic->elementSpec = (Z_ElementSpec *)
1521                 odr_malloc(out, sizeof(Z_ElementSpec));
1522             compo.u.complex->generic->elementSpec->which =
1523                 Z_ElementSpec_elementSetName;
1524             compo.u.complex->generic->elementSpec->u.elementSetName =
1525                 elementSetNames->u.generic;
1526         }
1527         compo.u.complex->num_dbSpecific = 0;
1528         compo.u.complex->dbSpecific = 0;
1529         compo.u.complex->num_recordSyntax = 0;
1530         compo.u.complex->recordSyntax = 0;
1531     }
1532     else if (elementSetNames)
1533     {
1534         req->recordComposition = &compo;
1535         compo.which = Z_RecordComp_simple;
1536         compo.u.simple = elementSetNames;
1537     }
1538     send_apdu(apdu);
1539     printf("Sent presentRequest (%d+%d).\n", setno, nos);
1540     return 2;
1541 }
1542
1543 void process_close(Z_Close *req)
1544 {
1545     Z_APDU *apdu = zget_APDU(out, Z_APDU_close);
1546     Z_Close *res = apdu->u.close;
1547
1548     static char *reasons[] =
1549     {
1550         "finished",
1551         "shutdown",
1552         "system problem",
1553         "cost limit reached",
1554         "resources",
1555         "security violation",
1556         "protocolError",
1557         "lack of activity",
1558         "peer abort",
1559         "unspecified"
1560     };
1561
1562     printf("Reason: %s, message: %s\n", reasons[*req->closeReason],
1563         req->diagnosticInformation ? req->diagnosticInformation : "NULL");
1564     if (sent_close)
1565     {
1566         cs_close (conn);
1567         conn = NULL;
1568         if (session_mem)
1569         {
1570             nmem_destroy (session_mem);
1571             session_mem = NULL;
1572         }
1573         sent_close = 0;
1574     }
1575     else
1576     {
1577         *res->closeReason = Z_Close_finished;
1578         send_apdu(apdu);
1579         printf("Sent response.\n");
1580         sent_close = 1;
1581     }
1582 }
1583
1584 static int cmd_show(char *arg)
1585 {
1586     if (!conn)
1587     {
1588         printf("Not connected yet\n");
1589         return 0;
1590     }
1591     if (!send_presentRequest(arg))
1592         return 0;
1593     return 2;
1594 }
1595
1596 int cmd_quit(char *arg)
1597 {
1598     printf("See you later, alligator.\n");
1599     xmalloc_trav ("");
1600     exit(0);
1601     return 0;
1602 }
1603
1604 int cmd_cancel(char *arg)
1605 {
1606     Z_APDU *apdu = zget_APDU(out, Z_APDU_triggerResourceControlRequest);
1607     Z_TriggerResourceControlRequest *req =
1608         apdu->u.triggerResourceControlRequest;
1609     bool_t rfalse = 0;
1610
1611     if (!conn)
1612     {
1613         printf("Session not initialized yet\n");
1614         return 0;
1615     }
1616     if (!ODR_MASK_GET(session->options, Z_Options_triggerResourceCtrl))
1617     {
1618         printf("Target doesn't support cancel (trigger resource ctrl)\n");
1619         return 0;
1620     }
1621     *req->requestedAction = Z_TriggerResourceCtrl_cancel;
1622     req->resultSetWanted = &rfalse;
1623
1624     send_apdu(apdu);
1625     printf("Sent cancel request\n");
1626     return 2;
1627 }
1628
1629 int send_scanrequest(const char *query, int pp, int num, const char *term)
1630 {
1631     Z_APDU *apdu = zget_APDU(out, Z_APDU_scanRequest);
1632     Z_ScanRequest *req = apdu->u.scanRequest;
1633     int use_rpn = 1;
1634 #if YAZ_MODULE_ccl
1635     int oid[OID_SIZE];
1636
1637     if (queryType == QueryType_CCL2RPN)
1638     {
1639         oident bib1;
1640         int error, pos;
1641         struct ccl_rpn_node *rpn;
1642
1643         rpn = ccl_find_str (bibset,  query, &error, &pos);
1644         if (error)
1645         {
1646             printf("CCL ERROR: %s\n", ccl_err_msg(error));
1647             return -1;
1648         }
1649         use_rpn = 0;
1650         bib1.proto = PROTO_Z3950;
1651         bib1.oclass = CLASS_ATTSET;
1652         bib1.value = VAL_BIB1;
1653         req->attributeSet = oid_ent_to_oid (&bib1, oid);
1654         if (!(req->termListAndStartPoint = ccl_scan_query (out, rpn)))
1655         {
1656             printf("Couldn't convert CCL to Scan term\n");
1657             return -1;
1658         }
1659         ccl_rpn_delete (rpn);
1660     }
1661 #endif
1662     if (use_rpn && !(req->termListAndStartPoint =
1663                      p_query_scan(out, protocol, &req->attributeSet, query)))
1664     {
1665         printf("Prefix query error\n");
1666         return -1;
1667     }
1668     if (term && *term)
1669     {
1670         if (req->termListAndStartPoint->term &&
1671             req->termListAndStartPoint->term->which == Z_Term_general &&
1672             req->termListAndStartPoint->term->u.general)
1673         {
1674             req->termListAndStartPoint->term->u.general->buf =
1675                 (unsigned char *) odr_strdup(out, term);
1676             req->termListAndStartPoint->term->u.general->len =
1677                 req->termListAndStartPoint->term->u.general->size =
1678                 strlen(term);
1679         }
1680     }
1681     req->referenceId = set_refid (out);
1682     req->num_databaseNames = num_databaseNames;
1683     req->databaseNames = databaseNames;
1684     req->numberOfTermsRequested = &num;
1685     req->preferredPositionInResponse = &pp;
1686     send_apdu(apdu);
1687     return 2;
1688 }
1689
1690 int send_sortrequest(char *arg, int newset)
1691 {
1692     Z_APDU *apdu = zget_APDU(out, Z_APDU_sortRequest);
1693     Z_SortRequest *req = apdu->u.sortRequest;
1694     Z_SortKeySpecList *sksl = (Z_SortKeySpecList *)
1695         odr_malloc (out, sizeof(*sksl));
1696     char setstring[32];
1697
1698     if (setnumber >= 0)
1699         sprintf (setstring, "%d", setnumber);
1700     else
1701         sprintf (setstring, "default");
1702
1703     req->referenceId = set_refid (out);
1704
1705     req->num_inputResultSetNames = 1;
1706     req->inputResultSetNames = (Z_InternationalString **)
1707         odr_malloc (out, sizeof(*req->inputResultSetNames));
1708     req->inputResultSetNames[0] = odr_strdup (out, setstring);
1709
1710     if (newset && setnumber >= 0)
1711         sprintf (setstring, "%d", ++setnumber);
1712
1713     req->sortedResultSetName = odr_strdup (out, setstring);
1714
1715     req->sortSequence = yaz_sort_spec (out, arg);
1716     if (!req->sortSequence)
1717     {
1718         printf ("Missing sort specifications\n");
1719         return -1;
1720     }
1721     send_apdu(apdu);
1722     return 2;
1723 }
1724
1725 void display_term(Z_TermInfo *t)
1726 {
1727     if (t->term->which == Z_Term_general)
1728     {
1729         printf("%.*s", t->term->u.general->len, t->term->u.general->buf);
1730         sprintf(last_scan_line, "%.*s", t->term->u.general->len,
1731             t->term->u.general->buf);
1732     }
1733     else
1734         printf("Term (not general)");
1735     if (t->globalOccurrences)
1736         printf (" (%d)\n", *t->globalOccurrences);
1737     else
1738         printf ("\n");
1739 }
1740
1741 void process_scanResponse(Z_ScanResponse *res)
1742 {
1743     int i;
1744     Z_Entry **entries = NULL;
1745     int num_entries = 0;
1746
1747     printf("Received ScanResponse\n");
1748     print_refid (res->referenceId);
1749     printf("%d entries", *res->numberOfEntriesReturned);
1750     if (res->positionOfTerm)
1751         printf (", position=%d", *res->positionOfTerm);
1752     printf ("\n");
1753     if (*res->scanStatus != Z_Scan_success)
1754         printf("Scan returned code %d\n", *res->scanStatus);
1755     if (!res->entries)
1756         return;
1757     if ((entries = res->entries->entries))
1758         num_entries = res->entries->num_entries;
1759     for (i = 0; i < num_entries; i++)
1760     {
1761         int pos_term = res->positionOfTerm ? *res->positionOfTerm : -1;
1762         if (entries[i]->which == Z_Entry_termInfo)
1763         {
1764             printf("%c ", i + 1 == pos_term ? '*' : ' ');
1765             display_term(entries[i]->u.termInfo);
1766         }
1767         else
1768             display_diagrecs(&entries[i]->u.surrogateDiagnostic, 1);
1769     }
1770     if (res->entries->nonsurrogateDiagnostics)
1771         display_diagrecs (res->entries->nonsurrogateDiagnostics,
1772                           res->entries->num_nonsurrogateDiagnostics);
1773 }
1774
1775 void process_sortResponse(Z_SortResponse *res)
1776 {
1777     printf("Received SortResponse: status=");
1778     switch (*res->sortStatus)
1779     {
1780     case Z_SortStatus_success:
1781         printf ("success"); break;
1782     case Z_SortStatus_partial_1:
1783         printf ("partial"); break;
1784     case Z_SortStatus_failure:
1785         printf ("failure"); break;
1786     default:
1787         printf ("unknown (%d)", *res->sortStatus);
1788     }
1789     printf ("\n");
1790     print_refid (res->referenceId);
1791     if (res->diagnostics)
1792         display_diagrecs(res->diagnostics,
1793                          res->num_diagnostics);
1794 }
1795
1796 void process_deleteResultSetResponse (Z_DeleteResultSetResponse *res)
1797 {
1798     printf("Got deleteResultSetResponse status=%d\n",
1799            *res->deleteOperationStatus);
1800     if (res->deleteListStatuses)
1801     {
1802         int i;
1803         for (i = 0; i < res->deleteListStatuses->num; i++)
1804         {
1805             printf ("%s status=%d\n", res->deleteListStatuses->elements[i]->id,
1806                     *res->deleteListStatuses->elements[i]->status);
1807         }
1808     }
1809 }
1810
1811 int cmd_sort_generic(char *arg, int newset)
1812 {
1813     if (!conn)
1814     {
1815         printf("Session not initialized yet\n");
1816         return 0;
1817     }
1818     if (!ODR_MASK_GET(session->options, Z_Options_sort))
1819     {
1820         printf("Target doesn't support sort\n");
1821         return 0;
1822     }
1823     if (*arg)
1824     {
1825         if (send_sortrequest(arg, newset) < 0)
1826             return 0;
1827         return 2;
1828     }
1829     return 0;
1830 }
1831
1832 int cmd_sort(char *arg)
1833 {
1834     return cmd_sort_generic (arg, 0);
1835 }
1836
1837 int cmd_sort_newset (char *arg)
1838 {
1839     return cmd_sort_generic (arg, 1);
1840 }
1841
1842 int cmd_scan(char *arg)
1843 {
1844     if (!conn)
1845     {
1846         printf("Session not initialized yet\n");
1847         return 0;
1848     }
1849     if (!ODR_MASK_GET(session->options, Z_Options_scan))
1850     {
1851         printf("Target doesn't support scan\n");
1852         return 0;
1853     }
1854     if (*arg)
1855     {
1856         strcpy (last_scan_query, arg);
1857         if (send_scanrequest(arg, 1, 20, 0) < 0)
1858             return 0;
1859     }
1860     else
1861     {
1862         if (send_scanrequest(last_scan_query, 1, 20, last_scan_line) < 0)
1863             return 0;
1864     }
1865     return 2;
1866 }
1867
1868 int cmd_schema(char *arg)
1869 {
1870     if (!arg || !*arg)
1871     {
1872         schema = VAL_NONE;
1873         return 1;
1874     }
1875     schema = oid_getvalbyname (arg);
1876     if (schema == VAL_NONE)
1877     {
1878         printf ("unknown schema\n");
1879         return 0;
1880     }
1881     return 1;
1882 }
1883
1884 int cmd_format(char *arg)
1885 {
1886     if (!arg || !*arg)
1887     {
1888         printf("Usage: format <recordsyntax>\n");
1889         return 0;
1890     }
1891     recordsyntax = oid_getvalbyname (arg);
1892     if (recordsyntax == VAL_NONE)
1893     {
1894         printf ("unknown record syntax\n");
1895         return 0;
1896     }
1897     return 1;
1898 }
1899
1900 int cmd_elements(char *arg)
1901 {
1902     static Z_ElementSetNames esn;
1903     static char what[100];
1904
1905     if (!arg || !*arg)
1906     {
1907         elementSetNames = 0;
1908         return 1;
1909     }
1910     strcpy(what, arg);
1911     esn.which = Z_ElementSetNames_generic;
1912     esn.u.generic = what;
1913     elementSetNames = &esn;
1914     return 1;
1915 }
1916
1917 int cmd_attributeset(char *arg)
1918 {
1919     char what[100];
1920
1921     if (!arg || !*arg)
1922     {
1923         printf("Usage: attributeset <setname>\n");
1924         return 0;
1925     }
1926     sscanf(arg, "%s", what);
1927     if (p_query_attset (what))
1928     {
1929         printf("Unknown attribute set name\n");
1930         return 0;
1931     }
1932     return 1;
1933 }
1934
1935 int cmd_querytype (char *arg)
1936 {
1937     if (!strcmp (arg, "ccl"))
1938         queryType = QueryType_CCL;
1939     else if (!strcmp (arg, "prefix") || !strcmp(arg, "rpn"))
1940         queryType = QueryType_Prefix;
1941 #if YAZ_MODULE_ccl
1942     else if (!strcmp (arg, "ccl2rpn") || !strcmp (arg, "cclrpn"))
1943         queryType = QueryType_CCL2RPN;
1944 #endif
1945     else
1946     {
1947         printf ("Querytype must be one of:\n");
1948         printf (" prefix         - Prefix query\n");
1949         printf (" ccl            - CCL query\n");
1950 #if YAZ_MODULE_ccl
1951         printf (" ccl2rpn        - CCL query converted to RPN\n");
1952 #endif
1953         return 0;
1954     }
1955     return 1;
1956 }
1957
1958 int cmd_refid (char *arg)
1959 {
1960     xfree (refid);
1961     refid = NULL;
1962     if (*arg)
1963     {
1964         refid = (char *) xmalloc (strlen(arg)+1);
1965         strcpy (refid, arg);
1966     }
1967     return 1;
1968 }
1969
1970 int cmd_close(char *arg)
1971 {
1972     Z_APDU *apdu;
1973     Z_Close *req;
1974     if (!conn)
1975         return 0;
1976
1977     apdu = zget_APDU(out, Z_APDU_close);
1978     req = apdu->u.close;
1979     *req->closeReason = Z_Close_finished;
1980     send_apdu(apdu);
1981     printf("Sent close request.\n");
1982     sent_close = 1;
1983     return 2;
1984 }
1985
1986 int cmd_packagename(char* arg)
1987 {
1988     xfree (esPackageName);
1989     esPackageName = NULL;
1990     if (*arg)
1991     {
1992         esPackageName = (char *) xmalloc (strlen(arg)+1);
1993         strcpy (esPackageName, arg);
1994     }
1995     return 1;
1996 }
1997
1998 int cmd_proxy(char* arg)
1999 {
2000     xfree (yazProxy);
2001     yazProxy = NULL;
2002     if (*arg)
2003     {
2004         yazProxy = (char *) xmalloc (strlen(arg)+1);
2005         strcpy (yazProxy, arg);
2006     }
2007     return 1;
2008 }
2009
2010 int cmd_source(char* arg)
2011 {
2012     /* first should open the file and read one line at a time.. */
2013     FILE* includeFile;
2014     char line[1024], *cp;
2015
2016     {
2017         char* args_end=(arg)+strlen(arg)-1;
2018         while(isspace(*args_end))
2019         {*args_end=0;
2020         --args_end;};
2021     }
2022
2023     REMOVE_TAILING_BLANKS(arg);
2024
2025     if(strlen(arg)<1) {
2026         fprintf(stderr,"Error in source command use a filename\n");
2027         return -1;
2028     }
2029
2030     includeFile = fopen (arg, "r");
2031
2032     if(!includeFile) {
2033         fprintf(stderr,"Unable to open file %s for reading\n",arg);
2034         return -1;
2035     }
2036
2037     while(!feof(includeFile)) {
2038         memset(line,0,sizeof(line));
2039         fgets(line,sizeof(line),includeFile);
2040
2041         if(strlen(line) < 2) continue;
2042         if(line[0] == '#') continue;
2043
2044         if ((cp = strrchr (line, '\n')))
2045             *cp = '\0';
2046
2047         process_cmd_line(line);
2048     }
2049
2050     if(fclose(includeFile)<0) {
2051         perror("unable to close include file");
2052         exit(1);
2053     }
2054     return 1;
2055 }
2056
2057 int cmd_subshell(char* args)
2058 {
2059     if(strlen(args))
2060         system(args);
2061     else
2062         system(getenv("SHELL"));
2063
2064     printf("\n");
2065     return 1;
2066 }
2067
2068 int cmd_set_apdufile(char* arg)
2069 {
2070     REMOVE_TAILING_BLANKS(arg);
2071
2072     if(apdu_file && apdu_file != stderr) { /* don't close stdout*/
2073         perror("unable to close apdu log file");
2074     };
2075     apdu_file=NULL;
2076
2077     if(strlen(arg)<1) {
2078         return 1;
2079     }
2080
2081     if(!strcmp(arg,"-"))
2082         apdu_file=stderr;
2083     else
2084         apdu_file=fopen(arg, "a");
2085
2086     if(!apdu_file) {
2087         perror("unable to open apdu log file no apdu log loaded");
2088     } else {
2089         odr_setprint(print, apdu_file);
2090     }
2091
2092     return 1;
2093 }
2094
2095 int cmd_set_cclfields(char* arg)
2096 {
2097 #if YAZ_MODULE_ccl
2098     FILE *inf;
2099
2100     REMOVE_TAILING_BLANKS(arg);
2101
2102     bibset = ccl_qual_mk ();
2103     inf = fopen (arg, "r");
2104     if (inf)
2105     {
2106         ccl_qual_file (bibset, inf);
2107         fclose (inf);
2108     }
2109 #else
2110     fprintf(stderr,"Not compiled with the yaz ccl module\n");
2111 #endif
2112
2113     return 1;
2114 }
2115
2116 int cmd_set_marcdump(char* arg)
2117 {
2118     if(marcdump && marcdump != stderr) { /* don't close stdout*/
2119         perror("unable to close apdu log file");
2120     };
2121     marcdump=NULL;
2122
2123     if(strlen(arg)<1) {
2124         return 1;
2125     }
2126
2127     if(!strcmp(arg,"-"))
2128         marcdump=stderr;
2129     else
2130         marcdump=fopen(arg, "a");
2131
2132     if(!marcdump) {
2133         perror("unable to open apdu marcdump file no marcdump done\n");
2134     }
2135
2136     return 1;
2137 }
2138
2139 int cmd_set_proxy(char* arg) {
2140     if(yazProxy) free(yazProxy);
2141     yazProxy=NULL;
2142
2143     if(strlen(arg) > 1) {
2144         yazProxy=strdup(arg);
2145     }
2146     return 1;
2147 }
2148
2149 /*
2150    this command takes 3 arge {name class oid}
2151  */
2152 int cmd_register_oid(char* args) {
2153     static struct {
2154         char* className;
2155         oid_class oclass;
2156     } oid_classes[] = {
2157         {"appctx",CLASS_APPCTX},
2158         {"absyn",CLASS_ABSYN},
2159         {"attset",CLASS_ATTSET},
2160         {"transyn",CLASS_TRANSYN},
2161         {"diagset",CLASS_DIAGSET},
2162         {"recsyn",CLASS_RECSYN},
2163         {"resform",CLASS_RESFORM},
2164         {"accform",CLASS_ACCFORM},
2165         {"extserv",CLASS_EXTSERV},
2166         {"userinfo",CLASS_USERINFO},
2167         {"elemspec",CLASS_ELEMSPEC},
2168         {"varset",CLASS_VARSET},
2169         {"schema",CLASS_SCHEMA},
2170         {"tagset",CLASS_TAGSET},
2171         {"general",CLASS_GENERAL},
2172         {0,0}
2173     };
2174     char oname_str[101], oclass_str[101], oid_str[101];
2175     char* name;
2176     int i;
2177     oid_class oidclass = CLASS_GENERAL;
2178     int val = 0, oid[OID_SIZE];
2179     struct oident * new_oident=NULL;
2180
2181     if (sscanf (args, "%100[^ ] %100[^ ] %100s",
2182                 oname_str,oclass_str, oid_str) < 1) {
2183         printf("Error in regristrate command \n");
2184         return 0;
2185     }
2186
2187     for (i = 0; oid_classes[i].className; i++) {
2188         if (!strcmp(oid_classes[i].className, oclass_str))
2189         {
2190             oidclass=oid_classes[i].oclass;
2191             break;
2192         }
2193     }
2194
2195     if(!(oid_classes[i].className)) {
2196         printf("Unknonwn oid class %s\n",oclass_str);
2197         return 0;
2198     }
2199
2200     i = 0;
2201     name = oid_str;
2202     val = 0;
2203
2204     while (isdigit (*name))
2205     {
2206         val = val*10 + (*name - '0');
2207         name++;
2208         if (*name == '.')
2209         {
2210             if (i < OID_SIZE-1)
2211                 oid[i++] = val;
2212             val = 0;
2213             name++;
2214         }
2215     }
2216     oid[i] = val;
2217     oid[i+1] = -1;
2218
2219     new_oident=oid_addent (oid,PROTO_GENERAL,oidclass,oname_str,VAL_DYNAMIC);
2220     if(strcmp(new_oident->desc,oname_str)) {
2221         fprintf(stderr,"oid is already named as %s, regristration faild\n",
2222                 new_oident->desc);
2223     }
2224     return 1;
2225 }
2226
2227 int cmd_push_command(char* arg) {
2228 #if HAVE_READLINE_HISTORY_H
2229     if(strlen(arg)>1)
2230         add_history(arg);
2231 #else
2232     fprintf(stderr,"Not compiled with the readline/history module\n");
2233 #endif
2234     return 1;
2235 }
2236
2237 void source_rcfile() {
2238     /*  Look for a $HOME/.yazclientrc and source it if it exists */
2239     struct stat statbuf;
2240     char buffer[1000];
2241     char* homedir=getenv("HOME");
2242     if(!homedir) return;
2243
2244     sprintf(buffer,"%s/.yazclientrc",homedir);
2245
2246     if(stat(buffer,&statbuf)==0) {
2247         cmd_source(buffer);
2248     }
2249
2250     if(stat(".yazclientrc",&statbuf)==0) {
2251         cmd_source(".yazclientrc");
2252     }
2253 }
2254
2255
2256 static void initialize(void)
2257 {
2258 #if YAZ_MODULE_ccl
2259     FILE *inf;
2260 #endif
2261     if (!(out = odr_createmem(ODR_ENCODE)) ||
2262         !(in = odr_createmem(ODR_DECODE)) ||
2263         !(print = odr_createmem(ODR_PRINT)))
2264     {
2265         fprintf(stderr, "failed to allocate ODR streams\n");
2266         exit(1);
2267     }
2268     oid_init();
2269
2270     setvbuf(stdout, 0, _IONBF, 0);
2271     if (apdu_file)
2272         odr_setprint(print, apdu_file);
2273
2274 #if YAZ_MODULE_ccl
2275     bibset = ccl_qual_mk ();
2276     inf = fopen (ccl_fields, "r");
2277     if (inf)
2278     {
2279         ccl_qual_file (bibset, inf);
2280         fclose (inf);
2281     }
2282 #endif
2283     cmd_base("Default");
2284
2285 #if HAVE_READLINE_READLINE_H
2286     rl_attempted_completion_function = (CPPFunction*)readline_completer;
2287 #endif
2288
2289     source_rcfile();
2290 }
2291
2292
2293 #if HAVE_GETTIMEOFDAY
2294 struct timeval tv_start, tv_end;
2295 #endif
2296
2297 void wait_and_handle_responce()
2298 {
2299
2300     int res;
2301     char *netbuffer= 0;
2302     int netbufferlen = 0;
2303     Z_APDU *apdu;
2304
2305
2306     if (conn)
2307     {
2308         do
2309         {
2310             if ((res = cs_get(conn, &netbuffer, &netbufferlen)) < 0)
2311             {
2312                 perror("cs_get");
2313                 exit(1);
2314             }
2315             if (!res)
2316             {
2317                 printf("Target closed connection.\n");
2318                 exit(1);
2319             }
2320             odr_reset(in); /* release APDU from last round */
2321             record_last = 0;
2322             odr_setbuf(in, netbuffer, res, 0);
2323             if (!z_APDU(in, &apdu, 0, 0))
2324             {
2325                 odr_perror(in, "Decoding incoming APDU");
2326                 fprintf(stderr, "[Near %d]\n", odr_offset(in));
2327                 fprintf(stderr, "Packet dump:\n---------\n");
2328                 odr_dumpBER(stderr, netbuffer, res);
2329                 fprintf(stderr, "---------\n");
2330                 if (apdu_file)
2331                     z_APDU(print, &apdu, 0, 0);
2332                 exit(1);
2333             }
2334             if (apdu_file && !z_APDU(print, &apdu, 0, 0))
2335             {
2336                 odr_perror(print, "Failed to print incoming APDU");
2337                 odr_reset(print);
2338                 continue;
2339             }
2340             switch(apdu->which)
2341             {
2342             case Z_APDU_initResponse:
2343                 process_initResponse(apdu->u.initResponse);
2344                 break;
2345             case Z_APDU_searchResponse:
2346                 process_searchResponse(apdu->u.searchResponse);
2347                 break;
2348             case Z_APDU_scanResponse:
2349                 process_scanResponse(apdu->u.scanResponse);
2350                 break;
2351             case Z_APDU_presentResponse:
2352                 print_refid (apdu->u.presentResponse->referenceId);
2353                 setno +=
2354                     *apdu->u.presentResponse->numberOfRecordsReturned;
2355                 if (apdu->u.presentResponse->records)
2356                     display_records(apdu->u.presentResponse->records);
2357                 else
2358                     printf("No records.\n");
2359                 printf ("nextResultSetPosition = %d\n",
2360                         *apdu->u.presentResponse->nextResultSetPosition);
2361                 break;
2362             case Z_APDU_sortResponse:
2363                 process_sortResponse(apdu->u.sortResponse);
2364                 break;
2365             case Z_APDU_extendedServicesResponse:
2366                 printf("Got extended services response\n");
2367                 process_ESResponse(apdu->u.extendedServicesResponse);
2368                 break;
2369             case Z_APDU_close:
2370                 printf("Target has closed the association.\n");
2371                 process_close(apdu->u.close);
2372                 break;
2373             case Z_APDU_resourceControlRequest:
2374                 process_resourceControlRequest
2375                     (apdu->u.resourceControlRequest);
2376                 break;
2377             case Z_APDU_deleteResultSetResponse:
2378                 process_deleteResultSetResponse(apdu->u.
2379                                                 deleteResultSetResponse);
2380                 break;
2381             default:
2382                 printf("Received unknown APDU type (%d).\n",
2383                        apdu->which);
2384                 exit(1);
2385             }
2386         }
2387         while (conn && cs_more(conn));
2388 #if HAVE_GETTIMEOFDAY
2389         gettimeofday (&tv_end, 0);
2390 #if 0
2391         printf ("S/U S/U=%ld/%ld %ld/%ld",
2392                 (long) tv_start.tv_sec,
2393                 (long) tv_start.tv_usec,
2394                 (long) tv_end.tv_sec,
2395                 (long) tv_end.tv_usec);
2396 #endif
2397         printf ("Elapsed: %.6f\n",
2398                 (double) tv_end.tv_usec / 1e6 + tv_end.tv_sec -
2399                 ((double) tv_start.tv_usec / 1e6 + tv_start.tv_sec));
2400 #endif
2401     }
2402     xfree (netbuffer);
2403 }
2404
2405
2406 static struct {
2407     char *cmd;
2408     int (*fun)(char *arg);
2409     char *ad;
2410     char *(*rl_completerfunction)(const char *text, int state);
2411     int complete_filenames;
2412 } cmd[] = {
2413     {"open", cmd_open, "('tcp'|'ssl')':<host>[':'<port>][/<db>]",NULL,0},
2414     {"quit", cmd_quit, "",NULL,0},
2415     {"find", cmd_find, "<query>",NULL,0},
2416     {"delete", cmd_delete, "<setname>",NULL,0},
2417     {"base", cmd_base, "<base-name>",NULL,0},
2418     {"show", cmd_show, "<rec#>['+'<#recs>['+'<setname>]]",NULL,0},
2419     {"scan", cmd_scan, "<term>",NULL,0},
2420     {"sort", cmd_sort, "<sortkey> <flag> <sortkey> <flag> ...",NULL,0},
2421     {"sort+", cmd_sort_newset, "<sortkey> <flag> <sortkey> <flag> ...",NULL,0},
2422     {"authentication", cmd_authentication, "<acctstring>",NULL,0},
2423     {"lslb", cmd_lslb, "<largeSetLowerBound>",NULL,0},
2424     {"ssub", cmd_ssub, "<smallSetUpperBound>",NULL,0},
2425     {"mspn", cmd_mspn, "<mediumSetPresentNumber>",NULL,0},
2426     {"status", cmd_status, "",NULL,0},
2427     {"setnames", cmd_setnames, "",NULL,0},
2428     {"cancel", cmd_cancel, "",NULL,0},
2429     {"format", cmd_format, "<recordsyntax>",complete_format,0},
2430     {"schema", cmd_schema, "<schema>",complete_schema,0},
2431     {"elements", cmd_elements, "<elementSetName>",NULL,0},
2432     {"close", cmd_close, "",NULL,0},
2433     {"attributeset", cmd_attributeset, "<attrset>",complete_attributeset,0},
2434     {"querytype", cmd_querytype, "<type>",complete_querytype,0},
2435     {"refid", cmd_refid, "<id>",NULL,0},
2436     {"itemorder", cmd_itemorder, "ill|item <itemno>",NULL,0},
2437     {"update", cmd_update, "<item>",NULL,0},
2438     {"packagename", cmd_packagename, "<packagename>",NULL,0},
2439     {"proxy", cmd_proxy, "[('tcp'|'ssl')]<host>[':'<port>]",NULL,0},
2440     {".", cmd_source, "<filename>",NULL,1},
2441     {"!", cmd_subshell, "Subshell command",NULL,0},
2442     {"set_apdufile", cmd_set_apdufile, "<filename>",NULL,0},
2443     {"set_marcdump", cmd_set_marcdump," <filename>",NULL,0},
2444     {"set_cclfields", cmd_set_cclfields,"<filename>",NULL,1},
2445     {"register_oid",cmd_register_oid,"<name> <class> <oid>",NULL,0},
2446     {"push_command",cmd_push_command,"<command>",command_generator,0},
2447     /* Server Admin Functions */
2448     {"adm-reindex", cmd_adm_reindex, "<database-name>",NULL,0},
2449     {"adm-truncate", cmd_adm_truncate, "('database'|'index')<object-name>",NULL,0},
2450     {"adm-create", cmd_adm_create, "",NULL,0},
2451     {"adm-drop", cmd_adm_drop, "('database'|'index')<object-name>",NULL,0},
2452     {"adm-import", cmd_adm_import, "<record-type> <dir> <pattern>",NULL,0},
2453     {"adm-refresh", cmd_adm_refresh, "",NULL,0},
2454     {"adm-commit", cmd_adm_commit, "",NULL,0},
2455     {"adm-shutdown", cmd_adm_shutdown, "",NULL,0},
2456     {"adm-startup", cmd_adm_startup, "",NULL,0},
2457     {0,0,0,0,0}
2458 };
2459
2460 void process_cmd_line(char* line)
2461 {
2462     int i,res;
2463     char word[32], arg[1024];
2464
2465 #if HAVE_GETTIMEOFDAY
2466     gettimeofday (&tv_start, 0);
2467 #endif
2468
2469     if ((res = sscanf(line, "%31s %1023[^;]", word, arg)) <= 0)
2470     {
2471         strcpy(word, last_cmd);
2472         *arg = '\0';
2473     }
2474     else if (res == 1)
2475         *arg = 0;
2476     strcpy(last_cmd, word);
2477
2478     /* removed tailing spaces from the arg command */
2479     {
2480         char* p;
2481         char* lastnonspace=NULL;
2482         p = arg;
2483
2484         for(;*p; ++p) {
2485             if(!isspace(*p)) {
2486                 lastnonspace = p;
2487             }
2488         }
2489         if(lastnonspace)
2490             *(++lastnonspace) = 0;
2491     }
2492
2493
2494     for (i = 0; cmd[i].cmd; i++)
2495         if (!strncmp(cmd[i].cmd, word, strlen(word)))
2496         {
2497             res = (*cmd[i].fun)(arg);
2498             break;
2499         }
2500     if (!cmd[i].cmd) /* dump our help-screen */
2501     {
2502         printf("Unknown command: %s.\n", word);
2503         printf("Currently recognized commands:\n");
2504         for (i = 0; cmd[i].cmd; i++)
2505             printf("   %s %s\n", cmd[i].cmd, cmd[i].ad);
2506         return;
2507     }
2508
2509     if(apdu_file) fflush(apdu_file);
2510
2511     if (res >= 2)
2512         wait_and_handle_responce();
2513
2514     if(apdu_file)
2515         fflush(apdu_file);
2516     if(marcdump)
2517         fflush(marcdump);
2518 }
2519
2520
2521 char *command_generator(const char *text, int state)
2522 {
2523     static idx; // index is the last used the last time command_generator was called
2524     if (state==0) {
2525         idx = 0;
2526     }
2527     for( ; cmd[idx].cmd; ++idx) {
2528         if (!strncmp(cmd[idx].cmd,text,strlen(text))) {
2529             ++idx;  /* skip this entry on the next run */
2530             return strdup(cmd[idx-1].cmd);
2531         };
2532     }
2533     return NULL;
2534 }
2535
2536
2537 /*
2538    This function only known how to complete on the first word
2539 */
2540 char ** readline_completer(char *text, int start, int end) {
2541 #if HAVE_READLINE_READLINE_H
2542
2543     if(start == 0) {
2544 #ifdef RL_READLINE_VERSION
2545         char** res=rl_completion_matches(text,
2546                                       command_generator);
2547 #else
2548         char** res=completion_matches(text,
2549                                       (CPFunction*)command_generator);
2550 #endif
2551         rl_attempted_completion_over = 1;
2552         return res;
2553     } else {
2554         char arg[1024],word[32];
2555         int i=0 ,res;
2556         if ((res = sscanf(rl_line_buffer, "%31s %1023[^;]", word, arg)) <= 0) {
2557             rl_attempted_completion_over = 1;
2558             return NULL;
2559         };
2560
2561         if(start != strlen(word) +1 ) {
2562             rl_attempted_completion_over = 1;
2563             return 0;
2564         }
2565         for (i = 0; cmd[i].cmd; i++) {
2566             if (!strncmp(cmd[i].cmd, word, strlen(word))) {
2567                 break;
2568             }
2569         }
2570
2571
2572         if(!cmd[i].complete_filenames)
2573             rl_attempted_completion_over = 1;
2574         if(cmd[i].rl_completerfunction) {
2575 #ifdef RL_READLINE_VERSION
2576             char** res=
2577                 rl_completion_matches(text,
2578                                    cmd[i].rl_completerfunction);
2579 #else
2580             char** res=
2581                 completion_matches(text,
2582                                    (CPFunction*)cmd[i].rl_completerfunction);
2583 #endif
2584             rl_attempted_completion_over = 1;
2585             return res;
2586         } else {
2587             rl_attempted_completion_over = 1;
2588             return 0;
2589         };
2590     };
2591 #else
2592     return 0;
2593 #endif
2594 };
2595
2596
2597 static void client(void)
2598 {
2599     char line[1024];
2600
2601     line[1023] = '\0';
2602
2603 #if HAVE_GETTIMEOFDAY
2604     gettimeofday (&tv_start, 0);
2605 #endif
2606
2607     while (1)
2608     {
2609         char *line_in = NULL;
2610 #if HAVE_READLINE_READLINE_H
2611         if (isatty(0))
2612         {
2613             line_in=readline(C_PROMPT);
2614             if (!line_in)
2615                 break;
2616 #if HAVE_READLINE_HISTORY_H
2617             if (*line_in)
2618                 add_history(line_in);
2619 #endif
2620             strncpy(line, line_in, 1023);
2621             free (line_in);
2622         }
2623 #endif
2624         if (!line_in)
2625         {
2626             char *end_p;
2627             printf (C_PROMPT);
2628             fflush(stdout);
2629             if (!fgets(line, 1023, stdin))
2630                 break;
2631             if ((end_p = strchr (line, '\n')))
2632                 *end_p = '\0';
2633         }
2634         process_cmd_line(line);
2635     }
2636 }
2637
2638 int main(int argc, char **argv)
2639 {
2640     char *prog = *argv;
2641     char *open_command = 0;
2642     char *auth_command = 0;
2643     char *arg;
2644     int ret;
2645
2646     while ((ret = options("k:c:a:m:v:p:u:", argv, argc, &arg)) != -2)
2647     {
2648         switch (ret)
2649         {
2650         case 0:
2651             if (!open_command)
2652             {
2653                 open_command = (char *) xmalloc (strlen(arg)+6);
2654                 strcpy (open_command, "open ");
2655                 strcat (open_command, arg);
2656             }
2657             break;
2658         case 'k':
2659             kilobytes = atoi(arg);
2660             break;
2661         case 'm':
2662             if (!(marcdump = fopen (arg, "a")))
2663             {
2664                 perror (arg);
2665                 exit (1);
2666             }
2667             break;
2668     case 'c':
2669         strncpy (ccl_fields, arg, sizeof(ccl_fields)-1);
2670         ccl_fields[sizeof(ccl_fields)-1] = '\0';
2671         break;
2672         case 'a':
2673             if (!strcmp(arg, "-"))
2674                 apdu_file=stderr;
2675             else
2676                 apdu_file=fopen(arg, "a");
2677             break;
2678     case 'p':
2679         yazProxy=strdup(arg);
2680         break;
2681         case 'u':
2682             if (!auth_command)
2683             {
2684                 auth_command = (char *) xmalloc (strlen(arg)+6);
2685                 strcpy (auth_command, "auth ");
2686                 strcat (auth_command, arg);
2687             }
2688             break;
2689         case 'v':
2690             yaz_log_init (yaz_log_mask_str(arg), "", NULL);
2691             break;
2692         default:
2693             fprintf (stderr, "Usage: %s [-m <marclog>] [ -a <apdulog>] "
2694                      "[-c cclfields]\n      [-p <proxy-addr>] [-u <auth>] "
2695                      "[-k size] [<server-addr>]\n",
2696                      prog);
2697             exit (1);
2698         }
2699     }
2700     initialize();
2701     if (auth_command)
2702     {
2703 #ifdef HAVE_GETTIMEOFDAY
2704         gettimeofday (&tv_start, 0);
2705 #endif
2706         process_cmd_line (auth_command);
2707 #if HAVE_READLINE_HISTORY_H
2708         add_history(auth_command);
2709 #endif
2710         xfree(auth_command);
2711     }
2712     if (open_command)
2713     {
2714 #ifdef HAVE_GETTIMEOFDAY
2715         gettimeofday (&tv_start, 0);
2716 #endif
2717         process_cmd_line (open_command);
2718 #if HAVE_READLINE_HISTORY_H
2719         add_history(open_command);
2720 #endif
2721         xfree(open_command);
2722     }
2723     client ();
2724     exit (0);
2725 }
2726
2727
2728
2729
2730 /*
2731  * Local variables:
2732  * tab-width: 4
2733  * c-basic-offset: 4
2734  * End:
2735  */