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