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