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