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