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