client/client.c

   1 /* This file is part of the YAZ toolkit.
   2  * Copyright (C) 1995-2010 Index Data
   3  * See the file LICENSE for details.
   4  */
   5 /** \file client.c
   6  *  \brief yaz-client program
   7  */
   8
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <assert.h>
  12 #include <time.h>
  13 #include <ctype.h>
  14 #ifndef WIN32
  15 #include <signal.h>
  16 #endif
  17 #if HAVE_SYS_TYPES_H
  18 #include <sys/types.h>
  19 #endif
  20 #if HAVE_LOCALE_H
  21 #include <locale.h>
  22 #endif
  23 #if HAVE_LANGINFO_H
  24 #include <langinfo.h>
  25 #endif
  26 #if HAVE_UNISTD_H
  27 #include <unistd.h>
  28 #endif
  29 #if HAVE_SYS_STAT_H
  30 #include <sys/stat.h>
  31 #endif
  32 #if HAVE_SYS_TIME_H
  33 #include <sys/time.h>
  34 #endif
  35
  36 #ifdef WIN32
  37 #include <sys/stat.h>
  38 #include <io.h>
  39 #include <windows.h>
  40 #define S_ISREG(x) (x & _S_IFREG)
  41 #define S_ISDIR(x) (x & _S_IFDIR)
  42 #endif
  43
  44 #include <yaz/yaz-util.h>
  45
  46 #include <yaz/comstack.h>
  47
  48 #include <yaz/oid_db.h>
  49 #define NO_OID 1
  50 #include <yaz/proto.h>
  51 #include <yaz/marcdisp.h>
  52 #include <yaz/diagbib1.h>
  53 #include <yaz/otherinfo.h>
  54 #include <yaz/charneg.h>
  55 #include <yaz/query-charset.h>
  56
  57 #include <yaz/pquery.h>
  58 #include <yaz/sortspec.h>
  59
  60 #include <yaz/ill.h>
  61 #include <yaz/srw.h>
  62 #include <yaz/yaz-ccl.h>
  63 #include <yaz/cql.h>
  64 #include <yaz/log.h>
  65 #include <yaz/facet.h>
  66
  67 #if HAVE_READLINE_READLINE_H
  68 #include <readline/readline.h>
  69 #endif
  70 #if HAVE_READLINE_HISTORY_H
  71 #include <readline/history.h>
  72 #endif
  73
  74
  75 #include "admin.h"
  76 #include "tabcomplete.h"
  77 #include "fhistory.h"
  78
  79 #define C_PROMPT "Z> "
  80
  81 static file_history_t file_history = 0;
  82
  83 static char sru_method[10] = "soap";
  84 static char sru_version[10] = "1.2";
  85 static char *codeset = 0;               /* character set for output */
  86 static int hex_dump = 0;
  87 static char *dump_file_prefix = 0;
  88 static ODR out, in, print;              /* encoding and decoding streams */
  89 #if YAZ_HAVE_XML2
  90 static ODR srw_sr_odr_out = 0;
  91 static Z_SRW_PDU *srw_sr = 0;
  92 #endif
  93 static FILE *apdu_file = 0;
  94 static FILE *ber_file = 0;
  95 static COMSTACK conn = 0;               /* our z-association */
  96
  97 static Z_IdAuthentication *auth = 0;    /* our current auth definition */
  98 static NMEM nmem_auth = NULL;
  99
 100 char *databaseNames[128];
 101 int num_databaseNames = 0;
 102 static Z_External *record_last = 0;
 103 static int setnumber = -1;              /* current result set number */
 104 static int smallSetUpperBound = 0;
 105 static int largeSetLowerBound = 1;
 106 static int mediumSetPresentNumber = 0;
 107 static Z_ElementSetNames *elementSetNames = 0;
 108 static Z_FacetList *facet_list = 0;
 109 static Odr_int setno = 1;                   /* current set offset */
 110 static enum oid_proto protocol = PROTO_Z3950;      /* current app protocol */
 111 #define RECORDSYNTAX_MAX 20
 112 static char *recordsyntax_list[RECORDSYNTAX_MAX];
 113 static int recordsyntax_size = 0;
 114
 115 static char *record_schema = 0;
 116 static int sent_close = 0;
 117 static NMEM session_mem = NULL;         /* memory handle for init-response */
 118 static Z_InitResponse *session_initResponse = 0;   /* session parameters */
 119 static char last_scan_line[512] = "0";
 120 static char last_scan_query[512] = "0";
 121 static char ccl_fields[512] = "default.bib";
 122 /* ### How can I set this path to use wherever YAZ is installed? */
 123 static char cql_fields[512] = "/usr/local/share/yaz/etc/pqf.properties";
 124 static char *esPackageName = 0;
 125 static char *yazProxy = 0;
 126 static int kilobytes = 1024;
 127 static char *negotiationCharset = 0;
 128 static int  negotiationCharsetRecords = 1;
 129 static int  negotiationCharsetVersion = 3;
 130 static char *outputCharset = 0;
 131 static char *marcCharset = 0;
 132 static char *queryCharset = 0;
 133 static char* yazLang = 0;
 134
 135 static char last_cmd[32] = "?";
 136 static FILE *marc_file = 0;
 137 static char *refid = NULL;
 138 static int auto_reconnect = 0;
 139 static int auto_wait = 1;
 140 static Odr_bitmask z3950_options;
 141 static int z3950_version = 3;
 142 static int scan_stepSize = 0;
 143 static int scan_position = 1;
 144 static int scan_size = 20;
 145 static char cur_host[200];
 146 static Odr_int last_hit_count = 0;
 147
 148 typedef enum {
 149     QueryType_Prefix,
 150     QueryType_CCL,
 151     QueryType_CCL2RPN,
 152     QueryType_CQL,
 153     QueryType_CQL2RPN
 154 } QueryType;
 155
 156 static QueryType queryType = QueryType_Prefix;
 157
 158 static CCL_bibset bibset;               /* CCL bibset handle */
 159 static cql_transform_t cqltrans = 0; /* CQL context-set handle */
 160
 161 #if HAVE_READLINE_COMPLETION_OVER
 162
 163 #else
 164 /* readline doesn't have this var. Define it ourselves. */
 165 int rl_attempted_completion_over = 0;
 166 #endif
 167
 168 #define maxOtherInfosSupported 10
 169 struct eoi {
 170     Odr_oid oid[OID_SIZE];
 171     char* value;
 172 } extraOtherInfos[maxOtherInfosSupported];
 173
 174 static void process_cmd_line(char* line);
 175 #if HAVE_READLINE_READLINE_H
 176 static char **readline_completer(char *text, int start, int end);
 177 #endif
 178 static char *command_generator(const char *text, int state);
 179 static int cmd_register_tab(const char* arg);
 180 static int cmd_querycharset(const char *arg);
 181
 182 static void close_session(void);
 183
 184 static void marc_file_write(const char *buf, size_t sz);
 185
 186 static void wait_and_handle_response(int one_response_only);
 187 static Z_GDU *get_HTTP_Request_url(ODR odr, const char *url);
 188
 189 ODR getODROutputStream(void)
 190 {
 191     return out;
 192 }
 193
 194 static const char* query_type_as_string(QueryType q)
 195 {
 196     switch (q)
 197     {
 198     case QueryType_Prefix: return "prefix (RPN sent to server)";
 199     case QueryType_CCL: return "CCL (CCL sent to server) ";
 200     case QueryType_CCL2RPN: return "CCL -> RPN (RPN sent to server)";
 201     case QueryType_CQL: return "CQL (CQL sent to server)";
 202     case QueryType_CQL2RPN: return "CQL -> RPN (RPN sent to server)";
 203     default:
 204         return "unknown Query type internal yaz-client error";
 205     }
 206 }
 207
 208 static void do_hex_dump(const char* buf, size_t len)
 209 {
 210     if (hex_dump)
 211     {
 212         size_t i;
 213         int x;
 214         for (i = 0; i < len ; i = i+16 )
 215         {
 216             printf(" %4.4ld ", (long) i);
 217             for (x = 0 ; i+x < len && x < 16; ++x)
 218             {
 219                 printf("%2.2X ",(unsigned int)((unsigned char)buf[i+x]));
 220             }
 221             printf("\n");
 222         }
 223     }
 224     if (dump_file_prefix)
 225     {
 226         static int no = 0;
 227         if (++no < 1000 && strlen(dump_file_prefix) < 500)
 228         {
 229             char fname[1024];
 230             FILE *of;
 231             sprintf(fname, "%s.%03d.raw", dump_file_prefix, no);
 232             of = fopen(fname, "wb");
 233
 234             if (fwrite(buf, 1, len, of) != len)
 235             {
 236                 printf("write failed for %s", fname);
 237             }
 238             if (fclose(of))
 239             {
 240                 printf("close failed for %s", fname);
 241             }
 242         }
 243     }
 244 }
 245
 246 static void add_otherInfos(Z_APDU *a)
 247 {
 248     Z_OtherInformation **oi;
 249     int i;
 250
 251     yaz_oi_APDU(a, &oi);
 252     if (facet_list)
 253         yaz_oi_set_facetlist(oi, out, facet_list);
 254     for (i = 0; i < maxOtherInfosSupported; ++i)
 255     {
 256         if (oid_oidlen(extraOtherInfos[i].oid) > 0)
 257             yaz_oi_set_string_oid(oi, out, extraOtherInfos[i].oid,
 258                                      1, extraOtherInfos[i].value);
 259     }
 260 }
 261
 262 int send_apdu(Z_APDU *a)
 263 {
 264     char *buf;
 265     int len;
 266
 267     add_otherInfos(a);
 268
 269     if (apdu_file)
 270     {
 271         z_APDU(print, &a, 0, 0);
 272         odr_reset(print);
 273     }
 274     if (!z_APDU(out, &a, 0, 0))
 275     {
 276         odr_perror(out, "Encoding APDU");
 277         close_session();
 278         return 0;
 279     }
 280     buf = odr_getbuf(out, &len, 0);
 281     if (ber_file)
 282         odr_dumpBER(ber_file, buf, len);
 283     /* printf ("sending APDU of size %d\n", len); */
 284     do_hex_dump(buf, len);
 285     if (cs_put(conn, buf, len) < 0)
 286     {
 287         fprintf(stderr, "cs_put: %s", cs_errmsg(cs_errno(conn)));
 288         close_session();
 289         return 0;
 290     }
 291     odr_reset(out); /* release the APDU structure  */
 292     return 1;
 293 }
 294
 295 static void print_stringn(const char *buf, size_t len)
 296 {
 297     size_t i;
 298     for (i = 0; i < len; i++)
 299         if ((buf[i] <= 126 && buf[i] >= 32) || strchr("\n\r\t\f", buf[i]))
 300             printf("%c", buf[i]);
 301         else
 302             printf("\\X%02X", ((const unsigned char *)buf)[i]);
 303 }
 304
 305 static void print_refid(Z_ReferenceId *id)
 306 {
 307     if (id)
 308     {
 309         printf("Reference Id: ");
 310         print_stringn((const char *) id->buf, id->len);
 311         printf("\n");
 312     }
 313 }
 314
 315 static Z_ReferenceId *set_refid(ODR out)
 316 {
 317     Z_ReferenceId *id;
 318     if (!refid)
 319         return 0;
 320     id = (Z_ReferenceId *) odr_malloc(out, sizeof(*id));
 321     id->size = id->len = strlen(refid);
 322     id->buf = (unsigned char *) odr_malloc(out, id->len);
 323     memcpy(id->buf, refid, id->len);
 324     return id;
 325 }
 326
 327 /* INIT SERVICE ------------------------------- */
 328
 329 static void send_Z3950_initRequest(const char* type_and_host)
 330 {
 331     Z_APDU *apdu = zget_APDU(out, Z_APDU_initRequest);
 332     Z_InitRequest *req = apdu->u.initRequest;
 333     int i;
 334
 335     req->options = &z3950_options;
 336
 337     ODR_MASK_ZERO(req->protocolVersion);
 338     for (i = 0; i<z3950_version; i++)
 339         ODR_MASK_SET(req->protocolVersion, i);
 340
 341     *req->maximumRecordSize = 1024*kilobytes;
 342     *req->preferredMessageSize = 1024*kilobytes;
 343
 344     req->idAuthentication = auth;
 345
 346     req->referenceId = set_refid(out);
 347
 348     if (yazProxy && type_and_host)
 349     {
 350         yaz_oi_set_string_oid(&req->otherInfo, out, yaz_oid_userinfo_proxy,
 351                               1, type_and_host);
 352     }
 353
 354     if (negotiationCharset || yazLang)
 355     {
 356         Z_OtherInformation **p;
 357         Z_OtherInformationUnit *p0;
 358
 359         yaz_oi_APDU(apdu, &p);
 360
 361         if ((p0=yaz_oi_update(p, out, NULL, 0, 0)))
 362         {
 363             ODR_MASK_SET(req->options, Z_Options_negotiationModel);
 364
 365             p0->which = Z_OtherInfo_externallyDefinedInfo;
 366             p0->information.externallyDefinedInfo =
 367                 yaz_set_proposal_charneg_list(out, ",",
 368                                               negotiationCharset,
 369                                               yazLang,
 370                                               negotiationCharsetRecords);
 371         }
 372     }
 373     if (send_apdu(apdu))
 374         printf("Sent initrequest.\n");
 375 }
 376
 377
 378 static void render_initUserInfo(Z_OtherInformation *ui1);
 379 static void render_diag(Z_DiagnosticFormat *diag);
 380
 381 static void pr_opt(const char *opt, void *clientData)
 382 {
 383     printf(" %s", opt);
 384 }
 385
 386 static int process_Z3950_initResponse(Z_InitResponse *res)
 387 {
 388     int ver = 0;
 389     /* save session parameters for later use */
 390     session_mem = odr_extract_mem(in);
 391     session_initResponse = res;
 392
 393     for (ver = 0; ver < 8; ver++)
 394         if (!ODR_MASK_GET(res->protocolVersion, ver))
 395             break;
 396
 397     if (!*res->result)
 398         printf("Connection rejected by v%d target.\n", ver);
 399     else
 400         printf("Connection accepted by v%d target.\n", ver);
 401     if (res->implementationId)
 402         printf("ID     : %s\n", res->implementationId);
 403     if (res->implementationName)
 404         printf("Name   : %s\n", res->implementationName);
 405     if (res->implementationVersion)
 406         printf("Version: %s\n", res->implementationVersion);
 407     if (res->userInformationField)
 408     {
 409         Z_External *uif = res->userInformationField;
 410         if (uif->which == Z_External_userInfo1)
 411             render_initUserInfo(uif->u.userInfo1);
 412         else
 413         {
 414             printf("UserInformationfield:\n");
 415             if (!z_External(print, (Z_External**)&uif, 0, 0))
 416             {
 417                 odr_perror(print, "Printing userinfo\n");
 418                 odr_reset(print);
 419             }
 420             if (uif->which == Z_External_octet)
 421             {
 422                 printf("Guessing visiblestring:\n");
 423                 printf("'%.*s'\n", uif->u.octet_aligned->len,
 424                        uif->u.octet_aligned->buf);
 425             }
 426             else if (uif->which == Z_External_single)
 427             {
 428                 Odr_any *sat = uif->u.single_ASN1_type;
 429                 if (!oid_oidcmp(uif->direct_reference,
 430                                 yaz_oid_userinfo_oclc_userinfo))
 431                 {
 432                     Z_OCLC_UserInformation *oclc_ui;
 433                     ODR decode = odr_createmem(ODR_DECODE);
 434                     odr_setbuf(decode, (char *) sat->buf, sat->len, 0);
 435                     if (!z_OCLC_UserInformation(decode, &oclc_ui, 0, 0))
 436                         printf("Bad OCLC UserInformation:\n");
 437                     else
 438                         printf("OCLC UserInformation:\n");
 439                     if (!z_OCLC_UserInformation(print, &oclc_ui, 0, 0))
 440                         printf("Bad OCLC UserInformation spec\n");
 441                     odr_destroy(decode);
 442                 }
 443                 else
 444                 {
 445                     /* Peek at any private Init-diagnostic APDUs */
 446                     printf("yaz-client ignoring unrecognised userInformationField: %d-octet External '%.*s'\n",
 447                            (int) sat->len, sat->len, sat->buf);
 448                 }
 449             }
 450             odr_reset(print);
 451         }
 452     }
 453     printf("Options:");
 454     yaz_init_opt_decode(res->options, pr_opt, 0);
 455     printf("\n");
 456
 457     if (ODR_MASK_GET(res->options, Z_Options_namedResultSets))
 458         setnumber = 0;
 459
 460     if (ODR_MASK_GET(res->options, Z_Options_negotiationModel))
 461     {
 462         Z_CharSetandLanguageNegotiation *p =
 463                 yaz_get_charneg_record(res->otherInfo);
 464
 465         if (p)
 466         {
 467             char *charset=NULL, *lang=NULL;
 468             int selected;
 469
 470             yaz_get_response_charneg(session_mem, p, &charset, &lang,
 471                                      &selected);
 472
 473             printf("Accepted character set : %s\n", charset ? charset:"none");
 474             printf("Accepted code language : %s\n", lang ? lang:"none");
 475             printf("Accepted records in ...: %d\n", selected );
 476
 477             if (outputCharset && charset)
 478             {
 479                 printf("Converting between %s and %s\n",
 480                        outputCharset, charset);
 481                 odr_set_charset(out, charset, outputCharset);
 482                 odr_set_charset(in, outputCharset, charset);
 483                 cmd_querycharset(charset);
 484             }
 485             else
 486             {
 487                 odr_set_charset(out, 0, 0);
 488                 odr_set_charset(in, 0, 0);
 489             }
 490         }
 491     }
 492     fflush(stdout);
 493     return 0;
 494 }
 495
 496
 497 static void render_initUserInfo(Z_OtherInformation *ui1)
 498 {
 499     int i;
 500     printf("Init response contains %d otherInfo unit%s:\n",
 501            ui1->num_elements, ui1->num_elements == 1 ? "" : "s");
 502
 503     for (i = 0; i < ui1->num_elements; i++)
 504     {
 505         Z_OtherInformationUnit *unit = ui1->list[i];
 506         printf("  %d: otherInfo unit contains ", i+1);
 507         if (unit->which == Z_OtherInfo_externallyDefinedInfo &&
 508             unit->information.externallyDefinedInfo &&
 509             unit->information.externallyDefinedInfo->which ==
 510             Z_External_diag1)
 511         {
 512             render_diag(unit->information.externallyDefinedInfo->u.diag1);
 513         }
 514         else if (unit->which != Z_OtherInfo_externallyDefinedInfo)
 515         {
 516             printf("unsupported otherInfo unit->which = %d\n", unit->which);
 517         }
 518         else
 519         {
 520             printf("unsupported otherInfo unit external %d\n",
 521                    unit->information.externallyDefinedInfo ?
 522                    unit->information.externallyDefinedInfo->which : -2);
 523         }
 524     }
 525 }
 526
 527
 528 /* ### should this share code with display_diagrecs()? */
 529 static void render_diag(Z_DiagnosticFormat *diag)
 530 {
 531     int i;
 532
 533     printf("%d diagnostic%s:\n", diag->num, diag->num == 1 ? "" : "s");
 534     for (i = 0; i < diag->num; i++)
 535     {
 536         Z_DiagnosticFormat_s *ds = diag->elements[i];
 537         printf("    %d: ", i+1);
 538         switch (ds->which)
 539         {
 540         case Z_DiagnosticFormat_s_defaultDiagRec: {
 541             Z_DefaultDiagFormat *dd = ds->u.defaultDiagRec;
 542             /* ### should check `dd->diagnosticSetId' */
 543             printf("code=" ODR_INT_PRINTF " (%s)", *dd->condition,
 544                    diagbib1_str(*dd->condition));
 545             /* Both types of addinfo are the same, so use type-pun */
 546             if (dd->u.v2Addinfo != 0)
 547                 printf(",\n\taddinfo='%s'", dd->u.v2Addinfo);
 548             break;
 549         }
 550         case Z_DiagnosticFormat_s_explicitDiagnostic:
 551             printf("Explicit diagnostic (not supported)");
 552             break;
 553         default:
 554             printf("Unrecognised diagnostic type %d", ds->which);
 555             break;
 556         }
 557
 558         if (ds->message != 0)
 559             printf(", message='%s'", ds->message);
 560         printf("\n");
 561     }
 562 }
 563
 564
 565 static int set_base(const char *arg)
 566 {
 567     int i;
 568     const char *cp;
 569
 570     for (i = 0; i<num_databaseNames; i++)
 571         xfree(databaseNames[i]);
 572     num_databaseNames = 0;
 573     while (1)
 574     {
 575         char *cp1;
 576         if (!(cp = strchr(arg, ' ')))
 577             cp = arg + strlen(arg);
 578         if (cp - arg < 1)
 579             break;
 580         databaseNames[num_databaseNames] = (char *)xmalloc(1 + cp - arg);
 581         memcpy(databaseNames[num_databaseNames], arg, cp - arg);
 582         databaseNames[num_databaseNames][cp - arg] = '\0';
 583
 584         for (cp1 = databaseNames[num_databaseNames]; *cp1 ; cp1++)
 585             if (*cp1 == '+')
 586                 *cp1 = ' ';
 587         num_databaseNames++;
 588
 589         if (!*cp)
 590             break;
 591         arg = cp+1;
 592     }
 593     if (num_databaseNames == 0)
 594     {
 595         num_databaseNames = 1;
 596         databaseNames[0] = xstrdup("");
 597     }
 598     return 1;
 599 }
 600
 601 static int parse_cmd_doc(const char **arg, ODR out, char **buf, int *len)
 602 {
 603     const char *sep;
 604     while (**arg && strchr(" \t\n\r\f", **arg))
 605         (*arg)++;
 606     if (**arg == '\0')
 607     {
 608         return 0;
 609     }
 610     else if ((*arg)[0] == '<')
 611     {
 612         size_t fsize;
 613         FILE *inf;
 614         const char *fname;
 615         const char *arg_start = ++(*arg);
 616
 617         while (**arg != '\0' && **arg != ' ')
 618             (*arg)++;
 619
 620         fname = odr_strdupn(out, arg_start, *arg - arg_start);
 621
 622         inf = fopen(fname, "rb");
 623         if (!inf)
 624         {
 625             printf("Couldn't open %s\n", fname);
 626             return 0;
 627         }
 628         if (fseek(inf, 0L, SEEK_END) == -1)
 629         {
 630             printf("Couldn't seek in %s\n", fname);
 631             fclose(inf);
 632             return 0;
 633         }
 634         fsize = ftell(inf);
 635         if (fseek(inf, 0L, SEEK_SET) == -1)
 636         {
 637             printf("Couldn't seek in %s\n", fname);
 638             fclose(inf);
 639             return 0;
 640         }
 641         *len = fsize;
 642         *buf = (char *) odr_malloc(out, fsize+1);
 643         (*buf)[fsize] = '\0';
 644         if (fread(*buf, 1, fsize, inf) != fsize)
 645         {
 646             printf("Unable to read %s\n", fname);
 647             fclose(inf);
 648             return 0;
 649         }
 650         fclose(inf);
 651     }
 652     else if ((*arg)[0] == '\"' && (sep=strchr(*arg+1, '"')))
 653     {
 654         (*arg)++;
 655         *len = sep - *arg;
 656         *buf = odr_strdupn(out, *arg, *len);
 657         (*arg) = sep+1;
 658     }
 659     else
 660     {
 661         const char *arg_start = *arg;
 662
 663         while (**arg != '\0' && **arg != ' ')
 664             (*arg)++;
 665
 666         *len = *arg - arg_start;
 667         *buf = odr_strdupn(out, arg_start, *len);
 668     }
 669     return 1;
 670 }
 671
 672 static int cmd_base(const char *arg)
 673 {
 674     if (!*arg)
 675     {
 676         printf("Usage: base <database> <database> ...\n");
 677         return 0;
 678     }
 679     return set_base(arg);
 680 }
 681
 682 static int session_connect_base(const char *arg, const char **basep)
 683 {
 684     void *add;
 685     char type_and_host[101];
 686     if (conn)
 687     {
 688         cs_close(conn);
 689         conn = 0;
 690     }
 691     if (session_mem)
 692     {
 693         nmem_destroy(session_mem);
 694         session_mem = NULL;
 695         session_initResponse = 0;
 696     }
 697     cs_get_host_args(arg, basep);
 698
 699     strncpy(type_and_host, arg, sizeof(type_and_host)-1);
 700     type_and_host[sizeof(type_and_host)-1] = '\0';
 701
 702     if (yazProxy)
 703         conn = cs_create_host(yazProxy, 1, &add);
 704     else
 705         conn = cs_create_host(arg, 1, &add);
 706     if (!conn)
 707     {
 708         printf("Could not resolve address %s\n", arg);
 709         return 0;
 710     }
 711 #if YAZ_HAVE_XML2
 712 #else
 713     if (conn->protocol == PROTO_HTTP)
 714     {
 715         printf("SRW/HTTP not enabled in this YAZ\n");
 716         cs_close(conn);
 717         conn = 0;
 718         return 0;
 719     }
 720 #endif
 721     protocol = conn->protocol;
 722     printf("Connecting...");
 723     fflush(stdout);
 724     if (cs_connect(conn, add) < 0)
 725     {
 726         printf("error = %s\n", cs_strerror(conn));
 727         cs_close(conn);
 728         conn = 0;
 729         return 0;
 730     }
 731     printf("OK.\n");
 732     cs_print_session_info(conn);
 733     if (protocol == PROTO_Z3950)
 734     {
 735         send_Z3950_initRequest(type_and_host);
 736         return 2;
 737     }
 738     return 0;
 739 }
 740
 741 static int session_connect(const char *arg)
 742 {
 743     int r;
 744     const char *basep = 0;
 745
 746     r = session_connect_base(arg, &basep);
 747     if (basep && *basep)
 748         set_base(basep);
 749     else if (protocol == PROTO_Z3950)
 750         set_base("Default");
 751     return r;
 752 }
 753
 754 static int cmd_open(const char *arg)
 755 {
 756     int r;
 757     if (arg)
 758     {
 759         strncpy(cur_host, arg, sizeof(cur_host)-1);
 760         cur_host[sizeof(cur_host)-1] = 0;
 761     }
 762     /* TODO Make facet definition survive the open command without crashing */
 763     /* TODO Fix deallocation */
 764     facet_list = 0;
 765
 766     set_base("");
 767     r = session_connect(cur_host);
 768     if (conn && conn->protocol == PROTO_HTTP)
 769         queryType = QueryType_CQL;
 770
 771
 772     return r;
 773 }
 774
 775 static int cmd_authentication(const char *arg)
 776 {
 777     char **args;
 778     int r;
 779
 780     nmem_reset(nmem_auth);
 781     nmem_strsplit_blank(nmem_auth, arg, &args, &r);
 782
 783     if (r == 0)
 784     {
 785         printf("Authentication set to null\n");
 786         auth = 0;
 787     }
 788     else if (r == 1)
 789     {
 790         auth = (Z_IdAuthentication *) nmem_malloc(nmem_auth, sizeof(*auth));
 791         if (!strcmp(args[0], "-"))
 792         {
 793             auth->which = Z_IdAuthentication_anonymous;
 794             auth->u.anonymous = odr_nullval();
 795             printf("Authentication set to Anonymous\n");
 796         }
 797         else
 798         {
 799             auth->which = Z_IdAuthentication_open;
 800             auth->u.open = args[0];
 801             printf("Authentication set to Open (%s)\n", args[0]);
 802         }
 803     }
 804     else if (r == 2)
 805     {
 806         auth = (Z_IdAuthentication *) nmem_malloc(nmem_auth, sizeof(*auth));
 807         auth->which = Z_IdAuthentication_idPass;
 808         auth->u.idPass = (Z_IdPass *)
 809             nmem_malloc(nmem_auth, sizeof(*auth->u.idPass));
 810         auth->u.idPass->groupId = NULL;
 811         auth->u.idPass->userId = !strcmp(args[0], "-") ? 0 : args[0];
 812         auth->u.idPass->password = !strcmp(args[1], "-") ? 0 : args[1];
 813         printf("Authentication set to User (%s), Pass (%s)\n",
 814                args[0], args[1]);
 815     }
 816     else if (r == 3)
 817     {
 818         auth = (Z_IdAuthentication*) nmem_malloc(nmem_auth, sizeof(*auth));
 819         auth->which = Z_IdAuthentication_idPass;
 820         auth->u.idPass = (Z_IdPass *)
 821             nmem_malloc(nmem_auth, sizeof(*auth->u.idPass));
 822         auth->u.idPass->groupId = args[1];
 823         auth->u.idPass->userId = args[0];
 824         auth->u.idPass->password = args[2];
 825         printf("Authentication set to User (%s), Group (%s), Pass (%s)\n",
 826                args[0], args[1], args[2]);
 827     }
 828     else
 829     {
 830         printf("Bad number of args to auth\n");
 831         auth = 0;
 832     }
 833
 834     return 1;
 835 }
 836
 837 /* SEARCH SERVICE ------------------------------ */
 838 static void display_record(Z_External *r);
 839
 840 static void print_record(const char *buf, size_t len)
 841 {
 842     size_t i = len;
 843     print_stringn(buf, len);
 844     /* add newline if not already added ... */
 845     if (i <= 0 || buf[i-1] != '\n')
 846         printf("\n");
 847 }
 848
 849 static void display_record(Z_External *r)
 850 {
 851     const Odr_oid *oid = r->direct_reference;
 852
 853     record_last = r;
 854     /*
 855      * Tell the user what we got.
 856      */
 857     if (oid)
 858     {
 859         oid_class oclass;
 860         char oid_name_buf[OID_STR_MAX];
 861         const char *oid_name
 862             =  yaz_oid_to_string_buf(oid, &oclass, oid_name_buf);
 863         printf("Record type: ");
 864         if (oid_name)
 865             printf("%s\n", oid_name);
 866     }
 867     /* Check if this is a known, ASN.1 type tucked away in an octet string */
 868     if (r->which == Z_External_octet)
 869     {
 870         Z_ext_typeent *type = z_ext_getentbyref(r->direct_reference);
 871         char *rr;
 872
 873         if (type)
 874         {
 875             /*
 876              * Call the given decoder to process the record.
 877              */
 878             odr_setbuf(in, (char*)r->u.octet_aligned->buf,
 879                 r->u.octet_aligned->len, 0);
 880             if (!(*type->fun)(in, &rr, 0, 0))
 881             {
 882                 odr_perror(in, "Decoding constructed record.");
 883                 fprintf(stdout, "[Near %ld]\n", (long) odr_offset(in));
 884                 fprintf(stdout, "Packet dump:\n---------\n");
 885                 odr_dumpBER(stdout, (char*)r->u.octet_aligned->buf,
 886                             r->u.octet_aligned->len);
 887                 fprintf(stdout, "---------\n");
 888
 889                 /* note just ignores the error ant print the bytes form the octet_aligned later */
 890             } else {
 891                 /*
 892                  * Note: we throw away the original, BER-encoded record here.
 893                  * Do something else with it if you want to keep it.
 894                  */
 895                 r->u.sutrs = (Z_SUTRS *) rr; /* we don't actually check the type here. */
 896                 r->which = type->what;
 897             }
 898         }
 899     }
 900     if (oid && r->which == Z_External_octet)
 901     {
 902         const char *octet_buf = (const char*)r->u.octet_aligned->buf;
 903         size_t octet_len = r->u.octet_aligned->len;
 904         if (!oid_oidcmp(oid, yaz_oid_recsyn_xml)
 905             || !oid_oidcmp(oid, yaz_oid_recsyn_xml)
 906             || !oid_oidcmp(oid, yaz_oid_recsyn_html))
 907         {
 908             fwrite(octet_buf, 1, octet_len, stdout);
 909         }
 910         else if (yaz_oid_is_iso2709(oid))
 911         {
 912             const char *result;
 913             size_t rlen;
 914             yaz_iconv_t cd = 0;
 915             yaz_marc_t mt = yaz_marc_create();
 916             const char *from = 0;
 917
 918             if (marcCharset && !strcmp(marcCharset, "auto"))
 919             {
 920                 if (!oid_oidcmp(oid, yaz_oid_recsyn_usmarc))
 921                 {
 922                     if (octet_buf[9] == 'a')
 923                         from = "UTF-8";
 924                     else
 925                         from = "MARC-8";
 926                 }
 927                 else
 928                     from = "ISO-8859-1";
 929             }
 930             else if (marcCharset)
 931                 from = marcCharset;
 932             if (outputCharset && from)
 933             {
 934                 cd = yaz_iconv_open(outputCharset, from);
 935                 printf("convert from %s to %s", from,
 936                        outputCharset);
 937                 if (!cd)
 938                     printf(" unsupported\n");
 939                 else
 940                 {
 941                     yaz_marc_iconv(mt, cd);
 942                     printf("\n");
 943                 }
 944             }
 945
 946             if (yaz_marc_decode_buf(mt, octet_buf, octet_len,
 947                                     &result, &rlen)> 0)
 948             {
 949                 if (fwrite(result, rlen, 1, stdout) != 1)
 950                 {
 951                     printf("write to stdout failed\n");
 952                 }
 953             }
 954             else
 955             {
 956                 printf("bad MARC. Dumping as it is:\n");
 957                 print_record(octet_buf, octet_len);
 958             }
 959             yaz_marc_destroy(mt);
 960             if (cd)
 961                 yaz_iconv_close(cd);
 962         }
 963         else
 964         {
 965             print_record(octet_buf, octet_len);
 966         }
 967         marc_file_write(octet_buf, r->u.octet_aligned->len);
 968     }
 969     else if (oid && !oid_oidcmp(oid, yaz_oid_recsyn_sutrs))
 970     {
 971         if (r->which != Z_External_sutrs)
 972         {
 973             printf("Expecting single SUTRS type for SUTRS.\n");
 974             return;
 975         }
 976         print_record((const char *) r->u.sutrs->buf, r->u.sutrs->len);
 977         marc_file_write((const char *) r->u.sutrs->buf, r->u.sutrs->len);
 978     }
 979     else if (oid && !oid_oidcmp(oid, yaz_oid_recsyn_grs_1))
 980     {
 981         WRBUF w;
 982         if (r->which != Z_External_grs1)
 983         {
 984             printf("Expecting single GRS type for GRS.\n");
 985             return;
 986         }
 987         w = wrbuf_alloc();
 988         yaz_display_grs1(w, r->u.grs1, 0);
 989         puts(wrbuf_cstr(w));
 990         wrbuf_destroy(w);
 991     }
 992     else if (oid && !oid_oidcmp(oid, yaz_oid_recsyn_opac))
 993     {
 994         int i;
 995         if (r->u.opac->bibliographicRecord)
 996             display_record(r->u.opac->bibliographicRecord);
 997         for (i = 0; i<r->u.opac->num_holdingsData; i++)
 998         {
 999             Z_HoldingsRecord *h = r->u.opac->holdingsData[i];
1000             if (h->which == Z_HoldingsRecord_marcHoldingsRecord)
1001             {
1002                 printf("MARC holdings %d\n", i);
1003                 display_record(h->u.marcHoldingsRecord);
1004             }
1005             else if (h->which == Z_HoldingsRecord_holdingsAndCirc)
1006             {
1007                 int j;
1008
1009                 Z_HoldingsAndCircData *data = h->u.holdingsAndCirc;
1010
1011                 printf("Data holdings %d\n", i);
1012                 if (data->typeOfRecord)
1013                     printf("typeOfRecord: %s\n", data->typeOfRecord);
1014                 if (data->encodingLevel)
1015                     printf("encodingLevel: %s\n", data->encodingLevel);
1016                 if (data->receiptAcqStatus)
1017                     printf("receiptAcqStatus: %s\n", data->receiptAcqStatus);
1018                 if (data->generalRetention)
1019                     printf("generalRetention: %s\n", data->generalRetention);
1020                 if (data->completeness)
1021                     printf("completeness: %s\n", data->completeness);
1022                 if (data->dateOfReport)
1023                     printf("dateOfReport: %s\n", data->dateOfReport);
1024                 if (data->nucCode)
1025                     printf("nucCode: %s\n", data->nucCode);
1026                 if (data->localLocation)
1027                     printf("localLocation: %s\n", data->localLocation);
1028                 if (data->shelvingLocation)
1029                     printf("shelvingLocation: %s\n", data->shelvingLocation);
1030                 if (data->callNumber)
1031                     printf("callNumber: %s\n", data->callNumber);
1032                 if (data->shelvingData)
1033                     printf("shelvingData: %s\n", data->shelvingData);
1034                 if (data->copyNumber)
1035                     printf("copyNumber: %s\n", data->copyNumber);
1036                 if (data->publicNote)
1037                     printf("publicNote: %s\n", data->publicNote);
1038                 if (data->reproductionNote)
1039                     printf("reproductionNote: %s\n", data->reproductionNote);
1040                 if (data->termsUseRepro)
1041                     printf("termsUseRepro: %s\n", data->termsUseRepro);
1042                 if (data->enumAndChron)
1043                     printf("enumAndChron: %s\n", data->enumAndChron);
1044                 for (j = 0; j<data->num_volumes; j++)
1045                 {
1046                     printf("volume %d\n", j);
1047                     if (data->volumes[j]->enumeration)
1048                         printf(" enumeration: %s\n",
1049                                 data->volumes[j]->enumeration);
1050                     if (data->volumes[j]->chronology)
1051                         printf(" chronology: %s\n",
1052                                 data->volumes[j]->chronology);
1053                     if (data->volumes[j]->enumAndChron)
1054                         printf(" enumAndChron: %s\n",
1055                                 data->volumes[j]->enumAndChron);
1056                 }
1057                 for (j = 0; j<data->num_circulationData; j++)
1058                 {
1059                     printf("circulation %d\n", j);
1060                     if (data->circulationData[j]->availableNow)
1061                         printf(" availableNow: %d\n",
1062                                 *data->circulationData[j]->availableNow);
1063                     if (data->circulationData[j]->availablityDate)
1064                         printf(" availabiltyDate: %s\n",
1065                                 data->circulationData[j]->availablityDate);
1066                     if (data->circulationData[j]->availableThru)
1067                         printf(" availableThru: %s\n",
1068                                 data->circulationData[j]->availableThru);
1069                     if (data->circulationData[j]->restrictions)
1070                         printf(" restrictions: %s\n",
1071                                 data->circulationData[j]->restrictions);
1072                     if (data->circulationData[j]->itemId)
1073                         printf(" itemId: %s\n",
1074                                 data->circulationData[j]->itemId);
1075                     if (data->circulationData[j]->renewable)
1076                         printf(" renewable: %d\n",
1077                                 *data->circulationData[j]->renewable);
1078                     if (data->circulationData[j]->onHold)
1079                         printf(" onHold: %d\n",
1080                                 *data->circulationData[j]->onHold);
1081                     if (data->circulationData[j]->enumAndChron)
1082                         printf(" enumAndChron: %s\n",
1083                                 data->circulationData[j]->enumAndChron);
1084                     if (data->circulationData[j]->midspine)
1085                         printf(" midspine: %s\n",
1086                                 data->circulationData[j]->midspine);
1087                     if (data->circulationData[j]->temporaryLocation)
1088                         printf(" temporaryLocation: %s\n",
1089                                 data->circulationData[j]->temporaryLocation);
1090                 }
1091             }
1092         }
1093     }
1094     else
1095     {
1096         printf("Unknown record representation.\n");
1097         if (!z_External(print, &r, 0, 0))
1098         {
1099             odr_perror(print, "Printing external");
1100             odr_reset(print);
1101         }
1102     }
1103 }
1104
1105 static void display_diagrecs(Z_DiagRec **pp, int num)
1106 {
1107     int i;
1108     Z_DefaultDiagFormat *r;
1109
1110     printf("Diagnostic message(s) from database:\n");
1111     for (i = 0; i<num; i++)
1112     {
1113         Z_DiagRec *p = pp[i];
1114         if (p->which != Z_DiagRec_defaultFormat)
1115         {
1116             printf("Diagnostic record not in default format.\n");
1117             return;
1118         }
1119         else
1120             r = p->u.defaultFormat;
1121
1122         if (!r->diagnosticSetId)
1123             printf("Missing diagset\n");
1124         else
1125         {
1126             oid_class oclass;
1127             char diag_name_buf[OID_STR_MAX];
1128             const char *diag_name = 0;
1129             diag_name = yaz_oid_to_string_buf
1130                 (r->diagnosticSetId, &oclass, diag_name_buf);
1131             if (oid_oidcmp(r->diagnosticSetId, yaz_oid_diagset_bib_1))
1132                 printf("Unknown diagset: %s\n", diag_name);
1133         }
1134         printf("    [" ODR_INT_PRINTF "] %s",
1135                *r->condition, diagbib1_str(*r->condition));
1136         switch (r->which)
1137         {
1138         case Z_DefaultDiagFormat_v2Addinfo:
1139             printf(" -- v2 addinfo '%s'\n", r->u.v2Addinfo);
1140             break;
1141         case Z_DefaultDiagFormat_v3Addinfo:
1142             printf(" -- v3 addinfo '%s'\n", r->u.v3Addinfo);
1143             break;
1144         }
1145     }
1146 }
1147
1148
1149 static void display_nameplusrecord(Z_NamePlusRecord *p)
1150 {
1151     if (p->databaseName)
1152         printf("[%s]", p->databaseName);
1153     if (p->which == Z_NamePlusRecord_surrogateDiagnostic)
1154         display_diagrecs(&p->u.surrogateDiagnostic, 1);
1155     else if (p->which == Z_NamePlusRecord_databaseRecord)
1156         display_record(p->u.databaseRecord);
1157 }
1158
1159 static void display_records(Z_Records *p)
1160 {
1161     int i;
1162
1163     if (p->which == Z_Records_NSD)
1164     {
1165         Z_DiagRec dr, *dr_p = &dr;
1166         dr.which = Z_DiagRec_defaultFormat;
1167         dr.u.defaultFormat = p->u.nonSurrogateDiagnostic;
1168         display_diagrecs(&dr_p, 1);
1169     }
1170     else if (p->which == Z_Records_multipleNSD)
1171         display_diagrecs(p->u.multipleNonSurDiagnostics->diagRecs,
1172                           p->u.multipleNonSurDiagnostics->num_diagRecs);
1173     else
1174     {
1175         printf("Records: %d\n", p->u.databaseOrSurDiagnostics->num_records);
1176         for (i = 0; i < p->u.databaseOrSurDiagnostics->num_records; i++)
1177             display_nameplusrecord(p->u.databaseOrSurDiagnostics->records[i]);
1178     }
1179 }
1180
1181 static int send_Z3950_deleteResultSetRequest(const char *arg)
1182 {
1183     char names[8][32];
1184     int i;
1185
1186     Z_APDU *apdu = zget_APDU(out, Z_APDU_deleteResultSetRequest);
1187     Z_DeleteResultSetRequest *req = apdu->u.deleteResultSetRequest;
1188
1189     req->referenceId = set_refid(out);
1190
1191     req->num_resultSetList =
1192         sscanf(arg, "%30s %30s %30s %30s %30s %30s %30s %30s",
1193                names[0], names[1], names[2], names[3],
1194                names[4], names[5], names[6], names[7]);
1195
1196     req->deleteFunction = odr_intdup(out, 0);
1197     if (req->num_resultSetList > 0)
1198     {
1199         *req->deleteFunction = Z_DeleteResultSetRequest_list;
1200         req->resultSetList = (char **)
1201             odr_malloc(out, sizeof(*req->resultSetList)*
1202                         req->num_resultSetList);
1203         for (i = 0; i<req->num_resultSetList; i++)
1204             req->resultSetList[i] = names[i];
1205     }
1206     else
1207     {
1208         *req->deleteFunction = Z_DeleteResultSetRequest_all;
1209         req->resultSetList = 0;
1210     }
1211
1212     send_apdu(apdu);
1213     printf("Sent deleteResultSetRequest.\n");
1214     return 2;
1215 }
1216
1217 #if YAZ_HAVE_XML2
1218 static int send_gdu(Z_GDU *gdu)
1219 {
1220     if (z_GDU(out, &gdu, 0, 0))
1221     {
1222         /* encode OK */
1223         char *buf_out;
1224         int len_out;
1225         int r;
1226         if (apdu_file)
1227         {
1228             if (!z_GDU(print, &gdu, 0, 0))
1229                 printf("Failed to print outgoing SRU package\n");
1230             odr_reset(print);
1231         }
1232         buf_out = odr_getbuf(out, &len_out, 0);
1233
1234         /* we don't odr_reset(out), since we may need the buffer again */
1235
1236         do_hex_dump(buf_out, len_out);
1237
1238         r = cs_put(conn, buf_out, len_out);
1239
1240         if (r >= 0)
1241             return 2;
1242     }
1243     return 0;
1244 }
1245
1246 static int send_srw_host_path(Z_SRW_PDU *sr, const char *host_port,
1247                               char *path)
1248 {
1249     const char *charset = negotiationCharset;
1250     Z_GDU *gdu;
1251
1252     gdu = z_get_HTTP_Request_host_path(out, host_port, path);
1253
1254     if (auth)
1255     {
1256         if (auth->which == Z_IdAuthentication_open)
1257         {
1258             char **darray;
1259             int num;
1260             nmem_strsplit(out->mem, "/", auth->u.open, &darray, &num);
1261             if (num >= 1)
1262                 sr->username = darray[0];
1263             if (num >= 2)
1264                 sr->password = darray[1];
1265         }
1266         else if (auth->which == Z_IdAuthentication_idPass)
1267         {
1268             sr->username = auth->u.idPass->userId;
1269             sr->password = auth->u.idPass->password;
1270         }
1271     }
1272
1273     if (!yaz_matchstr(sru_method, "get"))
1274     {
1275         yaz_sru_get_encode(gdu->u.HTTP_Request, sr, out, charset);
1276     }
1277     else if (!yaz_matchstr(sru_method, "post"))
1278     {
1279         yaz_sru_post_encode(gdu->u.HTTP_Request, sr, out, charset);
1280     }
1281     else if (!yaz_matchstr(sru_method, "soap"))
1282     {
1283         yaz_sru_soap_encode(gdu->u.HTTP_Request, sr, out, charset);
1284     }
1285     else if (!yaz_matchstr(sru_method, "solr"))
1286     {
1287         yaz_solr_encode_request(gdu->u.HTTP_Request, sr, out, charset);
1288     }
1289
1290     return send_gdu(gdu);
1291 }
1292
1293 static int send_srw(Z_SRW_PDU *sr)
1294 {
1295     char *path = yaz_encode_sru_dbpath_odr(out, databaseNames[0]);
1296     return send_srw_host_path(sr, cur_host, path);
1297 }
1298
1299 static int send_SRW_redirect(const char *uri, Z_HTTP_Response *cookie_hres)
1300 {
1301     const char *username = 0;
1302     const char *password = 0;
1303     struct Z_HTTP_Header *h;
1304     char *combined_cookies = 0;
1305     int combined_cookies_len = 0;
1306     Z_GDU *gdu = get_HTTP_Request_url(out, uri);
1307
1308     gdu->u.HTTP_Request->method = odr_strdup(out, "GET");
1309     z_HTTP_header_add(out, &gdu->u.HTTP_Request->headers, "Accept",
1310                       "text/xml");
1311
1312     for (h = cookie_hres->headers; h; h = h->next)
1313     {
1314         if (!strcmp(h->name, "Set-Cookie"))
1315         {
1316             char *cp;
1317
1318             if (!(cp = strchr(h->value, ';')))
1319                 cp = h->value + strlen(h->value);
1320             if (cp - h->value >= 1)
1321             {
1322                 combined_cookies = xrealloc(combined_cookies, combined_cookies_len + cp - h->value + 3);
1323                 memcpy(combined_cookies+combined_cookies_len, h->value, cp - h->value);
1324                 combined_cookies[combined_cookies_len + cp - h->value] = '\0';
1325                 strcat(combined_cookies,"; ");
1326                 combined_cookies_len = strlen(combined_cookies);
1327             }
1328         }
1329     }
1330     if (combined_cookies_len)
1331     {
1332         z_HTTP_header_add(out, &gdu->u.HTTP_Request->headers, "Cookie", combined_cookies);
1333         xfree(combined_cookies);
1334     }
1335
1336     if (auth)
1337     {
1338         if (auth->which == Z_IdAuthentication_open)
1339         {
1340             char **darray;
1341             int num;
1342             nmem_strsplit(out->mem, "/", auth->u.open, &darray, &num);
1343             if (num >= 1)
1344                 username = darray[0];
1345             if (num >= 2)
1346                 password = darray[1];
1347         }
1348         else if (auth->which == Z_IdAuthentication_idPass)
1349         {
1350             username = auth->u.idPass->userId;
1351             password = auth->u.idPass->password;
1352         }
1353     }
1354
1355     if (username && password)
1356     {
1357         z_HTTP_header_add_basic_auth(out, &gdu->u.HTTP_Request->headers,
1358                                      username, password);
1359     }
1360
1361     return send_gdu(gdu);
1362 }
1363 #endif
1364
1365 #if YAZ_HAVE_XML2
1366 static char *encode_SRW_term(ODR o, const char *q)
1367 {
1368     const char *in_charset = "ISO-8859-1";
1369     WRBUF w = wrbuf_alloc();
1370     yaz_iconv_t cd;
1371     char *res;
1372     if (outputCharset)
1373         in_charset = outputCharset;
1374     cd = yaz_iconv_open("UTF-8", in_charset);
1375     if (!cd)
1376     {
1377         wrbuf_destroy(w);
1378         return odr_strdup(o, q);
1379     }
1380     wrbuf_iconv_write(w, cd, q, strlen(q));
1381     if (wrbuf_len(w))
1382         res = odr_strdup(o, wrbuf_cstr(w));
1383     else
1384         res = odr_strdup(o, q);
1385     yaz_iconv_close(cd);
1386     wrbuf_destroy(w);
1387     return res;
1388 }
1389
1390
1391 static int send_SRW_scanRequest(const char *arg, int pos, int num)
1392 {
1393     Z_SRW_PDU *sr = 0;
1394
1395     /* regular requestse .. */
1396     sr = yaz_srw_get_pdu(out, Z_SRW_scan_request, sru_version);
1397
1398     switch (queryType)
1399     {
1400     case QueryType_CQL:
1401         sr->u.scan_request->query_type = Z_SRW_query_type_cql;
1402         sr->u.scan_request->scanClause.cql = encode_SRW_term(out, arg);
1403         break;
1404     case QueryType_Prefix:
1405         sr->u.scan_request->query_type = Z_SRW_query_type_pqf;
1406         sr->u.scan_request->scanClause.pqf = encode_SRW_term(out, arg);
1407         break;
1408     default:
1409         printf("Only CQL and PQF supported in SRW\n");
1410         return 0;
1411     }
1412     sr->u.scan_request->responsePosition = odr_intdup(out, pos);
1413     sr->u.scan_request->maximumTerms = odr_intdup(out, num);
1414     return send_srw(sr);
1415 }
1416
1417 static int send_SRW_searchRequest(const char *arg)
1418 {
1419     Z_SRW_PDU *sr = 0;
1420
1421     if (!srw_sr)
1422     {
1423         assert(srw_sr_odr_out == 0);
1424         srw_sr_odr_out = odr_createmem(ODR_ENCODE);
1425     }
1426     odr_reset(srw_sr_odr_out);
1427
1428     setno = 1;
1429
1430     /* save this for later .. when fetching individual records */
1431     srw_sr =  yaz_srw_get_pdu(srw_sr_odr_out, Z_SRW_searchRetrieve_request,
1432                               sru_version);
1433
1434     /* regular request .. */
1435     sr = yaz_srw_get_pdu(out, Z_SRW_searchRetrieve_request, sru_version);
1436
1437     switch (queryType)
1438     {
1439     case QueryType_CQL:
1440         srw_sr->u.request->query_type = Z_SRW_query_type_cql;
1441         srw_sr->u.request->query.cql = encode_SRW_term(srw_sr_odr_out, arg);
1442
1443         sr->u.request->query_type = Z_SRW_query_type_cql;
1444         sr->u.request->query.cql = encode_SRW_term(srw_sr_odr_out, arg);
1445         break;
1446     case QueryType_Prefix:
1447         srw_sr->u.request->query_type = Z_SRW_query_type_pqf;
1448         srw_sr->u.request->query.pqf = encode_SRW_term(srw_sr_odr_out, arg);
1449
1450         sr->u.request->query_type = Z_SRW_query_type_pqf;
1451         sr->u.request->query.pqf = encode_SRW_term(srw_sr_odr_out, arg);
1452         break;
1453     default:
1454         printf("Only CQL and PQF supported in SRW\n");
1455         return 0;
1456     }
1457     sr->u.request->maximumRecords = odr_intdup(out, 0);
1458     sr->u.request->facetList = facet_list;
1459     if (record_schema)
1460         sr->u.request->recordSchema = record_schema;
1461     if (recordsyntax_size == 1 && !yaz_matchstr(recordsyntax_list[0], "xml"))
1462         sr->u.request->recordPacking = "xml";
1463     return send_srw(sr);
1464 }
1465 #endif
1466
1467 static void query_charset_convert(Z_RPNQuery *q)
1468 {
1469     if (queryCharset && outputCharset)
1470     {
1471         yaz_iconv_t cd = yaz_iconv_open(queryCharset, outputCharset);
1472         if (!cd)
1473         {
1474             printf("Conversion from %s to %s unsupported\n",
1475                    outputCharset, queryCharset);
1476             return;
1477         }
1478         yaz_query_charset_convert_rpnquery(q, out, cd);
1479         yaz_iconv_close(cd);
1480     }
1481 }
1482
1483 static int send_Z3950_searchRequest(const char *arg)
1484 {
1485     Z_APDU *apdu = zget_APDU(out, Z_APDU_searchRequest);
1486     Z_SearchRequest *req = apdu->u.searchRequest;
1487     Z_Query query;
1488     struct ccl_rpn_node *rpn = NULL;
1489     int error, pos;
1490     char setstring[100];
1491     Z_RPNQuery *RPNquery;
1492     Odr_oct ccl_query;
1493     YAZ_PQF_Parser pqf_parser;
1494     Z_External *ext;
1495     QueryType myQueryType = queryType;
1496     char pqfbuf[512];
1497
1498     if (myQueryType == QueryType_CCL2RPN)
1499     {
1500         rpn = ccl_find_str(bibset, arg, &error, &pos);
1501         if (error)
1502         {
1503             printf("CCL ERROR: %s\n", ccl_err_msg(error));
1504             return 0;
1505         }
1506     }
1507     else if (myQueryType == QueryType_CQL2RPN)
1508     {
1509         /* ### All this code should be wrapped in a utility function */
1510         CQL_parser parser;
1511         struct cql_node *node;
1512         const char *addinfo;
1513         if (cqltrans == 0)
1514         {
1515             printf("Can't use CQL: no translation file.  Try set_cqlfile\n");
1516             return 0;
1517         }
1518         parser = cql_parser_create();
1519         if ((error = cql_parser_string(parser, arg)) != 0)
1520         {
1521             printf("Can't parse CQL: must be a syntax error\n");
1522             return 0;
1523         }
1524         node = cql_parser_result(parser);
1525         if ((error = cql_transform_buf(cqltrans, node, pqfbuf,
1526                                        sizeof pqfbuf)) != 0)
1527         {
1528             error = cql_transform_error(cqltrans, &addinfo);
1529             printf("Can't convert CQL to PQF: %s (addinfo=%s)\n",
1530                     cql_strerror(error), addinfo);
1531             return 0;
1532         }
1533         arg = pqfbuf;
1534         myQueryType = QueryType_Prefix;
1535     }
1536
1537     req->referenceId = set_refid(out);
1538     if (!strcmp(arg, "@big")) /* strictly for troublemaking */
1539     {
1540         static unsigned char big[2100];
1541         static Odr_oct bigo;
1542
1543         /* send a very big referenceid to test transport stack etc. */
1544         memset(big, 'A', 2100);
1545         bigo.len = bigo.size = 2100;
1546         bigo.buf = big;
1547         req->referenceId = &bigo;
1548     }
1549
1550     if (setnumber >= 0)
1551     {
1552         sprintf(setstring, "%d", ++setnumber);
1553         req->resultSetName = setstring;
1554     }
1555     *req->smallSetUpperBound = smallSetUpperBound;
1556     *req->largeSetLowerBound = largeSetLowerBound;
1557     *req->mediumSetPresentNumber = mediumSetPresentNumber;
1558     if (smallSetUpperBound > 0 || (largeSetLowerBound > 1 &&
1559         mediumSetPresentNumber > 0))
1560     {
1561         if (recordsyntax_size)
1562             req->preferredRecordSyntax =
1563                 yaz_string_to_oid_odr(yaz_oid_std(),
1564                                       CLASS_RECSYN, recordsyntax_list[0], out);
1565
1566         req->smallSetElementSetNames =
1567             req->mediumSetElementSetNames = elementSetNames;
1568     }
1569     req->num_databaseNames = num_databaseNames;
1570     req->databaseNames = databaseNames;
1571
1572     req->query = &query;
1573
1574     switch (myQueryType)
1575     {
1576     case QueryType_Prefix:
1577         query.which = Z_Query_type_1;
1578         pqf_parser = yaz_pqf_create();
1579         RPNquery = yaz_pqf_parse(pqf_parser, out, arg);
1580         if (!RPNquery)
1581         {
1582             const char *pqf_msg;
1583             size_t off;
1584             int code = yaz_pqf_error(pqf_parser, &pqf_msg, &off);
1585             int ioff = off;
1586             printf("%*s^\n", ioff+4, "");
1587             printf("Prefix query error: %s (code %d)\n", pqf_msg, code);
1588
1589             yaz_pqf_destroy(pqf_parser);
1590             return 0;
1591         }
1592         yaz_pqf_destroy(pqf_parser);
1593         query_charset_convert(RPNquery);
1594         query.u.type_1 = RPNquery;
1595         break;
1596     case QueryType_CCL:
1597         query.which = Z_Query_type_2;
1598         query.u.type_2 = &ccl_query;
1599         ccl_query.buf = (unsigned char*) arg;
1600         ccl_query.len = strlen(arg);
1601         break;
1602     case QueryType_CCL2RPN:
1603         query.which = Z_Query_type_1;
1604         RPNquery = ccl_rpn_query(out, rpn);
1605         if (!RPNquery)
1606         {
1607             printf("Couldn't convert from CCL to RPN\n");
1608             return 0;
1609         }
1610         query_charset_convert(RPNquery);
1611         query.u.type_1 = RPNquery;
1612         ccl_rpn_delete(rpn);
1613         break;
1614     case QueryType_CQL:
1615         query.which = Z_Query_type_104;
1616         ext = (Z_External *) odr_malloc(out, sizeof(*ext));
1617         ext->direct_reference = odr_getoidbystr(out, "1.2.840.10003.16.2");
1618         ext->indirect_reference = 0;
1619         ext->descriptor = 0;
1620         ext->which = Z_External_CQL;
1621         ext->u.cql = odr_strdup(out, arg);
1622         query.u.type_104 =  ext;
1623         break;
1624     default:
1625         printf("Unsupported query type\n");
1626         return 0;
1627     }
1628     if (send_apdu(apdu))
1629         printf("Sent searchRequest.\n");
1630     setno = 1;
1631     return 2;
1632 }
1633
1634 static void display_term(Z_Term *term)
1635 {
1636     switch (term->which)
1637     {
1638     case Z_Term_general:
1639         printf("    %.*s", term->u.general->len, term->u.general->buf);
1640         break;
1641     case Z_Term_characterString:
1642         printf("    %s", term->u.characterString);
1643         break;
1644     case Z_Term_numeric:
1645         printf("    " ODR_INT_PRINTF, *term->u.numeric);
1646         break;
1647     case Z_Term_null:
1648         printf("    null");
1649         break;
1650     }
1651 }
1652
1653 /* display Query Expression as part of searchResult-1 */
1654 static void display_queryExpression(const char *lead, Z_QueryExpression *qe)
1655 {
1656     if (!qe)
1657         return;
1658     printf(" %s=", lead);
1659     if (qe->which == Z_QueryExpression_term)
1660     {
1661         if (qe->u.term->queryTerm)
1662         {
1663             Z_Term *term = qe->u.term->queryTerm;
1664             display_term(term);
1665         }
1666     }
1667 }
1668
1669 static void display_facet(Z_FacetField *facet)
1670 {
1671     if (facet->attributes)
1672     {
1673         Z_AttributeList *al = facet->attributes;
1674         struct yaz_facet_attr attr_values;
1675         attr_values.errcode = 0;
1676         attr_values.limit = -1;
1677         attr_values.useattr = 0;
1678         attr_values.relation = "default";
1679
1680         yaz_facet_attr_get_z_attributes(al, &attr_values);
1681         if (!attr_values.errcode)
1682         {
1683             int term_index;
1684             printf("  %s (%d): \n", attr_values.useattr, /* attr_values.relation, attr_values.limit, */ facet->num_terms);
1685             for (term_index = 0 ; term_index < facet->num_terms; term_index++)
1686             {
1687                 Z_FacetTerm *facetTerm = facet->terms[term_index];
1688                 display_term(facetTerm->term);
1689                 printf(" (" NMEM_INT_PRINTF ")\n", *facetTerm->count);
1690             }
1691         }
1692
1693     }
1694 }
1695
1696 static void* display_facets(Z_FacetList *fl)
1697 {
1698     int index;
1699     printf("Facets(%d): \n", fl->num);
1700
1701     for (index = 0; index < fl->num ; index++)
1702     {
1703         display_facet(fl->elements[index]);
1704     }
1705     return 0;
1706 }
1707
1708 void display_searchResult1(Z_SearchInfoReport *sr)
1709 {
1710     int j;
1711     printf("SearchResult-1:");
1712     for (j = 0; j < sr->num; j++)
1713     {
1714         if (j)
1715             printf(",");
1716         if (!sr->elements[j]->subqueryExpression)
1717             printf("%d", j);
1718         display_queryExpression("term",
1719             sr->elements[j]->subqueryExpression);
1720         display_queryExpression("interpretation",
1721             sr->elements[j]->subqueryInterpretation);
1722         display_queryExpression("recommendation",
1723             sr->elements[j]->subqueryRecommendation);
1724         if (sr->elements[j]->subqueryCount)
1725             printf(" cnt=" ODR_INT_PRINTF,
1726                    *sr->elements[j]->subqueryCount);
1727         if (sr->elements[j]->subqueryId)
1728             printf(" id=%s ", sr->elements[j]->subqueryId);
1729     }
1730     printf("\n");
1731 }
1732
1733
1734
1735 /* see if we can find USR:SearchResult-1 */
1736 static void display_searchResult(Z_OtherInformation *o)
1737 {
1738     int i;
1739     if (!o)
1740         return ;
1741     for (i = 0; i < o->num_elements; i++)
1742     {
1743         if (o->list[i]->which == Z_OtherInfo_externallyDefinedInfo)
1744         {
1745             Z_External *ext = o->list[i]->information.externallyDefinedInfo;
1746
1747             if (ext->which == Z_External_searchResult1)
1748                 display_searchResult1(ext->u.searchResult1);
1749             else if  (ext->which == Z_External_userFacets)
1750                 display_facets(ext->u.facetList);
1751         }
1752     }
1753 }
1754
1755 static int process_Z3950_searchResponse(Z_SearchResponse *res)
1756 {
1757     printf("Received SearchResponse.\n");
1758     print_refid(res->referenceId);
1759     if (*res->searchStatus)
1760         printf("Search was a success.\n");
1761     else
1762         printf("Search was a bloomin' failure.\n");
1763     printf("Number of hits: " ODR_INT_PRINTF, *res->resultCount);
1764     last_hit_count = *res->resultCount;
1765     if (setnumber >= 0)
1766         printf(", setno %d", setnumber);
1767     putchar('\n');
1768     if (res->resultSetStatus)
1769     {
1770         printf("Result Set Status: ");
1771         switch (*res->resultSetStatus)
1772         {
1773         case Z_SearchResponse_subset:
1774             printf("subset"); break;
1775         case Z_SearchResponse_interim:
1776             printf("interim"); break;
1777         case Z_SearchResponse_none:
1778             printf("none"); break;
1779         case Z_SearchResponse_estimate:
1780             printf("estimate"); break;
1781         default:
1782             printf(ODR_INT_PRINTF, *res->resultSetStatus);
1783         }
1784         putchar('\n');
1785     }
1786     display_searchResult(res->additionalSearchInfo);
1787     printf("records returned: " ODR_INT_PRINTF "\n",
1788            *res->numberOfRecordsReturned);
1789     setno += *res->numberOfRecordsReturned;
1790     if (res->records)
1791         display_records(res->records);
1792     return 0;
1793 }
1794
1795 static void print_level(int iLevel)
1796 {
1797     int i;
1798     for (i = 0; i < iLevel * 4; i++)
1799         printf(" ");
1800 }
1801
1802 static void print_int(int iLevel, const char *pTag, Odr_int *pInt)
1803 {
1804     if (pInt != NULL)
1805     {
1806         print_level(iLevel);
1807         printf("%s: " ODR_INT_PRINTF "\n", pTag, *pInt);
1808     }
1809 }
1810
1811 static void print_bool(int iLevel, const char *pTag, Odr_bool *pInt)
1812 {
1813     if (pInt != NULL)
1814     {
1815         print_level(iLevel);
1816         printf("%s: %d\n", pTag, *pInt);
1817     }
1818 }
1819
1820 static void print_string(int iLevel, const char *pTag, const char *pString)
1821 {
1822     if (pString != NULL)
1823     {
1824         print_level(iLevel);
1825         printf("%s: %s\n", pTag, pString);
1826     }
1827 }
1828
1829 static void print_oid(int iLevel, const char *pTag, Odr_oid *pOid)
1830 {
1831     if (pOid != NULL)
1832     {
1833         Odr_oid *pInt = pOid;
1834
1835         print_level(iLevel);
1836         printf("%s:", pTag);
1837         for (; *pInt != -1; pInt++)
1838             printf(" %d", *pInt);
1839         printf("\n");
1840     }
1841 }
1842
1843 static void print_referenceId(int iLevel, Z_ReferenceId *referenceId)
1844 {
1845     if (referenceId != NULL)
1846     {
1847         int i;
1848
1849         print_level(iLevel);
1850         printf("Ref Id (%d, %d): ", referenceId->len, referenceId->size);
1851         for (i = 0; i < referenceId->len; i++)
1852             printf("%c", referenceId->buf[i]);
1853         printf("\n");
1854     }
1855 }
1856
1857 static void print_string_or_numeric(int iLevel, const char *pTag, Z_StringOrNumeric *pStringNumeric)
1858 {
1859     if (pStringNumeric != NULL)
1860     {
1861         switch (pStringNumeric->which)
1862         {
1863         case Z_StringOrNumeric_string:
1864             print_string(iLevel, pTag, pStringNumeric->u.string);
1865             break;
1866
1867         case Z_StringOrNumeric_numeric:
1868             print_int(iLevel, pTag, pStringNumeric->u.numeric);
1869             break;
1870
1871         default:
1872             print_level(iLevel);
1873             printf("%s: valid type for Z_StringOrNumeric\n", pTag);
1874             break;
1875         }
1876     }
1877 }
1878
1879 static void print_universe_report_duplicate(
1880     int iLevel,
1881     Z_UniverseReportDuplicate *pUniverseReportDuplicate)
1882 {
1883     if (pUniverseReportDuplicate != NULL)
1884     {
1885         print_level(iLevel);
1886         printf("Universe Report Duplicate: \n");
1887         iLevel++;
1888         print_string_or_numeric(iLevel, "Hit No",
1889                                 pUniverseReportDuplicate->hitno);
1890     }
1891 }
1892
1893 static void print_universe_report_hits(
1894     int iLevel,
1895     Z_UniverseReportHits *pUniverseReportHits)
1896 {
1897     if (pUniverseReportHits != NULL)
1898     {
1899         print_level(iLevel);
1900         printf("Universe Report Hits: \n");
1901         iLevel++;
1902         print_string_or_numeric(iLevel, "Database",
1903                                 pUniverseReportHits->database);
1904         print_string_or_numeric(iLevel, "Hits", pUniverseReportHits->hits);
1905     }
1906 }
1907
1908 static void print_universe_report(int iLevel, Z_UniverseReport *pUniverseReport)
1909 {
1910     if (pUniverseReport != NULL)
1911     {
1912         print_level(iLevel);
1913         printf("Universe Report: \n");
1914         iLevel++;
1915         print_int(iLevel, "Total Hits", pUniverseReport->totalHits);
1916         switch (pUniverseReport->which)
1917         {
1918         case Z_UniverseReport_databaseHits:
1919             print_universe_report_hits(iLevel,
1920                                        pUniverseReport->u.databaseHits);
1921             break;
1922
1923         case Z_UniverseReport_duplicate:
1924             print_universe_report_duplicate(iLevel,
1925                                             pUniverseReport->u.duplicate);
1926             break;
1927
1928         default:
1929             print_level(iLevel);
1930             printf("Type: %d\n", pUniverseReport->which);
1931             break;
1932         }
1933     }
1934 }
1935
1936 static void print_external(int iLevel, Z_External *pExternal)
1937 {
1938     if (pExternal != NULL)
1939     {
1940         print_level(iLevel);
1941         printf("External: \n");
1942         iLevel++;
1943         print_oid(iLevel, "Direct Reference", pExternal->direct_reference);
1944         print_int(iLevel, "InDirect Reference", pExternal->indirect_reference);
1945         print_string(iLevel, "Descriptor", pExternal->descriptor);
1946         switch (pExternal->which)
1947         {
1948         case Z_External_universeReport:
1949             print_universe_report(iLevel, pExternal->u.universeReport);
1950             break;
1951
1952         default:
1953             print_level(iLevel);
1954             printf("Type: %d\n", pExternal->which);
1955             break;
1956         }
1957     }
1958 }
1959
1960 static int process_Z3950_resourceControlRequest(Z_ResourceControlRequest *req)
1961 {
1962     printf("Received ResourceControlRequest.\n");
1963     print_referenceId(1, req->referenceId);
1964     print_bool(1, "Suspended Flag", req->suspendedFlag);
1965     print_int(1, "Partial Results Available", req->partialResultsAvailable);
1966     print_bool(1, "Response Required", req->responseRequired);
1967     print_bool(1, "Triggered Request Flag", req->triggeredRequestFlag);
1968     print_external(1, req->resourceReport);
1969     return 0;
1970 }
1971
1972 static void process_Z3950_ESResponse(Z_ExtendedServicesResponse *res)
1973 {
1974     printf("Status: ");
1975     switch (*res->operationStatus)
1976     {
1977     case Z_ExtendedServicesResponse_done:
1978         printf("done\n");
1979         break;
1980     case Z_ExtendedServicesResponse_accepted:
1981         printf("accepted\n");
1982         break;
1983     case Z_ExtendedServicesResponse_failure:
1984         printf("failure\n");
1985         display_diagrecs(res->diagnostics, res->num_diagnostics);
1986         break;
1987     default:
1988         printf("unknown\n");
1989     }
1990     if ( (*res->operationStatus != Z_ExtendedServicesResponse_failure) &&
1991         (res->num_diagnostics != 0) )
1992     {
1993         display_diagrecs(res->diagnostics, res->num_diagnostics);
1994     }
1995     print_refid (res->referenceId);
1996     if (res->taskPackage &&
1997         res->taskPackage->which == Z_External_extendedService)
1998     {
1999         Z_TaskPackage *taskPackage = res->taskPackage->u.extendedService;
2000         Odr_oct *id = taskPackage->targetReference;
2001         Z_External *ext = taskPackage->taskSpecificParameters;
2002
2003         if (id)
2004         {
2005             printf("Target Reference: ");
2006             print_stringn((const char *) id->buf, id->len);
2007             printf("\n");
2008         }
2009         if (ext->which == Z_External_update)
2010         {
2011             Z_IUUpdateTaskPackage *utp = ext->u.update->u.taskPackage;
2012             if (utp && utp->targetPart)
2013             {
2014                 Z_IUTargetPart *targetPart = utp->targetPart;
2015                 int i;
2016
2017                 for (i = 0; i<targetPart->num_taskPackageRecords;  i++)
2018                 {
2019
2020                     Z_IUTaskPackageRecordStructure *tpr =
2021                         targetPart->taskPackageRecords[i];
2022                     printf("task package record %d\n", i+1);
2023                     if (tpr->which == Z_IUTaskPackageRecordStructure_record)
2024                     {
2025                         display_record (tpr->u.record);
2026                     }
2027                     else
2028                     {
2029                         printf("other type\n");
2030                     }
2031                 }
2032             }
2033         }
2034         if (ext->which == Z_External_itemOrder)
2035         {
2036             Z_IOTaskPackage *otp = ext->u.itemOrder->u.taskPackage;
2037
2038             if (otp && otp->targetPart)
2039             {
2040                 if (otp->targetPart->itemRequest)
2041                 {
2042                     Z_External *ext = otp->targetPart->itemRequest;
2043                     if (ext->which == Z_External_octet)
2044                     {
2045                         Odr_oct *doc = ext->u.octet_aligned;
2046                         printf("Got itemRequest doc %.*s\n",
2047                                doc->len, doc->buf);
2048                     }
2049                 }
2050                 else if (otp->targetPart->statusOrErrorReport)
2051                 {
2052                     Z_External *ext = otp->targetPart->statusOrErrorReport;
2053                     if (ext->which == Z_External_octet)
2054                     {
2055                         Odr_oct *doc = ext->u.octet_aligned;
2056                         printf("Got Status or Error Report doc %.*s\n",
2057                                doc->len, doc->buf);
2058                     }
2059                 }
2060             }
2061         }
2062     }
2063     if (res->taskPackage && res->taskPackage->which == Z_External_octet)
2064     {
2065         Odr_oct *doc = res->taskPackage->u.octet_aligned;
2066         printf("%.*s\n", doc->len, doc->buf);
2067     }
2068 }
2069
2070 static const char *get_ill_element(void *clientData, const char *element)
2071 {
2072     return 0;
2073 }
2074
2075 static Z_External *create_external_itemRequest(void)
2076 {
2077     struct ill_get_ctl ctl;
2078     ILL_ItemRequest *req;
2079     Z_External *r = 0;
2080     int item_request_size = 0;
2081     char *item_request_buf = 0;
2082
2083     ctl.odr = out;
2084     ctl.clientData = 0;
2085     ctl.f = get_ill_element;
2086
2087     req = ill_get_ItemRequest(&ctl, "ill", 0);
2088     if (!req)
2089         printf("ill_get_ItemRequest failed\n");
2090
2091     if (!ill_ItemRequest(out, &req, 0, 0))
2092     {
2093         if (apdu_file)
2094         {
2095             ill_ItemRequest(print, &req, 0, 0);
2096             odr_reset(print);
2097         }
2098         item_request_buf = odr_getbuf (out, &item_request_size, 0);
2099         if (item_request_buf)
2100             odr_setbuf (out, item_request_buf, item_request_size, 1);
2101         printf("Couldn't encode ItemRequest, size %d\n", item_request_size);
2102         return 0;
2103     }
2104     else
2105     {
2106         r = (Z_External *) odr_malloc(out, sizeof(*r));
2107         r->direct_reference = odr_oiddup(out, yaz_oid_general_isoill_1);
2108         r->indirect_reference = 0;
2109         r->descriptor = 0;
2110         r->which = Z_External_single;
2111
2112         r->u.single_ASN1_type = (Odr_oct *)
2113             odr_malloc(out, sizeof(*r->u.single_ASN1_type));
2114         r->u.single_ASN1_type->buf = (unsigned char *)
2115         odr_malloc(out, item_request_size);
2116         r->u.single_ASN1_type->len = item_request_size;
2117         r->u.single_ASN1_type->size = item_request_size;
2118         memcpy(r->u.single_ASN1_type->buf, item_request_buf,
2119                 item_request_size);
2120
2121         do_hex_dump(item_request_buf,item_request_size);
2122     }
2123     return r;
2124 }
2125
2126 static Z_External *create_external_ILL_APDU(void)
2127 {
2128     struct ill_get_ctl ctl;
2129     ILL_APDU *ill_apdu;
2130     Z_External *r = 0;
2131     int ill_request_size = 0;
2132     char *ill_request_buf = 0;
2133
2134     ctl.odr = out;
2135     ctl.clientData = 0;
2136     ctl.f = get_ill_element;
2137
2138     ill_apdu = ill_get_APDU(&ctl, "ill", 0);
2139
2140     if (!ill_APDU (out, &ill_apdu, 0, 0))
2141     {
2142         if (apdu_file)
2143         {
2144             printf("-------------------\n");
2145             ill_APDU(print, &ill_apdu, 0, 0);
2146             odr_reset(print);
2147             printf("-------------------\n");
2148         }
2149         ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
2150         if (ill_request_buf)
2151             odr_setbuf (out, ill_request_buf, ill_request_size, 1);
2152         printf("Couldn't encode ILL-Request, size %d\n", ill_request_size);
2153         return 0;
2154     }
2155     else
2156     {
2157         ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
2158
2159         r = (Z_External *) odr_malloc(out, sizeof(*r));
2160         r->direct_reference = odr_oiddup(out, yaz_oid_general_isoill_1);
2161         r->indirect_reference = 0;
2162         r->descriptor = 0;
2163         r->which = Z_External_single;
2164
2165         r->u.single_ASN1_type = (Odr_oct *)
2166             odr_malloc(out, sizeof(*r->u.single_ASN1_type));
2167         r->u.single_ASN1_type->buf = (unsigned char *)
2168         odr_malloc(out, ill_request_size);
2169         r->u.single_ASN1_type->len = ill_request_size;
2170         r->u.single_ASN1_type->size = ill_request_size;
2171         memcpy(r->u.single_ASN1_type->buf, ill_request_buf, ill_request_size);
2172 /*         printf("len = %d\n", ill_request_size); */
2173 /*              do_hex_dump(ill_request_buf,ill_request_size); */
2174 /*              printf("--- end of extenal\n"); */
2175
2176     }
2177     return r;
2178 }
2179
2180
2181 static Z_External *create_ItemOrderExternal(const char *type, int itemno,
2182                                             const char *xml_buf,
2183                                             int xml_len)
2184 {
2185     Z_External *r = (Z_External *) odr_malloc(out, sizeof(Z_External));
2186     r->direct_reference = odr_oiddup(out, yaz_oid_extserv_item_order);
2187     r->indirect_reference = 0;
2188     r->descriptor = 0;
2189
2190     r->which = Z_External_itemOrder;
2191
2192     r->u.itemOrder = (Z_ItemOrder *) odr_malloc(out,sizeof(Z_ItemOrder));
2193     memset(r->u.itemOrder, 0, sizeof(Z_ItemOrder));
2194     r->u.itemOrder->which=Z_IOItemOrder_esRequest;
2195
2196     r->u.itemOrder->u.esRequest = (Z_IORequest *)
2197         odr_malloc(out,sizeof(Z_IORequest));
2198     memset(r->u.itemOrder->u.esRequest, 0, sizeof(Z_IORequest));
2199
2200     r->u.itemOrder->u.esRequest->toKeep = (Z_IOOriginPartToKeep *)
2201         odr_malloc(out,sizeof(Z_IOOriginPartToKeep));
2202     memset(r->u.itemOrder->u.esRequest->toKeep, 0, sizeof(Z_IOOriginPartToKeep));
2203     r->u.itemOrder->u.esRequest->notToKeep = (Z_IOOriginPartNotToKeep *)
2204         odr_malloc(out,sizeof(Z_IOOriginPartNotToKeep));
2205     memset(r->u.itemOrder->u.esRequest->notToKeep, 0, sizeof(Z_IOOriginPartNotToKeep));
2206
2207     r->u.itemOrder->u.esRequest->toKeep->supplDescription = NULL;
2208     r->u.itemOrder->u.esRequest->toKeep->contact = NULL;
2209     r->u.itemOrder->u.esRequest->toKeep->addlBilling = NULL;
2210
2211     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem =
2212         (Z_IOResultSetItem *) odr_malloc(out, sizeof(Z_IOResultSetItem));
2213     memset(r->u.itemOrder->u.esRequest->notToKeep->resultSetItem, 0, sizeof(Z_IOResultSetItem));
2214     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->resultSetId = "1";
2215
2216     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->item =
2217         odr_intdup(out, itemno);
2218     if (!strcmp (type, "item") || !strcmp(type, "2"))
2219     {
2220         printf("using item-request\n");
2221         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2222             create_external_itemRequest();
2223     }
2224     else if (!strcmp(type, "ill") || !strcmp(type, "1"))
2225     {
2226         printf("using ILL-request\n");
2227         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2228             create_external_ILL_APDU();
2229     }
2230     else if (!strcmp(type, "xml") || !strcmp(type, "3"))
2231     {
2232         printf("using XML ILL-request\n");
2233
2234         if (!xml_buf)
2235         {
2236             printf("no docoument added\n");
2237             r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
2238         }
2239         else
2240         {
2241             r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2242                 z_ext_record_oid(out, yaz_oid_recsyn_xml, xml_buf, xml_len);
2243         }
2244     }
2245     else
2246         r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
2247
2248     return r;
2249 }
2250
2251 static int send_Z3950_itemorder(const char *type, int itemno,
2252                                 const char *xml_buf, int xml_len)
2253 {
2254     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest);
2255     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2256
2257     req->referenceId = set_refid (out);
2258
2259     req->packageType = odr_oiddup(out, yaz_oid_extserv_item_order);
2260     req->packageName = esPackageName;
2261
2262     req->taskSpecificParameters = create_ItemOrderExternal(type, itemno,
2263                                                            xml_buf, xml_len);
2264     send_apdu(apdu);
2265     return 0;
2266 }
2267
2268 static int only_z3950(void)
2269 {
2270     if (!conn)
2271     {
2272         printf("Not connected yet\n");
2273         return 1;
2274     }
2275     if (protocol == PROTO_HTTP)
2276     {
2277         printf("Not supported by SRW\n");
2278         return 1;
2279     }
2280     return 0;
2281 }
2282
2283 static int is_SRW(void)
2284 {
2285     if (!conn)
2286     {
2287         printf("Not connected yet\n");
2288         return 1;
2289     }
2290     if (protocol == PROTO_HTTP && yaz_matchstr(sru_method, "solr"))
2291     {
2292         printf("Not supported by SRW\n");
2293         return 1;
2294     }
2295     return 0;
2296 }
2297
2298
2299 static int cmd_update_common(const char *arg, int version);
2300
2301 static int cmd_update(const char *arg)
2302 {
2303     return cmd_update_common(arg, 1);
2304 }
2305
2306 static int cmd_update0(const char *arg)
2307 {
2308     return cmd_update_common(arg, 0);
2309 }
2310
2311 static int send_Z3950_update(int version, int action_no, const char *recid,
2312                             char *rec_buf, int rec_len);
2313
2314 #if YAZ_HAVE_XML2
2315 static int send_SRW_update(int action_no, const char *recid,
2316                           char *rec_buf, int rec_len);
2317 #endif
2318
2319 static int cmd_update_common(const char *arg, int version)
2320 {
2321     char *action_buf;
2322     int action_len;
2323     char *recid_buf;
2324     int recid_len;
2325     const char *recid = 0;
2326     char *rec_buf;
2327     int rec_len;
2328     int action_no;
2329     int noread = 0;
2330
2331     if (parse_cmd_doc(&arg, out, &action_buf, &action_len) == 0)
2332     {
2333         printf("Use: update action recid [fname]\n");
2334         printf(" where action is one of insert,replace,delete.update\n");
2335         printf(" recid is some record ID. Use none for no ID\n");
2336         printf(" fname is file of record to be updated\n");
2337         return 0;
2338     }
2339
2340     if (parse_cmd_doc(&arg, out, &recid_buf, &recid_len) == 0)
2341     {
2342         printf("Missing recid\n");
2343         return 0;
2344     }
2345
2346     if (!strcmp(action_buf, "insert"))
2347         action_no = Z_IUOriginPartToKeep_recordInsert;
2348     else if (!strcmp(action_buf, "replace"))
2349         action_no = Z_IUOriginPartToKeep_recordReplace;
2350     else if (!strcmp(action_buf, "delete"))
2351         action_no = Z_IUOriginPartToKeep_recordDelete;
2352     else if (!strcmp(action_buf, "update"))
2353         action_no = Z_IUOriginPartToKeep_specialUpdate;
2354     else
2355     {
2356         printf("Bad action: %s\n", action_buf);
2357         printf("Possible values: insert, replace, delete, update\n");
2358         return 0;
2359     }
2360
2361     if (strcmp(recid_buf, "none")) /* none means no record ID */
2362         recid = recid_buf;
2363
2364     arg += noread;
2365     if (parse_cmd_doc(&arg, out, &rec_buf, &rec_len) == 0)
2366         return 0;
2367
2368 #if YAZ_HAVE_XML2
2369     if (protocol == PROTO_HTTP)
2370         return send_SRW_update(action_no, recid_buf, rec_buf, rec_len);
2371 #endif
2372     return send_Z3950_update(version, action_no, recid_buf, rec_buf, rec_len);
2373 }
2374
2375 #if YAZ_HAVE_XML2
2376 static int send_SRW_update(int action_no, const char *recid,
2377                            char *rec_buf, int rec_len)
2378 {
2379     if (!conn)
2380         session_connect(cur_host);
2381     if (!conn)
2382         return 0;
2383     else
2384     {
2385         Z_SRW_PDU *srw = yaz_srw_get(out, Z_SRW_update_request);
2386         Z_SRW_updateRequest *sr = srw->u.update_request;
2387
2388         switch (action_no)
2389         {
2390         case Z_IUOriginPartToKeep_recordInsert:
2391             sr->operation = "info:srw/action/1/create";
2392             break;
2393         case Z_IUOriginPartToKeep_recordReplace:
2394             sr->operation = "info:srw/action/1/replace";
2395             break;
2396         case Z_IUOriginPartToKeep_recordDelete:
2397             sr->operation = "info:srw/action/1/delete";
2398             break;
2399         }
2400         if (rec_buf)
2401         {
2402             sr->record = yaz_srw_get_record(out);
2403             sr->record->recordData_buf = rec_buf;
2404             sr->record->recordData_len = rec_len;
2405             sr->record->recordSchema = record_schema;
2406         }
2407         if (recid)
2408             sr->recordId = odr_strdup(out, recid);
2409         return send_srw(srw);
2410     }
2411 }
2412 #endif
2413
2414 static int send_Z3950_update(int version, int action_no, const char *recid,
2415                             char *rec_buf, int rec_len)
2416 {
2417     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest );
2418     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2419     Z_External *r;
2420     Z_External *record_this = 0;
2421     if (rec_buf)
2422         record_this = z_ext_record_oid(out, yaz_oid_recsyn_xml,
2423                                        rec_buf, rec_len);
2424     else
2425     {
2426         if (!record_last)
2427         {
2428             printf("No last record (update ignored)\n");
2429             return 0;
2430         }
2431         record_this = record_last;
2432     }
2433
2434     req->packageType = odr_oiddup(out, (version == 0 ?
2435        yaz_oid_extserv_database_update_first_version :
2436        yaz_oid_extserv_database_update));
2437
2438     req->packageName = esPackageName;
2439
2440     req->referenceId = set_refid (out);
2441
2442     r = req->taskSpecificParameters = (Z_External *)
2443         odr_malloc(out, sizeof(*r));
2444     r->direct_reference = req->packageType;
2445     r->indirect_reference = 0;
2446     r->descriptor = 0;
2447     if (version == 0)
2448     {
2449         Z_IU0OriginPartToKeep *toKeep;
2450         Z_IU0SuppliedRecords *notToKeep;
2451
2452         r->which = Z_External_update0;
2453         r->u.update0 = (Z_IU0Update *) odr_malloc(out, sizeof(*r->u.update0));
2454         r->u.update0->which = Z_IUUpdate_esRequest;
2455         r->u.update0->u.esRequest = (Z_IU0UpdateEsRequest *)
2456             odr_malloc(out, sizeof(*r->u.update0->u.esRequest));
2457         toKeep = r->u.update0->u.esRequest->toKeep = (Z_IU0OriginPartToKeep *)
2458             odr_malloc(out, sizeof(*r->u.update0->u.esRequest->toKeep));
2459
2460         toKeep->databaseName = databaseNames[0];
2461         toKeep->schema = 0;
2462         if (record_schema)
2463         {
2464             toKeep->schema = yaz_string_to_oid_odr(yaz_oid_std(),
2465                                                    CLASS_SCHEMA,
2466                                                    record_schema, out);
2467         }
2468         toKeep->elementSetName = 0;
2469
2470         toKeep->action = odr_intdup(out, action_no);
2471
2472         notToKeep = r->u.update0->u.esRequest->notToKeep = (Z_IU0SuppliedRecords *)
2473             odr_malloc(out, sizeof(*r->u.update0->u.esRequest->notToKeep));
2474         notToKeep->num = 1;
2475         notToKeep->elements = (Z_IU0SuppliedRecords_elem **)
2476             odr_malloc(out, sizeof(*notToKeep->elements));
2477         notToKeep->elements[0] = (Z_IU0SuppliedRecords_elem *)
2478             odr_malloc(out, sizeof(**notToKeep->elements));
2479         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
2480         if (recid)
2481         {
2482             notToKeep->elements[0]->u.opaque = (Odr_oct *)
2483                 odr_malloc(out, sizeof(Odr_oct));
2484             notToKeep->elements[0]->u.opaque->buf = (unsigned char *) recid;
2485             notToKeep->elements[0]->u.opaque->size = strlen(recid);
2486             notToKeep->elements[0]->u.opaque->len = strlen(recid);
2487         }
2488         else
2489             notToKeep->elements[0]->u.opaque = 0;
2490         notToKeep->elements[0]->supplementalId = 0;
2491         notToKeep->elements[0]->correlationInfo = 0;
2492         notToKeep->elements[0]->record = record_this;
2493     }
2494     else
2495     {
2496         Z_IUOriginPartToKeep *toKeep;
2497         Z_IUSuppliedRecords *notToKeep;
2498
2499         r->which = Z_External_update;
2500         r->u.update = (Z_IUUpdate *) odr_malloc(out, sizeof(*r->u.update));
2501         r->u.update->which = Z_IUUpdate_esRequest;
2502         r->u.update->u.esRequest = (Z_IUUpdateEsRequest *)
2503             odr_malloc(out, sizeof(*r->u.update->u.esRequest));
2504         toKeep = r->u.update->u.esRequest->toKeep = (Z_IUOriginPartToKeep *)
2505             odr_malloc(out, sizeof(*r->u.update->u.esRequest->toKeep));
2506
2507         toKeep->databaseName = databaseNames[0];
2508         toKeep->schema = 0;
2509         if (record_schema)
2510         {
2511             toKeep->schema = yaz_string_to_oid_odr(yaz_oid_std(),
2512                                                    CLASS_SCHEMA,
2513                                                    record_schema, out);
2514         }
2515         toKeep->elementSetName = 0;
2516         toKeep->actionQualifier = 0;
2517         toKeep->action = odr_intdup(out, action_no);
2518
2519         notToKeep = r->u.update->u.esRequest->notToKeep = (Z_IUSuppliedRecords *)
2520             odr_malloc(out, sizeof(*r->u.update->u.esRequest->notToKeep));
2521         notToKeep->num = 1;
2522         notToKeep->elements = (Z_IUSuppliedRecords_elem **)
2523             odr_malloc(out, sizeof(*notToKeep->elements));
2524         notToKeep->elements[0] = (Z_IUSuppliedRecords_elem *)
2525             odr_malloc(out, sizeof(**notToKeep->elements));
2526         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
2527         if (recid)
2528         {
2529             notToKeep->elements[0]->u.opaque = (Odr_oct *)
2530                 odr_malloc(out, sizeof(Odr_oct));
2531             notToKeep->elements[0]->u.opaque->buf = (unsigned char *) recid;
2532             notToKeep->elements[0]->u.opaque->size = strlen(recid);
2533             notToKeep->elements[0]->u.opaque->len = strlen(recid);
2534         }
2535         else
2536             notToKeep->elements[0]->u.opaque = 0;
2537         notToKeep->elements[0]->supplementalId = 0;
2538         notToKeep->elements[0]->correlationInfo = 0;
2539         notToKeep->elements[0]->record = record_this;
2540     }
2541
2542     send_apdu(apdu);
2543
2544     return 2;
2545 }
2546
2547 static int cmd_xmles(const char *arg)
2548 {
2549     if (only_z3950())
2550         return 1;
2551     else
2552     {
2553         char *asn_buf = 0;
2554         int noread = 0;
2555         Odr_oid *oid;
2556         char oid_str[51];
2557         Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest);
2558         Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2559
2560
2561         Z_External *ext = (Z_External *) odr_malloc(out, sizeof(*ext));
2562
2563         req->referenceId = set_refid (out);
2564         req->taskSpecificParameters = ext;
2565         ext->indirect_reference = 0;
2566         ext->descriptor = 0;
2567         ext->which = Z_External_octet;
2568         ext->u.single_ASN1_type = (Odr_oct *) odr_malloc(out, sizeof(Odr_oct));
2569         sscanf(arg, "%50s%n", oid_str, &noread);
2570         if (noread == 0)
2571         {
2572             printf("Missing OID for xmles\n");
2573             return 0;
2574         }
2575         arg += noread;
2576         if (parse_cmd_doc(&arg, out, &asn_buf,
2577                           &ext->u.single_ASN1_type->len) == 0)
2578             return 0;
2579
2580         ext->u.single_ASN1_type->buf = (unsigned char *) asn_buf;
2581
2582         oid = yaz_string_to_oid_odr(yaz_oid_std(),
2583                                     CLASS_EXTSERV, oid_str, out);
2584         if (!oid)
2585         {
2586             printf("Bad OID: %s\n", oid_str);
2587             return 0;
2588         }
2589
2590         req->packageType = oid;
2591
2592         ext->direct_reference = oid;
2593
2594         send_apdu(apdu);
2595
2596         return 2;
2597     }
2598 }
2599
2600 static int cmd_itemorder(const char *arg)
2601 {
2602     char type[12];
2603     int itemno;
2604     char *xml_buf = 0;
2605     int xml_len = 0;
2606     int no_read = 0;
2607
2608     if (only_z3950())
2609         return 1;
2610     if (sscanf(arg, "%10s %d%n", type, &itemno, &no_read) < 2)
2611         return 0;
2612     arg += no_read;
2613     parse_cmd_doc(&arg, out, &xml_buf, &xml_len);
2614
2615     fflush(stdout);
2616     send_Z3950_itemorder(type, itemno, xml_buf, xml_len);
2617     return 2;
2618 }
2619
2620 static void show_opt(const char *arg, void *clientData)
2621 {
2622     printf("%s ", arg);
2623 }
2624
2625 static int cmd_zversion(const char *arg)
2626 {
2627     if (*arg && arg)
2628         z3950_version = atoi(arg);
2629     else
2630         printf("version is %d\n", z3950_version);
2631     return 0;
2632 }
2633
2634 static int cmd_options(const char *arg)
2635 {
2636     if (*arg)
2637     {
2638         int r;
2639         int pos;
2640         r = yaz_init_opt_encode(&z3950_options, arg, &pos);
2641         if (r == -1)
2642             printf("Unknown option(s) near %s\n", arg+pos);
2643     }
2644     else
2645     {
2646         yaz_init_opt_decode(&z3950_options, show_opt, 0);
2647         printf("\n");
2648     }
2649     return 0;
2650 }
2651
2652 static int cmd_explain(const char *arg)
2653 {
2654     if (protocol != PROTO_HTTP)
2655         return 0;
2656 #if YAZ_HAVE_XML2
2657     if (!conn)
2658         session_connect(cur_host);
2659     if (conn)
2660     {
2661         Z_SRW_PDU *sr = 0;
2662
2663         setno = 1;
2664
2665         /* save this for later .. when fetching individual records */
2666         sr = yaz_srw_get(out, Z_SRW_explain_request);
2667         if (recordsyntax_size == 1
2668             && !yaz_matchstr(recordsyntax_list[0], "xml"))
2669             sr->u.explain_request->recordPacking = "xml";
2670         send_srw(sr);
2671         return 2;
2672     }
2673 #endif
2674     return 0;
2675 }
2676
2677 static int cmd_init(const char *arg)
2678 {
2679     if (*arg)
2680     {
2681         strncpy(cur_host, arg, sizeof(cur_host)-1);
2682         cur_host[sizeof(cur_host)-1] = 0;
2683     }
2684     if (only_z3950())
2685         return 1;
2686     send_Z3950_initRequest(cur_host);
2687     return 2;
2688 }
2689
2690 static Z_GDU *get_HTTP_Request_url(ODR odr, const char *url)
2691 {
2692     Z_GDU *p = z_get_HTTP_Request(odr);
2693     const char *host = url;
2694     const char *cp0 = strstr(host, "://");
2695     const char *cp1 = 0;
2696     if (cp0)
2697         cp0 = cp0+3;
2698     else
2699         cp0 = host;
2700
2701     cp1 = strchr(cp0, '/');
2702     if (!cp1)
2703         cp1 = cp0 + strlen(cp0);
2704
2705     if (cp0 && cp1)
2706     {
2707         char *h = (char*) odr_malloc(odr, cp1 - cp0 + 1);
2708         memcpy (h, cp0, cp1 - cp0);
2709         h[cp1-cp0] = '\0';
2710         z_HTTP_header_add(odr, &p->u.HTTP_Request->headers, "Host", h);
2711     }
2712     p->u.HTTP_Request->path = odr_strdup(odr, *cp1 ? cp1 : "/");
2713     return p;
2714 }
2715
2716 static WRBUF get_url(const char *uri, WRBUF username, WRBUF password,
2717                      int *code, int show_headers)
2718 {
2719     WRBUF result = 0;
2720     ODR out = odr_createmem(ODR_ENCODE);
2721     ODR in = odr_createmem(ODR_DECODE);
2722     Z_GDU *gdu = get_HTTP_Request_url(out, uri);
2723
2724     gdu->u.HTTP_Request->method = odr_strdup(out, "GET");
2725     if (username && password)
2726     {
2727         z_HTTP_header_add_basic_auth(out, &gdu->u.HTTP_Request->headers,
2728                                      wrbuf_cstr(username),
2729                                      wrbuf_cstr(password));
2730     }
2731     z_HTTP_header_add(out, &gdu->u.HTTP_Request->headers, "Accept",
2732                       "text/xml");
2733     if (!z_GDU(out, &gdu, 0, 0))
2734     {
2735         yaz_log(YLOG_WARN, "Can not encode HTTP request URL:%s", uri);
2736     }
2737     else
2738     {
2739         void *add;
2740         COMSTACK conn = cs_create_host(uri, 1, &add);
2741         if (cs_connect(conn, add) < 0)
2742             yaz_log(YLOG_WARN, "Can not connect to URL:%s", uri);
2743         else
2744         {
2745             int len;
2746             char *buf = odr_getbuf(out, &len, 0);
2747
2748             if (cs_put(conn, buf, len) < 0)
2749                 yaz_log(YLOG_WARN, "cs_put failed URL:%s", uri);
2750             else
2751             {
2752                 char *netbuffer = 0;
2753                 int netlen = 0;
2754                 int res = cs_get(conn, &netbuffer, &netlen);
2755                 if (res <= 0)
2756                 {
2757                     yaz_log(YLOG_WARN, "cs_get failed URL:%s", uri);
2758                 }
2759                 else
2760                 {
2761                     Z_GDU *gdu;
2762                     odr_setbuf(in, netbuffer, res, 0);
2763                     if (!z_GDU(in, &gdu, 0, 0)
2764                         || gdu->which != Z_GDU_HTTP_Response)
2765                     {
2766                         yaz_log(YLOG_WARN, "decode failed URL: %s", uri);
2767                     }
2768                     else
2769                     {
2770                         Z_HTTP_Response *res = gdu->u.HTTP_Response;
2771                         struct Z_HTTP_Header *h;
2772                         result = wrbuf_alloc();
2773                         if (show_headers)
2774                         {
2775
2776                             wrbuf_printf(result, "HTTP %d\n", res->code);
2777                             for (h = res->headers; h; h = h->next)
2778                                 wrbuf_printf(result, "%s: %s\n",
2779                                              h->name, h->value);
2780                         }
2781                         *code = res->code;
2782                         wrbuf_write(result, res->content_buf, res->content_len);
2783                     }
2784                 }
2785                 xfree(netbuffer);
2786             }
2787             cs_close(conn);
2788         }
2789     }
2790     odr_destroy(out);
2791     odr_destroy(in);
2792     return result;
2793 }
2794
2795
2796 static int cmd_url(const char *arg)
2797 {
2798     int code = 0;
2799     WRBUF res = get_url(arg, 0, 0, &code, 1);
2800     if (res)
2801     {
2802         if (wrbuf_len(res) > 1200)
2803         {
2804             fwrite(wrbuf_buf(res), 1, 1200, stdout);
2805             printf(".. out of %lld\n", (long long) wrbuf_len(res));
2806         }
2807         else
2808             puts(wrbuf_cstr(res));
2809         wrbuf_destroy(res);
2810     }
2811     return 0;
2812 }
2813
2814 static int cmd_sru(const char *arg)
2815 {
2816     if (!*arg)
2817     {
2818         printf("SRU method is: %s\n", sru_method);
2819         printf("SRU version is: %s\n", sru_version);
2820     }
2821     else
2822     {
2823         int r = sscanf(arg, "%9s %9s", sru_method, sru_version);
2824         if (r >= 1)
2825         {
2826             if (!yaz_matchstr(sru_method, "post"))
2827                 ;
2828             else if (!yaz_matchstr(sru_method, "get"))
2829                 ;
2830             else if (!yaz_matchstr(sru_method, "soap"))
2831                 ;
2832             else if (!yaz_matchstr(sru_method, "solr"))
2833                 ;
2834             else
2835             {
2836                 strcpy(sru_method, "soap");
2837                 printf("Unknown SRU method: %s\n", arg);
2838                 printf("Specify one of POST, GET, SOAP, SOLR\n");
2839             }
2840         }
2841     }
2842     return 0;
2843 }
2844
2845 static int cmd_find(const char *arg)
2846 {
2847     if (!*arg)
2848     {
2849         printf("Find what?\n");
2850         return 0;
2851     }
2852     if (protocol == PROTO_HTTP)
2853     {
2854 #if YAZ_HAVE_XML2
2855         if (!conn)
2856             session_connect(cur_host);
2857         if (!conn)
2858             return 0;
2859         if (!send_SRW_searchRequest(arg))
2860             return 0;
2861 #else
2862         return 0;
2863 #endif
2864     }
2865     else
2866     {
2867         if (*cur_host && auto_reconnect)
2868         {
2869             int i = 0;
2870             for (;;)
2871             {
2872                 if (conn)
2873                 {
2874                     if (!send_Z3950_searchRequest(arg))
2875                         return 0;
2876                     wait_and_handle_response(0);
2877                     if (conn)
2878                         break;
2879                 }
2880                 if (++i == 2)
2881                 {
2882                     printf("Unable to reconnect\n");
2883                     break;
2884                 }
2885                 session_connect(cur_host);
2886                 wait_and_handle_response(0);
2887             }
2888             return 0;
2889         }
2890         else if (conn)
2891         {
2892             if (!send_Z3950_searchRequest(arg))
2893                 return 0;
2894         }
2895         else
2896         {
2897             printf("Not connected yet\n");
2898             return 0;
2899         }
2900     }
2901     return 2;
2902 }
2903
2904 static int cmd_facets(const char *arg)
2905 {
2906     /* TODO Wrong odr. Loosing memory */
2907     ODR odr = odr_createmem(ODR_ENCODE);
2908     int size = 0;
2909     if (!*arg)
2910     {
2911         facet_list = 0;
2912         printf("Facets cleared.\n");
2913         return 0;
2914     }
2915     size = strlen(arg);
2916     if (is_SRW()) {
2917         printf("WARN: No supported for SRW/SRU.\n");
2918     }
2919     facet_list = yaz_pqf_parse_facet_list(odr, arg);
2920
2921     if (!facet_list)
2922     {
2923         printf("Invalid facet list: %s", arg);
2924         return 0;
2925     }
2926     return 1;
2927 }
2928
2929
2930 static int cmd_delete(const char *arg)
2931 {
2932     if (only_z3950())
2933         return 0;
2934     if (!send_Z3950_deleteResultSetRequest(arg))
2935         return 0;
2936     return 2;
2937 }
2938
2939 static int cmd_ssub(const char *arg)
2940 {
2941     if (!(smallSetUpperBound = atoi(arg)))
2942         return 0;
2943     return 1;
2944 }
2945
2946 static int cmd_lslb(const char *arg)
2947 {
2948     if (!(largeSetLowerBound = atoi(arg)))
2949         return 0;
2950     return 1;
2951 }
2952
2953 static int cmd_mspn(const char *arg)
2954 {
2955     if (!(mediumSetPresentNumber = atoi(arg)))
2956         return 0;
2957     return 1;
2958 }
2959
2960 static int cmd_status(const char *arg)
2961 {
2962     printf("smallSetUpperBound: %d\n", smallSetUpperBound);
2963     printf("largeSetLowerBound: %d\n", largeSetLowerBound);
2964     printf("mediumSetPresentNumber: %d\n", mediumSetPresentNumber);
2965     return 1;
2966 }
2967
2968 static int cmd_setnames(const char *arg)
2969 {
2970     if (*arg == '1')         /* enable ? */
2971         setnumber = 0;
2972     else if (*arg == '0')    /* disable ? */
2973         setnumber = -1;
2974     else if (setnumber < 0)  /* no args, toggle .. */
2975         setnumber = 0;
2976     else
2977         setnumber = -1;
2978
2979     if (setnumber >= 0)
2980         printf("Set numbering enabled.\n");
2981     else
2982         printf("Set numbering disabled.\n");
2983     return 1;
2984 }
2985
2986 /* PRESENT SERVICE ----------------------------- */
2987
2988 static int parse_show_args(const char *arg_c, char *setstring,
2989                            Odr_int *start, Odr_int *number)
2990 {
2991     char *end_ptr;
2992     Odr_int start_position;
2993
2994     if (setnumber >= 0)
2995         sprintf(setstring, "%d", setnumber);
2996     else
2997         *setstring = '\0';
2998
2999     if (!strcmp(arg_c, "all"))
3000     {
3001         *number = last_hit_count;
3002         *start = 1;
3003     }
3004     start_position = odr_strtol(arg_c, &end_ptr, 10);
3005     if (end_ptr == arg_c)
3006         return 1;
3007     *start = start_position;
3008     if (*end_ptr == '\0')
3009         return 1;
3010     while (isspace(*(unsigned char *)end_ptr))
3011         end_ptr++;
3012     if (*end_ptr != '+')
3013     {
3014         printf("Bad show arg: expected +. Got %s\n", end_ptr);
3015         return 0;
3016     }
3017     end_ptr++;
3018     arg_c = end_ptr;
3019     *number = odr_strtol(arg_c, &end_ptr, 10);
3020     if (end_ptr == arg_c)
3021     {
3022         printf("Bad show arg: expected number after +\n");
3023         return 0;
3024     }
3025     if (*end_ptr == '\0')
3026         return 1;
3027     while (isspace(*(unsigned char *)end_ptr))
3028         end_ptr++;
3029     if (*end_ptr != '+')
3030     {
3031         printf("Bad show arg: + expected. Got %s\n", end_ptr);
3032         return 0;
3033     }
3034     strcpy(setstring, end_ptr+1);
3035     return 1;
3036 }
3037
3038 static int send_Z3950_presentRequest(const char *arg)
3039 {
3040     Z_APDU *apdu = zget_APDU(out, Z_APDU_presentRequest);
3041     Z_PresentRequest *req = apdu->u.presentRequest;
3042     Z_RecordComposition compo;
3043     Odr_int nos = 1;
3044     char setstring[100];
3045
3046     req->referenceId = set_refid(out);
3047
3048     if (!parse_show_args(arg, setstring, &setno, &nos))
3049         return 0;
3050     if (*setstring)
3051         req->resultSetId = setstring;
3052
3053     req->resultSetStartPoint = &setno;
3054     req->numberOfRecordsRequested = &nos;
3055
3056     if (recordsyntax_size)
3057         req->preferredRecordSyntax =
3058             yaz_string_to_oid_odr(yaz_oid_std(),
3059                                   CLASS_RECSYN, recordsyntax_list[0], out);
3060
3061     if (record_schema || recordsyntax_size >= 2)
3062     {
3063         req->recordComposition = &compo;
3064         compo.which = Z_RecordComp_complex;
3065         compo.u.complex = (Z_CompSpec *)
3066             odr_malloc(out, sizeof(*compo.u.complex));
3067         compo.u.complex->selectAlternativeSyntax = (bool_t *)
3068             odr_malloc(out, sizeof(bool_t));
3069         *compo.u.complex->selectAlternativeSyntax = 0;
3070
3071         compo.u.complex->generic = (Z_Specification *)
3072             odr_malloc(out, sizeof(*compo.u.complex->generic));
3073
3074         compo.u.complex->generic->which = Z_Schema_oid;
3075         if (!record_schema)
3076             compo.u.complex->generic->schema.oid = 0;
3077         else
3078         {
3079             compo.u.complex->generic->schema.oid =
3080                 yaz_string_to_oid_odr(yaz_oid_std(),
3081                                       CLASS_SCHEMA, record_schema, out);
3082
3083             if (!compo.u.complex->generic->schema.oid)
3084             {
3085                 /* OID wasn't a schema! Try record syntax instead. */
3086                 compo.u.complex->generic->schema.oid = (Odr_oid *)
3087                     yaz_string_to_oid_odr(yaz_oid_std(),
3088                                           CLASS_RECSYN, record_schema, out);
3089             }
3090         }
3091         if (!elementSetNames)
3092             compo.u.complex->generic->elementSpec = 0;
3093         else
3094         {
3095             compo.u.complex->generic->elementSpec = (Z_ElementSpec *)
3096                 odr_malloc(out, sizeof(Z_ElementSpec));
3097             compo.u.complex->generic->elementSpec->which =
3098                 Z_ElementSpec_elementSetName;
3099             compo.u.complex->generic->elementSpec->u.elementSetName =
3100                 elementSetNames->u.generic;
3101         }
3102         compo.u.complex->num_dbSpecific = 0;
3103         compo.u.complex->dbSpecific = 0;
3104
3105         compo.u.complex->num_recordSyntax = 0;
3106         compo.u.complex->recordSyntax = 0;
3107         if (recordsyntax_size >= 2)
3108         {
3109             int i;
3110             compo.u.complex->num_recordSyntax = recordsyntax_size;
3111             compo.u.complex->recordSyntax = (Odr_oid **)
3112                 odr_malloc(out, recordsyntax_size * sizeof(Odr_oid*));
3113             for (i = 0; i < recordsyntax_size; i++)
3114             compo.u.complex->recordSyntax[i] =
3115                 yaz_string_to_oid_odr(yaz_oid_std(),
3116                                       CLASS_RECSYN, recordsyntax_list[i], out);
3117         }
3118     }
3119     else if (elementSetNames)
3120     {
3121         req->recordComposition = &compo;
3122         compo.which = Z_RecordComp_simple;
3123         compo.u.simple = elementSetNames;
3124     }
3125     send_apdu(apdu);
3126     printf("Sent presentRequest (" ODR_INT_PRINTF "+" ODR_INT_PRINTF ").\n",
3127            setno, nos);
3128     return 2;
3129 }
3130
3131 #if YAZ_HAVE_XML2
3132 static int send_SRW_presentRequest(const char *arg)
3133 {
3134     char setstring[100];
3135     Odr_int nos = 1;
3136     Z_SRW_PDU *sr = srw_sr;
3137
3138     if (!sr)
3139         return 0;
3140     if (!parse_show_args(arg, setstring, &setno, &nos))
3141         return 0;
3142     sr->u.request->startRecord = odr_intdup(out, setno);
3143     sr->u.request->maximumRecords = odr_intdup(out, nos);
3144     if (record_schema)
3145         sr->u.request->recordSchema = record_schema;
3146     if (recordsyntax_size == 1 && !yaz_matchstr(recordsyntax_list[0], "xml"))
3147         sr->u.request->recordPacking = "xml";
3148     return send_srw(sr);
3149 }
3150 #endif
3151
3152 static void close_session(void)
3153 {
3154     if (conn)
3155         cs_close(conn);
3156     conn = 0;
3157     sent_close = 0;
3158     odr_reset(out);
3159     odr_reset(in);
3160     odr_reset(print);
3161     last_hit_count = 0;
3162 }
3163
3164 static void process_Z3950_close(Z_Close *req)
3165 {
3166     Z_APDU *apdu = zget_APDU(out, Z_APDU_close);
3167     Z_Close *res = apdu->u.close;
3168
3169     static char *reasons[] =
3170     {
3171         "finished",
3172         "shutdown",
3173         "system problem",
3174         "cost limit reached",
3175         "resources",
3176         "security violation",
3177         "protocolError",
3178         "lack of activity",
3179         "peer abort",
3180         "unspecified"
3181     };
3182
3183     printf("Reason: %s, message: %s\n", reasons[*req->closeReason],
3184         req->diagnosticInformation ? req->diagnosticInformation : "NULL");
3185     if (sent_close)
3186         close_session();
3187     else
3188     {
3189         *res->closeReason = Z_Close_finished;
3190         send_apdu(apdu);
3191         printf("Sent response.\n");
3192         sent_close = 1;
3193     }
3194 }
3195
3196 static int cmd_show(const char *arg)
3197 {
3198     if (protocol == PROTO_HTTP)
3199     {
3200 #if YAZ_HAVE_XML2
3201         if (!conn)
3202             session_connect(cur_host);
3203         if (!conn)
3204             return 0;
3205         if (!send_SRW_presentRequest(arg))
3206             return 0;
3207 #else
3208         return 0;
3209 #endif
3210     }
3211     else
3212     {
3213         if (!conn)
3214         {
3215             printf("Not connected yet\n");
3216             return 0;
3217         }
3218         if (!send_Z3950_presentRequest(arg))
3219             return 0;
3220     }
3221     return 2;
3222 }
3223
3224 static void exit_client(int code)
3225 {
3226     file_history_save(file_history);
3227     file_history_destroy(&file_history);
3228     nmem_destroy(nmem_auth);
3229     exit(code);
3230 }
3231
3232 static int cmd_quit(const char *arg)
3233 {
3234     printf("See you later, alligator.\n");
3235     xmalloc_trav("");
3236     exit_client(0);
3237     return 0;
3238 }
3239
3240 static int cmd_cancel(const char *arg)
3241 {
3242     if (only_z3950())
3243         return 0;
3244     else
3245     {
3246         Z_APDU *apdu = zget_APDU(out, Z_APDU_triggerResourceControlRequest);
3247         Z_TriggerResourceControlRequest *req =
3248             apdu->u.triggerResourceControlRequest;
3249         bool_t rfalse = 0;
3250         char command[16];
3251
3252         *command = '\0';
3253         sscanf(arg, "%15s", command);
3254
3255         if (only_z3950())
3256             return 0;
3257         if (session_initResponse &&
3258             !ODR_MASK_GET(session_initResponse->options,
3259                           Z_Options_triggerResourceCtrl))
3260         {
3261             printf("Target doesn't support cancel (trigger resource ctrl)\n");
3262             return 0;
3263         }
3264         *req->requestedAction = Z_TriggerResourceControlRequest_cancel;
3265         req->resultSetWanted = &rfalse;
3266         req->referenceId = set_refid(out);
3267
3268         send_apdu(apdu);
3269         printf("Sent cancel request\n");
3270         if (!strcmp(command, "wait"))
3271             return 2;
3272         return 1;
3273     }
3274 }
3275
3276 static int cmd_cancel_find(const char *arg)
3277 {
3278     int fres;
3279     if (only_z3950())
3280         return 0;
3281     fres = cmd_find(arg);
3282     if (fres > 0)
3283     {
3284         return cmd_cancel("");
3285     };
3286     return fres;
3287 }
3288
3289 static int send_Z3950_scanrequest(const char *set,  const char *query,
3290                                   Odr_int pp, Odr_int num, const char *term)
3291 {
3292     Z_APDU *apdu = zget_APDU(out, Z_APDU_scanRequest);
3293     Z_ScanRequest *req = apdu->u.scanRequest;
3294
3295     if (only_z3950())
3296         return 0;
3297     if (queryType == QueryType_CCL2RPN)
3298     {
3299         int error, pos;
3300         struct ccl_rpn_node *rpn;
3301
3302         rpn = ccl_find_str(bibset,  query, &error, &pos);
3303         if (error)
3304         {
3305             printf("CCL ERROR: %s\n", ccl_err_msg(error));
3306             return -1;
3307         }
3308         req->attributeSet =
3309             yaz_string_to_oid_odr(yaz_oid_std(),
3310                                   CLASS_ATTSET, "Bib-1", out);
3311         if (!(req->termListAndStartPoint = ccl_scan_query(out, rpn)))
3312         {
3313             printf("Couldn't convert CCL to Scan term\n");
3314             return -1;
3315         }
3316         ccl_rpn_delete(rpn);
3317     }
3318     else
3319     {
3320         YAZ_PQF_Parser pqf_parser = yaz_pqf_create();
3321
3322
3323         if (!(req->termListAndStartPoint =
3324               yaz_pqf_scan(pqf_parser, out, &req->attributeSet, query)))
3325         {
3326             const char *pqf_msg;
3327             size_t off;
3328             int code = yaz_pqf_error(pqf_parser, &pqf_msg, &off);
3329             int ioff = off;
3330             printf("%*s^\n", ioff+7, "");
3331             printf("Prefix query error: %s (code %d)\n", pqf_msg, code);
3332             yaz_pqf_destroy(pqf_parser);
3333             return -1;
3334         }
3335         yaz_pqf_destroy(pqf_parser);
3336     }
3337     if (queryCharset && outputCharset)
3338     {
3339         yaz_iconv_t cd = yaz_iconv_open(queryCharset, outputCharset);
3340         if (!cd)
3341         {
3342             printf("Conversion from %s to %s unsupported\n",
3343                    outputCharset, queryCharset);
3344             return -1;
3345         }
3346         yaz_query_charset_convert_apt(req->termListAndStartPoint, out, cd);
3347         yaz_iconv_close(cd);
3348     }
3349     if (term && *term)
3350     {
3351         if (req->termListAndStartPoint->term &&
3352             req->termListAndStartPoint->term->which == Z_Term_general &&
3353             req->termListAndStartPoint->term->u.general)
3354         {
3355             req->termListAndStartPoint->term->u.general->buf =
3356                 (unsigned char *) odr_strdup(out, term);
3357             req->termListAndStartPoint->term->u.general->len =
3358                 req->termListAndStartPoint->term->u.general->size =
3359                 strlen(term);
3360         }
3361     }
3362     req->referenceId = set_refid(out);
3363     req->num_databaseNames = num_databaseNames;
3364     req->databaseNames = databaseNames;
3365     req->numberOfTermsRequested = &num;
3366     req->preferredPositionInResponse = &pp;
3367     req->stepSize = odr_intdup(out, scan_stepSize);
3368
3369     if (set)
3370         yaz_oi_set_string_oid(&req->otherInfo, out,
3371                               yaz_oid_userinfo_scan_set, 1, set);
3372
3373     send_apdu(apdu);
3374     return 2;
3375 }
3376
3377 static int send_sortrequest(const char *arg, int newset)
3378 {
3379     Z_APDU *apdu = zget_APDU(out, Z_APDU_sortRequest);
3380     Z_SortRequest *req = apdu->u.sortRequest;
3381     Z_SortKeySpecList *sksl = (Z_SortKeySpecList *)
3382         odr_malloc(out, sizeof(*sksl));
3383     char setstring[32];
3384
3385     if (only_z3950())
3386         return 0;
3387     if (setnumber >= 0)
3388         sprintf(setstring, "%d", setnumber);
3389     else
3390         sprintf(setstring, "default");
3391
3392     req->referenceId = set_refid(out);
3393
3394     req->num_inputResultSetNames = 1;
3395     req->inputResultSetNames = (Z_InternationalString **)
3396         odr_malloc(out, sizeof(*req->inputResultSetNames));
3397     req->inputResultSetNames[0] = odr_strdup(out, setstring);
3398
3399     if (newset && setnumber >= 0)
3400         sprintf(setstring, "%d", ++setnumber);
3401
3402     req->sortedResultSetName = odr_strdup(out, setstring);
3403
3404     req->sortSequence = yaz_sort_spec(out, arg);
3405     if (!req->sortSequence)
3406     {
3407         printf("Missing sort specifications\n");
3408         return -1;
3409     }
3410     send_apdu(apdu);
3411     return 2;
3412 }
3413
3414 static void display_term_info(Z_TermInfo *t)
3415 {
3416     if (t->displayTerm)
3417         printf("%s", t->displayTerm);
3418     else if (t->term->which == Z_Term_general)
3419         printf("%.*s", t->term->u.general->len, t->term->u.general->buf);
3420     else
3421         printf("Term (not general)");
3422     if (t->term->which == Z_Term_general)
3423         sprintf(last_scan_line, "%.*s", t->term->u.general->len,
3424             t->term->u.general->buf);
3425
3426     if (t->globalOccurrences)
3427         printf(" (" ODR_INT_PRINTF ")\n", *t->globalOccurrences);
3428     else
3429         printf("\n");
3430 }
3431
3432 static void process_Z3950_scanResponse(Z_ScanResponse *res)
3433 {
3434     int i;
3435     Z_Entry **entries = NULL;
3436     int num_entries = 0;
3437
3438     printf("Received ScanResponse\n");
3439     print_refid(res->referenceId);
3440     printf(ODR_INT_PRINTF " entries", *res->numberOfEntriesReturned);
3441     if (res->positionOfTerm)
3442         printf(", position=" ODR_INT_PRINTF, *res->positionOfTerm);
3443     printf("\n");
3444     if (*res->scanStatus != Z_Scan_success)
3445         printf("Scan returned code " ODR_INT_PRINTF "\n", *res->scanStatus);
3446     if (!res->entries)
3447         return;
3448     if ((entries = res->entries->entries))
3449         num_entries = res->entries->num_entries;
3450     for (i = 0; i < num_entries; i++)
3451     {
3452         int pos_term = res->positionOfTerm ? *res->positionOfTerm : -1;
3453         if (entries[i]->which == Z_Entry_termInfo)
3454         {
3455             printf("%c ", i + 1 == pos_term ? '*' : ' ');
3456             display_term_info(entries[i]->u.termInfo);
3457         }
3458         else
3459             display_diagrecs(&entries[i]->u.surrogateDiagnostic, 1);
3460     }
3461     if (res->entries->nonsurrogateDiagnostics)
3462         display_diagrecs(res->entries->nonsurrogateDiagnostics,
3463                           res->entries->num_nonsurrogateDiagnostics);
3464 }
3465
3466 static void process_Z3950_sortResponse(Z_SortResponse *res)
3467 {
3468     printf("Received SortResponse: status=");
3469     switch (*res->sortStatus)
3470     {
3471     case Z_SortResponse_success:
3472         printf("success"); break;
3473     case Z_SortResponse_partial_1:
3474         printf("partial"); break;
3475     case Z_SortResponse_failure:
3476         printf("failure"); break;
3477     default:
3478         printf("unknown (" ODR_INT_PRINTF ")", *res->sortStatus);
3479     }
3480     printf("\n");
3481     print_refid (res->referenceId);
3482     if (res->diagnostics)
3483         display_diagrecs(res->diagnostics,
3484                          res->num_diagnostics);
3485 }
3486
3487 static void process_Z3950_deleteResultSetResponse(
3488     Z_DeleteResultSetResponse *res)
3489 {
3490     printf("Got deleteResultSetResponse status=" ODR_INT_PRINTF "\n",
3491            *res->deleteOperationStatus);
3492     if (res->deleteListStatuses)
3493     {
3494         int i;
3495         for (i = 0; i < res->deleteListStatuses->num; i++)
3496         {
3497             printf("%s status=" ODR_INT_PRINTF "\n",
3498                    res->deleteListStatuses->elements[i]->id,
3499                    *res->deleteListStatuses->elements[i]->status);
3500         }
3501     }
3502 }
3503
3504 static int cmd_sort_generic(const char *arg, int newset)
3505 {
3506     if (only_z3950())
3507         return 0;
3508     if (session_initResponse &&
3509         !ODR_MASK_GET(session_initResponse->options, Z_Options_sort))
3510     {
3511         printf("Target doesn't support sort\n");
3512         return 0;
3513     }
3514     if (*arg)
3515     {
3516         if (send_sortrequest(arg, newset) < 0)
3517             return 0;
3518         return 2;
3519     }
3520     return 0;
3521 }
3522
3523 static int cmd_sort(const char *arg)
3524 {
3525     return cmd_sort_generic(arg, 0);
3526 }
3527
3528 static int cmd_sort_newset(const char *arg)
3529 {
3530     return cmd_sort_generic(arg, 1);
3531 }
3532
3533 static int cmd_scanstep(const char *arg)
3534 {
3535     scan_stepSize = atoi(arg);
3536     return 0;
3537 }
3538
3539 static int cmd_scanpos(const char *arg)
3540 {
3541     int r = sscanf(arg, "%d", &scan_position);
3542     if (r == 0)
3543         scan_position = 1;
3544     return 0;
3545 }
3546
3547 static int cmd_scansize(const char *arg)
3548 {
3549     int r = sscanf(arg, "%d", &scan_size);
3550     if (r == 0)
3551         scan_size = 20;
3552     return 0;
3553 }
3554
3555 static int cmd_scan_common(const char *set, const char *arg)
3556 {
3557     if (protocol == PROTO_HTTP)
3558     {
3559 #if YAZ_HAVE_XML2
3560         if (!conn)
3561             session_connect(cur_host);
3562         if (!conn)
3563             return 0;
3564         if (*arg)
3565         {
3566             if (send_SRW_scanRequest(arg, scan_position, scan_size) < 0)
3567                 return 0;
3568         }
3569         else
3570         {
3571             if (send_SRW_scanRequest(last_scan_line, 1, scan_size) < 0)
3572                 return 0;
3573         }
3574         return 2;
3575 #else
3576         return 0;
3577 #endif
3578     }
3579     else
3580     {
3581         if (*cur_host && !conn && auto_reconnect)
3582         {
3583             session_connect(cur_host);
3584             wait_and_handle_response(0);
3585         }
3586         if (!conn)
3587             return 0;
3588         if (session_initResponse &&
3589             !ODR_MASK_GET(session_initResponse->options, Z_Options_scan))
3590         {
3591             printf("Target doesn't support scan\n");
3592             return 0;
3593         }
3594         if (*arg)
3595         {
3596             strcpy(last_scan_query, arg);
3597             if (send_Z3950_scanrequest(set, arg,
3598                                        scan_position, scan_size, 0) < 0)
3599                 return 0;
3600         }
3601         else
3602         {
3603             if (send_Z3950_scanrequest(set, last_scan_query,
3604                                       1, scan_size, last_scan_line) < 0)
3605                 return 0;
3606         }
3607         return 2;
3608     }
3609 }
3610
3611 static int cmd_scan(const char *arg)
3612 {
3613     return cmd_scan_common(0, arg);
3614 }
3615
3616 static int cmd_setscan(const char *arg)
3617 {
3618     char setstring[100];
3619     int nor;
3620     if (sscanf(arg, "%99s%n", setstring, &nor) < 1)
3621     {
3622         printf("missing set for setscan\n");
3623         return 0;
3624     }
3625     return cmd_scan_common(setstring, arg + nor);
3626 }
3627
3628 static int cmd_schema(const char *arg)
3629 {
3630     xfree(record_schema);
3631     record_schema = 0;
3632     if (arg && *arg)
3633         record_schema = xstrdup(arg);
3634     return 1;
3635 }
3636
3637 static int cmd_format(const char *arg)
3638 {
3639     const char *cp = arg;
3640     int nor;
3641     int idx = 0;
3642     int i;
3643     char form_str[41];
3644     if (!arg || !*arg)
3645     {
3646         printf("Usage: format <recordsyntax>\n");
3647         return 0;
3648     }
3649     while (sscanf(cp, "%40s%n", form_str, &nor) >= 1 && nor > 0
3650            && idx < RECORDSYNTAX_MAX)
3651     {
3652         if (strcmp(form_str, "none") &&
3653             !yaz_string_to_oid_odr(yaz_oid_std(), CLASS_RECSYN, form_str, out))
3654         {
3655             printf("Bad format: %s\n", form_str);
3656             return 0;
3657         }
3658         cp += nor;
3659     }
3660     for (i = 0; i < recordsyntax_size; i++)
3661     {
3662         xfree(recordsyntax_list[i]);
3663         recordsyntax_list[i] = 0;
3664     }
3665
3666     cp = arg;
3667     while (sscanf(cp, "%40s%n", form_str, &nor) >= 1 && nor > 0
3668            && idx < RECORDSYNTAX_MAX)
3669     {
3670         if (!strcmp(form_str, "none"))
3671             break;
3672         recordsyntax_list[idx] = xstrdup(form_str);
3673         cp += nor;
3674         idx++;
3675     }
3676     recordsyntax_size = idx;
3677     return 1;
3678 }
3679
3680 static int cmd_elements(const char *arg)
3681 {
3682     static Z_ElementSetNames esn;
3683     static char what[100];
3684
3685     if (!arg || !*arg)
3686     {
3687         elementSetNames = 0;
3688         return 1;
3689     }
3690     strcpy(what, arg);
3691     esn.which = Z_ElementSetNames_generic;
3692     esn.u.generic = what;
3693     elementSetNames = &esn;
3694     return 1;
3695 }
3696
3697 static int cmd_querytype(const char *arg)
3698 {
3699     if (!strcmp(arg, "ccl"))
3700         queryType = QueryType_CCL;
3701     else if (!strcmp(arg, "prefix") || !strcmp(arg, "rpn"))
3702         queryType = QueryType_Prefix;
3703     else if (!strcmp(arg, "ccl2rpn") || !strcmp(arg, "cclrpn"))
3704         queryType = QueryType_CCL2RPN;
3705     else if (!strcmp(arg, "cql"))
3706         queryType = QueryType_CQL;
3707     else if (!strcmp(arg, "cql2rpn") || !strcmp(arg, "cqlrpn"))
3708         queryType = QueryType_CQL2RPN;
3709     else
3710     {
3711         printf("Querytype must be one of:\n");
3712         printf(" prefix         - Prefix query\n");
3713         printf(" ccl            - CCL query\n");
3714         printf(" ccl2rpn        - CCL query converted to RPN\n");
3715         printf(" cql            - CQL\n");
3716         printf(" cql2rpn        - CQL query converted to RPN\n");
3717         return 0;
3718     }
3719     return 1;
3720 }
3721
3722 static int cmd_refid(const char *arg)
3723 {
3724     xfree(refid);
3725     refid = NULL;
3726     if (*arg)
3727         refid = xstrdup(arg);
3728     return 1;
3729 }
3730
3731 static int cmd_close(const char *arg)
3732 {
3733     Z_APDU *apdu;
3734     Z_Close *req;
3735     if (only_z3950())
3736         return 0;
3737     apdu = zget_APDU(out, Z_APDU_close);
3738     req = apdu->u.close;
3739     *req->closeReason = Z_Close_finished;
3740     send_apdu(apdu);
3741     printf("Sent close request.\n");
3742     sent_close = 1;
3743     return 2;
3744 }
3745
3746 int cmd_packagename(const char* arg)
3747 {
3748     xfree(esPackageName);
3749     esPackageName = NULL;
3750     if (*arg)
3751         esPackageName = xstrdup(arg);
3752     return 1;
3753 }
3754
3755 static int cmd_proxy(const char* arg)
3756 {
3757     xfree(yazProxy);
3758     yazProxy = 0;
3759     if (*arg)
3760         yazProxy = xstrdup(arg);
3761     return 1;
3762 }
3763
3764 static int cmd_marccharset(const char *arg)
3765 {
3766     char l1[30];
3767
3768     *l1 = 0;
3769     if (sscanf(arg, "%29s", l1) < 1)
3770     {
3771         printf("MARC character set is `%s'\n",
3772                marcCharset ? marcCharset: "none");
3773         return 1;
3774     }
3775     xfree(marcCharset);
3776     marcCharset = 0;
3777     if (strcmp(l1, "-") && strcmp(l1, "none"))
3778         marcCharset = xstrdup(l1);
3779     return 1;
3780 }
3781
3782 static int cmd_querycharset(const char *arg)
3783 {
3784     char l1[30];
3785
3786     *l1 = 0;
3787     if (sscanf(arg, "%29s", l1) < 1)
3788     {
3789         printf("Query character set is `%s'\n",
3790                queryCharset ? queryCharset: "none");
3791         return 1;
3792     }
3793     xfree(queryCharset);
3794     queryCharset = 0;
3795     if (strcmp(l1, "-") && strcmp(l1, "none"))
3796         queryCharset = xstrdup(l1);
3797     return 1;
3798 }
3799
3800 static int cmd_displaycharset(const char *arg)
3801 {
3802     char l1[30];
3803
3804     *l1 = 0;
3805     if (sscanf(arg, "%29s", l1) < 1)
3806     {
3807         printf("Display character set is `%s'\n",
3808                outputCharset ? outputCharset: "none");
3809     }
3810     else
3811     {
3812         xfree(outputCharset);
3813         outputCharset = 0;
3814         if (!strcmp(l1, "auto") && codeset)
3815         {
3816             if (codeset)
3817             {
3818                 printf("Display character set: %s\n", codeset);
3819                 outputCharset = xstrdup(codeset);
3820             }
3821             else
3822                 printf("No codeset found on this system\n");
3823         }
3824         else if (strcmp(l1, "-") && strcmp(l1, "none"))
3825             outputCharset = xstrdup(l1);
3826     }
3827     return 1;
3828 }
3829
3830 static int cmd_negcharset(const char *arg)
3831 {
3832     char l1[30];
3833
3834     *l1 = 0;
3835     if (sscanf(arg, "%29s %d %d", l1, &negotiationCharsetRecords,
3836                &negotiationCharsetVersion) < 1)
3837     {
3838         printf("Negotiation character set `%s'\n",
3839                negotiationCharset ? negotiationCharset: "none");
3840         if (negotiationCharset)
3841         {
3842             printf("Records in charset %s\n", negotiationCharsetRecords ?
3843                    "yes" : "no");
3844             printf("Charneg version %d\n", negotiationCharsetVersion);
3845         }
3846     }
3847     else
3848     {
3849         xfree(negotiationCharset);
3850         negotiationCharset = NULL;
3851         if (*l1 && strcmp(l1, "-") && strcmp(l1, "none"))
3852         {
3853             negotiationCharset = xstrdup(l1);
3854             printf("Character set negotiation : %s\n", negotiationCharset);
3855         }
3856     }
3857     return 1;
3858 }
3859
3860 static int cmd_charset(const char* arg)
3861 {
3862     char l1[30], l2[30], l3[30], l4[30];
3863
3864     *l1 = *l2 = *l3 = *l4 = '\0';
3865     if (sscanf(arg, "%29s %29s %29s %29s", l1, l2, l3, l4) < 1)
3866     {
3867         cmd_negcharset("");
3868         cmd_displaycharset("");
3869         cmd_marccharset("");
3870         cmd_querycharset("");
3871     }
3872     else
3873     {
3874         cmd_negcharset(l1);
3875         if (*l2)
3876             cmd_displaycharset(l2);
3877         if (*l3)
3878             cmd_marccharset(l3);
3879         if (*l4)
3880             cmd_querycharset(l4);
3881     }
3882     return 1;
3883 }
3884
3885 static int cmd_lang(const char* arg)
3886 {
3887     if (*arg == '\0')
3888     {
3889         printf("Current language is `%s'\n", yazLang ? yazLang : "none");
3890         return 1;
3891     }
3892     xfree(yazLang);
3893     yazLang = NULL;
3894     if (*arg)
3895         yazLang = xstrdup(arg);
3896     return 1;
3897 }
3898
3899 static int cmd_source(const char* arg, int echo )
3900 {
3901     /* first should open the file and read one line at a time.. */
3902     FILE* includeFile;
3903     char line[102400], *cp;
3904
3905     if (strlen(arg) < 1)
3906     {
3907         fprintf(stderr, "Error in source command use a filename\n");
3908         return -1;
3909     }
3910
3911     includeFile = fopen(arg, "r");
3912
3913     if (!includeFile)
3914     {
3915         fprintf(stderr, "Unable to open file %s for reading\n",arg);
3916         return -1;
3917     }
3918
3919     while (fgets(line, sizeof(line), includeFile))
3920     {
3921         if (strlen(line) < 2)
3922             continue;
3923         if (line[0] == '#')
3924             continue;
3925
3926         if ((cp = strrchr(line, '\n')))
3927             *cp = '\0';
3928
3929         if (echo)
3930             printf("processing line: %s\n", line);
3931         process_cmd_line(line);
3932     }
3933
3934     if (fclose(includeFile))
3935     {
3936         perror("unable to close include file");
3937         exit(1);
3938     }
3939     return 1;
3940 }
3941
3942 static int cmd_source_echo(const char* arg)
3943 {
3944     cmd_source(arg, 1);
3945     return 1;
3946 }
3947
3948 static int cmd_subshell(const char* args)
3949 {
3950     int ret = system(strlen(args) ? args : getenv("SHELL"));
3951     printf("\n");
3952     if (ret)
3953     {
3954         printf("Exit %d\n", ret);
3955     }
3956     return 1;
3957 }
3958
3959 static int cmd_set_berfile(const char *arg)
3960 {
3961     if (ber_file && ber_file != stdout && ber_file != stderr)
3962         fclose(ber_file);
3963     if (!strcmp(arg, ""))
3964         ber_file = 0;
3965     else if (!strcmp(arg, "-"))
3966         ber_file = stdout;
3967     else
3968         ber_file = fopen(arg, "a");
3969     return 1;
3970 }
3971
3972 static int cmd_set_apdufile(const char *arg)
3973 {
3974     if (apdu_file && apdu_file != stderr && apdu_file != stderr)
3975         fclose(apdu_file);
3976     if (!strcmp(arg, ""))
3977         apdu_file = 0;
3978     else if (!strcmp(arg, "-"))
3979         apdu_file = stderr;
3980     else
3981     {
3982         apdu_file = fopen(arg, "a");
3983         if (!apdu_file)
3984             perror("unable to open apdu log file");
3985     }
3986     if (apdu_file)
3987         odr_setprint(print, apdu_file);
3988     return 1;
3989 }
3990
3991 static int cmd_set_cclfile(const char* arg)
3992 {
3993     FILE *inf;
3994
3995     bibset = ccl_qual_mk();
3996     inf = fopen(arg, "r");
3997     if (!inf)
3998         perror("unable to open CCL file");
3999     else
4000     {
4001         ccl_qual_file(bibset, inf);
4002         fclose(inf);
4003     }
4004     strcpy(ccl_fields,arg);
4005     return 0;
4006 }
4007
4008 static int cmd_set_cqlfile(const char* arg)
4009 {
4010     cql_transform_t newcqltrans;
4011
4012     if ((newcqltrans = cql_transform_open_fname(arg)) == 0)
4013     {
4014         perror("unable to open CQL file");
4015         return 0;
4016     }
4017     if (cqltrans != 0)
4018         cql_transform_close(cqltrans);
4019
4020     cqltrans = newcqltrans;
4021     strcpy(cql_fields, arg);
4022     return 0;
4023 }
4024
4025 static int cmd_set_auto_reconnect(const char* arg)
4026 {
4027     if (strlen(arg)==0)
4028         auto_reconnect = ! auto_reconnect;
4029     else if (strcmp(arg,"on")==0)
4030         auto_reconnect = 1;
4031     else if (strcmp(arg,"off")==0)
4032         auto_reconnect = 0;
4033     else
4034     {
4035         printf("Error use on or off\n");
4036         return 1;
4037     }
4038
4039     if (auto_reconnect)
4040         printf("Set auto reconnect enabled.\n");
4041     else
4042         printf("Set auto reconnect disabled.\n");
4043
4044     return 0;
4045 }
4046
4047 static int cmd_set_auto_wait(const char* arg)
4048 {
4049     if (strlen(arg)==0)
4050         auto_wait = ! auto_wait;
4051     else if (strcmp(arg,"on")==0)
4052         auto_wait = 1;
4053     else if (strcmp(arg,"off")==0)
4054         auto_wait = 0;
4055     else
4056     {
4057         printf("Error use on or off\n");
4058         return 1;
4059     }
4060
4061     if (auto_wait)
4062         printf("Set auto wait enabled.\n");
4063     else
4064         printf("Set auto wait disabled.\n");
4065
4066     return 0;
4067 }
4068
4069 static int cmd_set_marcdump(const char* arg)
4070 {
4071     if (marc_file && marc_file != stderr)
4072     { /* don't close stdout*/
4073         fclose(marc_file);
4074     }
4075
4076     if (!strcmp(arg, ""))
4077         marc_file = 0;
4078     else if (!strcmp(arg, "-"))
4079         marc_file = stderr;
4080     else
4081     {
4082         marc_file = fopen(arg, "a");
4083         if (!marc_file)
4084             perror("unable to open marc log file");
4085     }
4086     return 1;
4087 }
4088
4089 static void marc_file_write(const char *buf, size_t sz)
4090 {
4091     if (marc_file)
4092     {
4093         if (fwrite(buf, 1, sz, marc_file) != sz)
4094         {
4095             perror("marcfile write");
4096         }
4097     }
4098 }
4099 /*
4100    this command takes 3 arge {name class oid}
4101 */
4102 static int cmd_register_oid(const char* args)
4103 {
4104     static struct {
4105         char* className;
4106         oid_class oclass;
4107     } oid_classes[] = {
4108         {"appctx",CLASS_APPCTX},
4109         {"absyn",CLASS_ABSYN},
4110         {"attset",CLASS_ATTSET},
4111         {"transyn",CLASS_TRANSYN},
4112         {"diagset",CLASS_DIAGSET},
4113         {"recsyn",CLASS_RECSYN},
4114         {"resform",CLASS_RESFORM},
4115         {"accform",CLASS_ACCFORM},
4116         {"extserv",CLASS_EXTSERV},
4117         {"userinfo",CLASS_USERINFO},
4118         {"elemspec",CLASS_ELEMSPEC},
4119         {"varset",CLASS_VARSET},
4120         {"schema",CLASS_SCHEMA},
4121         {"tagset",CLASS_TAGSET},
4122         {"general",CLASS_GENERAL},
4123         {0,(enum oid_class) 0}
4124     };
4125     char oname_str[101], oclass_str[101], oid_str[101];
4126     int i;
4127     oid_class oidclass = CLASS_GENERAL;
4128     Odr_oid oid[OID_SIZE];
4129
4130     if (sscanf(args, "%100[^ ] %100[^ ] %100s",
4131                 oname_str,oclass_str, oid_str) < 1)
4132     {
4133         printf("Error in register command \n");
4134         return 0;
4135     }
4136
4137     for (i = 0; oid_classes[i].className; i++)
4138     {
4139         if (!strcmp(oid_classes[i].className, oclass_str))
4140         {
4141             oidclass=oid_classes[i].oclass;
4142             break;
4143         }
4144     }
4145
4146     if (!(oid_classes[i].className))
4147     {
4148         printf("Unknown oid class %s\n",oclass_str);
4149         return 0;
4150     }
4151
4152     oid_dotstring_to_oid(oid_str, oid);
4153
4154     if (yaz_oid_add(yaz_oid_std(), oidclass, oname_str, oid))
4155     {
4156         printf("oid %s already exists, registration failed\n",
4157                oname_str);
4158     }
4159     return 1;
4160 }
4161
4162 static int cmd_push_command(const char* arg)
4163 {
4164 #if HAVE_READLINE_HISTORY_H
4165     if (strlen(arg) > 1)
4166         add_history(arg);
4167 #else
4168     fprintf(stderr,"Not compiled with the readline/history module\n");
4169 #endif
4170     return 1;
4171 }
4172
4173 void source_rc_file(const char *rc_file)
4174 {
4175     /*  If rc_file != NULL, source that. Else
4176         Look for .yazclientrc and read it if it exists.
4177         If it does not exist, read  $HOME/.yazclientrc instead */
4178     struct stat statbuf;
4179
4180     if (rc_file)
4181     {
4182         if (stat(rc_file, &statbuf) == 0)
4183             cmd_source(rc_file, 0);
4184         else
4185         {
4186             fprintf(stderr, "yaz_client: cannot source '%s'\n", rc_file);
4187             exit(1);
4188         }
4189     }
4190     else
4191     {
4192         char fname[1000];
4193         strcpy(fname, ".yazclientrc");
4194         if (stat(fname, &statbuf)==0)
4195         {
4196             cmd_source(fname, 0);
4197         }
4198         else
4199         {
4200             const char* homedir = getenv("HOME");
4201             if (homedir)
4202             {
4203                 sprintf(fname, "%.800s/%s", homedir, ".yazclientrc");
4204                 if (stat(fname, &statbuf)==0)
4205                     cmd_source(fname, 0);
4206             }
4207         }
4208     }
4209 }
4210
4211 static void add_to_readline_history(void *client_data, const char *line)
4212 {
4213 #if HAVE_READLINE_HISTORY_H
4214     if (strlen(line))
4215         add_history(line);
4216 #endif
4217 }
4218
4219 static void initialize(const char *rc_file)
4220 {
4221     FILE *inf;
4222     int i;
4223
4224     if (!(out = odr_createmem(ODR_ENCODE)) ||
4225         !(in = odr_createmem(ODR_DECODE)) ||
4226         !(print = odr_createmem(ODR_PRINT)))
4227     {
4228         fprintf(stderr, "failed to allocate ODR streams\n");
4229         exit(1);
4230     }
4231
4232     setvbuf(stdout, 0, _IONBF, 0);
4233     if (apdu_file)
4234         odr_setprint(print, apdu_file);
4235
4236     bibset = ccl_qual_mk();
4237     inf = fopen(ccl_fields, "r");
4238     if (inf)
4239     {
4240         ccl_qual_file(bibset, inf);
4241         fclose(inf);
4242     }
4243
4244     cqltrans = cql_transform_open_fname(cql_fields);
4245     /* If this fails, no problem: we detect cqltrans == 0 later */
4246
4247 #if HAVE_READLINE_READLINE_H
4248     rl_attempted_completion_function =
4249         (char **(*)(const char *, int, int)) readline_completer;
4250 #endif
4251     for (i = 0; i < maxOtherInfosSupported; ++i)
4252     {
4253         extraOtherInfos[i].oid[0] = -1;
4254         extraOtherInfos[i].value = 0;
4255     }
4256
4257     cmd_format("usmarc");
4258
4259     file_history = file_history_new();
4260
4261     source_rc_file(rc_file);
4262
4263     file_history_load(file_history);
4264     file_history_trav(file_history, 0, add_to_readline_history);
4265 }
4266
4267
4268 #if HAVE_GETTIMEOFDAY
4269 struct timeval tv_start;
4270 #endif
4271
4272 #if YAZ_HAVE_XML2
4273 static void handle_srw_record(Z_SRW_record *rec)
4274 {
4275     if (rec->recordPosition)
4276     {
4277         printf("pos=" ODR_INT_PRINTF, *rec->recordPosition);
4278         setno = *rec->recordPosition + 1;
4279     }
4280     if (rec->recordSchema)
4281         printf(" schema=%s", rec->recordSchema);
4282     printf("\n");
4283     if (rec->recordData_buf && rec->recordData_len)
4284     {
4285         printf("%.*s", rec->recordData_len, rec->recordData_buf);
4286         marc_file_write(rec->recordData_buf, rec->recordData_len);
4287     }
4288     else
4289         printf("No data!");
4290     printf("\n");
4291 }
4292
4293 static void handle_srw_explain_response(Z_SRW_explainResponse *res)
4294 {
4295     handle_srw_record(&res->record);
4296 }
4297
4298 static void handle_srw_response(Z_SRW_searchRetrieveResponse *res)
4299 {
4300     int i;
4301
4302     printf("Received SRW SearchRetrieve Response\n");
4303
4304     for (i = 0; i<res->num_diagnostics; i++)
4305     {
4306         if (res->diagnostics[i].uri)
4307             printf("SRW diagnostic %s\n",
4308                     res->diagnostics[i].uri);
4309         else
4310             printf("SRW diagnostic missing or could not be decoded\n");
4311         if (res->diagnostics[i].message)
4312             printf("Message: %s\n", res->diagnostics[i].message);
4313         if (res->diagnostics[i].details)
4314             printf("Details: %s\n", res->diagnostics[i].details);
4315     }
4316     if (res->numberOfRecords)
4317         printf("Number of hits: " ODR_INT_PRINTF "\n", *res->numberOfRecords);
4318     for (i = 0; i<res->num_records; i++)
4319         handle_srw_record(res->records + i);
4320 }
4321
4322 static void handle_srw_scan_term(Z_SRW_scanTerm *term)
4323 {
4324     if (term->displayTerm)
4325         printf("%s:", term->displayTerm);
4326     else if (term->value)
4327         printf("%s:", term->value);
4328     else
4329         printf("No value:");
4330     if (term->numberOfRecords)
4331         printf(" " ODR_INT_PRINTF, *term->numberOfRecords);
4332     if (term->whereInList)
4333         printf(" %s", term->whereInList);
4334     if (term->value && term->displayTerm)
4335         printf(" %s", term->value);
4336
4337     strcpy(last_scan_line, term->value);
4338     printf("\n");
4339 }
4340
4341 static void handle_srw_scan_response(Z_SRW_scanResponse *res)
4342 {
4343     int i;
4344
4345     printf("Received SRW Scan Response\n");
4346
4347     for (i = 0; i<res->num_diagnostics; i++)
4348     {
4349         if (res->diagnostics[i].uri)
4350             printf("SRW diagnostic %s\n",
4351                     res->diagnostics[i].uri);
4352         else
4353             printf("SRW diagnostic missing or could not be decoded\n");
4354         if (res->diagnostics[i].message)
4355             printf("Message: %s\n", res->diagnostics[i].message);
4356         if (res->diagnostics[i].details)
4357             printf("Details: %s\n", res->diagnostics[i].details);
4358     }
4359     if (res->terms)
4360         for (i = 0; i<res->num_terms; i++)
4361             handle_srw_scan_term(res->terms + i);
4362 }
4363
4364 static void http_response(Z_HTTP_Response *hres)
4365 {
4366     int ret = -1;
4367     const char *connection_head = z_HTTP_header_lookup(hres->headers,
4368                                                        "Connection");
4369     if (hres->code != 200)
4370     {
4371         printf("HTTP Error Status=%d\n", hres->code);
4372     }
4373
4374     if (!yaz_srw_check_content_type(hres))
4375         printf("Content type does not appear to be XML\n");
4376     else
4377     {
4378         if (!yaz_matchstr(sru_method, "solr"))
4379         {
4380             Z_SRW_PDU *sr = 0;
4381             ODR o = odr_createmem(ODR_DECODE);
4382             ret = yaz_solr_decode_response(o, hres, &sr);
4383
4384             if (ret == 0 && sr->which == Z_SRW_searchRetrieve_response)
4385                 handle_srw_response(sr->u.response);
4386             else
4387             {
4388                 printf("Decoding of SOLR package failed\n");
4389                 ret = -1;
4390             }
4391             odr_destroy(o);
4392         }
4393         else
4394         {
4395             Z_SOAP *soap_package = 0;
4396             ODR o = odr_createmem(ODR_DECODE);
4397             Z_SOAP_Handler soap_handlers[3] = {
4398                 {YAZ_XMLNS_SRU_v1_1, 0, (Z_SOAP_fun) yaz_srw_codec},
4399                 {YAZ_XMLNS_UPDATE_v0_9, 0, (Z_SOAP_fun) yaz_ucp_codec},
4400                 {0, 0, 0}
4401             };
4402             ret = z_soap_codec(o, &soap_package,
4403                                &hres->content_buf, &hres->content_len,
4404                                soap_handlers);
4405             if (!ret && soap_package->which == Z_SOAP_generic)
4406             {
4407                 Z_SRW_PDU *sr = (Z_SRW_PDU *) soap_package->u.generic->p;
4408                 if (sr->which == Z_SRW_searchRetrieve_response)
4409                     handle_srw_response(sr->u.response);
4410                 else if (sr->which == Z_SRW_explain_response)
4411                     handle_srw_explain_response(sr->u.explain_response);
4412                 else if (sr->which == Z_SRW_scan_response)
4413                     handle_srw_scan_response(sr->u.scan_response);
4414                 else if (sr->which == Z_SRW_update_response)
4415                     printf("Got update response. Status: %s\n",
4416                            sr->u.update_response->operationStatus);
4417                 else
4418                 {
4419                     printf("Decoding of SRW package failed\n");
4420                     ret = -1;
4421                 }
4422             }
4423             else if (soap_package && (soap_package->which == Z_SOAP_fault
4424                                       || soap_package->which == Z_SOAP_error))
4425             {
4426                 printf("SOAP Fault code %s\n",
4427                        soap_package->u.fault->fault_code);
4428                 printf("SOAP Fault string %s\n",
4429                        soap_package->u.fault->fault_string);
4430                 if (soap_package->u.fault->details)
4431                     printf("SOAP Details %s\n",
4432                            soap_package->u.fault->details);
4433             }
4434             else
4435             {
4436                 printf("z_soap_codec failed. (no SOAP error)\n");
4437                 ret = -1;
4438             }
4439             odr_destroy(o);
4440         }
4441     }
4442     if (ret)
4443         close_session(); /* close session on error */
4444     else
4445     {
4446         if (!strcmp(hres->version, "1.0"))
4447         {
4448             /* HTTP 1.0: only if Keep-Alive we stay alive.. */
4449             if (!connection_head || strcmp(connection_head, "Keep-Alive"))
4450                 close_session();
4451         }
4452         else
4453         {
4454             /* HTTP 1.1: only if no close we stay alive .. */
4455             if (connection_head && !strcmp(connection_head, "close"))
4456                 close_session();
4457         }
4458     }
4459 }
4460 #endif
4461
4462 #define max_HTTP_redirects 2
4463
4464 static void wait_and_handle_response(int one_response_only)
4465 {
4466     int reconnect_ok = 1;
4467     int no_redirects = 0;
4468     int res;
4469     char *netbuffer= 0;
4470     int netbufferlen = 0;
4471 #if HAVE_GETTIMEOFDAY
4472     int got_tv_end = 0;
4473     struct timeval tv_end;
4474 #endif
4475     Z_GDU *gdu;
4476
4477     while(conn)
4478     {
4479         res = cs_get(conn, &netbuffer, &netbufferlen);
4480         if (reconnect_ok && res <= 0 && protocol == PROTO_HTTP)
4481         {
4482             cs_close(conn);
4483             conn = 0;
4484             session_connect(cur_host);
4485             reconnect_ok = 0;
4486             if (conn)
4487             {
4488                 char *buf_out;
4489                 int len_out;
4490
4491                 buf_out = odr_getbuf(out, &len_out, 0);
4492
4493                 do_hex_dump(buf_out, len_out);
4494
4495                 cs_put(conn, buf_out, len_out);
4496
4497                 odr_reset(out);
4498                 continue;
4499             }
4500         }
4501         else if (res <= 0)
4502         {
4503             printf("Target closed connection\n");
4504             close_session();
4505             break;
4506         }
4507 #if HAVE_GETTIMEOFDAY
4508         if (got_tv_end == 0)
4509             gettimeofday(&tv_end, 0); /* count first one only */
4510         got_tv_end++;
4511 #endif
4512         odr_reset(out);
4513         odr_reset(in); /* release APDU from last round */
4514         record_last = 0;
4515         do_hex_dump(netbuffer, res);
4516         odr_setbuf(in, netbuffer, res, 0);
4517
4518         if (!z_GDU(in, &gdu, 0, 0))
4519         {
4520             FILE *f = ber_file ? ber_file : stdout;
4521             odr_perror(in, "Decoding incoming APDU");
4522             fprintf(f, "[Near %ld]\n", (long) odr_offset(in));
4523             fprintf(f, "Packet dump:\n---------\n");
4524             odr_dumpBER(f, netbuffer, res);
4525             fprintf(f, "---------\n");
4526             if (apdu_file)
4527             {
4528                 z_GDU(print, &gdu, 0, 0);
4529                 odr_reset(print);
4530             }
4531             if (conn && cs_more(conn))
4532                 continue;
4533             break;
4534         }
4535         if (ber_file)
4536             odr_dumpBER(ber_file, netbuffer, res);
4537         if (apdu_file && !z_GDU(print, &gdu, 0, 0))
4538         {
4539             odr_perror(print, "Failed to print incoming APDU");
4540             odr_reset(print);
4541                 continue;
4542         }
4543         if (gdu->which == Z_GDU_Z3950)
4544         {
4545             Z_APDU *apdu = gdu->u.z3950;
4546             switch (apdu->which)
4547             {
4548             case Z_APDU_initResponse:
4549                 process_Z3950_initResponse(apdu->u.initResponse);
4550                 break;
4551             case Z_APDU_searchResponse:
4552                 process_Z3950_searchResponse(apdu->u.searchResponse);
4553                 break;
4554             case Z_APDU_scanResponse:
4555                 process_Z3950_scanResponse(apdu->u.scanResponse);
4556                 break;
4557             case Z_APDU_presentResponse:
4558                 print_refid(apdu->u.presentResponse->referenceId);
4559                 setno +=
4560                     *apdu->u.presentResponse->numberOfRecordsReturned;
4561                 if (apdu->u.presentResponse->records)
4562                     display_records(apdu->u.presentResponse->records);
4563                 else
4564                     printf("No records.\n");
4565                 printf("nextResultSetPosition = " ODR_INT_PRINTF "\n",
4566                         *apdu->u.presentResponse->nextResultSetPosition);
4567                 break;
4568             case Z_APDU_sortResponse:
4569                 process_Z3950_sortResponse(apdu->u.sortResponse);
4570                 break;
4571             case Z_APDU_extendedServicesResponse:
4572                 printf("Got extended services response\n");
4573                 process_Z3950_ESResponse(apdu->u.extendedServicesResponse);
4574                 break;
4575             case Z_APDU_close:
4576                 printf("Target has closed the association.\n");
4577                 process_Z3950_close(apdu->u.close);
4578                 break;
4579             case Z_APDU_resourceControlRequest:
4580                 process_Z3950_resourceControlRequest(
4581                     apdu->u.resourceControlRequest);
4582                 break;
4583             case Z_APDU_deleteResultSetResponse:
4584                 process_Z3950_deleteResultSetResponse(
4585                     apdu->u.deleteResultSetResponse);
4586                 break;
4587             default:
4588                 printf("Received unknown APDU type (%d).\n",
4589                        apdu->which);
4590                 close_session();
4591             }
4592         }
4593 #if YAZ_HAVE_XML2
4594         else if (gdu->which == Z_GDU_HTTP_Response)
4595         {
4596             Z_HTTP_Response *hres = gdu->u.HTTP_Response;
4597             int code = hres->code;
4598             const char *location = 0;
4599             if ((code == 301 || code == 302)
4600                 && no_redirects < max_HTTP_redirects
4601                 && !yaz_matchstr(sru_method, "get")
4602                 && (location = z_HTTP_header_lookup(hres->headers, "Location")))
4603             {
4604                 const char *base_tmp;
4605                 session_connect_base(location, &base_tmp);
4606                 no_redirects++;
4607                 if (conn)
4608                 {
4609                     if (send_SRW_redirect(location, hres) == 2)
4610                         continue;
4611                 }
4612                 printf("Redirect failed\n");
4613             }
4614             else
4615                 http_response(gdu->u.HTTP_Response);
4616         }
4617 #endif
4618         if (one_response_only)
4619             break;
4620         if (conn && !cs_more(conn))
4621             break;
4622     }
4623 #if HAVE_GETTIMEOFDAY
4624     if (got_tv_end)
4625     {
4626 #if 0
4627         printf("S/U S/U=%ld/%ld %ld/%ld",
4628                 (long) tv_start.tv_sec,
4629                 (long) tv_start.tv_usec,
4630                 (long) tv_end.tv_sec,
4631                 (long) tv_end.tv_usec);
4632 #endif
4633         printf("Elapsed: %.6f\n",
4634                 (double) tv_end.tv_usec / 1e6 + tv_end.tv_sec -
4635                 ((double) tv_start.tv_usec / 1e6 + tv_start.tv_sec));
4636     }
4637 #endif
4638     xfree(netbuffer);
4639 }
4640
4641 static int cmd_cclparse(const char* arg)
4642 {
4643     int error, pos;
4644     struct ccl_rpn_node *rpn=NULL;
4645
4646
4647     rpn = ccl_find_str(bibset, arg, &error, &pos);
4648
4649     if (error)
4650     {
4651         int ioff = 3+strlen(last_cmd)+1+pos;
4652         printf("%*s^ - ", ioff, " ");
4653         printf("%s\n", ccl_err_msg(error));
4654     }
4655     else
4656     {
4657         if (rpn)
4658         {
4659             ccl_pr_tree(rpn, stdout);
4660         }
4661     }
4662     if (rpn)
4663         ccl_rpn_delete(rpn);
4664
4665     printf("\n");
4666
4667     return 0;
4668 }
4669
4670 static int cmd_set_otherinfo(const char* args)
4671 {
4672     char oidstr[101], otherinfoString[101];
4673     int otherinfoNo;
4674     int sscan_res;
4675
4676     sscan_res = sscanf(args, "%d %100[^ ] %100s",
4677                         &otherinfoNo, oidstr, otherinfoString);
4678
4679     if (sscan_res > 0 && otherinfoNo >= maxOtherInfosSupported)
4680     {
4681         printf("Error otherinfo index too large (%d>=%d)\n",
4682                otherinfoNo,maxOtherInfosSupported);
4683         return 0;
4684     }
4685
4686
4687     if (sscan_res==1)
4688     {
4689         /* reset this otherinfo */
4690         extraOtherInfos[otherinfoNo].oid[0] = -1;
4691         xfree(extraOtherInfos[otherinfoNo].value);
4692         extraOtherInfos[otherinfoNo].value = 0;
4693         return 0;
4694     }
4695     if (sscan_res != 3)
4696     {
4697         printf("Error in set_otherinfo command \n");
4698         return 0;
4699     }
4700     else
4701     {
4702         NMEM oid_tmp = nmem_create();
4703         const Odr_oid *oid =
4704             yaz_string_to_oid_nmem(yaz_oid_std(),
4705                                    CLASS_GENERAL, oidstr, oid_tmp);
4706         oid_oidcpy(extraOtherInfos[otherinfoNo].oid, oid);
4707
4708         xfree(extraOtherInfos[otherinfoNo].value);
4709         extraOtherInfos[otherinfoNo].value = xstrdup(otherinfoString);
4710
4711         nmem_destroy(oid_tmp);
4712     }
4713
4714     return 0;
4715 }
4716
4717 static int cmd_sleep(const char* args )
4718 {
4719     int sec = atoi(args);
4720     if (sec > 0)
4721     {
4722 #ifdef WIN32
4723         Sleep(sec*1000);
4724 #else
4725         sleep(sec);
4726 #endif
4727         printf("Done sleeping %d seconds\n", sec);
4728     }
4729     return 1;
4730 }
4731
4732 static int cmd_list_otherinfo(const char* args)
4733 {
4734     int i;
4735
4736     if (strlen(args)>0)
4737     {
4738         i = atoi(args);
4739         if (i >= maxOtherInfosSupported)
4740         {
4741             printf("Error otherinfo index to large (%d>%d)\n",i,maxOtherInfosSupported);
4742             return 0;
4743         }
4744         if (extraOtherInfos[i].value)
4745         {
4746             char name_oid[OID_STR_MAX];
4747             oid_class oclass;
4748             const char *name =
4749                 yaz_oid_to_string_buf(extraOtherInfos[i].oid, &oclass,
4750                                       name_oid);
4751             printf("  otherinfo %d %s %s\n",
4752                    i, name ? name : "null",
4753                    extraOtherInfos[i].value);
4754         }
4755
4756     }
4757     else
4758     {
4759         for (i = 0; i < maxOtherInfosSupported; ++i)
4760         {
4761             if (extraOtherInfos[i].value)
4762             {
4763                 char name_oid[OID_STR_MAX];
4764                 oid_class oclass;
4765                 const char *name =
4766                     yaz_oid_to_string_buf(extraOtherInfos[i].oid, &oclass,
4767                                           name_oid);
4768                 printf("  otherinfo %d %s %s\n",
4769                        i, name ? name : "null",
4770                        extraOtherInfos[i].value);
4771             }
4772         }
4773     }
4774     return 0;
4775 }
4776
4777 static int cmd_list_all(const char* args)
4778 {
4779     int i;
4780
4781     /* connection options */
4782     if (conn)
4783         printf("Connected to         : %s\n", cur_host);
4784     else if (*cur_host)
4785         printf("Not connected to     : %s\n", cur_host);
4786     else
4787         printf("Not connected        : \n");
4788     if (yazProxy) printf("using proxy          : %s\n",yazProxy);
4789
4790     printf("auto_reconnect       : %s\n",auto_reconnect?"on":"off");
4791     printf("auto_wait            : %s\n",auto_wait?"on":"off");
4792
4793     if (!auth)
4794         printf("Authentication       : none\n");
4795     else
4796     {
4797         switch (auth->which)
4798         {
4799         case Z_IdAuthentication_idPass:
4800             printf("Authentication       : IdPass\n");
4801             printf("    Login User       : %s\n",auth->u.idPass->userId?auth->u.idPass->userId:"");
4802             printf("    Login Group      : %s\n",auth->u.idPass->groupId?auth->u.idPass->groupId:"");
4803             printf("    Password         : %s\n",auth->u.idPass->password?auth->u.idPass->password:"");
4804             break;
4805         case Z_IdAuthentication_open:
4806             printf("Authentication       : psOpen\n");
4807             printf("    Open string      : %s\n",auth->u.open);
4808             break;
4809         default:
4810             printf("Authentication       : Unknown\n");
4811         }
4812     }
4813     if (negotiationCharset)
4814         printf("Neg. Character set   : `%s'\n", negotiationCharset);
4815
4816     /* bases */
4817     printf("Bases                : ");
4818     for (i = 0; i<num_databaseNames; i++) printf("%s ",databaseNames[i]);
4819     printf("\n");
4820
4821     /* Query options */
4822     printf("CCL file             : %s\n",ccl_fields);
4823     printf("CQL file             : %s\n",cql_fields);
4824     printf("Query type           : %s\n",query_type_as_string(queryType));
4825
4826     printf("Named Result Sets    : %s\n",setnumber==-1?"off":"on");
4827
4828     /* piggy back options */
4829     printf("ssub/lslb/mspn       : %d/%d/%d\n",smallSetUpperBound,largeSetLowerBound,mediumSetPresentNumber);
4830
4831     /* print present related options */
4832     if (recordsyntax_size > 0)
4833     {
4834         printf("Format               : %s\n", recordsyntax_list[0]);
4835     }
4836     printf("Schema               : %s\n",record_schema ? record_schema : "not set");
4837     printf("Elements             : %s\n",elementSetNames?elementSetNames->u.generic:"");
4838
4839     /* loging options */
4840     printf("APDU log             : %s\n",apdu_file?"on":"off");
4841     printf("Record log           : %s\n",marc_file?"on":"off");
4842
4843     /* other infos */
4844     printf("Other Info: \n");
4845     cmd_list_otherinfo("");
4846
4847     return 0;
4848 }
4849
4850 static int cmd_clear_otherinfo(const char* args)
4851 {
4852     if (strlen(args) > 0)
4853     {
4854         int otherinfoNo = atoi(args);
4855         if (otherinfoNo >= maxOtherInfosSupported)
4856         {
4857             printf("Error otherinfo index too large (%d>=%d)\n",
4858                    otherinfoNo, maxOtherInfosSupported);
4859             return 0;
4860         }
4861         if (extraOtherInfos[otherinfoNo].value)
4862         {
4863             /* only clear if set. */
4864             extraOtherInfos[otherinfoNo].oid[0] = -1;
4865             xfree(extraOtherInfos[otherinfoNo].value);
4866             extraOtherInfos[otherinfoNo].value = 0;
4867         }
4868     }
4869     else
4870     {
4871         int i;
4872         for (i = 0; i < maxOtherInfosSupported; ++i)
4873         {
4874             if (extraOtherInfos[i].value)
4875             {
4876                 extraOtherInfos[i].oid[0] = -1;
4877                 xfree(extraOtherInfos[i].value);
4878                 extraOtherInfos[i].value = 0;
4879             }
4880         }
4881     }
4882     return 0;
4883 }
4884
4885 static int cmd_wait_response(const char *arg)
4886 {
4887     int i;
4888     int wait_for = atoi(arg);
4889     if (wait_for < 1)
4890         wait_for = 1;
4891
4892     for (i = 0 ; i < wait_for; ++i )
4893         wait_and_handle_response(1);
4894     return 0;
4895 }
4896
4897 static int cmd_help(const char *line);
4898
4899 typedef char *(*completerFunctionType)(const char *text, int state);
4900
4901 static struct {
4902     char *cmd;
4903     int (*fun)(const char *arg);
4904     char *ad;
4905         completerFunctionType rl_completerfunction;
4906     int complete_filenames;
4907     const char **local_tabcompletes;
4908 } cmd_array[] = {
4909     {"open", cmd_open, "('tcp'|'ssl')':<host>[':'<port>][/<db>]",NULL,0,NULL},
4910     {"quit", cmd_quit, "",NULL,0,NULL},
4911     {"find", cmd_find, "<query>",NULL,0,NULL},
4912     {"facets", cmd_facets, "<query>",NULL,0,NULL},
4913     {"delete", cmd_delete, "<setname>",NULL,0,NULL},
4914     {"base", cmd_base, "<base-name>",NULL,0,NULL},
4915     {"show", cmd_show, "<rec#>['+'<#recs>['+'<setname>]]",NULL,0,NULL},
4916     {"setscan", cmd_setscan, "<term>",NULL,0,NULL},
4917     {"scan", cmd_scan, "<term>",NULL,0,NULL},
4918     {"scanstep", cmd_scanstep, "<size>",NULL,0,NULL},
4919     {"scanpos", cmd_scanpos, "<size>",NULL,0,NULL},
4920     {"scansize", cmd_scansize, "<size>",NULL,0,NULL},
4921     {"sort", cmd_sort, "<sortkey> <flag> <sortkey> <flag> ...",NULL,0,NULL},
4922     {"sort+", cmd_sort_newset, "<sortkey> <flag> <sortkey> <flag> ...",NULL,0,NULL},
4923     {"authentication", cmd_authentication, "<acctstring>",NULL,0,NULL},
4924     {"lslb", cmd_lslb, "<largeSetLowerBound>",NULL,0,NULL},
4925     {"ssub", cmd_ssub, "<smallSetUpperBound>",NULL,0,NULL},
4926     {"mspn", cmd_mspn, "<mediumSetPresentNumber>",NULL,0,NULL},
4927     {"status", cmd_status, "",NULL,0,NULL},
4928     {"setnames", cmd_setnames, "",NULL,0,NULL},
4929     {"cancel", cmd_cancel, "",NULL,0,NULL},
4930     {"cancel_find", cmd_cancel_find, "<query>",NULL,0,NULL},
4931     {"format", cmd_format, "<recordsyntax>",complete_format,0,NULL},
4932     {"schema", cmd_schema, "<schema>",complete_schema,0,NULL},
4933     {"elements", cmd_elements, "<elementSetName>",NULL,0,NULL},
4934     {"close", cmd_close, "",NULL,0,NULL},
4935     {"querytype", cmd_querytype, "<type>",complete_querytype,0,NULL},
4936     {"refid", cmd_refid, "<id>",NULL,0,NULL},
4937     {"itemorder", cmd_itemorder, "ill|item|xml <itemno>",NULL,0,NULL},
4938     {"update", cmd_update, "<action> <recid> [<doc>]",NULL,0,NULL},
4939     {"update0", cmd_update0, "<action> <recid> [<doc>]",NULL,0,NULL},
4940     {"xmles", cmd_xmles, "<OID> <doc>",NULL,0,NULL},
4941     {"packagename", cmd_packagename, "<packagename>",NULL,0,NULL},
4942     {"proxy", cmd_proxy, "[('tcp'|'ssl')]<host>[':'<port>]",NULL,0,NULL},
4943     {"charset", cmd_charset, "<nego_charset> <output_charset>",NULL,0,NULL},
4944     {"negcharset", cmd_negcharset, "<nego_charset>",NULL,0,NULL},
4945     {"displaycharset", cmd_displaycharset, "<output_charset>",NULL,0,NULL},
4946     {"marccharset", cmd_marccharset, "<charset_name>",NULL,0,NULL},
4947     {"querycharset", cmd_querycharset, "<charset_name>",NULL,0,NULL},
4948     {"lang", cmd_lang, "<language_code>",NULL,0,NULL},
4949     {"source", cmd_source_echo, "<filename>",NULL,1,NULL},
4950     {".", cmd_source_echo, "<filename>",NULL,1,NULL},
4951     {"!", cmd_subshell, "Subshell command",NULL,1,NULL},
4952     {"set_apdufile", cmd_set_apdufile, "<filename>",NULL,1,NULL},
4953     {"set_berfile", cmd_set_berfile, "<filename>",NULL,1,NULL},
4954     {"set_marcdump", cmd_set_marcdump," <filename>",NULL,1,NULL},
4955     {"set_cclfile", cmd_set_cclfile," <filename>",NULL,1,NULL},
4956     {"set_cqlfile", cmd_set_cqlfile," <filename>",NULL,1,NULL},
4957     {"set_auto_reconnect", cmd_set_auto_reconnect," on|off",complete_auto_reconnect,1,NULL},
4958     {"set_auto_wait", cmd_set_auto_wait," on|off",complete_auto_reconnect,1,NULL},
4959     {"set_otherinfo", cmd_set_otherinfo,"<otherinfoinddex> <oid> <string>",NULL,0,NULL},
4960     {"sleep", cmd_sleep,"<seconds>",NULL,0,NULL},
4961     {"register_oid", cmd_register_oid,"<name> <class> <oid>",NULL,0,NULL},
4962     {"push_command", cmd_push_command,"<command>",command_generator,0,NULL},
4963     {"register_tab", cmd_register_tab,"<commandname> <tab>",command_generator,0,NULL},
4964     {"cclparse", cmd_cclparse,"<ccl find command>",NULL,0,NULL},
4965     {"list_otherinfo",cmd_list_otherinfo,"[otherinfoinddex]",NULL,0,NULL},
4966     {"list_all",cmd_list_all,"",NULL,0,NULL},
4967     {"clear_otherinfo",cmd_clear_otherinfo,"",NULL,0,NULL},
4968     {"wait_response",cmd_wait_response,"<number>",NULL,0,NULL},
4969     /* Server Admin Functions */
4970     {"adm-reindex", cmd_adm_reindex, "<database-name>",NULL,0,NULL},
4971     {"adm-truncate", cmd_adm_truncate, "('database'|'index')<object-name>",NULL,0,NULL},
4972     {"adm-create", cmd_adm_create, "",NULL,0,NULL},
4973     {"adm-drop", cmd_adm_drop, "('database'|'index')<object-name>",NULL,0,NULL},
4974     {"adm-import", cmd_adm_import, "<record-type> <dir> <pattern>",NULL,0,NULL},
4975     {"adm-refresh", cmd_adm_refresh, "",NULL,0,NULL},
4976     {"adm-commit", cmd_adm_commit, "",NULL,0,NULL},
4977     {"adm-shutdown", cmd_adm_shutdown, "",NULL,0,NULL},
4978     {"adm-startup", cmd_adm_startup, "",NULL,0,NULL},
4979     {"explain", cmd_explain, "", NULL, 0, NULL},
4980     {"options", cmd_options, "", NULL, 0, NULL},
4981     {"zversion", cmd_zversion, "", NULL, 0, NULL},
4982     {"help", cmd_help, "", NULL,0,NULL},
4983     {"init", cmd_init, "", NULL,0,NULL},
4984     {"sru", cmd_sru, "<method> <version>", NULL,0,NULL},
4985     {"url", cmd_url, "<url>", NULL,0,NULL},
4986     {"exit", cmd_quit, "",NULL,0,NULL},
4987     {0,0,0,0,0,0}
4988 };
4989
4990 static int cmd_help(const char *line)
4991 {
4992     int i;
4993     char topic[21];
4994
4995     *topic = 0;
4996     sscanf(line, "%20s", topic);
4997
4998     if (*topic == 0)
4999         printf("Commands:\n");
5000     for (i = 0; cmd_array[i].cmd; i++)
5001         if (*topic == 0 || strcmp(topic, cmd_array[i].cmd) == 0)
5002             printf("   %s %s\n", cmd_array[i].cmd, cmd_array[i].ad);
5003     if (!strcmp(topic, "find"))
5004     {
5005         printf("RPN:\n");
5006         printf(" \"term\"                        Simple Term\n");
5007         printf(" @attr [attset] type=value op  Attribute\n");
5008         printf(" @and opl opr                  And\n");
5009         printf(" @or opl opr                   Or\n");
5010         printf(" @not opl opr                  And-Not\n");
5011         printf(" @set set                      Result set\n");
5012         printf(" @prox exl dist ord rel uc ut  Proximity. Use help prox\n");
5013         printf("\n");
5014         printf("Bib-1 attribute types\n");
5015         printf("1=Use:         ");
5016         printf("4=Title 7=ISBN 8=ISSN 30=Date 62=Abstract 1003=Author 1016=Any\n");
5017         printf("2=Relation:    ");
5018         printf("1<   2<=  3=  4>=  5>  6!=  102=Relevance\n");
5019         printf("3=Position:    ");
5020         printf("1=First in Field  2=First in subfield  3=Any position\n");
5021         printf("4=Structure:   ");
5022         printf("1=Phrase  2=Word  3=Key  4=Year  5=Date  6=WordList\n");
5023         printf("5=Truncation:  ");
5024         printf("1=Right  2=Left  3=L&R  100=No  101=#  102=Re-1  103=Re-2\n");
5025         printf("6=Completeness:");
5026         printf("1=Incomplete subfield  2=Complete subfield  3=Complete field\n");
5027     }
5028     if (!strcmp(topic, "prox"))
5029     {
5030         printf("Proximity:\n");
5031         printf(" @prox exl dist ord rel uc ut\n");
5032         printf(" exl:  exclude flag . 0=include, 1=exclude.\n");
5033         printf(" dist: distance integer.\n");
5034         printf(" ord:  order flag. 0=unordered, 1=ordered.\n");
5035         printf(" rel:  relation integer. 1<  2<=  3= 4>=  5>  6!= .\n");
5036         printf(" uc:   unit class. k=known, p=private.\n");
5037         printf(" ut:   unit type. 1=character, 2=word, 3=sentence,\n");
5038         printf("        4=paragraph, 5=section, 6=chapter, 7=document,\n");
5039         printf("        8=element, 9=subelement, 10=elementType, 11=byte.\n");
5040         printf("\nExamples:\n");
5041         printf(" Search for a and b in-order at most 3 words apart:\n");
5042         printf("  @prox 0 3 1 2 k 2 a b\n");
5043         printf(" Search for any order of a and b next to each other:\n");
5044         printf("  @prox 0 1 0 3 k 2 a b\n");
5045     }
5046     return 1;
5047 }
5048
5049 static int cmd_register_tab(const char* arg)
5050 {
5051 #if HAVE_READLINE_READLINE_H
5052     char command[101], tabargument[101];
5053     int i;
5054     int num_of_tabs;
5055     const char** tabslist;
5056
5057     if (sscanf(arg, "%100s %100s", command, tabargument) < 1)
5058     {
5059         return 0;
5060     }
5061
5062     /* locate the amdn in the list */
5063     for (i = 0; cmd_array[i].cmd; i++)
5064     {
5065         if (!strncmp(cmd_array[i].cmd, command, strlen(command)))
5066             break;
5067     }
5068
5069     if (!cmd_array[i].cmd)
5070     {
5071         fprintf(stderr,"Unknown command %s\n",command);
5072         return 1;
5073     }
5074
5075
5076     if (!cmd_array[i].local_tabcompletes)
5077         cmd_array[i].local_tabcompletes = (const char **) calloc(1,sizeof(char**));
5078
5079     num_of_tabs=0;
5080
5081     tabslist = cmd_array[i].local_tabcompletes;
5082     for (; tabslist && *tabslist; tabslist++)
5083         num_of_tabs++;
5084
5085     cmd_array[i].local_tabcompletes = (const char **)
5086         realloc(cmd_array[i].local_tabcompletes,
5087                 (num_of_tabs+2)*sizeof(char**));
5088     tabslist = cmd_array[i].local_tabcompletes;
5089     tabslist[num_of_tabs] = strdup(tabargument);
5090     tabslist[num_of_tabs+1] = NULL;
5091 #endif
5092     return 1;
5093 }
5094
5095 static void process_cmd_line(char* line)
5096 {
5097     int i, res;
5098     char word[32], arg[10240];
5099
5100 #if HAVE_GETTIMEOFDAY
5101     gettimeofday(&tv_start, 0);
5102 #endif
5103
5104     if ((res = sscanf(line, "%31s %10239[^;]", word, arg)) <= 0)
5105     {
5106         strcpy(word, last_cmd);
5107         *arg = '\0';
5108     }
5109     else if (res == 1)
5110         *arg = 0;
5111     strcpy(last_cmd, word);
5112
5113     /* removed tailing spaces from the arg command */
5114     {
5115         char* p = arg;
5116         char* lastnonspace=NULL;
5117
5118         for (; *p; ++p)
5119         {
5120             if (!isspace(*(unsigned char *) p))
5121                 lastnonspace = p;
5122         }
5123         if (lastnonspace)
5124             *(++lastnonspace) = 0;
5125     }
5126
5127     for (i = 0; cmd_array[i].cmd; i++)
5128         if (!strncmp(cmd_array[i].cmd, word, strlen(word)))
5129         {
5130             res = (*cmd_array[i].fun)(arg);
5131             break;
5132         }
5133
5134     if (!cmd_array[i].cmd) /* dump our help-screen */
5135     {
5136         printf("Unknown command: %s.\n", word);
5137         printf("Type 'help' for list of commands\n");
5138         res = 1;
5139     }
5140
5141     if (apdu_file)
5142         fflush(apdu_file);
5143
5144     if (res >= 2 && auto_wait)
5145         wait_and_handle_response(0);
5146
5147     if (apdu_file)
5148         fflush(apdu_file);
5149     if (marc_file)
5150         fflush(marc_file);
5151 }
5152
5153 static char *command_generator(const char *text, int state)
5154 {
5155 #if HAVE_READLINE_READLINE_H
5156     static int idx;
5157     if (state == 0)
5158         idx = 0;
5159     for (; cmd_array[idx].cmd; ++idx)
5160     {
5161         if (!strncmp(cmd_array[idx].cmd, text, strlen(text)))
5162         {
5163             ++idx;  /* skip this entry on the next run */
5164             return strdup(cmd_array[idx-1].cmd);
5165         }
5166     }
5167 #endif
5168     return NULL;
5169 }
5170
5171 #if HAVE_READLINE_READLINE_H
5172 static const char** default_completer_list = NULL;
5173
5174 static char* default_completer(const char* text, int state)
5175 {
5176     return complete_from_list(default_completer_list, text, state);
5177 }
5178 #endif
5179
5180 #if HAVE_READLINE_READLINE_H
5181
5182 /*
5183    This function only known how to complete on the first word
5184 */
5185 static char **readline_completer(char *text, int start, int end)
5186 {
5187     completerFunctionType completerToUse;
5188
5189     if (start == 0)
5190     {
5191 #if HAVE_READLINE_RL_COMPLETION_MATCHES
5192         char** res = rl_completion_matches(text, command_generator);
5193 #else
5194         char** res = completion_matches(text,
5195                                         (CPFunction*)command_generator);
5196 #endif
5197         rl_attempted_completion_over = 1;
5198         return res;
5199     }
5200     else
5201     {
5202         char arg[10240],word[32];
5203         int i ,res;
5204         if ((res = sscanf(rl_line_buffer, "%31s %10239[^;]", word, arg)) <= 0)
5205         {
5206             rl_attempted_completion_over = 1;
5207             return NULL;
5208         }
5209
5210         for (i = 0; cmd_array[i].cmd; i++)
5211             if (!strncmp(cmd_array[i].cmd, word, strlen(word)))
5212                 break;
5213
5214         if (!cmd_array[i].cmd)
5215             return NULL;
5216
5217         default_completer_list = cmd_array[i].local_tabcompletes;
5218
5219         completerToUse = cmd_array[i].rl_completerfunction;
5220         if (!completerToUse)
5221         { /* if command completer is not defined use the default completer */
5222             completerToUse = default_completer;
5223         }
5224         if (completerToUse)
5225         {
5226 #ifdef HAVE_READLINE_RL_COMPLETION_MATCHES
5227             char** res=
5228                 rl_completion_matches(text, completerToUse);
5229 #else
5230             char** res=
5231                 completion_matches(text, (CPFunction*)completerToUse);
5232 #endif
5233             if (!cmd_array[i].complete_filenames)
5234                 rl_attempted_completion_over = 1;
5235             return res;
5236         }
5237         else
5238         {
5239             if (!cmd_array[i].complete_filenames)
5240                 rl_attempted_completion_over = 1;
5241             return 0;
5242         }
5243     }
5244 }
5245 #endif
5246
5247 #ifndef WIN32
5248 static void ctrl_c_handler(int x)
5249 {
5250     exit_client(0);
5251 }
5252 #endif
5253
5254 static void client(void)
5255 {
5256     char line[10240];
5257
5258     line[10239] = '\0';
5259
5260 #ifndef WIN32
5261     signal(SIGINT, ctrl_c_handler);
5262 #endif
5263
5264 #if HAVE_GETTIMEOFDAY
5265     gettimeofday(&tv_start, 0);
5266 #endif
5267
5268     while (1)
5269     {
5270         char *line_in = NULL;
5271 #if HAVE_READLINE_READLINE_H
5272         if (isatty(0))
5273         {
5274             line_in=readline(C_PROMPT);
5275             if (!line_in)
5276             {
5277                 putchar('\n');
5278                 break;
5279             }
5280 #if HAVE_READLINE_HISTORY_H
5281             if (*line_in)
5282                 add_history(line_in);
5283 #endif
5284             strncpy(line, line_in, sizeof(line)-1);
5285             free(line_in);
5286         }
5287 #endif
5288         if (!line_in)
5289         {
5290             char *end_p;
5291             printf(C_PROMPT);
5292             fflush(stdout);
5293             if (!fgets(line, sizeof(line)-1, stdin))
5294                 break;
5295             if ((end_p = strchr(line, '\n')))
5296                 *end_p = '\0';
5297         }
5298         if (isatty(0))
5299             file_history_add_line(file_history, line);
5300         process_cmd_line(line);
5301     }
5302 }
5303
5304 static void show_version(void)
5305 {
5306     char vstr[20], sha1_str[41];
5307
5308     yaz_version(vstr, sha1_str);
5309     printf("YAZ version: %s %s\n", YAZ_VERSION, YAZ_VERSION_SHA1);
5310     if (strcmp(sha1_str, YAZ_VERSION_SHA1))
5311         printf("YAZ DLL/SO: %s %s\n", vstr, sha1_str);
5312     exit(0);
5313 }
5314
5315 int main(int argc, char **argv)
5316 {
5317     char *prog = *argv;
5318     char *open_command = 0;
5319     char *auth_command = 0;
5320     char *arg;
5321     const char *rc_file = 0;
5322     int ret;
5323
5324 #if HAVE_LOCALE_H
5325     if (!setlocale(LC_CTYPE, ""))
5326         fprintf(stderr, "setlocale failed\n");
5327 #endif
5328 #if HAVE_LANGINFO_H
5329 #ifdef CODESET
5330     codeset = nl_langinfo(CODESET);
5331 #endif
5332 #endif
5333     if (codeset)
5334         outputCharset = xstrdup(codeset);
5335
5336     ODR_MASK_SET(&z3950_options, Z_Options_search);
5337     ODR_MASK_SET(&z3950_options, Z_Options_present);
5338     ODR_MASK_SET(&z3950_options, Z_Options_namedResultSets);
5339     ODR_MASK_SET(&z3950_options, Z_Options_triggerResourceCtrl);
5340     ODR_MASK_SET(&z3950_options, Z_Options_scan);
5341     ODR_MASK_SET(&z3950_options, Z_Options_sort);
5342     ODR_MASK_SET(&z3950_options, Z_Options_extendedServices);
5343     ODR_MASK_SET(&z3950_options, Z_Options_delSet);
5344
5345     nmem_auth = nmem_create();
5346
5347     while ((ret = options("k:c:q:a:b:m:v:p:u:t:Vxd:f:", argv, argc, &arg)) != -2)
5348     {
5349         switch (ret)
5350         {
5351         case 0:
5352             if (!open_command)
5353             {
5354                 open_command = (char *) xmalloc(strlen(arg)+6);
5355                 strcpy(open_command, "open ");
5356                 strcat(open_command, arg);
5357             }
5358             else
5359             {
5360                 fprintf(stderr, "%s: Specify at most one server address\n",
5361                         prog);
5362                 exit(1);
5363             }
5364             break;
5365         case 'a':
5366             if (!strcmp(arg, "-"))
5367                 apdu_file=stderr;
5368             else
5369                 apdu_file=fopen(arg, "a");
5370             break;
5371         case 'b':
5372             if (!strcmp(arg, "-"))
5373                 ber_file=stderr;
5374             else
5375                 ber_file=fopen(arg, "a");
5376             break;
5377         case 'c':
5378             strncpy(ccl_fields, arg, sizeof(ccl_fields)-1);
5379             ccl_fields[sizeof(ccl_fields)-1] = '\0';
5380             break;
5381         case 'd':
5382             dump_file_prefix = arg;
5383             break;
5384         case 'f':
5385             rc_file = arg;
5386             break;
5387         case 'k':
5388             kilobytes = atoi(arg);
5389             break;
5390         case 'm':
5391             if (!(marc_file = fopen(arg, "a")))
5392             {
5393                 perror(arg);
5394                 exit(1);
5395             }
5396             break;
5397         case 'p':
5398             yazProxy = xstrdup(arg);
5399             break;
5400         case 'q':
5401             strncpy(cql_fields, arg, sizeof(cql_fields)-1);
5402             cql_fields[sizeof(cql_fields)-1] = '\0';
5403             break;
5404         case 't':
5405             outputCharset = xstrdup(arg);
5406             break;
5407         case 'u':
5408             if (!auth_command)
5409             {
5410                 auth_command = (char *) xmalloc(strlen(arg)+6);
5411                 strcpy(auth_command, "auth ");
5412                 strcat(auth_command, arg);
5413             }
5414             break;
5415         case 'v':
5416             yaz_log_init(yaz_log_mask_str(arg), "", 0);
5417             break;
5418         case 'V':
5419             show_version();
5420             break;
5421         case 'x':
5422             hex_dump = 1;
5423             break;
5424         default:
5425             fprintf(stderr, "Usage: %s "
5426                      " [-a apdulog]"
5427                      " [-b berdump]"
5428                      " [-c cclfile]"
5429                      " [-d dump]"
5430                      " [-f cmdfile]"
5431                      " [-k size]"
5432                      " [-m marclog]"
5433                      " [-p proxy-addr]"
5434                      " [-q cqlfile]"
5435                      " [-t dispcharset]"
5436                      " [-u auth]"
5437                      " [-v loglevel]"
5438                      " [-V]"
5439                      " [-x]"
5440                      " [server-addr]\n",
5441                      prog);
5442             exit(1);
5443         }
5444     }
5445     initialize(rc_file);
5446     if (auth_command)
5447     {
5448 #ifdef HAVE_GETTIMEOFDAY
5449         gettimeofday(&tv_start, 0);
5450 #endif
5451         process_cmd_line(auth_command);
5452 #if HAVE_READLINE_HISTORY_H
5453         add_history(auth_command);
5454 #endif
5455         xfree(auth_command);
5456     }
5457     if (open_command)
5458     {
5459 #ifdef HAVE_GETTIMEOFDAY
5460         gettimeofday(&tv_start, 0);
5461 #endif
5462         process_cmd_line(open_command);
5463 #if HAVE_READLINE_HISTORY_H
5464         add_history(open_command);
5465 #endif
5466         xfree(open_command);
5467     }
5468     client();
5469     exit_client(0);
5470     return 0;
5471 }
5472 /*
5473  * Local variables:
5474  * c-basic-offset: 4
5475  * c-file-style: "Stroustrup"
5476  * indent-tabs-mode: nil
5477  * End:
5478  * vim: shiftwidth=4 tabstop=8 expandtab
5479  */
5480