diff options
author | phajdan.jr@chromium.org <phajdan.jr@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2013-03-21 01:40:53 +0000 |
---|---|---|
committer | phajdan.jr@chromium.org <phajdan.jr@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2013-03-21 01:40:53 +0000 |
commit | f2cb3cf86e4217581443d93e863c891a05066e60 (patch) | |
tree | c62110d2347d17e1b1a1f57fa758cf8bba72548b /net/base/host_resolver_impl_unittest.cc | |
parent | b8512263f9556ac705394e9da03d364980f0eadf (diff) | |
download | chromium_src-f2cb3cf86e4217581443d93e863c891a05066e60.zip chromium_src-f2cb3cf86e4217581443d93e863c891a05066e60.tar.gz chromium_src-f2cb3cf86e4217581443d93e863c891a05066e60.tar.bz2 |
net: move host_resolver files from net/base to net/dns
BUG=70818
Review URL: https://codereview.chromium.org/12518036
git-svn-id: svn://svn.chromium.org/chrome/trunk/src@189485 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'net/base/host_resolver_impl_unittest.cc')
-rw-r--r-- | net/base/host_resolver_impl_unittest.cc | 1481 |
1 files changed, 0 insertions, 1481 deletions
diff --git a/net/base/host_resolver_impl_unittest.cc b/net/base/host_resolver_impl_unittest.cc deleted file mode 100644 index d0c8fcb..0000000 --- a/net/base/host_resolver_impl_unittest.cc +++ /dev/null @@ -1,1481 +0,0 @@ -// Copyright (c) 2012 The Chromium Authors. All rights reserved. -// Use of this source code is governed by a BSD-style license that can be -// found in the LICENSE file. - -#include "net/base/host_resolver_impl.h" - -#include <algorithm> -#include <string> - -#include "base/bind.h" -#include "base/bind_helpers.h" -#include "base/memory/ref_counted.h" -#include "base/memory/scoped_vector.h" -#include "base/message_loop.h" -#include "base/string_util.h" -#include "base/stringprintf.h" -#include "base/synchronization/condition_variable.h" -#include "base/synchronization/lock.h" -#include "base/test/test_timeouts.h" -#include "base/time.h" -#include "net/base/address_list.h" -#include "net/base/host_cache.h" -#include "net/base/mock_host_resolver.h" -#include "net/base/net_errors.h" -#include "net/base/net_util.h" -#include "net/dns/dns_client.h" -#include "net/dns/dns_test_util.h" -#include "testing/gtest/include/gtest/gtest.h" - -namespace net { - -namespace { - -const size_t kMaxJobs = 10u; -const size_t kMaxRetryAttempts = 4u; - -PrioritizedDispatcher::Limits DefaultLimits() { - PrioritizedDispatcher::Limits limits(NUM_PRIORITIES, kMaxJobs); - return limits; -} - -HostResolverImpl::ProcTaskParams DefaultParams( - HostResolverProc* resolver_proc) { - return HostResolverImpl::ProcTaskParams(resolver_proc, kMaxRetryAttempts); -} - -// A HostResolverProc that pushes each host mapped into a list and allows -// waiting for a specific number of requests. Unlike RuleBasedHostResolverProc -// it never calls SystemHostResolverProc. By default resolves all hostnames to -// "127.0.0.1". After AddRule(), it resolves only names explicitly specified. -class MockHostResolverProc : public HostResolverProc { - public: - struct ResolveKey { - ResolveKey(const std::string& hostname, AddressFamily address_family) - : hostname(hostname), address_family(address_family) {} - bool operator<(const ResolveKey& other) const { - return address_family < other.address_family || - (address_family == other.address_family && hostname < other.hostname); - } - std::string hostname; - AddressFamily address_family; - }; - - typedef std::vector<ResolveKey> CaptureList; - - MockHostResolverProc() - : HostResolverProc(NULL), - num_requests_waiting_(0), - num_slots_available_(0), - requests_waiting_(&lock_), - slots_available_(&lock_) { - } - - // Waits until |count| calls to |Resolve| are blocked. Returns false when - // timed out. - bool WaitFor(unsigned count) { - base::AutoLock lock(lock_); - base::Time start_time = base::Time::Now(); - while (num_requests_waiting_ < count) { - requests_waiting_.TimedWait(TestTimeouts::action_timeout()); - if (base::Time::Now() > start_time + TestTimeouts::action_timeout()) - return false; - } - return true; - } - - // Signals |count| waiting calls to |Resolve|. First come first served. - void SignalMultiple(unsigned count) { - base::AutoLock lock(lock_); - num_slots_available_ += count; - slots_available_.Broadcast(); - } - - // Signals all waiting calls to |Resolve|. Beware of races. - void SignalAll() { - base::AutoLock lock(lock_); - num_slots_available_ = num_requests_waiting_; - slots_available_.Broadcast(); - } - - void AddRule(const std::string& hostname, AddressFamily family, - const AddressList& result) { - base::AutoLock lock(lock_); - rules_[ResolveKey(hostname, family)] = result; - } - - void AddRule(const std::string& hostname, AddressFamily family, - const std::string& ip_list) { - AddressList result; - int rv = ParseAddressList(ip_list, "", &result); - DCHECK_EQ(OK, rv); - AddRule(hostname, family, result); - } - - void AddRuleForAllFamilies(const std::string& hostname, - const std::string& ip_list) { - AddressList result; - int rv = ParseAddressList(ip_list, "", &result); - DCHECK_EQ(OK, rv); - AddRule(hostname, ADDRESS_FAMILY_UNSPECIFIED, result); - AddRule(hostname, ADDRESS_FAMILY_IPV4, result); - AddRule(hostname, ADDRESS_FAMILY_IPV6, result); - } - - virtual int Resolve(const std::string& hostname, - AddressFamily address_family, - HostResolverFlags host_resolver_flags, - AddressList* addrlist, - int* os_error) OVERRIDE { - base::AutoLock lock(lock_); - capture_list_.push_back(ResolveKey(hostname, address_family)); - ++num_requests_waiting_; - requests_waiting_.Broadcast(); - while (!num_slots_available_) - slots_available_.Wait(); - DCHECK_GT(num_requests_waiting_, 0u); - --num_slots_available_; - --num_requests_waiting_; - if (rules_.empty()) { - int rv = ParseAddressList("127.0.0.1", "", addrlist); - DCHECK_EQ(OK, rv); - return OK; - } - ResolveKey key(hostname, address_family); - if (rules_.count(key) == 0) - return ERR_NAME_NOT_RESOLVED; - *addrlist = rules_[key]; - return OK; - } - - CaptureList GetCaptureList() const { - CaptureList copy; - { - base::AutoLock lock(lock_); - copy = capture_list_; - } - return copy; - } - - bool HasBlockedRequests() const { - base::AutoLock lock(lock_); - return num_requests_waiting_ > num_slots_available_; - } - - protected: - virtual ~MockHostResolverProc() {} - - private: - mutable base::Lock lock_; - std::map<ResolveKey, AddressList> rules_; - CaptureList capture_list_; - unsigned num_requests_waiting_; - unsigned num_slots_available_; - base::ConditionVariable requests_waiting_; - base::ConditionVariable slots_available_; - - DISALLOW_COPY_AND_ASSIGN(MockHostResolverProc); -}; - -// A wrapper for requests to a HostResolver. -class Request { - public: - // Base class of handlers to be executed on completion of requests. - struct Handler { - virtual ~Handler() {} - virtual void Handle(Request* request) = 0; - }; - - Request(const HostResolver::RequestInfo& info, - size_t index, - HostResolver* resolver, - Handler* handler) - : info_(info), - index_(index), - resolver_(resolver), - handler_(handler), - quit_on_complete_(false), - result_(ERR_UNEXPECTED), - handle_(NULL) {} - - int Resolve() { - DCHECK(resolver_); - DCHECK(!handle_); - list_ = AddressList(); - result_ = resolver_->Resolve( - info_, &list_, base::Bind(&Request::OnComplete, base::Unretained(this)), - &handle_, BoundNetLog()); - if (!list_.empty()) - EXPECT_EQ(OK, result_); - return result_; - } - - int ResolveFromCache() { - DCHECK(resolver_); - DCHECK(!handle_); - return resolver_->ResolveFromCache(info_, &list_, BoundNetLog()); - } - - void Cancel() { - DCHECK(resolver_); - DCHECK(handle_); - resolver_->CancelRequest(handle_); - handle_ = NULL; - } - - const HostResolver::RequestInfo& info() const { return info_; } - size_t index() const { return index_; } - const AddressList& list() const { return list_; } - int result() const { return result_; } - bool completed() const { return result_ != ERR_IO_PENDING; } - bool pending() const { return handle_ != NULL; } - - bool HasAddress(const std::string& address, int port) const { - IPAddressNumber ip; - bool rv = ParseIPLiteralToNumber(address, &ip); - DCHECK(rv); - return std::find(list_.begin(), - list_.end(), - IPEndPoint(ip, port)) != list_.end(); - } - - // Returns the number of addresses in |list_|. - unsigned NumberOfAddresses() const { - return list_.size(); - } - - bool HasOneAddress(const std::string& address, int port) const { - return HasAddress(address, port) && (NumberOfAddresses() == 1u); - } - - // Returns ERR_UNEXPECTED if timed out. - int WaitForResult() { - if (completed()) - return result_; - base::CancelableClosure closure(MessageLoop::QuitClosure()); - MessageLoop::current()->PostDelayedTask(FROM_HERE, - closure.callback(), - TestTimeouts::action_max_timeout()); - quit_on_complete_ = true; - MessageLoop::current()->Run(); - bool did_quit = !quit_on_complete_; - quit_on_complete_ = false; - closure.Cancel(); - if (did_quit) - return result_; - else - return ERR_UNEXPECTED; - } - - private: - void OnComplete(int rv) { - EXPECT_TRUE(pending()); - EXPECT_EQ(ERR_IO_PENDING, result_); - EXPECT_NE(ERR_IO_PENDING, rv); - result_ = rv; - handle_ = NULL; - if (!list_.empty()) { - EXPECT_EQ(OK, result_); - EXPECT_EQ(info_.port(), list_.front().port()); - } - if (handler_) - handler_->Handle(this); - if (quit_on_complete_) { - MessageLoop::current()->Quit(); - quit_on_complete_ = false; - } - } - - HostResolver::RequestInfo info_; - size_t index_; - HostResolver* resolver_; - Handler* handler_; - bool quit_on_complete_; - - AddressList list_; - int result_; - HostResolver::RequestHandle handle_; - - DISALLOW_COPY_AND_ASSIGN(Request); -}; - -// Using LookupAttemptHostResolverProc simulate very long lookups, and control -// which attempt resolves the host. -class LookupAttemptHostResolverProc : public HostResolverProc { - public: - LookupAttemptHostResolverProc(HostResolverProc* previous, - int attempt_number_to_resolve, - int total_attempts) - : HostResolverProc(previous), - attempt_number_to_resolve_(attempt_number_to_resolve), - current_attempt_number_(0), - total_attempts_(total_attempts), - total_attempts_resolved_(0), - resolved_attempt_number_(0), - all_done_(&lock_) { - } - - // Test harness will wait for all attempts to finish before checking the - // results. - void WaitForAllAttemptsToFinish(const base::TimeDelta& wait_time) { - base::TimeTicks end_time = base::TimeTicks::Now() + wait_time; - { - base::AutoLock auto_lock(lock_); - while (total_attempts_resolved_ != total_attempts_ && - base::TimeTicks::Now() < end_time) { - all_done_.TimedWait(end_time - base::TimeTicks::Now()); - } - } - } - - // All attempts will wait for an attempt to resolve the host. - void WaitForAnAttemptToComplete() { - base::TimeDelta wait_time = base::TimeDelta::FromSeconds(60); - base::TimeTicks end_time = base::TimeTicks::Now() + wait_time; - { - base::AutoLock auto_lock(lock_); - while (resolved_attempt_number_ == 0 && base::TimeTicks::Now() < end_time) - all_done_.TimedWait(end_time - base::TimeTicks::Now()); - } - all_done_.Broadcast(); // Tell all waiting attempts to proceed. - } - - // Returns the number of attempts that have finished the Resolve() method. - int total_attempts_resolved() { return total_attempts_resolved_; } - - // Returns the first attempt that that has resolved the host. - int resolved_attempt_number() { return resolved_attempt_number_; } - - // HostResolverProc methods. - virtual int Resolve(const std::string& host, - AddressFamily address_family, - HostResolverFlags host_resolver_flags, - AddressList* addrlist, - int* os_error) OVERRIDE { - bool wait_for_right_attempt_to_complete = true; - { - base::AutoLock auto_lock(lock_); - ++current_attempt_number_; - if (current_attempt_number_ == attempt_number_to_resolve_) { - resolved_attempt_number_ = current_attempt_number_; - wait_for_right_attempt_to_complete = false; - } - } - - if (wait_for_right_attempt_to_complete) - // Wait for the attempt_number_to_resolve_ attempt to resolve. - WaitForAnAttemptToComplete(); - - int result = ResolveUsingPrevious(host, address_family, host_resolver_flags, - addrlist, os_error); - - { - base::AutoLock auto_lock(lock_); - ++total_attempts_resolved_; - } - - all_done_.Broadcast(); // Tell all attempts to proceed. - - // Since any negative number is considered a network error, with -1 having - // special meaning (ERR_IO_PENDING). We could return the attempt that has - // resolved the host as a negative number. For example, if attempt number 3 - // resolves the host, then this method returns -4. - if (result == OK) - return -1 - resolved_attempt_number_; - else - return result; - } - - protected: - virtual ~LookupAttemptHostResolverProc() {} - - private: - int attempt_number_to_resolve_; - int current_attempt_number_; // Incremented whenever Resolve is called. - int total_attempts_; - int total_attempts_resolved_; - int resolved_attempt_number_; - - // All attempts wait for right attempt to be resolve. - base::Lock lock_; - base::ConditionVariable all_done_; -}; - -} // namespace - -class HostResolverImplTest : public testing::Test { - public: - static const int kDefaultPort = 80; - - HostResolverImplTest() : proc_(new MockHostResolverProc()) {} - - protected: - // A Request::Handler which is a proxy to the HostResolverImplTest fixture. - struct Handler : public Request::Handler { - virtual ~Handler() {} - - // Proxy functions so that classes derived from Handler can access them. - Request* CreateRequest(const HostResolver::RequestInfo& info) { - return test->CreateRequest(info); - } - Request* CreateRequest(const std::string& hostname, int port) { - return test->CreateRequest(hostname, port); - } - Request* CreateRequest(const std::string& hostname) { - return test->CreateRequest(hostname); - } - ScopedVector<Request>& requests() { return test->requests_; } - - void DeleteResolver() { test->resolver_.reset(); } - - HostResolverImplTest* test; - }; - - void CreateResolver() { - resolver_.reset(new HostResolverImpl( - HostCache::CreateDefaultCache(), - DefaultLimits(), - DefaultParams(proc_), - NULL)); - } - - // This HostResolverImpl will only allow 1 outstanding resolve at a time and - // perform no retries. - void CreateSerialResolver() { - HostResolverImpl::ProcTaskParams params = DefaultParams(proc_); - params.max_retry_attempts = 0u; - PrioritizedDispatcher::Limits limits(NUM_PRIORITIES, 1); - resolver_.reset(new HostResolverImpl( - HostCache::CreateDefaultCache(), - limits, - params, - NULL)); - } - - // The Request will not be made until a call to |Resolve()|, and the Job will - // not start until released by |proc_->SignalXXX|. - Request* CreateRequest(const HostResolver::RequestInfo& info) { - Request* req = new Request(info, requests_.size(), resolver_.get(), - handler_.get()); - requests_.push_back(req); - return req; - } - - Request* CreateRequest(const std::string& hostname, - int port, - RequestPriority priority, - AddressFamily family) { - HostResolver::RequestInfo info(HostPortPair(hostname, port)); - info.set_priority(priority); - info.set_address_family(family); - return CreateRequest(info); - } - - Request* CreateRequest(const std::string& hostname, - int port, - RequestPriority priority) { - return CreateRequest(hostname, port, priority, ADDRESS_FAMILY_UNSPECIFIED); - } - - Request* CreateRequest(const std::string& hostname, int port) { - return CreateRequest(hostname, port, MEDIUM); - } - - Request* CreateRequest(const std::string& hostname) { - return CreateRequest(hostname, kDefaultPort); - } - - virtual void SetUp() OVERRIDE { - CreateResolver(); - } - - virtual void TearDown() OVERRIDE { - if (resolver_.get()) - EXPECT_EQ(0u, resolver_->num_running_jobs_for_tests()); - EXPECT_FALSE(proc_->HasBlockedRequests()); - } - - void set_handler(Handler* handler) { - handler_.reset(handler); - handler_->test = this; - } - - // Friendship is not inherited, so use proxies to access those. - size_t num_running_jobs() const { - DCHECK(resolver_.get()); - return resolver_->num_running_jobs_for_tests(); - } - - scoped_refptr<MockHostResolverProc> proc_; - scoped_ptr<HostResolverImpl> resolver_; - ScopedVector<Request> requests_; - - scoped_ptr<Handler> handler_; -}; - -TEST_F(HostResolverImplTest, AsynchronousLookup) { - proc_->AddRuleForAllFamilies("just.testing", "192.168.1.42"); - proc_->SignalMultiple(1u); - - Request* req = CreateRequest("just.testing", 80); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - EXPECT_EQ(OK, req->WaitForResult()); - - EXPECT_TRUE(req->HasOneAddress("192.168.1.42", 80)); - - EXPECT_EQ("just.testing", proc_->GetCaptureList()[0].hostname); -} - -TEST_F(HostResolverImplTest, FailedAsynchronousLookup) { - proc_->AddRuleForAllFamilies("", "0.0.0.0"); // Default to failures. - proc_->SignalMultiple(1u); - - Request* req = CreateRequest("just.testing", 80); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, req->WaitForResult()); - - EXPECT_EQ("just.testing", proc_->GetCaptureList()[0].hostname); - - // Also test that the error is not cached. - EXPECT_EQ(ERR_DNS_CACHE_MISS, req->ResolveFromCache()); -} - -TEST_F(HostResolverImplTest, AbortedAsynchronousLookup) { - Request* req0 = CreateRequest("just.testing", 80); - EXPECT_EQ(ERR_IO_PENDING, req0->Resolve()); - - EXPECT_TRUE(proc_->WaitFor(1u)); - - // Resolver is destroyed while job is running on WorkerPool. - resolver_.reset(); - - proc_->SignalAll(); - - // To ensure there was no spurious callback, complete with a new resolver. - CreateResolver(); - Request* req1 = CreateRequest("just.testing", 80); - EXPECT_EQ(ERR_IO_PENDING, req1->Resolve()); - - proc_->SignalMultiple(2u); - - EXPECT_EQ(OK, req1->WaitForResult()); - - // This request was canceled. - EXPECT_FALSE(req0->completed()); -} - -TEST_F(HostResolverImplTest, NumericIPv4Address) { - // Stevens says dotted quads with AI_UNSPEC resolve to a single sockaddr_in. - Request* req = CreateRequest("127.1.2.3", 5555); - EXPECT_EQ(OK, req->Resolve()); - - EXPECT_TRUE(req->HasOneAddress("127.1.2.3", 5555)); -} - -TEST_F(HostResolverImplTest, NumericIPv6Address) { - // Resolve a plain IPv6 address. Don't worry about [brackets], because - // the caller should have removed them. - Request* req = CreateRequest("2001:db8::1", 5555); - EXPECT_EQ(OK, req->Resolve()); - - EXPECT_TRUE(req->HasOneAddress("2001:db8::1", 5555)); -} - -TEST_F(HostResolverImplTest, EmptyHost) { - Request* req = CreateRequest("", 5555); - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, req->Resolve()); -} - -TEST_F(HostResolverImplTest, LongHost) { - Request* req = CreateRequest(std::string(4097, 'a'), 5555); - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, req->Resolve()); -} - -TEST_F(HostResolverImplTest, DeDupeRequests) { - // Start 5 requests, duplicating hosts "a" and "b". Since the resolver_proc is - // blocked, these should all pile up until we signal it. - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b", 81)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 82)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b", 83)->Resolve()); - - proc_->SignalMultiple(2u); // One for "a", one for "b". - - for (size_t i = 0; i < requests_.size(); ++i) { - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; - } -} - -TEST_F(HostResolverImplTest, CancelMultipleRequests) { - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b", 81)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 82)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b", 83)->Resolve()); - - // Cancel everything except request for ("a", 82). - requests_[0]->Cancel(); - requests_[1]->Cancel(); - requests_[2]->Cancel(); - requests_[4]->Cancel(); - - proc_->SignalMultiple(2u); // One for "a", one for "b". - - EXPECT_EQ(OK, requests_[3]->WaitForResult()); -} - -TEST_F(HostResolverImplTest, CanceledRequestsReleaseJobSlots) { - // Fill up the dispatcher and queue. - for (unsigned i = 0; i < kMaxJobs + 1; ++i) { - std::string hostname = "a_"; - hostname[1] = 'a' + i; - EXPECT_EQ(ERR_IO_PENDING, CreateRequest(hostname, 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest(hostname, 81)->Resolve()); - } - - EXPECT_TRUE(proc_->WaitFor(kMaxJobs)); - - // Cancel all but last two. - for (unsigned i = 0; i < requests_.size() - 2; ++i) { - requests_[i]->Cancel(); - } - - EXPECT_TRUE(proc_->WaitFor(kMaxJobs + 1)); - - proc_->SignalAll(); - - size_t num_requests = requests_.size(); - EXPECT_EQ(OK, requests_[num_requests - 1]->WaitForResult()); - EXPECT_EQ(OK, requests_[num_requests - 2]->result()); -} - -TEST_F(HostResolverImplTest, CancelWithinCallback) { - struct MyHandler : public Handler { - virtual void Handle(Request* req) OVERRIDE { - // Port 80 is the first request that the callback will be invoked for. - // While we are executing within that callback, cancel the other requests - // in the job and start another request. - if (req->index() == 0) { - // Once "a:80" completes, it will cancel "a:81" and "a:82". - requests()[1]->Cancel(); - requests()[2]->Cancel(); - } - } - }; - set_handler(new MyHandler()); - - for (size_t i = 0; i < 4; ++i) { - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 80 + i)->Resolve()) << i; - } - - proc_->SignalMultiple(2u); // One for "a". One for "finalrequest". - - EXPECT_EQ(OK, requests_[0]->WaitForResult()); - - Request* final_request = CreateRequest("finalrequest", 70); - EXPECT_EQ(ERR_IO_PENDING, final_request->Resolve()); - EXPECT_EQ(OK, final_request->WaitForResult()); - EXPECT_TRUE(requests_[3]->completed()); -} - -TEST_F(HostResolverImplTest, DeleteWithinCallback) { - struct MyHandler : public Handler { - virtual void Handle(Request* req) OVERRIDE { - EXPECT_EQ("a", req->info().hostname()); - EXPECT_EQ(80, req->info().port()); - - DeleteResolver(); - - // Quit after returning from OnCompleted (to give it a chance at - // incorrectly running the cancelled tasks). - MessageLoop::current()->PostTask(FROM_HERE, MessageLoop::QuitClosure()); - } - }; - set_handler(new MyHandler()); - - for (size_t i = 0; i < 4; ++i) { - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 80 + i)->Resolve()) << i; - } - - proc_->SignalMultiple(1u); // One for "a". - - // |MyHandler| will send quit message once all the requests have finished. - MessageLoop::current()->Run(); -} - -TEST_F(HostResolverImplTest, DeleteWithinAbortedCallback) { - struct MyHandler : public Handler { - virtual void Handle(Request* req) OVERRIDE { - EXPECT_EQ("a", req->info().hostname()); - EXPECT_EQ(80, req->info().port()); - - DeleteResolver(); - - // Quit after returning from OnCompleted (to give it a chance at - // incorrectly running the cancelled tasks). - MessageLoop::current()->PostTask(FROM_HERE, MessageLoop::QuitClosure()); - } - }; - set_handler(new MyHandler()); - - // This test assumes that the Jobs will be Aborted in order ["a", "b"] - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 80)->Resolve()); - // HostResolverImpl will be deleted before later Requests can complete. - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 81)->Resolve()); - // Job for 'b' will be aborted before it can complete. - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b", 82)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b", 83)->Resolve()); - - EXPECT_TRUE(proc_->WaitFor(1u)); - - // Triggering an IP address change. - NetworkChangeNotifier::NotifyObserversOfIPAddressChangeForTests(); - - // |MyHandler| will send quit message once all the requests have finished. - MessageLoop::current()->Run(); - - EXPECT_EQ(ERR_NETWORK_CHANGED, requests_[0]->result()); - EXPECT_EQ(ERR_IO_PENDING, requests_[1]->result()); - EXPECT_EQ(ERR_IO_PENDING, requests_[2]->result()); - EXPECT_EQ(ERR_IO_PENDING, requests_[3]->result()); - // Clean up. - proc_->SignalMultiple(requests_.size()); -} - -TEST_F(HostResolverImplTest, StartWithinCallback) { - struct MyHandler : public Handler { - virtual void Handle(Request* req) OVERRIDE { - if (req->index() == 0) { - // On completing the first request, start another request for "a". - // Since caching is disabled, this will result in another async request. - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 70)->Resolve()); - } - } - }; - set_handler(new MyHandler()); - - // Turn off caching for this host resolver. - resolver_.reset(new HostResolverImpl( - scoped_ptr<HostCache>(), - DefaultLimits(), - DefaultParams(proc_), - NULL)); - - for (size_t i = 0; i < 4; ++i) { - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 80 + i)->Resolve()) << i; - } - - proc_->SignalMultiple(2u); // One for "a". One for the second "a". - - EXPECT_EQ(OK, requests_[0]->WaitForResult()); - ASSERT_EQ(5u, requests_.size()); - EXPECT_EQ(OK, requests_.back()->WaitForResult()); - - EXPECT_EQ(2u, proc_->GetCaptureList().size()); -} - -TEST_F(HostResolverImplTest, BypassCache) { - struct MyHandler : public Handler { - virtual void Handle(Request* req) OVERRIDE { - if (req->index() == 0) { - // On completing the first request, start another request for "a". - // Since caching is enabled, this should complete synchronously. - std::string hostname = req->info().hostname(); - EXPECT_EQ(OK, CreateRequest(hostname, 70)->Resolve()); - EXPECT_EQ(OK, CreateRequest(hostname, 75)->ResolveFromCache()); - - // Ok good. Now make sure that if we ask to bypass the cache, it can no - // longer service the request synchronously. - HostResolver::RequestInfo info(HostPortPair(hostname, 71)); - info.set_allow_cached_response(false); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest(info)->Resolve()); - } else if (71 == req->info().port()) { - // Test is done. - MessageLoop::current()->Quit(); - } else { - FAIL() << "Unexpected request"; - } - } - }; - set_handler(new MyHandler()); - - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a", 80)->Resolve()); - proc_->SignalMultiple(3u); // Only need two, but be generous. - - // |verifier| will send quit message once all the requests have finished. - MessageLoop::current()->Run(); - EXPECT_EQ(2u, proc_->GetCaptureList().size()); -} - -// Test that IP address changes flush the cache. -TEST_F(HostResolverImplTest, FlushCacheOnIPAddressChange) { - proc_->SignalMultiple(2u); // One before the flush, one after. - - Request* req = CreateRequest("host1", 70); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - EXPECT_EQ(OK, req->WaitForResult()); - - req = CreateRequest("host1", 75); - EXPECT_EQ(OK, req->Resolve()); // Should complete synchronously. - - // Flush cache by triggering an IP address change. - NetworkChangeNotifier::NotifyObserversOfIPAddressChangeForTests(); - MessageLoop::current()->RunUntilIdle(); // Notification happens async. - - // Resolve "host1" again -- this time it won't be served from cache, so it - // will complete asynchronously. - req = CreateRequest("host1", 80); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - EXPECT_EQ(OK, req->WaitForResult()); -} - -// Test that IP address changes send ERR_NETWORK_CHANGED to pending requests. -TEST_F(HostResolverImplTest, AbortOnIPAddressChanged) { - Request* req = CreateRequest("host1", 70); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - - EXPECT_TRUE(proc_->WaitFor(1u)); - // Triggering an IP address change. - NetworkChangeNotifier::NotifyObserversOfIPAddressChangeForTests(); - MessageLoop::current()->RunUntilIdle(); // Notification happens async. - proc_->SignalAll(); - - EXPECT_EQ(ERR_NETWORK_CHANGED, req->WaitForResult()); - EXPECT_EQ(0u, resolver_->GetHostCache()->size()); -} - -// Obey pool constraints after IP address has changed. -TEST_F(HostResolverImplTest, ObeyPoolConstraintsAfterIPAddressChange) { - // Runs at most one job at a time. - CreateSerialResolver(); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("a")->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("b")->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("c")->Resolve()); - - EXPECT_TRUE(proc_->WaitFor(1u)); - // Triggering an IP address change. - NetworkChangeNotifier::NotifyObserversOfIPAddressChangeForTests(); - MessageLoop::current()->RunUntilIdle(); // Notification happens async. - proc_->SignalMultiple(3u); // Let the false-start go so that we can catch it. - - EXPECT_EQ(ERR_NETWORK_CHANGED, requests_[0]->WaitForResult()); - - EXPECT_EQ(1u, num_running_jobs()); - - EXPECT_FALSE(requests_[1]->completed()); - EXPECT_FALSE(requests_[2]->completed()); - - EXPECT_EQ(OK, requests_[2]->WaitForResult()); - EXPECT_EQ(OK, requests_[1]->result()); -} - -// Tests that a new Request made from the callback of a previously aborted one -// will not be aborted. -TEST_F(HostResolverImplTest, AbortOnlyExistingRequestsOnIPAddressChange) { - struct MyHandler : public Handler { - virtual void Handle(Request* req) OVERRIDE { - // Start new request for a different hostname to ensure that the order - // of jobs in HostResolverImpl is not stable. - std::string hostname; - if (req->index() == 0) - hostname = "zzz"; - else if (req->index() == 1) - hostname = "aaa"; - else if (req->index() == 2) - hostname = "eee"; - else - return; // A request started from within MyHandler. - EXPECT_EQ(ERR_IO_PENDING, CreateRequest(hostname)->Resolve()) << hostname; - } - }; - set_handler(new MyHandler()); - - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("bbb")->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("eee")->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("ccc")->Resolve()); - - // Wait until all are blocked; - EXPECT_TRUE(proc_->WaitFor(3u)); - // Trigger an IP address change. - NetworkChangeNotifier::NotifyObserversOfIPAddressChangeForTests(); - // This should abort all running jobs. - MessageLoop::current()->RunUntilIdle(); - EXPECT_EQ(ERR_NETWORK_CHANGED, requests_[0]->result()); - EXPECT_EQ(ERR_NETWORK_CHANGED, requests_[1]->result()); - EXPECT_EQ(ERR_NETWORK_CHANGED, requests_[2]->result()); - ASSERT_EQ(6u, requests_.size()); - // Unblock all calls to proc. - proc_->SignalMultiple(requests_.size()); - // Run until the re-started requests finish. - EXPECT_EQ(OK, requests_[3]->WaitForResult()); - EXPECT_EQ(OK, requests_[4]->WaitForResult()); - EXPECT_EQ(OK, requests_[5]->WaitForResult()); - // Verify that results of aborted Jobs were not cached. - EXPECT_EQ(6u, proc_->GetCaptureList().size()); - EXPECT_EQ(3u, resolver_->GetHostCache()->size()); -} - -// Tests that when the maximum threads is set to 1, requests are dequeued -// in order of priority. -TEST_F(HostResolverImplTest, HigherPriorityRequestsStartedFirst) { - CreateSerialResolver(); - - // Note that at this point the MockHostResolverProc is blocked, so any - // requests we make will not complete. - CreateRequest("req0", 80, LOW); - CreateRequest("req1", 80, MEDIUM); - CreateRequest("req2", 80, MEDIUM); - CreateRequest("req3", 80, LOW); - CreateRequest("req4", 80, HIGHEST); - CreateRequest("req5", 80, LOW); - CreateRequest("req6", 80, LOW); - CreateRequest("req5", 80, HIGHEST); - - for (size_t i = 0; i < requests_.size(); ++i) { - EXPECT_EQ(ERR_IO_PENDING, requests_[i]->Resolve()) << i; - } - - // Unblock the resolver thread so the requests can run. - proc_->SignalMultiple(requests_.size()); // More than needed. - - // Wait for all the requests to complete succesfully. - for (size_t i = 0; i < requests_.size(); ++i) { - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; - } - - // Since we have restricted to a single concurrent thread in the jobpool, - // the requests should complete in order of priority (with the exception - // of the first request, which gets started right away, since there is - // nothing outstanding). - MockHostResolverProc::CaptureList capture_list = proc_->GetCaptureList(); - ASSERT_EQ(7u, capture_list.size()); - - EXPECT_EQ("req0", capture_list[0].hostname); - EXPECT_EQ("req4", capture_list[1].hostname); - EXPECT_EQ("req5", capture_list[2].hostname); - EXPECT_EQ("req1", capture_list[3].hostname); - EXPECT_EQ("req2", capture_list[4].hostname); - EXPECT_EQ("req3", capture_list[5].hostname); - EXPECT_EQ("req6", capture_list[6].hostname); -} - -// Try cancelling a job which has not started yet. -TEST_F(HostResolverImplTest, CancelPendingRequest) { - CreateSerialResolver(); - - CreateRequest("req0", 80, LOWEST); - CreateRequest("req1", 80, HIGHEST); // Will cancel. - CreateRequest("req2", 80, MEDIUM); - CreateRequest("req3", 80, LOW); - CreateRequest("req4", 80, HIGHEST); // Will cancel. - CreateRequest("req5", 80, LOWEST); // Will cancel. - CreateRequest("req6", 80, MEDIUM); - - // Start all of the requests. - for (size_t i = 0; i < requests_.size(); ++i) { - EXPECT_EQ(ERR_IO_PENDING, requests_[i]->Resolve()) << i; - } - - // Cancel some requests - requests_[1]->Cancel(); - requests_[4]->Cancel(); - requests_[5]->Cancel(); - - // Unblock the resolver thread so the requests can run. - proc_->SignalMultiple(requests_.size()); // More than needed. - - // Wait for all the requests to complete succesfully. - for (size_t i = 0; i < requests_.size(); ++i) { - if (!requests_[i]->pending()) - continue; // Don't wait for the requests we cancelled. - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; - } - - // Verify that they called out the the resolver proc (which runs on the - // resolver thread) in the expected order. - MockHostResolverProc::CaptureList capture_list = proc_->GetCaptureList(); - ASSERT_EQ(4u, capture_list.size()); - - EXPECT_EQ("req0", capture_list[0].hostname); - EXPECT_EQ("req2", capture_list[1].hostname); - EXPECT_EQ("req6", capture_list[2].hostname); - EXPECT_EQ("req3", capture_list[3].hostname); -} - -// Test that when too many requests are enqueued, old ones start to be aborted. -TEST_F(HostResolverImplTest, QueueOverflow) { - CreateSerialResolver(); - - // Allow only 3 queued jobs. - const size_t kMaxPendingJobs = 3u; - resolver_->SetMaxQueuedJobs(kMaxPendingJobs); - - // Note that at this point the MockHostResolverProc is blocked, so any - // requests we make will not complete. - - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("req0", 80, LOWEST)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("req1", 80, HIGHEST)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("req2", 80, MEDIUM)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("req3", 80, MEDIUM)->Resolve()); - - // At this point, there are 3 enqueued jobs. - // Insertion of subsequent requests will cause evictions - // based on priority. - - EXPECT_EQ(ERR_HOST_RESOLVER_QUEUE_TOO_LARGE, - CreateRequest("req4", 80, LOW)->Resolve()); // Evicts itself! - - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("req5", 80, MEDIUM)->Resolve()); - EXPECT_EQ(ERR_HOST_RESOLVER_QUEUE_TOO_LARGE, requests_[2]->result()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("req6", 80, HIGHEST)->Resolve()); - EXPECT_EQ(ERR_HOST_RESOLVER_QUEUE_TOO_LARGE, requests_[3]->result()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("req7", 80, MEDIUM)->Resolve()); - EXPECT_EQ(ERR_HOST_RESOLVER_QUEUE_TOO_LARGE, requests_[5]->result()); - - // Unblock the resolver thread so the requests can run. - proc_->SignalMultiple(4u); - - // The rest should succeed. - EXPECT_EQ(OK, requests_[7]->WaitForResult()); - EXPECT_EQ(OK, requests_[0]->result()); - EXPECT_EQ(OK, requests_[1]->result()); - EXPECT_EQ(OK, requests_[6]->result()); - - // Verify that they called out the the resolver proc (which runs on the - // resolver thread) in the expected order. - MockHostResolverProc::CaptureList capture_list = proc_->GetCaptureList(); - ASSERT_EQ(4u, capture_list.size()); - - EXPECT_EQ("req0", capture_list[0].hostname); - EXPECT_EQ("req1", capture_list[1].hostname); - EXPECT_EQ("req6", capture_list[2].hostname); - EXPECT_EQ("req7", capture_list[3].hostname); - - // Verify that the evicted (incomplete) requests were not cached. - EXPECT_EQ(4u, resolver_->GetHostCache()->size()); - - for (size_t i = 0; i < requests_.size(); ++i) { - EXPECT_TRUE(requests_[i]->completed()) << i; - } -} - -// Tests that after changing the default AddressFamily to IPV4, requests -// with UNSPECIFIED address family map to IPV4. -TEST_F(HostResolverImplTest, SetDefaultAddressFamily_IPv4) { - CreateSerialResolver(); // To guarantee order of resolutions. - - proc_->AddRule("h1", ADDRESS_FAMILY_IPV4, "1.0.0.1"); - proc_->AddRule("h1", ADDRESS_FAMILY_IPV6, "::2"); - - resolver_->SetDefaultAddressFamily(ADDRESS_FAMILY_IPV4); - - CreateRequest("h1", 80, MEDIUM, ADDRESS_FAMILY_UNSPECIFIED); - CreateRequest("h1", 80, MEDIUM, ADDRESS_FAMILY_IPV4); - CreateRequest("h1", 80, MEDIUM, ADDRESS_FAMILY_IPV6); - - // Start all of the requests. - for (size_t i = 0; i < requests_.size(); ++i) { - EXPECT_EQ(ERR_IO_PENDING, requests_[i]->Resolve()) << i; - } - - proc_->SignalMultiple(requests_.size()); - - // Wait for all the requests to complete. - for (size_t i = 0u; i < requests_.size(); ++i) { - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; - } - - // Since the requests all had the same priority and we limited the thread - // count to 1, they should have completed in the same order as they were - // requested. Moreover, request0 and request1 will have been serviced by - // the same job. - - MockHostResolverProc::CaptureList capture_list = proc_->GetCaptureList(); - ASSERT_EQ(2u, capture_list.size()); - - EXPECT_EQ("h1", capture_list[0].hostname); - EXPECT_EQ(ADDRESS_FAMILY_IPV4, capture_list[0].address_family); - - EXPECT_EQ("h1", capture_list[1].hostname); - EXPECT_EQ(ADDRESS_FAMILY_IPV6, capture_list[1].address_family); - - // Now check that the correct resolved IP addresses were returned. - EXPECT_TRUE(requests_[0]->HasOneAddress("1.0.0.1", 80)); - EXPECT_TRUE(requests_[1]->HasOneAddress("1.0.0.1", 80)); - EXPECT_TRUE(requests_[2]->HasOneAddress("::2", 80)); -} - -// This is the exact same test as SetDefaultAddressFamily_IPv4, except the -// default family is set to IPv6 and the family of requests is flipped where -// specified. -TEST_F(HostResolverImplTest, SetDefaultAddressFamily_IPv6) { - CreateSerialResolver(); // To guarantee order of resolutions. - - // Don't use IPv6 replacements here since some systems don't support it. - proc_->AddRule("h1", ADDRESS_FAMILY_IPV4, "1.0.0.1"); - proc_->AddRule("h1", ADDRESS_FAMILY_IPV6, "::2"); - - resolver_->SetDefaultAddressFamily(ADDRESS_FAMILY_IPV6); - - CreateRequest("h1", 80, MEDIUM, ADDRESS_FAMILY_UNSPECIFIED); - CreateRequest("h1", 80, MEDIUM, ADDRESS_FAMILY_IPV6); - CreateRequest("h1", 80, MEDIUM, ADDRESS_FAMILY_IPV4); - - // Start all of the requests. - for (size_t i = 0; i < requests_.size(); ++i) { - EXPECT_EQ(ERR_IO_PENDING, requests_[i]->Resolve()) << i; - } - - proc_->SignalMultiple(requests_.size()); - - // Wait for all the requests to complete. - for (size_t i = 0u; i < requests_.size(); ++i) { - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; - } - - // Since the requests all had the same priority and we limited the thread - // count to 1, they should have completed in the same order as they were - // requested. Moreover, request0 and request1 will have been serviced by - // the same job. - - MockHostResolverProc::CaptureList capture_list = proc_->GetCaptureList(); - ASSERT_EQ(2u, capture_list.size()); - - EXPECT_EQ("h1", capture_list[0].hostname); - EXPECT_EQ(ADDRESS_FAMILY_IPV6, capture_list[0].address_family); - - EXPECT_EQ("h1", capture_list[1].hostname); - EXPECT_EQ(ADDRESS_FAMILY_IPV4, capture_list[1].address_family); - - // Now check that the correct resolved IP addresses were returned. - EXPECT_TRUE(requests_[0]->HasOneAddress("::2", 80)); - EXPECT_TRUE(requests_[1]->HasOneAddress("::2", 80)); - EXPECT_TRUE(requests_[2]->HasOneAddress("1.0.0.1", 80)); -} - -TEST_F(HostResolverImplTest, ResolveFromCache) { - proc_->AddRuleForAllFamilies("just.testing", "192.168.1.42"); - proc_->SignalMultiple(1u); // Need only one. - - HostResolver::RequestInfo info(HostPortPair("just.testing", 80)); - - // First hit will miss the cache. - EXPECT_EQ(ERR_DNS_CACHE_MISS, CreateRequest(info)->ResolveFromCache()); - - // This time, we fetch normally. - EXPECT_EQ(ERR_IO_PENDING, CreateRequest(info)->Resolve()); - EXPECT_EQ(OK, requests_[1]->WaitForResult()); - - // Now we should be able to fetch from the cache. - EXPECT_EQ(OK, CreateRequest(info)->ResolveFromCache()); - EXPECT_TRUE(requests_[2]->HasOneAddress("192.168.1.42", 80)); -} - -// Test the retry attempts simulating host resolver proc that takes too long. -TEST_F(HostResolverImplTest, MultipleAttempts) { - // Total number of attempts would be 3 and we want the 3rd attempt to resolve - // the host. First and second attempt will be forced to sleep until they get - // word that a resolution has completed. The 3rd resolution attempt will try - // to get done ASAP, and won't sleep.. - int kAttemptNumberToResolve = 3; - int kTotalAttempts = 3; - - scoped_refptr<LookupAttemptHostResolverProc> resolver_proc( - new LookupAttemptHostResolverProc( - NULL, kAttemptNumberToResolve, kTotalAttempts)); - - HostResolverImpl::ProcTaskParams params = DefaultParams(resolver_proc.get()); - - // Specify smaller interval for unresponsive_delay_ for HostResolverImpl so - // that unit test runs faster. For example, this test finishes in 1.5 secs - // (500ms * 3). - params.unresponsive_delay = base::TimeDelta::FromMilliseconds(500); - - resolver_.reset( - new HostResolverImpl(HostCache::CreateDefaultCache(), - DefaultLimits(), - params, - NULL)); - - // Resolve "host1". - HostResolver::RequestInfo info(HostPortPair("host1", 70)); - Request* req = CreateRequest(info); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - - // Resolve returns -4 to indicate that 3rd attempt has resolved the host. - EXPECT_EQ(-4, req->WaitForResult()); - - resolver_proc->WaitForAllAttemptsToFinish( - base::TimeDelta::FromMilliseconds(60000)); - MessageLoop::current()->RunUntilIdle(); - - EXPECT_EQ(resolver_proc->total_attempts_resolved(), kTotalAttempts); - EXPECT_EQ(resolver_proc->resolved_attempt_number(), kAttemptNumberToResolve); -} - -DnsConfig CreateValidDnsConfig() { - IPAddressNumber dns_ip; - bool rv = ParseIPLiteralToNumber("192.168.1.0", &dns_ip); - EXPECT_TRUE(rv); - - DnsConfig config; - config.nameservers.push_back(IPEndPoint(dns_ip, dns_protocol::kDefaultPort)); - EXPECT_TRUE(config.IsValid()); - return config; -} - -// Specialized fixture for tests of DnsTask. -class HostResolverImplDnsTest : public HostResolverImplTest { - protected: - virtual void SetUp() OVERRIDE { - AddDnsRule("er", dns_protocol::kTypeA, MockDnsClientRule::FAIL_SYNC); - AddDnsRule("er", dns_protocol::kTypeAAAA, MockDnsClientRule::FAIL_SYNC); - AddDnsRule("nx", dns_protocol::kTypeA, MockDnsClientRule::FAIL_ASYNC); - AddDnsRule("nx", dns_protocol::kTypeAAAA, MockDnsClientRule::FAIL_ASYNC); - AddDnsRule("ok", dns_protocol::kTypeA, MockDnsClientRule::OK); - AddDnsRule("ok", dns_protocol::kTypeAAAA, MockDnsClientRule::OK); - AddDnsRule("4ok", dns_protocol::kTypeA, MockDnsClientRule::OK); - AddDnsRule("4ok", dns_protocol::kTypeAAAA, MockDnsClientRule::EMPTY); - AddDnsRule("6ok", dns_protocol::kTypeA, MockDnsClientRule::EMPTY); - AddDnsRule("6ok", dns_protocol::kTypeAAAA, MockDnsClientRule::OK); - AddDnsRule("4nx", dns_protocol::kTypeA, MockDnsClientRule::OK); - AddDnsRule("4nx", dns_protocol::kTypeAAAA, MockDnsClientRule::FAIL_ASYNC); - CreateResolver(); - } - - void CreateResolver() { - resolver_.reset(new HostResolverImpl( - HostCache::CreateDefaultCache(), - DefaultLimits(), - DefaultParams(proc_), - NULL)); - resolver_->SetDnsClient(CreateMockDnsClient(DnsConfig(), dns_rules_)); - } - - // Adds a rule to |dns_rules_|. Must be followed by |CreateResolver| to apply. - void AddDnsRule(const std::string& prefix, - uint16 qtype, - MockDnsClientRule::Result result) { - dns_rules_.push_back(MockDnsClientRule(prefix, qtype, result)); - } - - void ChangeDnsConfig(const DnsConfig& config) { - NetworkChangeNotifier::SetDnsConfig(config); - // Notification is delivered asynchronously. - MessageLoop::current()->RunUntilIdle(); - } - - MockDnsClientRuleList dns_rules_; -}; - -// TODO(szym): Test AbortAllInProgressJobs due to DnsConfig change. - -// TODO(cbentzel): Test a mix of requests with different HostResolverFlags. - -// Test successful and fallback resolutions in HostResolverImpl::DnsTask. -TEST_F(HostResolverImplDnsTest, DnsTask) { - resolver_->SetDefaultAddressFamily(ADDRESS_FAMILY_IPV4); - - proc_->AddRuleForAllFamilies("er_succeed", "192.168.1.101"); - proc_->AddRuleForAllFamilies("nx_succeed", "192.168.1.102"); - // All other hostnames will fail in proc_. - - // Initially there is no config, so client should not be invoked. - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("ok_fail", 80)->Resolve()); - proc_->SignalMultiple(requests_.size()); - - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, requests_[0]->WaitForResult()); - - ChangeDnsConfig(CreateValidDnsConfig()); - - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("ok_fail", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("er_fail", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("nx_fail", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("er_succeed", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("nx_succeed", 80)->Resolve()); - - proc_->SignalMultiple(requests_.size()); - - for (size_t i = 1; i < requests_.size(); ++i) - EXPECT_NE(ERR_UNEXPECTED, requests_[i]->WaitForResult()) << i; - - EXPECT_EQ(OK, requests_[1]->result()); - // Resolved by MockDnsClient. - EXPECT_TRUE(requests_[1]->HasOneAddress("127.0.0.1", 80)); - // Fallback to ProcTask. - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, requests_[2]->result()); - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, requests_[3]->result()); - EXPECT_EQ(OK, requests_[4]->result()); - EXPECT_TRUE(requests_[4]->HasOneAddress("192.168.1.101", 80)); - EXPECT_EQ(OK, requests_[5]->result()); - EXPECT_TRUE(requests_[5]->HasOneAddress("192.168.1.102", 80)); -} - -TEST_F(HostResolverImplDnsTest, DnsTaskUnspec) { - ChangeDnsConfig(CreateValidDnsConfig()); - - proc_->AddRuleForAllFamilies("4nx", "192.168.1.101"); - // All other hostnames will fail in proc_. - - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("ok", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("4ok", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("6ok", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("4nx", 80)->Resolve()); - - proc_->SignalMultiple(requests_.size()); - - for (size_t i = 0; i < requests_.size(); ++i) - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; - - EXPECT_EQ(2u, requests_[0]->NumberOfAddresses()); - EXPECT_TRUE(requests_[0]->HasAddress("127.0.0.1", 80)); - EXPECT_TRUE(requests_[0]->HasAddress("::1", 80)); - EXPECT_EQ(1u, requests_[1]->NumberOfAddresses()); - EXPECT_TRUE(requests_[1]->HasAddress("127.0.0.1", 80)); - EXPECT_EQ(1u, requests_[2]->NumberOfAddresses()); - EXPECT_TRUE(requests_[2]->HasAddress("::1", 80)); - EXPECT_EQ(1u, requests_[3]->NumberOfAddresses()); - EXPECT_TRUE(requests_[3]->HasAddress("192.168.1.101", 80)); -} - -TEST_F(HostResolverImplDnsTest, ServeFromHosts) { - // Initially, use empty HOSTS file. - DnsConfig config = CreateValidDnsConfig(); - ChangeDnsConfig(config); - - proc_->AddRuleForAllFamilies("", ""); // Default to failures. - proc_->SignalMultiple(1u); // For the first request which misses. - - Request* req0 = CreateRequest("er_ipv4", 80); - EXPECT_EQ(ERR_IO_PENDING, req0->Resolve()); - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, req0->WaitForResult()); - - IPAddressNumber local_ipv4, local_ipv6; - ASSERT_TRUE(ParseIPLiteralToNumber("127.0.0.1", &local_ipv4)); - ASSERT_TRUE(ParseIPLiteralToNumber("::1", &local_ipv6)); - - DnsHosts hosts; - hosts[DnsHostsKey("er_ipv4", ADDRESS_FAMILY_IPV4)] = local_ipv4; - hosts[DnsHostsKey("er_ipv6", ADDRESS_FAMILY_IPV6)] = local_ipv6; - hosts[DnsHostsKey("er_both", ADDRESS_FAMILY_IPV4)] = local_ipv4; - hosts[DnsHostsKey("er_both", ADDRESS_FAMILY_IPV6)] = local_ipv6; - - // Update HOSTS file. - config.hosts = hosts; - ChangeDnsConfig(config); - - Request* req1 = CreateRequest("er_ipv4", 80); - EXPECT_EQ(OK, req1->Resolve()); - EXPECT_TRUE(req1->HasOneAddress("127.0.0.1", 80)); - - Request* req2 = CreateRequest("er_ipv6", 80); - EXPECT_EQ(OK, req2->Resolve()); - EXPECT_TRUE(req2->HasOneAddress("::1", 80)); - - Request* req3 = CreateRequest("er_both", 80); - EXPECT_EQ(OK, req3->Resolve()); - EXPECT_TRUE(req3->HasOneAddress("127.0.0.1", 80) || - req3->HasOneAddress("::1", 80)); - - // Requests with specified AddressFamily. - Request* req4 = CreateRequest("er_ipv4", 80, MEDIUM, ADDRESS_FAMILY_IPV4); - EXPECT_EQ(OK, req4->Resolve()); - EXPECT_TRUE(req4->HasOneAddress("127.0.0.1", 80)); - - Request* req5 = CreateRequest("er_ipv6", 80, MEDIUM, ADDRESS_FAMILY_IPV6); - EXPECT_EQ(OK, req5->Resolve()); - EXPECT_TRUE(req5->HasOneAddress("::1", 80)); - - // Request with upper case. - Request* req6 = CreateRequest("er_IPV4", 80); - EXPECT_EQ(OK, req6->Resolve()); - EXPECT_TRUE(req6->HasOneAddress("127.0.0.1", 80)); -} - -TEST_F(HostResolverImplDnsTest, BypassDnsTask) { - ChangeDnsConfig(CreateValidDnsConfig()); - - proc_->AddRuleForAllFamilies("", ""); // Default to failures. - - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("ok.local", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("ok.local.", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("oklocal", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("oklocal.", 80)->Resolve()); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest("ok", 80)->Resolve()); - - proc_->SignalMultiple(requests_.size()); - - for (size_t i = 0; i < 2; ++i) - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, requests_[i]->WaitForResult()) << i; - - for (size_t i = 2; i < requests_.size(); ++i) - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; -} - -TEST_F(HostResolverImplDnsTest, DisableDnsClientOnPersistentFailure) { - ChangeDnsConfig(CreateValidDnsConfig()); - - proc_->AddRuleForAllFamilies("", ""); // Default to failures. - - // Check that DnsTask works. - Request* req = CreateRequest("ok_1", 80); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - EXPECT_EQ(OK, req->WaitForResult()); - - for (unsigned i = 0; i < 20; ++i) { - // Use custom names to require separate Jobs. - std::string hostname = base::StringPrintf("err_%u", i); - // Ensure fallback to ProcTask succeeds. - proc_->AddRuleForAllFamilies(hostname, "192.168.1.101"); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest(hostname, 80)->Resolve()) << i; - } - - proc_->SignalMultiple(requests_.size()); - - for (size_t i = 0; i < requests_.size(); ++i) - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; - - ASSERT_FALSE(proc_->HasBlockedRequests()); - - // DnsTask should be disabled by now. - req = CreateRequest("ok_2", 80); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - proc_->SignalMultiple(1u); - EXPECT_EQ(ERR_NAME_NOT_RESOLVED, req->WaitForResult()); - - // Check that it is re-enabled after DNS change. - ChangeDnsConfig(CreateValidDnsConfig()); - req = CreateRequest("ok_3", 80); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - EXPECT_EQ(OK, req->WaitForResult()); -} - -TEST_F(HostResolverImplDnsTest, DontDisableDnsClientOnSporadicFailure) { - ChangeDnsConfig(CreateValidDnsConfig()); - - // |proc_| defaults to successes. - - // 20 failures interleaved with 20 successes. - for (unsigned i = 0; i < 40; ++i) { - // Use custom names to require separate Jobs. - std::string hostname = (i % 2) == 0 ? base::StringPrintf("err_%u", i) - : base::StringPrintf("ok_%u", i); - EXPECT_EQ(ERR_IO_PENDING, CreateRequest(hostname, 80)->Resolve()) << i; - } - - proc_->SignalMultiple(requests_.size()); - - for (size_t i = 0; i < requests_.size(); ++i) - EXPECT_EQ(OK, requests_[i]->WaitForResult()) << i; - - // Make |proc_| default to failures. - proc_->AddRuleForAllFamilies("", ""); - - // DnsTask should still be enabled. - Request* req = CreateRequest("ok_last", 80); - EXPECT_EQ(ERR_IO_PENDING, req->Resolve()); - EXPECT_EQ(OK, req->WaitForResult()); -} - -} // namespace net |