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