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