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