X-Git-Url: http://git.indexdata.com/?p=yazpp-moved-to-github.git;a=blobdiff_plain;f=src%2Fyaz-z-server.cpp;h=ae7e94ad3c08b4004b4f5cbc447da2f196457657;hp=38fedb46187c25dcf29722ed5e8089798c2d730a;hb=ffc71f400dd4cdb8e631186f8e31a93fc641abd7;hpb=86433d4fcd24b068f362aecf0b1d1666d32ac6a1 diff --git a/src/yaz-z-server.cpp b/src/yaz-z-server.cpp index 38fedb4..ae7e94a 100644 --- a/src/yaz-z-server.cpp +++ b/src/yaz-z-server.cpp @@ -1,9 +1,30 @@ /* - * Copyright (c) 2000, Index Data. + * Copyright (c) 2000-2001, Index Data. * See the file LICENSE for details. * * $Log: yaz-z-server.cpp,v $ - * Revision 1.5 2000-10-24 12:29:57 adam + * Revision 1.12 2001-04-25 19:40:18 adam + * Added refernceId handling for other services. + * + * Revision 1.11 2001/04/12 15:12:10 heikki + * minor ursula stuff + * + * Revision 1.10 2001/04/04 14:02:49 adam + * URSULA / Z-ruth service. + * + * Revision 1.9 2001/03/29 15:14:26 adam + * Minor updates. + * + * Revision 1.8 2001/03/27 14:47:45 adam + * New server facility scheme. + * + * Revision 1.7 2001/03/26 14:43:49 adam + * New threaded PDU association. + * + * Revision 1.6 2001/01/29 11:18:24 adam + * Server sets OPTIONS search and present. + * + * Revision 1.5 2000/10/24 12:29:57 adam * Fixed bug in proxy where a Yaz_ProxyClient could be owned by * two Yaz_Proxy's (fatal). * @@ -25,36 +46,122 @@ #include #include - Yaz_Z_Server::Yaz_Z_Server(IYaz_PDU_Observable *the_PDU_Observable) : Yaz_Z_Assoc(the_PDU_Observable) { + m_facilities = 0; +} + +Yaz_Z_Server::~Yaz_Z_Server() +{ + facility_reset(); +} + +void Yaz_Z_Server::facility_reset () +{ + Yaz_Z_Server_Facility_Info *p = m_facilities; + while (p) + { + Yaz_Z_Server_Facility_Info *p_next = p->m_next; + + delete [] p->m_name; + delete p; + p = p_next; + } + m_facilities = 0; +} + +void Yaz_Z_Server::facility_add(IYaz_Server_Facility *facility, + const char *name) +{ + Yaz_Z_Server_Facility_Info **p = &m_facilities; + while (*p) + p = &(*p)->m_next; + + *p = new Yaz_Z_Server_Facility_Info; + + (*p)->m_next = 0; + (*p)->m_name = new char [strlen(name)+1]; + strcpy ((*p)->m_name, name); + (*p)->m_facility = facility; +} + +void Yaz_Z_Server::recv_Z_PDU (Z_APDU *apdu_request) +{ + Yaz_Z_Server_Facility_Info *f = m_facilities; + + if (apdu_request->which == Z_APDU_initRequest) + { + Z_APDU *apdu_response = create_Z_PDU(Z_APDU_initResponse); + + Z_InitRequest *req = apdu_request->u.initRequest; + Z_InitResponse *resp = apdu_response->u.initResponse; + + if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_1)) + { + ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_1); + } + if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_2)) + { + ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_2); + } + if (ODR_MASK_GET(req->protocolVersion, Z_ProtocolVersion_3)) + { + ODR_MASK_SET(resp->protocolVersion, Z_ProtocolVersion_3); + } + while (f) + { + f->m_facility->init(this, req, resp); + f = f->m_next; + } + transfer_referenceId(apdu_request, apdu_response); + send_Z_PDU(apdu_response); + } + else + { + f = m_facilities; + int taken = 0; + while (f) + { + taken = f->m_facility->recv(this, apdu_request); + if (taken) + break; + f = f->m_next; + } + if (!taken) + { + yaz_log (LOG_LOG, "unhandled request = %d", apdu_request->which); + delete this; + } + } } /* * database record. */ -void Yaz_Z_Server::create_databaseRecord ( - Z_NamePlusRecord *rec, const char *dbname, int format, const void *buf, int len) +void Yaz_Z_ServerUtility::create_databaseRecord ( + ODR odr, Z_NamePlusRecord *rec, const char *dbname, int format, + const void *buf, int len) { - rec->databaseName = dbname ? odr_strdup (odr_encode(), dbname) : 0; + rec->databaseName = dbname ? odr_strdup (odr, dbname) : 0; rec->which = Z_NamePlusRecord_databaseRecord; - rec->u.databaseRecord = z_ext_record (odr_encode(), format, + rec->u.databaseRecord = z_ext_record (odr, format, (const char *) buf, len); } /* * surrogate diagnostic. */ -void Yaz_Z_Server::create_surrogateDiagnostics( - Z_NamePlusRecord *rec, const char *dbname, int error, char *const addinfo) +void Yaz_Z_ServerUtility::create_surrogateDiagnostics( + ODR odr, Z_NamePlusRecord *rec, const char *dbname, + int error, char *const addinfo) { int oid[OID_SIZE]; - int *err = (int *)odr_malloc (odr_encode(), sizeof(*err)); + int *err = (int *)odr_malloc (odr, sizeof(*err)); oident bib1; - Z_DiagRec *drec = (Z_DiagRec *)odr_malloc (odr_encode(), sizeof(*drec)); + Z_DiagRec *drec = (Z_DiagRec *)odr_malloc (odr, sizeof(*drec)); Z_DefaultDiagFormat *dr = (Z_DefaultDiagFormat *) - odr_malloc (odr_encode(), sizeof(*dr)); + odr_malloc (odr, sizeof(*dr)); bib1.proto = PROTO_Z3950; bib1.oclass = CLASS_DIAGSET; @@ -62,31 +169,31 @@ void Yaz_Z_Server::create_surrogateDiagnostics( yaz_log(LOG_DEBUG, "SurrogateDiagnotic: %d -- %s", error, addinfo); *err = error; - rec->databaseName = dbname ? odr_strdup (odr_encode(), dbname) : 0; + rec->databaseName = dbname ? odr_strdup (odr, dbname) : 0; rec->which = Z_NamePlusRecord_surrogateDiagnostic; rec->u.surrogateDiagnostic = drec; drec->which = Z_DiagRec_defaultFormat; drec->u.defaultFormat = dr; - dr->diagnosticSetId = odr_oiddup (odr_encode(), + dr->diagnosticSetId = odr_oiddup (odr, oid_ent_to_oid(&bib1, oid)); dr->condition = err; dr->which = Z_DefaultDiagFormat_v2Addinfo; - dr->u.v2Addinfo = odr_strdup (odr_encode(), addinfo ? addinfo : ""); + dr->u.v2Addinfo = odr_strdup (odr, addinfo ? addinfo : ""); } -Z_Records *Yaz_Z_Server::create_nonSurrogateDiagnostics ( - int error, const char *addinfo) +Z_Records *Yaz_Z_ServerUtility::create_nonSurrogateDiagnostics ( + ODR odr, int error, const char *addinfo) { int oid[OID_SIZE]; Z_Records *rec = (Z_Records *) - odr_malloc (odr_encode(), sizeof(*rec)); + odr_malloc (odr, sizeof(*rec)); oident bib1; int *err = (int *) - odr_malloc (odr_encode(), sizeof(*err)); + odr_malloc (odr, sizeof(*err)); Z_DiagRec *drec = (Z_DiagRec *) - odr_malloc (odr_encode(), sizeof(*drec)); + odr_malloc (odr, sizeof(*drec)); Z_DefaultDiagFormat *dr = (Z_DefaultDiagFormat *) - odr_malloc (odr_encode(), sizeof(*dr)); + odr_malloc (odr, sizeof(*dr)); bib1.proto = PROTO_Z3950; bib1.oclass = CLASS_DIAGSET; @@ -96,225 +203,9 @@ Z_Records *Yaz_Z_Server::create_nonSurrogateDiagnostics ( rec->which = Z_Records_NSD; rec->u.nonSurrogateDiagnostic = dr; dr->diagnosticSetId = - odr_oiddup (odr_encode(), oid_ent_to_oid(&bib1, oid)); + odr_oiddup (odr, oid_ent_to_oid(&bib1, oid)); dr->condition = err; dr->which = Z_DefaultDiagFormat_v2Addinfo; - dr->u.v2Addinfo = odr_strdup (odr_encode(), addinfo ? addinfo : ""); + dr->u.v2Addinfo = odr_strdup (odr, addinfo ? addinfo : ""); return rec; } - -Z_Records *Yaz_Z_Server::pack_records (const char *resultSetName, - int start, int xnum, - Z_RecordComposition *comp, - int *next, int *pres, - int *format) -{ - int recno, total_length = 0, toget = xnum, dumped_records = 0; - Z_Records *records = - (Z_Records *) odr_malloc (odr_encode(), sizeof(*records)); - Z_NamePlusRecordList *reclist = - (Z_NamePlusRecordList *) odr_malloc (odr_encode(), sizeof(*reclist)); - Z_NamePlusRecord **list = - (Z_NamePlusRecord **) odr_malloc (odr_encode(), sizeof(*list) * toget); - - records->which = Z_Records_DBOSD; - records->u.databaseOrSurDiagnostics = reclist; - reclist->num_records = 0; - reclist->records = list; - *pres = Z_PRES_SUCCESS; - *next = 0; - - yaz_log(LOG_LOG, "Request to pack %d+%d", start, toget); - yaz_log(LOG_LOG, "pms=%d, mrs=%d", m_preferredMessageSize, - m_maximumRecordSize); - for (recno = start; reclist->num_records < toget; recno++) - { - Z_NamePlusRecord *this_rec = - (Z_NamePlusRecord *) odr_malloc (odr_encode(), sizeof(*this_rec)); - this_rec->databaseName = 0; - this_rec->which = Z_NamePlusRecord_databaseRecord; - this_rec->u.databaseRecord = 0; - - int this_length = 0; - - recv_Z_record (resultSetName, recno, format, comp, this_rec, records); - - if (records->which != Z_Records_DBOSD) - { - *pres = Z_PRES_FAILURE; - return records; - } - - if (this_rec->which == Z_NamePlusRecord_databaseRecord && - this_rec->u.databaseRecord == 0) - { // handler did not return a record.. - create_surrogateDiagnostics(this_rec, 0, 14, 0); - } - /* - * we get the number of bytes allocated on the stream before any - * allocation done by the backend - this should give us a reasonable - * idea of the total size of the data so far. - */ - total_length = odr_total(odr_encode()) - dumped_records; - this_length = odr_total(odr_encode()) - total_length; - yaz_log(LOG_LOG, " fetched record, len=%d, total=%d", - this_length, total_length); - if (this_length + total_length > m_preferredMessageSize) - { - /* record is small enough, really */ - if (this_length <= m_preferredMessageSize) - { - yaz_log(LOG_LOG, " Dropped last normal-sized record"); - *pres = Z_PRES_PARTIAL_2; - break; - } - if (this_length >= m_maximumRecordSize) - { /* too big entirely */ - yaz_log(LOG_LOG, "Record > maxrcdsz"); - reclist->records[reclist->num_records] = this_rec; - create_surrogateDiagnostics(this_rec, - this_rec->databaseName, 17, 0); - reclist->num_records++; - *next = recno + 1; - dumped_records += this_length; - continue; - } - else /* record can only be fetched by itself */ - { - yaz_log(LOG_LOG, " Record > prefmsgsz"); - if (toget > 1) - { - yaz_log(LOG_DEBUG, " Dropped it"); - reclist->records[reclist->num_records] = this_rec; - create_surrogateDiagnostics(this_rec, - this_rec->databaseName, - 16, 0); - reclist->num_records++; - // *next = freq.last_in_set ? 0 : recno + 1; - *next = recno + 1; - dumped_records += this_length; - continue; - } - } - } - reclist->records[reclist->num_records] = this_rec; - reclist->num_records++; - *next = recno + 1; - } - return records; -} - -void Yaz_Z_Server::fetch_via_piggyback (Z_SearchRequest *req, - Z_SearchResponse *res) -{ - bool_t *sr = (bool_t *)odr_malloc (odr_encode(), sizeof(*sr)); - *sr = 1; - - int toget = 0; - - Z_RecordComposition comp, *compp = 0; - int hits = *res->resultCount; - - int *nulint = (int *)odr_malloc (odr_encode(), sizeof(*nulint)); - *nulint = 0; - - comp.which = Z_RecordComp_simple; - /* how many records does the user agent want, then? */ - if (hits <= *req->smallSetUpperBound) - { - toget = hits; - if ((comp.u.simple = req->smallSetElementSetNames)) - compp = ∁ - } - else if (hits < *req->largeSetLowerBound) - { - toget = *req->mediumSetPresentNumber; - if (toget > hits) - toget = hits; - if ((comp.u.simple = req->mediumSetElementSetNames)) - compp = ∁ - } - - if (toget && !res->records) - { - res->presentStatus = (int *) odr_malloc (odr_encode(), sizeof(int)); - *res->presentStatus = Z_PRES_SUCCESS; - res->records = - pack_records(req->resultSetName, 1, toget, compp, - res->nextResultSetPosition, - res->presentStatus, - req->preferredRecordSyntax); - if (!res->records) - return; - if (res->records->which == Z_Records_DBOSD) - *res->numberOfRecordsReturned = - res->records->u.databaseOrSurDiagnostics->num_records; - res->searchStatus = sr; - res->resultSetStatus = 0; - } - else - { - if (hits) - *res->nextResultSetPosition = 1; - res->numberOfRecordsReturned = nulint; - res->searchStatus = sr; - res->resultSetStatus = 0; - res->presentStatus = 0; - } -} - -void Yaz_Z_Server::fetch_via_present (Z_PresentRequest *req, - Z_PresentResponse *res) -{ - res->records = pack_records (req->resultSetId,*req->resultSetStartPoint, - *req->numberOfRecordsRequested, - req->recordComposition, - res->nextResultSetPosition, - res->presentStatus, - req->preferredRecordSyntax); - if (res->records->which == Z_Records_DBOSD) - *res->numberOfRecordsReturned = - res->records->u.databaseOrSurDiagnostics->num_records; -} - -void Yaz_Z_Server::recv_Z_PDU (Z_APDU *apdu_request) -{ - Z_APDU *apdu_response; - switch (apdu_request->which) - { - case Z_APDU_initRequest: - logf (LOG_LOG, "got InitRequest p=%p", this); - apdu_response = create_Z_PDU(Z_APDU_initResponse); - recv_Z_init (apdu_request->u.initRequest, - apdu_response->u.initResponse); - m_preferredMessageSize = - *apdu_request->u.initRequest->preferredMessageSize; - m_maximumRecordSize = - *apdu_request->u.initRequest->maximumRecordSize; - send_Z_PDU(apdu_response); - break; - case Z_APDU_searchRequest: - logf (LOG_LOG, "got SearchRequest p=%p", this); - apdu_response = create_Z_PDU(Z_APDU_searchResponse); - recv_Z_search (apdu_request->u.searchRequest, - apdu_response->u.searchResponse); - if (!apdu_response->u.searchResponse->records) - { - fetch_via_piggyback(apdu_request->u.searchRequest, - apdu_response->u.searchResponse); - } - send_Z_PDU(apdu_response); - break; - case Z_APDU_presentRequest: - logf (LOG_LOG, "got PresentRequest p=%p", this); - apdu_response = create_Z_PDU(Z_APDU_presentResponse); - recv_Z_present (apdu_request->u.presentRequest, - apdu_response->u.presentResponse); - if (!apdu_response->u.presentResponse->records) - fetch_via_present(apdu_request->u.presentRequest, - apdu_response->u.presentResponse); - send_Z_PDU(apdu_response); - break; - } -} -