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