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