1 /* This file is part of the yazpp toolkit.
2 * Copyright (C) Index Data and Mike Taylor
3 * See the file LICENSE for details.
12 #include <yazpp/ir-assoc.h>
13 #include <yaz/oid_db.h>
15 using namespace yazpp_1;
17 IR_Assoc::IR_Assoc(IPDU_Observable *the_PDU_Observable)
18 : Z_Assoc(the_PDU_Observable)
20 m_num_databaseNames = 0;
22 m_preferredRecordSyntax = 0;
23 m_elementSetNames = 0;
29 const char *db = "Default";
30 set_databaseNames(1, &db);
35 xfree(m_preferredRecordSyntax);
36 if (m_elementSetNames)
37 delete [] m_elementSetNames->u.generic;
38 delete [] m_elementSetNames;
44 void IR_Assoc::get_databaseNames (int *num, char ***list)
46 *num = m_num_databaseNames;
47 *list = m_databaseNames;
51 void IR_Assoc::set_databaseNames (int num, const char **list)
54 yaz_log (m_log, "IR_Assoc::set_databaseNames num=%d", num);
55 for (i = 0; i<m_num_databaseNames; i++)
56 delete [] m_databaseNames[i];
57 delete [] m_databaseNames;
58 m_num_databaseNames = num;
60 m_databaseNames = new char *[num];
61 for (i = 0; i<m_num_databaseNames; i++)
63 m_databaseNames[i] = new char[strlen(list[i])+1];
64 strcpy(m_databaseNames[i], list[i]);
68 void IR_Assoc::set_databaseNames(const char *dblist, const char *sep)
70 const char **list = new const char* [strlen(dblist)];
71 char *dbtmp = new char[strlen(dblist)+1];
72 strcpy(dbtmp, dblist);
75 for (char *cp = dbtmp; ; cp++)
76 if (*cp && !strchr(sep, *cp))
90 set_databaseNames (num, list);
95 void IR_Assoc::set_preferredRecordSyntax (const char *syntax)
97 xfree(m_preferredRecordSyntax);
98 m_preferredRecordSyntax = 0;
99 if (syntax && *syntax)
100 m_preferredRecordSyntax = xstrdup(syntax);
103 void IR_Assoc::get_preferredRecordSyntax (const char **dst)
105 if (m_preferredRecordSyntax)
106 *dst = m_preferredRecordSyntax;
111 void IR_Assoc::set_elementSetName (const char *elementSetName)
113 if (m_elementSetNames)
114 delete [] m_elementSetNames->u.generic;
115 delete m_elementSetNames;
116 m_elementSetNames = 0;
117 if (elementSetName && *elementSetName)
119 m_elementSetNames = new Z_ElementSetNames;
120 m_elementSetNames->which = Z_ElementSetNames_generic;
121 m_elementSetNames->u.generic = new char[strlen(elementSetName)+1];
122 strcpy (m_elementSetNames->u.generic, elementSetName);
126 void IR_Assoc::get_elementSetName (Z_ElementSetNames **elementSetNames)
128 *elementSetNames = m_elementSetNames;
131 void IR_Assoc::get_elementSetName (const char **elementSetName)
133 if (!m_elementSetNames ||
134 m_elementSetNames->which != Z_ElementSetNames_generic)
139 *elementSetName = m_elementSetNames->u.generic;
143 void IR_Assoc::recv_GDU(Z_GDU *apdu, int len)
145 if (apdu->which == Z_GDU_Z3950)
146 recv_Z_PDU(apdu->u.z3950, len);
149 void IR_Assoc::recv_Z_PDU(Z_APDU *apdu, int len)
151 yaz_log (m_log, "recv_Z_PDU %d bytes", len);
152 m_lastReceived = apdu->which;
155 case Z_APDU_initResponse:
156 yaz_log (m_log, "recv InitResponse");
157 recv_initResponse(apdu->u.initResponse);
159 case Z_APDU_initRequest:
160 yaz_log (m_log, "recv InitRequest");
161 recv_initRequest(apdu->u.initRequest);
163 case Z_APDU_searchRequest:
164 yaz_log (m_log, "recv searchRequest");
165 recv_searchRequest(apdu->u.searchRequest);
167 case Z_APDU_searchResponse:
168 yaz_log (m_log, "recv searchResponse");
169 recv_searchResponse(apdu->u.searchResponse);
171 case Z_APDU_presentRequest:
172 yaz_log (m_log, "recv presentRequest");
173 recv_presentRequest(apdu->u.presentRequest);
175 case Z_APDU_presentResponse:
176 yaz_log (m_log, "recv presentResponse");
177 recv_presentResponse(apdu->u.presentResponse);
179 case Z_APDU_extendedServicesResponse:
180 yaz_log (m_log, "recv extendedServiceResponse");
181 recv_extendedServicesResponse(apdu->u.extendedServicesResponse);
186 int IR_Assoc::send_searchRequest(Yaz_Z_Query *query,
190 Z_APDU *apdu = create_Z_PDU(Z_APDU_searchRequest);
191 Z_SearchRequest *req = apdu->u.searchRequest;
193 req->query = query->get_Z_Query();
196 get_databaseNames (&req->num_databaseNames, &req->databaseNames);
197 const char *recordSyntax;
198 get_preferredRecordSyntax(&recordSyntax);
199 if (recordSyntax && *recordSyntax)
201 req->preferredRecordSyntax
202 = yaz_string_to_oid_odr(yaz_oid_std(), CLASS_RECSYN, recordSyntax,
205 yaz_log (m_log, "send_searchRequest");
206 assert (req->otherInfo == 0);
209 set_otherInformationString(&req->otherInfo, yaz_oid_userinfo_cookie,
211 assert (req->otherInfo);
216 req->referenceId = getRefID(pRefId);
221 req->resultSetName = pResultSetId;
224 return send_Z_PDU(apdu, 0);
227 int IR_Assoc::send_presentRequest(Odr_int start,
232 Z_APDU *apdu = create_Z_PDU(Z_APDU_presentRequest);
233 Z_PresentRequest *req = apdu->u.presentRequest;
235 req->resultSetStartPoint = &start;
236 req->numberOfRecordsRequested = &number;
238 const char *recordSyntax;
239 get_preferredRecordSyntax (&recordSyntax);
240 if (recordSyntax && *recordSyntax)
242 req->preferredRecordSyntax =
243 yaz_string_to_oid_odr(yaz_oid_std(), CLASS_RECSYN, recordSyntax,
246 Z_RecordComposition compo;
247 Z_ElementSetNames *elementSetNames;
248 get_elementSetName (&elementSetNames);
251 req->recordComposition = &compo;
252 compo.which = Z_RecordComp_simple;
253 compo.u.simple = elementSetNames;
257 set_otherInformationString(&req->otherInfo, yaz_oid_userinfo_cookie,
262 req->referenceId = getRefID(pRefId);
267 req->resultSetId = pResultSetId;
270 return send_Z_PDU(apdu, 0);
273 void IR_Assoc::set_proxy(const char *str)
279 m_proxy = new char[strlen(str)+1];
280 strcpy (m_proxy, str);
284 void IR_Assoc::set_cookie(const char *str)
290 m_cookie = new char[strlen(str)+1];
291 strcpy(m_cookie, str);
295 const char *IR_Assoc::get_cookie()
300 void IR_Assoc::client(const char *addr)
303 m_host = new char[strlen(addr)+1];
304 strcpy(m_host, addr);
305 const char *dbpart = strchr(m_host, '/');
307 set_databaseNames (dbpart+1, "+ ");
308 Z_Assoc::client(m_proxy ? m_proxy : m_host);
311 const char *IR_Assoc::get_proxy()
316 const char *IR_Assoc::get_host()
321 void IR_Assoc::recv_searchRequest(Z_SearchRequest *searchRequest)
323 Z_APDU *apdu = create_Z_PDU(Z_APDU_searchResponse);
327 void IR_Assoc::recv_presentRequest(Z_PresentRequest *presentRequest)
329 Z_APDU *apdu = create_Z_PDU(Z_APDU_presentResponse);
333 void IR_Assoc::recv_initRequest(Z_InitRequest *initRequest)
335 Z_APDU *apdu = create_Z_PDU(Z_APDU_initResponse);
339 void IR_Assoc::recv_searchResponse (Z_SearchResponse *searchResponse)
343 void IR_Assoc::recv_presentResponse (Z_PresentResponse *presentResponse)
347 void IR_Assoc::recv_initResponse(Z_InitResponse *initResponse)
351 void IR_Assoc::recv_extendedServicesResponse(Z_ExtendedServicesResponse *ExtendedServicesResponse)
355 int IR_Assoc::get_lastReceived()
357 return m_lastReceived;
360 void IR_Assoc::set_lastReceived(int lastReceived)
362 m_lastReceived = lastReceived;
365 int IR_Assoc::send_initRequest(char* pRefId)
367 Z_APDU *apdu = create_Z_PDU(Z_APDU_initRequest);
368 Z_InitRequest *req = apdu->u.initRequest;
370 ODR_MASK_SET(req->options, Z_Options_search);
371 ODR_MASK_SET(req->options, Z_Options_present);
372 ODR_MASK_SET(req->options, Z_Options_namedResultSets);
373 ODR_MASK_SET(req->options, Z_Options_triggerResourceCtrl);
374 ODR_MASK_SET(req->options, Z_Options_scan);
375 ODR_MASK_SET(req->options, Z_Options_sort);
376 ODR_MASK_SET(req->options, Z_Options_extendedServices);
377 ODR_MASK_SET(req->options, Z_Options_delSet);
379 ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_1);
380 ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_2);
381 ODR_MASK_SET(req->protocolVersion, Z_ProtocolVersion_3);
385 req->referenceId = getRefID(pRefId);
388 if (m_proxy && m_host)
389 set_otherInformationString(&req->otherInfo, yaz_oid_userinfo_proxy,
392 set_otherInformationString(&req->otherInfo, yaz_oid_userinfo_cookie,
394 return send_Z_PDU(apdu, 0);
397 int IR_Assoc::send_deleteResultSetRequest(char* pResultSetId, char* pRefId)
399 char* ResultSetIds[1];
401 Z_APDU *apdu = create_Z_PDU(Z_APDU_deleteResultSetRequest);
402 Z_DeleteResultSetRequest *req = apdu->u.deleteResultSetRequest;
406 *req->deleteFunction = Z_DeleteResultSetRequest_list;
407 req->num_resultSetList = 1;
408 ResultSetIds[0] = pResultSetId;
409 req->resultSetList = ResultSetIds;
413 *req->deleteFunction = Z_DeleteResultSetRequest_all;
418 req->referenceId = getRefID(pRefId);
421 if (m_proxy && m_host)
422 set_otherInformationString(&req->otherInfo, yaz_oid_userinfo_proxy,
425 set_otherInformationString(&req->otherInfo, yaz_oid_userinfo_cookie,
428 return send_Z_PDU(apdu, 0);
435 * c-file-style: "Stroustrup"
436 * indent-tabs-mode: nil
438 * vim: shiftwidth=4 tabstop=8 expandtab