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