ztest/ztest.c

   1 /*
   2  * Copyright (c) 1995-2000, Index Data.
   3  * See the file LICENSE for details.
   4  *
   5  * NT Service interface by
   6  *    Chas Woodfield, Fretwell Downing Datasystems.
   7  *
   8  * $Log: ztest.c,v $
   9  * Revision 1.30  2000-01-13 23:05:50  adam
  10  * Fixed tagging for member requester-CHECKED-IN in ILL ASN.1 spec.
  11  *
  12  * Revision 1.29  2000/01/12 14:36:07  adam
  13  * Added printing stream (ODR) for backend functions.
  14  *
  15  * Revision 1.28  1999/12/16 23:36:19  adam
  16  * Implemented ILL protocol. Minor updates ASN.1 compiler.
  17  *
  18  * Revision 1.27  1999/11/30 13:47:12  adam
  19  * Improved installation. Moved header files to include/yaz.
  20  *
  21  * Revision 1.26  1999/08/27 09:40:32  adam
  22  * Renamed logf function to yaz_log. Removed VC++ project files.
  23  *
  24  * Revision 1.25  1999/06/01 14:29:12  adam
  25  * Work on Extended Services.
  26  *
  27  * Revision 1.24  1999/05/27 13:07:54  adam
  28  * Fix.
  29  *
  30  * Revision 1.23  1999/05/27 13:02:20  adam
  31  * Assigned OID for old DB Update (VAL_DBUPDATE0).
  32  *
  33  * Revision 1.22  1999/05/26 13:49:12  adam
  34  * DB Update implemented in client (very basic).
  35  *
  36  * Revision 1.21  1998/12/15 12:45:42  adam
  37  * Minor change.
  38  *
  39  * Revision 1.20  1998/12/14 14:48:05  adam
  40  * Fixed memory leak - happened when fetching MARC records.
  41  *
  42  * Revision 1.19  1998/10/20 15:16:22  adam
  43  * Minor change to prevent warning.
  44  *
  45  * Revision 1.18  1998/10/20 15:13:45  adam
  46  * Minor fix regarding output for Item Order.
  47  *
  48  * Revision 1.17  1998/10/18 22:33:35  quinn
  49  * Added diagnostic dump of Item Order Eservice.
  50  *
  51  * Revision 1.16  1998/10/15 08:26:23  adam
  52  * Added type cast to make C++ happy.
  53  *
  54  * Revision 1.15  1998/10/13 20:05:57  adam
  55  * Minor change.
  56  *
  57  * Revision 1.14  1998/10/13 16:12:25  adam
  58  * Added support for Surrogate Diagnostics for Scan Term entries.
  59  *
  60  * Revision 1.13  1998/08/19 16:10:09  adam
  61  * Changed som member names of DeleteResultSetRequest/Response.
  62  *
  63  * Revision 1.12  1998/07/20 12:38:44  adam
  64  * Implemented delete result set service to server API.
  65  *
  66  * Revision 1.11  1998/06/09 13:55:08  adam
  67  * Minor changes.
  68  *
  69  * Revision 1.10  1998/05/27 16:55:54  adam
  70  * Minor changes.
  71  *
  72  * Revision 1.9  1998/03/31 11:07:45  adam
  73  * Furhter work on UNIverse resource report.
  74  * Added Extended Services handling in frontend server.
  75  *
  76  * Revision 1.8  1998/02/11 11:53:36  adam
  77  * Changed code so that it compiles as C++.
  78  *
  79  * Revision 1.7  1998/02/10 11:03:57  adam
  80  * Added support for extended handlers in backend server interface.
  81  *
  82  * Revision 1.6  1998/01/29 13:16:02  adam
  83  * Added dummy sort in test server.
  84  *
  85  * Revision 1.5  1997/11/07 13:31:58  adam
  86  * Added NT Service name part of statserv_options_block. Moved NT
  87  * service utility to server library.
  88  *
  89  * Revision 1.4  1997/09/17 12:10:43  adam
  90  * YAZ version 1.4.
  91  *
  92  * Revision 1.3  1997/09/09 10:10:20  adam
  93  * Another MSV5.0 port. Changed projects to include proper
  94  * library/include paths.
  95  * Server starts server in test-mode when no options are given.
  96  *
  97  * Revision 1.2  1997/09/04 13:50:31  adam
  98  * Bug fix in ztest.
  99  *
 100  */
 101
 102 /*
 103  * Demonstration of simple server
 104  */
 105
 106 #include <stdio.h>
 107 #include <stdlib.h>
 108 #include <ctype.h>
 109 #include <assert.h>
 110
 111 #include <yaz/backend.h>
 112 #include <yaz/log.h>
 113
 114 #ifdef ASN_COMPILED
 115 #include <yaz/ill.h>
 116 #endif
 117
 118 Z_GenericRecord *read_grs1(FILE *f, ODR o);
 119
 120 int ztest_search (void *handle, bend_search_rr *rr);
 121 int ztest_sort (void *handle, bend_sort_rr *rr);
 122 int ztest_present (void *handle, bend_present_rr *rr);
 123 int ztest_esrequest (void *handle, bend_esrequest_rr *rr);
 124 int ztest_delete (void *handle, bend_delete_rr *rr);
 125
 126 bend_initresult *bend_init(bend_initrequest *q)
 127 {
 128     bend_initresult *r = (bend_initresult *) odr_malloc (q->stream, sizeof(*r));
 129     static char *dummy = "Hej fister";
 130
 131     r->errcode = 0;
 132     r->errstring = 0;
 133     r->handle = dummy;
 134     q->bend_sort = ztest_sort;       /* register sort handler */
 135     q->bend_search = ztest_search;   /* register search handler */
 136     q->bend_present = ztest_present; /* register present handle */
 137     q->bend_esrequest = ztest_esrequest;
 138     q->bend_delete = ztest_delete;
 139     return r;
 140 }
 141
 142 int ztest_search (void *handle, bend_search_rr *rr)
 143 {
 144     rr->hits = rand() % 22;
 145     return 0;
 146 }
 147
 148 int ztest_present (void *handle, bend_present_rr *rr)
 149 {
 150     return 0;
 151 }
 152
 153 int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
 154 {
 155     yaz_log(LOG_LOG, "function: %d", *rr->esr->function);
 156     if (rr->esr->packageName)
 157         yaz_log(LOG_LOG, "packagename: %s", rr->esr->packageName);
 158     yaz_log(LOG_LOG, "Waitaction: %d", *rr->esr->waitAction);
 159
 160     if (!rr->esr->taskSpecificParameters)
 161     {
 162         yaz_log (LOG_WARN, "No task specific parameters");
 163     }
 164     else if (rr->esr->taskSpecificParameters->which == Z_External_itemOrder)
 165     {
 166         Z_ItemOrder *it = rr->esr->taskSpecificParameters->u.itemOrder;
 167         yaz_log (LOG_LOG, "Received ItemOrder");
 168         switch (it->which)
 169         {
 170 #ifdef ASN_COMPILED
 171         case Z_IOItemOrder_esRequest:
 172 #else
 173         case Z_ItemOrder_esRequest:
 174 #endif
 175         {
 176             Z_IORequest *ir = it->u.esRequest;
 177             Z_IOOriginPartToKeep *k = ir->toKeep;
 178             Z_IOOriginPartNotToKeep *n = ir->notToKeep;
 179
 180             if (k && k->contact)
 181             {
 182                 if (k->contact->name)
 183                     yaz_log(LOG_LOG, "contact name %s", k->contact->name);
 184                 if (k->contact->phone)
 185                     yaz_log(LOG_LOG, "contact phone %s", k->contact->phone);
 186                 if (k->contact->email)
 187                     yaz_log(LOG_LOG, "contact email %s", k->contact->email);
 188             }
 189             if (k->addlBilling)
 190             {
 191                 yaz_log(LOG_LOG, "Billing info (not shown)");
 192             }
 193
 194             if (n->resultSetItem)
 195             {
 196                 yaz_log(LOG_LOG, "resultsetItem");
 197                 yaz_log(LOG_LOG, "setId: %s", n->resultSetItem->resultSetId);
 198                 yaz_log(LOG_LOG, "item: %d", *n->resultSetItem->item);
 199             }
 200 #ifdef ASN_COMPILED
 201             if (n->itemRequest)
 202             {
 203                 Z_External *r = (Z_External*) n->itemRequest;
 204                 ILL_ItemRequest *item_req = 0;
 205                 ILL_Request *ill_req = 0;
 206                 if (r->direct_reference)
 207                 {
 208                     oident *ent = oid_getentbyoid(r->direct_reference);
 209                     if (ent)
 210                         yaz_log(LOG_LOG, "OID %s", ent->desc);
 211                     if (ent && ent->value == VAL_ISO_ILL_1)
 212                     {
 213                         yaz_log (LOG_LOG, "ItemRequest");
 214                         if (r->which == ODR_EXTERNAL_single)
 215                         {
 216                             odr_setbuf(rr->decode,
 217                                        r->u.single_ASN1_type->buf,
 218                                        r->u.single_ASN1_type->len, 0);
 219
 220                             if (!ill_ItemRequest (rr->decode, &item_req, 0, 0))
 221                             {
 222                                 yaz_log (LOG_LOG,
 223                                     "Couldn't decode ItemRequest %s near %d",
 224                                        odr_errmsg(odr_geterror(rr->decode)),
 225                                        odr_offset(rr->decode));
 226                                 yaz_log(LOG_LOG, "PDU dump:");
 227                                 odr_dumpBER(log_file(),
 228                                      r->u.single_ASN1_type->buf,
 229                                      r->u.single_ASN1_type->len);
 230                             }
 231                             if (rr->print)
 232                             {
 233                                 ill_ItemRequest (rr->print, &item_req, 0,
 234                                     "ItemRequest");
 235                                 odr_reset (rr->print);
 236                             }
 237                         }
 238                         if (!item_req && r->which == ODR_EXTERNAL_single)
 239                         {
 240                             yaz_log (LOG_LOG, "ILLRequest");
 241                             odr_setbuf(rr->decode,
 242                                        r->u.single_ASN1_type->buf,
 243                                        r->u.single_ASN1_type->len, 0);
 244
 245                             if (!ill_Request (rr->decode, &ill_req, 0, 0))
 246                             {
 247                                 yaz_log (LOG_LOG,
 248                                     "Couldn't decode ILLRequest %s near %d",
 249                                        odr_errmsg(odr_geterror(rr->decode)),
 250                                        odr_offset(rr->decode));
 251                                 yaz_log(LOG_LOG, "PDU dump:");
 252                                 odr_dumpBER(log_file(),
 253                                      r->u.single_ASN1_type->buf,
 254                                      r->u.single_ASN1_type->len);
 255                             }
 256                             if (rr->print)
 257                             {
 258                                 ill_Request (rr->print, &ill_req, 0,
 259                                     "ILLRequest");
 260                                 odr_reset (rr->print);
 261                             }
 262                         }
 263                     }
 264                 }
 265                 if (item_req)
 266                 {
 267                     yaz_log (LOG_LOG, "ILL protocol version = %d",
 268                              *item_req->protocol_version_num);
 269                 }
 270             }
 271 #endif
 272         }
 273         break;
 274         }
 275     }
 276     else if (rr->esr->taskSpecificParameters->which == Z_External_update)
 277     {
 278         Z_IUUpdate *up = rr->esr->taskSpecificParameters->u.update;
 279         yaz_log (LOG_LOG, "Received DB Update");
 280         if (up->which == Z_IUUpdate_esRequest)
 281         {
 282             Z_IUUpdateEsRequest *esRequest = up->u.esRequest;
 283             Z_IUOriginPartToKeep *toKeep = esRequest->toKeep;
 284             Z_IUSuppliedRecords *notToKeep = esRequest->notToKeep;
 285
 286             yaz_log (LOG_LOG, "action");
 287             if (toKeep->action)
 288             {
 289                 switch (*toKeep->action)
 290                 {
 291                 case Z_IUOriginPartToKeep_recordInsert:
 292                     yaz_log (LOG_LOG, " recordInsert");
 293                     break;
 294                 case Z_IUOriginPartToKeep_recordReplace:
 295                     yaz_log (LOG_LOG, " recordUpdate");
 296                     break;
 297                 case Z_IUOriginPartToKeep_recordDelete:
 298                     yaz_log (LOG_LOG, " recordDelete");
 299                     break;
 300                 case Z_IUOriginPartToKeep_elementUpdate:
 301                     yaz_log (LOG_LOG, " elementUpdate");
 302                     break;
 303                 case Z_IUOriginPartToKeep_specialUpdate:
 304                     yaz_log (LOG_LOG, " specialUpdate");
 305                     break;
 306                 default:
 307                     yaz_log (LOG_LOG, " unknown (%d)", *toKeep->action);
 308                 }
 309             }
 310             if (toKeep->databaseName)
 311             {
 312                 yaz_log (LOG_LOG, "database: %s", toKeep->databaseName);
 313                 if (!strcmp(toKeep->databaseName, "fault"))
 314                 {
 315                     rr->errcode = 109;
 316                     rr->errstring = toKeep->databaseName;
 317                 }
 318                 if (!strcmp(toKeep->databaseName, "accept"))
 319                     rr->errcode = -1;
 320             }
 321             if (notToKeep)
 322             {
 323                 int i;
 324                 for (i = 0; i < notToKeep->num; i++)
 325                 {
 326                     Z_External *rec = notToKeep->elements[i]->record;
 327
 328                     if (rec->direct_reference)
 329                     {
 330                         struct oident *oident;
 331                         oident = oid_getentbyoid(rec->direct_reference);
 332                         if (oident)
 333                             yaz_log (LOG_LOG, "record %d type %s", i,
 334                                      oident->desc);
 335                     }
 336                     switch (rec->which)
 337                     {
 338                     case Z_External_sutrs:
 339                         if (rec->u.octet_aligned->len > 170)
 340                             yaz_log (LOG_LOG, "%d bytes:\n%.168s ...",
 341                                      rec->u.sutrs->len,
 342                                      rec->u.sutrs->buf);
 343                         else
 344                             yaz_log (LOG_LOG, "%d bytes:\n%s",
 345                                      rec->u.sutrs->len,
 346                                      rec->u.sutrs->buf);
 347                         break;
 348                     case Z_External_octet        :
 349                         if (rec->u.octet_aligned->len > 170)
 350                             yaz_log (LOG_LOG, "%d bytes:\n%.168s ...",
 351                                      rec->u.octet_aligned->len,
 352                                      rec->u.octet_aligned->buf);
 353                         else
 354                             yaz_log (LOG_LOG, "%d bytes\n%s",
 355                                      rec->u.octet_aligned->len,
 356                                      rec->u.octet_aligned->buf);
 357                     }
 358                 }
 359             }
 360         }
 361     }
 362     else
 363     {
 364         yaz_log (LOG_WARN, "Unknown Extended Service(%d)",
 365                  rr->esr->taskSpecificParameters->which);
 366
 367     }
 368     return 0;
 369 }
 370
 371 int ztest_delete (void *handle, bend_delete_rr *rr)
 372 {
 373     if (rr->num_setnames == 1 && !strcmp (rr->setnames[0], "1"))
 374         rr->delete_status = Z_DeleteStatus_success;
 375     else
 376         rr->delete_status = Z_DeleteStatus_resultSetDidNotExist;
 377     return 0;
 378 }
 379
 380 /* Obsolete bend_search, never called because handler is registered */
 381 bend_searchresult *bend_search(void *handle, bend_searchrequest *q, int *fd)
 382 {
 383     return 0;
 384 }
 385
 386 /* Our sort handler really doesn't sort... */
 387 int ztest_sort (void *handle, bend_sort_rr *rr)
 388 {
 389     rr->errcode = 0;
 390     rr->sort_status = Z_SortStatus_success;
 391     return 0;
 392 }
 393
 394 static int atoin (const char *buf, int n)
 395 {
 396     int val = 0;
 397     while (--n >= 0)
 398     {
 399         if (isdigit(*buf))
 400             val = val*10 + (*buf - '0');
 401         buf++;
 402     }
 403     return val;
 404 }
 405
 406 char *marc_read(FILE *inf, ODR odr)
 407 {
 408     char length_str[5];
 409     size_t size;
 410     char *buf;
 411
 412     if (fread (length_str, 1, 5, inf) != 5)
 413         return NULL;
 414     size = atoin (length_str, 5);
 415     if (size <= 6)
 416         return NULL;
 417     if (!(buf = (char*) odr_malloc (odr, size+1)))
 418         return NULL;
 419     if (fread (buf+5, 1, size-5, inf) != (size-5))
 420     {
 421         xfree (buf);
 422         return NULL;
 423     }
 424     memcpy (buf, length_str, 5);
 425     buf[size] = '\0';
 426     return buf;
 427 }
 428
 429 static char *dummy_database_record (int num, ODR odr)
 430 {
 431     FILE *inf = fopen ("dummy-records", "r");
 432     char *buf = 0;
 433
 434     if (!inf)
 435         return NULL;
 436     while (--num >= 0)
 437     {
 438         if (num == 98)
 439         {
 440             buf = (char*) odr_malloc(odr, 2101);
 441             assert(buf);
 442             memset(buf, 'A', 2100);
 443             buf[2100] = '\0';
 444             break;
 445         }
 446         else
 447             buf = marc_read (inf, odr);
 448         if (!num || !buf)
 449             break;
 450     }
 451     fclose(inf);
 452     if (num < 0)
 453         return 0;
 454     return buf;
 455 }
 456
 457 static Z_GenericRecord *dummy_grs_record (int num, ODR o)
 458 {
 459     FILE *f = fopen("dummy-grs", "r");
 460     char line[512];
 461     Z_GenericRecord *r = 0;
 462     int n;
 463
 464     if (!f)
 465         return 0;
 466     while (fgets(line, 512, f))
 467         if (*line == '#' && sscanf(line, "#%d", &n) == 1 && n == num)
 468         {
 469             r = read_grs1(f, o);
 470             break;
 471         }
 472     fclose(f);
 473     return r;
 474 }
 475
 476 bend_fetchresult *bend_fetch(void *handle, bend_fetchrequest *q, int *fd)
 477 {
 478     bend_fetchresult *r = (bend_fetchresult *)
 479                         odr_malloc (q->stream, sizeof(*r));
 480     char *cp;
 481     r->errstring = 0;
 482     r->last_in_set = 0;
 483     r->basename = "DUMMY";
 484     r->format = q->format;
 485     if (q->format == VAL_SUTRS)
 486     {
 487         char buf[100];
 488
 489         sprintf(buf, "This is dummy SUTRS record number %d\n", q->number);
 490         r->len = strlen(buf);
 491         r->record = (char *) odr_malloc (q->stream, r->len+1);
 492         strcpy(r->record, buf);
 493     }
 494     else if (q->format == VAL_GRS1)
 495     {
 496         r->len = -1;
 497         r->record = (char*) dummy_grs_record(q->number, q->stream);
 498         if (!r->record)
 499         {
 500             r->errcode = 13;
 501             return r;
 502         }
 503     }
 504     else if ((cp = dummy_database_record(q->number, q->stream)))
 505     {
 506         r->len = strlen(cp);
 507         r->record = cp;
 508         r->format = VAL_USMARC;
 509     }
 510     else
 511     {
 512         r->errcode = 13;
 513         return r;
 514     }
 515     r->errcode = 0;
 516     return r;
 517 }
 518
 519 /*
 520  * silly dummy-scan what reads words from a file.
 521  */
 522 bend_scanresult *bend_scan(void *handle, bend_scanrequest *q, int *fd)
 523 {
 524     bend_scanresult *r = (bend_scanresult *)
 525         odr_malloc (q->stream, sizeof(*r));
 526     static FILE *f = 0;
 527     static struct scan_entry list[200];
 528     static char entries[200][80];
 529     int hits[200];
 530     char term[80], *p;
 531     int i, pos;
 532
 533     r->errcode = 0;
 534     r->errstring = 0;
 535     r->entries = list;
 536     r->status = BEND_SCAN_SUCCESS;
 537     if (!f && !(f = fopen("dummy-words", "r")))
 538     {
 539         perror("dummy-words");
 540         exit(1);
 541     }
 542     if (q->term->term->which != Z_Term_general)
 543     {
 544         r->errcode = 229; /* unsupported term type */
 545         return r;
 546     }
 547     if (q->term->term->u.general->len >= 80)
 548     {
 549         r->errcode = 11; /* term too long */
 550         return r;
 551     }
 552     if (q->num_entries > 200)
 553     {
 554         r->errcode = 31;
 555         return r;
 556     }
 557     memcpy(term, q->term->term->u.general->buf, q->term->term->u.general->len);
 558     term[q->term->term->u.general->len] = '\0';
 559     for (p = term; *p; p++)
 560         if (islower(*p))
 561             *p = toupper(*p);
 562
 563     fseek(f, 0, SEEK_SET);
 564     r->num_entries = 0;
 565     for (i = 0, pos = 0; fscanf(f, " %79[^:]:%d", entries[pos], &hits[pos]) == 2;
 566         i++, pos < 199 ? pos++ : (pos = 0))
 567     {
 568         if (!r->num_entries && strcmp(entries[pos], term) >= 0) /* s-point fnd */
 569         {
 570             if ((r->term_position = q->term_position) > i + 1)
 571             {
 572                 r->term_position = i + 1;
 573                 r->status = BEND_SCAN_PARTIAL;
 574             }
 575             for (; r->num_entries < r->term_position; r->num_entries++)
 576             {
 577                 int po;
 578
 579                 po = pos - r->term_position + r->num_entries + 1; /* find pos */
 580                 if (po < 0)
 581                     po += 200;
 582
 583                 if (!strcmp (term, "SD") && r->num_entries == 2)
 584                 {
 585                     list[r->num_entries].term = entries[pos];
 586                     list[r->num_entries].occurrences = -1;
 587                     list[r->num_entries].errcode = 233;
 588                     list[r->num_entries].errstring = "SD for Scan Term";
 589                 }
 590                 else
 591                 {
 592                     list[r->num_entries].term = entries[po];
 593                     list[r->num_entries].occurrences = hits[po];
 594                 }
 595             }
 596         }
 597         else if (r->num_entries)
 598         {
 599             list[r->num_entries].term = entries[pos];
 600             list[r->num_entries].occurrences = hits[pos];
 601             r->num_entries++;
 602         }
 603         if (r->num_entries >= q->num_entries)
 604             break;
 605     }
 606     if (feof(f))
 607         r->status = BEND_SCAN_PARTIAL;
 608     return r;
 609 }
 610
 611 void bend_close(void *handle)
 612 {
 613     return;
 614 }
 615
 616 int main(int argc, char **argv)
 617 {
 618     return statserv_main(argc, argv);
 619 }