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