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