]> git.ipfire.org Git - thirdparty/kea.git/commitdiff
[1600] Add reused MessageRenderer,and make some clean up.
authorXie Jiagui <xiejiagui@cnnic.cn>
Tue, 6 Mar 2012 08:27:23 +0000 (16:27 +0800)
committerXie Jiagui <xiejiagui@cnnic.cn>
Tue, 6 Mar 2012 08:27:23 +0000 (16:27 +0800)
src/bin/auth/auth_srv.cc
src/bin/auth/tests/query_unittest.cc

index b533b8f2eadfaa4a6794eb2b9b3f6c94d95160db..73f559245c4dd5b610ef9f88681496b9b9acc959 100644 (file)
@@ -99,6 +99,8 @@ public:
 
     IOService io_service_;
 
+    MessageRenderer renderer_;
+    
     /// Currently non-configurable, but will be.
     static const uint16_t DEFAULT_LOCAL_UDPSIZE = 4096;
 
@@ -306,7 +308,7 @@ makeErrorMessage(MessagePtr message, OutputBufferPtr buffer,
     }
     for_each(questions.begin(), questions.end(), QuestionInserter(message));
     message->setRcode(rcode);
-
+    
     MessageRenderer renderer;
     renderer.setBuffer(buffer.get());
     if (tsig_context.get() != NULL) {
@@ -556,19 +558,18 @@ AuthSrvImpl::processNormalQuery(const IOMessage& io_message, MessagePtr message,
         return (true);
     }
 
-    MessageRenderer renderer;
-    renderer.setBuffer(buffer.get());
+    renderer_.setBuffer(buffer.get());
     const bool udp_buffer =
         (io_message.getSocket().getProtocol() == IPPROTO_UDP);
-    renderer.setLengthLimit(udp_buffer ? remote_bufsize : 65535);
+    renderer_.setLengthLimit(udp_buffer ? remote_bufsize : 65535);
     if (tsig_context.get() != NULL) {
-        message->toWire(renderer, *tsig_context);
+        message->toWire(renderer_, *tsig_context);
     } else {
-        message->toWire(renderer);
+        message->toWire(renderer_);
     }
-    renderer.setBuffer(NULL);
+    renderer_.setBuffer(NULL);
     LOG_DEBUG(auth_logger, DBG_AUTH_MESSAGES, AUTH_SEND_NORMAL_RESPONSE)
-              .arg(renderer.getLength()).arg(message->toText());
+              .arg(renderer_.getLength()).arg(message->toText());
 
     return (true);
 }
@@ -687,14 +688,13 @@ AuthSrvImpl::processNotify(const IOMessage& io_message, MessagePtr message,
     message->setHeaderFlag(Message::HEADERFLAG_AA);
     message->setRcode(Rcode::NOERROR());
 
-    MessageRenderer renderer;
-    renderer.setBuffer(buffer.get());
+    renderer_.setBuffer(buffer.get());
     if (tsig_context.get() != NULL) {
-        message->toWire(renderer, *tsig_context);
+        message->toWire(renderer_, *tsig_context);
     } else {
-        message->toWire(renderer);
+        message->toWire(renderer_);
     }
-    renderer.setBuffer(NULL);
+    renderer_.setBuffer(NULL);
     return (true);
 }
 
index 0c413a1fdde7940686de3ecfd4f050579b6e8642..89f7d042d6428859c1a8bb0ee2bd947425e7e48b 100644 (file)
@@ -1117,7 +1117,6 @@ TEST_F(QueryTest, secureUnsignedDelegationWithNSEC3OptOut) {
     // proof (and their RRSIGs).  The closest encloser is the apex (origin),
     // and with our faked hash the covering NSEC3 for the next closer
     // (= child zone name) is that for www.example.com.
-    cout << response << endl;
     responseCheck(response, Rcode::NOERROR(), 0, 0, 5, 0,
                   NULL,
                   (string(unsigned_delegation_txt) +