123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032 |
- #include <config.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <sys/time.h>
- #include <cstring>
- #include <boost/noncopyable.hpp>
- #include <boost/lexical_cast.hpp>
- #include <boost/bind.hpp>
- #include <boost/scoped_ptr.hpp>
- #include <boost/date_time/posix_time/posix_time_types.hpp>
- #include <gtest/gtest.h>
- #include <exceptions/exceptions.h>
- #include <dns/tests/unittest_util.h>
- #include <dns/rcode.h>
- #include <util/buffer.h>
- #include <util/unittests/resolver.h>
- #include <dns/message.h>
- #include <dns/rdataclass.h>
- #include <nsas/nameserver_address_store.h>
- #include <cache/resolver_cache.h>
- #include <resolve/resolve.h>
- #include <resolve/recursive_query.h>
- #include <asiodns/dns_lookup.h>
- #include <asiolink/io_socket.h>
- #include <asiolink/io_service.h>
- #include <asiolink/io_message.h>
- #include <asiolink/io_error.h>
- #include <asiolink/simple_callback.h>
- using isc::UnitTestUtil;
- using namespace std;
- using namespace isc::asiodns;
- using namespace isc::asiolink;
- using namespace isc::dns;
- using namespace isc::util;
- using boost::scoped_ptr;
- namespace isc {
- namespace asiodns {
- std::string
- deepestDelegation(Name name, RRClass rrclass,
- isc::cache::ResolverCache& cache);
- }
- }
- namespace {
- const char* const TEST_SERVER_PORT = "53535";
- const char* const TEST_CLIENT_PORT = "53536";
- const char* const TEST_IPV6_ADDR = "::1";
- const char* const TEST_IPV4_ADDR = "127.0.0.1";
- const uint8_t test_data[] = {0, 4, 1, 2, 3, 4};
- struct addrinfo*
- resolveAddress(const int protocol, const char* const addr,
- const char* const port)
- {
- struct addrinfo hints;
- memset(&hints, 0, sizeof(hints));
- hints.ai_family = AF_UNSPEC;
- hints.ai_socktype = (protocol == IPPROTO_UDP) ? SOCK_DGRAM : SOCK_STREAM;
- hints.ai_protocol = protocol;
- hints.ai_flags = AI_NUMERICSERV;
- struct addrinfo* res;
- const int error = getaddrinfo(addr, port, &hints, &res);
- if (error != 0) {
- isc_throw(IOError, "getaddrinfo failed: " << gai_strerror(error));
- }
- return (res);
- }
- struct addrinfo*
- resolveAddress(const int family, const int protocol, const bool client) {
- return (resolveAddress(protocol,
- (family == AF_INET6) ? TEST_IPV6_ADDR :
- TEST_IPV4_ADDR,
- client ? TEST_CLIENT_PORT : TEST_SERVER_PORT));
- }
- struct ScopedAddrInfo {
- ScopedAddrInfo(struct addrinfo* res) : res_(res) {}
- ~ScopedAddrInfo() { freeaddrinfo(res_); }
- struct addrinfo* res_;
- };
- struct ScopedSocket : private boost::noncopyable {
- ScopedSocket() : s_(-1) {}
- ScopedSocket(int s) : s_(s) {}
- ~ScopedSocket() {
- if (s_ >= 0) {
- close(s_);
- }
- }
- void reset(int new_s) {
- if (s_ >= 0) {
- close(s_);
- }
- s_ = new_s;
- }
- int release() {
- int s = s_;
- s_ = -1;
- return (s);
- }
- int s_;
- };
- class RecursiveQueryTest : public ::testing::Test {
- protected:
- RecursiveQueryTest();
- ~RecursiveQueryTest() {
-
-
- resolver_.reset();
- }
-
- void sendUDP(const int family) {
- ScopedAddrInfo sai(resolveAddress(family, IPPROTO_UDP, false));
- struct addrinfo* res = sai.res_;
- sock_.reset(socket(res->ai_family, res->ai_socktype,
- res->ai_protocol));
- if (sock_.s_ < 0) {
- isc_throw(IOError, "failed to open test socket");
- }
- const int cc = sendto(sock_.s_, test_data, sizeof(test_data), 0,
- res->ai_addr, res->ai_addrlen);
- if (cc != sizeof(test_data)) {
- isc_throw(IOError, "unexpected sendto result: " << cc);
- }
- io_service_->run();
- }
-
- void sendTCP(const int family) {
- ScopedAddrInfo sai(resolveAddress(family, IPPROTO_TCP, false));
- struct addrinfo* res = sai.res_;
- sock_.reset(socket(res->ai_family, res->ai_socktype,
- res->ai_protocol));
- if (sock_.s_ < 0) {
- isc_throw(IOError, "failed to open test socket");
- }
- if (connect(sock_.s_, res->ai_addr, res->ai_addrlen) < 0) {
- isc_throw(IOError, "failed to connect to the test server");
- }
- const int cc = send(sock_.s_, test_data, sizeof(test_data), 0);
- if (cc != sizeof(test_data)) {
- isc_throw(IOError, "unexpected send result: " << cc);
- }
- io_service_->run();
- }
-
-
-
- void recvUDP(const int family, void* buffer, size_t& size) {
- ScopedAddrInfo sai(resolveAddress(family, IPPROTO_UDP, true));
- struct addrinfo* res = sai.res_;
- sock_.reset(socket(res->ai_family, res->ai_socktype,
- res->ai_protocol));
- if (sock_.s_ < 0) {
- isc_throw(IOError, "failed to open test socket");
- }
- if (bind(sock_.s_, res->ai_addr, res->ai_addrlen) < 0) {
- isc_throw(IOError, "bind failed: " << strerror(errno));
- }
-
-
-
- io_service_->run_one();
-
- io_service_->run_one();
-
-
-
-
-
-
- const struct timeval timeo = { 10, 0 };
- int recv_options = 0;
- if (setsockopt(sock_.s_, SOL_SOCKET, SO_RCVTIMEO, &timeo,
- sizeof(timeo))) {
- if (errno == ENOPROTOOPT) {
-
-
-
-
-
- recv_options = MSG_DONTWAIT;
- } else {
- isc_throw(IOError, "set RCVTIMEO failed: " << strerror(errno));
- }
- }
- const int ret = recv(sock_.s_, buffer, size, recv_options);
- if (ret < 0) {
- isc_throw(IOError, "recvfrom failed: " << strerror(errno));
- }
-
-
- size = ret;
- }
- void
- addServer(const string& address, const char* const port, int protocol) {
- ScopedAddrInfo sai(resolveAddress(protocol, address.c_str(), port));
- struct addrinfo* res = sai.res_;
- const int family = res->ai_family;
- ScopedSocket sock(socket(res->ai_family, res->ai_socktype,
- res->ai_protocol));
- const int s = sock.s_;
- if (s < 0) {
- isc_throw(isc::Unexpected, "failed to open a test socket");
- }
- const int on = 1;
- if (family == AF_INET6) {
- if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on)) ==
- -1) {
- isc_throw(isc::Unexpected,
- "failed to set socket option(IPV6_V6ONLY)");
- }
- }
- if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1) {
- isc_throw(isc::Unexpected,
- "failed to set socket option(SO_REUSEADDR)");
- }
- if (bind(s, res->ai_addr, res->ai_addrlen) != 0) {
- isc_throw(isc::Unexpected, "failed to bind a test socket");
- }
- if (protocol == IPPROTO_TCP) {
- dns_service_->addServerTCPFromFD(sock.release(), family);
- } else {
- dns_service_->addServerUDPFromFD(sock.release(), family);
- }
- }
-
- void setDNSService(const string& address) {
- setDNSService();
- addServer(address, TEST_SERVER_PORT, IPPROTO_TCP);
- addServer(address, TEST_SERVER_PORT, IPPROTO_UDP);
- }
-
-
- void setDNSService(const bool use_ipv4, const bool use_ipv6) {
- setDNSService();
- if (use_ipv6) {
- addServer("::", TEST_SERVER_PORT, IPPROTO_TCP);
- addServer("::", TEST_SERVER_PORT, IPPROTO_UDP);
- }
- if (use_ipv4) {
- addServer("0.0.0.0", TEST_SERVER_PORT, IPPROTO_TCP);
- addServer("0.0.0.0", TEST_SERVER_PORT, IPPROTO_UDP);
- }
- }
-
-
- void setDNSService() {
- io_service_.reset(new IOService());
- callback_.reset(new ASIOCallBack(this));
- dns_service_.reset(new DNSService(*io_service_, callback_.get(), NULL,
- NULL));
- }
-
-
-
-
-
-
- void doTest(const int family, const int protocol) {
- if (protocol == IPPROTO_UDP) {
- sendUDP(family);
- } else {
- sendTCP(family);
- }
-
-
-
- EXPECT_NE(sock_.s_, callback_native_);
- EXPECT_EQ(protocol, callback_protocol_);
- EXPECT_EQ(family == AF_INET6 ? TEST_IPV6_ADDR : TEST_IPV4_ADDR,
- callback_address_);
- const uint8_t* expected_data =
- protocol == IPPROTO_UDP ? test_data : test_data + 2;
- const size_t expected_datasize =
- protocol == IPPROTO_UDP ? sizeof(test_data) :
- sizeof(test_data) - 2;
- EXPECT_PRED_FORMAT4(UnitTestUtil::matchWireData, &callback_data_[0],
- callback_data_.size(),
- expected_data, expected_datasize);
- }
- protected:
-
-
-
- class MockServer : public DNSServer {
- public:
- explicit MockServer(IOService& io_service,
- SimpleCallback* checkin = NULL,
- DNSLookup* lookup = NULL,
- DNSAnswer* answer = NULL) :
- io_(io_service),
- done_(false),
- message_(new Message(Message::PARSE)),
- answer_message_(new Message(Message::RENDER)),
- respbuf_(new OutputBuffer(0)),
- checkin_(checkin), lookup_(lookup), answer_(answer)
- {}
- void operator()(asio::error_code ec = asio::error_code(),
- size_t length = 0)
- {}
- void resume(const bool) {
-
- }
- DNSServer* clone() {
- MockServer* s = new MockServer(*this);
- return (s);
- }
- inline void asyncLookup() {
- if (lookup_) {
- (*lookup_)(*io_message_, message_, answer_message_,
- respbuf_, this);
- }
- }
- protected:
- IOService& io_;
- bool done_;
- private:
-
-
- boost::shared_ptr<isc::asiolink::IOMessage> io_message_;
- isc::dns::MessagePtr message_;
- isc::dns::MessagePtr answer_message_;
- isc::util::OutputBufferPtr respbuf_;
-
- const SimpleCallback* checkin_;
- const DNSLookup* lookup_;
- const DNSAnswer* answer_;
- };
-
- class MockServerStop : public MockServer {
- public:
- explicit MockServerStop(IOService& io_service, bool* done) :
- MockServer(io_service),
- done_(done)
- {}
- void resume(const bool done) {
- *done_ = done;
- io_.stop();
- }
- DNSServer* clone() {
- return (new MockServerStop(*this));
- }
- private:
- bool* done_;
- };
-
-
-
-
- class MockServerStop2 : public MockServer {
- public:
- explicit MockServerStop2(IOService& io_service,
- bool* done1, bool* done2) :
- MockServer(io_service),
- done1_(done1),
- done2_(done2),
- stopped_once_(false)
- {}
- void resume(const bool done) {
- if (stopped_once_) {
- *done2_ = done;
- io_.stop();
- } else {
- *done1_ = done;
- stopped_once_ = true;
- }
- }
- DNSServer* clone() {
- return (new MockServerStop2(*this));
- }
- private:
- bool* done1_;
- bool* done2_;
- bool stopped_once_;
- };
- private:
- class ASIOCallBack : public SimpleCallback {
- public:
- ASIOCallBack(RecursiveQueryTest* test_obj) : test_obj_(test_obj) {}
- void operator()(const IOMessage& io_message) const {
- test_obj_->callBack(io_message);
- }
- private:
- RecursiveQueryTest* test_obj_;
- };
- void callBack(const IOMessage& io_message) {
- callback_protocol_ = io_message.getSocket().getProtocol();
- callback_native_ = io_message.getSocket().getNative();
- callback_address_ =
- io_message.getRemoteEndpoint().getAddress().toText();
- callback_data_.assign(
- static_cast<const uint8_t*>(io_message.getData()),
- static_cast<const uint8_t*>(io_message.getData()) +
- io_message.getDataSize());
- io_service_->stop();
- }
- protected:
-
-
- scoped_ptr<IOService> io_service_;
- scoped_ptr<DNSService> dns_service_;
- scoped_ptr<isc::nsas::NameserverAddressStore> nsas_;
- isc::cache::ResolverCache cache_;
- scoped_ptr<ASIOCallBack> callback_;
- int callback_protocol_;
- int callback_native_;
- string callback_address_;
- vector<uint8_t> callback_data_;
- ScopedSocket sock_;
- boost::shared_ptr<isc::util::unittests::TestResolver> resolver_;
- };
- RecursiveQueryTest::RecursiveQueryTest() :
- dns_service_(NULL), callback_(NULL), callback_protocol_(0),
- callback_native_(-1), resolver_(new isc::util::unittests::TestResolver())
- {
- io_service_.reset(new IOService());
- setDNSService(true, true);
- nsas_.reset(new isc::nsas::NameserverAddressStore(resolver_));
- }
- TEST_F(RecursiveQueryTest, v6UDPSend) {
- doTest(AF_INET6, IPPROTO_UDP);
- }
- TEST_F(RecursiveQueryTest, v6TCPSend) {
- doTest(AF_INET6, IPPROTO_TCP);
- }
- TEST_F(RecursiveQueryTest, v4UDPSend) {
- doTest(AF_INET, IPPROTO_UDP);
- }
- TEST_F(RecursiveQueryTest, v4TCPSend) {
- doTest(AF_INET, IPPROTO_TCP);
- }
- TEST_F(RecursiveQueryTest, v6UDPSendSpecific) {
-
-
-
-
-
-
-
-
-
-
- setDNSService(TEST_IPV6_ADDR);
- doTest(AF_INET6, IPPROTO_UDP);
- }
- TEST_F(RecursiveQueryTest, v6TCPSendSpecific) {
- setDNSService(TEST_IPV6_ADDR);
- doTest(AF_INET6, IPPROTO_TCP);
- EXPECT_THROW(sendTCP(AF_INET), IOError);
- }
- TEST_F(RecursiveQueryTest, v4UDPSendSpecific) {
- setDNSService(TEST_IPV4_ADDR);
- doTest(AF_INET, IPPROTO_UDP);
- }
- TEST_F(RecursiveQueryTest, v4TCPSendSpecific) {
- setDNSService(TEST_IPV4_ADDR);
- doTest(AF_INET, IPPROTO_TCP);
- EXPECT_THROW(sendTCP(AF_INET6), IOError);
- }
- TEST_F(RecursiveQueryTest, v6AddServer) {
- setDNSService();
- addServer(TEST_IPV6_ADDR, TEST_SERVER_PORT, IPPROTO_TCP);
- doTest(AF_INET6, IPPROTO_TCP);
- EXPECT_THROW(sendTCP(AF_INET), IOError);
- }
- TEST_F(RecursiveQueryTest, v4AddServer) {
- setDNSService();
- addServer(TEST_IPV4_ADDR, TEST_SERVER_PORT, IPPROTO_TCP);
- doTest(AF_INET, IPPROTO_TCP);
- EXPECT_THROW(sendTCP(AF_INET6), IOError);
- }
- TEST_F(RecursiveQueryTest, clearServers) {
- setDNSService();
- dns_service_->clearServers();
- EXPECT_THROW(sendTCP(AF_INET), IOError);
- EXPECT_THROW(sendTCP(AF_INET6), IOError);
- }
- TEST_F(RecursiveQueryTest, v6TCPOnly) {
-
-
-
- setDNSService(false, true);
- EXPECT_THROW(sendTCP(AF_INET), IOError);
- }
- TEST_F(RecursiveQueryTest, v4TCPOnly) {
- setDNSService(true, false);
- EXPECT_THROW(sendTCP(AF_INET6), IOError);
- }
- vector<pair<string, uint16_t> >
- singleAddress(const string &address, uint16_t port) {
- vector<pair<string, uint16_t> > result;
- result.push_back(pair<string, uint16_t>(address, port));
- return (result);
- }
- TEST_F(RecursiveQueryTest, recursiveSetupV4) {
- setDNSService(true, false);
- uint16_t port = boost::lexical_cast<uint16_t>(TEST_CLIENT_PORT);
- EXPECT_NO_THROW(RecursiveQuery(*dns_service_,
- *nsas_, cache_,
- singleAddress(TEST_IPV4_ADDR, port),
- singleAddress(TEST_IPV4_ADDR, port)));
- }
- TEST_F(RecursiveQueryTest, recursiveSetupV6) {
- setDNSService(false, true);
- uint16_t port = boost::lexical_cast<uint16_t>(TEST_CLIENT_PORT);
- EXPECT_NO_THROW(RecursiveQuery(*dns_service_,
- *nsas_, cache_,
- singleAddress(TEST_IPV6_ADDR, port),
- singleAddress(TEST_IPV6_ADDR,port)));
- }
- TEST_F(RecursiveQueryTest, forwarderSend) {
- setDNSService(true, false);
-
-
- uint16_t port = boost::lexical_cast<uint16_t>(TEST_CLIENT_PORT);
- MockServer server(*io_service_);
- RecursiveQuery rq(*dns_service_,
- *nsas_, cache_,
- singleAddress(TEST_IPV4_ADDR, port),
- singleAddress(TEST_IPV4_ADDR, port));
- Question q(Name("example.com"), RRClass::IN(), RRType::TXT());
- Message query_message(Message::RENDER);
- isc::resolve::initResponseMessage(q, query_message);
- OutputBufferPtr buffer(new OutputBuffer(0));
- MessagePtr answer(new Message(Message::RENDER));
- rq.forward(ConstMessagePtr(&query_message), answer, buffer, &server);
- char data[4096];
- size_t size = sizeof(data);
- ASSERT_NO_THROW(recvUDP(AF_INET, data, size));
- Message m(Message::PARSE);
- InputBuffer ibuf(data, size);
-
- EXPECT_NO_THROW(m.parseHeader(ibuf));
- EXPECT_NO_THROW(m.fromWire(ibuf));
-
- QuestionPtr q2 = *m.beginQuestion();
- EXPECT_EQ(q.getName(), q2->getName());
- EXPECT_EQ(q.getType(), q2->getType());
- EXPECT_EQ(q.getClass(), q2->getClass());
- }
- int
- createTestSocket() {
- ScopedAddrInfo sai(resolveAddress(AF_INET, IPPROTO_UDP, true));
- struct addrinfo* res = sai.res_;
- ScopedSocket sock(socket(res->ai_family, res->ai_socktype,
- res->ai_protocol));
- if (sock.s_ < 0) {
- isc_throw(IOError, "failed to open test socket");
- }
- if (bind(sock.s_, res->ai_addr, res->ai_addrlen) < 0) {
- isc_throw(IOError, "failed to bind test socket");
- }
- return (sock.release());
- }
- int
- setSocketTimeout(int sock, size_t tv_sec, size_t tv_usec) {
- const struct timeval timeo = { tv_sec, tv_usec };
- int recv_options = 0;
- if (setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &timeo, sizeof(timeo))) {
- if (errno == ENOPROTOOPT) {
- recv_options = MSG_DONTWAIT;
- } else {
- isc_throw(IOError, "set RCVTIMEO failed: " << strerror(errno));
- }
- }
- return (recv_options);
- }
- bool tryRead(int sock, int recv_options, size_t max, int* num) {
- size_t i = 0;
- do {
- char inbuff[512];
- if (recv(sock, inbuff, sizeof(inbuff), recv_options) < 0) {
- return false;
- } else {
- ++i;
- ++*num;
- }
- } while (i < max);
- return true;
- }
- class MockResolverCallback : public isc::resolve::ResolverInterface::Callback {
- public:
- enum ResultValue {
- DEFAULT = 0,
- SUCCESS = 1,
- FAILURE = 2
- };
- MockResolverCallback(DNSServer* server):
- result(DEFAULT),
- server_(server->clone())
- {}
- ~MockResolverCallback() {
- delete server_;
- }
- void success(const isc::dns::MessagePtr response) {
- result = SUCCESS;
- server_->resume(true);
- }
- void failure() {
- result = FAILURE;
- server_->resume(false);
- }
- uint32_t result;
- private:
- DNSServer* server_;
- };
- TEST_F(RecursiveQueryTest, forwardQueryTimeout) {
-
- setDNSService();
-
- sock_.reset(createTestSocket());
-
- bool done(true);
- MockServerStop server(*io_service_, &done);
-
- const uint16_t port = boost::lexical_cast<uint16_t>(TEST_CLIENT_PORT);
- RecursiveQuery query(*dns_service_,
- *nsas_, cache_,
- singleAddress(TEST_IPV4_ADDR, port),
- singleAddress(TEST_IPV4_ADDR, port),
- 10, 4000, 3000, 2);
- Question question(Name("example.net"), RRClass::IN(), RRType::A());
- OutputBufferPtr buffer(new OutputBuffer(0));
- MessagePtr answer(new Message(Message::RENDER));
- Message query_message(Message::RENDER);
- isc::resolve::initResponseMessage(question, query_message);
- boost::shared_ptr<MockResolverCallback> callback(new MockResolverCallback(&server));
- query.forward(ConstMessagePtr(&query_message), answer, buffer, &server, callback);
-
- io_service_->run();
- EXPECT_EQ(callback->result, MockResolverCallback::FAILURE);
- }
- TEST_F(RecursiveQueryTest, forwardClientTimeout) {
-
- setDNSService();
- sock_.reset(createTestSocket());
-
- bool done1(true);
- MockServerStop server(*io_service_, &done1);
- MessagePtr answer(new Message(Message::RENDER));
-
- const uint16_t port = boost::lexical_cast<uint16_t>(TEST_CLIENT_PORT);
- RecursiveQuery query(*dns_service_,
- *nsas_, cache_,
- singleAddress(TEST_IPV4_ADDR, port),
- singleAddress(TEST_IPV4_ADDR, port),
- 1000, 10, 4000, 4);
- Question q(Name("example.net"), RRClass::IN(), RRType::A());
- OutputBufferPtr buffer(new OutputBuffer(0));
- Message query_message(Message::RENDER);
- isc::resolve::initResponseMessage(q, query_message);
- boost::shared_ptr<MockResolverCallback> callback(new MockResolverCallback(&server));
- query.forward(ConstMessagePtr(&query_message), answer, buffer, &server, callback);
-
- io_service_->run();
- EXPECT_EQ(callback->result, MockResolverCallback::FAILURE);
- }
- TEST_F(RecursiveQueryTest, forwardLookupTimeout) {
-
- setDNSService();
-
- sock_.reset(createTestSocket());
-
- bool done(true);
- MockServerStop server(*io_service_, &done);
- MessagePtr answer(new Message(Message::RENDER));
-
- const uint16_t port = boost::lexical_cast<uint16_t>(TEST_CLIENT_PORT);
- RecursiveQuery query(*dns_service_,
- *nsas_, cache_,
- singleAddress(TEST_IPV4_ADDR, port),
- singleAddress(TEST_IPV4_ADDR, port),
- 1000, 4000, 10, 5);
- Question question(Name("example.net"), RRClass::IN(), RRType::A());
- OutputBufferPtr buffer(new OutputBuffer(0));
- Message query_message(Message::RENDER);
- isc::resolve::initResponseMessage(question, query_message);
- boost::shared_ptr<MockResolverCallback> callback(new MockResolverCallback(&server));
- query.forward(ConstMessagePtr(&query_message), answer, buffer, &server, callback);
-
- io_service_->run();
- EXPECT_EQ(callback->result, MockResolverCallback::FAILURE);
- }
- TEST_F(RecursiveQueryTest, lowtimeouts) {
-
- setDNSService();
-
- sock_.reset(createTestSocket());
-
- bool done(true);
- MockServerStop server(*io_service_, &done);
- MessagePtr answer(new Message(Message::RENDER));
-
- const uint16_t port = boost::lexical_cast<uint16_t>(TEST_CLIENT_PORT);
- RecursiveQuery query(*dns_service_,
- *nsas_, cache_,
- singleAddress(TEST_IPV4_ADDR, port),
- singleAddress(TEST_IPV4_ADDR, port),
- 1, 1, 1, 1);
- Question question(Name("example.net"), RRClass::IN(), RRType::A());
- OutputBufferPtr buffer(new OutputBuffer(0));
- Message query_message(Message::RENDER);
- isc::resolve::initResponseMessage(question, query_message);
- boost::shared_ptr<MockResolverCallback> callback(new MockResolverCallback(&server));
- query.forward(ConstMessagePtr(&query_message), answer, buffer, &server, callback);
-
- io_service_->run();
- EXPECT_EQ(callback->result, MockResolverCallback::FAILURE);
- }
- TEST_F(RecursiveQueryTest, DISABLED_recursiveSendOk) {
- setDNSService(true, false);
- bool done;
-
- MockServerStop server(*io_service_, &done);
- vector<pair<string, uint16_t> > empty_vector;
- RecursiveQuery rq(*dns_service_, *nsas_, cache_, empty_vector,
- empty_vector, 10000, 0);
- Question q(Name("www.isc.org"), RRClass::IN(), RRType::A());
- OutputBufferPtr buffer(new OutputBuffer(0));
- MessagePtr answer(new Message(Message::RENDER));
- rq.resolve(q, answer, buffer, &server);
- io_service_->run();
-
- EXPECT_EQ(Rcode::NOERROR(), answer->getRcode());
- ASSERT_EQ(1, answer->getRRCount(Message::SECTION_ANSWER));
- RRsetPtr a = *answer->beginSection(Message::SECTION_ANSWER);
- EXPECT_EQ(q.getName(), a->getName());
- EXPECT_EQ(q.getType(), a->getType());
- EXPECT_EQ(q.getClass(), a->getClass());
- EXPECT_EQ(1, a->getRdataCount());
- }
- TEST_F(RecursiveQueryTest, DISABLED_recursiveSendNXDOMAIN) {
- setDNSService(true, false);
- bool done;
-
- MockServerStop server(*io_service_, &done);
- vector<pair<string, uint16_t> > empty_vector;
- RecursiveQuery rq(*dns_service_, *nsas_, cache_, empty_vector,
- empty_vector, 10000, 0);
- Question q(Name("wwwdoesnotexist.isc.org"), RRClass::IN(), RRType::A());
- OutputBufferPtr buffer(new OutputBuffer(0));
- MessagePtr answer(new Message(Message::RENDER));
- rq.resolve(q, answer, buffer, &server);
- io_service_->run();
-
- EXPECT_EQ(Rcode::NXDOMAIN(), answer->getRcode());
- EXPECT_EQ(0, answer->getRRCount(Message::SECTION_ANSWER));
- }
- TEST_F(RecursiveQueryTest, CachedNS) {
- setDNSService(true, true);
-
-
- EXPECT_EQ(".", deepestDelegation(Name("www.somewhere.deep.example.org"),
- RRClass::IN(), cache_));
-
-
-
- RRsetPtr nsUpper(new RRset(Name("example.org"), RRClass::IN(),
- RRType::NS(), RRTTL(300)));
- nsUpper->addRdata(rdata::generic::NS(Name("ns.example.org")));
- nsUpper->addRdata(rdata::generic::NS(Name("ns2.example.org")));
- RRsetPtr nsLower(new RRset(Name("somewhere.deep.example.org"),
- RRClass::IN(), RRType::NS(), RRTTL(300)));
- nsLower->addRdata(rdata::generic::NS(Name("ns.somewhere.deep.example.org"))
- );
- RRsetPtr nsIp(new RRset(Name("ns2.example.org"), RRClass::IN(),
- RRType::A(), RRTTL(300)));
- nsIp->addRdata(rdata::in::A("192.0.2.1"));
-
-
-
- ASSERT_TRUE(cache_.update(nsUpper));
- ASSERT_TRUE(cache_.update(nsLower));
- ASSERT_TRUE(cache_.update(nsIp));
- RRsetPtr deepest(cache_.lookupDeepestNS(Name(
- "www.somewhere.deep.example.org"), RRClass::IN()));
- ASSERT_NE(RRsetPtr(), deepest);
- ASSERT_EQ(nsLower->getName(), deepest->getName());
-
-
-
- EXPECT_EQ("example.org.",
- deepestDelegation(Name("www.somewhere.deep.example.org"),
- RRClass::IN(), cache_));
-
-
-
-
-
-
-
-
-
- vector<pair<string, uint16_t> > roots;
- roots.push_back(pair<string, uint16_t>("192.0.2.2", 53));
- RecursiveQuery rq(*dns_service_, *nsas_, cache_,
- vector<pair<string, uint16_t> >(), roots);
-
-
-
- Question q(Name("www.somewhere.deep.example.org"), RRClass::IN(),
- RRType::A());
- OutputBufferPtr buffer(new OutputBuffer(0));
- MessagePtr answer(new Message(Message::RENDER));
-
- MockServer server(*io_service_);
- rq.resolve(q, answer, buffer, &server);
-
-
-
- EXPECT_NO_THROW(EXPECT_EQ(nsUpper->getName(),
- (*resolver_)[0]->getName()) <<
- "It starts resolving at the wrong place") <<
- "It does not ask NSAS anything, how does it know where to send?";
- }
- }
|