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