src/filter_backend_test.cpp

   1 /* $Id: filter_backend_test.cpp,v 1.1 2005-10-25 11:48:30 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 Backend_test::Rep {
  31             friend class Backend_test;
  32
  33         private:
  34             bool m_support_named_result_sets;
  35         };
  36     }
  37 }
  38
  39 yf::Backend_test::Backend_test() {
  40     m_p = new Backend_test::Rep;
  41     m_p->m_support_named_result_sets = false;
  42 }
  43
  44 yf::Backend_test::~Backend_test() {
  45     delete m_p;
  46 }
  47
  48 void yf::Backend_test::process(Package &package) const
  49 {
  50     Z_GDU *gdu = package.request().get();
  51
  52     if (!gdu || gdu->which != Z_GDU_Z3950)
  53         package.move();
  54     else
  55     {
  56         Z_APDU *apdu_req = gdu->u.z3950;
  57         Z_APDU *apdu_res = 0;
  58         ODR odr = odr_createmem(ODR_ENCODE);
  59         if (apdu_req->which == Z_APDU_initRequest)
  60         {
  61             apdu_res = zget_APDU(odr, Z_APDU_initResponse);
  62             Z_InitRequest *req = apdu_req->u.initRequest;
  63             Z_InitResponse *resp = apdu_res->u.initResponse;
  64
  65             int i;
  66             static const int masks[] = {
  67                 Z_Options_search, Z_Options_present, 0
  68             };
  69             for (i = 0; masks[i]; i++)
  70                 if (ODR_MASK_GET(req->options, masks[i]))
  71                     ODR_MASK_SET(resp->options, masks[i]);
  72             if (m_p->m_support_named_result_sets)
  73             {
  74                 if (ODR_MASK_GET(req->options, Z_Options_namedResultSets))
  75                     ODR_MASK_SET(resp->options, Z_Options_namedResultSets);
  76                 else
  77                     m_p->m_support_named_result_sets = false;
  78             }
  79         }
  80         else if (apdu_req->which == Z_APDU_searchRequest)
  81         {
  82             apdu_res = zget_APDU(odr, Z_APDU_searchResponse);
  83             Z_SearchRequest *req = apdu_req->u.searchRequest;
  84             Z_SearchResponse *resp = apdu_res->u.searchResponse;
  85
  86             if (!m_p->m_support_named_result_sets &&
  87                 strcmp(req->resultSetName, "default"))
  88             {
  89                 Z_Records *rec = (Z_Records *)
  90                     odr_malloc(odr, sizeof(Z_Records));
  91                 resp->records = rec;
  92                 rec->which = Z_Records_NSD;
  93                 rec->u.nonSurrogateDiagnostic =
  94                     zget_DefaultDiagFormat(
  95                         odr, YAZ_BIB1_RESULT_SET_NAMING_UNSUPP, 0);
  96             }
  97             else
  98                 *resp->resultCount = 42;
  99         }
 100         else if (apdu_req->which == Z_APDU_presentRequest)
 101         {
 102             apdu_res = zget_APDU(odr, Z_APDU_presentResponse);
 103         }
 104         else
 105         {
 106             apdu_res = zget_APDU(odr, Z_APDU_close);
 107             *apdu_res->u.close->closeReason = Z_Close_protocolError;
 108             package.session().close();
 109         }
 110         if (apdu_res)
 111             package.response() = apdu_res;
 112         odr_destroy(odr);
 113     }
 114 }
 115
 116
 117 /*
 118  * Local variables:
 119  * c-basic-offset: 4
 120  * indent-tabs-mode: nil
 121  * c-file-style: "stroustrup"
 122  * End:
 123  * vim: shiftwidth=4 tabstop=8 expandtab
 124  */