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