[1539] make sure the forwarder connection is closed on destruction of authsrv.

diff --git a/src/bin/auth/auth_srv.cc b/src/bin/auth/auth_srv.cc
index a48f9da8dd4e73d550143b211e1e2151b04ac48f..609e572d68487889484a8159000762031441ef10 100644 (file)
--- a/src/bin/auth/auth_srv.cc
+++ b/src/bin/auth/auth_srv.cc
@@ -117,7 +117,11 @@ public:
     SocketSessionForwarderHolder(BaseSocketSessionForwarder& forwarder) :
         forwarder_(forwarder), connected_(false)
     {}
-    ~SocketSessionForwarderHolder() {}
+    ~SocketSessionForwarderHolder() {
+        if (connected_) {
+            forwarder_.close();
+        }
+    }
 
     void connect() {
         if (!connected_) {

diff --git a/src/bin/auth/tests/auth_srv_unittest.cc b/src/bin/auth/tests/auth_srv_unittest.cc
index 66c95ed777c1aa3da08200d063e5f0a88cb36adc..199ce64eb710cae9d6ebf054f7a1ba5a5d622f74 100644 (file)
--- a/src/bin/auth/tests/auth_srv_unittest.cc
+++ b/src/bin/auth/tests/auth_srv_unittest.cc
@@ -14,10 +14,6 @@
 
 #include <config.h>
 
-#include <vector>
-
-#include <boost/shared_ptr.hpp>
-
 #include <gtest/gtest.h>
 
 #include <dns/message.h>
@@ -45,6 +41,11 @@
 #include <testutils/portconfig.h>
 #include <testutils/socket_request.h>
 
+#include <boost/shared_ptr.hpp>
+#include <boost/scoped_ptr.hpp>
+
+#include <vector>
+
 using namespace std;
 using namespace isc::cc;
 using namespace isc::dns;
@@ -57,6 +58,7 @@ using namespace isc::asiolink;
 using namespace isc::testutils;
 using namespace isc::server_common::portconfig;
 using isc::UnitTestUtil;
+using boost::scoped_ptr;
 
 namespace {
 const char* const CONFIG_TESTDB =
@@ -1407,4 +1409,20 @@ TEST_F(AuthSrvTest, DDNSForward) {
     }
 }
 
+TEST_F(AuthSrvTest, DDNSForwardClose) {
+    scoped_ptr<AuthSrv> tmp_server(new AuthSrv(true, xfrout, ddns_forwarder));
+    UnitTestUtil::createRequestMessage(request_message, Opcode::UPDATE(),
+                                       default_qid, Name("example.com"),
+                                       RRClass::IN(), RRType::SOA());
+    createRequestPacket(request_message, IPPROTO_UDP);
+    tmp_server->processMessage(*io_message, *parse_message, *response_obuffer,
+                               &dnsserv);
+    EXPECT_FALSE(dnsserv.hasAnswer());
+    EXPECT_TRUE(ddns_forwarder.isConnected());
+
+    // Destroy the server.  The forwarder should close the connection.
+    tmp_server.reset();
+    EXPECT_FALSE(ddns_forwarder.isConnected());
+}
+
 }

diff --git a/src/lib/testutils/mockups.h b/src/lib/testutils/mockups.h
index 4f4c27b2ea3c4d4860ef092aceed3b745e5b0a39..cbfa864e93c0827962a9338c8aeb0254254db50f 100644 (file)
--- a/src/lib/testutils/mockups.h
+++ b/src/lib/testutils/mockups.h
@@ -223,7 +223,9 @@ public:
         }
         is_connected_ = true;
     }
-    virtual void close() {}
+    virtual void close() {
+        is_connected_ = false;
+    }
 #if 0
     virtual void push(int sock, int family, int type, int protocol,
                       const struct sockaddr& local_end,