Browse Source

[1600] Add reused MessageRenderer,and make some clean up.

Xie Jiagui 13 years ago
parent
commit
d5ec40dace
2 changed files with 13 additions and 14 deletions
  1. 13 13
      src/bin/auth/auth_srv.cc
  2. 0 1
      src/bin/auth/tests/query_unittest.cc

+ 13 - 13
src/bin/auth/auth_srv.cc

@@ -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);
 }
 

+ 0 - 1
src/bin/auth/tests/query_unittest.cc

@@ -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) +