X-Git-Url: http://sru.miketaylor.org.uk/?a=blobdiff_plain;f=src%2Ffilter_backend_test.cpp;h=bec5daf3f92461518de7baea8082c12d2640bc50;hb=7a90dfec2c8b5af3c39bbb7fafdf2294671b93d4;hp=f4f0b165cdf21d657d3e5f58e55353acecd58411;hpb=cf09b1981338f07c333e43e0a2c907012c841c5d;p=metaproxy-moved-to-github.git

diff --git a/src/filter_backend_test.cpp b/src/filter_backend_test.cpp
index f4f0b16..bec5daf 100644
--- a/src/filter_backend_test.cpp
+++ b/src/filter_backend_test.cpp
@@ -1,4 +1,4 @@
-/* $Id: filter_backend_test.cpp,v 1.1 2005-10-25 11:48:30 adam Exp $
+/* $Id: filter_backend_test.cpp,v 1.6 2005-10-29 15:54:29 adam Exp $
    Copyright (c) 2005, Index Data.
 
 %LICENSE%
@@ -27,22 +27,25 @@ namespace yf = yp2::filter;
 
 namespace yp2 {
     namespace filter {
+        class Session_info {
+            int dummy;
+        };
         class Backend_test::Rep {
             friend class Backend_test;
             
         private:
             bool m_support_named_result_sets;
+
+            session_map<Session_info> m_sessions;
         };
     }
 }
 
-yf::Backend_test::Backend_test() {
-    m_p = new Backend_test::Rep;
+yf::Backend_test::Backend_test() : m_p(new Backend_test::Rep) {
     m_p->m_support_named_result_sets = false;
 }
 
 yf::Backend_test::~Backend_test() {
-    delete m_p;
 }
 
 void yf::Backend_test::process(Package &package) const
@@ -56,7 +59,18 @@ void yf::Backend_test::process(Package &package) const
         Z_APDU *apdu_req = gdu->u.z3950;
         Z_APDU *apdu_res = 0;
         ODR odr = odr_createmem(ODR_ENCODE);
-        if (apdu_req->which == Z_APDU_initRequest)
+        
+        if (apdu_req->which != Z_APDU_initRequest && 
+            !m_p->m_sessions.exist(package.session()))
+        {
+            apdu_res = zget_APDU(odr, Z_APDU_close);            
+            *apdu_res->u.close->closeReason = Z_Close_protocolError;
+            apdu_res->u.close->diagnosticInformation =
+                odr_strdup(odr, "no init for filter_backend_test");
+            
+            package.session().close();
+        }
+        else if (apdu_req->which == Z_APDU_initRequest)
         {
             apdu_res = zget_APDU(odr, Z_APDU_initResponse);
             Z_InitRequest *req = apdu_req->u.initRequest;
@@ -64,9 +78,9 @@ void yf::Backend_test::process(Package &package) const
             
             int i;
             static const int masks[] = {
-                Z_Options_search, Z_Options_present, 0 
+                Z_Options_search, Z_Options_present, -1 
             };
-            for (i = 0; masks[i]; i++)
+            for (i = 0; masks[i] != -1; i++)
                 if (ODR_MASK_GET(req->options, masks[i]))
                     ODR_MASK_SET(resp->options, masks[i]);
             if (m_p->m_support_named_result_sets)
@@ -76,13 +90,27 @@ void yf::Backend_test::process(Package &package) const
                 else
                     m_p->m_support_named_result_sets = false;
             }
+            static const int versions[] = {
+                Z_ProtocolVersion_1,
+                Z_ProtocolVersion_2,
+                Z_ProtocolVersion_3,
+                -1
+            };
+            for (i = 0; versions[i] != -1; i++)
+                if (ODR_MASK_GET(req->protocolVersion, versions[i]))
+                    ODR_MASK_SET(resp->protocolVersion, versions[i]);
+                else
+                    break;
+
+            Session_info info;
+            m_p->m_sessions.create(info, package.session());
         }
         else if (apdu_req->which == Z_APDU_searchRequest)
-        { 
+        {
             apdu_res = zget_APDU(odr, Z_APDU_searchResponse);
             Z_SearchRequest *req = apdu_req->u.searchRequest;
             Z_SearchResponse *resp = apdu_res->u.searchResponse;
-
+                
             if (!m_p->m_support_named_result_sets && 
                 strcmp(req->resultSetName, "default"))
             {
@@ -105,12 +133,17 @@ void yf::Backend_test::process(Package &package) const
         {
             apdu_res = zget_APDU(odr, Z_APDU_close);            
             *apdu_res->u.close->closeReason = Z_Close_protocolError;
+            apdu_res->u.close->diagnosticInformation =
+                odr_strdup(odr, "bad APDU in filter_backend_test");
+            
             package.session().close();
         }
         if (apdu_res)
             package.response() = apdu_res;
         odr_destroy(odr);
     }
+    if (package.session().is_closed())
+        m_p->m_sessions.release(package.session());
 }