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