client/client.c

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