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