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