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