client/client.c

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