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