xfree/xmalloc used everywhere.
[idzebra-moved-to-github.git] / index / zserver.c
1 /*
2  * Copyright (C) 1994-1995, Index Data I/S 
3  * All rights reserved.
4  * Sebastian Hammer, Adam Dickmeiss
5  *
6  * $Log: zserver.c,v $
7  * Revision 1.8  1995-09-28 09:19:47  adam
8  * xfree/xmalloc used everywhere.
9  * Extract/retrieve method seems to work for text records.
10  *
11  * Revision 1.7  1995/09/27  16:17:32  adam
12  * More work on retrieve.
13  *
14  * Revision 1.6  1995/09/08  08:53:22  adam
15  * Record buffer maintained in server_info.
16  *
17  * Revision 1.5  1995/09/06  16:11:18  adam
18  * Option: only one word key per file.
19  *
20  * Revision 1.4  1995/09/06  10:33:04  adam
21  * More work on present. Some log messages removed.
22  *
23  * Revision 1.3  1995/09/05  15:28:40  adam
24  * More work on search engine.
25  *
26  * Revision 1.2  1995/09/04  12:33:43  adam
27  * Various cleanup. YAZ util used instead.
28  *
29  * Revision 1.1  1995/09/04  09:10:41  adam
30  * More work on index add/del/update.
31  * Merge sort implemented.
32  * Initial work on z39 server.
33  *
34  */
35 #include <stdio.h>
36 #include <assert.h>
37 #include <unistd.h>
38 #include <fcntl.h>
39
40 #include <recctrl.h>
41 #include <backend.h>
42 #include <dmalloc.h>
43 #include "zserver.h"
44
45 ZServerInfo server_info;
46
47 bend_initresult *bend_init (bend_initrequest *q)
48 {
49     static bend_initresult r;
50     static char *name = "zserver";
51
52     r.errcode = 0;
53     r.errstring = 0;
54     r.handle = name;
55
56     server_info.sets = NULL;
57     if (!(server_info.sys_idx_fd = open (FNAME_SYS_IDX, O_RDONLY)))
58     {
59         r.errcode = 1;
60         r.errstring = "dict_open fail: filedict";
61         return &r;
62     }
63     if (!(server_info.fileDict = dict_open (FNAME_FILE_DICT, 10, 0)))
64     {
65         r.errcode = 1;
66         r.errstring = "dict_open fail: filedict";
67         return &r;
68     }    
69     if (!(server_info.wordDict = dict_open (FNAME_WORD_DICT, 40, 0)))
70     {
71         dict_close (server_info.fileDict);
72         r.errcode = 1;
73         r.errstring = "dict_open fail: worddict";
74         return &r;
75     }    
76     if (!(server_info.wordIsam = is_open (FNAME_WORD_ISAM, key_compare, 0,
77                                           sizeof (struct it_key))))
78     {
79         dict_close (server_info.wordDict);
80         dict_close (server_info.fileDict);
81         r.errcode = 1;
82         r.errstring = "is_open fail: wordisam";
83         return &r;
84     }
85     server_info.recordBuf = NULL;
86     return &r;
87 }
88
89 bend_searchresult *bend_search (void *handle, bend_searchrequest *q, int *fd)
90 {
91     static bend_searchresult r;
92
93     r.errcode = 0;
94     r.errstring = 0;
95     r.hits = 0;
96
97     switch (q->query->which)
98     {
99     case Z_Query_type_1:
100         r.errcode = rpn_search (&server_info, q->query->u.type_1,
101                                 q->num_bases, q->basenames, q->setname,
102                                 &r.hits);
103         break;
104     default:
105         r.errcode = 107;
106     }
107     return &r;
108 }
109
110 static int record_read (int fd, char *buf, size_t count)
111 {
112     return read (fd, buf, count);
113 }
114
115 static int record_fetch (ZServerInfo *zi, int sysno, ODR stream,
116                           oid_value input_format, oid_value *output_format,
117                           char **rec_bufp, int *rec_lenp)
118 {
119     char record_info[SYS_IDX_ENTRY_LEN];
120     char *fname, *file_type;
121     RecType rt;
122     struct recRetrieveCtrl retrieveCtrl;
123
124     if (lseek (zi->sys_idx_fd, sysno * SYS_IDX_ENTRY_LEN,
125                SEEK_SET) == -1)
126     {
127         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: lseek of sys_idx");
128         exit (1);
129     }
130     if (read (zi->sys_idx_fd, record_info, SYS_IDX_ENTRY_LEN) == -1)
131     {
132         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: read of sys_idx");
133         exit (1);
134     }
135     file_type = record_info;
136     fname = record_info + strlen(record_info) + 1;
137     if (!(rt = recType_byName (file_type)))
138     {
139         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: Cannot handle type %s", 
140               file_type);
141         exit (1);
142     }
143     if ((retrieveCtrl.fd = open (fname, O_RDONLY)) == -1)
144     {
145         logf (LOG_FATAL|LOG_ERRNO, "Retrieve: Open record file %s", fname);
146         exit (1);
147     }
148     retrieveCtrl.odr = stream;
149     retrieveCtrl.readf = record_read;
150     retrieveCtrl.input_format = input_format;
151     (*rt->retrieve)(&retrieveCtrl);
152     *output_format = retrieveCtrl.output_format;
153     *rec_bufp = retrieveCtrl.rec_buf;
154     *rec_lenp = retrieveCtrl.rec_len;
155     close (retrieveCtrl.fd);
156     return 0;
157 }
158
159 bend_fetchresult *bend_fetch (void *handle, bend_fetchrequest *q, int *num)
160 {
161     static bend_fetchresult r;
162     int positions[2];
163     ZServerSetSysno *records;
164
165     r.errstring = 0;
166     r.last_in_set = 0;
167     r.basename = "base";
168
169     xfree (server_info.recordBuf);
170     server_info.recordBuf = NULL;
171     positions[0] = q->number;
172     records = resultSetSysnoGet (&server_info, q->setname, 1, positions);
173     if (!records)
174     {
175         logf (LOG_DEBUG, "resultSetRecordGet, error");
176         r.errcode = 13;
177         return &r;
178     }
179     if (!records[0].sysno)
180     {
181         r.errcode = 13;
182         logf (LOG_DEBUG, "Out of range. pos=%d", q->number);
183         return &r;
184     }
185     r.errcode = record_fetch (&server_info, records[0].sysno, q->stream,
186                               q->format, &r.format, &r.record, &r.len);
187     return &r;
188 }
189
190 bend_deleteresult *bend_delete (void *handle, bend_deleterequest *q, int *num)
191 {
192     return 0;
193 }
194
195 bend_scanresult *bend_scan (void *handle, bend_scanrequest *q, int *num)
196 {
197     static struct scan_entry list[200];
198     static char buf[200][200];
199     static bend_scanresult r;
200     int i;
201
202     r.term_position = q->term_position;
203     r.num_entries = q->num_entries;
204     r.entries = list;
205     for (i = 0; i < r.num_entries; i++)
206     {
207         list[i].term = buf[i];
208         sprintf(list[i].term, "term-%d", i+1);
209         list[i].occurrences = rand() % 100000;
210     }
211     r.errcode = 0;
212     r.errstring = 0;
213     return &r;
214 }
215
216 void bend_close (void *handle)
217 {
218     dict_close (server_info.fileDict);
219     dict_close (server_info.wordDict);
220     is_close (server_info.wordIsam);
221     close (server_info.sys_idx_fd);
222     xfree (server_info.recordBuf);
223     server_info.recordBuf = NULL;
224     return;
225 }
226
227 int main (int argc, char **argv)
228 {
229     char *base_name = "base";
230
231     if (!(common_resource = res_open (base_name)))
232     {
233         logf (LOG_FATAL, "Cannot open resource `%s'", base_name);
234         exit (1);
235     }
236     return statserv_main (argc, argv);
237 }