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