1 /* $Id: util.cpp,v 1.15 2006-05-15 11:43:01 adam Exp $
2 Copyright (c) 2005-2006, Index Data.
10 #include <yaz/pquery.h>
11 #include <yaz/otherinfo.h>
12 #include <yaz/querytowrbuf.h> // for yaz_query_to_wrbuf()
17 namespace mp = metaproxy_1;
19 std::string mp::util::database_name_normalize(const std::string &s)
23 for (i = 0; i < r.length(); i++)
26 if (ch >= 'A' && ch <= 'Z')
27 r[i] = ch + 'a' - 'A';
33 void mp::util::piggyback(int smallSetUpperBound,
34 int largeSetLowerBound,
35 int mediumSetPresentNumber,
37 int &number_to_present)
39 // deal with piggyback
41 if (result_set_size < smallSetUpperBound)
43 // small set . Return all records in set
44 number_to_present = result_set_size;
46 else if (result_set_size > largeSetLowerBound)
48 // large set . Return no records
49 number_to_present = 0;
53 // medium set . Return mediumSetPresentNumber records
54 number_to_present = mediumSetPresentNumber;
55 if (number_to_present > result_set_size)
56 number_to_present = result_set_size;
61 bool mp::util::pqf(ODR odr, Z_APDU *apdu, const std::string &q) {
62 YAZ_PQF_Parser pqf_parser = yaz_pqf_create();
64 Z_RPNQuery *rpn = yaz_pqf_parse(pqf_parser, odr, q.c_str());
67 yaz_pqf_destroy(pqf_parser);
70 yaz_pqf_destroy(pqf_parser);
71 Z_Query *query = (Z_Query *) odr_malloc(odr, sizeof(Z_Query));
72 query->which = Z_Query_type_1;
73 query->u.type_1 = rpn;
75 apdu->u.searchRequest->query = query;
80 std::string mp::util::zQueryToString(Z_Query *query)
82 std::string query_str = "";
84 if (query && query->which == Z_Query_type_1){
85 Z_RPNQuery *rpn = query->u.type_1;
89 // allocate wrbuf (strings in YAZ!)
90 WRBUF w = wrbuf_alloc();
93 yaz_rpnquery_to_wrbuf(w, rpn);
95 // from w to std::string
96 query_str = std::string(wrbuf_buf(w), wrbuf_len(w));
104 if (query && query->which == Z_Query_type_1){
106 // allocate wrbuf (strings in YAZ!)
107 WRBUF w = wrbuf_alloc();
110 yaz_query_to_wrbuf(w, query);
112 // from w to std::string
113 query_str = std::string(wrbuf_buf(w), wrbuf_len(w));
122 void mp::util::get_default_diag(Z_DefaultDiagFormat *r,
123 int &error_code, std::string &addinfo)
125 error_code = *r->condition;
128 case Z_DefaultDiagFormat_v2Addinfo:
129 addinfo = std::string(r->u.v2Addinfo);
131 case Z_DefaultDiagFormat_v3Addinfo:
132 addinfo = r->u.v3Addinfo;
137 void mp::util::get_init_diagnostics(Z_InitResponse *initrs,
138 int &error_code, std::string &addinfo)
140 Z_External *uif = initrs->userInformationField;
142 if (uif && uif->which == Z_External_userInfo1)
144 Z_OtherInformation *ui = uif->u.userInfo1;
146 for (i = 0; i < ui->num_elements; i++)
148 Z_OtherInformationUnit *unit = ui->list[i];
149 if (unit->which == Z_OtherInfo_externallyDefinedInfo &&
150 unit->information.externallyDefinedInfo &&
151 unit->information.externallyDefinedInfo->which ==
154 Z_DiagnosticFormat *diag =
155 unit->information.externallyDefinedInfo->u.diag1;
159 Z_DiagnosticFormat_s *ds = diag->elements[0];
160 if (ds->which == Z_DiagnosticFormat_s_defaultDiagRec)
161 mp::util::get_default_diag(ds->u.defaultDiagRec,
162 error_code, addinfo);
169 int mp::util::get_vhost_otherinfo(Z_OtherInformation **otherInformation,
171 std::list<std::string> &vhosts)
174 for (cat = 1; ; cat++)
176 // check virtual host
178 yaz_oi_get_string_oidval(otherInformation,
180 cat /* categoryValue */,
181 remove_flag /* delete flag */);
184 vhosts.push_back(std::string(vhost));
190 void mp::util::set_vhost_otherinfo(Z_OtherInformation **otherInformation,
192 const std::list<std::string> &vhosts)
195 std::list<std::string>::const_iterator it = vhosts.begin();
196 for (cat = 1; it != vhosts.end() ; cat++, it++)
198 yaz_oi_set_string_oidval(otherInformation, odr,
199 VAL_PROXY, cat, it->c_str());
203 void mp::util::split_zurl(std::string zurl, std::string &host,
204 std::list<std::string> &db)
206 const char *zurl_cstr = zurl.c_str();
207 const char *sep = strchr(zurl_cstr, '/');
211 host = std::string(zurl_cstr, sep - zurl_cstr);
213 const char *cp1 = sep+1;
216 const char *cp2 = strchr(cp1, '+');
218 db.push_back(std::string(cp1, cp2 - cp1));
221 db.push_back(std::string(cp1));
233 bool mp::util::set_databases_from_zurl(ODR odr, std::string zurl,
234 int *db_num, char ***db_strings)
237 std::list<std::string> dblist;
239 split_zurl(zurl, host, dblist);
241 if (dblist.size() == 0)
243 *db_num = dblist.size();
244 *db_strings = (char **) odr_malloc(odr, sizeof(char*) * (*db_num));
246 std::list<std::string>::const_iterator it = dblist.begin();
247 for (int i = 0; it != dblist.end(); it++, i++)
248 (*db_strings)[i] = odr_strdup(odr, it->c_str());
252 mp::odr::odr(int type)
254 m_odr = odr_createmem(type);
259 m_odr = odr_createmem(ODR_ENCODE);
267 mp::odr::operator ODR() const
272 Z_APDU *mp::odr::create_close(Z_APDU *in_apdu,
273 int reason, const char *addinfo)
275 Z_APDU *apdu = create_APDU(Z_APDU_close, in_apdu);
277 *apdu->u.close->closeReason = reason;
279 apdu->u.close->diagnosticInformation = odr_strdup(m_odr, addinfo);
283 Z_APDU *mp::odr::create_APDU(int type, Z_APDU *in_apdu)
285 return mp::util::create_APDU(m_odr, type, in_apdu);
288 Z_APDU *mp::util::create_APDU(ODR odr, int type, Z_APDU *in_apdu)
290 Z_APDU *out_apdu = zget_APDU(odr, type);
292 Z_ReferenceId **id_to = mp::util::get_referenceId(out_apdu);
296 Z_ReferenceId **id_from = mp::util::get_referenceId(in_apdu);
297 if (id_from && *id_from && id_to)
299 *id_to = (Z_ReferenceId*) odr_malloc (odr, sizeof(**id_to));
300 (*id_to)->size = (*id_to)->len = (*id_from)->len;
301 (*id_to)->buf = (unsigned char*) odr_malloc(odr, (*id_to)->len);
302 memcpy((*id_to)->buf, (*id_from)->buf, (*id_to)->len);
310 Z_APDU *mp::odr::create_initResponse(Z_APDU *in_apdu,
311 int error, const char *addinfo)
313 Z_APDU *apdu = create_APDU(Z_APDU_initResponse, in_apdu);
316 apdu->u.initResponse->userInformationField =
317 zget_init_diagnostics(m_odr, error, addinfo);
318 *apdu->u.initResponse->result = 0;
323 Z_APDU *mp::odr::create_searchResponse(Z_APDU *in_apdu,
324 int error, const char *addinfo)
326 Z_APDU *apdu = create_APDU(Z_APDU_searchResponse, in_apdu);
329 Z_Records *rec = (Z_Records *) odr_malloc(m_odr, sizeof(Z_Records));
330 *apdu->u.searchResponse->searchStatus = 0;
331 apdu->u.searchResponse->records = rec;
332 rec->which = Z_Records_NSD;
333 rec->u.nonSurrogateDiagnostic =
334 zget_DefaultDiagFormat(m_odr, error, addinfo);
340 Z_APDU *mp::odr::create_presentResponse(Z_APDU *in_apdu,
341 int error, const char *addinfo)
343 Z_APDU *apdu = create_APDU(Z_APDU_presentResponse, in_apdu);
346 Z_Records *rec = (Z_Records *) odr_malloc(m_odr, sizeof(Z_Records));
347 apdu->u.presentResponse->records = rec;
349 rec->which = Z_Records_NSD;
350 rec->u.nonSurrogateDiagnostic =
351 zget_DefaultDiagFormat(m_odr, error, addinfo);
352 *apdu->u.presentResponse->presentStatus = Z_PresentStatus_failure;
357 Z_APDU *mp::odr::create_scanResponse(Z_APDU *in_apdu,
358 int error, const char *addinfo)
360 Z_APDU *apdu = create_APDU(Z_APDU_scanResponse, in_apdu);
361 Z_ScanResponse *res = apdu->u.scanResponse;
362 res->entries = (Z_ListEntries *) odr_malloc(m_odr, sizeof(*res->entries));
363 res->entries->num_entries = 0;
364 res->entries->entries = 0;
368 *res->scanStatus = Z_Scan_failure;
370 res->entries->num_nonsurrogateDiagnostics = 1;
371 res->entries->nonsurrogateDiagnostics = (Z_DiagRec **)
372 odr_malloc(m_odr, sizeof(Z_DiagRec *));
373 res->entries->nonsurrogateDiagnostics[0] =
374 zget_DiagRec(m_odr, error, addinfo);
378 res->entries->num_nonsurrogateDiagnostics = 0;
379 res->entries->nonsurrogateDiagnostics = 0;
384 Z_GDU *mp::odr::create_HTTP_Response(mp::Session &session,
385 Z_HTTP_Request *hreq, int code)
387 const char *response_version = "1.0";
388 bool keepalive = false;
389 if (!strcmp(hreq->version, "1.0"))
391 const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
392 if (v && !strcmp(v, "Keep-Alive"))
396 response_version = "1.0";
400 const char *v = z_HTTP_header_lookup(hreq->headers, "Connection");
401 if (v && !strcmp(v, "close"))
405 response_version = "1.1";
408 Z_GDU *gdu = z_get_HTTP_Response(m_odr, code);
409 Z_HTTP_Response *hres = gdu->u.HTTP_Response;
410 hres->version = odr_strdup(m_odr, response_version);
412 z_HTTP_header_add(m_odr, &hres->headers, "Connection", "Keep-Alive");
417 Z_ReferenceId **mp::util::get_referenceId(Z_APDU *apdu)
421 case Z_APDU_initRequest:
422 return &apdu->u.initRequest->referenceId;
423 case Z_APDU_initResponse:
424 return &apdu->u.initResponse->referenceId;
425 case Z_APDU_searchRequest:
426 return &apdu->u.searchRequest->referenceId;
427 case Z_APDU_searchResponse:
428 return &apdu->u.searchResponse->referenceId;
429 case Z_APDU_presentRequest:
430 return &apdu->u.presentRequest->referenceId;
431 case Z_APDU_presentResponse:
432 return &apdu->u.presentResponse->referenceId;
433 case Z_APDU_deleteResultSetRequest:
434 return &apdu->u.deleteResultSetRequest->referenceId;
435 case Z_APDU_deleteResultSetResponse:
436 return &apdu->u.deleteResultSetResponse->referenceId;
437 case Z_APDU_accessControlRequest:
438 return &apdu->u.accessControlRequest->referenceId;
439 case Z_APDU_accessControlResponse:
440 return &apdu->u.accessControlResponse->referenceId;
441 case Z_APDU_resourceControlRequest:
442 return &apdu->u.resourceControlRequest->referenceId;
443 case Z_APDU_resourceControlResponse:
444 return &apdu->u.resourceControlResponse->referenceId;
445 case Z_APDU_triggerResourceControlRequest:
446 return &apdu->u.triggerResourceControlRequest->referenceId;
447 case Z_APDU_resourceReportRequest:
448 return &apdu->u.resourceReportRequest->referenceId;
449 case Z_APDU_resourceReportResponse:
450 return &apdu->u.resourceReportResponse->referenceId;
451 case Z_APDU_scanRequest:
452 return &apdu->u.scanRequest->referenceId;
453 case Z_APDU_scanResponse:
454 return &apdu->u.scanResponse->referenceId;
455 case Z_APDU_sortRequest:
456 return &apdu->u.sortRequest->referenceId;
457 case Z_APDU_sortResponse:
458 return &apdu->u.sortResponse->referenceId;
459 case Z_APDU_segmentRequest:
460 return &apdu->u.segmentRequest->referenceId;
461 case Z_APDU_extendedServicesRequest:
462 return &apdu->u.extendedServicesRequest->referenceId;
463 case Z_APDU_extendedServicesResponse:
464 return &apdu->u.extendedServicesResponse->referenceId;
466 return &apdu->u.close->referenceId;
475 * indent-tabs-mode: nil
476 * c-file-style: "stroustrup"
478 * vim: shiftwidth=4 tabstop=8 expandtab