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