ztest/ztest.c

   1 /* This file is part of the YAZ toolkit.
   2  * Copyright (C) 1995-2010 Index Data
   3  * See the file LICENSE for details.
   4  */
   5
   6 /** \file
   7  * \brief yaz-ztest Generic Frontend Server
   8  */
   9
  10 #include <stdio.h>
  11 #include <math.h>
  12 #include <stdlib.h>
  13 #include <ctype.h>
  14 #if HAVE_UNISTD_H
  15 #include <unistd.h>
  16 #endif
  17
  18 #include <yaz/log.h>
  19 #include <yaz/backend.h>
  20 #include <yaz/ill.h>
  21 #include <yaz/diagbib1.h>
  22
  23 #include "ztest.h"
  24
  25 static int log_level=0;
  26 static int log_level_set=0;
  27
  28 struct delay {
  29     double d1;
  30     double d2;
  31 };
  32
  33 struct result_set {
  34     char *name;
  35     char *db;
  36     Odr_int hits;
  37     struct delay search_delay;
  38     struct delay present_delay;
  39     struct delay fetch_delay;
  40     struct result_set *next;
  41 };
  42
  43 struct session_handle {
  44     struct result_set *result_sets;
  45 };
  46
  47 int ztest_search(void *handle, bend_search_rr *rr);
  48 int ztest_sort(void *handle, bend_sort_rr *rr);
  49 int ztest_present(void *handle, bend_present_rr *rr);
  50 int ztest_esrequest(void *handle, bend_esrequest_rr *rr);
  51 int ztest_delete(void *handle, bend_delete_rr *rr);
  52
  53 static struct result_set *get_set(struct session_handle *sh, const char *name)
  54 {
  55     struct result_set *set = sh->result_sets;
  56     for (; set; set = set->next)
  57         if (!strcmp(name, set->name))
  58             return set;
  59     return 0;
  60 }
  61
  62 static void remove_sets(struct session_handle *sh)
  63 {
  64     struct result_set *set = sh->result_sets;
  65     while (set)
  66     {
  67         struct result_set *set_next = set->next;
  68         xfree(set->name);
  69         xfree(set->db);
  70         xfree(set);
  71         set = set_next;
  72     }
  73     sh->result_sets = 0;
  74 }
  75
  76 /** \brief use term value as hit count
  77     \param s RPN structure
  78     \return >= 0: search term number or -1: not found
  79
  80     Traverse RPN tree 'in order' and use term value as hit count.
  81     Only terms  that looks a numeric is used.. Returns -1 if
  82     no sub tree has a hit count term
  83 */
  84 static Odr_int get_term_hit(Z_RPNStructure *s)
  85 {
  86     Odr_int h = -1;
  87     switch(s->which)
  88     {
  89     case Z_RPNStructure_simple:
  90         if (s->u.simple->which == Z_Operand_APT)
  91         {
  92             Z_AttributesPlusTerm *apt = s->u.simple->u.attributesPlusTerm;
  93             if (apt->term->which == Z_Term_general)
  94             {
  95                 Odr_oct *oct = apt->term->u.general;
  96                 if (oct->len > 0 && oct->buf[0] >= '0' && oct->buf[0] <= '9')
  97                 {
  98                     WRBUF hits_str = wrbuf_alloc();
  99                     wrbuf_write(hits_str, (const char *) oct->buf, oct->len);
 100                     h = odr_atoi(wrbuf_cstr(hits_str));
 101                     wrbuf_destroy(hits_str);
 102                 }
 103             }
 104         }
 105         break;
 106     case Z_RPNStructure_complex:
 107         h = get_term_hit(s->u.complex->s1);
 108         if (h == -1)
 109             h = get_term_hit(s->u.complex->s2);
 110         break;
 111     }
 112     return h;
 113 }
 114
 115 /** \brief gets hit count for numeric terms in RPN queries
 116     \param q RPN Query
 117     \return number of hits (random or number for term)
 118
 119     This is just for testing.. A real database of course uses
 120     the content of a database to establish a value.. In our case, we
 121     have a way to trigger a certain hit count. Good for testing of
 122     client applications etc
 123 */
 124 static Odr_int get_hit_count(Z_Query *q)
 125 {
 126     Odr_int h = -1;
 127     if (q->which == Z_Query_type_1 || q->which == Z_Query_type_101)
 128         h = get_term_hit(q->u.type_1->RPNStructure);
 129     if (h == -1)
 130         h = rand() % 24;
 131     return h;
 132 }
 133
 134 /** \brief checks if it's a dummy Slow database
 135     \param basename database name to check
 136     \param association backend association (or NULL if not available)
 137     \retval 1 is slow database
 138     \retval 0 is not a slow database
 139
 140     The Slow database is for testing.. It allows us to simulate
 141     a slow server...
 142 */
 143 static int check_slow(const char *basename, bend_association association)
 144 {
 145     if (strncmp(basename, "Slow", 4) == 0)
 146     {
 147 #if HAVE_UNISTD_H
 148         int i, w = 3;
 149         if (basename[4])
 150             sscanf(basename+4, "%d", &w);
 151         /* wait up to 3 seconds and check if connection is still alive */
 152         for (i = 0; i < w; i++)
 153         {
 154             if (association && !bend_assoc_is_alive(association))
 155             {
 156                 yaz_log(YLOG_LOG, "search aborted");
 157                 break;
 158             }
 159             sleep(1);
 160         }
 161 #endif
 162         return 1;
 163     }
 164     return 0;
 165 }
 166
 167 static int strcmp_prefix(const char *s, const char *p)
 168 {
 169     size_t l = strlen(p);
 170     if (strlen(s) >= l && !memcmp(s, p, l))
 171         return 1;
 172     return 0;
 173 }
 174
 175 static void init_delay(struct delay *delayp)
 176 {
 177     delayp->d1 = delayp->d2 = 0.0;
 178 }
 179
 180 static int parse_delay(struct delay *delayp, const char *value)
 181 {
 182     if (sscanf(value, "%lf:%lf", &delayp->d1, &delayp->d2) == 2)
 183         ;
 184     else if (sscanf(value, "%lf", &delayp->d1) == 1)
 185         delayp->d2 = 0.0;
 186     else
 187         return -1;
 188     return 0;
 189 }
 190
 191 static void do_delay(const struct delay *delayp)
 192 {
 193     double d = delayp->d1;
 194
 195     if (d > 0.0)
 196     {
 197         struct timeval tv;
 198         if (delayp->d2 > d)
 199             d += (rand()) * (delayp->d2 - d) / RAND_MAX;
 200         tv.tv_sec = floor(d);
 201         tv.tv_usec = (d - floor(d)) * 1000000;
 202         select(0, 0, 0, 0, &tv);
 203     }
 204 }
 205
 206 int ztest_search(void *handle, bend_search_rr *rr)
 207 {
 208     struct session_handle *sh = (struct session_handle*) handle;
 209     struct result_set *new_set;
 210     const char *db, *db_sep;
 211
 212     if (rr->num_bases != 1)
 213     {
 214         rr->errcode = YAZ_BIB1_COMBI_OF_SPECIFIED_DATABASES_UNSUPP;
 215         return 0;
 216     }
 217
 218     db = rr->basenames[0];
 219
 220     /* Allow Default, db.* and Slow */
 221     if (strcmp_prefix(db, "Default"))
 222         ;  /* Default is OK in our test */
 223     else if (strcmp_prefix(db, "db"))
 224         ;  /* db.* is OK in our test */
 225     else if (check_slow(rr->basenames[0], rr->association))
 226     {
 227         rr->estimated_hit_count = 1;
 228     }
 229     else
 230     {
 231         rr->errcode = YAZ_BIB1_DATABASE_UNAVAILABLE;
 232         rr->errstring = rr->basenames[0];
 233         return 0;
 234     }
 235
 236     new_set = get_set(sh, rr->setname);
 237     if (new_set)
 238     {
 239         if (!rr->replace_set)
 240         {
 241             rr->errcode = YAZ_BIB1_RESULT_SET_EXISTS_AND_REPLACE_INDICATOR_OFF;
 242             return 0;
 243         }
 244         xfree(new_set->db);
 245     }
 246     else
 247     {
 248         new_set = xmalloc(sizeof(*new_set));
 249         new_set->next = sh->result_sets;
 250         sh->result_sets = new_set;
 251         new_set->name = xstrdup(rr->setname);
 252     }
 253     new_set->hits = 0;
 254     new_set->db = xstrdup(db);
 255     init_delay(&new_set->search_delay);
 256     init_delay(&new_set->present_delay);
 257     init_delay(&new_set->fetch_delay);
 258
 259     db_sep = strchr(db, '?');
 260     if (db_sep)
 261     {
 262         char **names;
 263         char **values;
 264         int no_parms = yaz_uri_to_array(db_sep+1, rr->stream, &names, &values);
 265         int i;
 266         for (i = 0; i < no_parms; i++)
 267         {
 268             const char *name = names[i];
 269             const char *value = values[i];
 270             if (!strcmp(name, "seed"))
 271                 srandom(atoi(value));
 272             else if (!strcmp(name, "search-delay"))
 273                 parse_delay(&new_set->search_delay, value);
 274             else if (!strcmp(name, "present-delay"))
 275                 parse_delay(&new_set->present_delay, value);
 276             else if (!strcmp(name, "fetch-delay"))
 277                 parse_delay(&new_set->fetch_delay, value);
 278             else
 279             {
 280                 rr->errcode = YAZ_BIB1_SERVICE_UNSUPP_FOR_THIS_DATABASE;
 281                 rr->errstring = odr_strdup(rr->stream, name);
 282             }
 283         }
 284     }
 285
 286     if (rr->extra_args)
 287     {
 288         Z_SRW_extra_arg *a;
 289         WRBUF response_xml = wrbuf_alloc();
 290         wrbuf_puts(response_xml, "<extra>");
 291         for (a = rr->extra_args; a; a = a->next)
 292         {
 293             wrbuf_puts(response_xml, "<extra name=\"");
 294             wrbuf_xmlputs(response_xml, a->name);
 295             wrbuf_puts(response_xml, "\"");
 296             if (a->value)
 297             {
 298                 wrbuf_puts(response_xml, " value=\"");
 299                 wrbuf_xmlputs(response_xml, a->value);
 300                 wrbuf_puts(response_xml, "\"");
 301             }
 302             wrbuf_puts(response_xml, "/>");
 303         }
 304         wrbuf_puts(response_xml, "</extra>");
 305         rr->extra_response_data =
 306             odr_strdup(rr->stream, wrbuf_cstr(response_xml));
 307         wrbuf_destroy(response_xml);
 308     }
 309     rr->hits = get_hit_count(rr->query);
 310
 311     do_delay(&new_set->search_delay);
 312     new_set->hits = rr->hits;
 313
 314     return 0;
 315 }
 316
 317
 318 /* this huge function handles extended services */
 319 int ztest_esrequest(void *handle, bend_esrequest_rr *rr)
 320 {
 321     /* user-defined handle - created in bend_init */
 322     int *counter = (int*) handle;
 323
 324     yaz_log(log_level, "ESRequest no %d", *counter);
 325
 326     (*counter)++;
 327
 328     if (rr->esr->packageName)
 329         yaz_log(log_level, "packagename: %s", rr->esr->packageName);
 330     yaz_log(log_level, "Waitaction: " ODR_INT_PRINTF, *rr->esr->waitAction);
 331
 332
 333     yaz_log(log_level, "function: " ODR_INT_PRINTF, *rr->esr->function);
 334
 335     if (!rr->esr->taskSpecificParameters)
 336     {
 337         yaz_log(log_level, "No task specific parameters");
 338     }
 339     else if (rr->esr->taskSpecificParameters->which == Z_External_itemOrder)
 340     {
 341         Z_ItemOrder *it = rr->esr->taskSpecificParameters->u.itemOrder;
 342         yaz_log(log_level, "Received ItemOrder");
 343         if (it->which == Z_IOItemOrder_esRequest)
 344         {
 345             Z_IORequest *ir = it->u.esRequest;
 346             Z_IOOriginPartToKeep *k = ir->toKeep;
 347             Z_IOOriginPartNotToKeep *n = ir->notToKeep;
 348             const char *xml_in_response = 0;
 349
 350             if (k && k->contact)
 351             {
 352                 if (k->contact->name)
 353                     yaz_log(log_level, "contact name %s", k->contact->name);
 354                 if (k->contact->phone)
 355                     yaz_log(log_level, "contact phone %s", k->contact->phone);
 356                 if (k->contact->email)
 357                     yaz_log(log_level, "contact email %s", k->contact->email);
 358             }
 359             if (k->addlBilling)
 360             {
 361                 yaz_log(log_level, "Billing info (not shown)");
 362             }
 363
 364             if (n->resultSetItem)
 365             {
 366                 yaz_log(log_level, "resultsetItem");
 367                 yaz_log(log_level, "setId: %s", n->resultSetItem->resultSetId);
 368                 yaz_log(log_level, "item: " ODR_INT_PRINTF, *n->resultSetItem->item);
 369             }
 370             if (n->itemRequest)
 371             {
 372                 Z_External *r = (Z_External*) n->itemRequest;
 373                 ILL_ItemRequest *item_req = 0;
 374                 ILL_APDU *ill_apdu = 0;
 375                 if (r->direct_reference)
 376                 {
 377                     char oid_name_str[OID_STR_MAX];
 378                     oid_class oclass;
 379                     const char *oid_name =
 380                         yaz_oid_to_string_buf(r->direct_reference,
 381                                               &oclass, oid_name_str);
 382                     if (oid_name)
 383                         yaz_log(log_level, "OID %s", oid_name);
 384                     if (!oid_oidcmp(r->direct_reference, yaz_oid_recsyn_xml))
 385                     {
 386                         yaz_log(log_level, "ILL XML request");
 387                         if (r->which == Z_External_octet)
 388                             yaz_log(log_level, "%.*s",
 389                                     r->u.octet_aligned->len,
 390                                     r->u.octet_aligned->buf);
 391                         xml_in_response = "<dummy>x</dummy>";
 392                     }
 393                     if (!oid_oidcmp(r->direct_reference,
 394                                     yaz_oid_general_isoill_1))
 395                     {
 396                         yaz_log(log_level, "Decode ItemRequest begin");
 397                         if (r->which == ODR_EXTERNAL_single)
 398                         {
 399                             odr_setbuf(rr->decode,
 400                                        (char *) r->u.single_ASN1_type->buf,
 401                                        r->u.single_ASN1_type->len, 0);
 402
 403                             if (!ill_ItemRequest(rr->decode, &item_req, 0, 0))
 404                             {
 405                                 yaz_log(log_level,
 406                                         "Couldn't decode ItemRequest %s near %ld",
 407                                         odr_errmsg(odr_geterror(rr->decode)),
 408                                         (long) odr_offset(rr->decode));
 409                             }
 410                             else
 411                                 yaz_log(log_level, "Decode ItemRequest OK");
 412                             if (rr->print)
 413                             {
 414                                 ill_ItemRequest(rr->print, &item_req, 0,
 415                                                 "ItemRequest");
 416                                 odr_reset(rr->print);
 417                             }
 418                         }
 419                         if (!item_req && r->which == ODR_EXTERNAL_single)
 420                         {
 421                             yaz_log(log_level, "Decode ILL APDU begin");
 422                             odr_setbuf(rr->decode,
 423                                        (char*) r->u.single_ASN1_type->buf,
 424                                        r->u.single_ASN1_type->len, 0);
 425
 426                             if (!ill_APDU(rr->decode, &ill_apdu, 0, 0))
 427                             {
 428                                 yaz_log(log_level,
 429                                         "Couldn't decode ILL APDU %s near %ld",
 430                                         odr_errmsg(odr_geterror(rr->decode)),
 431                                         (long) odr_offset(rr->decode));
 432                                 yaz_log(log_level, "PDU dump:");
 433                                 odr_dumpBER(yaz_log_file(),
 434                                             (char *) r->u.single_ASN1_type->buf,
 435                                             r->u.single_ASN1_type->len);
 436                             }
 437                             else
 438                                 yaz_log(log_level, "Decode ILL APDU OK");
 439                             if (rr->print)
 440                             {
 441                                 ill_APDU(rr->print, &ill_apdu, 0,
 442                                          "ILL APDU");
 443                                 odr_reset(rr->print);
 444                             }
 445                         }
 446                     }
 447                 }
 448                 if (item_req)
 449                 {
 450                     yaz_log(log_level, "ILL protocol version = "
 451                             ODR_INT_PRINTF,
 452                             *item_req->protocol_version_num);
 453                 }
 454             }
 455             if (k)
 456             {
 457
 458                 Z_External *ext = (Z_External *)
 459                     odr_malloc(rr->stream, sizeof(*ext));
 460                 Z_IUOriginPartToKeep *keep = (Z_IUOriginPartToKeep *)
 461                     odr_malloc(rr->stream, sizeof(*keep));
 462                 Z_IOTargetPart *targetPart = (Z_IOTargetPart *)
 463                     odr_malloc(rr->stream, sizeof(*targetPart));
 464
 465                 rr->taskPackage = (Z_TaskPackage *)
 466                     odr_malloc(rr->stream, sizeof(*rr->taskPackage));
 467                 rr->taskPackage->packageType =
 468                     odr_oiddup(rr->stream, rr->esr->packageType);
 469                 rr->taskPackage->packageName = 0;
 470                 rr->taskPackage->userId = 0;
 471                 rr->taskPackage->retentionTime = 0;
 472                 rr->taskPackage->permissions = 0;
 473                 rr->taskPackage->description = 0;
 474                 rr->taskPackage->targetReference = (Odr_oct *)
 475                     odr_malloc(rr->stream, sizeof(Odr_oct));
 476                 rr->taskPackage->targetReference->buf =
 477                     (unsigned char *) odr_strdup(rr->stream, "911");
 478                 rr->taskPackage->targetReference->len =
 479                     rr->taskPackage->targetReference->size =
 480                     strlen((char *) (rr->taskPackage->targetReference->buf));
 481                 rr->taskPackage->creationDateTime = 0;
 482                 rr->taskPackage->taskStatus = odr_intdup(rr->stream, 0);
 483                 rr->taskPackage->packageDiagnostics = 0;
 484                 rr->taskPackage->taskSpecificParameters = ext;
 485
 486                 ext->direct_reference =
 487                     odr_oiddup(rr->stream, rr->esr->packageType);
 488                 ext->indirect_reference = 0;
 489                 ext->descriptor = 0;
 490                 ext->which = Z_External_itemOrder;
 491                 ext->u.itemOrder = (Z_ItemOrder *)
 492                     odr_malloc(rr->stream, sizeof(*ext->u.update));
 493                 ext->u.itemOrder->which = Z_IOItemOrder_taskPackage;
 494                 ext->u.itemOrder->u.taskPackage =  (Z_IOTaskPackage *)
 495                     odr_malloc(rr->stream, sizeof(Z_IOTaskPackage));
 496                 ext->u.itemOrder->u.taskPackage->originPart = k;
 497                 ext->u.itemOrder->u.taskPackage->targetPart = targetPart;
 498
 499                 if (xml_in_response)
 500                     targetPart->itemRequest =
 501                         z_ext_record_xml(rr->stream, xml_in_response,
 502                                          strlen(xml_in_response));
 503                 else
 504                     targetPart->itemRequest = 0;
 505
 506                 targetPart->statusOrErrorReport = 0;
 507                 targetPart->auxiliaryStatus = 0;
 508             }
 509         }
 510     }
 511     else if (rr->esr->taskSpecificParameters->which == Z_External_update)
 512     {
 513         Z_IUUpdate *up = rr->esr->taskSpecificParameters->u.update;
 514         yaz_log(log_level, "Received DB Update");
 515         if (up->which == Z_IUUpdate_esRequest)
 516         {
 517             Z_IUUpdateEsRequest *esRequest = up->u.esRequest;
 518             Z_IUOriginPartToKeep *toKeep = esRequest->toKeep;
 519             Z_IUSuppliedRecords *notToKeep = esRequest->notToKeep;
 520
 521             yaz_log(log_level, "action");
 522             if (toKeep->action)
 523             {
 524                 switch (*toKeep->action)
 525                 {
 526                 case Z_IUOriginPartToKeep_recordInsert:
 527                     yaz_log(log_level, " recordInsert");
 528                     break;
 529                 case Z_IUOriginPartToKeep_recordReplace:
 530                     yaz_log(log_level, " recordReplace");
 531                     break;
 532                 case Z_IUOriginPartToKeep_recordDelete:
 533                     yaz_log(log_level, " recordDelete");
 534                     break;
 535                 case Z_IUOriginPartToKeep_elementUpdate:
 536                     yaz_log(log_level, " elementUpdate");
 537                     break;
 538                 case Z_IUOriginPartToKeep_specialUpdate:
 539                     yaz_log(log_level, " specialUpdate");
 540                     break;
 541                 default:
 542                     yaz_log(log_level, " unknown (" ODR_INT_PRINTF ")",
 543                             *toKeep->action);
 544                 }
 545             }
 546             if (toKeep->databaseName)
 547             {
 548                 yaz_log(log_level, "database: %s", toKeep->databaseName);
 549                 if (!strcmp(toKeep->databaseName, "fault"))
 550                 {
 551                     rr->errcode = YAZ_BIB1_DATABASE_UNAVAILABLE;
 552                     rr->errstring = toKeep->databaseName;
 553                 }
 554                 if (!strcmp(toKeep->databaseName, "accept"))
 555                     rr->errcode = -1;
 556             }
 557             if (toKeep)
 558             {
 559                 Z_External *ext = (Z_External *)
 560                     odr_malloc(rr->stream, sizeof(*ext));
 561                 Z_IUOriginPartToKeep *keep = (Z_IUOriginPartToKeep *)
 562                     odr_malloc(rr->stream, sizeof(*keep));
 563                 Z_IUTargetPart *targetPart = (Z_IUTargetPart *)
 564                     odr_malloc(rr->stream, sizeof(*targetPart));
 565
 566                 rr->taskPackage = (Z_TaskPackage *)
 567                     odr_malloc(rr->stream, sizeof(*rr->taskPackage));
 568                 rr->taskPackage->packageType =
 569                     odr_oiddup(rr->stream, rr->esr->packageType);
 570                 rr->taskPackage->packageName = 0;
 571                 rr->taskPackage->userId = 0;
 572                 rr->taskPackage->retentionTime = 0;
 573                 rr->taskPackage->permissions = 0;
 574                 rr->taskPackage->description = 0;
 575                 rr->taskPackage->targetReference = (Odr_oct *)
 576                     odr_malloc(rr->stream, sizeof(Odr_oct));
 577                 rr->taskPackage->targetReference->buf =
 578                     (unsigned char *) odr_strdup(rr->stream, "123");
 579                 rr->taskPackage->targetReference->len =
 580                     rr->taskPackage->targetReference->size =
 581                     strlen((char *) (rr->taskPackage->targetReference->buf));
 582                 rr->taskPackage->creationDateTime = 0;
 583                 rr->taskPackage->taskStatus = odr_intdup(rr->stream, 0);
 584                 rr->taskPackage->packageDiagnostics = 0;
 585                 rr->taskPackage->taskSpecificParameters = ext;
 586
 587                 ext->direct_reference =
 588                     odr_oiddup(rr->stream, rr->esr->packageType);
 589                 ext->indirect_reference = 0;
 590                 ext->descriptor = 0;
 591                 ext->which = Z_External_update;
 592                 ext->u.update = (Z_IUUpdate *)
 593                     odr_malloc(rr->stream, sizeof(*ext->u.update));
 594                 ext->u.update->which = Z_IUUpdate_taskPackage;
 595                 ext->u.update->u.taskPackage =  (Z_IUUpdateTaskPackage *)
 596                     odr_malloc(rr->stream, sizeof(Z_IUUpdateTaskPackage));
 597                 ext->u.update->u.taskPackage->originPart = keep;
 598                 ext->u.update->u.taskPackage->targetPart = targetPart;
 599
 600                 keep->action = odr_intdup(rr->stream, *toKeep->action);
 601                 keep->databaseName =
 602                     odr_strdup(rr->stream, toKeep->databaseName);
 603                 keep->schema = 0;
 604                 keep->elementSetName = 0;
 605                 keep->actionQualifier = 0;
 606
 607                 targetPart->updateStatus = odr_intdup(rr->stream, 1);
 608                 targetPart->num_globalDiagnostics = 0;
 609                 targetPart->globalDiagnostics = (Z_DiagRec **) odr_nullval();
 610                 targetPart->num_taskPackageRecords = 1;
 611                 targetPart->taskPackageRecords =
 612                     (Z_IUTaskPackageRecordStructure **)
 613                     odr_malloc(rr->stream,
 614                                sizeof(Z_IUTaskPackageRecordStructure *));
 615                 targetPart->taskPackageRecords[0] =
 616                     (Z_IUTaskPackageRecordStructure *)
 617                     odr_malloc(rr->stream,
 618                                sizeof(Z_IUTaskPackageRecordStructure));
 619
 620                 targetPart->taskPackageRecords[0]->which =
 621                     Z_IUTaskPackageRecordStructure_record;
 622                 targetPart->taskPackageRecords[0]->u.record =
 623                     z_ext_record_sutrs(rr->stream, "test", 4);
 624                 targetPart->taskPackageRecords[0]->correlationInfo = 0;
 625                 targetPart->taskPackageRecords[0]->recordStatus =
 626                     odr_intdup(rr->stream,
 627                                Z_IUTaskPackageRecordStructure_success);
 628                 targetPart->taskPackageRecords[0]->num_supplementalDiagnostics
 629                     = 0;
 630
 631                 targetPart->taskPackageRecords[0]->supplementalDiagnostics = 0;
 632             }
 633             if (notToKeep)
 634             {
 635                 int i;
 636                 for (i = 0; i < notToKeep->num; i++)
 637                 {
 638                     Z_External *rec = notToKeep->elements[i]->record;
 639
 640                     if (rec->direct_reference)
 641                     {
 642                         char oid_name_str[OID_STR_MAX];
 643                         const char *oid_name
 644                             = oid_name = yaz_oid_to_string_buf(
 645                                 rec->direct_reference, 0,
 646                                 oid_name_str);
 647                         if (oid_name)
 648                             yaz_log(log_level, "record %d type %s", i,
 649                                     oid_name);
 650                     }
 651                     switch (rec->which)
 652                     {
 653                     case Z_External_sutrs:
 654                         if (rec->u.octet_aligned->len > 170)
 655                             yaz_log(log_level, "%d bytes:\n%.168s ...",
 656                                     rec->u.sutrs->len,
 657                                     rec->u.sutrs->buf);
 658                         else
 659                             yaz_log(log_level, "%d bytes:\n%s",
 660                                     rec->u.sutrs->len,
 661                                     rec->u.sutrs->buf);
 662                         break;
 663                     case Z_External_octet        :
 664                         if (rec->u.octet_aligned->len > 170)
 665                             yaz_log(log_level, "%d bytes:\n%.168s ...",
 666                                     rec->u.octet_aligned->len,
 667                                     rec->u.octet_aligned->buf);
 668                         else
 669                             yaz_log(log_level, "%d bytes\n%s",
 670                                     rec->u.octet_aligned->len,
 671                                     rec->u.octet_aligned->buf);
 672                     }
 673                 }
 674             }
 675         }
 676     }
 677     return 0;
 678 }
 679
 680 /* result set delete */
 681 int ztest_delete(void *handle, bend_delete_rr *rr)
 682 {
 683     if (rr->num_setnames == 1 && !strcmp(rr->setnames[0], "1"))
 684         rr->delete_status = Z_DeleteStatus_success;
 685     else
 686         rr->delete_status = Z_DeleteStatus_resultSetDidNotExist;
 687     return 0;
 688 }
 689
 690 /* Our sort handler really doesn't sort... */
 691 int ztest_sort(void *handle, bend_sort_rr *rr)
 692 {
 693     rr->errcode = 0;
 694     rr->sort_status = Z_SortResponse_success;
 695     return 0;
 696 }
 697
 698
 699 /* present request handler */
 700 int ztest_present(void *handle, bend_present_rr *rr)
 701 {
 702     struct session_handle *sh = (struct session_handle*) handle;
 703     struct result_set *set = get_set(sh, rr->setname);
 704
 705     if (!set)
 706     {
 707         rr->errcode = YAZ_BIB1_SPECIFIED_RESULT_SET_DOES_NOT_EXIST;
 708         rr->errstring = odr_strdup(rr->stream, rr->setname);
 709         return 0;
 710     }
 711     do_delay(&set->present_delay);
 712     return 0;
 713 }
 714
 715 /* retrieval of a single record (present, and piggy back search) */
 716 int ztest_fetch(void *handle, bend_fetch_rr *r)
 717 {
 718     struct session_handle *sh = (struct session_handle*) handle;
 719     char *cp;
 720     const Odr_oid *oid = r->request_format;
 721     struct result_set *set = get_set(sh, r->setname);
 722
 723     if (!set)
 724     {
 725         r->errcode = YAZ_BIB1_SPECIFIED_RESULT_SET_DOES_NOT_EXIST;
 726         r->errstring = odr_strdup(r->stream, r->setname);
 727         return 0;
 728     }
 729     do_delay(&set->fetch_delay);
 730     r->last_in_set = 0;
 731     r->basename = set->db;
 732     r->output_format = r->request_format;
 733
 734     if (r->number < 1 || r->number > set->hits)
 735     {
 736         r->errcode = YAZ_BIB1_PRESENT_REQUEST_OUT_OF_RANGE;
 737         return 0;
 738     }
 739     if (!oid || yaz_oid_is_iso2709(oid))
 740     {
 741         cp = dummy_marc_record(r->number, r->stream);
 742         if (!cp)
 743         {
 744             r->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 745             r->surrogate_flag = 1;
 746             return 0;
 747         }
 748         else
 749         {
 750             r->len = strlen(cp);
 751             r->record = cp;
 752             r->output_format = odr_oiddup(r->stream, yaz_oid_recsyn_usmarc);
 753         }
 754     }
 755     else if (!oid_oidcmp(oid, yaz_oid_recsyn_opac))
 756     {
 757         cp = dummy_marc_record(r->number, r->stream);
 758         if (!cp)
 759         {
 760             r->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 761             r->surrogate_flag = 1;
 762             return 0;
 763         }
 764         r->record = (char *) dummy_opac(r->number, r->stream, cp);
 765         r->len = -1;
 766     }
 767     else if (!oid_oidcmp(oid, yaz_oid_recsyn_sutrs))
 768     {
 769         /* this section returns a small record */
 770         char buf[100];
 771
 772         sprintf(buf, "This is dummy SUTRS record number %d\n", r->number);
 773
 774         r->len = strlen(buf);
 775         r->record = (char *) odr_malloc(r->stream, r->len+1);
 776         strcpy(r->record, buf);
 777     }
 778     else if (!oid_oidcmp(oid, yaz_oid_recsyn_grs_1))
 779     {
 780         r->len = -1;
 781         r->record = (char*) dummy_grs_record(r->number, r->stream);
 782         if (!r->record)
 783         {
 784             r->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 785             r->surrogate_flag = 1;
 786             return 0;
 787         }
 788     }
 789     else if (!oid_oidcmp(oid, yaz_oid_recsyn_postscript))
 790     {
 791         char fname[20];
 792         FILE *f;
 793         long size;
 794
 795         sprintf(fname, "part.%d.ps", r->number);
 796         f = fopen(fname, "rb");
 797         if (!f)
 798         {
 799             r->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 800             r->surrogate_flag = 1;
 801             return 0;
 802         }
 803         fseek(f, 0L, SEEK_END);
 804         size = ftell(f);
 805         if (size <= 0 || size >= 5000000)
 806         {
 807             r->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 808             r->surrogate_flag = 1;
 809         }
 810         fseek(f, 0L, SEEK_SET);
 811         r->record = (char*) odr_malloc(r->stream, size);
 812         r->len = size;
 813         if (fread(r->record, size, 1, f) != 1)
 814         {
 815             r->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 816             r->surrogate_flag = 1;
 817         }
 818         fclose(f);
 819     }
 820     else if (!oid_oidcmp(oid, yaz_oid_recsyn_xml))
 821     {
 822         if ((cp = dummy_xml_record(r->number, r->stream)))
 823         {
 824             r->len = strlen(cp);
 825             r->record = cp;
 826         }
 827         else
 828         {
 829             r->errcode = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS;
 830             r->surrogate_flag = 1;
 831             return 0;
 832         }
 833     }
 834     else
 835     {
 836         char buf[OID_STR_MAX];
 837         r->errcode = YAZ_BIB1_RECORD_SYNTAX_UNSUPP;
 838         r->errstring = odr_strdup(r->stream, oid_oid_to_dotstring(oid, buf));
 839         return 0;
 840     }
 841     r->errcode = 0;
 842     return 0;
 843 }
 844
 845 /*
 846  * silly dummy-scan what reads words from a file.
 847  */
 848 int ztest_scan(void *handle, bend_scan_rr *q)
 849 {
 850     static FILE *f = 0;
 851     static struct scan_entry list[200];
 852     static char entries[200][80];
 853     int hits[200];
 854     char term[80], *p;
 855     int i, pos;
 856     int term_position_req = q->term_position;
 857     int num_entries_req = q->num_entries;
 858
 859     /* Throw Database unavailable if other than Default or Slow */
 860     if (!yaz_matchstr(q->basenames[0], "Default"))
 861         ;  /* Default is OK in our test */
 862     else if (check_slow(q->basenames[0], 0 /* no assoc for scan */))
 863         ;
 864     else
 865     {
 866         q->errcode = YAZ_BIB1_DATABASE_UNAVAILABLE;
 867         q->errstring = q->basenames[0];
 868         return 0;
 869     }
 870
 871     q->errcode = 0;
 872     q->errstring = 0;
 873     q->entries = list;
 874     q->status = BEND_SCAN_SUCCESS;
 875     if (!f && !(f = fopen("dummy-words", "r")))
 876     {
 877         perror("dummy-words");
 878         exit(1);
 879     }
 880     if (q->num_entries > 200)
 881     {
 882         q->errcode = YAZ_BIB1_RESOURCES_EXHAUSTED_NO_RESULTS_AVAILABLE;
 883         return 0;
 884     }
 885     if (q->term)
 886     {
 887         int len;
 888         if (q->term->term->which != Z_Term_general)
 889         {
 890             q->errcode = YAZ_BIB1_TERM_TYPE_UNSUPP;
 891             return 0;
 892         }
 893         if (*q->step_size != 0)
 894         {
 895             q->errcode = YAZ_BIB1_ONLY_ZERO_STEP_SIZE_SUPPORTED_FOR_SCAN;
 896             return 0;
 897         }
 898         len = q->term->term->u.general->len;
 899         if (len >= (int ) sizeof(term))
 900             len = sizeof(term)-1;
 901         memcpy(term, q->term->term->u.general->buf, len);
 902         term[len] = '\0';
 903     }
 904     else if (q->scanClause)
 905     {
 906         strncpy(term, q->scanClause, sizeof(term)-1);
 907         term[sizeof(term)-1] = '\0';
 908     }
 909     else
 910         strcpy(term, "0");
 911
 912     for (p = term; *p; p++)
 913         if (islower(*(unsigned char *) p))
 914             *p = toupper(*p);
 915
 916     fseek(f, 0, SEEK_SET);
 917     q->num_entries = 0;
 918
 919     for (i = 0, pos = 0; fscanf(f, " %79[^:]:%d", entries[pos], &hits[pos]) == 2;
 920          i++, pos < 199 ? pos++ : (pos = 0))
 921     {
 922         if (!q->num_entries && strcmp(entries[pos], term) >= 0) /* s-point fnd */
 923         {
 924             if ((q->term_position = term_position_req) > i + 1)
 925             {
 926                 q->term_position = i + 1;
 927                 q->status = BEND_SCAN_PARTIAL;
 928             }
 929             for (; q->num_entries < q->term_position; q->num_entries++)
 930             {
 931                 int po;
 932
 933                 po = pos - q->term_position + q->num_entries+1; /* find pos */
 934                 if (po < 0)
 935                     po += 200;
 936
 937                 if (!strcmp(term, "SD") && q->num_entries == 2)
 938                 {
 939                     list[q->num_entries].term = entries[pos];
 940                     list[q->num_entries].occurrences = -1;
 941                     list[q->num_entries].errcode =
 942                         YAZ_BIB1_SCAN_UNSUPP_VALUE_OF_POSITION_IN_RESPONSE;
 943                     list[q->num_entries].errstring = "SD for Scan Term";
 944                 }
 945                 else
 946                 {
 947                     list[q->num_entries].term = entries[po];
 948                     list[q->num_entries].occurrences = hits[po];
 949                 }
 950             }
 951         }
 952         else if (q->num_entries)
 953         {
 954             list[q->num_entries].term = entries[pos];
 955             list[q->num_entries].occurrences = hits[pos];
 956             q->num_entries++;
 957         }
 958         if (q->num_entries >= num_entries_req)
 959             break;
 960     }
 961     if (feof(f))
 962         q->status = BEND_SCAN_PARTIAL;
 963     return 0;
 964 }
 965
 966 int ztest_explain(void *handle, bend_explain_rr *rr)
 967 {
 968     if (rr->database && !strcmp(rr->database, "Default"))
 969     {
 970         rr->explain_buf = "<explain>\n"
 971             "\t<serverInfo>\n"
 972             "\t\t<host>localhost</host>\n"
 973             "\t\t<port>210</port>\n"
 974             "\t</serverInfo>\n"
 975             "</explain>\n";
 976     }
 977     return 0;
 978 }
 979
 980 int ztest_update(void *handle, bend_update_rr *rr)
 981 {
 982     rr->operation_status = "success";
 983     return 0;
 984 }
 985
 986 bend_initresult *bend_init(bend_initrequest *q)
 987 {
 988     bend_initresult *r = (bend_initresult *)
 989         odr_malloc(q->stream, sizeof(*r));
 990     struct session_handle *sh = xmalloc(sizeof(*sh));
 991
 992     sh->result_sets = 0;
 993
 994     if (!log_level_set)
 995     {
 996         log_level=yaz_log_module_level("ztest");
 997         log_level_set=1;
 998     }
 999
1000     r->errcode = 0;
1001     r->errstring = 0;
1002     r->handle = sh;                         /* tell GFS about our handle */
1003     q->bend_sort = ztest_sort;              /* register sort handler */
1004     q->bend_search = ztest_search;          /* register search handler */
1005     q->bend_present = ztest_present;        /* register present handle */
1006     q->bend_esrequest = ztest_esrequest;
1007     q->bend_delete = ztest_delete;
1008     q->bend_fetch = ztest_fetch;
1009     q->bend_scan = ztest_scan;
1010 #if 0
1011     q->bend_explain = ztest_explain;
1012 #endif
1013     q->bend_srw_scan = ztest_scan;
1014     q->bend_srw_update = ztest_update;
1015
1016     q->query_charset = "ISO-8859-1";
1017     q->records_in_same_charset = 0;
1018
1019     return r;
1020 }
1021
1022 void bend_close(void *handle)
1023 {
1024     struct session_handle *sh = (struct session_handle*) handle;
1025     remove_sets(sh);
1026     xfree(sh);              /* release our session */
1027     return;
1028 }
1029
1030 int main(int argc, char **argv)
1031 {
1032     return statserv_main(argc, argv, bend_init, bend_close);
1033 }
1034 /*
1035  * Local variables:
1036  * c-basic-offset: 4
1037  * c-file-style: "Stroustrup"
1038  * indent-tabs-mode: nil
1039  * End:
1040  * vim: shiftwidth=4 tabstop=8 expandtab
1041  */
1042