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