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