src/filter_backend_test.cpp

   1 /* $Id: filter_backend_test.cpp,v 1.6 2005-10-29 15:54:29 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 "filter_backend_test.hpp"
  16
  17 #include <yaz/zgdu.h>
  18 #include <yaz/log.h>
  19 #include <yaz/otherinfo.h>
  20 #include <yaz/diagbib1.h>
  21
  22 #include <list>
  23 #include <map>
  24 #include <iostream>
  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 yf::Backend_test::Backend_test() : m_p(new Backend_test::Rep) {
  45     m_p->m_support_named_result_sets = false;
  46 }
  47
  48 yf::Backend_test::~Backend_test() {
  49 }
  50
  51 void yf::Backend_test::process(Package &package) const
  52 {
  53     Z_GDU *gdu = package.request().get();
  54
  55     if (!gdu || gdu->which != Z_GDU_Z3950)
  56         package.move();
  57     else
  58     {
  59         Z_APDU *apdu_req = gdu->u.z3950;
  60         Z_APDU *apdu_res = 0;
  61         ODR odr = odr_createmem(ODR_ENCODE);
  62
  63         if (apdu_req->which != Z_APDU_initRequest &&
  64             !m_p->m_sessions.exist(package.session()))
  65         {
  66             apdu_res = zget_APDU(odr, Z_APDU_close);
  67             *apdu_res->u.close->closeReason = Z_Close_protocolError;
  68             apdu_res->u.close->diagnosticInformation =
  69                 odr_strdup(odr, "no init for filter_backend_test");
  70
  71             package.session().close();
  72         }
  73         else if (apdu_req->which == Z_APDU_initRequest)
  74         {
  75             apdu_res = zget_APDU(odr, Z_APDU_initResponse);
  76             Z_InitRequest *req = apdu_req->u.initRequest;
  77             Z_InitResponse *resp = apdu_res->u.initResponse;
  78
  79             int i;
  80             static const int masks[] = {
  81                 Z_Options_search, Z_Options_present, -1
  82             };
  83             for (i = 0; masks[i] != -1; i++)
  84                 if (ODR_MASK_GET(req->options, masks[i]))
  85                     ODR_MASK_SET(resp->options, masks[i]);
  86             if (m_p->m_support_named_result_sets)
  87             {
  88                 if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
  89                     ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
  90                 else
  91                     m_p->m_support_named_result_sets = false;
  92             }
  93             static const int versions[] = {
  94                 Z_ProtocolVersion_1,
  95                 Z_ProtocolVersion_2,
  96                 Z_ProtocolVersion_3,
  97                 -1
  98             };
  99             for (i = 0; versions[i] != -1; i++)
 100                 if (ODR_MASK_GET(req->protocolVersion, versions[i]))
 101                     ODR_MASK_SET(resp->protocolVersion, versions[i]);
 102                 else
 103                     break;
 104
 105             Session_info info;
 106             m_p->m_sessions.create(info, package.session());
 107         }
 108         else if (apdu_req->which == Z_APDU_searchRequest)
 109         {
 110             apdu_res = zget_APDU(odr, Z_APDU_searchResponse);
 111             Z_SearchRequest *req = apdu_req->u.searchRequest;
 112             Z_SearchResponse *resp = apdu_res->u.searchResponse;
 113
 114             if (!m_p->m_support_named_result_sets &&
 115                 strcmp(req->resultSetName, "default"))
 116             {
 117                 Z_Records *rec = (Z_Records *)
 118                     odr_malloc(odr, sizeof(Z_Records));
 119                 resp->records = rec;
 120                 rec->which = Z_Records_NSD;
 121                 rec->u.nonSurrogateDiagnostic =
 122                     zget_DefaultDiagFormat(
 123                         odr, YAZ_BIB1_RESULT_SET_NAMING_UNSUPP, 0);
 124             }
 125             else
 126                 *resp->resultCount = 42;
 127         }
 128         else if (apdu_req->which == Z_APDU_presentRequest)
 129         {
 130             apdu_res = zget_APDU(odr, Z_APDU_presentResponse);
 131         }
 132         else
 133         {
 134             apdu_res = zget_APDU(odr, Z_APDU_close);
 135             *apdu_res->u.close->closeReason = Z_Close_protocolError;
 136             apdu_res->u.close->diagnosticInformation =
 137                 odr_strdup(odr, "bad APDU in filter_backend_test");
 138
 139             package.session().close();
 140         }
 141         if (apdu_res)
 142             package.response() = apdu_res;
 143         odr_destroy(odr);
 144     }
 145     if (package.session().is_closed())
 146         m_p->m_sessions.release(package.session());
 147 }
 148
 149
 150 /*
 151  * Local variables:
 152  * c-basic-offset: 4
 153  * indent-tabs-mode: nil
 154  * c-file-style: "stroustrup"
 155  * End:
 156  * vim: shiftwidth=4 tabstop=8 expandtab
 157  */