1 /* $Id: retrieve.c,v 1.21 2004-05-27 09:28:01 adam Exp $
2 Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003,2004
5 This file is part of the Zebra server.
7 Zebra is free software; you can redistribute it and/or modify it under
8 the terms of the GNU General Public License as published by the Free
9 Software Foundation; either version 2, or (at your option) any later
12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
17 You should have received a copy of the GNU General Public License
18 along with Zebra; see the file LICENSE.zebra. If not, write to the
19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
37 int zebra_record_ext_read (void *fh, char *buf, size_t count)
39 struct zebra_fetch_control *fc = (struct zebra_fetch_control *) fh;
40 return read (fc->fd, buf, count);
43 off_t zebra_record_ext_seek (void *fh, off_t offset)
45 struct zebra_fetch_control *fc = (struct zebra_fetch_control *) fh;
46 return lseek (fc->fd, offset + fc->record_offset, SEEK_SET);
49 off_t zebra_record_ext_tell (void *fh)
51 struct zebra_fetch_control *fc = (struct zebra_fetch_control *) fh;
52 return lseek (fc->fd, 0, SEEK_CUR) - fc->record_offset;
55 off_t zebra_record_int_seek (void *fh, off_t offset)
57 struct zebra_fetch_control *fc = (struct zebra_fetch_control *) fh;
58 return (off_t) (fc->record_int_pos = offset);
61 off_t zebra_record_int_tell (void *fh)
63 struct zebra_fetch_control *fc = (struct zebra_fetch_control *) fh;
64 return (off_t) fc->record_int_pos;
67 int zebra_record_int_read (void *fh, char *buf, size_t count)
69 struct zebra_fetch_control *fc = (struct zebra_fetch_control *) fh;
70 int l = fc->record_int_len - fc->record_int_pos;
73 l = (l < (int) count) ? l : count;
74 memcpy (buf, fc->record_int_buf + fc->record_int_pos, l);
75 fc->record_int_pos += l;
79 void zebra_record_int_end (void *fh, off_t off)
81 struct zebra_fetch_control *fc = (struct zebra_fetch_control *) fh;
85 int zebra_record_fetch (ZebraHandle zh, int sysno, int score, ODR stream,
86 oid_value input_format, Z_RecordComposition *comp,
87 oid_value *output_format, char **rec_bufp,
88 int *rec_lenp, char **basenamep)
91 char *fname, *file_type, *basename;
93 struct recRetrieveCtrl retrieveCtrl;
95 struct zebra_fetch_control fc;
96 RecordAttr *recordAttr;
100 rec = rec_get (zh->reg->records, sysno);
103 logf (LOG_DEBUG, "rec_get fail on sysno=%d", sysno);
107 recordAttr = rec_init_attr (zh->reg->zei, rec);
109 file_type = rec->info[recInfo_fileType];
110 fname = rec->info[recInfo_filename];
111 basename = rec->info[recInfo_databaseName];
112 *basenamep = (char *) odr_malloc (stream, strlen(basename)+1);
113 strcpy (*basenamep, basename);
115 if (comp && comp->which == Z_RecordComp_simple &&
116 comp->u.simple->which == Z_ElementSetNames_generic)
118 if (!strcmp (comp->u.simple->u.generic, "R"))
121 if (!(rt = recType_byName (zh->reg->recTypes,
122 file_type, subType, &clientData)))
124 logf (LOG_WARN, "Retrieve: Cannot handle type %s", file_type);
127 logf (LOG_DEBUG, "retrieve localno=%d score=%d", sysno, score);
128 retrieveCtrl.fh = &fc;
130 retrieveCtrl.fname = fname;
131 if (rec->size[recInfo_storeData] > 0)
133 retrieveCtrl.readf = zebra_record_int_read;
134 retrieveCtrl.seekf = zebra_record_int_seek;
135 retrieveCtrl.tellf = zebra_record_int_tell;
136 fc.record_int_len = rec->size[recInfo_storeData];
137 fc.record_int_buf = rec->info[recInfo_storeData];
138 fc.record_int_pos = 0;
139 logf (LOG_DEBUG, "Internal retrieve. %d bytes", fc.record_int_len);
142 *output_format = VAL_SUTRS;
143 *rec_lenp = rec->size[recInfo_storeData];
144 *rec_bufp = (char *) odr_malloc(stream, *rec_lenp);
145 memcpy(*rec_bufp, rec->info[recInfo_storeData], *rec_lenp);
154 if (zh->path_reg && !yaz_is_abspath (fname))
156 strcpy (full_rep, zh->path_reg);
157 strcat (full_rep, "/");
158 strcat (full_rep, fname);
161 strcpy (full_rep, fname);
163 if ((fc.fd = open (full_rep, O_BINARY|O_RDONLY)) == -1)
165 logf (LOG_WARN|LOG_ERRNO, "Retrieve fail; missing file: %s",
170 fc.record_offset = recordAttr->recordOffset;
172 retrieveCtrl.readf = zebra_record_ext_read;
173 retrieveCtrl.seekf = zebra_record_ext_seek;
174 retrieveCtrl.tellf = zebra_record_ext_tell;
176 zebra_record_ext_seek (retrieveCtrl.fh, 0);
179 *output_format = VAL_SUTRS;
180 *rec_lenp = recordAttr->recordSize;
181 *rec_bufp = (char *) odr_malloc(stream, *rec_lenp);
182 zebra_record_ext_read(&fc, *rec_bufp, *rec_lenp);
188 retrieveCtrl.subType = subType;
189 retrieveCtrl.localno = sysno;
190 retrieveCtrl.score = score;
191 retrieveCtrl.recordSize = recordAttr->recordSize;
192 retrieveCtrl.odr = stream;
193 retrieveCtrl.input_format = retrieveCtrl.output_format = input_format;
194 retrieveCtrl.comp = comp;
195 retrieveCtrl.encoding = zh->record_encoding;
196 retrieveCtrl.diagnostic = 0;
197 retrieveCtrl.dh = zh->reg->dh;
198 retrieveCtrl.res = zh->res;
199 retrieveCtrl.rec_buf = 0;
200 retrieveCtrl.rec_len = -1;
202 (*rt->retrieve)(clientData, &retrieveCtrl);
203 *output_format = retrieveCtrl.output_format;
204 *rec_bufp = (char *) retrieveCtrl.rec_buf;
205 *rec_lenp = retrieveCtrl.rec_len;
210 return retrieveCtrl.diagnostic;