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