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,
2059                                             const char *xml_buf,
2060                                             int xml_len)
2061 {
2062     Z_External *r = (Z_External *) odr_malloc(out, sizeof(Z_External));
2063     r->direct_reference = odr_oiddup(out, yaz_oid_extserv_item_order);
2064     r->indirect_reference = 0;
2065     r->descriptor = 0;
2066
2067     r->which = Z_External_itemOrder;
2068
2069     r->u.itemOrder = (Z_ItemOrder *) odr_malloc(out,sizeof(Z_ItemOrder));
2070     memset(r->u.itemOrder, 0, sizeof(Z_ItemOrder));
2071     r->u.itemOrder->which=Z_IOItemOrder_esRequest;
2072
2073     r->u.itemOrder->u.esRequest = (Z_IORequest *)
2074         odr_malloc(out,sizeof(Z_IORequest));
2075     memset(r->u.itemOrder->u.esRequest, 0, sizeof(Z_IORequest));
2076
2077     r->u.itemOrder->u.esRequest->toKeep = (Z_IOOriginPartToKeep *)
2078         odr_malloc(out,sizeof(Z_IOOriginPartToKeep));
2079     memset(r->u.itemOrder->u.esRequest->toKeep, 0, sizeof(Z_IOOriginPartToKeep));
2080     r->u.itemOrder->u.esRequest->notToKeep = (Z_IOOriginPartNotToKeep *)
2081         odr_malloc(out,sizeof(Z_IOOriginPartNotToKeep));
2082     memset(r->u.itemOrder->u.esRequest->notToKeep, 0, sizeof(Z_IOOriginPartNotToKeep));
2083
2084     r->u.itemOrder->u.esRequest->toKeep->supplDescription = NULL;
2085     r->u.itemOrder->u.esRequest->toKeep->contact = NULL;
2086     r->u.itemOrder->u.esRequest->toKeep->addlBilling = NULL;
2087
2088     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem =
2089         (Z_IOResultSetItem *) odr_malloc(out, sizeof(Z_IOResultSetItem));
2090     memset(r->u.itemOrder->u.esRequest->notToKeep->resultSetItem, 0, sizeof(Z_IOResultSetItem));
2091     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->resultSetId = "1";
2092
2093     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->item =
2094         (int *) odr_malloc(out, sizeof(int));
2095     *r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->item = itemno;
2096
2097     if (!strcmp (type, "item") || !strcmp(type, "2"))
2098     {
2099         printf ("using item-request\n");
2100         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2101             create_external_itemRequest();
2102     }
2103     else if (!strcmp(type, "ill") || !strcmp(type, "1"))
2104     {
2105         printf ("using ILL-request\n");
2106         r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2107             create_external_ILL_APDU(ILL_APDU_ILL_Request);
2108     }
2109     else if (!strcmp(type, "xml") || !strcmp(type, "3"))
2110     {
2111         printf("using XML ILL-request\n");
2112
2113         if (!xml_buf)
2114         {
2115             printf("no docoument added\n");
2116             r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
2117         }
2118         else
2119         {
2120             r->u.itemOrder->u.esRequest->notToKeep->itemRequest =
2121                 z_ext_record_oid(out, yaz_oid_recsyn_xml, xml_buf, xml_len);
2122         }
2123     }
2124     else
2125         r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
2126
2127     return r;
2128 }
2129
2130 static int send_itemorder(const char *type, int itemno,
2131                           const char *xml_buf, int xml_len)
2132 {
2133     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest);
2134     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2135
2136     req->referenceId = set_refid (out);
2137
2138     req->packageType = odr_oiddup(out, yaz_oid_extserv_item_order);
2139     req->packageName = esPackageName;
2140
2141     req->taskSpecificParameters = create_ItemOrderExternal(type, itemno,
2142                                                            xml_buf, xml_len);
2143     send_apdu(apdu);
2144     return 0;
2145 }
2146
2147 static int only_z3950(void)
2148 {
2149     if (!conn)
2150     {
2151         printf ("Not connected yet\n");
2152         return 1;
2153     }
2154     if (protocol == PROTO_HTTP)
2155     {
2156         printf ("Not supported by SRW\n");
2157         return 1;
2158     }
2159     return 0;
2160 }
2161
2162 static int cmd_update_common(const char *arg, int version);
2163
2164 static int cmd_update(const char *arg)
2165 {
2166     return cmd_update_common(arg, 1);
2167 }
2168
2169 static int cmd_update0(const char *arg)
2170 {
2171     return cmd_update_common(arg, 0);
2172 }
2173
2174 static int cmd_update_Z3950(int version, int action_no, const char *recid,
2175                             char *rec_buf, int rec_len);
2176
2177 static int cmd_update_SRW(int action_no, const char *recid,
2178                           char *rec_buf, int rec_len);
2179
2180 static int cmd_update_common(const char *arg, int version)
2181 {
2182     char *action_buf;
2183     int action_len;
2184     char *recid_buf;
2185     int recid_len;
2186     const char *recid = 0;
2187     char *rec_buf;
2188     int rec_len;
2189     int action_no;
2190     int noread = 0;
2191
2192     if (parse_cmd_doc(&arg, out, &action_buf, &action_len) == 0)
2193     {
2194         printf("Use: update action recid [fname]\n");
2195         printf(" where action is one of insert,replace,delete.update\n");
2196         printf(" recid is some record ID. Use none for no ID\n");
2197         printf(" fname is file of record to be updated\n");
2198         return 0;
2199     }
2200
2201     if (parse_cmd_doc(&arg, out, &recid_buf, &recid_len) == 0)
2202     {
2203         printf("Missing recid\n");
2204         return 0;
2205     }
2206
2207     if (!strcmp(action_buf, "insert"))
2208         action_no = Z_IUOriginPartToKeep_recordInsert;
2209     else if (!strcmp(action_buf, "replace"))
2210         action_no = Z_IUOriginPartToKeep_recordReplace;
2211     else if (!strcmp(action_buf, "delete"))
2212         action_no = Z_IUOriginPartToKeep_recordDelete;
2213     else if (!strcmp(action_buf, "update"))
2214         action_no = Z_IUOriginPartToKeep_specialUpdate;
2215     else
2216     {
2217         printf ("Bad action: %s\n", action_buf);
2218         printf ("Possible values: insert, replace, delete, update\n");
2219         return 0;
2220     }
2221
2222     if (strcmp(recid_buf, "none")) /* none means no record ID */
2223         recid = recid_buf;
2224
2225     arg += noread;
2226     if (parse_cmd_doc(&arg, out, &rec_buf, &rec_len) == 0)
2227         return 0;
2228
2229 #if YAZ_HAVE_XML2
2230     if (protocol == PROTO_HTTP)
2231         return cmd_update_SRW(action_no, recid_buf, rec_buf, rec_len);
2232 #endif
2233     return cmd_update_Z3950(version, action_no, recid_buf, rec_buf, rec_len);
2234 }
2235
2236 #if YAZ_HAVE_XML2
2237 static int cmd_update_SRW(int action_no, const char *recid,
2238                           char *rec_buf, int rec_len)
2239 {
2240     if (!conn)
2241         session_connect(cur_host);
2242     if (!conn)
2243         return 0;
2244     else
2245     {
2246         Z_SRW_PDU *srw = yaz_srw_get(out, Z_SRW_update_request);
2247         Z_SRW_updateRequest *sr = srw->u.update_request;
2248
2249         switch(action_no)
2250         {
2251         case Z_IUOriginPartToKeep_recordInsert:
2252             sr->operation = "info:srw/action/1/create";
2253             break;
2254         case Z_IUOriginPartToKeep_recordReplace:
2255             sr->operation = "info:srw/action/1/replace";
2256             break;
2257         case Z_IUOriginPartToKeep_recordDelete:
2258             sr->operation = "info:srw/action/1/delete";
2259             break;
2260         }
2261         if (rec_buf)
2262         {
2263             sr->record = yaz_srw_get_record(out);
2264             sr->record->recordData_buf = rec_buf;
2265             sr->record->recordData_len = rec_len;
2266             sr->record->recordSchema = record_schema;
2267         }
2268         if (recid)
2269             sr->recordId = odr_strdup(out, recid);
2270         return send_srw(srw);
2271     }
2272 }
2273 #endif
2274
2275 static int cmd_update_Z3950(int version, int action_no, const char *recid,
2276                             char *rec_buf, int rec_len)
2277 {
2278     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest );
2279     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2280     Z_External *r;
2281     Z_External *record_this = 0;
2282     if (rec_buf)
2283         record_this = z_ext_record_oid(out, yaz_oid_recsyn_xml,
2284                                        rec_buf, rec_len);
2285     else
2286     {
2287         if (!record_last)
2288         {
2289             printf ("No last record (update ignored)\n");
2290             return 0;
2291         }
2292         record_this = record_last;
2293     }
2294
2295     req->packageType = odr_oiddup(out, (version == 0 ?
2296        yaz_oid_extserv_database_update_first_version :
2297        yaz_oid_extserv_database_update));
2298
2299     req->packageName = esPackageName;
2300
2301     req->referenceId = set_refid (out);
2302
2303     r = req->taskSpecificParameters = (Z_External *)
2304         odr_malloc (out, sizeof(*r));
2305     r->direct_reference = req->packageType;
2306     r->indirect_reference = 0;
2307     r->descriptor = 0;
2308     if (version == 0)
2309     {
2310         Z_IU0OriginPartToKeep *toKeep;
2311         Z_IU0SuppliedRecords *notToKeep;
2312
2313         r->which = Z_External_update0;
2314         r->u.update0 = (Z_IU0Update *) odr_malloc(out, sizeof(*r->u.update0));
2315         r->u.update0->which = Z_IUUpdate_esRequest;
2316         r->u.update0->u.esRequest = (Z_IU0UpdateEsRequest *)
2317             odr_malloc(out, sizeof(*r->u.update0->u.esRequest));
2318         toKeep = r->u.update0->u.esRequest->toKeep = (Z_IU0OriginPartToKeep *)
2319             odr_malloc(out, sizeof(*r->u.update0->u.esRequest->toKeep));
2320
2321         toKeep->databaseName = databaseNames[0];
2322         toKeep->schema = 0;
2323         if (record_schema)
2324         {
2325             toKeep->schema = yaz_string_to_oid_odr(yaz_oid_std(),
2326                                                    CLASS_SCHEMA,
2327                                                    record_schema, out);
2328         }
2329         toKeep->elementSetName = 0;
2330
2331         toKeep->action = (int *) odr_malloc(out, sizeof(*toKeep->action));
2332         *toKeep->action = action_no;
2333
2334         notToKeep = r->u.update0->u.esRequest->notToKeep = (Z_IU0SuppliedRecords *)
2335             odr_malloc(out, sizeof(*r->u.update0->u.esRequest->notToKeep));
2336         notToKeep->num = 1;
2337         notToKeep->elements = (Z_IU0SuppliedRecords_elem **)
2338             odr_malloc(out, sizeof(*notToKeep->elements));
2339         notToKeep->elements[0] = (Z_IU0SuppliedRecords_elem *)
2340             odr_malloc(out, sizeof(**notToKeep->elements));
2341         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
2342         if (recid)
2343         {
2344             notToKeep->elements[0]->u.opaque = (Odr_oct *)
2345                 odr_malloc (out, sizeof(Odr_oct));
2346             notToKeep->elements[0]->u.opaque->buf = (unsigned char *) recid;
2347             notToKeep->elements[0]->u.opaque->size = strlen(recid);
2348             notToKeep->elements[0]->u.opaque->len = strlen(recid);
2349         }
2350         else
2351             notToKeep->elements[0]->u.opaque = 0;
2352         notToKeep->elements[0]->supplementalId = 0;
2353         notToKeep->elements[0]->correlationInfo = 0;
2354         notToKeep->elements[0]->record = record_this;
2355     }
2356     else
2357     {
2358         Z_IUOriginPartToKeep *toKeep;
2359         Z_IUSuppliedRecords *notToKeep;
2360
2361         r->which = Z_External_update;
2362         r->u.update = (Z_IUUpdate *) odr_malloc(out, sizeof(*r->u.update));
2363         r->u.update->which = Z_IUUpdate_esRequest;
2364         r->u.update->u.esRequest = (Z_IUUpdateEsRequest *)
2365             odr_malloc(out, sizeof(*r->u.update->u.esRequest));
2366         toKeep = r->u.update->u.esRequest->toKeep = (Z_IUOriginPartToKeep *)
2367             odr_malloc(out, sizeof(*r->u.update->u.esRequest->toKeep));
2368
2369         toKeep->databaseName = databaseNames[0];
2370         toKeep->schema = 0;
2371         if (record_schema)
2372         {
2373             toKeep->schema = yaz_string_to_oid_odr(yaz_oid_std(),
2374                                                    CLASS_SCHEMA,
2375                                                    record_schema, out);
2376         }
2377         toKeep->elementSetName = 0;
2378         toKeep->actionQualifier = 0;
2379         toKeep->action = (int *) odr_malloc(out, sizeof(*toKeep->action));
2380         *toKeep->action = action_no;
2381
2382         notToKeep = r->u.update->u.esRequest->notToKeep = (Z_IUSuppliedRecords *)
2383             odr_malloc(out, sizeof(*r->u.update->u.esRequest->notToKeep));
2384         notToKeep->num = 1;
2385         notToKeep->elements = (Z_IUSuppliedRecords_elem **)
2386             odr_malloc(out, sizeof(*notToKeep->elements));
2387         notToKeep->elements[0] = (Z_IUSuppliedRecords_elem *)
2388             odr_malloc(out, sizeof(**notToKeep->elements));
2389         notToKeep->elements[0]->which = Z_IUSuppliedRecords_elem_opaque;
2390         if (recid)
2391         {
2392             notToKeep->elements[0]->u.opaque = (Odr_oct *)
2393                 odr_malloc (out, sizeof(Odr_oct));
2394             notToKeep->elements[0]->u.opaque->buf = (unsigned char *) recid;
2395             notToKeep->elements[0]->u.opaque->size = strlen(recid);
2396             notToKeep->elements[0]->u.opaque->len = strlen(recid);
2397         }
2398         else
2399             notToKeep->elements[0]->u.opaque = 0;
2400         notToKeep->elements[0]->supplementalId = 0;
2401         notToKeep->elements[0]->correlationInfo = 0;
2402         notToKeep->elements[0]->record = record_this;
2403     }
2404
2405     send_apdu(apdu);
2406
2407     return 2;
2408 }
2409
2410 static int cmd_xmles(const char *arg)
2411 {
2412     if (only_z3950())
2413         return 1;
2414     else
2415     {
2416         char *asn_buf = 0;
2417         int noread = 0;
2418         Odr_oid *oid;
2419         char oid_str[51];
2420         Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest);
2421         Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
2422
2423
2424         Z_External *ext = (Z_External *) odr_malloc(out, sizeof(*ext));
2425
2426         req->referenceId = set_refid (out);
2427         req->taskSpecificParameters = ext;
2428         ext->indirect_reference = 0;
2429         ext->descriptor = 0;
2430         ext->which = Z_External_octet;
2431         ext->u.single_ASN1_type = (Odr_oct *) odr_malloc (out, sizeof(Odr_oct));
2432         sscanf(arg, "%50s%n", oid_str, &noread);
2433         if (noread == 0)
2434         {
2435             printf("Missing OID for xmles\n");
2436             return 0;
2437         }
2438         arg += noread;
2439         if (parse_cmd_doc(&arg, out, &asn_buf,
2440                           &ext->u.single_ASN1_type->len) == 0)
2441             return 0;
2442
2443         ext->u.single_ASN1_type->buf = (unsigned char *) asn_buf;
2444
2445         oid = yaz_string_to_oid_odr(yaz_oid_std(),
2446                                     CLASS_EXTSERV, oid_str, out);
2447         if (!oid)
2448         {
2449             printf("Bad OID: %s\n", oid_str);
2450             return 0;
2451         }
2452
2453         req->packageType = oid;
2454
2455         ext->direct_reference = oid;
2456
2457         send_apdu(apdu);
2458
2459         return 2;
2460     }
2461 }
2462
2463 static int cmd_itemorder(const char *arg)
2464 {
2465     char type[12];
2466     int itemno;
2467     char *xml_buf = 0;
2468     int xml_len = 0;
2469     int no_read = 0;
2470
2471     if (only_z3950())
2472         return 1;
2473     if (sscanf(arg, "%10s %d%n", type, &itemno, &no_read) < 2)
2474         return 0;
2475     arg += no_read;
2476     parse_cmd_doc(&arg, out, &xml_buf, &xml_len);
2477
2478     fflush(stdout);
2479     send_itemorder(type, itemno, xml_buf, xml_len);
2480     return 2;
2481 }
2482
2483 static void show_opt(const char *arg, void *clientData)
2484 {
2485     printf ("%s ", arg);
2486 }
2487
2488 static int cmd_zversion(const char *arg)
2489 {
2490     if (*arg && arg)
2491         z3950_version = atoi(arg);
2492     else
2493         printf ("version is %d\n", z3950_version);
2494     return 0;
2495 }
2496
2497 static int cmd_options(const char *arg)
2498 {
2499     if (*arg)
2500     {
2501         int r;
2502         int pos;
2503         r = yaz_init_opt_encode(&z3950_options, arg, &pos);
2504         if (r == -1)
2505             printf("Unknown option(s) near %s\n", arg+pos);
2506     }
2507     else
2508     {
2509         yaz_init_opt_decode(&z3950_options, show_opt, 0);
2510         printf ("\n");
2511     }
2512     return 0;
2513 }
2514
2515 static int cmd_explain(const char *arg)
2516 {
2517     if (protocol != PROTO_HTTP)
2518         return 0;
2519 #if YAZ_HAVE_XML2
2520     if (!conn)
2521         session_connect(cur_host);
2522     if (conn)
2523     {
2524         Z_SRW_PDU *sr = 0;
2525
2526         setno = 1;
2527
2528         /* save this for later .. when fetching individual records */
2529         sr = yaz_srw_get(out, Z_SRW_explain_request);
2530         if (recordsyntax_size == 1
2531             && !yaz_matchstr(recordsyntax_list[0], "xml"))
2532             sr->u.explain_request->recordPacking = "xml";
2533         send_srw(sr);
2534         return 2;
2535     }
2536 #endif
2537     return 0;
2538 }
2539
2540 static int cmd_init(const char *arg)
2541 {
2542     if (*arg)
2543     {
2544         strncpy (cur_host, arg, sizeof(cur_host)-1);
2545         cur_host[sizeof(cur_host)-1] = 0;
2546     }
2547     if (only_z3950())
2548         return 1;
2549     send_initRequest(cur_host);
2550     return 2;
2551 }
2552
2553 static int cmd_sru(const char *arg)
2554 {
2555     if (!*arg)
2556     {
2557         printf("SRU method is: %s\n", sru_method);
2558         printf("SRU version is: %s\n", sru_version);
2559     }
2560     else
2561     {
2562         int r;
2563         r = sscanf(arg, "%9s %9s", sru_method, sru_version);
2564         if (r >= 1)
2565         {
2566             if (!yaz_matchstr(sru_method, "post"))
2567                 ;
2568             else if (!yaz_matchstr(sru_method, "get"))
2569                 ;
2570             else if (!yaz_matchstr(sru_method, "soap"))
2571                 ;
2572             else
2573             {
2574                 strcpy(sru_method, "soap");
2575                 printf("Unknown SRU method: %s\n", arg);
2576                 printf("Specify one of POST, GET, SOAP\n");
2577             }
2578         }
2579     }
2580     return 0;
2581 }
2582
2583 static int cmd_find(const char *arg)
2584 {
2585     if (!*arg)
2586     {
2587         printf("Find what?\n");
2588         return 0;
2589     }
2590     if (protocol == PROTO_HTTP)
2591     {
2592 #if YAZ_HAVE_XML2
2593         if (!conn)
2594             session_connect(cur_host);
2595         if (!conn)
2596             return 0;
2597         if (!send_SRW_searchRequest(arg))
2598             return 0;
2599 #else
2600         return 0;
2601 #endif
2602     }
2603     else
2604     {
2605         if (!conn)
2606         {
2607             try_reconnect();
2608
2609             if (!conn) {
2610                 printf("Not connected yet\n");
2611                 return 0;
2612             }
2613         }
2614         if (!send_searchRequest(arg))
2615             return 0;
2616     }
2617     return 2;
2618 }
2619
2620 static int cmd_delete(const char *arg)
2621 {
2622     if (only_z3950())
2623         return 0;
2624     if (!send_deleteResultSetRequest(arg))
2625         return 0;
2626     return 2;
2627 }
2628
2629 static int cmd_ssub(const char *arg)
2630 {
2631     if (!(smallSetUpperBound = atoi(arg)))
2632         return 0;
2633     return 1;
2634 }
2635
2636 static int cmd_lslb(const char *arg)
2637 {
2638     if (!(largeSetLowerBound = atoi(arg)))
2639         return 0;
2640     return 1;
2641 }
2642
2643 static int cmd_mspn(const char *arg)
2644 {
2645     if (!(mediumSetPresentNumber = atoi(arg)))
2646         return 0;
2647     return 1;
2648 }
2649
2650 static int cmd_status(const char *arg)
2651 {
2652     printf("smallSetUpperBound: %d\n", smallSetUpperBound);
2653     printf("largeSetLowerBound: %d\n", largeSetLowerBound);
2654     printf("mediumSetPresentNumber: %d\n", mediumSetPresentNumber);
2655     return 1;
2656 }
2657
2658 static int cmd_setnames(const char *arg)
2659 {
2660     if (*arg == '1')         /* enable ? */
2661         setnumber = 0;
2662     else if (*arg == '0')    /* disable ? */
2663         setnumber = -1;
2664     else if (setnumber < 0)  /* no args, toggle .. */
2665         setnumber = 0;
2666     else
2667         setnumber = -1;
2668
2669     if (setnumber >= 0)
2670         printf("Set numbering enabled.\n");
2671     else
2672         printf("Set numbering disabled.\n");
2673     return 1;
2674 }
2675
2676 /* PRESENT SERVICE ----------------------------- */
2677
2678 static void parse_show_args(const char *arg_c, char *setstring,
2679                             int *start, int *number)
2680 {
2681     char arg[40];
2682     char *p;
2683
2684     strncpy(arg, arg_c, sizeof(arg)-1);
2685     arg[sizeof(arg)-1] = '\0';
2686
2687     if ((p = strchr(arg, '+')))
2688     {
2689         *number = atoi(p + 1);
2690         *p = '\0';
2691     }
2692     if (*arg)
2693     {
2694         if (!strcmp(arg, "all"))
2695         {
2696             *number = last_hit_count;
2697             *start = 1;
2698         }
2699         else
2700             *start = atoi(arg);
2701     }
2702     if (p && (p=strchr(p+1, '+')))
2703         strcpy (setstring, p+1);
2704     else if (setnumber >= 0)
2705         sprintf(setstring, "%d", setnumber);
2706     else
2707         *setstring = '\0';
2708 }
2709
2710 static int send_presentRequest(const char *arg)
2711 {
2712     Z_APDU *apdu = zget_APDU(out, Z_APDU_presentRequest);
2713     Z_PresentRequest *req = apdu->u.presentRequest;
2714     Z_RecordComposition compo;
2715     int nos = 1;
2716     char setstring[100];
2717
2718     req->referenceId = set_refid (out);
2719
2720     parse_show_args(arg, setstring, &setno, &nos);
2721     if (*setstring)
2722         req->resultSetId = setstring;
2723
2724     req->resultSetStartPoint = &setno;
2725     req->numberOfRecordsRequested = &nos;
2726
2727     if (recordsyntax_size)
2728         req->preferredRecordSyntax =
2729             yaz_string_to_oid_odr(yaz_oid_std(),
2730                                   CLASS_RECSYN, recordsyntax_list[0], out);
2731
2732     if (record_schema || recordsyntax_size >= 2)
2733     {
2734         req->recordComposition = &compo;
2735         compo.which = Z_RecordComp_complex;
2736         compo.u.complex = (Z_CompSpec *)
2737             odr_malloc(out, sizeof(*compo.u.complex));
2738         compo.u.complex->selectAlternativeSyntax = (bool_t *)
2739             odr_malloc(out, sizeof(bool_t));
2740         *compo.u.complex->selectAlternativeSyntax = 0;
2741
2742         compo.u.complex->generic = (Z_Specification *)
2743             odr_malloc(out, sizeof(*compo.u.complex->generic));
2744
2745         compo.u.complex->generic->which = Z_Schema_oid;
2746         if (!record_schema)
2747             compo.u.complex->generic->schema.oid = 0;
2748         else
2749         {
2750             compo.u.complex->generic->schema.oid =
2751                 yaz_string_to_oid_odr(yaz_oid_std(),
2752                                       CLASS_SCHEMA, record_schema, out);
2753
2754             if (!compo.u.complex->generic->schema.oid)
2755             {
2756                 /* OID wasn't a schema! Try record syntax instead. */
2757                 compo.u.complex->generic->schema.oid = (Odr_oid *)
2758                     yaz_string_to_oid_odr(yaz_oid_std(),
2759                                           CLASS_RECSYN, record_schema, out);
2760             }
2761         }
2762         if (!elementSetNames)
2763             compo.u.complex->generic->elementSpec = 0;
2764         else
2765         {
2766             compo.u.complex->generic->elementSpec = (Z_ElementSpec *)
2767                 odr_malloc(out, sizeof(Z_ElementSpec));
2768             compo.u.complex->generic->elementSpec->which =
2769                 Z_ElementSpec_elementSetName;
2770             compo.u.complex->generic->elementSpec->u.elementSetName =
2771                 elementSetNames->u.generic;
2772         }
2773         compo.u.complex->num_dbSpecific = 0;
2774         compo.u.complex->dbSpecific = 0;
2775
2776         compo.u.complex->num_recordSyntax = 0;
2777         compo.u.complex->recordSyntax = 0;
2778         if (recordsyntax_size >= 2)
2779         {
2780             int i;
2781             compo.u.complex->num_recordSyntax = recordsyntax_size;
2782             compo.u.complex->recordSyntax = (Odr_oid **)
2783                 odr_malloc(out, recordsyntax_size * sizeof(Odr_oid*));
2784             for (i = 0; i < recordsyntax_size; i++)
2785             compo.u.complex->recordSyntax[i] =
2786                 yaz_string_to_oid_odr(yaz_oid_std(),
2787                                       CLASS_RECSYN, recordsyntax_list[i], out);
2788         }
2789     }
2790     else if (elementSetNames)
2791     {
2792         req->recordComposition = &compo;
2793         compo.which = Z_RecordComp_simple;
2794         compo.u.simple = elementSetNames;
2795     }
2796     send_apdu(apdu);
2797     printf("Sent presentRequest (%d+%d).\n", setno, nos);
2798     return 2;
2799 }
2800
2801 #if YAZ_HAVE_XML2
2802 static int send_SRW_presentRequest(const char *arg)
2803 {
2804     char setstring[100];
2805     int nos = 1;
2806     Z_SRW_PDU *sr = srw_sr;
2807
2808     if (!sr)
2809         return 0;
2810     parse_show_args(arg, setstring, &setno, &nos);
2811     sr->u.request->startRecord = odr_intdup(out, setno);
2812     sr->u.request->maximumRecords = odr_intdup(out, nos);
2813     if (record_schema)
2814         sr->u.request->recordSchema = record_schema;
2815     if (recordsyntax_size == 1 && !yaz_matchstr(recordsyntax_list[0], "xml"))
2816         sr->u.request->recordPacking = "xml";
2817     return send_srw(sr);
2818 }
2819 #endif
2820
2821 static void close_session (void)
2822 {
2823     if (conn)
2824         cs_close (conn);
2825     conn = 0;
2826     sent_close = 0;
2827     odr_reset(out);
2828     odr_reset(in);
2829     odr_reset(print);
2830     last_hit_count = 0;
2831 }
2832
2833 void process_close(Z_Close *req)
2834 {
2835     Z_APDU *apdu = zget_APDU(out, Z_APDU_close);
2836     Z_Close *res = apdu->u.close;
2837
2838     static char *reasons[] =
2839     {
2840         "finished",
2841         "shutdown",
2842         "system problem",
2843         "cost limit reached",
2844         "resources",
2845         "security violation",
2846         "protocolError",
2847         "lack of activity",
2848         "peer abort",
2849         "unspecified"
2850     };
2851
2852     printf("Reason: %s, message: %s\n", reasons[*req->closeReason],
2853         req->diagnosticInformation ? req->diagnosticInformation : "NULL");
2854     if (sent_close)
2855         close_session ();
2856     else
2857     {
2858         *res->closeReason = Z_Close_finished;
2859         send_apdu(apdu);
2860         printf("Sent response.\n");
2861         sent_close = 1;
2862     }
2863 }
2864
2865 static int cmd_show(const char *arg)
2866 {
2867     if (protocol == PROTO_HTTP)
2868     {
2869 #if YAZ_HAVE_XML2
2870         if (!conn)
2871             session_connect(cur_host);
2872         if (!conn)
2873             return 0;
2874         if (!send_SRW_presentRequest(arg))
2875             return 0;
2876 #else
2877         return 0;
2878 #endif
2879     }
2880     else
2881     {
2882         if (!conn)
2883         {
2884             printf("Not connected yet\n");
2885             return 0;
2886         }
2887         if (!send_presentRequest(arg))
2888             return 0;
2889     }
2890     return 2;
2891 }
2892
2893 void exit_client(int code)
2894 {
2895     file_history_save(file_history);
2896     file_history_destroy(&file_history);
2897     exit(code);
2898 }
2899
2900 int cmd_quit(const char *arg)
2901 {
2902     printf("See you later, alligator.\n");
2903     xmalloc_trav ("");
2904     exit_client(0);
2905     return 0;
2906 }
2907
2908 int cmd_cancel(const char *arg)
2909 {
2910     Z_APDU *apdu = zget_APDU(out, Z_APDU_triggerResourceControlRequest);
2911     Z_TriggerResourceControlRequest *req =
2912         apdu->u.triggerResourceControlRequest;
2913     bool_t rfalse = 0;
2914     char command[16];
2915
2916     *command = '\0';
2917     sscanf(arg, "%15s", command);
2918
2919     if (only_z3950())
2920         return 0;
2921     if (session_initResponse &&
2922         !ODR_MASK_GET(session_initResponse->options,
2923                       Z_Options_triggerResourceCtrl))
2924     {
2925         printf("Target doesn't support cancel (trigger resource ctrl)\n");
2926         return 0;
2927     }
2928     *req->requestedAction = Z_TriggerResourceControlRequest_cancel;
2929     req->resultSetWanted = &rfalse;
2930     req->referenceId = set_refid (out);
2931
2932     send_apdu(apdu);
2933     printf("Sent cancel request\n");
2934     if (!strcmp(command, "wait"))
2935          return 2;
2936     return 1;
2937 }
2938
2939
2940 int cmd_cancel_find(const char *arg) {
2941     int fres;
2942     fres=cmd_find(arg);
2943     if( fres > 0 ) {
2944         return cmd_cancel("");
2945     };
2946     return fres;
2947 }
2948
2949 int send_scanrequest(const char *set,  const char *query,
2950                      int pp, int num, const char *term)
2951 {
2952     Z_APDU *apdu = zget_APDU(out, Z_APDU_scanRequest);
2953     Z_ScanRequest *req = apdu->u.scanRequest;
2954
2955     if (only_z3950())
2956         return 0;
2957     if (queryType == QueryType_CCL2RPN)
2958     {
2959         int error, pos;
2960         struct ccl_rpn_node *rpn;
2961
2962         rpn = ccl_find_str (bibset,  query, &error, &pos);
2963         if (error)
2964         {
2965             printf("CCL ERROR: %s\n", ccl_err_msg(error));
2966             return -1;
2967         }
2968         req->attributeSet =
2969             yaz_string_to_oid_odr(yaz_oid_std(),
2970                                   CLASS_ATTSET, "Bib-1", out);
2971         if (!(req->termListAndStartPoint = ccl_scan_query (out, rpn)))
2972         {
2973             printf("Couldn't convert CCL to Scan term\n");
2974             return -1;
2975         }
2976         ccl_rpn_delete (rpn);
2977     }
2978     else
2979     {
2980         YAZ_PQF_Parser pqf_parser = yaz_pqf_create ();
2981
2982         if (!(req->termListAndStartPoint =
2983               yaz_pqf_scan(pqf_parser, out, &req->attributeSet, query)))
2984         {
2985             const char *pqf_msg;
2986             size_t off;
2987             int code = yaz_pqf_error (pqf_parser, &pqf_msg, &off);
2988             int ioff = off;
2989             printf("%*s^\n", ioff+7, "");
2990             printf("Prefix query error: %s (code %d)\n", pqf_msg, code);
2991             yaz_pqf_destroy (pqf_parser);
2992             return -1;
2993         }
2994         yaz_pqf_destroy (pqf_parser);
2995     }
2996     if (queryCharset && outputCharset)
2997     {
2998         yaz_iconv_t cd = yaz_iconv_open(queryCharset, outputCharset);
2999         if (!cd)
3000         {
3001             printf("Conversion from %s to %s unsupported\n",
3002                    outputCharset, queryCharset);
3003             return -1;
3004         }
3005         yaz_query_charset_convert_apt(req->termListAndStartPoint, out, cd);
3006         yaz_iconv_close(cd);
3007     }
3008     if (term && *term)
3009     {
3010         if (req->termListAndStartPoint->term &&
3011             req->termListAndStartPoint->term->which == Z_Term_general &&
3012             req->termListAndStartPoint->term->u.general)
3013         {
3014             req->termListAndStartPoint->term->u.general->buf =
3015                 (unsigned char *) odr_strdup(out, term);
3016             req->termListAndStartPoint->term->u.general->len =
3017                 req->termListAndStartPoint->term->u.general->size =
3018                 strlen(term);
3019         }
3020     }
3021     req->referenceId = set_refid (out);
3022     req->num_databaseNames = num_databaseNames;
3023     req->databaseNames = databaseNames;
3024     req->numberOfTermsRequested = &num;
3025     req->preferredPositionInResponse = &pp;
3026     req->stepSize = odr_intdup(out, scan_stepSize);
3027
3028     if (set)
3029         yaz_oi_set_string_oid(&req->otherInfo, out,
3030                               yaz_oid_userinfo_scan_set, 1, set);
3031
3032     send_apdu(apdu);
3033     return 2;
3034 }
3035
3036 int send_sortrequest(const char *arg, int newset)
3037 {
3038     Z_APDU *apdu = zget_APDU(out, Z_APDU_sortRequest);
3039     Z_SortRequest *req = apdu->u.sortRequest;
3040     Z_SortKeySpecList *sksl = (Z_SortKeySpecList *)
3041         odr_malloc (out, sizeof(*sksl));
3042     char setstring[32];
3043
3044     if (only_z3950())
3045         return 0;
3046     if (setnumber >= 0)
3047         sprintf (setstring, "%d", setnumber);
3048     else
3049         sprintf (setstring, "default");
3050
3051     req->referenceId = set_refid (out);
3052
3053     req->num_inputResultSetNames = 1;
3054     req->inputResultSetNames = (Z_InternationalString **)
3055         odr_malloc (out, sizeof(*req->inputResultSetNames));
3056     req->inputResultSetNames[0] = odr_strdup (out, setstring);
3057
3058     if (newset && setnumber >= 0)
3059         sprintf (setstring, "%d", ++setnumber);
3060
3061     req->sortedResultSetName = odr_strdup (out, setstring);
3062
3063     req->sortSequence = yaz_sort_spec (out, arg);
3064     if (!req->sortSequence)
3065     {
3066         printf ("Missing sort specifications\n");
3067         return -1;
3068     }
3069     send_apdu(apdu);
3070     return 2;
3071 }
3072
3073 void display_term(Z_TermInfo *t)
3074 {
3075     if (t->displayTerm)
3076         printf("%s", t->displayTerm);
3077     else if (t->term->which == Z_Term_general)
3078         printf("%.*s", t->term->u.general->len, t->term->u.general->buf);
3079     else
3080         printf("Term (not general)");
3081     if (t->term->which == Z_Term_general)
3082         sprintf(last_scan_line, "%.*s", t->term->u.general->len,
3083             t->term->u.general->buf);
3084
3085     if (t->globalOccurrences)
3086         printf (" (%d)\n", *t->globalOccurrences);
3087     else
3088         printf ("\n");
3089 }
3090
3091 void process_scanResponse(Z_ScanResponse *res)
3092 {
3093     int i;
3094     Z_Entry **entries = NULL;
3095     int num_entries = 0;
3096
3097     printf("Received ScanResponse\n");
3098     print_refid (res->referenceId);
3099     printf("%d entries", *res->numberOfEntriesReturned);
3100     if (res->positionOfTerm)
3101         printf (", position=%d", *res->positionOfTerm);
3102     printf ("\n");
3103     if (*res->scanStatus != Z_Scan_success)
3104         printf("Scan returned code %d\n", *res->scanStatus);
3105     if (!res->entries)
3106         return;
3107     if ((entries = res->entries->entries))
3108         num_entries = res->entries->num_entries;
3109     for (i = 0; i < num_entries; i++)
3110     {
3111         int pos_term = res->positionOfTerm ? *res->positionOfTerm : -1;
3112         if (entries[i]->which == Z_Entry_termInfo)
3113         {
3114             printf("%c ", i + 1 == pos_term ? '*' : ' ');
3115             display_term(entries[i]->u.termInfo);
3116         }
3117         else
3118             display_diagrecs(&entries[i]->u.surrogateDiagnostic, 1);
3119     }
3120     if (res->entries->nonsurrogateDiagnostics)
3121         display_diagrecs (res->entries->nonsurrogateDiagnostics,
3122                           res->entries->num_nonsurrogateDiagnostics);
3123 }
3124
3125 void process_sortResponse(Z_SortResponse *res)
3126 {
3127     printf("Received SortResponse: status=");
3128     switch (*res->sortStatus)
3129     {
3130     case Z_SortResponse_success:
3131         printf ("success"); break;
3132     case Z_SortResponse_partial_1:
3133         printf ("partial"); break;
3134     case Z_SortResponse_failure:
3135         printf ("failure"); break;
3136     default:
3137         printf ("unknown (%d)", *res->sortStatus);
3138     }
3139     printf ("\n");
3140     print_refid (res->referenceId);
3141     if (res->diagnostics)
3142         display_diagrecs(res->diagnostics,
3143                          res->num_diagnostics);
3144 }
3145
3146 void process_deleteResultSetResponse (Z_DeleteResultSetResponse *res)
3147 {
3148     printf("Got deleteResultSetResponse status=%d\n",
3149            *res->deleteOperationStatus);
3150     if (res->deleteListStatuses)
3151     {
3152         int i;
3153         for (i = 0; i < res->deleteListStatuses->num; i++)
3154         {
3155             printf ("%s status=%d\n", res->deleteListStatuses->elements[i]->id,
3156                     *res->deleteListStatuses->elements[i]->status);
3157         }
3158     }
3159 }
3160
3161 int cmd_sort_generic(const char *arg, int newset)
3162 {
3163     if (only_z3950())
3164         return 0;
3165     if (session_initResponse &&
3166         !ODR_MASK_GET(session_initResponse->options, Z_Options_sort))
3167     {
3168         printf("Target doesn't support sort\n");
3169         return 0;
3170     }
3171     if (*arg)
3172     {
3173         if (send_sortrequest(arg, newset) < 0)
3174             return 0;
3175         return 2;
3176     }
3177     return 0;
3178 }
3179
3180 int cmd_sort(const char *arg)
3181 {
3182     return cmd_sort_generic (arg, 0);
3183 }
3184
3185 int cmd_sort_newset (const char *arg)
3186 {
3187     return cmd_sort_generic (arg, 1);
3188 }
3189
3190 int cmd_scanstep(const char *arg)
3191 {
3192     scan_stepSize = atoi(arg);
3193     return 0;
3194 }
3195
3196 int cmd_scanpos(const char *arg)
3197 {
3198     int r = sscanf(arg, "%d", &scan_position);
3199     if (r == 0)
3200         scan_position = 1;
3201     return 0;
3202 }
3203
3204 int cmd_scansize(const char *arg)
3205 {
3206     int r = sscanf(arg, "%d", &scan_size);
3207     if (r == 0)
3208         scan_size = 20;
3209     return 0;
3210 }
3211
3212 static int cmd_scan_common(const char *set, const char *arg)
3213 {
3214     if (protocol == PROTO_HTTP)
3215     {
3216 #if YAZ_HAVE_XML2
3217         if (!conn)
3218             session_connect(cur_host);
3219         if (!conn)
3220             return 0;
3221         if (*arg)
3222         {
3223             if (send_SRW_scanRequest(arg, scan_position, scan_size) < 0)
3224                 return 0;
3225         }
3226         else
3227         {
3228             if (send_SRW_scanRequest(last_scan_line, 1, scan_size) < 0)
3229                 return 0;
3230         }
3231         return 2;
3232 #else
3233         return 0;
3234 #endif
3235     }
3236     else
3237     {
3238         if (!conn)
3239         {
3240             try_reconnect();
3241
3242             if (!conn) {
3243                 printf("Session not initialized yet\n");
3244                 return 0;
3245             }
3246         }
3247         if (session_initResponse &&
3248             !ODR_MASK_GET(session_initResponse->options, Z_Options_scan))
3249         {
3250             printf("Target doesn't support scan\n");
3251             return 0;
3252         }
3253         if (*arg)
3254         {
3255             strcpy (last_scan_query, arg);
3256             if (send_scanrequest(set, arg,
3257                                  scan_position, scan_size, 0) < 0)
3258                 return 0;
3259         }
3260         else
3261         {
3262             if (send_scanrequest(set, last_scan_query,
3263                                  1, scan_size, last_scan_line) < 0)
3264                 return 0;
3265         }
3266         return 2;
3267     }
3268 }
3269
3270 int cmd_scan(const char *arg)
3271 {
3272     return cmd_scan_common(0, arg);
3273 }
3274
3275 int cmd_setscan(const char *arg)
3276 {
3277     char setstring[100];
3278     int nor;
3279     if (sscanf(arg, "%99s%n", setstring, &nor) < 1)
3280     {
3281         printf("missing set for setscan\n");
3282         return 0;
3283     }
3284     return cmd_scan_common(setstring, arg + nor);
3285 }
3286
3287 int cmd_schema(const char *arg)
3288 {
3289     xfree(record_schema);
3290     record_schema = 0;
3291     if (arg && *arg)
3292         record_schema = xstrdup(arg);
3293     return 1;
3294 }
3295
3296 int cmd_format(const char *arg)
3297 {
3298     const char *cp = arg;
3299     int nor;
3300     int idx = 0;
3301     int i;
3302     char form_str[41];
3303     if (!arg || !*arg)
3304     {
3305         printf("Usage: format <recordsyntax>\n");
3306         return 0;
3307     }
3308     for (i = 0; i < recordsyntax_size; i++)
3309     {
3310         xfree(recordsyntax_list[i]);
3311         recordsyntax_list[i] = 0;
3312     }
3313
3314     while (sscanf(cp, "%40s%n", form_str, &nor) >= 1 && nor > 0
3315            && idx < RECORDSYNTAX_MAX)
3316     {
3317         if (!strcmp(form_str, "none"))
3318             break;
3319         recordsyntax_list[idx] = xstrdup(form_str);
3320         cp += nor;
3321         idx++;
3322     }
3323     recordsyntax_size = idx;
3324     return 1;
3325 }
3326
3327 int cmd_elements(const char *arg)
3328 {
3329     static Z_ElementSetNames esn;
3330     static char what[100];
3331
3332     if (!arg || !*arg)
3333     {
3334         elementSetNames = 0;
3335         return 1;
3336     }
3337     strcpy(what, arg);
3338     esn.which = Z_ElementSetNames_generic;
3339     esn.u.generic = what;
3340     elementSetNames = &esn;
3341     return 1;
3342 }
3343
3344 int cmd_querytype (const char *arg)
3345 {
3346     if (!strcmp (arg, "ccl"))
3347         queryType = QueryType_CCL;
3348     else if (!strcmp (arg, "prefix") || !strcmp(arg, "rpn"))
3349         queryType = QueryType_Prefix;
3350     else if (!strcmp (arg, "ccl2rpn") || !strcmp (arg, "cclrpn"))
3351         queryType = QueryType_CCL2RPN;
3352     else if (!strcmp(arg, "cql"))
3353         queryType = QueryType_CQL;
3354     else if (!strcmp (arg, "cql2rpn") || !strcmp (arg, "cqlrpn"))
3355         queryType = QueryType_CQL2RPN;
3356     else
3357     {
3358         printf ("Querytype must be one of:\n");
3359         printf (" prefix         - Prefix query\n");
3360         printf (" ccl            - CCL query\n");
3361         printf (" ccl2rpn        - CCL query converted to RPN\n");
3362         printf (" cql            - CQL\n");
3363         printf (" cql2rpn        - CQL query converted to RPN\n");
3364         return 0;
3365     }
3366     return 1;
3367 }
3368
3369 int cmd_refid (const char *arg)
3370 {
3371     xfree (refid);
3372     refid = NULL;
3373     if (*arg)
3374         refid = xstrdup (arg);
3375     return 1;
3376 }
3377
3378 int cmd_close(const char *arg)
3379 {
3380     Z_APDU *apdu;
3381     Z_Close *req;
3382     if (only_z3950())
3383         return 0;
3384     apdu = zget_APDU(out, Z_APDU_close);
3385     req = apdu->u.close;
3386     *req->closeReason = Z_Close_finished;
3387     send_apdu(apdu);
3388     printf("Sent close request.\n");
3389     sent_close = 1;
3390     return 2;
3391 }
3392
3393 int cmd_packagename(const char* arg)
3394 {
3395     xfree (esPackageName);
3396     esPackageName = NULL;
3397     if (*arg)
3398         esPackageName = xstrdup(arg);
3399     return 1;
3400 }
3401
3402 int cmd_proxy(const char* arg)
3403 {
3404     xfree(yazProxy);
3405     yazProxy = 0;
3406     if (*arg)
3407         yazProxy = xstrdup (arg);
3408     return 1;
3409 }
3410
3411 int cmd_marccharset(const char *arg)
3412 {
3413     char l1[30];
3414
3415     *l1 = 0;
3416     if (sscanf(arg, "%29s", l1) < 1)
3417     {
3418         printf("MARC character set is `%s'\n",
3419                marcCharset ? marcCharset: "none");
3420         return 1;
3421     }
3422     xfree (marcCharset);
3423     marcCharset = 0;
3424     if (strcmp(l1, "-") && strcmp(l1, "none"))
3425         marcCharset = xstrdup(l1);
3426     return 1;
3427 }
3428
3429 int cmd_querycharset(const char *arg)
3430 {
3431     char l1[30];
3432
3433     *l1 = 0;
3434     if (sscanf(arg, "%29s", l1) < 1)
3435     {
3436         printf("Query character set is `%s'\n",
3437                queryCharset ? queryCharset: "none");
3438         return 1;
3439     }
3440     xfree (queryCharset);
3441     queryCharset = 0;
3442     if (strcmp(l1, "-") && strcmp(l1, "none"))
3443         queryCharset = xstrdup(l1);
3444     return 1;
3445 }
3446
3447 int cmd_displaycharset(const char *arg)
3448 {
3449     char l1[30];
3450
3451     *l1 = 0;
3452     if (sscanf(arg, "%29s", l1) < 1)
3453     {
3454         printf("Display character set is `%s'\n",
3455                outputCharset ? outputCharset: "none");
3456     }
3457     else
3458     {
3459         xfree (outputCharset);
3460         outputCharset = 0;
3461         if (!strcmp(l1, "auto") && codeset)
3462         {
3463             if (codeset)
3464             {
3465                 printf ("Display character set: %s\n", codeset);
3466                 outputCharset = xstrdup(codeset);
3467             }
3468             else
3469                 printf ("No codeset found on this system\n");
3470         }
3471         else if (strcmp(l1, "-") && strcmp(l1, "none"))
3472             outputCharset = xstrdup(l1);
3473     }
3474     return 1;
3475 }
3476
3477 int cmd_negcharset(const char *arg)
3478 {
3479     char l1[30];
3480
3481     *l1 = 0;
3482     if (sscanf(arg, "%29s %d %d", l1, &negotiationCharsetRecords,
3483                &negotiationCharsetVersion) < 1)
3484     {
3485         printf("Negotiation character set `%s'\n",
3486                negotiationCharset ? negotiationCharset: "none");
3487         if (negotiationCharset)
3488         {
3489             printf("Records in charset %s\n", negotiationCharsetRecords ?
3490                    "yes" : "no");
3491             printf("Charneg version %d\n", negotiationCharsetVersion);
3492         }
3493     }
3494     else
3495     {
3496         xfree (negotiationCharset);
3497         negotiationCharset = NULL;
3498         if (*l1 && strcmp(l1, "-") && strcmp(l1, "none"))
3499         {
3500             negotiationCharset = xstrdup(l1);
3501             printf ("Character set negotiation : %s\n", negotiationCharset);
3502         }
3503     }
3504     return 1;
3505 }
3506
3507 int cmd_charset(const char* arg)
3508 {
3509     char l1[30], l2[30], l3[30], l4[30];
3510
3511     *l1 = *l2 = *l3 = *l4 = '\0';
3512     if (sscanf(arg, "%29s %29s %29s %29s", l1, l2, l3, l4) < 1)
3513     {
3514         cmd_negcharset("");
3515         cmd_displaycharset("");
3516         cmd_marccharset("");
3517         cmd_querycharset("");
3518     }
3519     else
3520     {
3521         cmd_negcharset(l1);
3522         if (*l2)
3523             cmd_displaycharset(l2);
3524         if (*l3)
3525             cmd_marccharset(l3);
3526         if (*l4)
3527             cmd_querycharset(l4);
3528     }
3529     return 1;
3530 }
3531
3532 int cmd_lang(const char* arg)
3533 {
3534     if (*arg == '\0') {
3535         printf("Current language is `%s'\n", yazLang ? yazLang : "none");
3536         return 1;
3537     }
3538     xfree (yazLang);
3539     yazLang = NULL;
3540     if (*arg)
3541         yazLang = xstrdup(arg);
3542     return 1;
3543 }
3544
3545 int cmd_source(const char* arg, int echo )
3546 {
3547     /* first should open the file and read one line at a time.. */
3548     FILE* includeFile;
3549     char line[102400], *cp;
3550
3551     if(strlen(arg)<1) {
3552         fprintf(stderr,"Error in source command use a filename\n");
3553         return -1;
3554     }
3555
3556     includeFile = fopen (arg, "r");
3557
3558     if(!includeFile) {
3559         fprintf(stderr,"Unable to open file %s for reading\n",arg);
3560         return -1;
3561     }
3562
3563     while(!feof(includeFile)) {
3564         memset(line,0,sizeof(line));
3565         fgets(line,sizeof(line),includeFile);
3566
3567         if(strlen(line) < 2) continue;
3568         if(line[0] == '#') continue;
3569
3570         if ((cp = strrchr (line, '\n')))
3571             *cp = '\0';
3572
3573         if( echo ) {
3574             printf( "processing line: %s\n",line );
3575         };
3576         process_cmd_line(line);
3577     }
3578
3579     if(fclose(includeFile)<0) {
3580         perror("unable to close include file");
3581         exit(1);
3582     }
3583     return 1;
3584 }
3585
3586 int cmd_source_echo(const char* arg)
3587 {
3588     cmd_source(arg, 1);
3589     return 1;
3590 }
3591
3592 int cmd_source_noecho(const char* arg)
3593 {
3594     cmd_source(arg, 0);
3595     return 1;
3596 }
3597
3598
3599 int cmd_subshell(const char* args)
3600 {
3601     if(strlen(args))
3602         system(args);
3603     else
3604         system(getenv("SHELL"));
3605
3606     printf("\n");
3607     return 1;
3608 }
3609
3610 int cmd_set_berfile(const char *arg)
3611 {
3612     if (ber_file && ber_file != stdout && ber_file != stderr)
3613         fclose(ber_file);
3614     if (!strcmp(arg, ""))
3615         ber_file = 0;
3616     else if (!strcmp(arg, "-"))
3617         ber_file = stdout;
3618     else
3619         ber_file = fopen(arg, "a");
3620     return 1;
3621 }
3622
3623 int cmd_set_apdufile(const char *arg)
3624 {
3625     if(apdu_file && apdu_file != stderr && apdu_file != stderr)
3626         fclose(apdu_file);
3627     if (!strcmp(arg, ""))
3628         apdu_file = 0;
3629     else if (!strcmp(arg, "-"))
3630         apdu_file = stderr;
3631     else
3632     {
3633         apdu_file = fopen(arg, "a");
3634         if (!apdu_file)
3635             perror("unable to open apdu log file");
3636     }
3637     if (apdu_file)
3638         odr_setprint(print, apdu_file);
3639     return 1;
3640 }
3641
3642 int cmd_set_cclfile(const char* arg)
3643 {
3644     FILE *inf;
3645
3646     bibset = ccl_qual_mk ();
3647     inf = fopen (arg, "r");
3648     if (!inf)
3649         perror("unable to open CCL file");
3650     else
3651     {
3652         ccl_qual_file (bibset, inf);
3653         fclose (inf);
3654     }
3655     strcpy(ccl_fields,arg);
3656     return 0;
3657 }
3658
3659 int cmd_set_cqlfile(const char* arg)
3660 {
3661     cql_transform_t newcqltrans;
3662
3663     if ((newcqltrans = cql_transform_open_fname(arg)) == 0) {
3664         perror("unable to open CQL file");
3665         return 0;
3666     }
3667     if (cqltrans != 0)
3668         cql_transform_close(cqltrans);
3669
3670     cqltrans = newcqltrans;
3671     strcpy(cql_fields, arg);
3672     return 0;
3673 }
3674
3675 int cmd_set_auto_reconnect(const char* arg)
3676 {
3677     if(strlen(arg)==0) {
3678         auto_reconnect = ! auto_reconnect;
3679     } else if(strcmp(arg,"on")==0) {
3680         auto_reconnect = 1;
3681     } else if(strcmp(arg,"off")==0) {
3682         auto_reconnect = 0;
3683     } else {
3684         printf("Error use on or off\n");
3685         return 1;
3686     }
3687
3688     if (auto_reconnect)
3689         printf("Set auto reconnect enabled.\n");
3690     else
3691         printf("Set auto reconnect disabled.\n");
3692
3693     return 0;
3694 }
3695
3696
3697 int cmd_set_auto_wait(const char* arg)
3698 {
3699     if(strlen(arg)==0) {
3700         auto_wait = ! auto_wait;
3701     } else if(strcmp(arg,"on")==0) {
3702         auto_wait = 1;
3703     } else if(strcmp(arg,"off")==0) {
3704         auto_wait = 0;
3705     } else {
3706         printf("Error use on or off\n");
3707         return 1;
3708     }
3709
3710     if (auto_wait)
3711         printf("Set auto wait enabled.\n");
3712     else
3713         printf("Set auto wait disabled.\n");
3714
3715     return 0;
3716 }
3717
3718 int cmd_set_marcdump(const char* arg)
3719 {
3720     if(marc_file && marc_file != stderr) { /* don't close stdout*/
3721         fclose(marc_file);
3722     }
3723
3724     if (!strcmp(arg, ""))
3725         marc_file = 0;
3726     else if (!strcmp(arg, "-"))
3727         marc_file = stderr;
3728     else
3729     {
3730         marc_file = fopen(arg, "a");
3731         if (!marc_file)
3732             perror("unable to open marc log file");
3733     }
3734     return 1;
3735 }
3736
3737 /*
3738    this command takes 3 arge {name class oid}
3739 */
3740 int cmd_register_oid(const char* args) {
3741     static struct {
3742         char* className;
3743         oid_class oclass;
3744     } oid_classes[] = {
3745         {"appctx",CLASS_APPCTX},
3746         {"absyn",CLASS_ABSYN},
3747         {"attset",CLASS_ATTSET},
3748         {"transyn",CLASS_TRANSYN},
3749         {"diagset",CLASS_DIAGSET},
3750         {"recsyn",CLASS_RECSYN},
3751         {"resform",CLASS_RESFORM},
3752         {"accform",CLASS_ACCFORM},
3753         {"extserv",CLASS_EXTSERV},
3754         {"userinfo",CLASS_USERINFO},
3755         {"elemspec",CLASS_ELEMSPEC},
3756         {"varset",CLASS_VARSET},
3757         {"schema",CLASS_SCHEMA},
3758         {"tagset",CLASS_TAGSET},
3759         {"general",CLASS_GENERAL},
3760         {0,(enum oid_class) 0}
3761     };
3762     char oname_str[101], oclass_str[101], oid_str[101];
3763     int i;
3764     oid_class oidclass = CLASS_GENERAL;
3765     Odr_oid oid[OID_SIZE];
3766
3767     if (sscanf (args, "%100[^ ] %100[^ ] %100s",
3768                 oname_str,oclass_str, oid_str) < 1) {
3769         printf("Error in register command \n");
3770         return 0;
3771     }
3772
3773     for (i = 0; oid_classes[i].className; i++) {
3774         if (!strcmp(oid_classes[i].className, oclass_str))
3775         {
3776             oidclass=oid_classes[i].oclass;
3777             break;
3778         }
3779     }
3780
3781     if(!(oid_classes[i].className)) {
3782         printf("Unknown oid class %s\n",oclass_str);
3783         return 0;
3784     }
3785
3786     oid_dotstring_to_oid(oid_str, oid);
3787
3788     if (yaz_oid_add(yaz_oid_std(), oidclass, oname_str, oid))
3789     {
3790         printf("oid %s already exists, registration failed\n",
3791                oname_str);
3792     }
3793     return 1;
3794 }
3795
3796 int cmd_push_command(const char* arg)
3797 {
3798 #if HAVE_READLINE_HISTORY_H
3799     if(strlen(arg)>1)
3800         add_history(arg);
3801 #else
3802     fprintf(stderr,"Not compiled with the readline/history module\n");
3803 #endif
3804     return 1;
3805 }
3806
3807 void source_rc_file(const char *rc_file)
3808 {
3809     /*  If rc_file != NULL, source that. Else
3810         Look for .yazclientrc and read it if it exists.
3811         If it does not exist, read  $HOME/.yazclientrc instead */
3812     struct stat statbuf;
3813
3814     if (rc_file)
3815     {
3816         if (stat(rc_file, &statbuf) == 0)
3817             cmd_source(rc_file, 0);
3818         else
3819         {
3820             fprintf(stderr, "yaz_client: cannot source '%s'\n", rc_file);
3821             exit(1);
3822         }
3823     }
3824     else
3825     {
3826         char fname[1000];
3827         strcpy(fname, ".yazclientrc");
3828         if (stat(fname, &statbuf)==0)
3829         {
3830             cmd_source(fname, 0);
3831         }
3832         else
3833         {
3834             const char* homedir = getenv("HOME");
3835             if (homedir)
3836             {
3837                 sprintf(fname, "%.800s/%s", homedir, ".yazclientrc");
3838                 if (stat(fname, &statbuf)==0)
3839                     cmd_source(fname, 0);
3840             }
3841         }
3842     }
3843 }
3844
3845 void add_to_readline_history(void *client_data, const char *line)
3846 {
3847 #if HAVE_READLINE_HISTORY_H
3848     if (strlen(line))
3849         add_history(line);
3850 #endif
3851 }
3852
3853 static void initialize(const char *rc_file)
3854 {
3855     FILE *inf;
3856     int i;
3857
3858     if (!(out = odr_createmem(ODR_ENCODE)) ||
3859         !(in = odr_createmem(ODR_DECODE)) ||
3860         !(print = odr_createmem(ODR_PRINT)))
3861     {
3862         fprintf(stderr, "failed to allocate ODR streams\n");
3863         exit(1);
3864     }
3865
3866     setvbuf(stdout, 0, _IONBF, 0);
3867     if (apdu_file)
3868         odr_setprint(print, apdu_file);
3869
3870     bibset = ccl_qual_mk ();
3871     inf = fopen (ccl_fields, "r");
3872     if (inf)
3873     {
3874         ccl_qual_file (bibset, inf);
3875         fclose (inf);
3876     }
3877
3878     cqltrans = cql_transform_open_fname(cql_fields);
3879     /* If this fails, no problem: we detect cqltrans == 0 later */
3880
3881 #if HAVE_READLINE_READLINE_H
3882     rl_attempted_completion_function =
3883         (char **(*)(const char *, int, int)) readline_completer;
3884 #endif
3885     for(i = 0; i < maxOtherInfosSupported; ++i) {
3886         extraOtherInfos[i].oid[0] = -1;
3887         extraOtherInfos[i].value = 0;
3888     }
3889
3890     cmd_format("usmarc");
3891
3892     source_rc_file(rc_file);
3893
3894     file_history = file_history_new();
3895     file_history_load(file_history);
3896     file_history_trav(file_history, 0, add_to_readline_history);
3897 }
3898
3899
3900 #if HAVE_GETTIMEOFDAY
3901 struct timeval tv_start;
3902 #endif
3903
3904 #if YAZ_HAVE_XML2
3905 static void handle_srw_record(Z_SRW_record *rec)
3906 {
3907     if (rec->recordPosition)
3908     {
3909         printf ("pos=%d", *rec->recordPosition);
3910         setno = *rec->recordPosition + 1;
3911     }
3912     if (rec->recordSchema)
3913         printf (" schema=%s", rec->recordSchema);
3914     printf ("\n");
3915     if (rec->recordData_buf && rec->recordData_len)
3916     {
3917         fwrite(rec->recordData_buf, 1, rec->recordData_len, stdout);
3918         if (marc_file)
3919             fwrite (rec->recordData_buf, 1, rec->recordData_len, marc_file);
3920     }
3921     else
3922         printf ("No data!");
3923     printf("\n");
3924 }
3925
3926 static void handle_srw_explain_response(Z_SRW_explainResponse *res)
3927 {
3928     handle_srw_record(&res->record);
3929 }
3930
3931 static void handle_srw_response(Z_SRW_searchRetrieveResponse *res)
3932 {
3933     int i;
3934
3935     printf ("Received SRW SearchRetrieve Response\n");
3936
3937     for (i = 0; i<res->num_diagnostics; i++)
3938     {
3939         if (res->diagnostics[i].uri)
3940             printf ("SRW diagnostic %s\n",
3941                     res->diagnostics[i].uri);
3942         else
3943             printf ("SRW diagnostic missing or could not be decoded\n");
3944         if (res->diagnostics[i].message)
3945             printf ("Message: %s\n", res->diagnostics[i].message);
3946         if (res->diagnostics[i].details)
3947             printf ("Details: %s\n", res->diagnostics[i].details);
3948     }
3949     if (res->numberOfRecords)
3950         printf ("Number of hits: %d\n", *res->numberOfRecords);
3951     for (i = 0; i<res->num_records; i++)
3952         handle_srw_record(res->records + i);
3953 }
3954
3955 static void handle_srw_scan_term(Z_SRW_scanTerm *term)
3956 {
3957     if (term->displayTerm)
3958         printf("%s:", term->displayTerm);
3959     else if (term->value)
3960         printf("%s:", term->value);
3961     else
3962         printf("No value:");
3963     if (term->numberOfRecords)
3964         printf(" %d", *term->numberOfRecords);
3965     if (term->whereInList)
3966         printf(" %s", term->whereInList);
3967     if (term->value && term->displayTerm)
3968         printf(" %s", term->value);
3969
3970     strcpy(last_scan_line, term->value);
3971     printf("\n");
3972 }
3973
3974 static void handle_srw_scan_response(Z_SRW_scanResponse *res)
3975 {
3976     int i;
3977
3978     printf ("Received SRW Scan Response\n");
3979
3980     for (i = 0; i<res->num_diagnostics; i++)
3981     {
3982         if (res->diagnostics[i].uri)
3983             printf ("SRW diagnostic %s\n",
3984                     res->diagnostics[i].uri);
3985         else
3986             printf ("SRW diagnostic missing or could not be decoded\n");
3987         if (res->diagnostics[i].message)
3988             printf ("Message: %s\n", res->diagnostics[i].message);
3989         if (res->diagnostics[i].details)
3990             printf ("Details: %s\n", res->diagnostics[i].details);
3991     }
3992     if (res->terms)
3993         for (i = 0; i<res->num_terms; i++)
3994             handle_srw_scan_term(res->terms + i);
3995 }
3996
3997 static void http_response(Z_HTTP_Response *hres)
3998 {
3999     int ret = -1;
4000     const char *connection_head = z_HTTP_header_lookup(hres->headers,
4001                                                        "Connection");
4002     if (!yaz_srw_check_content_type(hres))
4003         printf("Content type does not appear to be XML\n");
4004     else
4005     {
4006         Z_SOAP *soap_package = 0;
4007         ODR o = odr_createmem(ODR_DECODE);
4008         Z_SOAP_Handler soap_handlers[3] = {
4009             {YAZ_XMLNS_SRU_v1_1, 0, (Z_SOAP_fun) yaz_srw_codec},
4010             {YAZ_XMLNS_UPDATE_v0_9, 0, (Z_SOAP_fun) yaz_ucp_codec},
4011             {0, 0, 0}
4012         };
4013         ret = z_soap_codec(o, &soap_package,
4014                            &hres->content_buf, &hres->content_len,
4015                            soap_handlers);
4016         if (!ret && soap_package->which == Z_SOAP_generic)
4017         {
4018             Z_SRW_PDU *sr = (Z_SRW_PDU *) soap_package->u.generic->p;
4019             if (sr->which == Z_SRW_searchRetrieve_response)
4020                 handle_srw_response(sr->u.response);
4021             else if (sr->which == Z_SRW_explain_response)
4022                 handle_srw_explain_response(sr->u.explain_response);
4023             else if (sr->which == Z_SRW_scan_response)
4024                 handle_srw_scan_response(sr->u.scan_response);
4025             else if (sr->which == Z_SRW_update_response)
4026                 printf("Got update response. Status: %s\n",
4027                        sr->u.update_response->operationStatus);
4028             else
4029                 ret = -1;
4030         }
4031         else if (soap_package && (soap_package->which == Z_SOAP_fault
4032                                   || soap_package->which == Z_SOAP_error))
4033         {
4034             printf ("HTTP Error Status=%d\n", hres->code);
4035             printf ("SOAP Fault code %s\n",
4036                     soap_package->u.fault->fault_code);
4037             printf ("SOAP Fault string %s\n",
4038                     soap_package->u.fault->fault_string);
4039             if (soap_package->u.fault->details)
4040                 printf ("SOAP Details %s\n",
4041                         soap_package->u.fault->details);
4042         }
4043         else
4044         {
4045             printf("z_soap_codec failed. (no SOAP error)\n");
4046             ret = -1;
4047         }
4048         odr_destroy(o);
4049     }
4050     if (ret)
4051     {
4052         if (hres->code != 200)
4053         {
4054             printf ("HTTP Error Status=%d\n", hres->code);
4055         }
4056         else
4057         {
4058             printf ("Decoding of SRW package failed\n");
4059         }
4060         close_session();
4061     }
4062     else
4063     {
4064         if (!strcmp(hres->version, "1.0"))
4065         {
4066             /* HTTP 1.0: only if Keep-Alive we stay alive.. */
4067             if (!connection_head || strcmp(connection_head, "Keep-Alive"))
4068                 close_session();
4069         }
4070         else
4071         {
4072             /* HTTP 1.1: only if no close we stay alive .. */
4073             if (connection_head && !strcmp(connection_head, "close"))
4074                 close_session();
4075         }
4076     }
4077 }
4078 #endif
4079
4080 void wait_and_handle_response(int one_response_only)
4081 {
4082     int reconnect_ok = 1;
4083     int res;
4084     char *netbuffer= 0;
4085     int netbufferlen = 0;
4086 #if HAVE_GETTIMEOFDAY
4087     int got_tv_end = 0;
4088     struct timeval tv_end;
4089 #endif
4090     Z_GDU *gdu;
4091
4092     while(conn)
4093     {
4094         res = cs_get(conn, &netbuffer, &netbufferlen);
4095         if (reconnect_ok && res <= 0 && protocol == PROTO_HTTP)
4096         {
4097             cs_close(conn);
4098             conn = 0;
4099             session_connect(cur_host);
4100             reconnect_ok = 0;
4101             if (conn)
4102             {
4103                 char *buf_out;
4104                 int len_out;
4105
4106                 buf_out = odr_getbuf(out, &len_out, 0);
4107
4108                 do_hex_dump(buf_out, len_out);
4109
4110                 cs_put(conn, buf_out, len_out);
4111
4112                 odr_reset(out);
4113                 continue;
4114             }
4115         }
4116         else if (res <= 0)
4117         {
4118             printf("Target closed connection\n");
4119             close_session();
4120             break;
4121         }
4122 #if HAVE_GETTIMEOFDAY
4123         if (got_tv_end == 0)
4124             gettimeofday (&tv_end, 0); /* count first one only */
4125         got_tv_end++;
4126 #endif
4127         odr_reset(out);
4128         odr_reset(in); /* release APDU from last round */
4129         record_last = 0;
4130         do_hex_dump(netbuffer, res);
4131         odr_setbuf(in, netbuffer, res, 0);
4132
4133         if (!z_GDU(in, &gdu, 0, 0))
4134         {
4135             FILE *f = ber_file ? ber_file : stdout;
4136             odr_perror(in, "Decoding incoming APDU");
4137             fprintf(f, "[Near %ld]\n", (long) odr_offset(in));
4138             fprintf(f, "Packet dump:\n---------\n");
4139             odr_dumpBER(f, netbuffer, res);
4140             fprintf(f, "---------\n");
4141             if (apdu_file)
4142             {
4143                 z_GDU(print, &gdu, 0, 0);
4144                 odr_reset(print);
4145             }
4146             if (conn && cs_more(conn))
4147                 continue;
4148             break;
4149         }
4150         if (ber_file)
4151             odr_dumpBER(ber_file, netbuffer, res);
4152         if (apdu_file && !z_GDU(print, &gdu, 0, 0))
4153         {
4154             odr_perror(print, "Failed to print incoming APDU");
4155             odr_reset(print);
4156                 continue;
4157         }
4158         if (gdu->which == Z_GDU_Z3950)
4159         {
4160             Z_APDU *apdu = gdu->u.z3950;
4161             switch(apdu->which)
4162             {
4163             case Z_APDU_initResponse:
4164                 process_initResponse(apdu->u.initResponse);
4165                 break;
4166             case Z_APDU_searchResponse:
4167                 process_searchResponse(apdu->u.searchResponse);
4168                 break;
4169             case Z_APDU_scanResponse:
4170                 process_scanResponse(apdu->u.scanResponse);
4171                 break;
4172             case Z_APDU_presentResponse:
4173                 print_refid (apdu->u.presentResponse->referenceId);
4174                 setno +=
4175                     *apdu->u.presentResponse->numberOfRecordsReturned;
4176                 if (apdu->u.presentResponse->records)
4177                     display_records(apdu->u.presentResponse->records);
4178                 else
4179                     printf("No records.\n");
4180                 printf ("nextResultSetPosition = %d\n",
4181                         *apdu->u.presentResponse->nextResultSetPosition);
4182                 break;
4183             case Z_APDU_sortResponse:
4184                 process_sortResponse(apdu->u.sortResponse);
4185                 break;
4186             case Z_APDU_extendedServicesResponse:
4187                 printf("Got extended services response\n");
4188                 process_ESResponse(apdu->u.extendedServicesResponse);
4189                 break;
4190             case Z_APDU_close:
4191                 printf("Target has closed the association.\n");
4192                 process_close(apdu->u.close);
4193                 break;
4194             case Z_APDU_resourceControlRequest:
4195                 process_resourceControlRequest
4196                     (apdu->u.resourceControlRequest);
4197                 break;
4198             case Z_APDU_deleteResultSetResponse:
4199                 process_deleteResultSetResponse(apdu->u.
4200                                                 deleteResultSetResponse);
4201                 break;
4202             default:
4203                 printf("Received unknown APDU type (%d).\n",
4204                        apdu->which);
4205                 close_session ();
4206             }
4207         }
4208 #if YAZ_HAVE_XML2
4209         else if (gdu->which == Z_GDU_HTTP_Response)
4210         {
4211             http_response(gdu->u.HTTP_Response);
4212         }
4213 #endif
4214         if (one_response_only)
4215             break;
4216         if (conn && !cs_more(conn))
4217             break;
4218     }
4219 #if HAVE_GETTIMEOFDAY
4220     if (got_tv_end)
4221     {
4222 #if 0
4223         printf ("S/U S/U=%ld/%ld %ld/%ld",
4224                 (long) tv_start.tv_sec,
4225                 (long) tv_start.tv_usec,
4226                 (long) tv_end.tv_sec,
4227                 (long) tv_end.tv_usec);
4228 #endif
4229         printf ("Elapsed: %.6f\n",
4230                 (double) tv_end.tv_usec / 1e6 + tv_end.tv_sec -
4231                 ((double) tv_start.tv_usec / 1e6 + tv_start.tv_sec));
4232     }
4233 #endif
4234     xfree (netbuffer);
4235 }
4236
4237
4238 int cmd_cclparse(const char* arg)
4239 {
4240     int error, pos;
4241     struct ccl_rpn_node *rpn=NULL;
4242
4243
4244     rpn = ccl_find_str (bibset, arg, &error, &pos);
4245
4246     if (error) {
4247         int ioff = 3+strlen(last_cmd)+1+pos;
4248         printf ("%*s^ - ", ioff, " ");
4249         printf ("%s\n", ccl_err_msg (error));
4250     }
4251     else
4252     {
4253         if (rpn)
4254         {
4255             ccl_pr_tree(rpn, stdout);
4256         }
4257     }
4258     if (rpn)
4259         ccl_rpn_delete(rpn);
4260
4261     printf ("\n");
4262
4263     return 0;
4264 }
4265
4266
4267 int cmd_set_otherinfo(const char* args)
4268 {
4269     char oidstr[101], otherinfoString[101];
4270     int otherinfoNo;
4271     int sscan_res;
4272
4273     sscan_res = sscanf (args, "%d %100[^ ] %100s",
4274                         &otherinfoNo, oidstr, otherinfoString);
4275
4276     if (sscan_res > 0 && otherinfoNo >= maxOtherInfosSupported) {
4277         printf("Error otherinfo index too large (%d>=%d)\n",
4278                otherinfoNo,maxOtherInfosSupported);
4279         return 0;
4280     }
4281
4282
4283     if (sscan_res==1)
4284     {
4285         /* reset this otherinfo */
4286         extraOtherInfos[otherinfoNo].oid[0] = -1;
4287         xfree(extraOtherInfos[otherinfoNo].value);
4288         extraOtherInfos[otherinfoNo].value = 0;
4289         return 0;
4290     }
4291     if (sscan_res != 3) {
4292         printf("Error in set_otherinfo command \n");
4293         return 0;
4294     }
4295     else
4296     {
4297         NMEM oid_tmp = nmem_create();
4298         const Odr_oid *oid =
4299             yaz_string_to_oid_nmem(yaz_oid_std(),
4300                                    CLASS_GENERAL, oidstr, oid_tmp);
4301         oid_oidcpy(extraOtherInfos[otherinfoNo].oid, oid);
4302
4303         xfree(extraOtherInfos[otherinfoNo].value);
4304         extraOtherInfos[otherinfoNo].value = xstrdup(otherinfoString);
4305
4306         nmem_destroy(oid_tmp);
4307     }
4308
4309     return 0;
4310 }
4311
4312 int cmd_sleep(const char* args )
4313 {
4314     int sec=atoi(args);
4315     if( sec > 0 ) {
4316 #ifdef WIN32
4317         Sleep(sec*1000);
4318 #else
4319         sleep(sec);
4320 #endif
4321         printf("Done sleeping %d seconds\n", sec);
4322     }
4323     return 1;
4324 }
4325
4326 int cmd_list_otherinfo(const char* args)
4327 {
4328     int i;
4329
4330     if (strlen(args)>0)
4331     {
4332         i = atoi(args);
4333         if (i >= maxOtherInfosSupported)
4334         {
4335             printf("Error otherinfo index to large (%d>%d)\n",i,maxOtherInfosSupported);
4336             return 0;
4337         }
4338         if (extraOtherInfos[i].value)
4339         {
4340             char name_oid[OID_STR_MAX];
4341             oid_class oclass;
4342             const char *name =
4343                 yaz_oid_to_string_buf(extraOtherInfos[i].oid, &oclass,
4344                                       name_oid);
4345             printf("  otherinfo %d %s %s\n",
4346                    i, name ? name : "null",
4347                    extraOtherInfos[i].value);
4348         }
4349
4350     }
4351     else
4352     {
4353         for(i = 0; i < maxOtherInfosSupported; ++i)
4354         {
4355             if (extraOtherInfos[i].value)
4356             {
4357                 char name_oid[OID_STR_MAX];
4358                 oid_class oclass;
4359                 const char *name =
4360                     yaz_oid_to_string_buf(extraOtherInfos[i].oid, &oclass,
4361                                           name_oid);
4362                 printf("  otherinfo %d %s %s\n",
4363                        i, name ? name : "null",
4364                        extraOtherInfos[i].value);
4365             }
4366         }
4367     }
4368     return 0;
4369 }
4370
4371
4372 int cmd_list_all(const char* args) {
4373     int i;
4374
4375     /* connection options */
4376     if(conn) {
4377         printf("Connected to         : %s\n",last_open_command);
4378     } else {
4379         if(last_open_command)
4380             printf("Not connected to     : %s\n",last_open_command);
4381         else
4382             printf("Not connected        : \n");
4383
4384     }
4385     if(yazProxy) printf("using proxy          : %s\n",yazProxy);
4386
4387     printf("auto_reconnect       : %s\n",auto_reconnect?"on":"off");
4388     printf("auto_wait            : %s\n",auto_wait?"on":"off");
4389
4390     if (!auth) {
4391         printf("Authentication       : none\n");
4392     } else {
4393         switch(auth->which) {
4394         case Z_IdAuthentication_idPass:
4395             printf("Authentication       : IdPass\n");
4396             printf("    Login User       : %s\n",auth->u.idPass->userId?auth->u.idPass->userId:"");
4397             printf("    Login Group      : %s\n",auth->u.idPass->groupId?auth->u.idPass->groupId:"");
4398             printf("    Password         : %s\n",auth->u.idPass->password?auth->u.idPass->password:"");
4399             break;
4400         case Z_IdAuthentication_open:
4401             printf("Authentication       : psOpen\n");
4402             printf("    Open string      : %s\n",auth->u.open);
4403             break;
4404         default:
4405             printf("Authentication       : Unknown\n");
4406         }
4407     }
4408     if (negotiationCharset)
4409         printf("Neg. Character set   : `%s'\n", negotiationCharset);
4410
4411     /* bases */
4412     printf("Bases                : ");
4413     for (i = 0; i<num_databaseNames; i++) printf("%s ",databaseNames[i]);
4414     printf("\n");
4415
4416     /* Query options */
4417     printf("CCL file             : %s\n",ccl_fields);
4418     printf("CQL file             : %s\n",cql_fields);
4419     printf("Query type           : %s\n",query_type_as_string(queryType));
4420
4421     printf("Named Result Sets    : %s\n",setnumber==-1?"off":"on");
4422
4423     /* piggy back options */
4424     printf("ssub/lslb/mspn       : %d/%d/%d\n",smallSetUpperBound,largeSetLowerBound,mediumSetPresentNumber);
4425
4426     /* print present related options */
4427     if (recordsyntax_size > 0)
4428     {
4429         printf("Format               : %s\n", recordsyntax_list[0]);
4430     }
4431     printf("Schema               : %s\n",record_schema ? record_schema : "not set");
4432     printf("Elements             : %s\n",elementSetNames?elementSetNames->u.generic:"");
4433
4434     /* loging options */
4435     printf("APDU log             : %s\n",apdu_file?"on":"off");
4436     printf("Record log           : %s\n",marc_file?"on":"off");
4437
4438     /* other infos */
4439     printf("Other Info: \n");
4440     cmd_list_otherinfo("");
4441
4442     return 0;
4443 }
4444
4445 int cmd_clear_otherinfo(const char* args)
4446 {
4447     if(strlen(args)>0) {
4448         int otherinfoNo = atoi(args);
4449         if (otherinfoNo >= maxOtherInfosSupported) {
4450             printf("Error otherinfo index too large (%d>=%d)\n",
4451                    otherinfoNo, maxOtherInfosSupported);
4452             return 0;
4453         }
4454         if (extraOtherInfos[otherinfoNo].value)
4455         {
4456             /* only clear if set. */
4457             extraOtherInfos[otherinfoNo].oid[0] = -1;
4458             xfree(extraOtherInfos[otherinfoNo].value);
4459             extraOtherInfos[otherinfoNo].value = 0;
4460         }
4461     } else {
4462         int i;
4463         for(i = 0; i < maxOtherInfosSupported; ++i)
4464         {
4465             if (extraOtherInfos[i].value)
4466             {
4467                 extraOtherInfos[i].oid[0] = -1;
4468                 xfree(extraOtherInfos[i].value);
4469                 extraOtherInfos[i].value = 0;
4470             }
4471         }
4472     }
4473     return 0;
4474 }
4475
4476 int cmd_wait_response(const char *arg)
4477 {
4478     int wait_for = atoi(arg);
4479     int i=0;
4480     if( wait_for < 1 ) {
4481         wait_for = 1;
4482     };
4483
4484     for( i=0 ; i < wait_for ; ++i ) {
4485         wait_and_handle_response(1);
4486     };
4487     return 0;
4488 }
4489
4490 static int cmd_help (const char *line);
4491
4492 typedef char *(*completerFunctionType)(const char *text, int state);
4493
4494 static struct {
4495     char *cmd;
4496     int (*fun)(const char *arg);
4497     char *ad;
4498         completerFunctionType rl_completerfunction;
4499     int complete_filenames;
4500     const char **local_tabcompletes;
4501 } cmd_array[] = {
4502     {"open", cmd_open, "('tcp'|'ssl')':<host>[':'<port>][/<db>]",NULL,0,NULL},
4503     {"quit", cmd_quit, "",NULL,0,NULL},
4504     {"find", cmd_find, "<query>",NULL,0,NULL},
4505     {"delete", cmd_delete, "<setname>",NULL,0,NULL},
4506     {"base", cmd_base, "<base-name>",NULL,0,NULL},
4507     {"show", cmd_show, "<rec#>['+'<#recs>['+'<setname>]]",NULL,0,NULL},
4508     {"setscan", cmd_setscan, "<term>",NULL,0,NULL},
4509     {"scan", cmd_scan, "<term>",NULL,0,NULL},
4510     {"scanstep", cmd_scanstep, "<size>",NULL,0,NULL},
4511     {"scanpos", cmd_scanpos, "<size>",NULL,0,NULL},
4512     {"scansize", cmd_scansize, "<size>",NULL,0,NULL},
4513     {"sort", cmd_sort, "<sortkey> <flag> <sortkey> <flag> ...",NULL,0,NULL},
4514     {"sort+", cmd_sort_newset, "<sortkey> <flag> <sortkey> <flag> ...",NULL,0,NULL},
4515     {"authentication", cmd_authentication, "<acctstring>",NULL,0,NULL},
4516     {"lslb", cmd_lslb, "<largeSetLowerBound>",NULL,0,NULL},
4517     {"ssub", cmd_ssub, "<smallSetUpperBound>",NULL,0,NULL},
4518     {"mspn", cmd_mspn, "<mediumSetPresentNumber>",NULL,0,NULL},
4519     {"status", cmd_status, "",NULL,0,NULL},
4520     {"setnames", cmd_setnames, "",NULL,0,NULL},
4521     {"cancel", cmd_cancel, "",NULL,0,NULL},
4522     {"cancel_find", cmd_cancel_find, "<query>",NULL,0,NULL},
4523     {"format", cmd_format, "<recordsyntax>",complete_format,0,NULL},
4524     {"schema", cmd_schema, "<schema>",complete_schema,0,NULL},
4525     {"elements", cmd_elements, "<elementSetName>",NULL,0,NULL},
4526     {"close", cmd_close, "",NULL,0,NULL},
4527     {"querytype", cmd_querytype, "<type>",complete_querytype,0,NULL},
4528     {"refid", cmd_refid, "<id>",NULL,0,NULL},
4529     {"itemorder", cmd_itemorder, "ill|item|xml <itemno>",NULL,0,NULL},
4530     {"update", cmd_update, "<action> <recid> [<doc>]",NULL,0,NULL},
4531     {"update0", cmd_update0, "<action> <recid> [<doc>]",NULL,0,NULL},
4532     {"xmles", cmd_xmles, "<OID> <doc>",NULL,0,NULL},
4533     {"packagename", cmd_packagename, "<packagename>",NULL,0,NULL},
4534     {"proxy", cmd_proxy, "[('tcp'|'ssl')]<host>[':'<port>]",NULL,0,NULL},
4535     {"charset", cmd_charset, "<nego_charset> <output_charset>",NULL,0,NULL},
4536     {"negcharset", cmd_negcharset, "<nego_charset>",NULL,0,NULL},
4537     {"displaycharset", cmd_displaycharset, "<output_charset>",NULL,0,NULL},
4538     {"marccharset", cmd_marccharset, "<charset_name>",NULL,0,NULL},
4539     {"querycharset", cmd_querycharset, "<charset_name>",NULL,0,NULL},
4540     {"lang", cmd_lang, "<language_code>",NULL,0,NULL},
4541     {"source", cmd_source_echo, "<filename>",NULL,1,NULL},
4542     {".", cmd_source_echo, "<filename>",NULL,1,NULL},
4543     {"!", cmd_subshell, "Subshell command",NULL,1,NULL},
4544     {"set_apdufile", cmd_set_apdufile, "<filename>",NULL,1,NULL},
4545     {"set_berfile", cmd_set_berfile, "<filename>",NULL,1,NULL},
4546     {"set_marcdump", cmd_set_marcdump," <filename>",NULL,1,NULL},
4547     {"set_cclfile", cmd_set_cclfile," <filename>",NULL,1,NULL},
4548     {"set_cqlfile", cmd_set_cqlfile," <filename>",NULL,1,NULL},
4549     {"set_auto_reconnect", cmd_set_auto_reconnect," on|off",complete_auto_reconnect,1,NULL},
4550     {"set_auto_wait", cmd_set_auto_wait," on|off",complete_auto_reconnect,1,NULL},
4551     {"set_otherinfo", cmd_set_otherinfo,"<otherinfoinddex> <oid> <string>",NULL,0,NULL},
4552     {"sleep", cmd_sleep,"<seconds>",NULL,0,NULL},
4553     {"register_oid", cmd_register_oid,"<name> <class> <oid>",NULL,0,NULL},
4554     {"push_command", cmd_push_command,"<command>",command_generator,0,NULL},
4555     {"register_tab", cmd_register_tab,"<commandname> <tab>",command_generator,0,NULL},
4556     {"cclparse", cmd_cclparse,"<ccl find command>",NULL,0,NULL},
4557     {"list_otherinfo",cmd_list_otherinfo,"[otherinfoinddex]",NULL,0,NULL},
4558     {"list_all",cmd_list_all,"",NULL,0,NULL},
4559     {"clear_otherinfo",cmd_clear_otherinfo,"",NULL,0,NULL},
4560     {"wait_response",cmd_wait_response,"<number>",NULL,0,NULL},
4561     /* Server Admin Functions */
4562     {"adm-reindex", cmd_adm_reindex, "<database-name>",NULL,0,NULL},
4563     {"adm-truncate", cmd_adm_truncate, "('database'|'index')<object-name>",NULL,0,NULL},
4564     {"adm-create", cmd_adm_create, "",NULL,0,NULL},
4565     {"adm-drop", cmd_adm_drop, "('database'|'index')<object-name>",NULL,0,NULL},
4566     {"adm-import", cmd_adm_import, "<record-type> <dir> <pattern>",NULL,0,NULL},
4567     {"adm-refresh", cmd_adm_refresh, "",NULL,0,NULL},
4568     {"adm-commit", cmd_adm_commit, "",NULL,0,NULL},
4569     {"adm-shutdown", cmd_adm_shutdown, "",NULL,0,NULL},
4570     {"adm-startup", cmd_adm_startup, "",NULL,0,NULL},
4571     {"explain", cmd_explain, "", NULL, 0, NULL},
4572     {"options", cmd_options, "", NULL, 0, NULL},
4573     {"zversion", cmd_zversion, "", NULL, 0, NULL},
4574     {"help", cmd_help, "", NULL,0,NULL},
4575     {"init", cmd_init, "", NULL,0,NULL},
4576     {"sru", cmd_sru, "<method> <version>", NULL,0,NULL},
4577     {"exit", cmd_quit, "",NULL,0,NULL},
4578     {0,0,0,0,0,0}
4579 };
4580
4581 static int cmd_help (const char *line)
4582 {
4583     int i;
4584     char topic[21];
4585
4586     *topic = 0;
4587     sscanf (line, "%20s", topic);
4588
4589     if (*topic == 0)
4590         printf("Commands:\n");
4591     for (i = 0; cmd_array[i].cmd; i++)
4592         if (*topic == 0 || strcmp (topic, cmd_array[i].cmd) == 0)
4593             printf("   %s %s\n", cmd_array[i].cmd, cmd_array[i].ad);
4594     if (!strcmp(topic, "find"))
4595     {
4596         printf("RPN:\n");
4597         printf(" \"term\"                        Simple Term\n");
4598         printf(" @attr [attset] type=value op  Attribute\n");
4599         printf(" @and opl opr                  And\n");
4600         printf(" @or opl opr                   Or\n");
4601         printf(" @not opl opr                  And-Not\n");
4602         printf(" @set set                      Result set\n");
4603         printf(" @prox exl dist ord rel uc ut  Proximity. Use help prox\n");
4604         printf("\n");
4605         printf("Bib-1 attribute types\n");
4606         printf("1=Use:         ");
4607         printf("4=Title 7=ISBN 8=ISSN 30=Date 62=Abstract 1003=Author 1016=Any\n");
4608         printf("2=Relation:    ");
4609         printf("1<   2<=  3=  4>=  5>  6!=  102=Relevance\n");
4610         printf("3=Position:    ");
4611         printf("1=First in Field  2=First in subfield  3=Any position\n");
4612         printf("4=Structure:   ");
4613         printf("1=Phrase  2=Word  3=Key  4=Year  5=Date  6=WordList\n");
4614         printf("5=Truncation:  ");
4615         printf("1=Right  2=Left  3=L&R  100=No  101=#  102=Re-1  103=Re-2\n");
4616         printf("6=Completeness:");
4617         printf("1=Incomplete subfield  2=Complete subfield  3=Complete field\n");
4618     }
4619     if (!strcmp(topic, "prox"))
4620     {
4621         printf("Proximity:\n");
4622         printf(" @prox exl dist ord rel uc ut\n");
4623         printf(" exl:  exclude flag . 0=include, 1=exclude.\n");
4624         printf(" dist: distance integer.\n");
4625         printf(" ord:  order flag. 0=unordered, 1=ordered.\n");
4626         printf(" rel:  relation integer. 1<  2<=  3= 4>=  5>  6!= .\n");
4627         printf(" uc:   unit class. k=known, p=private.\n");
4628         printf(" ut:   unit type. 1=character, 2=word, 3=sentence,\n");
4629         printf("        4=paragraph, 5=section, 6=chapter, 7=document,\n");
4630         printf("        8=element, 9=subelement, 10=elementType, 11=byte.\n");
4631         printf("\nExamples:\n");
4632         printf(" Search for a and b in-order at most 3 words apart:\n");
4633         printf("  @prox 0 3 1 2 k 2 a b\n");
4634         printf(" Search for any order of a and b next to each other:\n");
4635         printf("  @prox 0 1 0 3 k 2 a b\n");
4636     }
4637     return 1;
4638 }
4639
4640 int cmd_register_tab(const char* arg)
4641 {
4642 #if HAVE_READLINE_READLINE_H
4643     char command[101], tabargument[101];
4644     int i;
4645     int num_of_tabs;
4646     const char** tabslist;
4647
4648     if (sscanf (arg, "%100s %100s", command, tabargument) < 1) {
4649         return 0;
4650     }
4651
4652     /* locate the amdn in the list */
4653     for (i = 0; cmd_array[i].cmd; i++) {
4654         if (!strncmp(cmd_array[i].cmd, command, strlen(command))) {
4655             break;
4656         }
4657     }
4658
4659     if (!cmd_array[i].cmd) {
4660         fprintf(stderr,"Unknown command %s\n",command);
4661         return 1;
4662     }
4663
4664
4665     if (!cmd_array[i].local_tabcompletes)
4666         cmd_array[i].local_tabcompletes = (const char **) calloc(1,sizeof(char**));
4667
4668     num_of_tabs=0;
4669
4670     tabslist = cmd_array[i].local_tabcompletes;
4671     for(; tabslist && *tabslist; tabslist++) {
4672         num_of_tabs++;
4673     }
4674
4675     cmd_array[i].local_tabcompletes = (const char **)
4676         realloc(cmd_array[i].local_tabcompletes,
4677                 (num_of_tabs+2)*sizeof(char**));
4678     tabslist = cmd_array[i].local_tabcompletes;
4679     tabslist[num_of_tabs] = strdup(tabargument);
4680     tabslist[num_of_tabs+1] = NULL;
4681 #endif
4682     return 1;
4683 }
4684
4685
4686 void process_cmd_line(char* line)
4687 {
4688     int i, res;
4689     char word[32], arg[10240];
4690
4691 #if HAVE_GETTIMEOFDAY
4692     gettimeofday (&tv_start, 0);
4693 #endif
4694
4695     if ((res = sscanf(line, "%31s %10239[^;]", word, arg)) <= 0)
4696     {
4697         strcpy(word, last_cmd);
4698         *arg = '\0';
4699     }
4700     else if (res == 1)
4701         *arg = 0;
4702     strcpy(last_cmd, word);
4703
4704     /* removed tailing spaces from the arg command */
4705     {
4706         char* p = arg;
4707         char* lastnonspace=NULL;
4708
4709         for(;*p; ++p) {
4710             if(!isspace(*(unsigned char *) p)) {
4711                 lastnonspace = p;
4712             }
4713         }
4714         if(lastnonspace)
4715             *(++lastnonspace) = 0;
4716     }
4717
4718     for (i = 0; cmd_array[i].cmd; i++)
4719         if (!strncmp(cmd_array[i].cmd, word, strlen(word)))
4720         {
4721             res = (*cmd_array[i].fun)(arg);
4722             break;
4723         }
4724
4725     if (!cmd_array[i].cmd) /* dump our help-screen */
4726     {
4727         printf("Unknown command: %s.\n", word);
4728         printf("Type 'help' for list of commands\n");
4729         res = 1;
4730     }
4731
4732     if(apdu_file) fflush(apdu_file);
4733
4734     if (res >= 2 && auto_wait)
4735         wait_and_handle_response(0);
4736
4737     if(apdu_file)
4738         fflush(apdu_file);
4739     if(marc_file)
4740         fflush(marc_file);
4741 }
4742
4743 static char *command_generator(const char *text, int state)
4744 {
4745 #if HAVE_READLINE_READLINE_H
4746     static int idx;
4747     if (state==0) {
4748         idx = 0;
4749     }
4750     for( ; cmd_array[idx].cmd; ++idx) {
4751         if (!strncmp(cmd_array[idx].cmd, text, strlen(text))) {
4752             ++idx;  /* skip this entry on the next run */
4753             return strdup(cmd_array[idx-1].cmd);
4754         }
4755     }
4756 #endif
4757     return NULL;
4758 }
4759
4760 #if HAVE_READLINE_READLINE_H
4761 static const char** default_completer_list = NULL;
4762
4763 static char* default_completer(const char* text, int state)
4764 {
4765     return complete_from_list(default_completer_list, text, state);
4766 }
4767 #endif
4768
4769 #if HAVE_READLINE_READLINE_H
4770
4771 /*
4772    This function only known how to complete on the first word
4773 */
4774 char **readline_completer(char *text, int start, int end)
4775 {
4776     completerFunctionType completerToUse;
4777
4778     if(start == 0) {
4779 #if HAVE_READLINE_RL_COMPLETION_MATCHES
4780         char** res = rl_completion_matches(text, command_generator);
4781 #else
4782         char** res = completion_matches(text,
4783                                         (CPFunction*)command_generator);
4784 #endif
4785         rl_attempted_completion_over = 1;
4786         return res;
4787     } else {
4788         char arg[10240],word[32];
4789         int i=0 ,res;
4790         if ((res = sscanf(rl_line_buffer, "%31s %10239[^;]", word, arg)) <= 0) {
4791             rl_attempted_completion_over = 1;
4792             return NULL;
4793         }
4794
4795         for (i = 0; cmd_array[i].cmd; i++)
4796             if (!strncmp(cmd_array[i].cmd, word, strlen(word)))
4797                 break;
4798
4799         if(!cmd_array[i].cmd)
4800             return NULL;
4801
4802         default_completer_list = cmd_array[i].local_tabcompletes;
4803
4804         completerToUse = cmd_array[i].rl_completerfunction;
4805         if (!completerToUse)
4806         { /* if command completer is not defined use the default completer */
4807             completerToUse = default_completer;
4808         }
4809         if (completerToUse) {
4810 #ifdef HAVE_READLINE_RL_COMPLETION_MATCHES
4811             char** res=
4812                 rl_completion_matches(text, completerToUse);
4813 #else
4814             char** res=
4815                 completion_matches(text, (CPFunction*)completerToUse);
4816 #endif
4817             if (!cmd_array[i].complete_filenames)
4818                 rl_attempted_completion_over = 1;
4819             return res;
4820         } else {
4821             if (!cmd_array[i].complete_filenames)
4822                 rl_attempted_completion_over = 1;
4823             return 0;
4824         }
4825     }
4826 }
4827 #endif
4828
4829 #ifndef WIN32
4830 void ctrl_c_handler(int x)
4831 {
4832     exit_client(0);
4833 }
4834 #endif
4835
4836 static void client(void)
4837 {
4838     char line[10240];
4839
4840     line[10239] = '\0';
4841
4842 #ifndef WIN32
4843     signal(SIGINT, ctrl_c_handler);
4844 #endif
4845
4846 #if HAVE_GETTIMEOFDAY
4847     gettimeofday (&tv_start, 0);
4848 #endif
4849
4850     while (1)
4851     {
4852         char *line_in = NULL;
4853 #if HAVE_READLINE_READLINE_H
4854         if (isatty(0))
4855         {
4856             line_in=readline(C_PROMPT);
4857             if (!line_in)
4858                 break;
4859 #if HAVE_READLINE_HISTORY_H
4860             if (*line_in)
4861                 add_history(line_in);
4862 #endif
4863             strncpy(line, line_in, sizeof(line)-1);
4864             free(line_in);
4865         }
4866 #endif
4867         if (!line_in)
4868         {
4869             char *end_p;
4870             printf (C_PROMPT);
4871             fflush(stdout);
4872             if (!fgets(line, sizeof(line)-1, stdin))
4873                 break;
4874             if ((end_p = strchr (line, '\n')))
4875                 *end_p = '\0';
4876         }
4877         if (isatty(0))
4878             file_history_add_line(file_history, line);
4879         process_cmd_line(line);
4880     }
4881 }
4882
4883 static void show_version(void)
4884 {
4885     char vstr[20];
4886
4887     yaz_version(vstr, 0);
4888     printf ("YAZ version: %s\n", YAZ_VERSION);
4889     if (strcmp(vstr, YAZ_VERSION))
4890         printf ("YAZ DLL/SO: %s\n", vstr);
4891     exit(0);
4892 }
4893
4894 int main(int argc, char **argv)
4895 {
4896     char *prog = *argv;
4897     char *open_command = 0;
4898     char *auth_command = 0;
4899     char *arg;
4900     const char *rc_file = 0;
4901     int ret;
4902
4903 #if HAVE_LOCALE_H
4904     if (!setlocale(LC_CTYPE, ""))
4905         fprintf (stderr, "setlocale failed\n");
4906 #endif
4907 #if HAVE_LANGINFO_H
4908 #ifdef CODESET
4909     codeset = nl_langinfo(CODESET);
4910 #endif
4911 #endif
4912     if (codeset)
4913         outputCharset = xstrdup(codeset);
4914
4915     ODR_MASK_SET(&z3950_options, Z_Options_search);
4916     ODR_MASK_SET(&z3950_options, Z_Options_present);
4917     ODR_MASK_SET(&z3950_options, Z_Options_namedResultSets);
4918     ODR_MASK_SET(&z3950_options, Z_Options_triggerResourceCtrl);
4919     ODR_MASK_SET(&z3950_options, Z_Options_scan);
4920     ODR_MASK_SET(&z3950_options, Z_Options_sort);
4921     ODR_MASK_SET(&z3950_options, Z_Options_extendedServices);
4922     ODR_MASK_SET(&z3950_options, Z_Options_delSet);
4923     ODR_MASK_SET(&z3950_options, Z_Options_negotiationModel);
4924
4925     while ((ret = options("k:c:q:a:b:m:v:p:u:t:Vxd:f:", argv, argc, &arg)) != -2)
4926     {
4927         switch (ret)
4928         {
4929         case 0:
4930             if (!open_command)
4931             {
4932                 open_command = (char *) xmalloc (strlen(arg)+6);
4933                 strcpy (open_command, "open ");
4934                 strcat (open_command, arg);
4935             }
4936             else
4937             {
4938                 fprintf(stderr, "%s: Specify at most one server address\n",
4939                         prog);
4940                 exit(1);
4941             }
4942             break;
4943         case 'a':
4944             if (!strcmp(arg, "-"))
4945                 apdu_file=stderr;
4946             else
4947                 apdu_file=fopen(arg, "a");
4948             break;
4949         case 'b':
4950             if (!strcmp(arg, "-"))
4951                 ber_file=stderr;
4952             else
4953                 ber_file=fopen(arg, "a");
4954             break;
4955         case 'c':
4956             strncpy (ccl_fields, arg, sizeof(ccl_fields)-1);
4957             ccl_fields[sizeof(ccl_fields)-1] = '\0';
4958             break;
4959         case 'd':
4960             dump_file_prefix = arg;
4961             break;
4962         case 'f':
4963             rc_file = arg;
4964             break;
4965         case 'k':
4966             kilobytes = atoi(arg);
4967             break;
4968         case 'm':
4969             if (!(marc_file = fopen (arg, "a")))
4970             {
4971                 perror (arg);
4972                 exit (1);
4973             }
4974             break;
4975         case 'p':
4976             yazProxy = xstrdup(arg);
4977             break;
4978         case 'q':
4979             strncpy (cql_fields, arg, sizeof(cql_fields)-1);
4980             cql_fields[sizeof(cql_fields)-1] = '\0';
4981             break;
4982         case 't':
4983             outputCharset = xstrdup(arg);
4984             break;
4985         case 'u':
4986             if (!auth_command)
4987             {
4988                 auth_command = (char *) xmalloc (strlen(arg)+6);
4989                 strcpy (auth_command, "auth ");
4990                 strcat (auth_command, arg);
4991             }
4992             break;
4993         case 'v':
4994             yaz_log_init(yaz_log_mask_str(arg), "", 0);
4995             break;
4996         case 'V':
4997             show_version();
4998             break;
4999         case 'x':
5000             hex_dump = 1;
5001             break;
5002         default:
5003             fprintf (stderr, "Usage: %s "
5004                      " [-a apdulog]"
5005                      " [-b berdump]"
5006                      " [-c cclfile]"
5007                      " [-d dump]"
5008                      " [-f cmdfile]"
5009                      " [-k size]"
5010                      " [-m marclog]"
5011                      " [-p proxy-addr]"
5012                      " [-q cqlfile]"
5013                      " [-t dispcharset]"
5014                      " [-u auth]"
5015                      " [-v loglevel]"
5016                      " [-V]"
5017                      " [-x]"
5018                      " [server-addr]\n",
5019                      prog);
5020             exit (1);
5021         }
5022     }
5023     initialize(rc_file);
5024     if (auth_command)
5025     {
5026 #ifdef HAVE_GETTIMEOFDAY
5027         gettimeofday (&tv_start, 0);
5028 #endif
5029         process_cmd_line (auth_command);
5030 #if HAVE_READLINE_HISTORY_H
5031         add_history(auth_command);
5032 #endif
5033         xfree(auth_command);
5034     }
5035     if (open_command)
5036     {
5037 #ifdef HAVE_GETTIMEOFDAY
5038         gettimeofday (&tv_start, 0);
5039 #endif
5040         process_cmd_line (open_command);
5041 #if HAVE_READLINE_HISTORY_H
5042         add_history(open_command);
5043 #endif
5044         xfree(open_command);
5045     }
5046     client();
5047     exit_client(0);
5048     return 0;
5049 }
5050 /*
5051  * Local variables:
5052  * c-basic-offset: 4
5053  * indent-tabs-mode: nil
5054  * End:
5055  * vim: shiftwidth=4 tabstop=8 expandtab
5056  */
5057