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