client/client.c

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