Fix infinite loop (bug introduced by previous commit)
[yaz-moved-to-github.git] / retrieval / d1_marc.c
1 /*
2  * Copyright (c) 1995-2002, Index Data.
3  * See the file LICENSE for details.
4  * Sebastian Hammer, Adam Dickmeiss
5  *
6  * $Id: d1_marc.c,v 1.26 2002-08-28 19:10:56 adam Exp $
7  */
8
9
10 #include <assert.h>
11 #include <stdlib.h>
12 #include <string.h>
13
14 #include <yaz/oid.h>
15 #include <yaz/log.h>
16 #include <yaz/marcdisp.h>
17 #include <yaz/readconf.h>
18 #include <yaz/xmalloc.h>
19 #include <yaz/data1.h>
20 #include <yaz/tpath.h>
21
22 data1_marctab *data1_read_marctab (data1_handle dh, const char *file)
23 {
24     FILE *f;
25     NMEM mem = data1_nmem_get (dh);
26     data1_marctab *res = (data1_marctab *)nmem_malloc(mem, sizeof(*res));
27     char line[512], *argv[50];
28     int lineno = 0;
29     int argc;
30     
31     if (!(f = data1_path_fopen(dh, file, "r")))
32     {
33         yaz_log(LOG_WARN|LOG_ERRNO, "%s", file);
34         return 0;
35     }
36
37     res->name = 0;
38     res->reference = VAL_NONE;
39     res->next = 0;
40     res->length_data_entry = 4;
41     res->length_starting = 5;
42     res->length_implementation = 0;
43     strcpy(res->future_use, "4");
44
45     strcpy(res->record_status, "n");
46     strcpy(res->implementation_codes, "    ");
47     res->indicator_length = 2;
48     res->identifier_length = 2;
49     res->force_indicator_length = -1;
50     res->force_identifier_length = -1;
51     strcpy(res->user_systems, "z  ");
52     
53     while ((argc = readconf_line(f, &lineno, line, 512, argv, 50)))
54         if (!strcmp(*argv, "name"))
55         {
56             if (argc != 2)
57             {
58                 yaz_log(LOG_WARN, "%s:%d:Missing arg for %s", file, lineno,
59                         *argv);
60                 continue;
61             }
62             res->name = nmem_strdup(mem, argv[1]);
63         }
64         else if (!strcmp(*argv, "reference"))
65         {
66             if (argc != 2)
67             {
68                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
69                         *argv);
70                 continue;
71             }
72             if ((res->reference = oid_getvalbyname(argv[1])) == VAL_NONE)
73             {
74                 yaz_log(LOG_WARN, "%s:%d: Unknown tagset reference '%s'",
75                         file, lineno, argv[1]);
76                 continue;
77             }
78         }
79         else if (!strcmp(*argv, "length-data-entry"))
80         {
81             if (argc != 2)
82             {
83                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
84                         *argv);
85                 continue;
86             }
87             res->length_data_entry = atoi(argv[1]);
88         }
89         else if (!strcmp(*argv, "length-starting"))
90         {
91             if (argc != 2)
92             {
93                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
94                         *argv);
95                 continue;
96             }
97             res->length_starting = atoi(argv[1]);
98         }
99         else if (!strcmp(*argv, "length-implementation"))
100         {
101             if (argc != 2)
102             {
103                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
104                         *argv);
105                 continue;
106             }
107             res->length_implementation = atoi(argv[1]);
108         }
109         else if (!strcmp(*argv, "future-use"))
110         {
111             if (argc != 2)
112             {
113                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
114                         *argv);
115                 continue;
116             }
117             strncpy(res->future_use, argv[1], 2);
118         }
119         else if (!strcmp(*argv, "force-indicator-length"))
120         {
121             if (argc != 2)
122             {
123                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
124                         *argv);
125                 continue;
126             }
127             res->force_indicator_length = atoi(argv[1]);
128         }
129         else if (!strcmp(*argv, "force-identifier-length"))
130         {
131             if (argc != 2)
132             {
133                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
134                         *argv);
135                 continue;
136             }
137             res->force_identifier_length = atoi(argv[1]);
138         }
139         else
140             yaz_log(LOG_WARN, "%s:%d: Unknown directive '%s'", file, lineno,
141                     *argv);
142
143     fclose(f);
144     return res;
145 }
146
147
148 /*
149  * Locate some data under this node. This routine should handle variants
150  * prettily.
151  */
152 static char *get_data(data1_node *n, int *len)
153 {
154     char *r;
155     data1_node *np = 0;
156
157     while (n)
158     {
159         if (n->which == DATA1N_data)
160         {
161             int i;
162             *len = n->u.data.len;
163
164             for (i = 0; i<*len; i++)
165                 if (!d1_isspace(n->u.data.data[i]))
166                     break;
167             while (*len && d1_isspace(n->u.data.data[*len - 1]))
168                 (*len)--;
169             *len = *len - i;
170             if (*len > 0)
171                 return n->u.data.data + i;
172         }
173         if (n->which == DATA1N_tag)
174             n = n->child;
175         else if (n->which == DATA1N_data)
176             n = n->next;
177         else
178             break;      
179     }
180     r = "";
181     *len = strlen(r);
182     return r;
183 }
184
185 static void memint (char *p, int val, int len)
186 {
187     char buf[10];
188
189     if (len == 1)
190         *p = val + '0';
191     else
192     {
193         sprintf (buf, "%08d", val);
194         memcpy (p, buf+8-len, len);
195     }
196 }
197
198 static int is_indicator (data1_marctab *p, data1_node *subf)
199 {
200 #if 1
201     if (p->indicator_length != 2 ||
202         (subf && subf->which == DATA1N_tag && strlen(subf->u.tag.tag) == 2))
203         return 1;
204 #else
205     if (subf && subf->which == DATA1N_tag && subf->child->which == DATA1N_tag)
206         return 1;
207 #endif
208     return 0;
209 }
210
211 static int nodetomarc(data1_handle dh,
212                       data1_marctab *p, data1_node *n, int selected,
213                       char **buf, int *size)
214 {
215     int len = 26;
216     int dlen;
217     int base_address = 25;
218     int entry_p, data_p;
219     char *op;
220     data1_node *field, *subf;
221
222     yaz_log (LOG_DEBUG, "nodetomarc");
223
224     for (field = n->child; field; field = field->next)
225     {
226         int is00X = 0;
227
228         if (field->which != DATA1N_tag)
229         {
230             yaz_log(LOG_WARN, "Malformed field composition for marc output.");
231             return -1;
232         }
233         if (selected && !field->u.tag.node_selected)
234             continue;
235
236         subf = field->child;
237         if (!subf)
238             continue;
239
240         len += 4 + p->length_data_entry + p->length_starting
241             + p->length_implementation;
242         base_address += 3 + p->length_data_entry + p->length_starting
243             + p->length_implementation;
244
245         if (subf->which == DATA1N_data)
246             is00X = 1;
247         if (!data1_is_xmlmode(dh))
248         {
249             if (subf->which == DATA1N_tag && !strcmp(subf->u.tag.tag, "@"))
250                 is00X = 1;
251         }
252             
253         
254         if (!is00X)
255             len += p->indicator_length;  
256         /*  we'll allow no indicator if length is not 2 */
257         if (is_indicator (p, subf))
258             subf = subf->child;
259
260         for (; subf; subf = subf->next)
261         {
262             if (!is00X)
263                 len += p->identifier_length;
264             get_data(subf, &dlen);
265             len += dlen;
266         }
267     }
268
269     if (!*buf)
270         *buf = (char *)xmalloc(*size = len);
271     else if (*size <= len)
272         *buf = (char *)xrealloc(*buf, *size = len);
273         
274     op = *buf;
275     memint (op, len, 5);
276     memcpy (op+5, p->record_status, 1);
277     memcpy (op+6, p->implementation_codes, 4);
278     memint (op+10, p->indicator_length, 1);
279     memint (op+11, p->identifier_length, 1);
280     memint (op+12, base_address, 5);
281     memcpy (op+17, p->user_systems, 3);
282     memint (op+20, p->length_data_entry, 1);
283     memint (op+21, p->length_starting, 1);
284     memint (op+22, p->length_implementation, 1);
285     memcpy (op+23, p->future_use, 1);
286     
287     entry_p = 24;
288     data_p = base_address;
289
290     for (field = n->child; field; field = field->next)
291     {
292         int is00X = 0;
293
294         int data_0 = data_p;
295         char *indicator_data = "    ";
296         if (selected && !field->u.tag.node_selected)
297             continue;
298
299         subf = field->child;
300         if (!subf)
301             continue;
302
303         if (subf->which == DATA1N_data)
304             is00X = 1;
305         if (!data1_is_xmlmode(dh))
306         {
307             if (subf->which == DATA1N_tag && !strcmp(subf->u.tag.tag, "@"))
308                 is00X = 1;
309         }
310
311         if (is_indicator (p, subf))
312         {
313             indicator_data = subf->u.tag.tag;
314             subf = subf->child;
315         }
316         if (!is00X)
317         {
318             memcpy (op + data_p, indicator_data, p->indicator_length);
319             data_p += p->indicator_length;
320         }
321         for (; subf; subf = subf->next)
322         {
323             char *data;
324
325             if (!is00X)
326             {
327                 const char *identifier = "a";
328                 if (subf->which != DATA1N_tag)
329                     yaz_log(LOG_WARN, "Malformed fields for marc output.");
330                 else
331                     identifier = subf->u.tag.tag;
332                 op[data_p] = ISO2709_IDFS;
333                 memcpy (op + data_p+1, identifier, p->identifier_length-1);
334                 data_p += p->identifier_length;
335             }
336             data = get_data(subf, &dlen);
337             memcpy (op + data_p, data, dlen);
338             data_p += dlen;
339         }
340         op[data_p++] = ISO2709_FS;
341
342         memcpy (op + entry_p, field->u.tag.tag, 3);
343         entry_p += 3;
344         memint (op + entry_p, data_p - data_0, p->length_data_entry);
345         entry_p += p->length_data_entry;
346         memint (op + entry_p, data_0 - base_address, p->length_starting);
347         entry_p += p->length_starting;
348         entry_p += p->length_implementation;
349     }
350     op[entry_p++] = ISO2709_FS;
351     assert (entry_p == base_address);
352     op[data_p++] = ISO2709_RS;
353     assert (data_p == len);
354     return len;
355 }
356
357 char *data1_nodetomarc(data1_handle dh, data1_marctab *p, data1_node *n,
358                        int selected, int *len)
359 {
360     int *size;
361     char **buf = data1_get_map_buf (dh, &size);
362
363     n = data1_get_root_tag (dh, n);
364     if (!n)
365         return 0;
366     *len = nodetomarc(dh, p, n, selected, buf, size);
367     return *buf;
368 }