ztest/ztest.c

   1 /*
   2  * Copyright (c) 1995-1998, Index Data.
   3  * See the file LICENSE for details.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * NT Service interface by
   7  *    Chas Woodfield, Fretwell Downing Datasystems.
   8  *
   9  * $Log: ztest.c,v $
  10  * Revision 1.13  1998-08-19 16:10:09  adam
  11  * Changed som member names of DeleteResultSetRequest/Response.
  12  *
  13  * Revision 1.12  1998/07/20 12:38:44  adam
  14  * Implemented delete result set service to server API.
  15  *
  16  * Revision 1.11  1998/06/09 13:55:08  adam
  17  * Minor changes.
  18  *
  19  * Revision 1.10  1998/05/27 16:55:54  adam
  20  * Minor changes.
  21  *
  22  * Revision 1.9  1998/03/31 11:07:45  adam
  23  * Furhter work on UNIverse resource report.
  24  * Added Extended Services handling in frontend server.
  25  *
  26  * Revision 1.8  1998/02/11 11:53:36  adam
  27  * Changed code so that it compiles as C++.
  28  *
  29  * Revision 1.7  1998/02/10 11:03:57  adam
  30  * Added support for extended handlers in backend server interface.
  31  *
  32  * Revision 1.6  1998/01/29 13:16:02  adam
  33  * Added dummy sort in test server.
  34  *
  35  * Revision 1.5  1997/11/07 13:31:58  adam
  36  * Added NT Service name part of statserv_options_block. Moved NT
  37  * service utility to server library.
  38  *
  39  * Revision 1.4  1997/09/17 12:10:43  adam
  40  * YAZ version 1.4.
  41  *
  42  * Revision 1.3  1997/09/09 10:10:20  adam
  43  * Another MSV5.0 port. Changed projects to include proper
  44  * library/include paths.
  45  * Server starts server in test-mode when no options are given.
  46  *
  47  * Revision 1.2  1997/09/04 13:50:31  adam
  48  * Bug fix in ztest.
  49  *
  50  */
  51
  52 /*
  53  * Demonstration of simple server
  54  */
  55
  56 #include <stdio.h>
  57 #include <stdlib.h>
  58 #include <ctype.h>
  59 #include <assert.h>
  60
  61 #include <backend.h>
  62 #include <xmalloc.h>
  63 #include <proto.h>
  64 #include <log.h>
  65
  66 Z_GenericRecord *read_grs1(FILE *f, ODR o);
  67
  68 int ztest_search (void *handle, bend_search_rr *rr);
  69 int ztest_sort (void *handle, bend_sort_rr *rr);
  70 int ztest_present (void *handle, bend_present_rr *rr);
  71 int ztest_esrequest (void *handle, bend_esrequest_rr *rr);
  72 int ztest_delete (void *handle, bend_delete_rr *rr);
  73
  74 bend_initresult *bend_init(bend_initrequest *q)
  75 {
  76     bend_initresult *r = (bend_initresult *) odr_malloc (q->stream, sizeof(*r));
  77     static char *dummy = "Hej fister";
  78
  79     r->errcode = 0;
  80     r->errstring = 0;
  81     r->handle = dummy;
  82     q->bend_sort = ztest_sort;       /* register sort handler */
  83     q->bend_search = ztest_search;   /* register search handler */
  84     q->bend_present = ztest_present; /* register present handle */
  85     q->bend_esrequest = ztest_esrequest;
  86     q->bend_delete = ztest_delete;
  87     return r;
  88 }
  89
  90 int ztest_search (void *handle, bend_search_rr *rr)
  91 {
  92     rr->hits = rand() % 22;
  93     return 0;
  94 }
  95
  96 int ztest_present (void *handle, bend_present_rr *rr)
  97 {
  98     return 0;
  99 }
 100
 101 int ztest_esrequest (void *handle, bend_esrequest_rr *rr)
 102 {
 103     rr->errcode = 0;
 104     return 0;
 105 }
 106
 107 int ztest_delete (void *handle, bend_delete_rr *rr)
 108 {
 109     if (rr->num_setnames == 1 && !strcmp (rr->setnames[0], "1"))
 110         rr->delete_status = Z_DeleteStatus_success;
 111     else
 112         rr->delete_status = Z_DeleteStatus_resultSetDidNotExist;
 113     return 0;
 114 }
 115
 116 /* Obsolete bend_search, never called because handler is registered */
 117 bend_searchresult *bend_search(void *handle, bend_searchrequest *q, int *fd)
 118 {
 119     return 0;
 120 }
 121
 122 /* Our sort handler really doesn't sort... */
 123 int ztest_sort (void *handle, bend_sort_rr *rr)
 124 {
 125     rr->errcode = 0;
 126     rr->sort_status = Z_SortStatus_success;
 127     return 0;
 128 }
 129
 130 static int atoin (const char *buf, int n)
 131 {
 132     int val = 0;
 133     while (--n >= 0)
 134     {
 135         if (isdigit(*buf))
 136             val = val*10 + (*buf - '0');
 137         buf++;
 138     }
 139     return val;
 140 }
 141
 142 char *marc_read(FILE *inf)
 143 {
 144     char length_str[5];
 145     size_t size;
 146     char *buf;
 147
 148     if (fread (length_str, 1, 5, inf) != 5)
 149         return NULL;
 150     size = atoin (length_str, 5);
 151     if (size <= 6)
 152         return NULL;
 153     if (!(buf = (char*) xmalloc (size+1)))
 154         return NULL;
 155     if (fread (buf+5, 1, size-5, inf) != (size-5))
 156     {
 157         xfree (buf);
 158         return NULL;
 159     }
 160     memcpy (buf, length_str, 5);
 161     buf[size] = '\0';
 162     return buf;
 163 }
 164
 165 static char *dummy_database_record (int num)
 166 {
 167     FILE *inf = fopen ("dummy-records", "r");
 168     char *buf = 0;
 169
 170     if (!inf)
 171         return NULL;
 172     while (--num >= 0)
 173     {
 174         if (buf)
 175            xfree(buf);
 176         if (num == 98)
 177         {
 178             buf = (char*) xmalloc(2101);
 179             assert(buf);
 180             memset(buf, 'A', 2100);
 181             buf[2100] = '\0';
 182             break;
 183         }
 184         else
 185             buf = marc_read (inf);
 186         if (!num || !buf)
 187             break;
 188     }
 189     fclose(inf);
 190     if (num < 0)
 191         return 0;
 192     return buf;
 193 }
 194
 195 static Z_GenericRecord *dummy_grs_record (int num, ODR o)
 196 {
 197     FILE *f = fopen("dummy-grs", "r");
 198     char line[512];
 199     Z_GenericRecord *r = 0;
 200     int n;
 201
 202     if (!f)
 203         return 0;
 204     while (fgets(line, 512, f))
 205         if (*line == '#' && sscanf(line, "#%d", &n) == 1 && n == num)
 206         {
 207             r = read_grs1(f, o);
 208             break;
 209         }
 210     fclose(f);
 211     return r;
 212 }
 213
 214 bend_fetchresult *bend_fetch(void *handle, bend_fetchrequest *q, int *fd)
 215 {
 216     bend_fetchresult *r = (bend_fetchresult *)
 217                         odr_malloc (q->stream, sizeof(*r));
 218     char *cp;
 219     r->errstring = 0;
 220     r->last_in_set = 0;
 221     r->basename = "DUMMY";
 222     r->format = q->format;
 223     if (q->format == VAL_SUTRS)
 224     {
 225         char buf[100];
 226
 227         sprintf(buf, "This is dummy SUTRS record number %d\n", q->number);
 228         r->len = strlen(buf);
 229         r->record = odr_malloc (q->stream, r->len+1);
 230         strcpy(r->record, buf);
 231     }
 232     else if (q->format == VAL_GRS1)
 233     {
 234         r->len = -1;
 235         r->record = (char*) dummy_grs_record(q->number, q->stream);
 236         if (!r->record)
 237         {
 238             r->errcode = 13;
 239             return r;
 240         }
 241     }
 242     else if ((cp = dummy_database_record(q->number)))
 243     {
 244         r->len = strlen(cp);
 245         r->record = (char *) odr_malloc (q->stream, 1+r->len);
 246         strcpy (r->record, cp);
 247         r->format = VAL_USMARC;
 248     }
 249     else
 250     {
 251         r->errcode = 13;
 252         return r;
 253     }
 254     r->errcode = 0;
 255     return r;
 256 }
 257
 258 /*
 259  * silly dummy-scan what reads words from a file.
 260  */
 261 bend_scanresult *bend_scan(void *handle, bend_scanrequest *q, int *fd)
 262 {
 263     bend_scanresult *r = (bend_scanresult *) odr_malloc (q->stream, sizeof(*r));
 264     static FILE *f = 0;
 265     static struct scan_entry list[200];
 266     static char entries[200][80];
 267     int hits[200];
 268     char term[80], *p;
 269     int i, pos;
 270
 271     r->errcode = 0;
 272     r->errstring = 0;
 273     r->entries = list;
 274     r->status = BEND_SCAN_SUCCESS;
 275     if (!f && !(f = fopen("dummy-words", "r")))
 276     {
 277         perror("dummy-words");
 278         exit(1);
 279     }
 280     if (q->term->term->which != Z_Term_general)
 281     {
 282         r->errcode = 229; /* unsupported term type */
 283         return r;
 284     }
 285     if (q->term->term->u.general->len >= 80)
 286     {
 287         r->errcode = 11; /* term too long */
 288         return r;
 289     }
 290     if (q->num_entries > 200)
 291     {
 292         r->errcode = 31;
 293         return r;
 294     }
 295     memcpy(term, q->term->term->u.general->buf, q->term->term->u.general->len);
 296     term[q->term->term->u.general->len] = '\0';
 297     for (p = term; *p; p++)
 298         if (islower(*p))
 299             *p = toupper(*p);
 300
 301     fseek(f, 0, SEEK_SET);
 302     r->num_entries = 0;
 303     for (i = 0, pos = 0; fscanf(f, " %79[^:]:%d", entries[pos], &hits[pos]) == 2;
 304         i++, pos < 199 ? pos++ : (pos = 0))
 305     {
 306         if (!r->num_entries && strcmp(entries[pos], term) >= 0) /* s-point fnd */
 307         {
 308             if ((r->term_position = q->term_position) > i + 1)
 309             {
 310                 r->term_position = i + 1;
 311                 r->status = BEND_SCAN_PARTIAL;
 312             }
 313             for (; r->num_entries < r->term_position; r->num_entries++)
 314             {
 315                 int po;
 316
 317                 po = pos - r->term_position + r->num_entries + 1; /* find pos */
 318                 if (po < 0)
 319                     po += 200;
 320                 list[r->num_entries].term = entries[po];
 321                 list[r->num_entries].occurrences = hits[po];
 322             }
 323         }
 324         else if (r->num_entries)
 325         {
 326             list[r->num_entries].term = entries[pos];
 327             list[r->num_entries].occurrences = hits[pos];
 328             r->num_entries++;
 329         }
 330         if (r->num_entries >= q->num_entries)
 331             break;
 332     }
 333     if (feof(f))
 334         r->status = BEND_SCAN_PARTIAL;
 335     return r;
 336 }
 337
 338 void bend_close(void *handle)
 339 {
 340     return;
 341 }
 342
 343 int main(int argc, char **argv)
 344 {
 345     return statserv_main(argc, argv);
 346 }