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