client/client.c

   1 /* This file is part of the YAZ toolkit.
   2  * Copyright (C) 1995-2012 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 = 64 * 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 static int pretty_xml = 0;
 150 static Odr_int sru_maximumRecords = 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     sru_maximumRecords = 0;
1483     sr->u.request->maximumRecords = odr_intdup(out, 0);
1484     sr->u.request->facetList = facet_list;
1485     if (record_schema)
1486         sr->u.request->recordSchema = record_schema;
1487     if (recordsyntax_size == 1 && !yaz_matchstr(recordsyntax_list[0], "xml"))
1488         sr->u.request->recordPacking = "xml";
1489     return send_srw(sr);
1490 }
1491 #endif
1492
1493 static void query_charset_convert(Z_RPNQuery *q)
1494 {
1495     if (queryCharset && outputCharset)
1496     {
1497         yaz_iconv_t cd = yaz_iconv_open(queryCharset, outputCharset);
1498         if (!cd)
1499         {
1500             printf("Conversion from %s to %s unsupported\n",
1501                    outputCharset, queryCharset);
1502             return;
1503         }
1504         yaz_query_charset_convert_rpnquery(q, out, cd);
1505         yaz_iconv_close(cd);
1506     }
1507 }
1508
1509 static int send_Z3950_searchRequest(const char *arg)
1510 {
1511     Z_APDU *apdu = zget_APDU(out, Z_APDU_searchRequest);
1512     Z_SearchRequest *req = apdu->u.searchRequest;
1513     Z_Query query;
1514     struct ccl_rpn_node *rpn = NULL;
1515     int error, pos;
1516     char setstring[100];
1517     Z_RPNQuery *RPNquery;
1518     Odr_oct ccl_query;
1519     YAZ_PQF_Parser pqf_parser;
1520     Z_External *ext;
1521     QueryType myQueryType = queryType;
1522     char pqfbuf[512];
1523
1524     if (myQueryType == QueryType_CCL2RPN)
1525     {
1526         rpn = ccl_find_str(bibset, arg, &error, &pos);
1527         if (error)
1528         {
1529             printf("CCL ERROR: %s\n", ccl_err_msg(error));
1530             return 0;
1531         }
1532     }
1533     else if (myQueryType == QueryType_CQL2RPN)
1534     {
1535         /* ### All this code should be wrapped in a utility function */
1536         CQL_parser parser;
1537         struct cql_node *node;
1538         const char *addinfo;
1539         if (cqltrans == 0)
1540         {
1541             printf("Can't use CQL: no translation file.  Try set_cqlfile\n");
1542             return 0;
1543         }
1544         parser = cql_parser_create();
1545         if ((error = cql_parser_string(parser, arg)) != 0)
1546         {
1547             printf("Can't parse CQL: must be a syntax error\n");
1548             return 0;
1549         }
1550         node = cql_parser_result(parser);
1551         if ((error = cql_transform_buf(cqltrans, node, pqfbuf,
1552                                        sizeof pqfbuf)) != 0)
1553         {
1554             error = cql_transform_error(cqltrans, &addinfo);
1555             printf("Can't convert CQL to PQF: %s (addinfo=%s)\n",
1556                     cql_strerror(error), addinfo);
1557             return 0;
1558         }
1559         arg = pqfbuf;
1560         myQueryType = QueryType_Prefix;
1561     }
1562
1563     req->referenceId = set_refid(out);
1564     if (!strcmp(arg, "@big")) /* strictly for troublemaking */
1565     {
1566         static unsigned char big[2100];
1567         static Odr_oct bigo;
1568
1569         /* send a very big referenceid to test transport stack etc. */
1570         memset(big, 'A', 2100);
1571         bigo.len = bigo.size = 2100;
1572         bigo.buf = big;
1573         req->referenceId = &bigo;
1574     }
1575
1576     if (setnumber >= 0)
1577     {
1578         sprintf(setstring, "%d", ++setnumber);
1579         req->resultSetName = setstring;
1580     }
1581     *req->smallSetUpperBound = smallSetUpperBound;
1582     *req->largeSetLowerBound = largeSetLowerBound;
1583     *req->mediumSetPresentNumber = mediumSetPresentNumber;
1584     if (smallSetUpperBound > 0 || (largeSetLowerBound > 1 &&
1585         mediumSetPresentNumber > 0))
1586     {
1587         if (recordsyntax_size)
1588             req->preferredRecordSyntax =
1589                 yaz_string_to_oid_odr(yaz_oid_std(),
1590                                       CLASS_RECSYN, recordsyntax_list[0], out);
1591
1592         req->smallSetElementSetNames =
1593             req->mediumSetElementSetNames = elementSetNames;
1594     }
1595     req->num_databaseNames = num_databaseNames;
1596     req->databaseNames = databaseNames;
1597
1598     req->query = &query;
1599
1600     switch (myQueryType)
1601     {
1602     case QueryType_Prefix:
1603         query.which = Z_Query_type_1;
1604         pqf_parser = yaz_pqf_create();
1605         RPNquery = yaz_pqf_parse(pqf_parser, out, arg);
1606         if (!RPNquery)
1607         {
1608             const char *pqf_msg;
1609             size_t off;
1610             int code = yaz_pqf_error(pqf_parser, &pqf_msg, &off);
1611             int ioff = off;
1612             printf("%*s^\n", ioff+4, "");
1613             printf("Prefix query error: %s (code %d)\n", pqf_msg, code);
1614
1615             yaz_pqf_destroy(pqf_parser);
1616             return 0;
1617         }
1618         yaz_pqf_destroy(pqf_parser);
1619         query_charset_convert(RPNquery);
1620         query.u.type_1 = RPNquery;
1621         break;
1622     case QueryType_CCL:
1623         query.which = Z_Query_type_2;
1624         query.u.type_2 = &ccl_query;
1625         ccl_query.buf = (unsigned char*) arg;
1626         ccl_query.len = strlen(arg);
1627         break;
1628     case QueryType_CCL2RPN:
1629         query.which = Z_Query_type_1;
1630         RPNquery = ccl_rpn_query(out, rpn);
1631         if (!RPNquery)
1632         {
1633             printf("Couldn't convert from CCL to RPN\n");
1634             return 0;
1635         }
1636         query_charset_convert(RPNquery);
1637         query.u.type_1 = RPNquery;
1638         ccl_rpn_delete(rpn);
1639         break;
1640     case QueryType_CQL:
1641         query.which = Z_Query_type_104;
1642         ext = (Z_External *) odr_malloc(out, sizeof(*ext));
1643         ext->direct_reference = odr_getoidbystr(out, "1.2.840.10003.16.2");
1644         ext->indirect_reference = 0;
1645         ext->descriptor = 0;
1646         ext->which = Z_External_CQL;
1647         ext->u.cql = odr_strdup(out, arg);
1648         query.u.type_104 =  ext;
1649         break;
1650     default:
1651         printf("Unsupported query type\n");
1652         return 0;
1653     }
1654     if (send_apdu(apdu))
1655         printf("Sent searchRequest.\n");
1656     setno = 1;
1657     return 2;
1658 }
1659
1660 static void display_term(Z_Term *term)
1661 {
1662     switch (term->which)
1663     {
1664     case Z_Term_general:
1665         printf("%.*s", term->u.general->len, term->u.general->buf);
1666         break;
1667     case Z_Term_characterString:
1668         printf("%s", term->u.characterString);
1669         break;
1670     case Z_Term_numeric:
1671         printf(ODR_INT_PRINTF, *term->u.numeric);
1672         break;
1673     case Z_Term_null:
1674         printf("null");
1675         break;
1676     }
1677 }
1678
1679 /* display Query Expression as part of searchResult-1 */
1680 static void display_queryExpression(const char *lead, Z_QueryExpression *qe)
1681 {
1682     if (!qe)
1683         return;
1684     printf(" %s=", lead);
1685     if (qe->which == Z_QueryExpression_term)
1686     {
1687         if (qe->u.term->queryTerm)
1688         {
1689             Z_Term *term = qe->u.term->queryTerm;
1690             display_term(term);
1691         }
1692     }
1693 }
1694
1695 static void display_facet(Z_FacetField *facet)
1696 {
1697     if (facet->attributes)
1698     {
1699         Z_AttributeList *al = facet->attributes;
1700         struct yaz_facet_attr attr_values;
1701         attr_values.errcode = 0;
1702         attr_values.limit = -1;
1703         attr_values.useattr = "NONE";
1704         attr_values.relation = "default";
1705
1706         yaz_facet_attr_get_z_attributes(al, &attr_values);
1707         if (!attr_values.errcode)
1708         {
1709             int term_index;
1710             printf("  %s (%d): \n", attr_values.useattr, facet->num_terms);
1711             for (term_index = 0 ; term_index < facet->num_terms; term_index++)
1712             {
1713                 Z_FacetTerm *facetTerm = facet->terms[term_index];
1714                 printf("    ");
1715                 display_term(facetTerm->term);
1716                 printf(" (" NMEM_INT_PRINTF ")\n", *facetTerm->count);
1717             }
1718         }
1719
1720     }
1721 }
1722
1723 static void* display_facets(Z_FacetList *fl)
1724 {
1725     int index;
1726     printf("Facets(%d): \n", fl->num);
1727
1728     for (index = 0; index < fl->num ; index++)
1729     {
1730         display_facet(fl->elements[index]);
1731     }
1732     return 0;
1733 }
1734
1735 void display_searchResult1(Z_SearchInfoReport *sr)
1736 {
1737     int j;
1738     printf("SearchResult-1:");
1739     for (j = 0; j < sr->num; j++)
1740     {
1741         if (j)
1742             printf(",");
1743         if (!sr->elements[j]->subqueryExpression)
1744             printf("%d", j);
1745         display_queryExpression("term",
1746             sr->elements[j]->subqueryExpression);
1747         display_queryExpression("interpretation",
1748             sr->elements[j]->subqueryInterpretation);
1749         display_queryExpression("recommendation",
1750             sr->elements[j]->subqueryRecommendation);
1751         if (sr->elements[j]->subqueryCount)
1752             printf(" cnt=" ODR_INT_PRINTF,
1753                    *sr->elements[j]->subqueryCount);
1754         if (sr->elements[j]->subqueryId)
1755             printf(" id=%s ", sr->elements[j]->subqueryId);
1756     }
1757     printf("\n");
1758 }
1759
1760
1761
1762 /* see if we can find USR:SearchResult-1 */
1763 static void display_searchResult(Z_OtherInformation *o)
1764 {
1765     int i;
1766     if (!o)
1767         return ;
1768     for (i = 0; i < o->num_elements; i++)
1769     {
1770         if (o->list[i]->which == Z_OtherInfo_externallyDefinedInfo)
1771         {
1772             Z_External *ext = o->list[i]->information.externallyDefinedInfo;
1773
1774             if (ext->which == Z_External_searchResult1)
1775                 display_searchResult1(ext->u.searchResult1);
1776             else if  (ext->which == Z_External_userFacets)
1777                 display_facets(ext->u.facetList);
1778         }
1779     }
1780 }
1781
1782 static int process_Z3950_searchResponse(Z_SearchResponse *res)
1783 {
1784     printf("Received SearchResponse.\n");
1785     print_refid(res->referenceId);
1786     if (*res->searchStatus)
1787         printf("Search was a success.\n");
1788     else
1789         printf("Search was a bloomin' failure.\n");
1790     printf("Number of hits: " ODR_INT_PRINTF, *res->resultCount);
1791     last_hit_count = *res->resultCount;
1792     if (setnumber >= 0)
1793         printf(", setno %d", setnumber);
1794     putchar('\n');
1795     if (res->resultSetStatus)
1796     {
1797         printf("Result Set Status: ");
1798         switch (*res->resultSetStatus)
1799         {
1800         case Z_SearchResponse_subset:
1801             printf("subset"); break;
1802         case Z_SearchResponse_interim:
1803             printf("interim"); break;
1804         case Z_SearchResponse_none:
1805             printf("none"); break;
1806         case Z_SearchResponse_estimate:
1807             printf("estimate"); break;
1808         default:
1809             printf(ODR_INT_PRINTF, *res->resultSetStatus);
1810         }
1811         putchar('\n');
1812     }
1813     display_searchResult(res->additionalSearchInfo);
1814     printf("records returned: " ODR_INT_PRINTF "\n",
1815            *res->numberOfRecordsReturned);
1816     setno += *res->numberOfRecordsReturned;
1817     if (res->records)
1818         display_records(res->records);
1819     return 0;
1820 }
1821
1822 static void print_level(int iLevel)
1823 {
1824     int i;
1825     for (i = 0; i < iLevel * 4; i++)
1826         printf(" ");
1827 }
1828
1829 static void print_int(int iLevel, const char *pTag, Odr_int *pInt)
1830 {
1831     if (pInt != NULL)
1832     {
1833         print_level(iLevel);
1834         printf("%s: " ODR_INT_PRINTF "\n", pTag, *pInt);
1835     }
1836 }
1837
1838 static void print_bool(int iLevel, const char *pTag, Odr_bool *pInt)
1839 {
1840     if (pInt != NULL)
1841     {
1842         print_level(iLevel);
1843         printf("%s: %d\n", pTag, *pInt);
1844     }
1845 }
1846
1847 static void print_string(int iLevel, const char *pTag, const char *pString)
1848 {
1849     if (pString != NULL)
1850     {
1851         print_level(iLevel);
1852         printf("%s: %s\n", pTag, pString);
1853     }
1854 }
1855
1856 static void print_oid(int iLevel, const char *pTag, Odr_oid *pOid)
1857 {
1858     if (pOid != NULL)
1859     {
1860         Odr_oid *pInt = pOid;
1861
1862         print_level(iLevel);
1863         printf("%s:", pTag);
1864         for (; *pInt != -1; pInt++)
1865             printf(" %d", *pInt);
1866         printf("\n");
1867     }
1868 }
1869
1870 static void print_referenceId(int iLevel, Z_ReferenceId *referenceId)
1871 {
1872     if (referenceId != NULL)
1873     {
1874         int i;
1875
1876         print_level(iLevel);
1877         printf("Ref Id (%d, %d): ", referenceId->len, referenceId->size);
1878         for (i = 0; i < referenceId->len; i++)
1879             printf("%c", referenceId->buf[i]);
1880         printf("\n");
1881     }
1882 }
1883
1884 static void print_string_or_numeric(int iLevel, const char *pTag, Z_StringOrNumeric *pStringNumeric)
1885 {
1886     if (pStringNumeric != NULL)
1887     {
1888         switch (pStringNumeric->which)
1889         {
1890         case Z_StringOrNumeric_string:
1891             print_string(iLevel, pTag, pStringNumeric->u.string);
1892             break;
1893
1894         case Z_StringOrNumeric_numeric:
1895             print_int(iLevel, pTag, pStringNumeric->u.numeric);
1896             break;
1897
1898         default:
1899             print_level(iLevel);
1900             printf("%s: valid type for Z_StringOrNumeric\n", pTag);
1901             break;
1902         }
1903     }
1904 }
1905
1906 static void print_universe_report_duplicate(
1907     int iLevel,
1908     Z_UniverseReportDuplicate *pUniverseReportDuplicate)
1909 {
1910     if (pUniverseReportDuplicate != NULL)
1911     {
1912         print_level(iLevel);
1913         printf("Universe Report Duplicate: \n");
1914         iLevel++;
1915         print_string_or_numeric(iLevel, "Hit No",
1916                                 pUniverseReportDuplicate->hitno);
1917     }
1918 }
1919
1920 static void print_universe_report_hits(
1921     int iLevel,
1922     Z_UniverseReportHits *pUniverseReportHits)
1923 {
1924     if (pUniverseReportHits != NULL)
1925     {
1926         print_level(iLevel);
1927         printf("Universe Report Hits: \n");
1928         iLevel++;
1929         print_string_or_numeric(iLevel, "Database",
1930                                 pUniverseReportHits->database);
1931         print_string_or_numeric(iLevel, "Hits", pUniverseReportHits->hits);
1932     }
1933 }
1934
1935 static void print_universe_report(int iLevel, Z_UniverseReport *pUniverseReport)
1936 {
1937     if (pUniverseReport != NULL)
1938     {
1939         print_level(iLevel);
1940         printf("Universe Report: \n");
1941         iLevel++;
1942         print_int(iLevel, "Total Hits", pUniverseReport->totalHits);
1943         switch (pUniverseReport->which)
1944         {
1945         case Z_UniverseReport_databaseHits:
1946             print_universe_report_hits(iLevel,
1947                                        pUniverseReport->u.databaseHits);
1948             break;
1949
1950         case Z_UniverseReport_duplicate:
1951             print_universe_report_duplicate(iLevel,
1952                                             pUniverseReport->u.duplicate);
1953             break;
1954
1955         default:
1956             print_level(iLevel);
1957             printf("Type: %d\n", pUniverseReport->which);
1958             break;
1959         }
1960     }
1961 }
1962
1963 static void print_external(int iLevel, Z_External *pExternal)
1964 {
1965     if (pExternal != NULL)
1966     {
1967         print_level(iLevel);
1968         printf("External: \n");
1969         iLevel++;
1970         print_oid(iLevel, "Direct Reference", pExternal->direct_reference);
1971         print_int(iLevel, "InDirect Reference", pExternal->indirect_reference);
1972         print_string(iLevel, "Descriptor", pExternal->descriptor);
1973         switch (pExternal->which)
1974         {
1975         case Z_External_universeReport:
1976             print_universe_report(iLevel, pExternal->u.universeReport);
1977             break;
1978
1979         default:
1980             print_level(iLevel);
1981             printf("Type: %d\n", pExternal->which);
1982             break;
1983         }
1984     }
1985 }
1986
1987 static int process_Z3950_resourceControlRequest(Z_ResourceControlRequest *req)
1988 {
1989     printf("Received ResourceControlRequest.\n");
1990     print_referenceId(1, req->referenceId);
1991     print_bool(1, "Suspended Flag", req->suspendedFlag);
1992     print_int(1, "Partial Results Available", req->partialResultsAvailable);
1993     print_bool(1, "Response Required", req->responseRequired);
1994     print_bool(1, "Triggered Request Flag", req->triggeredRequestFlag);
1995     print_external(1, req->resourceReport);
1996     return 0;
1997 }
1998
1999 static void process_Z3950_ESResponse(Z_ExtendedServicesResponse *res)
2000 {
2001     printf("Status: ");
2002     switch (*res->operationStatus)
2003     {
2004     case Z_ExtendedServicesResponse_done:
2005         printf("done\n");
2006         break;
2007     case Z_ExtendedServicesResponse_accepted:
2008         printf("accepted\n");
2009         break;
2010     case Z_ExtendedServicesResponse_failure:
2011         printf("failure\n");
2012         display_diagrecs(res->diagnostics, res->num_diagnostics);
2013         break;
2014     default:
2015         printf("unknown\n");
2016     }
2017     if ( (*res->operationStatus != Z_ExtendedServicesResponse_failure) &&
2018         (res->num_diagnostics != 0) )
2019     {
2020         display_diagrecs(res->diagnostics, res->num_diagnostics);
2021     }
2022     print_refid (res->referenceId);
2023     if (res->taskPackage &&
2024         res->taskPackage->which == Z_External_extendedService)
2025     {
2026         Z_TaskPackage *taskPackage = res->taskPackage->u.extendedService;
2027         Odr_oct *id = taskPackage->targetReference;
2028         Z_External *ext = taskPackage->taskSpecificParameters;
2029
2030         if (id)
2031         {
2032             printf("Target Reference: ");
2033             print_stringn((const char *) id->buf, id->len);
2034             printf("\n");
2035         }
2036         if (ext->which == Z_External_update)
2037         {
2038             Z_IUUpdateTaskPackage *utp = ext->u.update->u.taskPackage;
2039             if (utp && utp->targetPart)
2040             {
2041                 Z_IUTargetPart *targetPart = utp->targetPart;
2042                 int i;
2043
2044                 for (i = 0; i<targetPart->num_taskPackageRecords;  i++)
2045                 {
2046
2047                     Z_IUTaskPackageRecordStructure *tpr =
2048                         targetPart->taskPackageRecords[i];
2049                     printf("task package record %d\n", i+1);
2050                     if (tpr->which == Z_IUTaskPackageRecordStructure_record)
2051                     {
2052                         display_record (tpr->u.record);
2053                     }
2054                     else
2055                     {
2056                         printf("other type\n");
2057                     }
2058                 }
2059             }
2060         }
2061         if (ext->which == Z_External_itemOrder)
2062         {
2063             Z_IOTaskPackage *otp = ext->u.itemOrder->u.taskPackage;
2064
2065             if (otp && otp->targetPart)
2066             {
2067                 if (otp->targetPart->itemRequest)
2068                 {
2069                     Z_External *ext = otp->targetPart->itemRequest;
2070                     if (ext->which == Z_External_octet)
2071                     {
2072                         Odr_oct *doc = ext->u.octet_aligned;
2073                         printf("Got itemRequest doc %.*s\n",
2074                                doc->len, doc->buf);
2075                     }
2076                 }
2077                 else if (otp->targetPart->statusOrErrorReport)
2078                 {
2079                     Z_External *ext = otp->targetPart->statusOrErrorReport;
2080                     if (ext->which == Z_External_octet)
2081                     {
2082                         Odr_oct *doc = ext->u.octet_aligned;
2083                         printf("Got Status or Error Report doc %.*s\n",
2084                                doc->len, doc->buf);
2085                     }
2086                 }
2087             }
2088         }
2089     }
2090     if (res->taskPackage && res->taskPackage->which == Z_External_octet)
2091     {
2092         Odr_oct *doc = res->taskPackage->u.octet_aligned;
2093         printf("%.*s\n", doc->len, doc->buf);
2094     }
2095 }
2096
2097 static const char *get_ill_element(void *clientData, const char *element)
2098 {
2099     return 0;
2100 }
2101
2102 static Z_External *create_external_itemRequest(void)
2103 {
2104     struct ill_get_ctl ctl;
2105     ILL_ItemRequest *req;
2106     Z_External *r = 0;
2107     int item_request_size = 0;
2108     char *item_request_buf = 0;
2109
2110     ctl.odr = out;
2111     ctl.clientData = 0;
2112     ctl.f = get_ill_element;
2113
2114     req = ill_get_ItemRequest(&ctl, "ill", 0);
2115     if (!req)
2116         printf("ill_get_ItemRequest failed\n");
2117
2118     if (!ill_ItemRequest(out, &req, 0, 0))
2119     {
2120         if (apdu_file)
2121         {
2122             ill_ItemRequest(print, &req, 0, 0);
2123             odr_reset(print);
2124         }
2125         item_request_buf = odr_getbuf (out, &item_request_size, 0);
2126         if (item_request_buf)
2127             odr_setbuf (out, item_request_buf, item_request_size, 1);
2128         printf("Couldn't encode ItemRequest, size %d\n", item_request_size);
2129         return 0;
2130     }
2131     else
2132     {
2133         item_request_buf = odr_getbuf (out, &item_request_size, 0);
2134         r = (Z_External *) odr_malloc(out, sizeof(*r));
2135         r->direct_reference = odr_oiddup(out, yaz_oid_general_isoill_1);
2136         r->indirect_reference = 0;
2137         r->descriptor = 0;
2138         r->which = Z_External_single;
2139
2140         r->u.single_ASN1_type = (Odr_oct *)
2141             odr_malloc(out, sizeof(*r->u.single_ASN1_type));
2142         r->u.single_ASN1_type->buf = (unsigned char *)
2143         odr_malloc(out, item_request_size);
2144         r->u.single_ASN1_type->len = item_request_size;
2145         r->u.single_ASN1_type->size = item_request_size;
2146         memcpy(r->u.single_ASN1_type->buf, item_request_buf,
2147                 item_request_size);
2148
2149         do_hex_dump(item_request_buf,item_request_size);
2150     }
2151     return r;
2152 }
2153
2154 static Z_External *create_external_ILL_APDU(void)
2155 {
2156     struct ill_get_ctl ctl;
2157     ILL_APDU *ill_apdu;
2158     Z_External *r = 0;
2159     int ill_request_size = 0;
2160     char *ill_request_buf = 0;
2161
2162     ctl.odr = out;
2163     ctl.clientData = 0;
2164     ctl.f = get_ill_element;
2165
2166     ill_apdu = ill_get_APDU(&ctl, "ill", 0);
2167
2168     if (!ill_APDU (out, &ill_apdu, 0, 0))
2169     {
2170         if (apdu_file)
2171         {
2172             printf("-------------------\n");
2173             ill_APDU(print, &ill_apdu, 0, 0);
2174             odr_reset(print);
2175             printf("-------------------\n");
2176         }
2177         ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
2178         if (ill_request_buf)
2179             odr_setbuf (out, ill_request_buf, ill_request_size, 1);
2180         printf("Couldn't encode ILL-Request, size %d\n", ill_request_size);
2181         return 0;
2182     }
2183     else
2184     {
2185         ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
2186
2187         r = (Z_External *) odr_malloc(out, sizeof(*r));
2188         r->direct_reference = odr_oiddup(out, yaz_oid_general_isoill_1);
2189         r->indirect_reference = 0;
2190         r->descriptor = 0;
2191         r->which = Z_External_single;
2192
2193         r->u.single_ASN1_type = (Odr_oct *)
2194             odr_malloc(out, sizeof(*r->u.single_ASN1_type));
2195         r->u.single_ASN1_type->buf = (unsigned char *)
2196         odr_malloc(out, ill_request_size);
2197         r->u.single_ASN1_type->len = ill_request_size;
2198         r->u.single_ASN1_type->size = ill_request_size;
2199         memcpy(r->u.single_ASN1_type->buf, ill_request_buf, ill_request_size);
2200 /*         printf("len = %d\n", ill_request_size); */
2201 /*              do_hex_dump(ill_request_buf,ill_request_size); */
2202 /*              printf("--- end of extenal\n"); */
2203
2204     }
2205     return r;
2206 }
2207
2208
2209 static Z_External *create_ItemOrderExternal(const char *type, int itemno,
2210                                             const char *xml_buf,
2211                                             int xml_len)
2212 {
2213     Z_External *r = (Z_External *) odr_malloc(out, sizeof(Z_External));
2214     r->direct_reference = odr_oiddup(out, yaz_oid_extserv_item_order);
2215     r->indirect_reference = 0;
2216     r->descriptor = 0;
2217
2218     r->which = Z_External_itemOrder;
2219
2220     r->u.itemOrder = (Z_ItemOrder *) odr_malloc(out,sizeof(Z_ItemOrder));
2221     memset(r->u.itemOrder, 0, sizeof(Z_ItemOrder));
2222     r->u.itemOrder->which=Z_IOItemOrder_esRequest;
2223
2224     r->u.itemOrder->u.esRequest = (Z_IORequest *)
2225         odr_malloc(out,sizeof(Z_IORequest));
2226     memset(r->u.itemOrder->u.esRequest, 0, sizeof(Z_IORequest));
2227
2228     r->u.itemOrder->u.esRequest->toKeep = (Z_IOOriginPartToKeep *)
2229         odr_malloc(out,sizeof(Z_IOOriginPartToKeep));
2230     memset(r->u.itemOrder->u.esRequest->toKeep, 0, sizeof(Z_IOOriginPartToKeep));
2231     r->u.itemOrder->u.esRequest->notToKeep = (Z_IOOriginPartNotToKeep *)
2232         odr_malloc(out,sizeof(Z_IOOriginPartNotToKeep));
2233     memset(r->u.itemOrder->u.esRequest->notToKeep, 0, sizeof(Z_IOOriginPartNotToKeep));
2234
2235     r->u.itemOrder->u.esRequest->toKeep->supplDescription = NULL;
2236     r->u.itemOrder->u.esRequest->toKeep->contact = NULL;
2237     r->u.itemOrder->u.esRequest->toKeep->addlBilling = NULL;
2238
2239     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem =
2240         (Z_IOResultSetItem *) odr_malloc(out, sizeof(Z_IOResultSetItem));
2241     memset(r->u.itemOrder->u.esRequest->notToKeep->resultSetItem, 0, sizeof(Z_IOResultSetItem));
2242     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->resultSetId = "1";
2243
2244     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->item =
2245         odr_intdup(out, itemno);
2246     if (!strcmp (type, "item") || !strcmp(type, "2"))
2247     {
2248         printf("using item-request\n");
2249         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2250             create_external_itemRequest();
2251     }
2252     else if (!strcmp(type, "ill") || !strcmp(type, "1"))
2253     {
2254         printf("using ILL-request\n");
2255         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2256             create_external_ILL_APDU();
2257     }
2258     else if (!strcmp(type, "xml") || !strcmp(type, "3"))
2259     {
2260         printf("using XML ILL-request\n");
2261
2262         if (!xml_buf)
2263         {
2264             printf("no docoument added\n");
2265             r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
2266         }
2267         else
2268         {
2269             r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2270                 z_ext_record_oid(out, yaz_oid_recsyn_xml, xml_buf, xml_len);
2271         }
2272     }
2273     else
2274         r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
2275
2276     return r;
2277 }
2278
2279 static int send_Z3950_itemorder(const char *type, int itemno,
2280                                 const char *xml_buf, int xml_len)
2281 {
2282     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest);
2283     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2284
2285     req->referenceId = set_refid (out);
2286
2287     req->packageType = odr_oiddup(out, yaz_oid_extserv_item_order);
2288     req->packageName = esPackageName;
2289
2290     req->taskSpecificParameters = create_ItemOrderExternal(type, itemno,
2291                                                            xml_buf, xml_len);
2292     send_apdu(apdu);
2293     return 0;
2294 }
2295
2296 static int only_z3950(void)
2297 {
2298     if (!conn)
2299     {
2300         printf("Not connected yet\n");
2301         return 1;
2302     }
2303     if (protocol == PROTO_HTTP)
2304     {
2305         printf("Not supported by SRW\n");
2306         return 1;
2307     }
2308     return 0;
2309 }
2310
2311 static int cmd_update_common(const char *arg, int version);
2312
2313 static int cmd_update(const char *arg)
2314 {
2315     return cmd_update_common(arg, 1);
2316 }
2317
2318 static int cmd_update0(const char *arg)
2319 {
2320     return cmd_update_common(arg, 0);
2321 }
2322
2323 static int send_Z3950_update(int version, int action_no, const char *recid,
2324                             char *rec_buf, int rec_len);
2325
2326 #if YAZ_HAVE_XML2
2327 static int send_SRW_update(int action_no, const char *recid,
2328                           char *rec_buf, int rec_len);
2329 #endif
2330
2331 static int cmd_update_common(const char *arg, int version)
2332 {
2333     char *action_buf;
2334     int action_len;
2335     char *recid_buf;
2336     int recid_len;
2337     const char *recid = 0;
2338     char *rec_buf;
2339     int rec_len;
2340     int action_no;
2341     int noread = 0;
2342
2343     if (parse_cmd_doc(&arg, out, &action_buf, &action_len) == 0)
2344     {
2345         printf("Use: update action recid [fname]\n");
2346         printf(" where action is one of insert,replace,delete.update\n");
2347         printf(" recid is some record ID. Use none for no ID\n");
2348         printf(" fname is file of record to be updated\n");
2349         return 0;
2350     }
2351
2352     if (parse_cmd_doc(&arg, out, &recid_buf, &recid_len) == 0)
2353     {
2354         printf("Missing recid\n");
2355         return 0;
2356     }
2357
2358     if (!strcmp(action_buf, "insert"))
2359         action_no = Z_IUOriginPartToKeep_recordInsert;
2360     else if (!strcmp(action_buf, "replace"))
2361         action_no = Z_IUOriginPartToKeep_recordReplace;
2362     else if (!strcmp(action_buf, "delete"))
2363         action_no = Z_IUOriginPartToKeep_recordDelete;
2364     else if (!strcmp(action_buf, "update"))
2365         action_no = Z_IUOriginPartToKeep_specialUpdate;
2366     else
2367     {
2368         printf("Bad action: %s\n", action_buf);
2369         printf("Possible values: insert, replace, delete, update\n");
2370         return 0;
2371     }
2372
2373     if (strcmp(recid_buf, "none")) /* none means no record ID */
2374         recid = recid_buf;
2375
2376     arg += noread;
2377     if (parse_cmd_doc(&arg, out, &rec_buf, &rec_len) == 0)
2378         return 0;
2379
2380 #if YAZ_HAVE_XML2
2381     if (protocol == PROTO_HTTP)
2382         return send_SRW_update(action_no, recid, rec_buf, rec_len);
2383 #endif
2384     return send_Z3950_update(version, action_no, recid, rec_buf, rec_len);
2385 }
2386
2387 #if YAZ_HAVE_XML2
2388 static int send_SRW_update(int action_no, const char *recid,
2389                            char *rec_buf, int rec_len)
2390 {
2391     if (!conn)
2392         session_connect(cur_host);
2393     if (!conn)
2394         return 0;
2395     else
2396     {
2397         Z_SRW_PDU *srw = yaz_srw_get(out, Z_SRW_update_request);
2398         Z_SRW_updateRequest *sr = srw->u.update_request;
2399
2400         switch (action_no)
2401         {
2402         case Z_IUOriginPartToKeep_recordInsert:
2403             sr->operation = "info:srw/action/1/create";
2404             break;
2405         case Z_IUOriginPartToKeep_recordReplace:
2406             sr->operation = "info:srw/action/1/replace";
2407             break;
2408         case Z_IUOriginPartToKeep_recordDelete:
2409             sr->operation = "info:srw/action/1/delete";
2410             break;
2411         }
2412         if (rec_buf)
2413         {
2414             sr->record = yaz_srw_get_record(out);
2415             sr->record->recordData_buf = rec_buf;
2416             sr->record->recordData_len = rec_len;
2417             sr->record->recordSchema = record_schema;
2418         }
2419         if (recid)
2420             sr->recordId = odr_strdup(out, recid);
2421         return send_srw(srw);
2422     }
2423 }
2424 #endif
2425
2426 static int send_Z3950_update(int version, int action_no, const char *recid,
2427                             char *rec_buf, int rec_len)
2428 {
2429     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest );
2430     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2431     Z_External *r;
2432     Z_External *record_this = 0;
2433     if (rec_buf)
2434         record_this = z_ext_record_oid(out, yaz_oid_recsyn_xml,
2435                                        rec_buf, rec_len);
2436     else
2437     {
2438         if (!record_last)
2439         {
2440             printf("No last record (update ignored)\n");
2441             return 0;
2442         }
2443         record_this = record_last;
2444     }
2445
2446     req->packageType = odr_oiddup(out, (version == 0 ?
2447        yaz_oid_extserv_database_update_first_version :
2448        yaz_oid_extserv_database_update));
2449
2450     req->packageName = esPackageName;
2451
2452     req->referenceId = set_refid (out);
2453
2454     r = req->taskSpecificParameters = (Z_External *)
2455         odr_malloc(out, sizeof(*r));
2456     r->direct_reference = req->packageType;
2457     r->indirect_reference = 0;
2458     r->descriptor = 0;
2459     if (version == 0)
2460     {
2461         Z_IU0OriginPartToKeep *toKeep;
2462         Z_IU0SuppliedRecords *notToKeep;
2463
2464         r->which = Z_External_update0;
2465         r->u.update0 = (Z_IU0Update *) odr_malloc(out, sizeof(*r->u.update0));
2466         r->u.update0->which = Z_IUUpdate_esRequest;
2467         r->u.update0->u.esRequest = (Z_IU0UpdateEsRequest *)
2468             odr_malloc(out, sizeof(*r->u.update0->u.esRequest));
2469         toKeep = r->u.update0->u.esRequest->toKeep = (Z_IU0OriginPartToKeep *)
2470             odr_malloc(out, sizeof(*r->u.update0->u.esRequest->toKeep));
2471
2472         toKeep->databaseName = databaseNames[0];
2473         toKeep->schema = 0;
2474         if (record_schema)
2475         {
2476             toKeep->schema = yaz_string_to_oid_odr(yaz_oid_std(),
2477                                                    CLASS_SCHEMA,
2478                                                    record_schema, out);
2479         }
2480         toKeep->elementSetName = 0;
2481
2482         toKeep->action = odr_intdup(out, action_no);
2483
2484         notToKeep = r->u.update0->u.esRequest->notToKeep = (Z_IU0SuppliedRecords *)
2485             odr_malloc(out, sizeof(*r->u.update0->u.esRequest->notToKeep));
2486         notToKeep->num = 1;
2487         notToKeep->elements = (Z_IU0SuppliedRecords_elem **)
2488             odr_malloc(out, sizeof(*notToKeep->elements));
2489         notToKeep->elements[0] = (Z_IU0SuppliedRecords_elem *)
2490             odr_malloc(out, sizeof(**notToKeep->elements));
2491         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
2492         if (recid)
2493         {
2494             notToKeep->elements[0]->u.opaque = (Odr_oct *)
2495                 odr_malloc(out, sizeof(Odr_oct));
2496             notToKeep->elements[0]->u.opaque->buf = (unsigned char *) recid;
2497             notToKeep->elements[0]->u.opaque->size = strlen(recid);
2498             notToKeep->elements[0]->u.opaque->len = strlen(recid);
2499         }
2500         else
2501             notToKeep->elements[0]->u.opaque = 0;
2502         notToKeep->elements[0]->supplementalId = 0;
2503         notToKeep->elements[0]->correlationInfo = 0;
2504         notToKeep->elements[0]->record = record_this;
2505     }
2506     else
2507     {
2508         Z_IUOriginPartToKeep *toKeep;
2509         Z_IUSuppliedRecords *notToKeep;
2510
2511         r->which = Z_External_update;
2512         r->u.update = (Z_IUUpdate *) odr_malloc(out, sizeof(*r->u.update));
2513         r->u.update->which = Z_IUUpdate_esRequest;
2514         r->u.update->u.esRequest = (Z_IUUpdateEsRequest *)
2515             odr_malloc(out, sizeof(*r->u.update->u.esRequest));
2516         toKeep = r->u.update->u.esRequest->toKeep = (Z_IUOriginPartToKeep *)
2517             odr_malloc(out, sizeof(*r->u.update->u.esRequest->toKeep));
2518
2519         toKeep->databaseName = databaseNames[0];
2520         toKeep->schema = 0;
2521         if (record_schema)
2522         {
2523             toKeep->schema = yaz_string_to_oid_odr(yaz_oid_std(),
2524                                                    CLASS_SCHEMA,
2525                                                    record_schema, out);
2526         }
2527         toKeep->elementSetName = 0;
2528         toKeep->actionQualifier = 0;
2529         toKeep->action = odr_intdup(out, action_no);
2530
2531         notToKeep = r->u.update->u.esRequest->notToKeep = (Z_IUSuppliedRecords *)
2532             odr_malloc(out, sizeof(*r->u.update->u.esRequest->notToKeep));
2533         notToKeep->num = 1;
2534         notToKeep->elements = (Z_IUSuppliedRecords_elem **)
2535             odr_malloc(out, sizeof(*notToKeep->elements));
2536         notToKeep->elements[0] = (Z_IUSuppliedRecords_elem *)
2537             odr_malloc(out, sizeof(**notToKeep->elements));
2538         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
2539         if (recid)
2540         {
2541             notToKeep->elements[0]->u.opaque = (Odr_oct *)
2542                 odr_malloc(out, sizeof(Odr_oct));
2543             notToKeep->elements[0]->u.opaque->buf = (unsigned char *) recid;
2544             notToKeep->elements[0]->u.opaque->size = strlen(recid);
2545             notToKeep->elements[0]->u.opaque->len = strlen(recid);
2546         }
2547         else
2548             notToKeep->elements[0]->u.opaque = 0;
2549         notToKeep->elements[0]->supplementalId = 0;
2550         notToKeep->elements[0]->correlationInfo = 0;
2551         notToKeep->elements[0]->record = record_this;
2552     }
2553
2554     send_apdu(apdu);
2555
2556     return 2;
2557 }
2558
2559 static int cmd_xmles(const char *arg)
2560 {
2561     if (only_z3950())
2562         return 1;
2563     else
2564     {
2565         char *asn_buf = 0;
2566         int noread = 0;
2567         Odr_oid *oid;
2568         char oid_str[51];
2569         Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest);
2570         Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2571
2572
2573         Z_External *ext = (Z_External *) odr_malloc(out, sizeof(*ext));
2574
2575         req->referenceId = set_refid (out);
2576         req->taskSpecificParameters = ext;
2577         ext->indirect_reference = 0;
2578         ext->descriptor = 0;
2579         ext->which = Z_External_octet;
2580         ext->u.single_ASN1_type = (Odr_oct *) odr_malloc(out, sizeof(Odr_oct));
2581         sscanf(arg, "%50s%n", oid_str, &noread);
2582         if (noread == 0)
2583         {
2584             printf("Missing OID for xmles\n");
2585             return 0;
2586         }
2587         arg += noread;
2588         if (parse_cmd_doc(&arg, out, &asn_buf,
2589                           &ext->u.single_ASN1_type->len) == 0)
2590             return 0;
2591
2592         ext->u.single_ASN1_type->buf = (unsigned char *) asn_buf;
2593
2594         oid = yaz_string_to_oid_odr(yaz_oid_std(),
2595                                     CLASS_EXTSERV, oid_str, out);
2596         if (!oid)
2597         {
2598             printf("Bad OID: %s\n", oid_str);
2599             return 0;
2600         }
2601
2602         req->packageType = oid;
2603
2604         ext->direct_reference = oid;
2605
2606         send_apdu(apdu);
2607
2608         return 2;
2609     }
2610 }
2611
2612 static int cmd_itemorder(const char *arg)
2613 {
2614     char type[12];
2615     int itemno;
2616     char *xml_buf = 0;
2617     int xml_len = 0;
2618     int no_read = 0;
2619
2620     if (only_z3950())
2621         return 1;
2622     if (sscanf(arg, "%10s %d%n", type, &itemno, &no_read) < 2)
2623         return 0;
2624     arg += no_read;
2625     parse_cmd_doc(&arg, out, &xml_buf, &xml_len);
2626
2627     fflush(stdout);
2628     send_Z3950_itemorder(type, itemno, xml_buf, xml_len);
2629     return 2;
2630 }
2631
2632 static void show_opt(const char *arg, void *clientData)
2633 {
2634     printf("%s ", arg);
2635 }
2636
2637 static int cmd_zversion(const char *arg)
2638 {
2639     if (*arg && arg)
2640         z3950_version = atoi(arg);
2641     else
2642         printf("version is %d\n", z3950_version);
2643     return 0;
2644 }
2645
2646 static int cmd_options(const char *arg)
2647 {
2648     if (*arg)
2649     {
2650         int r;
2651         int pos;
2652         r = yaz_init_opt_encode(&z3950_options, arg, &pos);
2653         if (r == -1)
2654             printf("Unknown option(s) near %s\n", arg+pos);
2655     }
2656     else
2657     {
2658         yaz_init_opt_decode(&z3950_options, show_opt, 0);
2659         printf("\n");
2660     }
2661     return 0;
2662 }
2663
2664 static int cmd_explain(const char *arg)
2665 {
2666     if (protocol != PROTO_HTTP)
2667         return 0;
2668 #if YAZ_HAVE_XML2
2669     if (!conn)
2670         session_connect(cur_host);
2671     if (conn)
2672     {
2673         Z_SRW_PDU *sr = 0;
2674
2675         setno = 1;
2676
2677         /* save this for later .. when fetching individual records */
2678         sr = yaz_srw_get(out, Z_SRW_explain_request);
2679         if (recordsyntax_size == 1
2680             && !yaz_matchstr(recordsyntax_list[0], "xml"))
2681             sr->u.explain_request->recordPacking = "xml";
2682         send_srw(sr);
2683         return 2;
2684     }
2685 #endif
2686     return 0;
2687 }
2688
2689 static int cmd_init(const char *arg)
2690 {
2691     if (*arg)
2692     {
2693         strncpy(cur_host, arg, sizeof(cur_host)-1);
2694         cur_host[sizeof(cur_host)-1] = 0;
2695     }
2696     if (only_z3950())
2697         return 1;
2698     send_Z3950_initRequest(cur_host);
2699     return 2;
2700 }
2701
2702 static Z_GDU *get_HTTP_Request_url(ODR odr, const char *url)
2703 {
2704     Z_GDU *p = z_get_HTTP_Request(odr);
2705     const char *host = url;
2706     const char *cp0 = strstr(host, "://");
2707     const char *cp1 = 0;
2708     if (cp0)
2709         cp0 = cp0+3;
2710     else
2711         cp0 = host;
2712
2713     cp1 = strchr(cp0, '/');
2714     if (!cp1)
2715         cp1 = cp0 + strlen(cp0);
2716
2717     if (cp0 && cp1)
2718     {
2719         char *h = (char*) odr_malloc(odr, cp1 - cp0 + 1);
2720         memcpy (h, cp0, cp1 - cp0);
2721         h[cp1-cp0] = '\0';
2722         z_HTTP_header_add(odr, &p->u.HTTP_Request->headers, "Host", h);
2723     }
2724     p->u.HTTP_Request->path = odr_strdup(odr, *cp1 ? cp1 : "/");
2725     return p;
2726 }
2727
2728 static WRBUF get_url(const char *uri, WRBUF username, WRBUF password,
2729                      int *code, int show_headers)
2730 {
2731     WRBUF result = 0;
2732     ODR out = odr_createmem(ODR_ENCODE);
2733     ODR in = odr_createmem(ODR_DECODE);
2734     Z_GDU *gdu = get_HTTP_Request_url(out, uri);
2735
2736     gdu->u.HTTP_Request->method = odr_strdup(out, "GET");
2737     if (username && password)
2738     {
2739         z_HTTP_header_add_basic_auth(out, &gdu->u.HTTP_Request->headers,
2740                                      wrbuf_cstr(username),
2741                                      wrbuf_cstr(password));
2742     }
2743     z_HTTP_header_add(out, &gdu->u.HTTP_Request->headers, "Accept",
2744                       "text/xml");
2745     if (!z_GDU(out, &gdu, 0, 0))
2746     {
2747         yaz_log(YLOG_WARN, "Can not encode HTTP request URL:%s", uri);
2748     }
2749     else
2750     {
2751         void *add;
2752         COMSTACK conn = cs_create_host(uri, 1, &add);
2753         if (cs_connect(conn, add) < 0)
2754             yaz_log(YLOG_WARN, "Can not connect to URL:%s", uri);
2755         else
2756         {
2757             int len;
2758             char *buf = odr_getbuf(out, &len, 0);
2759
2760             if (cs_put(conn, buf, len) < 0)
2761                 yaz_log(YLOG_WARN, "cs_put failed URL:%s", uri);
2762             else
2763             {
2764                 char *netbuffer = 0;
2765                 int netlen = 0;
2766                 int res = cs_get(conn, &netbuffer, &netlen);
2767                 if (res <= 0)
2768                 {
2769                     yaz_log(YLOG_WARN, "cs_get failed URL:%s", uri);
2770                 }
2771                 else
2772                 {
2773                     Z_GDU *gdu;
2774                     odr_setbuf(in, netbuffer, res, 0);
2775                     if (!z_GDU(in, &gdu, 0, 0)
2776                         || gdu->which != Z_GDU_HTTP_Response)
2777                     {
2778                         yaz_log(YLOG_WARN, "decode failed URL: %s", uri);
2779                     }
2780                     else
2781                     {
2782                         Z_HTTP_Response *res = gdu->u.HTTP_Response;
2783                         struct Z_HTTP_Header *h;
2784                         result = wrbuf_alloc();
2785                         if (show_headers)
2786                         {
2787
2788                             wrbuf_printf(result, "HTTP %d\n", res->code);
2789                             for (h = res->headers; h; h = h->next)
2790                                 wrbuf_printf(result, "%s: %s\n",
2791                                              h->name, h->value);
2792                         }
2793                         *code = res->code;
2794                         wrbuf_write(result, res->content_buf, res->content_len);
2795                     }
2796                 }
2797                 xfree(netbuffer);
2798             }
2799             cs_close(conn);
2800         }
2801     }
2802     odr_destroy(out);
2803     odr_destroy(in);
2804     return result;
2805 }
2806
2807
2808 static int cmd_url(const char *arg)
2809 {
2810     int code = 0;
2811     WRBUF res = get_url(arg, 0, 0, &code, 1);
2812     if (res)
2813     {
2814         if (wrbuf_len(res) > 1200)
2815         {
2816             fwrite(wrbuf_buf(res), 1, 1200, stdout);
2817             printf(".. out of %lld\n", (long long) wrbuf_len(res));
2818         }
2819         else
2820             puts(wrbuf_cstr(res));
2821         wrbuf_destroy(res);
2822     }
2823     return 0;
2824 }
2825
2826 static int cmd_sru(const char *arg)
2827 {
2828     if (!*arg)
2829     {
2830         printf("SRU method is: %s\n", sru_method);
2831         printf("SRU version is: %s\n", sru_version);
2832     }
2833     else
2834     {
2835         int r = sscanf(arg, "%9s %9s", sru_method, sru_version);
2836         if (r >= 1)
2837         {
2838             if (!yaz_matchstr(sru_method, "post"))
2839                 ;
2840             else if (!yaz_matchstr(sru_method, "get"))
2841                 ;
2842             else if (!yaz_matchstr(sru_method, "soap"))
2843                 ;
2844             else if (!yaz_matchstr(sru_method, "solr"))
2845                 ;
2846             else
2847             {
2848                 strcpy(sru_method, "soap");
2849                 printf("Unknown SRU method: %s\n", arg);
2850                 printf("Specify one of POST, GET, SOAP, SOLR\n");
2851             }
2852         }
2853     }
2854     return 0;
2855 }
2856
2857 static int cmd_find(const char *arg)
2858 {
2859     if (!*arg)
2860     {
2861         printf("Find what?\n");
2862         return 0;
2863     }
2864     if (protocol == PROTO_HTTP)
2865     {
2866 #if YAZ_HAVE_XML2
2867         if (!conn)
2868             session_connect(cur_host);
2869         if (!conn)
2870             return 0;
2871         if (!send_SRW_searchRequest(arg))
2872             return 0;
2873 #else
2874         return 0;
2875 #endif
2876     }
2877     else
2878     {
2879         if (*cur_host && auto_reconnect)
2880         {
2881             int i = 0;
2882             for (;;)
2883             {
2884                 if (conn)
2885                 {
2886                     if (!send_Z3950_searchRequest(arg))
2887                         return 0;
2888                     wait_and_handle_response(0);
2889                     if (conn)
2890                         break;
2891                 }
2892                 if (++i == 2)
2893                 {
2894                     printf("Unable to reconnect\n");
2895                     break;
2896                 }
2897                 session_connect(cur_host);
2898                 wait_and_handle_response(0);
2899             }
2900             return 0;
2901         }
2902         else if (conn)
2903         {
2904             if (!send_Z3950_searchRequest(arg))
2905                 return 0;
2906         }
2907         else
2908         {
2909             printf("Not connected yet\n");
2910             return 0;
2911         }
2912     }
2913     return 2;
2914 }
2915
2916 static int cmd_facets(const char *arg)
2917 {
2918     if (!facet_odr)
2919         facet_odr = odr_createmem(ODR_ENCODE);
2920     odr_reset(facet_odr);
2921
2922     if (!*arg)
2923     {
2924         facet_list = 0;
2925         printf("Facets cleared.\n");
2926         return 0;
2927     }
2928     facet_list = yaz_pqf_parse_facet_list(facet_odr, arg);
2929     if (!facet_list)
2930     {
2931         printf("Invalid facet list: %s", arg);
2932         return 0;
2933     }
2934     return 1;
2935 }
2936
2937 static int cmd_delete(const char *arg)
2938 {
2939     if (only_z3950())
2940         return 0;
2941     if (!send_Z3950_deleteResultSetRequest(arg))
2942         return 0;
2943     return 2;
2944 }
2945
2946 static int cmd_ssub(const char *arg)
2947 {
2948     if (!(smallSetUpperBound = atoi(arg)))
2949         return 0;
2950     return 1;
2951 }
2952
2953 static int cmd_lslb(const char *arg)
2954 {
2955     if (!(largeSetLowerBound = atoi(arg)))
2956         return 0;
2957     return 1;
2958 }
2959
2960 static int cmd_mspn(const char *arg)
2961 {
2962     if (!(mediumSetPresentNumber = atoi(arg)))
2963         return 0;
2964     return 1;
2965 }
2966
2967 static int cmd_status(const char *arg)
2968 {
2969     printf("smallSetUpperBound: %d\n", smallSetUpperBound);
2970     printf("largeSetLowerBound: %d\n", largeSetLowerBound);
2971     printf("mediumSetPresentNumber: %d\n", mediumSetPresentNumber);
2972     return 1;
2973 }
2974
2975 static int cmd_setnames(const char *arg)
2976 {
2977     if (*arg == '1')         /* enable ? */
2978         setnumber = 0;
2979     else if (*arg == '0')    /* disable ? */
2980         setnumber = -1;
2981     else if (setnumber < 0)  /* no args, toggle .. */
2982         setnumber = 0;
2983     else
2984         setnumber = -1;
2985
2986     if (setnumber >= 0)
2987         printf("Set numbering enabled.\n");
2988     else
2989         printf("Set numbering disabled.\n");
2990     return 1;
2991 }
2992
2993 /* PRESENT SERVICE ----------------------------- */
2994
2995 size_t check_token(const char *haystack, const char *token)
2996 {
2997     size_t len = strlen(token);
2998     size_t extra;
2999     if (strncmp(haystack, token, len))
3000         return 0;
3001     for (extra = 0; haystack[extra + len] != '\0'; extra++)
3002         if (!strchr(" \r\n\t", haystack[extra + len]))
3003         {
3004             if (extra)
3005                 break;
3006             else
3007                 return 0;  /* no whitespace after token */
3008         }
3009     return extra + len;
3010 }
3011
3012 static int parse_show_args(const char *arg_c, char *setstring,
3013                            Odr_int *start, Odr_int *number)
3014 {
3015     char *end_ptr;
3016     Odr_int start_position;
3017     size_t token_len;
3018
3019     if (setnumber >= 0)
3020         sprintf(setstring, "%d", setnumber);
3021     else
3022         *setstring = '\0';
3023
3024     token_len = check_token(arg_c, "format");
3025     if (token_len)
3026     {
3027         pretty_xml = 1;
3028         arg_c += token_len;
3029     }
3030     else
3031         pretty_xml = 0;
3032
3033     token_len = check_token(arg_c, "all");
3034     if (token_len)
3035     {
3036         *number = last_hit_count;
3037         *start = 1;
3038         return 1;
3039     }
3040     start_position = odr_strtol(arg_c, &end_ptr, 10);
3041     if (end_ptr == arg_c)
3042         return 1;
3043     *start = start_position;
3044     if (*end_ptr == '\0')
3045         return 1;
3046     while (yaz_isspace(*end_ptr))
3047         end_ptr++;
3048     if (*end_ptr != '+')
3049     {
3050         printf("Bad show arg: expected +. Got %s\n", end_ptr);
3051         return 0;
3052     }
3053     end_ptr++;
3054     arg_c = end_ptr;
3055     *number = odr_strtol(arg_c, &end_ptr, 10);
3056     if (end_ptr == arg_c)
3057     {
3058         printf("Bad show arg: expected number after +\n");
3059         return 0;
3060     }
3061     if (*end_ptr == '\0')
3062         return 1;
3063     while (yaz_isspace(*end_ptr))
3064         end_ptr++;
3065     if (*end_ptr != '+')
3066     {
3067         printf("Bad show arg: + expected. Got %s\n", end_ptr);
3068         return 0;
3069     }
3070     strcpy(setstring, end_ptr+1);
3071     return 1;
3072 }
3073
3074 static int send_Z3950_presentRequest(const char *arg)
3075 {
3076     Z_APDU *apdu = zget_APDU(out, Z_APDU_presentRequest);
3077     Z_PresentRequest *req = apdu->u.presentRequest;
3078     Z_RecordComposition compo;
3079     Odr_int nos = 1;
3080     char setstring[100];
3081
3082     req->referenceId = set_refid(out);
3083
3084     if (!parse_show_args(arg, setstring, &setno, &nos))
3085         return 0;
3086     if (*setstring)
3087         req->resultSetId = setstring;
3088
3089     req->resultSetStartPoint = &setno;
3090     req->numberOfRecordsRequested = &nos;
3091
3092     if (recordsyntax_size)
3093         req->preferredRecordSyntax =
3094             yaz_string_to_oid_odr(yaz_oid_std(),
3095                                   CLASS_RECSYN, recordsyntax_list[0], out);
3096
3097     if (record_schema || recordsyntax_size >= 2)
3098     {
3099         req->recordComposition = &compo;
3100         compo.which = Z_RecordComp_complex;
3101         compo.u.complex = (Z_CompSpec *)
3102             odr_malloc(out, sizeof(*compo.u.complex));
3103         compo.u.complex->selectAlternativeSyntax = (bool_t *)
3104             odr_malloc(out, sizeof(bool_t));
3105         *compo.u.complex->selectAlternativeSyntax = 0;
3106
3107         compo.u.complex->generic = (Z_Specification *)
3108             odr_malloc(out, sizeof(*compo.u.complex->generic));
3109
3110         compo.u.complex->generic->which = Z_Schema_oid;
3111         if (!record_schema)
3112             compo.u.complex->generic->schema.oid = 0;
3113         else
3114         {
3115             compo.u.complex->generic->schema.oid =
3116                 yaz_string_to_oid_odr(yaz_oid_std(),
3117                                       CLASS_SCHEMA, record_schema, out);
3118
3119             if (!compo.u.complex->generic->schema.oid)
3120             {
3121                 /* OID wasn't a schema! Try record syntax instead. */
3122                 compo.u.complex->generic->schema.oid = (Odr_oid *)
3123                     yaz_string_to_oid_odr(yaz_oid_std(),
3124                                           CLASS_RECSYN, record_schema, out);
3125             }
3126         }
3127         if (!elementSetNames)
3128             compo.u.complex->generic->elementSpec = 0;
3129         else
3130         {
3131             compo.u.complex->generic->elementSpec = (Z_ElementSpec *)
3132                 odr_malloc(out, sizeof(Z_ElementSpec));
3133             compo.u.complex->generic->elementSpec->which =
3134                 Z_ElementSpec_elementSetName;
3135             compo.u.complex->generic->elementSpec->u.elementSetName =
3136                 elementSetNames->u.generic;
3137         }
3138         compo.u.complex->num_dbSpecific = 0;
3139         compo.u.complex->dbSpecific = 0;
3140
3141         compo.u.complex->num_recordSyntax = 0;
3142         compo.u.complex->recordSyntax = 0;
3143         if (recordsyntax_size >= 2)
3144         {
3145             int i;
3146             compo.u.complex->num_recordSyntax = recordsyntax_size;
3147             compo.u.complex->recordSyntax = (Odr_oid **)
3148                 odr_malloc(out, recordsyntax_size * sizeof(Odr_oid*));
3149             for (i = 0; i < recordsyntax_size; i++)
3150             compo.u.complex->recordSyntax[i] =
3151                 yaz_string_to_oid_odr(yaz_oid_std(),
3152                                       CLASS_RECSYN, recordsyntax_list[i], out);
3153         }
3154     }
3155     else if (elementSetNames)
3156     {
3157         req->recordComposition = &compo;
3158         compo.which = Z_RecordComp_simple;
3159         compo.u.simple = elementSetNames;
3160     }
3161     send_apdu(apdu);
3162     printf("Sent presentRequest (" ODR_INT_PRINTF "+" ODR_INT_PRINTF ").\n",
3163            setno, nos);
3164     return 2;
3165 }
3166
3167 #if YAZ_HAVE_XML2
3168 static int send_SRW_presentRequest(const char *arg)
3169 {
3170     char setstring[100];
3171     Odr_int nos = 1;
3172     Z_SRW_PDU *sr = srw_sr;
3173
3174     if (!sr)
3175         return 0;
3176     if (!parse_show_args(arg, setstring, &setno, &nos))
3177         return 0;
3178     sr->u.request->startRecord = odr_intdup(out, setno);
3179     sru_maximumRecords = nos;
3180     sr->u.request->maximumRecords = odr_intdup(out, nos);
3181     if (record_schema)
3182         sr->u.request->recordSchema = record_schema;
3183     if (recordsyntax_size == 1 && !yaz_matchstr(recordsyntax_list[0], "xml"))
3184         sr->u.request->recordPacking = "xml";
3185     return send_srw(sr);
3186 }
3187 #endif
3188
3189 static void close_session(void)
3190 {
3191     if (conn)
3192         cs_close(conn);
3193     conn = 0;
3194     sent_close = 0;
3195     odr_reset(out);
3196     odr_reset(in);
3197     odr_reset(print);
3198     last_hit_count = 0;
3199 }
3200
3201 static void process_Z3950_close(Z_Close *req)
3202 {
3203     Z_APDU *apdu = zget_APDU(out, Z_APDU_close);
3204     Z_Close *res = apdu->u.close;
3205
3206     static char *reasons[] =
3207     {
3208         "finished",
3209         "shutdown",
3210         "system problem",
3211         "cost limit reached",
3212         "resources",
3213         "security violation",
3214         "protocolError",
3215         "lack of activity",
3216         "peer abort",
3217         "unspecified"
3218     };
3219
3220     printf("Reason: %s, message: %s\n", reasons[*req->closeReason],
3221         req->diagnosticInformation ? req->diagnosticInformation : "NULL");
3222     if (sent_close)
3223         close_session();
3224     else
3225     {
3226         *res->closeReason = Z_Close_finished;
3227         send_apdu(apdu);
3228         printf("Sent response.\n");
3229         sent_close = 1;
3230     }
3231 }
3232
3233 static int cmd_show(const char *arg)
3234 {
3235     if (protocol == PROTO_HTTP)
3236     {
3237 #if YAZ_HAVE_XML2
3238         if (!conn)
3239             session_connect(cur_host);
3240         if (!conn)
3241             return 0;
3242         if (!send_SRW_presentRequest(arg))
3243             return 0;
3244 #else
3245         return 0;
3246 #endif
3247     }
3248     else
3249     {
3250         if (!conn)
3251         {
3252             printf("Not connected yet\n");
3253             return 0;
3254         }
3255         if (!send_Z3950_presentRequest(arg))
3256             return 0;
3257     }
3258     return 2;
3259 }
3260
3261 static void exit_client(int code)
3262 {
3263     file_history_save(file_history);
3264     file_history_destroy(&file_history);
3265     nmem_destroy(nmem_auth);
3266     exit(code);
3267 }
3268
3269 static int cmd_quit(const char *arg)
3270 {
3271     printf("See you later, alligator.\n");
3272     xmalloc_trav("");
3273     exit_client(0);
3274     return 0;
3275 }
3276
3277 static int cmd_cancel(const char *arg)
3278 {
3279     if (only_z3950())
3280         return 0;
3281     else
3282     {
3283         Z_APDU *apdu = zget_APDU(out, Z_APDU_triggerResourceControlRequest);
3284         Z_TriggerResourceControlRequest *req =
3285             apdu->u.triggerResourceControlRequest;
3286         bool_t rfalse = 0;
3287         char command[16];
3288
3289         *command = '\0';
3290         sscanf(arg, "%15s", command);
3291
3292         if (only_z3950())
3293             return 0;
3294         if (session_initResponse &&
3295             !ODR_MASK_GET(session_initResponse->options,
3296                           Z_Options_triggerResourceCtrl))
3297         {
3298             printf("Target doesn't support cancel (trigger resource ctrl)\n");
3299             return 0;
3300         }
3301         *req->requestedAction = Z_TriggerResourceControlRequest_cancel;
3302         req->resultSetWanted = &rfalse;
3303         req->referenceId = set_refid(out);
3304
3305         send_apdu(apdu);
3306         printf("Sent cancel request\n");
3307         if (!strcmp(command, "wait"))
3308             return 2;
3309         return 1;
3310     }
3311 }
3312
3313 static int cmd_cancel_find(const char *arg)
3314 {
3315     int fres;
3316     if (only_z3950())
3317         return 0;
3318     fres = cmd_find(arg);
3319     if (fres > 0)
3320     {
3321         return cmd_cancel("");
3322     };
3323     return fres;
3324 }
3325
3326 static int send_Z3950_scanrequest(const char *set,  const char *query,
3327                                   Odr_int *pos, Odr_int num, const char *term)
3328 {
3329     Z_APDU *apdu = zget_APDU(out, Z_APDU_scanRequest);
3330     Z_ScanRequest *req = apdu->u.scanRequest;
3331
3332     if (only_z3950())
3333         return 0;
3334     printf("query: %s\n", query);
3335     if (queryType == QueryType_CCL2RPN)
3336     {
3337         int error, pos;
3338         struct ccl_rpn_node *rpn;
3339
3340         rpn = ccl_find_str(bibset,  query, &error, &pos);
3341         if (error)
3342         {
3343             printf("CCL ERROR: %s\n", ccl_err_msg(error));
3344             return -1;
3345         }
3346         req->attributeSet =
3347             yaz_string_to_oid_odr(yaz_oid_std(),
3348                                   CLASS_ATTSET, "Bib-1", out);
3349         if (!(req->termListAndStartPoint = ccl_scan_query(out, rpn)))
3350         {
3351             printf("Couldn't convert CCL to Scan term\n");
3352             return -1;
3353         }
3354         ccl_rpn_delete(rpn);
3355     }
3356     else
3357     {
3358         YAZ_PQF_Parser pqf_parser = yaz_pqf_create();
3359
3360
3361         if (!(req->termListAndStartPoint =
3362               yaz_pqf_scan(pqf_parser, out, &req->attributeSet, query)))
3363         {
3364             const char *pqf_msg;
3365             size_t off;
3366             int code = yaz_pqf_error(pqf_parser, &pqf_msg, &off);
3367             int ioff = off;
3368             printf("%*s^\n", ioff+7, "");
3369             printf("Prefix query error: %s (code %d)\n", pqf_msg, code);
3370             yaz_pqf_destroy(pqf_parser);
3371             return -1;
3372         }
3373         yaz_pqf_destroy(pqf_parser);
3374     }
3375     if (queryCharset && outputCharset)
3376     {
3377         yaz_iconv_t cd = yaz_iconv_open(queryCharset, outputCharset);
3378         if (!cd)
3379         {
3380             printf("Conversion from %s to %s unsupported\n",
3381                    outputCharset, queryCharset);
3382             return -1;
3383         }
3384         yaz_query_charset_convert_apt(req->termListAndStartPoint, out, cd);
3385         yaz_iconv_close(cd);
3386     }
3387     if (term && *term)
3388     {
3389         if (req->termListAndStartPoint->term &&
3390             req->termListAndStartPoint->term->which == Z_Term_general &&
3391             req->termListAndStartPoint->term->u.general)
3392         {
3393             req->termListAndStartPoint->term->u.general->buf =
3394                 (unsigned char *) odr_strdup(out, term);
3395             req->termListAndStartPoint->term->u.general->len =
3396                 req->termListAndStartPoint->term->u.general->size =
3397                 strlen(term);
3398         }
3399     }
3400     req->referenceId = set_refid(out);
3401     req->num_databaseNames = num_databaseNames;
3402     req->databaseNames = databaseNames;
3403     req->numberOfTermsRequested = &num;
3404     req->preferredPositionInResponse = pos;
3405     req->stepSize = odr_intdup(out, scan_stepSize);
3406
3407     if (set)
3408         yaz_oi_set_string_oid(&req->otherInfo, out,
3409                               yaz_oid_userinfo_scan_set, 1, set);
3410
3411     send_apdu(apdu);
3412     return 2;
3413 }
3414
3415 static int send_sortrequest(const char *arg, int newset)
3416 {
3417     Z_APDU *apdu = zget_APDU(out, Z_APDU_sortRequest);
3418     Z_SortRequest *req = apdu->u.sortRequest;
3419     Z_SortKeySpecList *sksl = (Z_SortKeySpecList *)
3420         odr_malloc(out, sizeof(*sksl));
3421     char setstring[32];
3422
3423     if (only_z3950())
3424         return 0;
3425     if (setnumber >= 0)
3426         sprintf(setstring, "%d", setnumber);
3427     else
3428         sprintf(setstring, "default");
3429
3430     req->referenceId = set_refid(out);
3431
3432     req->num_inputResultSetNames = 1;
3433     req->inputResultSetNames = (Z_InternationalString **)
3434         odr_malloc(out, sizeof(*req->inputResultSetNames));
3435     req->inputResultSetNames[0] = odr_strdup(out, setstring);
3436
3437     if (newset && setnumber >= 0)
3438         sprintf(setstring, "%d", ++setnumber);
3439
3440     req->sortedResultSetName = odr_strdup(out, setstring);
3441
3442     req->sortSequence = yaz_sort_spec(out, arg);
3443     if (!req->sortSequence)
3444     {
3445         printf("Missing sort specifications\n");
3446         return -1;
3447     }
3448     send_apdu(apdu);
3449     return 2;
3450 }
3451
3452 static void display_term_info(Z_TermInfo *t)
3453 {
3454     if (t->displayTerm)
3455         printf("%s", t->displayTerm);
3456     else if (t->term->which == Z_Term_general)
3457         printf("%.*s", t->term->u.general->len, t->term->u.general->buf);
3458     else
3459         printf("Term (not general)");
3460     if (t->term->which == Z_Term_general)
3461         sprintf(last_scan_line, "%.*s", t->term->u.general->len,
3462             t->term->u.general->buf);
3463
3464     if (t->globalOccurrences)
3465         printf(" (" ODR_INT_PRINTF ")\n", *t->globalOccurrences);
3466     else
3467         printf("\n");
3468 }
3469
3470 static void process_Z3950_scanResponse(Z_ScanResponse *res)
3471 {
3472     int i;
3473     Z_Entry **entries = NULL;
3474     int num_entries = 0;
3475
3476     printf("Received ScanResponse\n");
3477     print_refid(res->referenceId);
3478     printf(ODR_INT_PRINTF " entries", *res->numberOfEntriesReturned);
3479     if (res->positionOfTerm)
3480         printf(", position=" ODR_INT_PRINTF, *res->positionOfTerm);
3481     printf("\n");
3482     if (*res->scanStatus != Z_Scan_success)
3483         printf("Scan returned code " ODR_INT_PRINTF "\n", *res->scanStatus);
3484     if (!res->entries)
3485         return;
3486     if ((entries = res->entries->entries))
3487         num_entries = res->entries->num_entries;
3488     for (i = 0; i < num_entries; i++)
3489     {
3490         int pos_term = res->positionOfTerm ? *res->positionOfTerm : -1;
3491         if (entries[i]->which == Z_Entry_termInfo)
3492         {
3493             printf("%c ", i + 1 == pos_term ? '*' : ' ');
3494             display_term_info(entries[i]->u.termInfo);
3495         }
3496         else
3497             display_diagrecs(&entries[i]->u.surrogateDiagnostic, 1);
3498     }
3499     if (res->entries->nonsurrogateDiagnostics)
3500         display_diagrecs(res->entries->nonsurrogateDiagnostics,
3501                           res->entries->num_nonsurrogateDiagnostics);
3502 }
3503
3504 static void process_Z3950_sortResponse(Z_SortResponse *res)
3505 {
3506     printf("Received SortResponse: status=");
3507     switch (*res->sortStatus)
3508     {
3509     case Z_SortResponse_success:
3510         printf("success"); break;
3511     case Z_SortResponse_partial_1:
3512         printf("partial"); break;
3513     case Z_SortResponse_failure:
3514         printf("failure"); break;
3515     default:
3516         printf("unknown (" ODR_INT_PRINTF ")", *res->sortStatus);
3517     }
3518     printf("\n");
3519     print_refid (res->referenceId);
3520     if (res->diagnostics)
3521         display_diagrecs(res->diagnostics,
3522                          res->num_diagnostics);
3523 }
3524
3525 static void process_Z3950_deleteResultSetResponse(
3526     Z_DeleteResultSetResponse *res)
3527 {
3528     printf("Got deleteResultSetResponse status=" ODR_INT_PRINTF "\n",
3529            *res->deleteOperationStatus);
3530     if (res->deleteListStatuses)
3531     {
3532         int i;
3533         for (i = 0; i < res->deleteListStatuses->num; i++)
3534         {
3535             printf("%s status=" ODR_INT_PRINTF "\n",
3536                    res->deleteListStatuses->elements[i]->id,
3537                    *res->deleteListStatuses->elements[i]->status);
3538         }
3539     }
3540 }
3541
3542 static int cmd_sort_generic(const char *arg, int newset)
3543 {
3544     if (only_z3950())
3545         return 0;
3546     if (session_initResponse &&
3547         !ODR_MASK_GET(session_initResponse->options, Z_Options_sort))
3548     {
3549         printf("Target doesn't support sort\n");
3550         return 0;
3551     }
3552     if (*arg)
3553     {
3554         if (send_sortrequest(arg, newset) < 0)
3555             return 0;
3556         return 2;
3557     }
3558     return 0;
3559 }
3560
3561 static int cmd_sort(const char *arg)
3562 {
3563     return cmd_sort_generic(arg, 0);
3564 }
3565
3566 static int cmd_sort_newset(const char *arg)
3567 {
3568     return cmd_sort_generic(arg, 1);
3569 }
3570
3571 static int cmd_scanstep(const char *arg)
3572 {
3573     scan_stepSize = atoi(arg);
3574     return 0;
3575 }
3576
3577 static int cmd_scanpos(const char *arg)
3578 {
3579     if (!strcmp(arg, "none"))
3580         strcpy(scan_position, "none");
3581     else
3582     {
3583         int dummy;
3584         int r = sscanf(arg, "%d", &dummy);
3585         if (r == 1 && strlen(arg) < sizeof(scan_position)-1)
3586             strcpy(scan_position, arg);
3587         else
3588             printf("specify number of none for scanpos\n");
3589     }
3590     return 0;
3591 }
3592
3593 static int cmd_scansize(const char *arg)
3594 {
3595     int r = sscanf(arg, "%d", &scan_size);
3596     if (r == 0)
3597         scan_size = 20;
3598     return 0;
3599 }
3600
3601 static int cmd_scan_common(const char *set, const char *arg)
3602 {
3603     Odr_int pos, *pos_p = 0;
3604     const char *scan_term = 0;
3605     const char *scan_query = 0;
3606
3607     if (!*arg)
3608     {
3609         pos = 1;
3610         pos_p = &pos;
3611         scan_query = last_scan_query;
3612         scan_term = last_scan_line;
3613     }
3614     else
3615     {
3616         strcpy(last_scan_query, arg);
3617         scan_query = arg;
3618         if (strcmp(scan_position, "none"))
3619         {
3620             pos = odr_atoi(scan_position);
3621             pos_p = &pos;
3622         }
3623     }
3624
3625     if (protocol == PROTO_HTTP)
3626     {
3627 #if YAZ_HAVE_XML2
3628         if (!conn)
3629             session_connect(cur_host);
3630         if (!conn)
3631             return 0;
3632         if (send_SRW_scanRequest(scan_query, pos_p, scan_size) < 0)
3633             return 0;
3634         return 2;
3635 #else
3636         return 0;
3637 #endif
3638     }
3639     else
3640     {
3641         if (*cur_host && !conn && auto_reconnect)
3642         {
3643             session_connect(cur_host);
3644             wait_and_handle_response(0);
3645         }
3646         if (!conn)
3647             return 0;
3648         if (session_initResponse &&
3649             !ODR_MASK_GET(session_initResponse->options, Z_Options_scan))
3650         {
3651             printf("Target doesn't support scan\n");
3652             return 0;
3653         }
3654         if (send_Z3950_scanrequest(set, scan_query, pos_p,
3655                                    scan_size, scan_term) < 0)
3656             return 0;
3657         return 2;
3658     }
3659 }
3660
3661 static int cmd_scan(const char *arg)
3662 {
3663     return cmd_scan_common(0, arg);
3664 }
3665
3666 static int cmd_setscan(const char *arg)
3667 {
3668     char setstring[100];
3669     int nor;
3670     if (sscanf(arg, "%99s%n", setstring, &nor) < 1)
3671     {
3672         printf("missing set for setscan\n");
3673         return 0;
3674     }
3675     return cmd_scan_common(setstring, arg + nor);
3676 }
3677
3678 static int cmd_schema(const char *arg)
3679 {
3680     xfree(record_schema);
3681     record_schema = 0;
3682     if (arg && *arg)
3683         record_schema = xstrdup(arg);
3684     return 1;
3685 }
3686
3687 static int cmd_format(const char *arg)
3688 {
3689     const char *cp = arg;
3690     int nor;
3691     int idx = 0;
3692     int i;
3693     char form_str[41];
3694     if (!arg || !*arg)
3695     {
3696         printf("Usage: format <recordsyntax>\n");
3697         return 0;
3698     }
3699     while (sscanf(cp, "%40s%n", form_str, &nor) >= 1 && nor > 0
3700            && idx < RECORDSYNTAX_MAX)
3701     {
3702         if (strcmp(form_str, "none") &&
3703             !yaz_string_to_oid_odr(yaz_oid_std(), CLASS_RECSYN, form_str, out))
3704         {
3705             printf("Bad format: %s\n", form_str);
3706             return 0;
3707         }
3708         cp += nor;
3709     }
3710     for (i = 0; i < recordsyntax_size; i++)
3711     {
3712         xfree(recordsyntax_list[i]);
3713         recordsyntax_list[i] = 0;
3714     }
3715
3716     cp = arg;
3717     while (sscanf(cp, "%40s%n", form_str, &nor) >= 1 && nor > 0
3718            && idx < RECORDSYNTAX_MAX)
3719     {
3720         if (!strcmp(form_str, "none"))
3721             break;
3722         recordsyntax_list[idx] = xstrdup(form_str);
3723         cp += nor;
3724         idx++;
3725     }
3726     recordsyntax_size = idx;
3727     return 1;
3728 }
3729
3730 static int cmd_elements(const char *arg)
3731 {
3732     static Z_ElementSetNames esn;
3733     static char what[100];
3734
3735     if (!arg || !*arg)
3736     {
3737         elementSetNames = 0;
3738         return 1;
3739     }
3740     strcpy(what, arg);
3741     esn.which = Z_ElementSetNames_generic;
3742     esn.u.generic = what;
3743     elementSetNames = &esn;
3744     return 1;
3745 }
3746
3747 static int cmd_querytype(const char *arg)
3748 {
3749     if (!strcmp(arg, "ccl"))
3750         queryType = QueryType_CCL;
3751     else if (!strcmp(arg, "prefix") || !strcmp(arg, "rpn"))
3752         queryType = QueryType_Prefix;
3753     else if (!strcmp(arg, "ccl2rpn") || !strcmp(arg, "cclrpn"))
3754         queryType = QueryType_CCL2RPN;
3755     else if (!strcmp(arg, "cql"))
3756         queryType = QueryType_CQL;
3757     else if (!strcmp(arg, "cql2rpn") || !strcmp(arg, "cqlrpn"))
3758         queryType = QueryType_CQL2RPN;
3759     else
3760     {
3761         printf("Querytype must be one of:\n");
3762         printf(" prefix         - Prefix query\n");
3763         printf(" ccl            - CCL query\n");
3764         printf(" ccl2rpn        - CCL query converted to RPN\n");
3765         printf(" cql            - CQL\n");
3766         printf(" cql2rpn        - CQL query converted to RPN\n");
3767         return 0;
3768     }
3769     return 1;
3770 }
3771
3772 static int cmd_refid(const char *arg)
3773 {
3774     xfree(refid);
3775     refid = NULL;
3776     if (*arg)
3777         refid = xstrdup(arg);
3778     return 1;
3779 }
3780
3781 static int cmd_close(const char *arg)
3782 {
3783     Z_APDU *apdu;
3784     Z_Close *req;
3785     if (only_z3950())
3786         return 0;
3787     apdu = zget_APDU(out, Z_APDU_close);
3788     req = apdu->u.close;
3789     *req->closeReason = Z_Close_finished;
3790     send_apdu(apdu);
3791     printf("Sent close request.\n");
3792     sent_close = 1;
3793     return 2;
3794 }
3795
3796 int cmd_packagename(const char* arg)
3797 {
3798     xfree(esPackageName);
3799     esPackageName = NULL;
3800     if (*arg)
3801         esPackageName = xstrdup(arg);
3802     return 1;
3803 }
3804
3805 static int cmd_proxy(const char* arg)
3806 {
3807     xfree(yazProxy);
3808     yazProxy = 0;
3809     if (*arg)
3810         yazProxy = xstrdup(arg);
3811     return 1;
3812 }
3813
3814 static int cmd_marccharset(const char *arg)
3815 {
3816     char l1[30];
3817
3818     *l1 = 0;
3819     if (sscanf(arg, "%29s", l1) < 1)
3820     {
3821         printf("MARC character set is `%s'\n",
3822                marcCharset ? marcCharset: "none");
3823         return 1;
3824     }
3825     xfree(marcCharset);
3826     marcCharset = 0;
3827     if (strcmp(l1, "-") && strcmp(l1, "none"))
3828         marcCharset = xstrdup(l1);
3829     return 1;
3830 }
3831
3832 static int cmd_querycharset(const char *arg)
3833 {
3834     char l1[30];
3835
3836     *l1 = 0;
3837     if (sscanf(arg, "%29s", l1) < 1)
3838     {
3839         printf("Query character set is `%s'\n",
3840                queryCharset ? queryCharset: "none");
3841         return 1;
3842     }
3843     xfree(queryCharset);
3844     queryCharset = 0;
3845     if (strcmp(l1, "-") && strcmp(l1, "none"))
3846         queryCharset = xstrdup(l1);
3847     return 1;
3848 }
3849
3850 static int cmd_displaycharset(const char *arg)
3851 {
3852     char l1[30];
3853
3854     *l1 = 0;
3855     if (sscanf(arg, "%29s", l1) < 1)
3856     {
3857         printf("Display character set is `%s'\n",
3858                outputCharset ? outputCharset: "none");
3859     }
3860     else
3861     {
3862         xfree(outputCharset);
3863         outputCharset = 0;
3864         if (!strcmp(l1, "auto") && codeset)
3865         {
3866             if (codeset)
3867             {
3868                 printf("Display character set: %s\n", codeset);
3869                 outputCharset = xstrdup(codeset);
3870             }
3871             else
3872                 printf("No codeset found on this system\n");
3873         }
3874         else if (strcmp(l1, "-") && strcmp(l1, "none"))
3875             outputCharset = xstrdup(l1);
3876     }
3877     return 1;
3878 }
3879
3880 static int cmd_negcharset(const char *arg)
3881 {
3882     char l1[30];
3883
3884     *l1 = 0;
3885     if (sscanf(arg, "%29s %d %d", l1, &negotiationCharsetRecords,
3886                &negotiationCharsetVersion) < 1)
3887     {
3888         printf("Negotiation character set `%s'\n",
3889                negotiationCharset ? negotiationCharset: "none");
3890         if (negotiationCharset)
3891         {
3892             printf("Records in charset %s\n", negotiationCharsetRecords ?
3893                    "yes" : "no");
3894             printf("Charneg version %d\n", negotiationCharsetVersion);
3895         }
3896     }
3897     else
3898     {
3899         xfree(negotiationCharset);
3900         negotiationCharset = NULL;
3901         if (*l1 && strcmp(l1, "-") && strcmp(l1, "none"))
3902         {
3903             negotiationCharset = xstrdup(l1);
3904             printf("Character set negotiation : %s\n", negotiationCharset);
3905         }
3906     }
3907     return 1;
3908 }
3909
3910 static int cmd_charset(const char* arg)
3911 {
3912     char l1[30], l2[30], l3[30], l4[30];
3913
3914     *l1 = *l2 = *l3 = *l4 = '\0';
3915     if (sscanf(arg, "%29s %29s %29s %29s", l1, l2, l3, l4) < 1)
3916     {
3917         cmd_negcharset("");
3918         cmd_displaycharset("");
3919         cmd_marccharset("");
3920         cmd_querycharset("");
3921     }
3922     else
3923     {
3924         cmd_negcharset(l1);
3925         if (*l2)
3926             cmd_displaycharset(l2);
3927         if (*l3)
3928             cmd_marccharset(l3);
3929         if (*l4)
3930             cmd_querycharset(l4);
3931     }
3932     return 1;
3933 }
3934
3935 static int cmd_lang(const char* arg)
3936 {
3937     if (*arg == '\0')
3938     {
3939         printf("Current language is `%s'\n", yazLang ? yazLang : "none");
3940         return 1;
3941     }
3942     xfree(yazLang);
3943     yazLang = NULL;
3944     if (*arg)
3945         yazLang = xstrdup(arg);
3946     return 1;
3947 }
3948
3949 static int cmd_source(const char* arg, int echo )
3950 {
3951     /* first should open the file and read one line at a time.. */
3952     FILE* includeFile;
3953     char line[102400], *cp;
3954
3955     if (strlen(arg) < 1)
3956     {
3957         fprintf(stderr, "Error in source command use a filename\n");
3958         return -1;
3959     }
3960
3961     includeFile = fopen(arg, "r");
3962
3963     if (!includeFile)
3964     {
3965         fprintf(stderr, "Unable to open file %s for reading\n",arg);
3966         return -1;
3967     }
3968
3969     while (fgets(line, sizeof(line), includeFile))
3970     {
3971         if (strlen(line) < 2)
3972             continue;
3973         if (line[0] == '#')
3974             continue;
3975
3976         if ((cp = strrchr(line, '\n')))
3977             *cp = '\0';
3978
3979         if (echo)
3980             printf("processing line: %s\n", line);
3981         process_cmd_line(line);
3982     }
3983
3984     if (fclose(includeFile))
3985     {
3986         perror("unable to close include file");
3987         exit(1);
3988     }
3989     return 1;
3990 }
3991
3992 static int cmd_source_echo(const char* arg)
3993 {
3994     cmd_source(arg, 1);
3995     return 1;
3996 }
3997
3998 static int cmd_subshell(const char* args)
3999 {
4000     int ret = system(strlen(args) ? args : getenv("SHELL"));
4001     printf("\n");
4002     if (ret)
4003     {
4004         printf("Exit %d\n", ret);
4005     }
4006     return 1;
4007 }
4008
4009 static int cmd_set_berfile(const char *arg)
4010 {
4011     if (ber_file && ber_file != stdout && ber_file != stderr)
4012         fclose(ber_file);
4013     if (!strcmp(arg, ""))
4014         ber_file = 0;
4015     else if (!strcmp(arg, "-"))
4016         ber_file = stdout;
4017     else
4018         ber_file = fopen(arg, "a");
4019     return 1;
4020 }
4021
4022 static int cmd_set_apdufile(const char *arg)
4023 {
4024     if (apdu_file && apdu_file != stderr && apdu_file != stderr)
4025         fclose(apdu_file);
4026     if (!strcmp(arg, ""))
4027         apdu_file = 0;
4028     else if (!strcmp(arg, "-"))
4029         apdu_file = stderr;
4030     else
4031     {
4032         apdu_file = fopen(arg, "a");
4033         if (!apdu_file)
4034             perror("unable to open apdu log file");
4035     }
4036     if (apdu_file)
4037         odr_setprint(print, apdu_file);
4038     return 1;
4039 }
4040
4041 static int cmd_set_cclfile(const char* arg)
4042 {
4043     FILE *inf;
4044
4045     bibset = ccl_qual_mk();
4046     inf = fopen(arg, "r");
4047     if (!inf)
4048         perror("unable to open CCL file");
4049     else
4050     {
4051         ccl_qual_file(bibset, inf);
4052         fclose(inf);
4053     }
4054     strcpy(ccl_fields,arg);
4055     return 0;
4056 }
4057
4058 static int cmd_set_cqlfile(const char* arg)
4059 {
4060     cql_transform_t newcqltrans;
4061
4062     if ((newcqltrans = cql_transform_open_fname(arg)) == 0)
4063     {
4064         perror("unable to open CQL file");
4065         return 0;
4066     }
4067     if (cqltrans != 0)
4068         cql_transform_close(cqltrans);
4069
4070     cqltrans = newcqltrans;
4071     strcpy(cql_fields, arg);
4072     return 0;
4073 }
4074
4075 static int cmd_set_auto_reconnect(const char* arg)
4076 {
4077     if (strlen(arg)==0)
4078         auto_reconnect = ! auto_reconnect;
4079     else if (strcmp(arg,"on")==0)
4080         auto_reconnect = 1;
4081     else if (strcmp(arg,"off")==0)
4082         auto_reconnect = 0;
4083     else
4084     {
4085         printf("Error use on or off\n");
4086         return 1;
4087     }
4088
4089     if (auto_reconnect)
4090         printf("Set auto reconnect enabled.\n");
4091     else
4092         printf("Set auto reconnect disabled.\n");
4093
4094     return 0;
4095 }
4096
4097 static int cmd_set_auto_wait(const char* arg)
4098 {
4099     if (strlen(arg)==0)
4100         auto_wait = ! auto_wait;
4101     else if (strcmp(arg,"on")==0)
4102         auto_wait = 1;
4103     else if (strcmp(arg,"off")==0)
4104         auto_wait = 0;
4105     else
4106     {
4107         printf("Error use on or off\n");
4108         return 1;
4109     }
4110
4111     if (auto_wait)
4112         printf("Set auto wait enabled.\n");
4113     else
4114         printf("Set auto wait disabled.\n");
4115
4116     return 0;
4117 }
4118
4119 static int cmd_set_marcdump(const char* arg)
4120 {
4121     if (marc_file && marc_file != stderr)
4122     { /* don't close stdout*/
4123         fclose(marc_file);
4124     }
4125
4126     if (!strcmp(arg, ""))
4127         marc_file = 0;
4128     else if (!strcmp(arg, "-"))
4129         marc_file = stderr;
4130     else
4131     {
4132         marc_file = fopen(arg, "a");
4133         if (!marc_file)
4134             perror("unable to open marc log file");
4135     }
4136     return 1;
4137 }
4138
4139 static void marc_file_write(const char *buf, size_t sz)
4140 {
4141     if (marc_file)
4142     {
4143         if (fwrite(buf, 1, sz, marc_file) != sz)
4144         {
4145             perror("marcfile write");
4146         }
4147     }
4148 }
4149 /*
4150    this command takes 3 arge {name class oid}
4151 */
4152 static int cmd_register_oid(const char* args)
4153 {
4154     static struct {
4155         char* className;
4156         oid_class oclass;
4157     } oid_classes[] = {
4158         {"appctx",CLASS_APPCTX},
4159         {"absyn",CLASS_ABSYN},
4160         {"attset",CLASS_ATTSET},
4161         {"transyn",CLASS_TRANSYN},
4162         {"diagset",CLASS_DIAGSET},
4163         {"recsyn",CLASS_RECSYN},
4164         {"resform",CLASS_RESFORM},
4165         {"accform",CLASS_ACCFORM},
4166         {"extserv",CLASS_EXTSERV},
4167         {"userinfo",CLASS_USERINFO},
4168         {"elemspec",CLASS_ELEMSPEC},
4169         {"varset",CLASS_VARSET},
4170         {"schema",CLASS_SCHEMA},
4171         {"tagset",CLASS_TAGSET},
4172         {"general",CLASS_GENERAL},
4173         {0,(enum oid_class) 0}
4174     };
4175     char oname_str[101], oclass_str[101], oid_str[101];
4176     int i;
4177     oid_class oidclass = CLASS_GENERAL;
4178     Odr_oid oid[OID_SIZE];
4179
4180     if (sscanf(args, "%100[^ ] %100[^ ] %100s",
4181                 oname_str,oclass_str, oid_str) < 1)
4182     {
4183         printf("Error in register command \n");
4184         return 0;
4185     }
4186
4187     for (i = 0; oid_classes[i].className; i++)
4188     {
4189         if (!strcmp(oid_classes[i].className, oclass_str))
4190         {
4191             oidclass=oid_classes[i].oclass;
4192             break;
4193         }
4194     }
4195
4196     if (!(oid_classes[i].className))
4197     {
4198         printf("Unknown oid class %s\n",oclass_str);
4199         return 0;
4200     }
4201
4202     oid_dotstring_to_oid(oid_str, oid);
4203
4204     if (yaz_oid_add(yaz_oid_std(), oidclass, oname_str, oid))
4205     {
4206         printf("oid %s already exists, registration failed\n",
4207                oname_str);
4208     }
4209     return 1;
4210 }
4211
4212 static int cmd_push_command(const char* arg)
4213 {
4214 #if HAVE_READLINE_HISTORY_H
4215     if (strlen(arg) > 1)
4216         add_history(arg);
4217 #else
4218     fprintf(stderr,"Not compiled with the readline/history module\n");
4219 #endif
4220     return 1;
4221 }
4222
4223 void source_rc_file(const char *rc_file)
4224 {
4225     /*  If rc_file != NULL, source that. Else
4226         Look for .yazclientrc and read it if it exists.
4227         If it does not exist, read  $HOME/.yazclientrc instead */
4228     struct stat statbuf;
4229
4230     if (rc_file)
4231     {
4232         if (stat(rc_file, &statbuf) == 0)
4233             cmd_source(rc_file, 0);
4234         else
4235         {
4236             fprintf(stderr, "yaz_client: cannot source '%s'\n", rc_file);
4237             exit(1);
4238         }
4239     }
4240     else
4241     {
4242         char fname[1000];
4243         strcpy(fname, ".yazclientrc");
4244         if (stat(fname, &statbuf)==0)
4245         {
4246             cmd_source(fname, 0);
4247         }
4248         else
4249         {
4250             const char* homedir = getenv("HOME");
4251             if (homedir)
4252             {
4253                 sprintf(fname, "%.800s/%s", homedir, ".yazclientrc");
4254                 if (stat(fname, &statbuf)==0)
4255                     cmd_source(fname, 0);
4256             }
4257         }
4258     }
4259 }
4260
4261 static void add_to_readline_history(void *client_data, const char *line)
4262 {
4263 #if HAVE_READLINE_HISTORY_H
4264     if (strlen(line))
4265         add_history(line);
4266 #endif
4267 }
4268
4269 static void initialize(const char *rc_file)
4270 {
4271     FILE *inf;
4272     int i;
4273
4274     if (!(out = odr_createmem(ODR_ENCODE)) ||
4275         !(in = odr_createmem(ODR_DECODE)) ||
4276         !(print = odr_createmem(ODR_PRINT)))
4277     {
4278         fprintf(stderr, "failed to allocate ODR streams\n");
4279         exit(1);
4280     }
4281
4282     strcpy(scan_position, "1");
4283
4284     setvbuf(stdout, 0, _IONBF, 0);
4285     if (apdu_file)
4286         odr_setprint(print, apdu_file);
4287
4288     bibset = ccl_qual_mk();
4289     inf = fopen(ccl_fields, "r");
4290     if (inf)
4291     {
4292         ccl_qual_file(bibset, inf);
4293         fclose(inf);
4294     }
4295
4296     cqltrans = cql_transform_open_fname(cql_fields);
4297     /* If this fails, no problem: we detect cqltrans == 0 later */
4298
4299 #if HAVE_READLINE_READLINE_H
4300     rl_attempted_completion_function =
4301         (char **(*)(const char *, int, int)) readline_completer;
4302 #endif
4303     for (i = 0; i < maxOtherInfosSupported; ++i)
4304     {
4305         extraOtherInfos[i].oid[0] = -1;
4306         extraOtherInfos[i].value = 0;
4307     }
4308
4309     cmd_format("usmarc");
4310
4311     file_history = file_history_new();
4312
4313     source_rc_file(rc_file);
4314
4315     file_history_load(file_history);
4316     file_history_trav(file_history, 0, add_to_readline_history);
4317 }
4318
4319
4320 #if HAVE_GETTIMEOFDAY
4321 struct timeval tv_start;
4322 #endif
4323
4324 #if YAZ_HAVE_XML2
4325 static void handle_srw_record(Z_SRW_record *rec)
4326 {
4327     if (rec->recordPosition)
4328     {
4329         printf("pos=" ODR_INT_PRINTF, *rec->recordPosition);
4330         setno = *rec->recordPosition + 1;
4331     }
4332     if (rec->recordSchema)
4333         printf(" schema=%s", rec->recordSchema);
4334     printf("\n");
4335     if (rec->recordData_buf && rec->recordData_len)
4336     {
4337         print_xml_record(rec->recordData_buf, rec->recordData_len);
4338         marc_file_write(rec->recordData_buf, rec->recordData_len);
4339     }
4340     else
4341         printf("No data!");
4342     printf("\n");
4343 }
4344
4345 static void handle_srw_explain_response(Z_SRW_explainResponse *res)
4346 {
4347     handle_srw_record(&res->record);
4348 }
4349
4350 static void handle_srw_response(Z_SRW_searchRetrieveResponse *res)
4351 {
4352     int i;
4353
4354     printf("Received SRW SearchRetrieve Response\n");
4355
4356     for (i = 0; i<res->num_diagnostics; i++)
4357     {
4358         if (res->diagnostics[i].uri)
4359             printf("SRW diagnostic %s\n",
4360                     res->diagnostics[i].uri);
4361         else
4362             printf("SRW diagnostic missing or could not be decoded\n");
4363         if (res->diagnostics[i].message)
4364             printf("Message: %s\n", res->diagnostics[i].message);
4365         if (res->diagnostics[i].details)
4366             printf("Details: %s\n", res->diagnostics[i].details);
4367     }
4368     if (res->numberOfRecords)
4369         printf("Number of hits: " ODR_INT_PRINTF "\n", *res->numberOfRecords);
4370     if (res->facetList)
4371         display_facets(res->facetList);
4372     if (res->suggestions)
4373         printf("Suggestions:\n%s\n", res->suggestions);
4374     for (i = 0; i < res->num_records; i++)
4375     {
4376         if (i >= sru_maximumRecords)
4377         {
4378             printf("SRU server returns extra records. Skipping "
4379                    ODR_INT_PRINTF " records.\n",
4380                    res->num_records - sru_maximumRecords);
4381             break;
4382         }
4383         handle_srw_record(res->records + i);
4384     }
4385 }
4386
4387 static void handle_srw_scan_term(Z_SRW_scanTerm *term)
4388 {
4389     if (term->displayTerm)
4390         printf("%s:", term->displayTerm);
4391     else if (term->value)
4392         printf("%s:", term->value);
4393     else
4394         printf("No value:");
4395     if (term->numberOfRecords)
4396         printf(" " ODR_INT_PRINTF, *term->numberOfRecords);
4397     if (term->whereInList)
4398         printf(" %s", term->whereInList);
4399     if (term->value && term->displayTerm)
4400         printf(" %s", term->value);
4401
4402     strcpy(last_scan_line, term->value);
4403     printf("\n");
4404 }
4405
4406 static void handle_srw_scan_response(Z_SRW_scanResponse *res)
4407 {
4408     int i;
4409
4410     printf("Received SRW Scan Response\n");
4411
4412     for (i = 0; i<res->num_diagnostics; i++)
4413     {
4414         if (res->diagnostics[i].uri)
4415             printf("SRW diagnostic %s\n",
4416                     res->diagnostics[i].uri);
4417         else
4418             printf("SRW diagnostic missing or could not be decoded\n");
4419         if (res->diagnostics[i].message)
4420             printf("Message: %s\n", res->diagnostics[i].message);
4421         if (res->diagnostics[i].details)
4422             printf("Details: %s\n", res->diagnostics[i].details);
4423     }
4424     if (res->terms)
4425         for (i = 0; i<res->num_terms; i++)
4426             handle_srw_scan_term(res->terms + i);
4427 }
4428
4429 static void http_response(Z_HTTP_Response *hres)
4430 {
4431     int ret = -1;
4432     const char *connection_head = z_HTTP_header_lookup(hres->headers,
4433                                                        "Connection");
4434     if (hres->code != 200)
4435     {
4436         printf("HTTP Error Status=%d\n", hres->code);
4437     }
4438
4439     if (!yaz_srw_check_content_type(hres))
4440         printf("Content type does not appear to be XML\n");
4441     else
4442     {
4443         if (!yaz_matchstr(sru_method, "solr"))
4444         {
4445             Z_SRW_PDU *sr = 0;
4446             ODR o = odr_createmem(ODR_DECODE);
4447             ret = yaz_solr_decode_response(o, hres, &sr);
4448
4449             if (ret == 0 && sr->which == Z_SRW_searchRetrieve_response)
4450                 handle_srw_response(sr->u.response);
4451             else
4452             {
4453                 printf("Decoding of SOLR package failed\n");
4454                 ret = -1;
4455             }
4456             odr_destroy(o);
4457         }
4458         else
4459         {
4460             Z_SOAP *soap_package = 0;
4461             ODR o = odr_createmem(ODR_DECODE);
4462             Z_SOAP_Handler soap_handlers[4] = {
4463                 {YAZ_XMLNS_SRU_v1_1, 0, (Z_SOAP_fun) yaz_srw_codec},
4464                 {YAZ_XMLNS_SRU_v2_response, 0, (Z_SOAP_fun) yaz_srw_codec},
4465                 {YAZ_XMLNS_UPDATE_v0_9, 0, (Z_SOAP_fun) yaz_ucp_codec},
4466                 {0, 0, 0}
4467             };
4468             ret = z_soap_codec(o, &soap_package,
4469                                &hres->content_buf, &hres->content_len,
4470                                soap_handlers);
4471             if (!ret && soap_package->which == Z_SOAP_generic)
4472             {
4473                 Z_SRW_PDU *sr = (Z_SRW_PDU *) soap_package->u.generic->p;
4474                 if (sr->which == Z_SRW_searchRetrieve_response)
4475                     handle_srw_response(sr->u.response);
4476                 else if (sr->which == Z_SRW_explain_response)
4477                     handle_srw_explain_response(sr->u.explain_response);
4478                 else if (sr->which == Z_SRW_scan_response)
4479                     handle_srw_scan_response(sr->u.scan_response);
4480                 else if (sr->which == Z_SRW_update_response)
4481                     printf("Got update response. Status: %s\n",
4482                            sr->u.update_response->operationStatus);
4483                 else
4484                 {
4485                     printf("Decoding of SRW package failed\n");
4486                     ret = -1;
4487                 }
4488             }
4489             else if (soap_package && (soap_package->which == Z_SOAP_fault
4490                                       || soap_package->which == Z_SOAP_error))
4491             {
4492                 printf("SOAP Fault code %s\n",
4493                        soap_package->u.fault->fault_code);
4494                 printf("SOAP Fault string %s\n",
4495                        soap_package->u.fault->fault_string);
4496                 if (soap_package->u.fault->details)
4497                     printf("SOAP Details %s\n",
4498                            soap_package->u.fault->details);
4499             }
4500             else
4501             {
4502                 printf("z_soap_codec failed. (no SOAP error)\n");
4503                 ret = -1;
4504             }
4505             odr_destroy(o);
4506         }
4507     }
4508     if (ret)
4509         close_session(); /* close session on error */
4510     else
4511     {
4512         if (!strcmp(hres->version, "1.0"))
4513         {
4514             /* HTTP 1.0: only if Keep-Alive we stay alive.. */
4515             if (!connection_head || strcmp(connection_head, "Keep-Alive"))
4516                 close_session();
4517         }
4518         else
4519         {
4520             /* HTTP 1.1: only if no close we stay alive .. */
4521             if (connection_head && !strcmp(connection_head, "close"))
4522                 close_session();
4523         }
4524     }
4525 }
4526 #endif
4527
4528 #define max_HTTP_redirects 2
4529
4530 static void wait_and_handle_response(int one_response_only)
4531 {
4532     int reconnect_ok = 1;
4533     int no_redirects = 0;
4534     int res;
4535     char *netbuffer= 0;
4536     int netbufferlen = 0;
4537 #if HAVE_GETTIMEOFDAY
4538     int got_tv_end = 0;
4539     struct timeval tv_end;
4540 #endif
4541     Z_GDU *gdu;
4542
4543     while(conn)
4544     {
4545         res = cs_get(conn, &netbuffer, &netbufferlen);
4546         if (reconnect_ok && res <= 0 && protocol == PROTO_HTTP)
4547         {
4548             cs_close(conn);
4549             conn = 0;
4550             session_connect(cur_host);
4551             reconnect_ok = 0;
4552             if (conn)
4553             {
4554                 char *buf_out;
4555                 int len_out;
4556
4557                 buf_out = odr_getbuf(out, &len_out, 0);
4558
4559                 do_hex_dump(buf_out, len_out);
4560
4561                 cs_put(conn, buf_out, len_out);
4562
4563                 odr_reset(out);
4564                 continue;
4565             }
4566         }
4567         else if (res <= 0)
4568         {
4569             printf("Target closed connection\n");
4570             close_session();
4571             break;
4572         }
4573 #if HAVE_GETTIMEOFDAY
4574         if (got_tv_end == 0)
4575             gettimeofday(&tv_end, 0); /* count first one only */
4576         got_tv_end++;
4577 #endif
4578         odr_reset(out);
4579         odr_reset(in); /* release APDU from last round */
4580         record_last = 0;
4581         do_hex_dump(netbuffer, res);
4582         odr_setbuf(in, netbuffer, res, 0);
4583
4584         if (!z_GDU(in, &gdu, 0, 0))
4585         {
4586             FILE *f = ber_file ? ber_file : stdout;
4587             odr_perror(in, "Decoding incoming APDU");
4588             fprintf(f, "[Near %ld]\n", (long) odr_offset(in));
4589             fprintf(f, "Packet dump:\n---------\n");
4590             odr_dumpBER(f, netbuffer, res);
4591             fprintf(f, "---------\n");
4592             if (apdu_file)
4593             {
4594                 z_GDU(print, &gdu, 0, 0);
4595                 odr_reset(print);
4596             }
4597             if (conn && cs_more(conn))
4598                 continue;
4599             break;
4600         }
4601         if (ber_file)
4602             odr_dumpBER(ber_file, netbuffer, res);
4603         if (apdu_file && !z_GDU(print, &gdu, 0, 0))
4604         {
4605             odr_perror(print, "Failed to print incoming APDU");
4606             odr_reset(print);
4607                 continue;
4608         }
4609         if (gdu->which == Z_GDU_Z3950)
4610         {
4611             Z_APDU *apdu = gdu->u.z3950;
4612             switch (apdu->which)
4613             {
4614             case Z_APDU_initResponse:
4615                 process_Z3950_initResponse(apdu->u.initResponse);
4616                 break;
4617             case Z_APDU_searchResponse:
4618                 process_Z3950_searchResponse(apdu->u.searchResponse);
4619                 break;
4620             case Z_APDU_scanResponse:
4621                 process_Z3950_scanResponse(apdu->u.scanResponse);
4622                 break;
4623             case Z_APDU_presentResponse:
4624                 print_refid(apdu->u.presentResponse->referenceId);
4625                 setno +=
4626                     *apdu->u.presentResponse->numberOfRecordsReturned;
4627                 if (apdu->u.presentResponse->records)
4628                     display_records(apdu->u.presentResponse->records);
4629                 else
4630                     printf("No records.\n");
4631                 printf("nextResultSetPosition = " ODR_INT_PRINTF "\n",
4632                         *apdu->u.presentResponse->nextResultSetPosition);
4633                 break;
4634             case Z_APDU_sortResponse:
4635                 process_Z3950_sortResponse(apdu->u.sortResponse);
4636                 break;
4637             case Z_APDU_extendedServicesResponse:
4638                 printf("Got extended services response\n");
4639                 process_Z3950_ESResponse(apdu->u.extendedServicesResponse);
4640                 break;
4641             case Z_APDU_close:
4642                 printf("Target has closed the association.\n");
4643                 process_Z3950_close(apdu->u.close);
4644                 break;
4645             case Z_APDU_resourceControlRequest:
4646                 process_Z3950_resourceControlRequest(
4647                     apdu->u.resourceControlRequest);
4648                 break;
4649             case Z_APDU_deleteResultSetResponse:
4650                 process_Z3950_deleteResultSetResponse(
4651                     apdu->u.deleteResultSetResponse);
4652                 break;
4653             default:
4654                 printf("Received unknown APDU type (%d).\n",
4655                        apdu->which);
4656                 close_session();
4657             }
4658         }
4659 #if YAZ_HAVE_XML2
4660         else if (gdu->which == Z_GDU_HTTP_Response)
4661         {
4662             Z_HTTP_Response *hres = gdu->u.HTTP_Response;
4663             int code = hres->code;
4664             const char *location = 0;
4665             if ((code == 301 || code == 302)
4666                 && no_redirects < max_HTTP_redirects
4667                 && !yaz_matchstr(sru_method, "get")
4668                 && (location = z_HTTP_header_lookup(hres->headers, "Location")))
4669             {
4670                 const char *base_tmp;
4671                 session_connect_base(location, &base_tmp);
4672                 no_redirects++;
4673                 if (conn)
4674                 {
4675                     if (send_SRW_redirect(location, hres) == 2)
4676                         continue;
4677                 }
4678                 printf("Redirect failed\n");
4679             }
4680             else
4681                 http_response(gdu->u.HTTP_Response);
4682         }
4683 #endif
4684         if (one_response_only)
4685             break;
4686         if (conn && !cs_more(conn))
4687             break;
4688     }
4689 #if HAVE_GETTIMEOFDAY
4690     if (got_tv_end)
4691     {
4692 #if 0
4693         printf("S/U S/U=%ld/%ld %ld/%ld",
4694                 (long) tv_start.tv_sec,
4695                 (long) tv_start.tv_usec,
4696                 (long) tv_end.tv_sec,
4697                 (long) tv_end.tv_usec);
4698 #endif
4699         printf("Elapsed: %.6f\n",
4700                 (double) tv_end.tv_usec / 1e6 + tv_end.tv_sec -
4701                 ((double) tv_start.tv_usec / 1e6 + tv_start.tv_sec));
4702     }
4703 #endif
4704     xfree(netbuffer);
4705 }
4706
4707 static int cmd_cclparse(const char* arg)
4708 {
4709     int error, pos;
4710     struct ccl_rpn_node *rpn=NULL;
4711
4712
4713     rpn = ccl_find_str(bibset, arg, &error, &pos);
4714
4715     if (error)
4716     {
4717         int ioff = 3+strlen(last_cmd)+1+pos;
4718         printf("%*s^ - ", ioff, " ");
4719         printf("%s\n", ccl_err_msg(error));
4720     }
4721     else
4722     {
4723         if (rpn)
4724         {
4725             ccl_pr_tree(rpn, stdout);
4726         }
4727     }
4728     if (rpn)
4729         ccl_rpn_delete(rpn);
4730
4731     printf("\n");
4732
4733     return 0;
4734 }
4735
4736 static int cmd_set_otherinfo(const char* args)
4737 {
4738     char oidstr[101], otherinfoString[101];
4739     int otherinfoNo;
4740     int sscan_res;
4741
4742     sscan_res = sscanf(args, "%d %100[^ ] %100s",
4743                         &otherinfoNo, oidstr, otherinfoString);
4744
4745     if (sscan_res > 0 && otherinfoNo >= maxOtherInfosSupported)
4746     {
4747         printf("Error otherinfo index too large (%d>=%d)\n",
4748                otherinfoNo,maxOtherInfosSupported);
4749         return 0;
4750     }
4751
4752
4753     if (sscan_res==1)
4754     {
4755         /* reset this otherinfo */
4756         extraOtherInfos[otherinfoNo].oid[0] = -1;
4757         xfree(extraOtherInfos[otherinfoNo].value);
4758         extraOtherInfos[otherinfoNo].value = 0;
4759         return 0;
4760     }
4761     if (sscan_res != 3)
4762     {
4763         printf("Error in set_otherinfo command \n");
4764         return 0;
4765     }
4766     else
4767     {
4768         NMEM oid_tmp = nmem_create();
4769         const Odr_oid *oid =
4770             yaz_string_to_oid_nmem(yaz_oid_std(),
4771                                    CLASS_GENERAL, oidstr, oid_tmp);
4772         oid_oidcpy(extraOtherInfos[otherinfoNo].oid, oid);
4773
4774         xfree(extraOtherInfos[otherinfoNo].value);
4775         extraOtherInfos[otherinfoNo].value = xstrdup(otherinfoString);
4776
4777         nmem_destroy(oid_tmp);
4778     }
4779
4780     return 0;
4781 }
4782
4783 static int cmd_sleep(const char* args )
4784 {
4785     int sec = atoi(args);
4786     if (sec > 0)
4787     {
4788 #ifdef WIN32
4789         Sleep(sec*1000);
4790 #else
4791         sleep(sec);
4792 #endif
4793         printf("Done sleeping %d seconds\n", sec);
4794     }
4795     return 1;
4796 }
4797
4798 static int cmd_list_otherinfo(const char* args)
4799 {
4800     int i;
4801
4802     if (strlen(args)>0)
4803     {
4804         i = atoi(args);
4805         if (i >= maxOtherInfosSupported)
4806         {
4807             printf("Error otherinfo index to large (%d>%d)\n",i,maxOtherInfosSupported);
4808             return 0;
4809         }
4810         if (extraOtherInfos[i].value)
4811         {
4812             char name_oid[OID_STR_MAX];
4813             oid_class oclass;
4814             const char *name =
4815                 yaz_oid_to_string_buf(extraOtherInfos[i].oid, &oclass,
4816                                       name_oid);
4817             printf("  otherinfo %d %s %s\n",
4818                    i, name ? name : "null",
4819                    extraOtherInfos[i].value);
4820         }
4821
4822     }
4823     else
4824     {
4825         for (i = 0; i < maxOtherInfosSupported; ++i)
4826         {
4827             if (extraOtherInfos[i].value)
4828             {
4829                 char name_oid[OID_STR_MAX];
4830                 oid_class oclass;
4831                 const char *name =
4832                     yaz_oid_to_string_buf(extraOtherInfos[i].oid, &oclass,
4833                                           name_oid);
4834                 printf("  otherinfo %d %s %s\n",
4835                        i, name ? name : "null",
4836                        extraOtherInfos[i].value);
4837             }
4838         }
4839     }
4840     return 0;
4841 }
4842
4843 static int cmd_list_all(const char* args)
4844 {
4845     int i;
4846
4847     /* connection options */
4848     if (conn)
4849         printf("Connected to         : %s\n", cur_host);
4850     else if (*cur_host)
4851         printf("Not connected to     : %s\n", cur_host);
4852     else
4853         printf("Not connected        : \n");
4854     if (yazProxy) printf("using proxy          : %s\n",yazProxy);
4855
4856     printf("auto_reconnect       : %s\n",auto_reconnect?"on":"off");
4857     printf("auto_wait            : %s\n",auto_wait?"on":"off");
4858
4859     if (!auth)
4860         printf("Authentication       : none\n");
4861     else
4862     {
4863         switch (auth->which)
4864         {
4865         case Z_IdAuthentication_idPass:
4866             printf("Authentication       : IdPass\n");
4867             printf("    Login User       : %s\n",auth->u.idPass->userId?auth->u.idPass->userId:"");
4868             printf("    Login Group      : %s\n",auth->u.idPass->groupId?auth->u.idPass->groupId:"");
4869             printf("    Password         : %s\n",auth->u.idPass->password?auth->u.idPass->password:"");
4870             break;
4871         case Z_IdAuthentication_open:
4872             printf("Authentication       : psOpen\n");
4873             printf("    Open string      : %s\n",auth->u.open);
4874             break;
4875         default:
4876             printf("Authentication       : Unknown\n");
4877         }
4878     }
4879     if (negotiationCharset)
4880         printf("Neg. Character set   : `%s'\n", negotiationCharset);
4881
4882     /* bases */
4883     printf("Bases                : ");
4884     for (i = 0; i<num_databaseNames; i++) printf("%s ",databaseNames[i]);
4885     printf("\n");
4886
4887     /* Query options */
4888     printf("CCL file             : %s\n",ccl_fields);
4889     printf("CQL file             : %s\n",cql_fields);
4890     printf("Query type           : %s\n",query_type_as_string(queryType));
4891
4892     printf("Named Result Sets    : %s\n",setnumber==-1?"off":"on");
4893
4894     /* piggy back options */
4895     printf("ssub/lslb/mspn       : %d/%d/%d\n",smallSetUpperBound,largeSetLowerBound,mediumSetPresentNumber);
4896
4897     /* print present related options */
4898     if (recordsyntax_size > 0)
4899     {
4900         printf("Format               : %s\n", recordsyntax_list[0]);
4901     }
4902     printf("Schema               : %s\n",record_schema ? record_schema : "not set");
4903     printf("Elements             : %s\n",elementSetNames?elementSetNames->u.generic:"");
4904
4905     /* loging options */
4906     printf("APDU log             : %s\n",apdu_file?"on":"off");
4907     printf("Record log           : %s\n",marc_file?"on":"off");
4908
4909     /* other infos */
4910     printf("Other Info: \n");
4911     cmd_list_otherinfo("");
4912
4913     return 0;
4914 }
4915
4916 static int cmd_clear_otherinfo(const char* args)
4917 {
4918     if (strlen(args) > 0)
4919     {
4920         int otherinfoNo = atoi(args);
4921         if (otherinfoNo >= maxOtherInfosSupported)
4922         {
4923             printf("Error otherinfo index too large (%d>=%d)\n",
4924                    otherinfoNo, maxOtherInfosSupported);
4925             return 0;
4926         }
4927         if (extraOtherInfos[otherinfoNo].value)
4928         {
4929             /* only clear if set. */
4930             extraOtherInfos[otherinfoNo].oid[0] = -1;
4931             xfree(extraOtherInfos[otherinfoNo].value);
4932             extraOtherInfos[otherinfoNo].value = 0;
4933         }
4934     }
4935     else
4936     {
4937         int i;
4938         for (i = 0; i < maxOtherInfosSupported; ++i)
4939         {
4940             if (extraOtherInfos[i].value)
4941             {
4942                 extraOtherInfos[i].oid[0] = -1;
4943                 xfree(extraOtherInfos[i].value);
4944                 extraOtherInfos[i].value = 0;
4945             }
4946         }
4947     }
4948     return 0;
4949 }
4950
4951 static int cmd_wait_response(const char *arg)
4952 {
4953     int i;
4954     int wait_for = atoi(arg);
4955     if (wait_for < 1)
4956         wait_for = 1;
4957
4958     for (i = 0 ; i < wait_for; ++i )
4959         wait_and_handle_response(1);
4960     return 0;
4961 }
4962
4963 static int cmd_help(const char *line);
4964
4965 typedef char *(*completerFunctionType)(const char *text, int state);
4966
4967 static struct {
4968     char *cmd;
4969     int (*fun)(const char *arg);
4970     char *ad;
4971         completerFunctionType rl_completerfunction;
4972     int complete_filenames;
4973     const char **local_tabcompletes;
4974 } cmd_array[] = {
4975     {"open", cmd_open, "('tcp'|'ssl')':<host>[':'<port>][/<db>]",NULL,0,NULL},
4976     {"quit", cmd_quit, "",NULL,0,NULL},
4977     {"find", cmd_find, "<query>",NULL,0,NULL},
4978     {"facets", cmd_facets, "<query>",NULL,0,NULL},
4979     {"delete", cmd_delete, "<setname>",NULL,0,NULL},
4980     {"base", cmd_base, "<base-name>",NULL,0,NULL},
4981     {"show", cmd_show, "<rec#>['+'<#recs>['+'<setname>]]",NULL,0,NULL},
4982     {"setscan", cmd_setscan, "<term>",NULL,0,NULL},
4983     {"scan", cmd_scan, "<term>",NULL,0,NULL},
4984     {"scanstep", cmd_scanstep, "<size>",NULL,0,NULL},
4985     {"scanpos", cmd_scanpos, "<size>",NULL,0,NULL},
4986     {"scansize", cmd_scansize, "<size>",NULL,0,NULL},
4987     {"sort", cmd_sort, "<sortkey> <flag> <sortkey> <flag> ...",NULL,0,NULL},
4988     {"sort+", cmd_sort_newset, "<sortkey> <flag> <sortkey> <flag> ...",NULL,0,NULL},
4989     {"authentication", cmd_authentication, "<acctstring>",NULL,0,NULL},
4990     {"lslb", cmd_lslb, "<largeSetLowerBound>",NULL,0,NULL},
4991     {"ssub", cmd_ssub, "<smallSetUpperBound>",NULL,0,NULL},
4992     {"mspn", cmd_mspn, "<mediumSetPresentNumber>",NULL,0,NULL},
4993     {"status", cmd_status, "",NULL,0,NULL},
4994     {"setnames", cmd_setnames, "",NULL,0,NULL},
4995     {"cancel", cmd_cancel, "",NULL,0,NULL},
4996     {"cancel_find", cmd_cancel_find, "<query>",NULL,0,NULL},
4997     {"format", cmd_format, "<recordsyntax>",complete_format,0,NULL},
4998     {"schema", cmd_schema, "<schema>",complete_schema,0,NULL},
4999     {"elements", cmd_elements, "<elementSetName>",NULL,0,NULL},
5000     {"close", cmd_close, "",NULL,0,NULL},
5001     {"querytype", cmd_querytype, "<type>",complete_querytype,0,NULL},
5002     {"refid", cmd_refid, "<id>",NULL,0,NULL},
5003     {"itemorder", cmd_itemorder, "ill|item|xml <itemno>",NULL,0,NULL},
5004     {"update", cmd_update, "<action> <recid> [<doc>]",NULL,0,NULL},
5005     {"update0", cmd_update0, "<action> <recid> [<doc>]",NULL,0,NULL},
5006     {"xmles", cmd_xmles, "<OID> <doc>",NULL,0,NULL},
5007     {"packagename", cmd_packagename, "<packagename>",NULL,0,NULL},
5008     {"proxy", cmd_proxy, "[('tcp'|'ssl')]<host>[':'<port>]",NULL,0,NULL},
5009     {"charset", cmd_charset, "<nego_charset> <output_charset>",NULL,0,NULL},
5010     {"negcharset", cmd_negcharset, "<nego_charset>",NULL,0,NULL},
5011     {"displaycharset", cmd_displaycharset, "<output_charset>",NULL,0,NULL},
5012     {"marccharset", cmd_marccharset, "<charset_name>",NULL,0,NULL},
5013     {"querycharset", cmd_querycharset, "<charset_name>",NULL,0,NULL},
5014     {"lang", cmd_lang, "<language_code>",NULL,0,NULL},
5015     {"source", cmd_source_echo, "<filename>",NULL,1,NULL},
5016     {".", cmd_source_echo, "<filename>",NULL,1,NULL},
5017     {"!", cmd_subshell, "Subshell command",NULL,1,NULL},
5018     {"set_apdufile", cmd_set_apdufile, "<filename>",NULL,1,NULL},
5019     {"set_berfile", cmd_set_berfile, "<filename>",NULL,1,NULL},
5020     {"set_marcdump", cmd_set_marcdump," <filename>",NULL,1,NULL},
5021     {"set_cclfile", cmd_set_cclfile," <filename>",NULL,1,NULL},
5022     {"set_cqlfile", cmd_set_cqlfile," <filename>",NULL,1,NULL},
5023     {"set_auto_reconnect", cmd_set_auto_reconnect," on|off",complete_auto_reconnect,1,NULL},
5024     {"set_auto_wait", cmd_set_auto_wait," on|off",complete_auto_reconnect,1,NULL},
5025     {"set_otherinfo", cmd_set_otherinfo,"<otherinfoinddex> <oid> <string>",NULL,0,NULL},
5026     {"sleep", cmd_sleep,"<seconds>",NULL,0,NULL},
5027     {"register_oid", cmd_register_oid,"<name> <class> <oid>",NULL,0,NULL},
5028     {"push_command", cmd_push_command,"<command>",command_generator,0,NULL},
5029     {"register_tab", cmd_register_tab,"<commandname> <tab>",command_generator,0,NULL},
5030     {"cclparse", cmd_cclparse,"<ccl find command>",NULL,0,NULL},
5031     {"list_otherinfo",cmd_list_otherinfo,"[otherinfoinddex]",NULL,0,NULL},
5032     {"list_all",cmd_list_all,"",NULL,0,NULL},
5033     {"clear_otherinfo",cmd_clear_otherinfo,"",NULL,0,NULL},
5034     {"wait_response",cmd_wait_response,"<number>",NULL,0,NULL},
5035     /* Server Admin Functions */
5036     {"adm-reindex", cmd_adm_reindex, "<database-name>",NULL,0,NULL},
5037     {"adm-truncate", cmd_adm_truncate, "('database'|'index')<object-name>",NULL,0,NULL},
5038     {"adm-create", cmd_adm_create, "",NULL,0,NULL},
5039     {"adm-drop", cmd_adm_drop, "('database'|'index')<object-name>",NULL,0,NULL},
5040     {"adm-import", cmd_adm_import, "<record-type> <dir> <pattern>",NULL,0,NULL},
5041     {"adm-refresh", cmd_adm_refresh, "",NULL,0,NULL},
5042     {"adm-commit", cmd_adm_commit, "",NULL,0,NULL},
5043     {"adm-shutdown", cmd_adm_shutdown, "",NULL,0,NULL},
5044     {"adm-startup", cmd_adm_startup, "",NULL,0,NULL},
5045     {"explain", cmd_explain, "", NULL, 0, NULL},
5046     {"options", cmd_options, "", NULL, 0, NULL},
5047     {"zversion", cmd_zversion, "", NULL, 0, NULL},
5048     {"help", cmd_help, "", NULL,0,NULL},
5049     {"init", cmd_init, "", NULL,0,NULL},
5050     {"sru", cmd_sru, "<method> <version>", NULL,0,NULL},
5051     {"url", cmd_url, "<url>", NULL,0,NULL},
5052     {"exit", cmd_quit, "",NULL,0,NULL},
5053     {0,0,0,0,0,0}
5054 };
5055
5056 static int cmd_help(const char *line)
5057 {
5058     int i;
5059     char topic[21];
5060
5061     *topic = 0;
5062     sscanf(line, "%20s", topic);
5063
5064     if (*topic == 0)
5065         printf("Commands:\n");
5066     for (i = 0; cmd_array[i].cmd; i++)
5067         if (*topic == 0 || strcmp(topic, cmd_array[i].cmd) == 0)
5068             printf("   %s %s\n", cmd_array[i].cmd, cmd_array[i].ad);
5069     if (!strcmp(topic, "find"))
5070     {
5071         printf("RPN:\n");
5072         printf(" \"term\"                        Simple Term\n");
5073         printf(" @attr [attset] type=value op  Attribute\n");
5074         printf(" @and opl opr                  And\n");
5075         printf(" @or opl opr                   Or\n");
5076         printf(" @not opl opr                  And-Not\n");
5077         printf(" @set set                      Result set\n");
5078         printf(" @prox exl dist ord rel uc ut  Proximity. Use help prox\n");
5079         printf("\n");
5080         printf("Bib-1 attribute types\n");
5081         printf("1=Use:         ");
5082         printf("4=Title 7=ISBN 8=ISSN 30=Date 62=Abstract 1003=Author 1016=Any\n");
5083         printf("2=Relation:    ");
5084         printf("1<   2<=  3=  4>=  5>  6!=  102=Relevance\n");
5085         printf("3=Position:    ");
5086         printf("1=First in Field  2=First in subfield  3=Any position\n");
5087         printf("4=Structure:   ");
5088         printf("1=Phrase  2=Word  3=Key  4=Year  5=Date  6=WordList\n");
5089         printf("5=Truncation:  ");
5090         printf("1=Right  2=Left  3=L&R  100=No  101=#  102=Re-1  103=Re-2\n");
5091         printf("6=Completeness:");
5092         printf("1=Incomplete subfield  2=Complete subfield  3=Complete field\n");
5093     }
5094     if (!strcmp(topic, "prox"))
5095     {
5096         printf("Proximity:\n");
5097         printf(" @prox exl dist ord rel uc ut\n");
5098         printf(" exl:  exclude flag . 0=include, 1=exclude.\n");
5099         printf(" dist: distance integer.\n");
5100         printf(" ord:  order flag. 0=unordered, 1=ordered.\n");
5101         printf(" rel:  relation integer. 1<  2<=  3= 4>=  5>  6!= .\n");
5102         printf(" uc:   unit class. k=known, p=private.\n");
5103         printf(" ut:   unit type. 1=character, 2=word, 3=sentence,\n");
5104         printf("        4=paragraph, 5=section, 6=chapter, 7=document,\n");
5105         printf("        8=element, 9=subelement, 10=elementType, 11=byte.\n");
5106         printf("\nExamples:\n");
5107         printf(" Search for a and b in-order at most 3 words apart:\n");
5108         printf("  @prox 0 3 1 2 k 2 a b\n");
5109         printf(" Search for any order of a and b next to each other:\n");
5110         printf("  @prox 0 1 0 3 k 2 a b\n");
5111     }
5112     return 1;
5113 }
5114
5115 static int cmd_register_tab(const char* arg)
5116 {
5117 #if HAVE_READLINE_READLINE_H
5118     char command[101], tabargument[101];
5119     int i;
5120     int num_of_tabs;
5121     const char** tabslist;
5122
5123     if (sscanf(arg, "%100s %100s", command, tabargument) < 1)
5124     {
5125         return 0;
5126     }
5127
5128     /* locate the amdn in the list */
5129     for (i = 0; cmd_array[i].cmd; i++)
5130     {
5131         if (!strncmp(cmd_array[i].cmd, command, strlen(command)))
5132             break;
5133     }
5134
5135     if (!cmd_array[i].cmd)
5136     {
5137         fprintf(stderr,"Unknown command %s\n",command);
5138         return 1;
5139     }
5140
5141
5142     if (!cmd_array[i].local_tabcompletes)
5143         cmd_array[i].local_tabcompletes = (const char **) calloc(1,sizeof(char**));
5144
5145     num_of_tabs=0;
5146
5147     tabslist = cmd_array[i].local_tabcompletes;
5148     for (; tabslist && *tabslist; tabslist++)
5149         num_of_tabs++;
5150
5151     cmd_array[i].local_tabcompletes = (const char **)
5152         realloc(cmd_array[i].local_tabcompletes,
5153                 (num_of_tabs+2)*sizeof(char**));
5154     tabslist = cmd_array[i].local_tabcompletes;
5155     tabslist[num_of_tabs] = strdup(tabargument);
5156     tabslist[num_of_tabs+1] = NULL;
5157 #endif
5158     return 1;
5159 }
5160
5161 static void process_cmd_line(char* line)
5162 {
5163     int i, res;
5164     char word[32], arg[10240];
5165
5166 #if HAVE_GETTIMEOFDAY
5167     gettimeofday(&tv_start, 0);
5168 #endif
5169
5170     if ((res = sscanf(line, "%31s %10239[^;]", word, arg)) <= 0)
5171     {
5172         strcpy(word, last_cmd);
5173         *arg = '\0';
5174     }
5175     else if (res == 1)
5176         *arg = 0;
5177     strcpy(last_cmd, word);
5178
5179     /* removed tailing spaces from the arg command */
5180     {
5181         char* p = arg;
5182         char* lastnonspace=NULL;
5183
5184         for (; *p; ++p)
5185         {
5186             if (!yaz_isspace(*p))
5187                 lastnonspace = p;
5188         }
5189         if (lastnonspace)
5190             *(++lastnonspace) = 0;
5191     }
5192
5193     for (i = 0; cmd_array[i].cmd; i++)
5194         if (!strncmp(cmd_array[i].cmd, word, strlen(word)))
5195         {
5196             res = (*cmd_array[i].fun)(arg);
5197             break;
5198         }
5199
5200     if (!cmd_array[i].cmd) /* dump our help-screen */
5201     {
5202         printf("Unknown command: %s.\n", word);
5203         printf("Type 'help' for list of commands\n");
5204         res = 1;
5205     }
5206
5207     if (apdu_file)
5208         fflush(apdu_file);
5209
5210     if (res >= 2 && auto_wait)
5211         wait_and_handle_response(0);
5212
5213     if (apdu_file)
5214         fflush(apdu_file);
5215     if (marc_file)
5216         fflush(marc_file);
5217 }
5218
5219 static char *command_generator(const char *text, int state)
5220 {
5221 #if HAVE_READLINE_READLINE_H
5222     static int idx;
5223     if (state == 0)
5224         idx = 0;
5225     for (; cmd_array[idx].cmd; ++idx)
5226     {
5227         if (!strncmp(cmd_array[idx].cmd, text, strlen(text)))
5228         {
5229             ++idx;  /* skip this entry on the next run */
5230             return strdup(cmd_array[idx-1].cmd);
5231         }
5232     }
5233 #endif
5234     return NULL;
5235 }
5236
5237 #if HAVE_READLINE_READLINE_H
5238 static const char** default_completer_list = NULL;
5239
5240 static char* default_completer(const char* text, int state)
5241 {
5242     return complete_from_list(default_completer_list, text, state);
5243 }
5244 #endif
5245
5246 #if HAVE_READLINE_READLINE_H
5247
5248 /*
5249    This function only known how to complete on the first word
5250 */
5251 static char **readline_completer(char *text, int start, int end)
5252 {
5253     completerFunctionType completerToUse;
5254
5255     if (start == 0)
5256     {
5257 #if HAVE_READLINE_RL_COMPLETION_MATCHES
5258         char** res = rl_completion_matches(text, command_generator);
5259 #else
5260         char** res = completion_matches(text,
5261                                         (CPFunction*)command_generator);
5262 #endif
5263         rl_attempted_completion_over = 1;
5264         return res;
5265     }
5266     else
5267     {
5268         char arg[10240],word[32];
5269         int i ,res;
5270         if ((res = sscanf(rl_line_buffer, "%31s %10239[^;]", word, arg)) <= 0)
5271         {
5272             rl_attempted_completion_over = 1;
5273             return NULL;
5274         }
5275
5276         for (i = 0; cmd_array[i].cmd; i++)
5277             if (!strncmp(cmd_array[i].cmd, word, strlen(word)))
5278                 break;
5279
5280         if (!cmd_array[i].cmd)
5281             return NULL;
5282
5283         default_completer_list = cmd_array[i].local_tabcompletes;
5284
5285         completerToUse = cmd_array[i].rl_completerfunction;
5286         if (!completerToUse)
5287         { /* if command completer is not defined use the default completer */
5288             completerToUse = default_completer;
5289         }
5290         if (completerToUse)
5291         {
5292 #ifdef HAVE_READLINE_RL_COMPLETION_MATCHES
5293             char** res=
5294                 rl_completion_matches(text, completerToUse);
5295 #else
5296             char** res=
5297                 completion_matches(text, (CPFunction*)completerToUse);
5298 #endif
5299             if (!cmd_array[i].complete_filenames)
5300                 rl_attempted_completion_over = 1;
5301             return res;
5302         }
5303         else
5304         {
5305             if (!cmd_array[i].complete_filenames)
5306                 rl_attempted_completion_over = 1;
5307             return 0;
5308         }
5309     }
5310 }
5311 #endif
5312
5313 #ifndef WIN32
5314 static void ctrl_c_handler(int x)
5315 {
5316     exit_client(0);
5317 }
5318 #endif
5319
5320 static void client(void)
5321 {
5322     char line[10240];
5323
5324     line[10239] = '\0';
5325
5326 #ifndef WIN32
5327     signal(SIGINT, ctrl_c_handler);
5328 #endif
5329
5330 #if HAVE_GETTIMEOFDAY
5331     gettimeofday(&tv_start, 0);
5332 #endif
5333
5334     while (1)
5335     {
5336         char *line_in = NULL;
5337 #if HAVE_READLINE_READLINE_H
5338         if (isatty(0))
5339         {
5340             line_in=readline(C_PROMPT);
5341             if (!line_in)
5342             {
5343                 putchar('\n');
5344                 break;
5345             }
5346 #if HAVE_READLINE_HISTORY_H
5347             if (*line_in)
5348                 add_history(line_in);
5349 #endif
5350             strncpy(line, line_in, sizeof(line)-1);
5351             free(line_in);
5352         }
5353 #endif
5354         if (!line_in)
5355         {
5356             char *end_p;
5357             printf(C_PROMPT);
5358             fflush(stdout);
5359             if (!fgets(line, sizeof(line)-1, stdin))
5360                 break;
5361             if ((end_p = strchr(line, '\n')))
5362                 *end_p = '\0';
5363         }
5364         if (isatty(0))
5365             file_history_add_line(file_history, line);
5366         process_cmd_line(line);
5367     }
5368 }
5369
5370 static void show_version(void)
5371 {
5372     char vstr[20], sha1_str[41];
5373
5374     yaz_version(vstr, sha1_str);
5375     printf("YAZ version: %s %s\n", YAZ_VERSION, YAZ_VERSION_SHA1);
5376     if (strcmp(sha1_str, YAZ_VERSION_SHA1))
5377         printf("YAZ DLL/SO: %s %s\n", vstr, sha1_str);
5378     exit(0);
5379 }
5380
5381 int main(int argc, char **argv)
5382 {
5383     char *prog = *argv;
5384     char *open_command = 0;
5385     char *auth_command = 0;
5386     char *arg;
5387     const char *rc_file = 0;
5388     int ret;
5389
5390 #if HAVE_LOCALE_H
5391     if (!setlocale(LC_CTYPE, ""))
5392         fprintf(stderr, "setlocale failed\n");
5393 #endif
5394 #if HAVE_LANGINFO_H
5395 #ifdef CODESET
5396     codeset = nl_langinfo(CODESET);
5397 #endif
5398 #endif
5399     if (codeset)
5400         outputCharset = xstrdup(codeset);
5401
5402     ODR_MASK_SET(&z3950_options, Z_Options_search);
5403     ODR_MASK_SET(&z3950_options, Z_Options_present);
5404     ODR_MASK_SET(&z3950_options, Z_Options_namedResultSets);
5405     ODR_MASK_SET(&z3950_options, Z_Options_triggerResourceCtrl);
5406     ODR_MASK_SET(&z3950_options, Z_Options_scan);
5407     ODR_MASK_SET(&z3950_options, Z_Options_sort);
5408     ODR_MASK_SET(&z3950_options, Z_Options_extendedServices);
5409     ODR_MASK_SET(&z3950_options, Z_Options_delSet);
5410
5411     nmem_auth = nmem_create();
5412
5413     while ((ret = options("k:c:q:a:b:m:v:p:u:t:Vxd:f:", argv, argc, &arg)) != -2)
5414     {
5415         switch (ret)
5416         {
5417         case 0:
5418             if (!open_command)
5419             {
5420                 open_command = (char *) xmalloc(strlen(arg)+6);
5421                 strcpy(open_command, "open ");
5422                 strcat(open_command, arg);
5423             }
5424             else
5425             {
5426                 fprintf(stderr, "%s: Specify at most one server address\n",
5427                         prog);
5428                 exit(1);
5429             }
5430             break;
5431         case 'a':
5432             if (!strcmp(arg, "-"))
5433                 apdu_file=stderr;
5434             else
5435                 apdu_file=fopen(arg, "a");
5436             break;
5437         case 'b':
5438             if (!strcmp(arg, "-"))
5439                 ber_file=stderr;
5440             else
5441                 ber_file=fopen(arg, "a");
5442             break;
5443         case 'c':
5444             strncpy(ccl_fields, arg, sizeof(ccl_fields)-1);
5445             ccl_fields[sizeof(ccl_fields)-1] = '\0';
5446             break;
5447         case 'd':
5448             dump_file_prefix = arg;
5449             break;
5450         case 'f':
5451             rc_file = arg;
5452             break;
5453         case 'k':
5454             kilobytes = atoi(arg);
5455             break;
5456         case 'm':
5457             if (!(marc_file = fopen(arg, "a")))
5458             {
5459                 perror(arg);
5460                 exit(1);
5461             }
5462             break;
5463         case 'p':
5464             yazProxy = xstrdup(arg);
5465             break;
5466         case 'q':
5467             strncpy(cql_fields, arg, sizeof(cql_fields)-1);
5468             cql_fields[sizeof(cql_fields)-1] = '\0';
5469             break;
5470         case 't':
5471             outputCharset = xstrdup(arg);
5472             break;
5473         case 'u':
5474             if (!auth_command)
5475             {
5476                 auth_command = (char *) xmalloc(strlen(arg)+6);
5477                 strcpy(auth_command, "auth ");
5478                 strcat(auth_command, arg);
5479             }
5480             break;
5481         case 'v':
5482             yaz_log_init(yaz_log_mask_str(arg), "", 0);
5483             break;
5484         case 'V':
5485             show_version();
5486             break;
5487         case 'x':
5488             hex_dump = 1;
5489             break;
5490         default:
5491             fprintf(stderr, "Usage: %s "
5492                      " [-a apdulog]"
5493                      " [-b berdump]"
5494                      " [-c cclfile]"
5495                      " [-d dump]"
5496                      " [-f cmdfile]"
5497                      " [-k size]"
5498                      " [-m marclog]"
5499                      " [-p proxy-addr]"
5500                      " [-q cqlfile]"
5501                      " [-t dispcharset]"
5502                      " [-u auth]"
5503                      " [-v loglevel]"
5504                      " [-V]"
5505                      " [-x]"
5506                      " [server-addr]\n",
5507                      prog);
5508             exit(1);
5509         }
5510     }
5511     initialize(rc_file);
5512     if (auth_command)
5513     {
5514 #ifdef HAVE_GETTIMEOFDAY
5515         gettimeofday(&tv_start, 0);
5516 #endif
5517         process_cmd_line(auth_command);
5518 #if HAVE_READLINE_HISTORY_H
5519         add_history(auth_command);
5520 #endif
5521         xfree(auth_command);
5522     }
5523     if (open_command)
5524     {
5525 #ifdef HAVE_GETTIMEOFDAY
5526         gettimeofday(&tv_start, 0);
5527 #endif
5528         process_cmd_line(open_command);
5529 #if HAVE_READLINE_HISTORY_H
5530         add_history(open_command);
5531 #endif
5532         xfree(open_command);
5533     }
5534     client();
5535     exit_client(0);
5536     return 0;
5537 }
5538 /*
5539  * Local variables:
5540  * c-basic-offset: 4
5541  * c-file-style: "Stroustrup"
5542  * indent-tabs-mode: nil
5543  * End:
5544  * vim: shiftwidth=4 tabstop=8 expandtab
5545  */
5546