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