From: Xie Jiagui Date: Tue, 6 Mar 2012 08:27:23 +0000 (+0800) Subject: [1600] Add reused MessageRenderer,and make some clean up. X-Git-Tag: trac2351_base~226^2~116^2~132 X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=d5ec40dace9fddaaec9873cfca2d670e8d35650a;p=thirdparty%2Fkea.git [1600] Add reused MessageRenderer,and make some clean up. --- diff --git a/src/bin/auth/auth_srv.cc b/src/bin/auth/auth_srv.cc index b533b8f2ea..73f559245c 100644 --- a/src/bin/auth/auth_srv.cc +++ b/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); } diff --git a/src/bin/auth/tests/query_unittest.cc b/src/bin/auth/tests/query_unittest.cc index 0c413a1fdd..89f7d042d6 100644 --- a/src/bin/auth/tests/query_unittest.cc +++ b/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) +