Added commands for sending shutdown and startup admin requests via the admin ES.
[yaz-moved-to-github.git] / client / client.c
1 /*
2  * Copyright (c) 1995-2000, Index Data
3  * See the file LICENSE for details.
4  * Sebastian Hammer, Adam Dickmeiss
5  *
6  * $Log: client.c,v $
7  * Revision 1.98  2000-03-16 13:55:49  ian
8  * Added commands for sending shutdown and startup admin requests via the admin ES.
9  *
10  * Revision 1.97  2000/03/14 14:06:04  ian
11  * Minor change to order of debugging output for send_apdu,
12  * fixed encoding of admin request.
13  *
14  * Revision 1.96  2000/03/14 09:27:07  ian
15  * Added code to enable sending of admin extended service requests
16  *
17  * Revision 1.95  2000/02/28 11:20:05  adam
18  * Using autoconf. New definitions: YAZ_BEGIN_CDECL/YAZ_END_CDECL.
19  *
20  * Revision 1.94  2000/01/31 13:15:21  adam
21  * Removed uses of assert(3). Cleanup of ODR. CCL parser update so
22  * that some characters are not surrounded by spaces in resulting term.
23  * ILL-code updates.
24  *
25  * Revision 1.93  2000/01/15 09:39:50  adam
26  * Implemented ill_get_ILLRequest. More ILL testing for client.
27  *
28  * Revision 1.92  1999/12/21 16:24:48  adam
29  * More robust ISO2709 handling (in case of real bad formats).
30  *
31  * Revision 1.91  1999/12/16 23:36:19  adam
32  * Implemented ILL protocol. Minor updates ASN.1 compiler.
33  *
34  * Revision 1.90  1999/11/30 13:47:11  adam
35  * Improved installation. Moved header files to include/yaz.
36  *
37  * Revision 1.89  1999/11/04 14:58:44  adam
38  * Added status elements for backend delete result set handler.
39  * Updated delete result result set command for client.
40  *
41  * Revision 1.88  1999/10/11 10:00:29  adam
42  * Modified printing of records.
43  *
44  * Revision 1.87  1999/08/27 09:40:32  adam
45  * Renamed logf function to yaz_log. Removed VC++ project files.
46  *
47  * Revision 1.86  1999/07/06 12:13:35  adam
48  * Added "schema" command.
49  *
50  * Revision 1.85  1999/06/16 11:55:24  adam
51  * Added APDU log to client.
52  *
53  * Revision 1.84  1999/06/01 14:29:11  adam
54  * Work on Extended Services.
55  *
56  * Revision 1.83  1999/05/26 15:24:26  adam
57  * Fixed minor bugs regarding DB Update (introduced by previous commit).
58  *
59  * Revision 1.82  1999/05/26 13:49:12  adam
60  * DB Update implemented in client (very basic).
61  *
62  * Revision 1.81  1999/04/20 09:56:48  adam
63  * Added 'name' paramter to encoder/decoder routines (typedef Odr_fun).
64  * Modified all encoders/decoders to reflect this change.
65  *
66  * Revision 1.80  1999/03/31 11:18:24  adam
67  * Implemented odr_strdup. Added Reference ID to backend server API.
68  *
69  * Revision 1.79  1999/03/23 14:14:25  adam
70  * Yet another fix.
71  *
72  * Revision 1.78  1999/03/18 12:57:18  adam
73  * Minor fix.
74  *
75  * Revision 1.77  1999/03/11 15:26:51  adam
76  * Fixed bug (introduced by previous commit).
77  *
78  * Revision 1.76  1999/03/11 11:12:07  adam
79  * Added GNU readline support. HTML display in client.
80  *
81  * Revision 1.75  1999/02/01 15:37:32  adam
82  * Fixed minor bug introduced by previous commit.
83  *
84  * Revision 1.74  1999/02/01 15:35:21  adam
85  * Added XML display.
86  *
87  * Revision 1.73  1998/10/20 13:55:43  quinn
88  * Fixed Scan bug in asn and client
89  *
90  * Revision 1.72  1998/10/20 13:23:15  quinn
91  * changed preferred pos to 1
92  *
93  * Revision 1.71  1998/10/20 13:21:43  adam
94  * Fixed scan response handler.
95  *
96  * Revision 1.70  1998/09/22 09:40:37  adam
97  * Minor changes in sort spec.
98  *
99  * Revision 1.69  1998/08/19 16:10:06  adam
100  * Changed som member names of DeleteResultSetRequest/Response.
101  *
102  * Revision 1.68  1998/07/20 12:37:06  adam
103  * Added 'delete <resultset>' command. Changed open command so that
104  * it reconnects if already connected.
105  *
106  * Revision 1.67  1998/06/09 13:55:06  adam
107  * Minor changes.
108  *
109  * Revision 1.66  1998/05/18 13:06:53  adam
110  * Changed the way attribute sets are handled by the retriaval module.
111  * Extended Explain conversion / schema.
112  * Modified server and client to work with ASN.1 compiled protocol handlers.
113  *
114  * Revision 1.65  1998/03/31 15:13:19  adam
115  * Development towards compiled ASN.1.
116  *
117  * Revision 1.64  1998/03/31 11:07:44  adam
118  * Furhter work on UNIverse resource report.
119  * Added Extended Services handling in frontend server.
120  *
121  * Revision 1.63  1998/03/05 08:05:10  adam
122  * Added a few casts to make C++ happy.
123  *
124  * Revision 1.62  1998/02/11 11:53:33  adam
125  * Changed code so that it compiles as C++.
126  *
127  * Revision 1.61  1998/02/10 11:03:06  adam
128  * Implemented command refid. Client prints reference-ID's, when present,
129  * in responses.
130  *
131  * Revision 1.60  1998/01/29 14:08:52  adam
132  * Better sort diagnostics.
133  *
134  * Revision 1.59  1998/01/29 13:17:56  adam
135  * Added sort.
136  *
137  * Revision 1.58  1998/01/07 13:51:45  adam
138  * Minor change.
139  *
140  * Revision 1.57  1998/01/07 12:58:22  adam
141  * Using fgets instead of gets.
142  *
143  * Revision 1.56  1997/11/05 09:18:31  adam
144  * The client handles records with no associated syntax.
145  *
146  * Revision 1.55  1997/10/31 12:20:08  adam
147  * Improved memory debugging for xmalloc/nmem.c. References to NMEM
148  * instead of ODR in n ESPEC-1 handling in source d1_espec.c.
149  * Bug fix: missing fclose in data1_read_espec1.
150  *
151  * Revision 1.54  1997/10/27 13:52:46  adam
152  * Header yaz-util includes all YAZ utility header files.
153  *
154  * Revision 1.53  1997/09/29 13:18:59  adam
155  * Added function, oid_ent_to_oid, to replace the function
156  * oid_getoidbyent, which is not thread safe.
157  *
158  * Revision 1.52  1997/09/29 07:20:31  adam
159  * Client code uses nmem_init.
160  *
161  * Revision 1.51  1997/09/26 09:41:55  adam
162  * Updated client to handle multiple diagnostics.
163  *
164  * Revision 1.50  1997/09/17 12:10:29  adam
165  * YAZ version 1.4.
166  *
167  * Revision 1.49  1997/09/04 13:45:17  adam
168  * Added UNImarc to list of available syntaxes.
169  *
170  * Revision 1.48  1997/09/01 08:48:44  adam
171  * New windows NT/95 port using MSV5.0. Only a few changes made
172  * to avoid warnings. Sub project created: client.dsp.
173  *
174  * Revision 1.47  1997/07/01 13:49:56  adam
175  * Take care of case when invalid target is specified on command line.
176  *
177  * Revision 1.46  1997/06/23 10:30:18  adam
178  * Added call to ccl_rpn_delete in search. Added ODR stream "out"
179  * as parameter to ccl_rpn_query to release RPN query.
180  *
181  * Revision 1.45  1997/05/14 06:53:29  adam
182  * C++ support.
183  *
184  * Revision 1.44  1997/05/05 11:20:35  adam
185  * Client uses "options" utility and marc dump filename may be specified
186  * as an option (-m <file>).
187  *
188  * Revision 1.43  1996/11/08 11:03:26  adam
189  * Client accepts multiple database names.
190  *
191  * Revision 1.42  1996/10/08 10:44:57  quinn
192  * Resolved conflicts.
193  *
194  * Revision 1.41  1996/10/07  15:29:03  quinn
195  * Work
196  *
197  * Revision 1.40  1996/08/29  14:19:34  quinn
198  * Fixed conflict (CVS)
199  *
200  * Revision 1.39  1996/08/27  10:43:22  quinn
201  * Made select() optional
202  *
203  * Revision 1.38  1996/08/12  14:09:11  adam
204  * Default prefix query attribute set defined by using p_query_attset.
205  *
206  * Revision 1.37  1996/07/06  19:58:29  quinn
207  * System headerfiles gathered in yconfig
208  *
209  * Revision 1.36  1996/06/10  08:53:47  quinn
210  * Added Summary
211  *
212  * Revision 1.35  1996/06/03  09:45:50  quinn
213  * Added display of OIDs in the GRS routine.
214  *
215  * Revision 1.34  1996/05/09  07:26:49  quinn
216  * *** empty log message ***
217  *
218  * Revision 1.33  1996/05/09  07:25:22  quinn
219  * Small
220  *
221  * Revision 1.32  1996/03/15  11:05:33  adam
222  * The user can set the preferred query type (prefix, ccl, ..) with the
223  * querytype command.
224  *
225  * Revision 1.31  1996/02/20  12:51:54  quinn
226  * Fixed problems with EXTERNAL.
227  *
228  * Revision 1.30  1996/02/12  18:18:09  quinn
229  * Fidgeting.
230  *
231  * Revision 1.29  1996/01/02  08:57:25  quinn
232  * Changed enums in the ASN.1 .h files to #defines. Changed oident.class to oclass
233  *
234  * Revision 1.28  1995/12/14  11:09:31  quinn
235  * Added Explain record syntax to the format command.
236  *
237  * Revision 1.27  1995/12/12  16:37:02  quinn
238  * Added destroy element to data1_node.
239  *
240  * Revision 1.26  1995/12/12  14:11:00  quinn
241  * Minimal.
242  *
243  * Revision 1.25  1995/11/13  09:27:22  quinn
244  * Fiddling with the variant stuff.
245  *
246  * Revision 1.24  1995/10/30  12:41:13  quinn
247  * Added hostname lookup for server.
248  *
249  * Revision 1.23  1995/10/18  16:12:30  quinn
250  * Better diagnostics.
251  *
252  * Revision 1.22  1995/10/11  14:49:12  quinn
253  * Smallish.
254  *
255  * Revision 1.21  1995/09/29  17:01:47  quinn
256  * More Windows work
257  *
258  * Revision 1.20  1995/08/29  14:24:13  quinn
259  * Added second half of close-handshake
260  *
261  * Revision 1.19  1995/08/29  11:17:28  quinn
262  * Added code to receive close
263  *
264  * Revision 1.18  1995/08/28  12:21:27  quinn
265  * Client can now ask for simple element set names.
266  *
267  * Revision 1.17  1995/08/17  12:45:02  quinn
268  * Fixed minor problems with GRS-1. Added support in c&s.
269  *
270  * Revision 1.16  1995/08/15  12:00:04  quinn
271  * Updated External
272  *
273  * Revision 1.15  1995/06/22  09:28:03  quinn
274  * Fixed bug in SUTRS processing.
275  *
276  * Revision 1.14  1995/06/19  12:37:41  quinn
277  * Added BER dumper.
278  *
279  * Revision 1.13  1995/06/16  10:29:11  quinn
280  * *** empty log message ***
281  *
282  * Revision 1.12  1995/06/15  07:44:57  quinn
283  * Moving to v3.
284  *
285  * Revision 1.11  1995/06/14  15:26:40  quinn
286  * *** empty log message ***
287  *
288  * Revision 1.10  1995/06/06  14:56:58  quinn
289  * Better diagnostics.
290  *
291  * Revision 1.9  1995/06/06  08:15:19  quinn
292  * Cosmetic.
293  *
294  * Revision 1.8  1995/06/05  10:52:22  quinn
295  * Added SCAN.
296  *
297  * Revision 1.7  1995/06/02  09:50:09  quinn
298  * Smallish.
299  *
300  * Revision 1.6  1995/05/31  08:29:21  quinn
301  * Nothing significant.
302  *
303  * Revision 1.5  1995/05/29  08:10:47  quinn
304  * Moved oid.c to util.
305  *
306  * Revision 1.4  1995/05/22  15:30:13  adam
307  * Client uses prefix query notation.
308  *
309  * Revision 1.3  1995/05/22  15:06:53  quinn
310  * *** empty log message ***
311  *
312  * Revision 1.2  1995/05/22  14:56:40  quinn
313  * *** empty log message ***
314  *
315  * Revision 1.1  1995/05/22  11:30:31  quinn
316  * Added prettier client.
317  *
318  *
319  */
320
321 /*
322  * This is the obligatory little toy client, whose primary purpose is
323  * to illustrate the use of the YAZ service-level API.
324  */
325
326 #include <stdio.h>
327 #include <stdlib.h>
328 #include <time.h>
329
330 #include <yaz/yaz-util.h>
331
332 #include <yaz/tcpip.h>
333 #ifdef USE_XTIMOSI
334 #include <yaz/xmosi.h>
335 #endif
336
337 #include <yaz/proto.h>
338 #include <yaz/marcdisp.h>
339 #include <yaz/diagbib1.h>
340
341 #include <yaz/pquery.h>
342
343 #ifdef ASN_COMPILED
344 #include <yaz/ill.h>
345 #endif
346
347 #if CCL2RPN
348 #include <yaz/yaz-ccl.h>
349 #endif
350
351 #if HAVE_READLINE_READLINE_H
352 #include <readline/readline.h>
353 #endif
354 #if HAVE_READLINE_HISTORY_H
355 #include <readline/history.h>
356 #endif
357
358 #include "admin.h"
359
360 #define C_PROMPT "Z> "
361
362 static ODR out, in, print;              /* encoding and decoding streams */
363 static FILE *apdu_file = 0;
364 static COMSTACK conn = 0;               /* our z-association */
365 static Z_IdAuthentication *auth = 0;    /* our current auth definition */
366 static char *databaseNames[128];
367 static Z_External *record_last = 0;
368 static int num_databaseNames = 0;
369 static int setnumber = 0;               /* current result set number */
370 static int smallSetUpperBound = 0;
371 static int largeSetLowerBound = 1;
372 static int mediumSetPresentNumber = 0;
373 static Z_ElementSetNames *elementSetNames = 0; 
374 static int setno = 1;                   /* current set offset */
375 static enum oid_proto protocol = PROTO_Z3950;      /* current app protocol */
376 static enum oid_value recordsyntax = VAL_USMARC;
377 static enum oid_value schema = VAL_NONE;
378 static int sent_close = 0;
379 static NMEM session_mem = NULL;      /* memory handle for init-response */
380 static Z_InitResponse *session = 0;     /* session parameters */
381 static char last_scan[512] = "0";
382 static char last_cmd[100] = "?";
383 static FILE *marcdump = 0;
384 static char *refid = NULL;
385
386 typedef enum {
387     QueryType_Prefix,
388     QueryType_CCL,
389     QueryType_CCL2RPN
390 } QueryType;
391
392 static QueryType queryType = QueryType_Prefix;
393
394 #if CCL2RPN
395 static CCL_bibset bibset;               /* CCL bibset handle */
396 #endif
397
398 ODR getODROutputStream()
399 {
400     return out;
401 }
402
403 void send_apdu(Z_APDU *a)
404 {
405     char *buf;
406     int len;
407
408     if (apdu_file)
409     {
410         z_APDU(print, &a, 0, 0);
411         odr_reset(print);
412     }
413     if (!z_APDU(out, &a, 0, 0))
414     {
415         odr_perror(out, "Encoding APDU");
416         exit(1);
417     }
418     buf = odr_getbuf(out, &len, 0);
419     /* printf ("sending APDU of size %d\n", len); */
420     if (cs_put(conn, buf, len) < 0)
421     {
422         fprintf(stderr, "cs_put: %s", cs_errmsg(cs_errno(conn)));
423         exit(1);
424     }
425     odr_reset(out); /* release the APDU structure  */
426 }
427
428 static void print_refid (Z_ReferenceId *id)
429 {
430     if (id)
431     {
432         printf ("ReferenceId: '%.*s'\n", id->len, id->buf);
433     }
434 }
435
436 static Z_ReferenceId *set_refid (ODR out)
437 {
438     Z_ReferenceId *id;
439     if (!refid)
440         return 0;
441     id = (Z_ReferenceId *) odr_malloc (out, sizeof(*id));
442     id->size = id->len = strlen(refid);
443     id->buf = (unsigned char *) odr_malloc (out, id->len);
444     memcpy (id->buf, refid, id->len);
445     return id;
446 }   
447
448 /* INIT SERVICE ------------------------------- */
449
450 static void send_initRequest()
451 {
452     Z_APDU *apdu = zget_APDU(out, Z_APDU_initRequest);
453     Z_InitRequest *req = apdu->u.initRequest;
454
455     ODR_MASK_SET(req->options, Z_Options_search);
456     ODR_MASK_SET(req->options, Z_Options_present);
457     ODR_MASK_SET(req->options, Z_Options_namedResultSets);
458     ODR_MASK_SET(req->options, Z_Options_triggerResourceCtrl);
459     ODR_MASK_SET(req->options, Z_Options_scan);
460     ODR_MASK_SET(req->options, Z_Options_sort);
461     ODR_MASK_SET(req->options, Z_Options_extendedServices);
462     ODR_MASK_SET(req->options, Z_Options_delSet);
463
464     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_1);
465     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_2);
466     ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_3);
467
468     *req->maximumRecordSize = 1024*1024;
469     *req->preferredMessageSize = 1024*1024;
470
471     req->idAuthentication = auth;
472
473     send_apdu(apdu);
474     printf("Sent initrequest.\n");
475 }
476
477 static int process_initResponse(Z_InitResponse *res)
478 {
479     /* save session parameters for later use */
480     session_mem = odr_extract_mem(in);
481     session = res;
482
483     if (!*res->result)
484         printf("Connection rejected by target.\n");
485     else
486         printf("Connection accepted by target.\n");
487     if (res->implementationId)
488         printf("ID     : %s\n", res->implementationId);
489     if (res->implementationName)
490         printf("Name   : %s\n", res->implementationName);
491     if (res->implementationVersion)
492         printf("Version: %s\n", res->implementationVersion);
493     if (res->userInformationField)
494     {
495         printf("UserInformationfield:\n");
496         if (!z_External(print, (Z_External**)&res-> userInformationField,
497             0, 0))
498         {
499             odr_perror(print, "Printing userinfo\n");
500             odr_reset(print);
501         }
502         if (res->userInformationField->which == Z_External_octet)
503         {
504             printf("Guessing visiblestring:\n");
505             printf("'%s'\n", res->userInformationField->u. octet_aligned->buf);
506         }
507         odr_reset (print);
508     }
509     return 0;
510 }
511
512 static int cmd_base(char *arg)
513 {
514     int i;
515     char *cp;
516
517     if (!*arg)
518     {
519         printf("Usage: base <database> <database> ...\n");
520         return 0;
521     }
522     for (i = 0; i<num_databaseNames; i++)
523         xfree (databaseNames[i]);
524     num_databaseNames = 0;
525     while (1)
526     {
527         if (!(cp = strchr(arg, ' ')))
528             cp = arg + strlen(arg);
529         if (cp - arg < 1)
530             break;
531         databaseNames[num_databaseNames] = (char *)xmalloc (1 + cp - arg);
532         memcpy (databaseNames[num_databaseNames], arg, cp - arg);
533         databaseNames[num_databaseNames++][cp - arg] = '\0';
534         if (!*cp)
535             break;
536         arg = cp+1;
537     }
538     return 1;
539 }
540
541
542 int cmd_open(char *arg)
543 {
544     void *add;
545     char type[100], addr[100], base[100];
546     CS_TYPE t;
547
548     if (conn)
549     {
550         printf("Already connected.\n");
551
552         cs_close (conn);
553         conn = NULL;
554         if (session_mem)
555         {
556             nmem_destroy (session_mem);
557             session_mem = NULL;
558         }
559     }
560     base[0] = '\0';
561     if (!*arg || sscanf(arg, "%[^:]:%[^/]/%s", type, addr, base) < 2)
562     {
563         fprintf(stderr, "Usage: open (osi|tcp) ':' [tsel '/']host[':'port]\n");
564         return 0;
565     }
566     if (*base)
567         cmd_base (base);
568     if (!strcmp(type, "tcp"))
569     {
570         t = tcpip_type;
571         protocol = PROTO_Z3950;
572     }
573     else
574 #ifdef USE_XTIMOSI
575     if (!strcmp(type, "osi"))
576     {
577         t = mosi_type;
578         protocol = PROTO_SR;
579     }
580     else
581 #endif
582     {
583         fprintf(stderr, "Bad type: %s\n", type);
584         return 0;
585     }
586     if (!(conn = cs_create(t, 1, protocol)))
587     {
588         perror("cs_create");
589         return 0;
590     }
591     if (!(add = cs_straddr(conn, addr)))
592     {
593         perror(arg);
594         return 0;
595     }
596     printf("Connecting...");
597     fflush(stdout);
598     if (cs_connect(conn, add) < 0)
599     {
600         perror("connect");
601         cs_close(conn);
602         conn = 0;
603         return 0;
604     }
605     printf("Ok.\n");
606     send_initRequest();
607     return 2;
608 }
609
610 int cmd_authentication(char *arg)
611 {
612     static Z_IdAuthentication au;
613     static char open[256];
614
615     if (!*arg)
616     {
617         printf("Auth field set to null\n");
618         auth = 0;
619         return 1;
620     }
621     auth = &au;
622     au.which = Z_IdAuthentication_open;
623     au.u.open = open;
624     strcpy(open, arg);
625     return 1;
626 }
627
628 /* SEARCH SERVICE ------------------------------ */
629
630 static void display_variant(Z_Variant *v, int level)
631 {
632     int i;
633
634     for (i = 0; i < v->num_triples; i++)
635     {
636         printf("%*sclass=%d,type=%d", level * 4, "", *v->triples[i]->zclass,
637             *v->triples[i]->type);
638         if (v->triples[i]->which == Z_Triple_internationalString)
639             printf(",value=%s\n", v->triples[i]->value.internationalString);
640         else
641             printf("\n");
642     }
643 }
644
645 static void display_grs1(Z_GenericRecord *r, int level)
646 {
647     int i;
648
649     if (!r)
650         return;
651     for (i = 0; i < r->num_elements; i++)
652     {
653         Z_TaggedElement *t;
654
655         printf("%*s", level * 4, "");
656         t = r->elements[i];
657         printf("(");
658         if (t->tagType)
659             printf("%d,", *t->tagType);
660         else
661             printf("?,");
662         if (t->tagValue->which == Z_StringOrNumeric_numeric)
663             printf("%d) ", *t->tagValue->u.numeric);
664         else
665             printf("%s) ", t->tagValue->u.string);
666         if (t->content->which == Z_ElementData_subtree)
667         {
668             printf("\n");
669             display_grs1(t->content->u.subtree, level+1);
670         }
671         else if (t->content->which == Z_ElementData_string)
672             printf("%s\n", t->content->u.string);
673         else if (t->content->which == Z_ElementData_numeric)
674             printf("%d\n", *t->content->u.numeric);
675         else if (t->content->which == Z_ElementData_oid)
676         {
677             int *ip = t->content->u.oid;
678             oident *oent;
679
680             if ((oent = oid_getentbyoid(t->content->u.oid)))
681                 printf("OID: %s\n", oent->desc);
682             else
683             {
684                 printf("{");
685                 while (ip && *ip >= 0)
686                     printf(" %d", *(ip++));
687                 printf(" }\n");
688             }
689         }
690         else if (t->content->which == Z_ElementData_noDataRequested)
691             printf("[No data requested]\n");
692         else if (t->content->which == Z_ElementData_elementEmpty)
693             printf("[Element empty]\n");
694         else if (t->content->which == Z_ElementData_elementNotThere)
695             printf("[Element not there]\n");
696         else
697             printf("??????\n");
698         if (t->appliedVariant)
699             display_variant(t->appliedVariant, level+1);
700         if (t->metaData && t->metaData->supportedVariants)
701         {
702             int c;
703
704             printf("%*s---- variant list\n", (level+1)*4, "");
705             for (c = 0; c < t->metaData->num_supportedVariants; c++)
706             {
707                 printf("%*svariant #%d\n", (level+1)*4, "", c);
708                 display_variant(t->metaData->supportedVariants[c], level + 2);
709             }
710         }
711     }
712 }
713
714 static void print_record(const unsigned char *buf, size_t len)
715 {
716    size_t i;
717    for (i = 0; i<len; i++)
718        if ((buf[i] <= 126 && buf[i] >= 32) || strchr ("\n\r\t\f", buf[i]))
719            fputc (buf[i], stdout);
720        else
721            printf ("\\X%02X", buf[i]);
722    /* add newline if not already added ... */
723    if (i <= 0 || buf[i-1] != '\n')
724        fputc ('\n', stdout);
725 }
726
727 static void display_record(Z_DatabaseRecord *p)
728 {
729     Z_External *r = (Z_External*) p;
730     oident *ent = oid_getentbyoid(r->direct_reference);
731
732     record_last = r;
733     /*
734      * Tell the user what we got.
735      */
736     if (r->direct_reference)
737     {
738         printf("Record type: ");
739         if (ent)
740             printf("%s\n", ent->desc);
741         else if (!odr_oid(print, &r->direct_reference, 0, 0))
742         {
743             odr_perror(print, "print oid");
744             odr_reset(print);
745         }
746     }
747     /* Check if this is a known, ASN.1 type tucked away in an octet string */
748     if (ent && r->which == Z_External_octet)
749     {
750         Z_ext_typeent *type = z_ext_getentbyref(ent->value);
751         void *rr;
752
753         if (type)
754         {
755             /*
756              * Call the given decoder to process the record.
757              */
758             odr_setbuf(in, (char*)p->u.octet_aligned->buf,
759                 p->u.octet_aligned->len, 0);
760             if (!(*type->fun)(in, (char **)&rr, 0, 0))
761             {
762                 odr_perror(in, "Decoding constructed record.");
763                 fprintf(stderr, "[Near %d]\n", odr_offset(in));
764                 fprintf(stderr, "Packet dump:\n---------\n");
765                 odr_dumpBER(stderr, (char*)p->u.octet_aligned->buf,
766                     p->u.octet_aligned->len);
767                 fprintf(stderr, "---------\n");
768                 exit(1);
769             }
770             /*
771              * Note: we throw away the original, BER-encoded record here.
772              * Do something else with it if you want to keep it.
773              */
774             r->u.sutrs = (Z_SUTRS *) rr; /* we don't actually check the type here. */
775             r->which = type->what;
776         }
777     }
778     if (ent && ent->value == VAL_SOIF)
779         print_record((const unsigned char *) r->u.octet_aligned->buf, r->u.octet_aligned->len);
780     else if (r->which == Z_External_octet && p->u.octet_aligned->len)
781     {
782         const char *octet_buf = (char*)p->u.octet_aligned->buf;
783         if (ent->value == VAL_TEXT_XML || ent->value == VAL_APPLICATION_XML ||
784             ent->value == VAL_HTML)
785             print_record((const unsigned char *) octet_buf,
786                          p->u.octet_aligned->len);
787         else
788         {
789             if (marc_display (octet_buf, NULL) <= 0)
790             {
791                 printf ("ISO2709 decoding failed, dumping record as is:\n");
792                 print_record((const unsigned char*) octet_buf,
793                               p->u.octet_aligned->len);
794             }
795         }
796         if (marcdump)
797             fwrite (octet_buf, 1, p->u.octet_aligned->len, marcdump);
798     }
799     else if (ent && ent->value == VAL_SUTRS)
800     {
801         if (r->which != Z_External_sutrs)
802         {
803             printf("Expecting single SUTRS type for SUTRS.\n");
804             return;
805         }
806         print_record(r->u.sutrs->buf, r->u.sutrs->len);
807     }
808     else if (ent && ent->value == VAL_GRS1)
809     {
810         if (r->which != Z_External_grs1)
811         {
812             printf("Expecting single GRS type for GRS.\n");
813             return;
814         }
815         display_grs1(r->u.grs1, 0);
816     }
817     else 
818     {
819         printf("Unknown record representation.\n");
820         if (!z_External(print, &r, 0, 0))
821         {
822             odr_perror(print, "Printing external");
823             odr_reset(print);
824         }
825     }
826 }
827
828
829 static void display_diagrecs(Z_DiagRec **pp, int num)
830 {
831     int i;
832     oident *ent;
833     Z_DefaultDiagFormat *r;
834
835     printf("Diagnostic message(s) from database:\n");
836     for (i = 0; i<num; i++)
837     {
838         Z_DiagRec *p = pp[i];
839         if (p->which != Z_DiagRec_defaultFormat)
840         {
841             printf("Diagnostic record not in default format.\n");
842             return;
843         }
844         else
845             r = p->u.defaultFormat;
846         if (!(ent = oid_getentbyoid(r->diagnosticSetId)) ||
847             ent->oclass != CLASS_DIAGSET || ent->value != VAL_BIB1)
848             printf("Missing or unknown diagset\n");
849         printf("    [%d] %s", *r->condition, diagbib1_str(*r->condition));
850 #ifdef ASN_COMPILED
851         switch (r->which)
852         {
853         case Z_DefaultDiagFormat_v2Addinfo:
854             printf (" -- v2 addinfo '%s'\n", r->u.v2Addinfo);
855             break;
856         case Z_DefaultDiagFormat_v3Addinfo:
857             printf (" -- v3 addinfo '%s'\n", r->u.v3Addinfo);
858             break;
859         }
860 #else
861         if (r->addinfo && *r->addinfo)
862             printf(" -- '%s'\n", r->addinfo);
863         else
864             printf("\n");
865 #endif
866     }
867 }
868
869
870 static void display_nameplusrecord(Z_NamePlusRecord *p)
871 {
872     if (p->databaseName)
873         printf("[%s]", p->databaseName);
874     if (p->which == Z_NamePlusRecord_surrogateDiagnostic)
875         display_diagrecs(&p->u.surrogateDiagnostic, 1);
876     else if (p->which == Z_NamePlusRecord_databaseRecord)
877         display_record(p->u.databaseRecord);
878 }
879
880 static void display_records(Z_Records *p)
881 {
882     int i;
883
884     if (p->which == Z_Records_NSD)
885     {
886 #ifdef ASN_COMPILED
887         Z_DiagRec dr, *dr_p = &dr;
888         dr.which = Z_DiagRec_defaultFormat;
889         dr.u.defaultFormat = p->u.nonSurrogateDiagnostic;
890         display_diagrecs (&dr_p, 1);
891 #else
892         display_diagrecs (&p->u.nonSurrogateDiagnostic, 1);
893 #endif
894     }
895     else if (p->which == Z_Records_multipleNSD)
896         display_diagrecs (p->u.multipleNonSurDiagnostics->diagRecs,
897                           p->u.multipleNonSurDiagnostics->num_diagRecs);
898     else 
899     {
900         printf("Records: %d\n", p->u.databaseOrSurDiagnostics->num_records);
901         for (i = 0; i < p->u.databaseOrSurDiagnostics->num_records; i++)
902             display_nameplusrecord(p->u.databaseOrSurDiagnostics->records[i]);
903     }
904 }
905
906 static int send_deleteResultSetRequest(char *arg)
907 {
908     char names[8][32];
909     int i;
910
911     Z_APDU *apdu = zget_APDU(out, Z_APDU_deleteResultSetRequest);
912     Z_DeleteResultSetRequest *req = apdu->u.deleteResultSetRequest;
913
914     req->referenceId = set_refid (out);
915
916     req->num_resultSetList =
917         sscanf (arg, "%30s %30s %30s %30s %30s %30s %30s %30s",
918                 names[0], names[1], names[2], names[3],
919                 names[4], names[5], names[6], names[7]);
920
921     req->deleteFunction = (int *)
922         odr_malloc (out, sizeof(*req->deleteFunction));
923     if (req->num_resultSetList > 0)
924     {
925         *req->deleteFunction = Z_DeleteRequest_list;
926         req->resultSetList = (char **)
927             odr_malloc (out, sizeof(*req->resultSetList)*
928                         req->num_resultSetList);
929         for (i = 0; i<req->num_resultSetList; i++)
930             req->resultSetList[i] = names[i];
931     }
932     else
933     {
934         *req->deleteFunction = Z_DeleteRequest_all;
935         req->resultSetList = 0;
936     }
937     
938     send_apdu(apdu);
939     printf("Sent deleteResultSetRequest.\n");
940     return 2;
941 }
942
943 static int send_searchRequest(char *arg)
944 {
945     Z_APDU *apdu = zget_APDU(out, Z_APDU_searchRequest);
946     Z_SearchRequest *req = apdu->u.searchRequest;
947     Z_Query query;
948     int oid[OID_SIZE];
949 #if CCL2RPN
950     struct ccl_rpn_node *rpn = NULL;
951     int error, pos;
952     oident bib1;
953 #endif
954     char setstring[100];
955     Z_RPNQuery *RPNquery;
956     Odr_oct ccl_query;
957
958 #if CCL2RPN
959     if (queryType == QueryType_CCL2RPN)
960     {
961         rpn = ccl_find_str(bibset, arg, &error, &pos);
962         if (error)
963         {
964             printf("CCL ERROR: %s\n", ccl_err_msg(error));
965             return 0;
966         }
967     }
968 #endif
969     req->referenceId = set_refid (out);
970     if (!strcmp(arg, "@big")) /* strictly for troublemaking */
971     {
972         static unsigned char big[2100];
973         static Odr_oct bigo;
974
975         /* send a very big referenceid to test transport stack etc. */
976         memset(big, 'A', 2100);
977         bigo.len = bigo.size = 2100;
978         bigo.buf = big;
979         req->referenceId = &bigo;
980     }
981     
982     if (setnumber >= 0)
983     {
984         sprintf(setstring, "%d", ++setnumber);
985         req->resultSetName = setstring;
986     }
987     *req->smallSetUpperBound = smallSetUpperBound;
988     *req->largeSetLowerBound = largeSetLowerBound;
989     *req->mediumSetPresentNumber = mediumSetPresentNumber;
990     if (smallSetUpperBound > 0 || (largeSetLowerBound > 1 &&
991         mediumSetPresentNumber > 0))
992     {
993         oident prefsyn;
994
995         prefsyn.proto = protocol;
996         prefsyn.oclass = CLASS_RECSYN;
997         prefsyn.value = recordsyntax;
998         req->preferredRecordSyntax =
999             odr_oiddup(out, oid_ent_to_oid(&prefsyn, oid));
1000         req->smallSetElementSetNames =
1001             req->mediumSetElementSetNames = elementSetNames;
1002     }
1003     req->num_databaseNames = num_databaseNames;
1004     req->databaseNames = databaseNames;
1005
1006     req->query = &query;
1007
1008     switch (queryType)
1009     {
1010     case QueryType_Prefix:
1011         query.which = Z_Query_type_1;
1012         RPNquery = p_query_rpn (out, protocol, arg);
1013         if (!RPNquery)
1014         {
1015             printf("Prefix query error\n");
1016             return 0;
1017         }
1018         query.u.type_1 = RPNquery;
1019         break;
1020     case QueryType_CCL:
1021         query.which = Z_Query_type_2;
1022         query.u.type_2 = &ccl_query;
1023         ccl_query.buf = (unsigned char*) arg;
1024         ccl_query.len = strlen(arg);
1025         break;
1026 #if CCL2RPN
1027     case QueryType_CCL2RPN:
1028         query.which = Z_Query_type_1;
1029         RPNquery = ccl_rpn_query(out, rpn);
1030         if (!RPNquery)
1031         {
1032             printf ("Couldn't convert from CCL to RPN\n");
1033             return 0;
1034         }
1035         bib1.proto = protocol;
1036         bib1.oclass = CLASS_ATTSET;
1037         bib1.value = VAL_BIB1;
1038         RPNquery->attributeSetId = oid_ent_to_oid(&bib1, oid);
1039         query.u.type_1 = RPNquery;
1040         ccl_rpn_delete (rpn);
1041         break;
1042 #endif
1043     default:
1044         printf ("Unsupported query type\n");
1045         return 0;
1046     }
1047     send_apdu(apdu);
1048     setno = 1;
1049     printf("Sent searchRequest.\n");
1050     return 2;
1051 }
1052
1053 static int process_searchResponse(Z_SearchResponse *res)
1054 {
1055     printf ("Received SearchResponse.\n");
1056     print_refid (res->referenceId);
1057     if (*res->searchStatus)
1058         printf("Search was a success.\n");
1059     else
1060         printf("Search was a bloomin' failure.\n");
1061     printf("Number of hits: %d, setno %d\n",
1062         *res->resultCount, setnumber);
1063     printf("records returned: %d\n",
1064         *res->numberOfRecordsReturned);
1065     setno += *res->numberOfRecordsReturned;
1066     if (res->records)
1067         display_records(res->records);
1068     return 0;
1069 }
1070
1071 static void print_level(int iLevel)
1072 {
1073     int i;
1074     for (i = 0; i < iLevel * 4; i++)
1075         printf(" ");
1076 }
1077
1078 static void print_int(int iLevel, const char *pTag, int *pInt)
1079 {
1080     if (pInt != NULL)
1081     {
1082         print_level(iLevel);
1083         printf("%s: %d\n", pTag, *pInt);
1084     }
1085 }
1086
1087 static void print_string(int iLevel, const char *pTag, const char *pString)
1088 {
1089     if (pString != NULL)
1090     {
1091         print_level(iLevel);
1092         printf("%s: %s\n", pTag, pString);
1093     }
1094 }
1095
1096 static void print_oid(int iLevel, const char *pTag, Odr_oid *pOid)
1097 {
1098     if (pOid != NULL)
1099     {
1100         int *pInt = pOid;
1101
1102         print_level(iLevel);
1103         printf("%s:", pTag);
1104         for (; *pInt != -1; pInt++)
1105             printf(" %d", *pInt);
1106         printf("\n");
1107     }
1108 }
1109
1110 static void print_referenceId(int iLevel, Z_ReferenceId *referenceId)
1111 {
1112     if (referenceId != NULL)
1113     {
1114         int i;
1115
1116         print_level(iLevel);
1117         printf("Ref Id (%d, %d): ", referenceId->len, referenceId->size);
1118         for (i = 0; i < referenceId->len; i++)
1119             printf("%c", referenceId->buf[i]);
1120         printf("\n");
1121     }
1122 }
1123
1124 static void print_string_or_numeric(int iLevel, const char *pTag, Z_StringOrNumeric *pStringNumeric)
1125 {
1126     if (pStringNumeric != NULL)
1127     {
1128         switch (pStringNumeric->which)
1129         {
1130             case Z_StringOrNumeric_string:
1131                 print_string(iLevel, pTag, pStringNumeric->u.string);
1132                 break;
1133
1134             case Z_StringOrNumeric_numeric:
1135                 print_int(iLevel, pTag, pStringNumeric->u.numeric);
1136                 break;
1137
1138             default:
1139                 print_level(iLevel);
1140                 printf("%s: valid type for Z_StringOrNumeric\n", pTag);
1141                 break;
1142         }
1143     }
1144 }
1145
1146 static void print_universe_report_duplicate(int iLevel, Z_UniverseReportDuplicate *pUniverseReportDuplicate)
1147 {
1148     if (pUniverseReportDuplicate != NULL)
1149     {
1150         print_level(iLevel);
1151         printf("Universe Report Duplicate: \n");
1152         iLevel++;
1153         print_string_or_numeric(iLevel, "Hit No", pUniverseReportDuplicate->hitno);
1154     }
1155 }
1156
1157 static void print_universe_report_hits(int iLevel, Z_UniverseReportHits *pUniverseReportHits)
1158 {
1159     if (pUniverseReportHits != NULL)
1160     {
1161         print_level(iLevel);
1162         printf("Universe Report Hits: \n");
1163         iLevel++;
1164         print_string_or_numeric(iLevel, "Database", pUniverseReportHits->database);
1165         print_string_or_numeric(iLevel, "Hits", pUniverseReportHits->hits);
1166     }
1167 }
1168
1169 static void print_universe_report(int iLevel, Z_UniverseReport *pUniverseReport)
1170 {
1171     if (pUniverseReport != NULL)
1172     {
1173         print_level(iLevel);
1174         printf("Universe Report: \n");
1175         iLevel++;
1176         print_int(iLevel, "Total Hits", pUniverseReport->totalHits);
1177         switch (pUniverseReport->which)
1178         {
1179             case Z_UniverseReport_databaseHits:
1180                 print_universe_report_hits(iLevel, pUniverseReport->u.databaseHits);
1181                 break;
1182
1183             case Z_UniverseReport_duplicate:
1184                 print_universe_report_duplicate(iLevel, pUniverseReport->u.duplicate);
1185                 break;
1186
1187             default:
1188                 print_level(iLevel);
1189                 printf("Type: %d\n", pUniverseReport->which);
1190                 break;
1191         }
1192     }
1193 }
1194
1195 static void print_external(int iLevel, Z_External *pExternal)
1196 {
1197     if (pExternal != NULL)
1198     {
1199         print_level(iLevel);
1200         printf("External: \n");
1201         iLevel++;
1202         print_oid(iLevel, "Direct Reference", pExternal->direct_reference);
1203         print_int(iLevel, "InDirect Reference", pExternal->indirect_reference);
1204         print_string(iLevel, "Descriptor", pExternal->descriptor);
1205         switch (pExternal->which)
1206         {
1207             case Z_External_universeReport:
1208                 print_universe_report(iLevel, pExternal->u.universeReport);
1209                 break;
1210
1211             default:
1212                 print_level(iLevel);
1213                 printf("Type: %d\n", pExternal->which);
1214                 break;
1215         }
1216     }
1217 }
1218
1219 static int process_resourceControlRequest (Z_ResourceControlRequest *req)
1220 {
1221     printf ("Received ResourceControlRequest.\n");
1222     print_referenceId(1, req->referenceId);
1223     print_int(1, "Suspended Flag", req->suspendedFlag);
1224     print_int(1, "Partial Results Available", req->partialResultsAvailable);
1225     print_int(1, "Response Required", req->responseRequired);
1226     print_int(1, "Triggered Request Flag", req->triggeredRequestFlag);
1227     print_external(1, req->resourceReport);
1228     return 0;
1229 }
1230
1231 void process_ESResponse(Z_ExtendedServicesResponse *res)
1232 {
1233     printf("process_ESResponse status=");
1234     switch (*res->operationStatus)
1235     {
1236     case Z_ExtendedServicesResponse_done:
1237         printf ("done\n");
1238         break;
1239     case Z_ExtendedServicesResponse_accepted:
1240         printf ("accepted\n");
1241         break;
1242     case Z_ExtendedServicesResponse_failure:
1243         printf ("failure\n");
1244         display_diagrecs(res->diagnostics, res->num_diagnostics);
1245         break;
1246     }
1247 }
1248
1249 #ifdef ASN_COMPILED
1250
1251 const char *get_ill_element (void *clientData, const char *element)
1252 {
1253     /* printf ("asking for %s\n", element); */
1254     if (!strcmp (element, "ill,transaction-id,transaction-group-qualifier"))
1255         return "1";
1256     if (!strcmp (element, "ill,transaction-id,transaction-qualifier"))
1257         return "1";
1258     return 0;
1259 }
1260
1261 static Z_External *create_external_itemRequest()
1262 {
1263     struct ill_get_ctl ctl;
1264     ILL_ItemRequest *req;
1265     Z_External *r = 0;
1266     int item_request_size = 0;
1267     char *item_request_buf = 0;
1268
1269     ctl.odr = out;
1270     ctl.clientData = 0;
1271     ctl.f = get_ill_element;
1272
1273     req = ill_get_ItemRequest(&ctl, "ill", 0);
1274
1275     if (!ill_ItemRequest (out, &req, 0, 0))
1276     {
1277         if (apdu_file)
1278         {
1279             ill_ItemRequest(print, &req, 0, 0);
1280             odr_reset(print);
1281         }
1282         item_request_buf = odr_getbuf (out, &item_request_size, 0);
1283         if (item_request_buf)
1284             odr_setbuf (out, item_request_buf, item_request_size, 1);
1285         printf ("Couldn't encode ItemRequest, size %d\n", item_request_size);
1286         return 0;
1287     }
1288     else
1289     {
1290         oident oid;
1291         
1292         item_request_buf = odr_getbuf (out, &item_request_size, 0);
1293         oid.proto = PROTO_GENERAL;
1294         oid.oclass = CLASS_GENERAL;
1295         oid.value = VAL_ISO_ILL_1;
1296         
1297         r = (Z_External *) odr_malloc (out, sizeof(*r));
1298         r->direct_reference = odr_oiddup(out,oid_getoidbyent(&oid)); 
1299         r->indirect_reference = 0;
1300         r->descriptor = 0;
1301         r->which = Z_External_single;
1302         
1303         r->u.single_ASN1_type = (Odr_oct *)
1304             odr_malloc (out, sizeof(*r->u.single_ASN1_type));
1305         r->u.single_ASN1_type->buf = odr_malloc (out, item_request_size);
1306         r->u.single_ASN1_type->len = item_request_size;
1307         r->u.single_ASN1_type->size = item_request_size;
1308         memcpy (r->u.single_ASN1_type->buf, item_request_buf,
1309                 item_request_size);
1310         printf ("len = %d\n", item_request_size);
1311     }
1312     return r;
1313 }
1314 #else
1315 static Z_External *create_external_itemRequest()
1316 {
1317     return 0;
1318 }
1319 #endif
1320
1321 #ifdef ASN_COMPILED
1322 static Z_External *create_external_ILLRequest()
1323 {
1324     struct ill_get_ctl ctl;
1325     ILL_Request *req;
1326     Z_External *r = 0;
1327     int ill_request_size = 0;
1328     char *ill_request_buf = 0;
1329         
1330     ctl.odr = out;
1331     ctl.clientData = 0;
1332     ctl.f = get_ill_element;
1333
1334     req = ill_get_ILLRequest(&ctl, "ill", 0);
1335
1336     if (!ill_Request (out, &req, 0, 0))
1337     {
1338         if (apdu_file)
1339         {
1340             printf ("-------------------\n");
1341             ill_Request(print, &req, 0, 0);
1342             odr_reset(print);
1343             printf ("-------------------\n");
1344         }
1345         ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
1346         if (ill_request_buf)
1347             odr_setbuf (out, ill_request_buf, ill_request_size, 1);
1348         printf ("Couldn't encode ILL-Request, size %d\n", ill_request_size);
1349         return 0;
1350     }
1351     else
1352     {
1353         oident oid;
1354         ill_request_buf = odr_getbuf (out, &ill_request_size, 0);
1355         
1356         oid.proto = PROTO_GENERAL;
1357         oid.oclass = CLASS_GENERAL;
1358         oid.value = VAL_ISO_ILL_1;
1359         
1360         r = (Z_External *) odr_malloc (out, sizeof(*r));
1361         r->direct_reference = odr_oiddup(out,oid_getoidbyent(&oid)); 
1362         r->indirect_reference = 0;
1363         r->descriptor = 0;
1364         r->which = Z_External_single;
1365         
1366         r->u.single_ASN1_type = (Odr_oct *)
1367             odr_malloc (out, sizeof(*r->u.single_ASN1_type));
1368         r->u.single_ASN1_type->buf = odr_malloc (out, ill_request_size);
1369         r->u.single_ASN1_type->len = ill_request_size;
1370         r->u.single_ASN1_type->size = ill_request_size;
1371         memcpy (r->u.single_ASN1_type->buf, ill_request_buf, ill_request_size);
1372         printf ("len = %d\n", ill_request_size);
1373     }
1374     return r;
1375 }
1376 #else
1377 static Z_External *create_external_ILLRequest()
1378 {
1379     return 0;
1380 }
1381 #endif
1382
1383
1384 static Z_External *create_ItemOrderExternal(const char *type, int itemno)
1385 {
1386     Z_External *r = (Z_External *) odr_malloc(out, sizeof(Z_External));
1387     oident ItemOrderRequest;
1388   
1389     ItemOrderRequest.proto = PROTO_Z3950;
1390     ItemOrderRequest.oclass = CLASS_EXTSERV;
1391     ItemOrderRequest.value = VAL_ITEMORDER;
1392  
1393     r->direct_reference = odr_oiddup(out,oid_getoidbyent(&ItemOrderRequest)); 
1394     r->indirect_reference = 0;
1395     r->descriptor = 0;
1396
1397     r->which = Z_External_itemOrder;
1398
1399     r->u.itemOrder = (Z_ItemOrder *) odr_malloc(out,sizeof(Z_ItemOrder));
1400     memset(r->u.itemOrder, 0, sizeof(Z_ItemOrder));
1401 #ifdef ASN_COMPILED
1402     r->u.itemOrder->which=Z_IOItemOrder_esRequest;
1403 #else
1404     r->u.itemOrder->which=Z_ItemOrder_esRequest;
1405 #endif
1406
1407     r->u.itemOrder->u.esRequest = (Z_IORequest *) 
1408         odr_malloc(out,sizeof(Z_IORequest));
1409     memset(r->u.itemOrder->u.esRequest, 0, sizeof(Z_IORequest));
1410
1411     r->u.itemOrder->u.esRequest->toKeep = (Z_IOOriginPartToKeep *)
1412         odr_malloc(out,sizeof(Z_IOOriginPartToKeep));
1413     memset(r->u.itemOrder->u.esRequest->toKeep, 0, sizeof(Z_IOOriginPartToKeep));
1414     r->u.itemOrder->u.esRequest->notToKeep = (Z_IOOriginPartNotToKeep *)
1415         odr_malloc(out,sizeof(Z_IOOriginPartNotToKeep));
1416     memset(r->u.itemOrder->u.esRequest->notToKeep, 0, sizeof(Z_IOOriginPartNotToKeep));
1417
1418     r->u.itemOrder->u.esRequest->toKeep->supplDescription = NULL;
1419     r->u.itemOrder->u.esRequest->toKeep->contact = NULL;
1420     r->u.itemOrder->u.esRequest->toKeep->addlBilling = NULL;
1421
1422     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem =
1423         (Z_IOResultSetItem *) odr_malloc(out, sizeof(Z_IOResultSetItem));
1424     memset(r->u.itemOrder->u.esRequest->notToKeep->resultSetItem, 0, sizeof(Z_IOResultSetItem));
1425     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->resultSetId = "1";
1426
1427     r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->item =
1428         (int *) odr_malloc(out, sizeof(int));
1429     *r->u.itemOrder->u.esRequest->notToKeep->resultSetItem->item = itemno;
1430
1431     switch (*type)
1432     {
1433     case '2':
1434         printf ("using item-request\n");
1435         r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 
1436             create_external_itemRequest();
1437         break;
1438     case '1':
1439         printf ("using ILL-request\n");
1440         r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 
1441             create_external_ILLRequest();
1442         break;
1443     default:
1444         r->u.itemOrder->u.esRequest->notToKeep->itemRequest = 0;
1445     }
1446     return r;
1447 }
1448
1449 static int send_itemorder(const char *type, int itemno)
1450 {
1451     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest );
1452     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
1453     oident ItemOrderRequest;
1454
1455     ItemOrderRequest.proto = PROTO_Z3950;
1456     ItemOrderRequest.oclass = CLASS_EXTSERV;
1457     ItemOrderRequest.value = VAL_ITEMORDER;
1458     req->packageType = odr_oiddup(out,oid_getoidbyent(&ItemOrderRequest));
1459     req->packageName = "1.Extendedserveq";
1460
1461     req->taskSpecificParameters = create_ItemOrderExternal(type, itemno);
1462
1463     send_apdu(apdu);
1464     return 0;
1465 }
1466
1467 static int cmd_update(char *arg)
1468 {
1469     Z_APDU *apdu = zget_APDU(out, Z_APDU_extendedServicesRequest );
1470     Z_ExtendedServicesRequest *req = apdu->u.extendedServicesRequest;
1471     Z_External *r;
1472     int oid[OID_SIZE];
1473     Z_IUOriginPartToKeep *toKeep;
1474     Z_IUSuppliedRecords *notToKeep;
1475     oident update_oid;
1476     printf ("Update request\n");
1477     fflush(stdout);
1478
1479     if (!record_last)
1480         return 0;
1481     update_oid.proto = PROTO_Z3950;
1482     update_oid.oclass = CLASS_EXTSERV;
1483     update_oid.value = VAL_DBUPDATE;
1484     oid_ent_to_oid (&update_oid, oid);
1485     req->packageType = odr_oiddup(out,oid);
1486     req->packageName = "1.Extendedserveq";
1487
1488     r = req->taskSpecificParameters = (Z_External *)
1489         odr_malloc (out, sizeof(*r));
1490     r->direct_reference = odr_oiddup(out,oid);
1491     r->indirect_reference = 0;
1492     r->descriptor = 0;
1493     r->which = Z_External_update;
1494     r->u.update = (Z_IUUpdate *) odr_malloc(out, sizeof(*r->u.update));
1495     r->u.update->which = Z_IUUpdate_esRequest;
1496     r->u.update->u.esRequest = (Z_IUUpdateEsRequest *)
1497         odr_malloc(out, sizeof(*r->u.update->u.esRequest));
1498     toKeep = r->u.update->u.esRequest->toKeep = (Z_IUOriginPartToKeep *)
1499         odr_malloc(out, sizeof(*r->u.update->u.esRequest->toKeep));
1500     toKeep->databaseName = databaseNames[0];
1501     toKeep->schema = 0;
1502     toKeep->elementSetName = 0;
1503     toKeep->actionQualifier = 0;
1504     toKeep->action = (int *) odr_malloc(out, sizeof(*toKeep->action));
1505     *toKeep->action = Z_IUOriginPartToKeep_recordInsert;
1506
1507     notToKeep = r->u.update->u.esRequest->notToKeep = (Z_IUSuppliedRecords *)
1508         odr_malloc(out, sizeof(*r->u.update->u.esRequest->notToKeep));
1509     notToKeep->num = 1;
1510     notToKeep->elements = (Z_IUSuppliedRecords_elem **)
1511         odr_malloc(out, sizeof(*notToKeep->elements));
1512     notToKeep->elements[0] = (Z_IUSuppliedRecords_elem *)
1513         odr_malloc(out, sizeof(**notToKeep->elements));
1514     notToKeep->elements[0]->u.number = 0;
1515     notToKeep->elements[0]->supplementalId = 0;
1516     notToKeep->elements[0]->correlationInfo = 0;
1517     notToKeep->elements[0]->record = record_last;
1518     
1519     send_apdu(apdu);
1520
1521     return 2;
1522 }
1523
1524 /* II : Added to do DALI Item Order Extended services request */
1525 static int cmd_itemorder(char *arg)
1526 {
1527     char type[12];
1528     int itemno;
1529
1530     if (sscanf (arg, "%10s %d", type, &itemno) != 2)
1531         return 0;
1532
1533     printf("Item order request\n");
1534     fflush(stdout);
1535     send_itemorder(type, itemno);
1536     return(2);
1537 }
1538
1539 static int cmd_find(char *arg)
1540 {
1541     if (!*arg)
1542     {
1543         printf("Find what?\n");
1544         return 0;
1545     }
1546     if (!conn)
1547     {
1548         printf("Not connected yet\n");
1549         return 0;
1550     }
1551     if (!send_searchRequest(arg))
1552         return 0;
1553     return 2;
1554 }
1555
1556 static int cmd_delete(char *arg)
1557 {
1558     if (!conn)
1559     {
1560         printf("Not connected yet\n");
1561         return 0;
1562     }
1563     if (!send_deleteResultSetRequest(arg))
1564         return 0;
1565     return 2;
1566 }
1567
1568 static int cmd_ssub(char *arg)
1569 {
1570     if (!(smallSetUpperBound = atoi(arg)))
1571         return 0;
1572     return 1;
1573 }
1574
1575 static int cmd_lslb(char *arg)
1576 {
1577     if (!(largeSetLowerBound = atoi(arg)))
1578         return 0;
1579     return 1;
1580 }
1581
1582 static int cmd_mspn(char *arg)
1583 {
1584     if (!(mediumSetPresentNumber = atoi(arg)))
1585         return 0;
1586     return 1;
1587 }
1588
1589 static int cmd_status(char *arg)
1590 {
1591     printf("smallSetUpperBound: %d\n", smallSetUpperBound);
1592     printf("largeSetLowerBound: %d\n", largeSetLowerBound);
1593     printf("mediumSetPresentNumber: %d\n", mediumSetPresentNumber);
1594     return 1;
1595 }
1596
1597 static int cmd_setnames(char *arg)
1598 {
1599     if (setnumber < 0)
1600     {
1601         printf("Set numbering enabled.\n");
1602         setnumber = 0;
1603     }
1604     else
1605     {
1606         printf("Set numbering disabled.\n");
1607         setnumber = -1;
1608     }
1609     return 1;
1610 }
1611
1612 /* PRESENT SERVICE ----------------------------- */
1613
1614 static int send_presentRequest(char *arg)
1615 {
1616     Z_APDU *apdu = zget_APDU(out, Z_APDU_presentRequest);
1617     Z_PresentRequest *req = apdu->u.presentRequest;
1618     Z_RecordComposition compo;
1619     oident prefsyn;
1620     int nos = 1;
1621     int oid[OID_SIZE];
1622     char *p;
1623     char setstring[100];
1624
1625     req->referenceId = set_refid (out);
1626     if ((p = strchr(arg, '+')))
1627     {
1628         nos = atoi(p + 1);
1629         *p = 0;
1630     }
1631     if (*arg)
1632         setno = atoi(arg);
1633     if (p && (p=strchr(p+1, '+')))
1634     {
1635         strcpy (setstring, p+1);
1636         req->resultSetId = setstring;
1637     }
1638     else if (setnumber >= 0)
1639     {
1640         sprintf(setstring, "%d", setnumber);
1641         req->resultSetId = setstring;
1642     }
1643 #if 0
1644     if (1)
1645     {
1646         static Z_Range range;
1647         static Z_Range *rangep = &range;
1648     req->num_ranges = 1;
1649 #endif
1650     req->resultSetStartPoint = &setno;
1651     req->numberOfRecordsRequested = &nos;
1652     prefsyn.proto = protocol;
1653     prefsyn.oclass = CLASS_RECSYN;
1654     prefsyn.value = recordsyntax;
1655     req->preferredRecordSyntax =
1656         odr_oiddup (out, oid_ent_to_oid(&prefsyn, oid));
1657
1658     if (schema != VAL_NONE)
1659     {
1660         oident prefschema;
1661
1662         prefschema.proto = protocol;
1663         prefschema.oclass = CLASS_SCHEMA;
1664         prefschema.value = schema;
1665
1666         req->recordComposition = &compo;
1667         compo.which = Z_RecordComp_complex;
1668         compo.u.complex = (Z_CompSpec *)
1669             odr_malloc(out, sizeof(*compo.u.complex));
1670         compo.u.complex->selectAlternativeSyntax = (bool_t *) 
1671             odr_malloc(out, sizeof(bool_t));
1672         *compo.u.complex->selectAlternativeSyntax = 0;
1673
1674         compo.u.complex->generic = (Z_Specification *)
1675             odr_malloc(out, sizeof(*compo.u.complex->generic));
1676         compo.u.complex->generic->schema = (Odr_oid *)
1677             odr_oiddup(out, oid_ent_to_oid(&prefschema, oid));
1678         if (!compo.u.complex->generic->schema)
1679         {
1680             /* OID wasn't a schema! Try record syntax instead. */
1681             prefschema.oclass = CLASS_RECSYN;
1682             compo.u.complex->generic->schema = (Odr_oid *)
1683                 odr_oiddup(out, oid_ent_to_oid(&prefschema, oid));
1684         }
1685         if (!elementSetNames)
1686             compo.u.complex->generic->elementSpec = 0;
1687         else
1688         {
1689             compo.u.complex->generic->elementSpec = (Z_ElementSpec *)
1690                 odr_malloc(out, sizeof(Z_ElementSpec));
1691             compo.u.complex->generic->elementSpec->which =
1692                 Z_ElementSpec_elementSetName;
1693             compo.u.complex->generic->elementSpec->u.elementSetName =
1694                 elementSetNames->u.generic;
1695         }
1696         compo.u.complex->num_dbSpecific = 0;
1697         compo.u.complex->dbSpecific = 0;
1698         compo.u.complex->num_recordSyntax = 0;
1699         compo.u.complex->recordSyntax = 0;
1700     }
1701     else if (elementSetNames)
1702     {
1703         req->recordComposition = &compo;
1704         compo.which = Z_RecordComp_simple;
1705         compo.u.simple = elementSetNames;
1706     }
1707     send_apdu(apdu);
1708     printf("Sent presentRequest (%d+%d).\n", setno, nos);
1709     return 2;
1710 }
1711
1712 void process_close(Z_Close *req)
1713 {
1714     Z_APDU *apdu = zget_APDU(out, Z_APDU_close);
1715     Z_Close *res = apdu->u.close;
1716
1717     static char *reasons[] =
1718     {
1719         "finished",
1720         "shutdown",
1721         "system problem",
1722         "cost limit reached",
1723         "resources",
1724         "security violation",
1725         "protocolError",
1726         "lack of activity",
1727         "peer abort",
1728         "unspecified"
1729     };
1730
1731     printf("Reason: %s, message: %s\n", reasons[*req->closeReason],
1732         req->diagnosticInformation ? req->diagnosticInformation : "NULL");
1733     if (sent_close)
1734     {
1735         cs_close (conn);
1736         conn = NULL;
1737         if (session_mem)
1738         {
1739             nmem_destroy (session_mem);
1740             session_mem = NULL;
1741         }
1742         sent_close = 0;
1743     }
1744     else
1745     {
1746         *res->closeReason = Z_Close_finished;
1747         send_apdu(apdu);
1748         printf("Sent response.\n");
1749         sent_close = 1;
1750     }
1751 }
1752
1753 static int cmd_show(char *arg)
1754 {
1755     if (!conn)
1756     {
1757         printf("Not connected yet\n");
1758         return 0;
1759     }
1760     if (!send_presentRequest(arg))
1761         return 0;
1762     return 2;
1763 }
1764
1765 int cmd_quit(char *arg)
1766 {
1767     printf("See you later, alligator.\n");
1768     exit(0);
1769     return 0;
1770 }
1771
1772 int cmd_cancel(char *arg)
1773 {
1774     Z_APDU *apdu = zget_APDU(out, Z_APDU_triggerResourceControlRequest);
1775     Z_TriggerResourceControlRequest *req =
1776         apdu->u.triggerResourceControlRequest;
1777     bool_t rfalse = 0;
1778     
1779     if (!conn)
1780     {
1781         printf("Session not initialized yet\n");
1782         return 0;
1783     }
1784     if (!ODR_MASK_GET(session->options, Z_Options_triggerResourceCtrl))
1785     {
1786         printf("Target doesn't support cancel (trigger resource ctrl)\n");
1787         return 0;
1788     }
1789     *req->requestedAction = Z_TriggerResourceCtrl_cancel;
1790     req->resultSetWanted = &rfalse;
1791
1792     send_apdu(apdu);
1793     printf("Sent cancel request\n");
1794     return 2;
1795 }
1796
1797 int send_scanrequest(char *string, int pp, int num)
1798 {
1799     Z_APDU *apdu = zget_APDU(out, Z_APDU_scanRequest);
1800     Z_ScanRequest *req = apdu->u.scanRequest;
1801     
1802     if (!(req->termListAndStartPoint =
1803           p_query_scan(out, protocol, &req->attributeSet, string)))
1804     {
1805         printf("Prefix query error\n");
1806         return -1;
1807     }
1808     req->referenceId = set_refid (out);
1809     req->num_databaseNames = num_databaseNames;
1810     req->databaseNames = databaseNames;
1811     req->numberOfTermsRequested = &num;
1812     req->preferredPositionInResponse = &pp;
1813     send_apdu(apdu);
1814     return 2;
1815 }
1816
1817 int send_sortrequest(char *arg, int newset)
1818 {
1819     Z_APDU *apdu = zget_APDU(out, Z_APDU_sortRequest);
1820     Z_SortRequest *req = apdu->u.sortRequest;
1821     Z_SortKeySpecList *sksl = (Z_SortKeySpecList *)
1822         odr_malloc (out, sizeof(*sksl));
1823     char setstring[32];
1824     char sort_string[32], sort_flags[32];
1825     int off;
1826     int oid[OID_SIZE];
1827     oident bib1;
1828
1829     if (setnumber >= 0)
1830         sprintf (setstring, "%d", setnumber);
1831     else
1832         sprintf (setstring, "default");
1833
1834     req->referenceId = set_refid (out);
1835
1836 #ifdef ASN_COMPILED
1837     req->num_inputResultSetNames = 1;
1838     req->inputResultSetNames = (Z_InternationalString **)
1839         odr_malloc (out, sizeof(*req->inputResultSetNames));
1840     req->inputResultSetNames[0] = odr_strdup (out, setstring);
1841 #else
1842     req->inputResultSetNames =
1843         (Z_StringList *)odr_malloc (out, sizeof(*req->inputResultSetNames));
1844     req->inputResultSetNames->num_strings = 1;
1845     req->inputResultSetNames->strings =
1846         (char **)odr_malloc (out, sizeof(*req->inputResultSetNames->strings));
1847     req->inputResultSetNames->strings[0] =
1848         odr_strdup (out, setstring);
1849 #endif
1850
1851     if (newset && setnumber >= 0)
1852         sprintf (setstring, "%d", ++setnumber);
1853
1854     req->sortedResultSetName = odr_strdup (out, setstring);
1855
1856     req->sortSequence = sksl;
1857     sksl->num_specs = 0;
1858     sksl->specs = (Z_SortKeySpec **)odr_malloc (out, sizeof(sksl->specs) * 20);
1859     
1860     bib1.proto = protocol;
1861     bib1.oclass = CLASS_ATTSET;
1862     bib1.value = VAL_BIB1;
1863     while ((sscanf (arg, "%31s %31s%n", sort_string, sort_flags, &off)) == 2 
1864            && off > 1)
1865     {
1866         int i;
1867         char *sort_string_sep;
1868         Z_SortKeySpec *sks = (Z_SortKeySpec *)odr_malloc (out, sizeof(*sks));
1869         Z_SortKey *sk = (Z_SortKey *)odr_malloc (out, sizeof(*sk));
1870
1871         arg += off;
1872         sksl->specs[sksl->num_specs++] = sks;
1873         sks->sortElement = (Z_SortElement *)odr_malloc (out, sizeof(*sks->sortElement));
1874         sks->sortElement->which = Z_SortElement_generic;
1875         sks->sortElement->u.generic = sk;
1876         
1877         if ((sort_string_sep = strchr (sort_string, '=')))
1878         {
1879             Z_AttributeElement *el = (Z_AttributeElement *)odr_malloc (out, sizeof(*el));
1880             sk->which = Z_SortKey_sortAttributes;
1881             sk->u.sortAttributes =
1882                 (Z_SortAttributes *)odr_malloc (out, sizeof(*sk->u.sortAttributes));
1883             sk->u.sortAttributes->id = oid_ent_to_oid(&bib1, oid);
1884             sk->u.sortAttributes->list =
1885                 (Z_AttributeList *)odr_malloc (out, sizeof(*sk->u.sortAttributes->list));
1886             sk->u.sortAttributes->list->num_attributes = 1;
1887             sk->u.sortAttributes->list->attributes =
1888                 (Z_AttributeElement **)odr_malloc (out,
1889                             sizeof(*sk->u.sortAttributes->list->attributes));
1890             sk->u.sortAttributes->list->attributes[0] = el;
1891             el->attributeSet = 0;
1892             el->attributeType = (int *)odr_malloc (out, sizeof(*el->attributeType));
1893             *el->attributeType = atoi (sort_string);
1894             el->which = Z_AttributeValue_numeric;
1895             el->value.numeric = (int *)odr_malloc (out, sizeof(*el->value.numeric));
1896             *el->value.numeric = atoi (sort_string_sep + 1);
1897         }
1898         else
1899         {
1900             sk->which = Z_SortKey_sortField;
1901             sk->u.sortField = odr_strdup (out, sort_string);
1902         }
1903         sks->sortRelation = (int *)odr_malloc (out, sizeof(*sks->sortRelation));
1904         *sks->sortRelation = Z_SortRelation_ascending;
1905         sks->caseSensitivity = (int *)odr_malloc (out, sizeof(*sks->caseSensitivity));
1906         *sks->caseSensitivity = Z_SortCase_caseSensitive;
1907
1908 #ifdef ASN_COMPILED
1909         sks->which = Z_SortKeySpec_null;
1910         sks->u.null = odr_nullval ();
1911 #else
1912         sks->missingValueAction = NULL;
1913 #endif
1914
1915         for (i = 0; sort_flags[i]; i++)
1916         {
1917             switch (sort_flags[i])
1918             {
1919             case 'a':
1920             case 'A':
1921             case '>':
1922                 *sks->sortRelation = Z_SortRelation_descending;
1923                 break;
1924             case 'd':
1925             case 'D':
1926             case '<':
1927                 *sks->sortRelation = Z_SortRelation_ascending;
1928                 break;
1929             case 'i':
1930             case 'I':
1931                 *sks->caseSensitivity = Z_SortCase_caseInsensitive;
1932                 break;
1933             case 'S':
1934             case 's':
1935                 *sks->caseSensitivity = Z_SortCase_caseSensitive;
1936                 break;
1937             }
1938         }
1939     }
1940     if (!sksl->num_specs)
1941     {
1942         printf ("Missing sort specifications\n");
1943         return -1;
1944     }
1945     send_apdu(apdu);
1946     return 2;
1947 }
1948
1949 void display_term(Z_TermInfo *t)
1950 {
1951     if (t->term->which == Z_Term_general)
1952     {
1953         printf("%.*s (%d)\n", t->term->u.general->len, t->term->u.general->buf,
1954             t->globalOccurrences ? *t->globalOccurrences : -1);
1955         sprintf(last_scan, "%.*s", t->term->u.general->len,
1956             t->term->u.general->buf);
1957     }
1958     else
1959         printf("Term type not general.\n");
1960 }
1961
1962 void process_scanResponse(Z_ScanResponse *res)
1963 {
1964     int i;
1965     Z_Entry **entries = NULL;
1966     int num_entries = 0;
1967    
1968     printf("Received ScanResponse\n"); 
1969     print_refid (res->referenceId);
1970     printf("%d entries", *res->numberOfEntriesReturned);
1971     if (res->positionOfTerm)
1972         printf (", position=%d", *res->positionOfTerm); 
1973     printf ("\n");
1974     if (*res->scanStatus != Z_Scan_success)
1975         printf("Scan returned code %d\n", *res->scanStatus);
1976     if (!res->entries)
1977         return;
1978     if ((entries = res->entries->entries))
1979         num_entries = res->entries->num_entries;
1980     for (i = 0; i < num_entries; i++)
1981     {
1982         int pos_term = res->positionOfTerm ? *res->positionOfTerm : -1;
1983         if (entries[i]->which == Z_Entry_termInfo)
1984         {
1985             printf("%c ", i + 1 == pos_term ? '*' : ' ');
1986             display_term(entries[i]->u.termInfo);
1987         }
1988         else
1989             display_diagrecs(&entries[i]->u.surrogateDiagnostic, 1);
1990     }
1991     if (res->entries->nonsurrogateDiagnostics)
1992         display_diagrecs (res->entries->nonsurrogateDiagnostics,
1993                           res->entries->num_nonsurrogateDiagnostics);
1994 }
1995
1996 void process_sortResponse(Z_SortResponse *res)
1997 {
1998     printf("Received SortResponse: status=");
1999     switch (*res->sortStatus)
2000     {
2001     case Z_SortStatus_success:
2002         printf ("success"); break;
2003     case Z_SortStatus_partial_1:
2004         printf ("partial"); break;
2005     case Z_SortStatus_failure:
2006         printf ("failure"); break;
2007     default:
2008         printf ("unknown (%d)", *res->sortStatus);
2009     }
2010     printf ("\n");
2011     print_refid (res->referenceId);
2012 #ifdef ASN_COMPILED
2013     if (res->diagnostics)
2014         display_diagrecs(res->diagnostics,
2015                          res->num_diagnostics);
2016 #else
2017     if (res->diagnostics)
2018         display_diagrecs(res->diagnostics->diagRecs,
2019                          res->diagnostics->num_diagRecs);
2020 #endif
2021 }
2022
2023 void process_deleteResultSetResponse (Z_DeleteResultSetResponse *res)
2024 {
2025     printf("Got deleteResultSetResponse status=%d\n",
2026            *res->deleteOperationStatus);
2027     if (res->deleteListStatuses)
2028     {
2029         int i;
2030         for (i = 0; i < res->deleteListStatuses->num; i++)
2031         {
2032             printf ("%s status=%d\n", res->deleteListStatuses->elements[i]->id,
2033                     *res->deleteListStatuses->elements[i]->status);
2034         }
2035     }
2036 }
2037
2038 int cmd_sort_generic(char *arg, int newset)
2039 {
2040     if (!conn)
2041     {
2042         printf("Session not initialized yet\n");
2043         return 0;
2044     }
2045     if (!ODR_MASK_GET(session->options, Z_Options_sort))
2046     {
2047         printf("Target doesn't support sort\n");
2048         return 0;
2049     }
2050     if (*arg)
2051     {
2052         if (send_sortrequest(arg, newset) < 0)
2053             return 0;
2054         return 2;
2055     }
2056     return 0;
2057 }
2058
2059 int cmd_sort(char *arg)
2060 {
2061     return cmd_sort_generic (arg, 0);
2062 }
2063
2064 int cmd_sort_newset (char *arg)
2065 {
2066     return cmd_sort_generic (arg, 1);
2067 }
2068
2069 int cmd_scan(char *arg)
2070 {
2071     if (!conn)
2072     {
2073         printf("Session not initialized yet\n");
2074         return 0;
2075     }
2076     if (!ODR_MASK_GET(session->options, Z_Options_scan))
2077     {
2078         printf("Target doesn't support scan\n");
2079         return 0;
2080     }
2081     if (*arg)
2082     {
2083         if (send_scanrequest(arg, 1, 20) < 0)
2084             return 0;
2085     }
2086     else
2087         if (send_scanrequest(last_scan, 1, 20) < 0)
2088             return 0;
2089     return 2;
2090 }
2091
2092 int cmd_schema(char *arg)
2093 {
2094     if (!arg || !*arg)
2095     {
2096         schema = VAL_NONE;
2097         return 1;
2098     }
2099     schema = oid_getvalbyname (arg);
2100     if (schema == VAL_NONE)
2101     {
2102         printf ("unknown schema\n");
2103         return 0;
2104     }
2105     return 1;
2106 }
2107
2108 int cmd_format(char *arg)
2109 {
2110     if (!arg || !*arg)
2111     {
2112         printf("Usage: format <recordsyntax>\n");
2113         return 0;
2114     }
2115     recordsyntax = oid_getvalbyname (arg);
2116     if (recordsyntax == VAL_NONE)
2117     {
2118         printf ("unknown record syntax\n");
2119         return 0;
2120     }
2121     return 1;
2122 }
2123
2124 int cmd_elements(char *arg)
2125 {
2126     static Z_ElementSetNames esn;
2127     static char what[100];
2128
2129     if (!arg || !*arg)
2130     {
2131         elementSetNames = 0;
2132         return 1;
2133     }
2134     strcpy(what, arg);
2135     esn.which = Z_ElementSetNames_generic;
2136     esn.u.generic = what;
2137     elementSetNames = &esn;
2138     return 1;
2139 }
2140
2141 int cmd_attributeset(char *arg)
2142 {
2143     char what[100];
2144
2145     if (!arg || !*arg)
2146     {
2147         printf("Usage: attributeset <setname>\n");
2148         return 0;
2149     }
2150     sscanf(arg, "%s", what);
2151     if (p_query_attset (what))
2152     {
2153         printf("Unknown attribute set name\n");
2154         return 0;
2155     }
2156     return 1;
2157 }
2158
2159 int cmd_querytype (char *arg)
2160 {
2161     if (!strcmp (arg, "ccl"))
2162         queryType = QueryType_CCL;
2163     else if (!strcmp (arg, "prefix") || !strcmp(arg, "rpn"))
2164         queryType = QueryType_Prefix;
2165 #if CCL2RPN
2166     else if (!strcmp (arg, "ccl2rpn") || !strcmp (arg, "cclrpn"))
2167         queryType = QueryType_CCL2RPN;
2168 #endif
2169     else
2170     {
2171         printf ("Querytype must be one of:\n");
2172         printf (" prefix         - Prefix query\n");
2173         printf (" ccl            - CCL query\n");
2174 #if CCL2RPN
2175         printf (" ccl2rpn        - CCL query converted to RPN\n");
2176 #endif
2177         return 0;
2178     }
2179     return 1;
2180 }
2181
2182 int cmd_refid (char *arg)
2183 {
2184     xfree (refid);
2185     refid = NULL;
2186     if (*arg)
2187     {
2188         refid = (char *) xmalloc (strlen(arg)+1);
2189         strcpy (refid, arg);
2190     }
2191     return 1;
2192 }
2193
2194 int cmd_close(char *arg)
2195 {
2196     Z_APDU *apdu;
2197     Z_Close *req;
2198     if (!conn)
2199         return 0;
2200
2201     apdu = zget_APDU(out, Z_APDU_close);
2202     req = apdu->u.close;
2203     *req->closeReason = Z_Close_finished;
2204     send_apdu(apdu);
2205     printf("Sent close request.\n");
2206     sent_close = 1;
2207     return 2;
2208 }
2209
2210 static void initialize(void)
2211 {
2212 #if CCL2RPN
2213     FILE *inf;
2214 #endif
2215     nmem_init();
2216     if (!(out = odr_createmem(ODR_ENCODE)) ||
2217         !(in = odr_createmem(ODR_DECODE)) ||
2218         !(print = odr_createmem(ODR_PRINT)))
2219     {
2220         fprintf(stderr, "failed to allocate ODR streams\n");
2221         exit(1);
2222     }
2223     setvbuf(stdout, 0, _IONBF, 0);
2224     if (apdu_file)
2225         odr_setprint(print, apdu_file);
2226
2227 #if CCL2RPN
2228     bibset = ccl_qual_mk (); 
2229     inf = fopen ("default.bib", "r");
2230     if (inf)
2231     {
2232         ccl_qual_file (bibset, inf);
2233         fclose (inf);
2234     }
2235 #endif
2236     cmd_base("Default");
2237 }
2238
2239 static int client(int wait)
2240 {
2241     static struct {
2242         char *cmd;
2243         int (*fun)(char *arg);
2244         char *ad;
2245     } cmd[] = {
2246         {"open", cmd_open, "('tcp'|'osi')':'[<tsel>'/']<host>[':'<port>]"},
2247         {"quit", cmd_quit, ""},
2248         {"find", cmd_find, "<query>"},
2249         {"delete", cmd_delete, "<setname>"},
2250         {"base", cmd_base, "<base-name>"},
2251         {"show", cmd_show, "<rec#>['+'<#recs>['+'<setname>]]"},
2252         {"scan", cmd_scan, "<term>"},
2253         {"sort", cmd_sort, "<sortkey> <flag> <sortkey> <flag> ..."},
2254         {"sort+", cmd_sort_newset, "<sortkey> <flag> <sortkey> <flag> ..."},
2255         {"authentication", cmd_authentication, "<acctstring>"},
2256         {"lslb", cmd_lslb, "<largeSetLowerBound>"},
2257         {"ssub", cmd_ssub, "<smallSetUpperBound>"},
2258         {"mspn", cmd_mspn, "<mediumSetPresentNumber>"},
2259         {"status", cmd_status, ""},
2260         {"setnames", cmd_setnames, ""},
2261         {"cancel", cmd_cancel, ""},
2262         {"format", cmd_format, "<recordsyntax>"},
2263         {"schema", cmd_schema, "<schema>"},
2264         {"elements", cmd_elements, "<elementSetName>"},
2265         {"close", cmd_close, ""},
2266         {"attributeset", cmd_attributeset, "<attrset>"},
2267         {"querytype", cmd_querytype, "<type>"},
2268         {"refid", cmd_refid, "<id>"},
2269         {"itemorder", cmd_itemorder, "1|2 <item>"},
2270         {"update", cmd_update, "<item>"},
2271         /* Server Admin Functions */
2272         {"adm-reindex", cmd_adm_reindex, "<database-name>"},
2273         {"adm-truncate", cmd_adm_truncate, "('database'|'index')<object-name>"},
2274         {"adm-create", cmd_adm_create, "<database-name>"},
2275         {"adm-delete", cmd_adm_delete, "('database'|'index')<object-name>"},
2276         {"adm-import", cmd_adm_import, "<database-name> <record-type> <source-file-name>"},
2277         {"adm-refresh", cmd_adm_refresh, "<database-name>"},
2278         {"adm-commit", cmd_adm_commit, ""},
2279         {"adm-shutdown", cmd_adm_shutdown, ""},
2280         {"adm-startup", cmd_adm_startup, ""},
2281         {0,0}
2282     };
2283     char *netbuffer= 0;
2284     int netbufferlen = 0;
2285     int i;
2286     Z_APDU *apdu;
2287
2288     while (1)
2289     {
2290         int res;
2291 #ifdef USE_SELECT
2292         fd_set input;
2293 #endif
2294         char line[1024], word[1024], arg[1024];
2295         
2296 #ifdef USE_SELECT
2297         FD_ZERO(&input);
2298         FD_SET(0, &input);
2299         if (conn)
2300             FD_SET(cs_fileno(conn), &input);
2301         if ((res = select(20, &input, 0, 0, 0)) < 0)
2302         {
2303             perror("select");
2304             exit(1);
2305         }
2306         if (!res)
2307             continue;
2308         if (!wait && FD_ISSET(0, &input))
2309 #else
2310         if (!wait)
2311 #endif
2312         {
2313 #if HAVE_READLINE_READLINE_H
2314             char* line_in;
2315             line_in=readline(C_PROMPT);
2316             if (!line_in)
2317                 break;
2318 #if HAVE_READLINE_HISTORY_H
2319             if (*line_in)
2320                 add_history(line_in);
2321 #endif
2322             strcpy(line,line_in);
2323             free (line_in);
2324 #else    
2325             char *end_p;
2326             printf (C_PROMPT);
2327             fflush(stdout);
2328             if (!fgets(line, 1023, stdin))
2329                 break;
2330             if ((end_p = strchr (line, '\n')))
2331                 *end_p = '\0';
2332 #endif 
2333             if ((res = sscanf(line, "%s %[^;]", word, arg)) <= 0)
2334             {
2335                 strcpy(word, last_cmd);
2336                 *arg = '\0';
2337             }
2338             else if (res == 1)
2339                 *arg = 0;
2340             strcpy(last_cmd, word);
2341             for (i = 0; cmd[i].cmd; i++)
2342                 if (!strncmp(cmd[i].cmd, word, strlen(word)))
2343                 {
2344                     res = (*cmd[i].fun)(arg);
2345                     break;
2346                 }
2347             if (!cmd[i].cmd) /* dump our help-screen */
2348             {
2349                 printf("Unknown command: %s.\n", word);
2350                 printf("Currently recognized commands:\n");
2351                 for (i = 0; cmd[i].cmd; i++)
2352                     printf("   %s %s\n", cmd[i].cmd, cmd[i].ad);
2353                 res = 1;
2354             }
2355             if (res < 2)
2356             {
2357                 continue;
2358             }
2359         }
2360         wait = 0;
2361 #ifdef USE_SELECT
2362         if (conn && FD_ISSET(cs_fileno(conn), &input))
2363 #endif
2364         {
2365             do
2366             {
2367                 if ((res = cs_get(conn, &netbuffer, &netbufferlen)) < 0)
2368                 {
2369                     perror("cs_get");
2370                     exit(1);
2371                 }
2372                 if (!res)
2373                 {
2374                     printf("Target closed connection.\n");
2375                     exit(1);
2376                 }
2377                 odr_reset(in); /* release APDU from last round */
2378                 record_last = 0;
2379                 odr_setbuf(in, netbuffer, res, 0);
2380                 if (!z_APDU(in, &apdu, 0, 0))
2381                 {
2382                     odr_perror(in, "Decoding incoming APDU");
2383                     fprintf(stderr, "[Near %d]\n", odr_offset(in));
2384                     fprintf(stderr, "Packet dump:\n---------\n");
2385                     odr_dumpBER(stderr, netbuffer, res);
2386                     fprintf(stderr, "---------\n");
2387                     if (apdu_file)
2388                         z_APDU(print, &apdu, 0, 0);
2389                     exit(1);
2390                 }
2391                 if (apdu_file && !z_APDU(print, &apdu, 0, 0))
2392                 {
2393                     odr_perror(print, "Failed to print incoming APDU");
2394                     odr_reset(print);
2395                     continue;
2396                 }
2397                 switch(apdu->which)
2398                 {
2399                 case Z_APDU_initResponse:
2400                     process_initResponse(apdu->u.initResponse);
2401                     break;
2402                 case Z_APDU_searchResponse:
2403                     process_searchResponse(apdu->u.searchResponse);
2404                     break;
2405                 case Z_APDU_scanResponse:
2406                     process_scanResponse(apdu->u.scanResponse);
2407                     break;
2408                 case Z_APDU_presentResponse:
2409                     print_refid (apdu->u.presentResponse->referenceId);
2410                     setno +=
2411                         *apdu->u.presentResponse->numberOfRecordsReturned;
2412                     if (apdu->u.presentResponse->records)
2413                         display_records(apdu->u.presentResponse->records);
2414                     else
2415                         printf("No records.\n");
2416                     printf ("nextResultSetPosition = %d\n",
2417                         *apdu->u.presentResponse->nextResultSetPosition);
2418                     break;
2419                 case Z_APDU_sortResponse:
2420                     process_sortResponse(apdu->u.sortResponse);
2421                     break;
2422                 case Z_APDU_extendedServicesResponse:
2423                     printf("Got extended services response\n");
2424                     process_ESResponse(apdu->u.extendedServicesResponse);
2425                     break;
2426                 case Z_APDU_close:
2427                     printf("Target has closed the association.\n");
2428                     process_close(apdu->u.close);
2429                     break;
2430                 case Z_APDU_resourceControlRequest:
2431                     process_resourceControlRequest
2432                         (apdu->u.resourceControlRequest);
2433                     break;
2434                 case Z_APDU_deleteResultSetResponse:
2435                     process_deleteResultSetResponse(apdu->u.
2436                                                     deleteResultSetResponse);
2437                     break;
2438                 default:
2439                     printf("Received unknown APDU type (%d).\n", 
2440                            apdu->which);
2441                     exit(1);
2442                 }
2443             }
2444             while (conn && cs_more(conn));
2445         }
2446     }
2447     return 0;
2448 }
2449
2450 int main(int argc, char **argv)
2451 {
2452     char *prog = *argv;
2453     char *arg;
2454     int ret;
2455     int opened = 0;
2456
2457     while ((ret = options("a:m:v:", argv, argc, &arg)) != -2)
2458     {
2459         switch (ret)
2460         {
2461         case 0:
2462             if (!opened)
2463             {
2464                 initialize ();
2465                 if (cmd_open (arg) == 2)
2466                     opened = 1;
2467             }
2468             break;
2469         case 'm':
2470             if (!(marcdump = fopen (arg, "a")))
2471             {
2472                 perror (arg);
2473                 exit (1);
2474             }
2475             break;
2476         case 'a':
2477             if (!strcmp(arg, "-"))
2478                 apdu_file=stderr;
2479             else
2480                 apdu_file=fopen(arg, "a");
2481             break;
2482         case 'v':
2483             log_init (log_mask_str(arg), "", NULL);
2484             break;
2485         default:
2486             fprintf (stderr, "Usage: %s [-m <marclog>] [ -m <apdulog>] "
2487                              "[<server-addr>]\n",
2488                      prog);
2489             exit (1);
2490         }
2491     }
2492     if (!opened)
2493         initialize ();
2494     return client (opened);
2495 }
2496
2497