1 /* $Id: filter_backend_test.cpp,v 1.24 2007-04-13 09:57:51 adam Exp $
2 Copyright (c) 2005-2007, Index Data.
4 See the LICENSE file for details
10 #include "package.hpp"
12 #include "filter_backend_test.hpp"
19 #include <boost/thread/mutex.hpp>
23 #include <yaz/otherinfo.h>
24 #include <yaz/diagbib1.h>
25 #include <yaz/oid_db.h>
27 namespace mp = metaproxy_1;
28 namespace yf = mp::filter;
31 namespace metaproxy_1 {
36 class BackendTest::Rep {
37 friend class BackendTest;
40 ODR odr, Odr_oid *preferredRecordSyntax,
41 Z_ElementSetNames *esn,
42 int start, int number, int &error_code, std::string &addinfo,
43 int *number_returned, int *next_position);
45 bool m_support_named_result_sets;
47 session_map<Session_info> m_sessions;
53 static const int result_set_size = 42;
55 // an ISO2709 USMARC/MARC21 record that we return..
56 static const char *marc_record =
57 "\x30\x30\x33\x36\x36\x6E\x61\x6D\x20\x20\x32\x32\x30\x30\x31\x36"
58 "\x39\x38\x61\x20\x34\x35\x30\x30\x30\x30\x31\x30\x30\x31\x33\x30"
59 "\x30\x30\x30\x30\x30\x30\x33\x30\x30\x30\x34\x30\x30\x30\x31\x33"
60 "\x30\x30\x35\x30\x30\x31\x37\x30\x30\x30\x31\x37\x30\x30\x38\x30"
61 "\x30\x34\x31\x30\x30\x30\x33\x34\x30\x31\x30\x30\x30\x31\x37\x30"
62 "\x30\x31\x37\x39\x30\x34\x30\x30\x30\x31\x33\x30\x30\x30\x37\x35"
63 "\x30\x35\x30\x30\x30\x31\x32\x30\x30\x30\x38\x38\x31\x30\x30\x30"
64 "\x30\x31\x37\x30\x30\x31\x30\x30\x32\x34\x35\x30\x30\x33\x30\x30"
65 "\x30\x31\x31\x37\x32\x36\x30\x30\x30\x31\x32\x30\x30\x31\x34\x37"
66 "\x32\x36\x33\x30\x30\x30\x39\x30\x30\x31\x35\x39\x33\x30\x30\x30"
67 "\x30\x31\x31\x30\x30\x31\x36\x38\x1E\x20\x20\x20\x31\x31\x32\x32"
68 "\x34\x34\x36\x36\x20\x1E\x44\x4C\x43\x1E\x30\x30\x30\x30\x30\x30"
69 "\x30\x30\x30\x30\x30\x30\x30\x30\x2E\x30\x1E\x39\x31\x30\x37\x31"
70 "\x30\x63\x31\x39\x39\x31\x30\x37\x30\x31\x6E\x6A\x75\x20\x20\x20"
71 "\x20\x20\x20\x20\x20\x20\x20\x20\x30\x30\x30\x31\x30\x20\x65\x6E"
72 "\x67\x20\x20\x1E\x20\x20\x1F\x61\x44\x4C\x43\x1F\x63\x44\x4C\x43"
73 "\x1E\x30\x30\x1F\x61\x31\x32\x33\x2D\x78\x79\x7A\x1E\x31\x30\x1F"
74 "\x61\x4A\x61\x63\x6B\x20\x43\x6F\x6C\x6C\x69\x6E\x73\x1E\x31\x30"
75 "\x1F\x61\x48\x6F\x77\x20\x74\x6F\x20\x70\x72\x6F\x67\x72\x61\x6D"
76 "\x20\x61\x20\x63\x6F\x6D\x70\x75\x74\x65\x72\x1E\x31\x20\x1F\x61"
77 "\x50\x65\x6E\x67\x75\x69\x6E\x1E\x20\x20\x1F\x61\x38\x37\x31\x30"
78 "\x1E\x20\x20\x1F\x61\x70\x2E\x20\x63\x6D\x2E\x1E\x20\x20\x1F\x61"
79 "\x20\x20\x20\x31\x31\x32\x32\x34\x34\x36\x36\x20\x1E\x1D";
82 yf::BackendTest::BackendTest() : m_p(new BackendTest::Rep) {
83 m_p->m_support_named_result_sets = false;
86 yf::BackendTest::~BackendTest() {
89 Z_Records *yf::BackendTest::Rep::fetch(
90 ODR odr, Odr_oid *preferredRecordSyntax,
91 Z_ElementSetNames *esn,
92 int start, int number, int &error_code, std::string &addinfo,
93 int *number_returned, int *next_position)
95 const char *element_set_name = "F"; // default to use
97 if (number + start - 1 > result_set_size || start < 1)
99 error_code = YAZ_BIB1_PRESENT_REQUEST_OUT_OF_RANGE;
103 const char *name_oid = OID_STR_USMARC; // default if syntax is given
104 if (preferredRecordSyntax)
107 yaz_oid_to_string(yaz_oid_std(), preferredRecordSyntax, 0);
108 if (name_oid && !strcmp(name_oid, OID_STR_USMARC))
110 else if (name_oid && !strcmp(name_oid, OID_STR_XML))
114 error_code = YAZ_BIB1_RECORD_SYNTAX_UNSUPP;
119 // no element set, "B" and "F" are supported
122 if (esn->which != Z_ElementSetNames_generic)
125 = YAZ_BIB1_SPECIFIED_ELEMENT_SET_NAME_NOT_VALID_FOR_SPECIFIED_;
128 element_set_name = esn->u.generic;
130 if (!strcmp(element_set_name, "B")
131 && !strcmp(name_oid, OID_STR_USMARC))
133 else if (!strcmp(element_set_name, "F")
134 && !strcmp(name_oid, OID_STR_USMARC))
136 else if (!strncmp(element_set_name, "FF", 2)
137 && !strcmp(name_oid, OID_STR_XML))
138 ; // Huge XML test record
142 = YAZ_BIB1_SPECIFIED_ELEMENT_SET_NAME_NOT_VALID_FOR_SPECIFIED_;
143 addinfo = std::string(element_set_name);
146 Z_Records *rec = (Z_Records *) odr_malloc(odr, sizeof(Z_Records));
147 rec->which = Z_Records_DBOSD;
148 rec->u.databaseOrSurDiagnostics = (Z_NamePlusRecordList *)
149 odr_malloc(odr, sizeof(Z_NamePlusRecordList));
150 rec->u.databaseOrSurDiagnostics->num_records = number;
151 rec->u.databaseOrSurDiagnostics->records = (Z_NamePlusRecord **)
152 odr_malloc(odr, sizeof(Z_NamePlusRecord *) * number);
154 for (i = 0; i<number; i++)
156 rec->u.databaseOrSurDiagnostics->records[i] = (Z_NamePlusRecord *)
157 odr_malloc(odr, sizeof(Z_NamePlusRecord));
158 Z_NamePlusRecord *npr = rec->u.databaseOrSurDiagnostics->records[i];
159 npr->databaseName = 0;
160 npr->which = Z_NamePlusRecord_databaseRecord;
162 if (!strncmp(element_set_name, "FF", 2))
163 { // Huge XML test record
165 if (element_set_name[2])
166 sz = atoi(element_set_name+2) * 1024;
169 char *tmp_rec = (char*) xmalloc(sz);
171 memset(tmp_rec, 'a', sz);
172 memcpy(tmp_rec, "<a>", 3);
173 memcpy(tmp_rec + sz - 4, "</a>", 4);
175 npr->u.databaseRecord = z_ext_record_xml(odr, tmp_rec, sz);
180 char *tmp_rec = odr_strdup(odr, marc_record);
182 sprintf(offset_str, "test__%09d_", i+start);
183 memcpy(tmp_rec+186, offset_str, strlen(offset_str));
184 npr->u.databaseRecord = z_ext_record_usmarc(
185 odr, tmp_rec, strlen(tmp_rec));
189 *number_returned = number;
190 if (start + number > result_set_size)
193 *next_position = start + number;
197 void yf::BackendTest::process(Package &package) const
199 Z_GDU *gdu = package.request().get();
201 if (!gdu || gdu->which != Z_GDU_Z3950)
205 Z_APDU *apdu_req = gdu->u.z3950;
206 Z_APDU *apdu_res = 0;
209 if (apdu_req->which != Z_APDU_initRequest &&
210 !m_p->m_sessions.exist(package.session()))
212 apdu_res = odr.create_close(apdu_req,
213 Z_Close_protocolError,
214 "no init for filter_backend_test");
215 package.session().close();
217 else if (apdu_req->which == Z_APDU_initRequest)
219 apdu_res = odr.create_initResponse(apdu_req, 0, 0);
220 Z_InitRequest *req = apdu_req->u.initRequest;
221 Z_InitResponse *resp = apdu_res->u.initResponse;
223 resp->implementationName = "backend_test";
224 if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
225 m_p->m_support_named_result_sets = true;
228 static const int masks[] = {
229 Z_Options_search, Z_Options_present,
230 Z_Options_namedResultSets, -1
232 for (i = 0; masks[i] != -1; i++)
233 if (ODR_MASK_GET(req->options, masks[i]))
234 ODR_MASK_SET(resp->options, masks[i]);
235 static const int versions[] = {
241 for (i = 0; versions[i] != -1; i++)
242 if (ODR_MASK_GET(req->protocolVersion, versions[i]))
243 ODR_MASK_SET(resp->protocolVersion, versions[i]);
248 m_p->m_sessions.create(info, package.session());
250 else if (apdu_req->which == Z_APDU_searchRequest)
252 Z_SearchRequest *req = apdu_req->u.searchRequest;
254 if (!m_p->m_support_named_result_sets &&
255 strcmp(req->resultSetName, "default"))
258 odr.create_searchResponse(
259 apdu_req, YAZ_BIB1_RESULT_SET_NAMING_UNSUPP, 0);
263 Z_Records *records = 0;
264 int number_returned = 0;
265 int next_position = 0;
270 mp::util::piggyback(*req->smallSetUpperBound,
271 *req->largeSetLowerBound,
272 *req->mediumSetPresentNumber,
277 { // not a large set for sure
278 Z_ElementSetNames *esn;
279 if (number > *req->smallSetUpperBound)
280 esn = req->mediumSetElementSetNames;
282 esn = req->smallSetElementSetNames;
283 records = m_p->fetch(
284 odr, req->preferredRecordSyntax, esn,
293 odr.create_searchResponse(
294 apdu_req, error_code, addinfo.c_str());
295 Z_SearchResponse *resp = apdu_res->u.searchResponse;
296 *resp->resultCount = result_set_size;
301 odr.create_searchResponse(apdu_req, 0, 0);
302 Z_SearchResponse *resp = apdu_res->u.searchResponse;
303 *resp->resultCount = result_set_size;
304 *resp->numberOfRecordsReturned = number_returned;
305 *resp->nextResultSetPosition = next_position;
306 resp->records = records;
310 else if (apdu_req->which == Z_APDU_presentRequest)
312 Z_PresentRequest *req = apdu_req->u.presentRequest;
313 int number_returned = 0;
314 int next_position = 0;
317 Z_ElementSetNames *esn = 0;
319 if (req->recordComposition)
321 if (req->recordComposition->which == Z_RecordComp_simple)
322 esn = req->recordComposition->u.simple;
326 odr.create_presentResponse(
328 YAZ_BIB1_ONLY_A_SINGLE_ELEMENT_SET_NAME_SUPPORTED,
330 package.response() = apdu_res;
334 Z_Records *records = m_p->fetch(
335 odr, req->preferredRecordSyntax, esn,
336 *req->resultSetStartPoint, *req->numberOfRecordsRequested,
344 odr.create_presentResponse(apdu_req, error_code,
350 odr.create_presentResponse(apdu_req, 0, 0);
351 Z_PresentResponse *resp = apdu_res->u.presentResponse;
352 resp->records = records;
353 *resp->numberOfRecordsReturned = number_returned;
354 *resp->nextResultSetPosition = next_position;
359 apdu_res = odr.create_close(apdu_req,
360 Z_Close_protocolError,
361 "backend_test: unhandled APDU");
362 package.session().close();
365 package.response() = apdu_res;
367 if (package.session().is_closed())
368 m_p->m_sessions.release(package.session());
371 static mp::filter::Base* filter_creator()
373 return new mp::filter::BackendTest;
377 struct metaproxy_1_filter_struct metaproxy_1_filter_backend_test = {
388 * indent-tabs-mode: nil
389 * c-file-style: "stroustrup"
391 * vim: shiftwidth=4 tabstop=8 expandtab