src/filter_backend_test.cpp

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