client/client.c

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