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