client/client.c

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