src/filter_backend_test.cpp

   1 /* $Id: filter_backend_test.cpp,v 1.8 2005-10-30 18:51:20 adam Exp $
   2    Copyright (c) 2005, Index Data.
   3
   4 %LICENSE%
   5  */
   6
   7 #include "config.hpp"
   8
   9 #include "filter.hpp"
  10 #include "router.hpp"
  11 #include "package.hpp"
  12
  13 #include <boost/thread/mutex.hpp>
  14
  15 #include "util.hpp"
  16 #include "filter_backend_test.hpp"
  17
  18 #include <yaz/zgdu.h>
  19 #include <yaz/log.h>
  20 #include <yaz/otherinfo.h>
  21 #include <yaz/diagbib1.h>
  22
  23 #include <list>
  24 #include <map>
  25 #include <iostream>
  26
  27 namespace yf = yp2::filter;
  28
  29 namespace yp2 {
  30     namespace filter {
  31         class Session_info {
  32             int dummy;
  33         };
  34         class Backend_test::Rep {
  35             friend class Backend_test;
  36
  37         private:
  38             bool m_support_named_result_sets;
  39
  40             session_map<Session_info> m_sessions;
  41         };
  42     }
  43 }
  44
  45 yf::Backend_test::Backend_test() : m_p(new Backend_test::Rep) {
  46     m_p->m_support_named_result_sets = false;
  47 }
  48
  49 yf::Backend_test::~Backend_test() {
  50 }
  51
  52 void yf::Backend_test::process(Package &package) const
  53 {
  54     Z_GDU *gdu = package.request().get();
  55
  56     if (!gdu || gdu->which != Z_GDU_Z3950)
  57         package.move();
  58     else
  59     {
  60         Z_APDU *apdu_req = gdu->u.z3950;
  61         Z_APDU *apdu_res = 0;
  62         yp2::odr odr;
  63
  64         if (apdu_req->which != Z_APDU_initRequest &&
  65             !m_p->m_sessions.exist(package.session()))
  66         {
  67             apdu_res = odr.create_close(Z_Close_protocolError,
  68                                         "no init for filter_backend_test");
  69             package.session().close();
  70         }
  71         else if (apdu_req->which == Z_APDU_initRequest)
  72         {
  73             apdu_res = zget_APDU(odr, Z_APDU_initResponse);
  74             Z_InitRequest *req = apdu_req->u.initRequest;
  75             Z_InitResponse *resp = apdu_res->u.initResponse;
  76
  77             int i;
  78             static const int masks[] = {
  79                 Z_Options_search, Z_Options_present, -1
  80             };
  81             for (i = 0; masks[i] != -1; i++)
  82                 if (ODR_MASK_GET(req->options, masks[i]))
  83                     ODR_MASK_SET(resp->options, masks[i]);
  84             if (m_p->m_support_named_result_sets)
  85             {
  86                 if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
  87                     ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
  88                 else
  89                     m_p->m_support_named_result_sets = false;
  90             }
  91             static const int versions[] = {
  92                 Z_ProtocolVersion_1,
  93                 Z_ProtocolVersion_2,
  94                 Z_ProtocolVersion_3,
  95                 -1
  96             };
  97             for (i = 0; versions[i] != -1; i++)
  98                 if (ODR_MASK_GET(req->protocolVersion, versions[i]))
  99                     ODR_MASK_SET(resp->protocolVersion, versions[i]);
 100                 else
 101                     break;
 102
 103             Session_info info;
 104             m_p->m_sessions.create(info, package.session());
 105         }
 106         else if (apdu_req->which == Z_APDU_searchRequest)
 107         {
 108             apdu_res = zget_APDU(odr, Z_APDU_searchResponse);
 109             Z_SearchRequest *req = apdu_req->u.searchRequest;
 110             Z_SearchResponse *resp = apdu_res->u.searchResponse;
 111
 112             if (!m_p->m_support_named_result_sets &&
 113                 strcmp(req->resultSetName, "default"))
 114             {
 115                 Z_Records *rec = (Z_Records *)
 116                     odr_malloc(odr, sizeof(Z_Records));
 117                 resp->records = rec;
 118                 rec->which = Z_Records_NSD;
 119                 rec->u.nonSurrogateDiagnostic =
 120                     zget_DefaultDiagFormat(
 121                         odr, YAZ_BIB1_RESULT_SET_NAMING_UNSUPP, 0);
 122             }
 123             else
 124                 *resp->resultCount = 42;
 125         }
 126         else if (apdu_req->which == Z_APDU_presentRequest)
 127         {
 128             apdu_res = zget_APDU(odr, Z_APDU_presentResponse);
 129         }
 130         else
 131         {
 132             apdu_res = odr.create_close(Z_Close_protocolError,
 133                                         "bad APDU in filter_backend_test");
 134             package.session().close();
 135         }
 136         if (apdu_res)
 137             package.response() = apdu_res;
 138     }
 139     if (package.session().is_closed())
 140         m_p->m_sessions.release(package.session());
 141 }
 142
 143
 144 /*
 145  * Local variables:
 146  * c-basic-offset: 4
 147  * indent-tabs-mode: nil
 148  * c-file-style: "stroustrup"
 149  * End:
 150  * vim: shiftwidth=4 tabstop=8 expandtab
 151  */