1 // Copyright 2014 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef CHROMEOS_NETWORK_HOST_RESOLVER_IMPL_CHROMEOS_H_ 6 #define CHROMEOS_NETWORK_HOST_RESOLVER_IMPL_CHROMEOS_H_ 7 8 #include "base/memory/ref_counted.h" 9 #include "base/threading/thread_checker.h" 10 #include "chromeos/chromeos_export.h" 11 #include "net/dns/host_resolver_impl.h" 12 13 namespace base { 14 class MessageLoopProxy; 15 } 16 17 namespace chromeos { 18 class NetworkStateHandler; 19 } 20 21 namespace chromeos { 22 23 // HostResolverImplChromeOS overrides HostResolverImpl::Resolve in order to 24 // provide the correct IP addresses for localhost using the chromeos 25 // NetworkHandler interface. ('hostname' only returns 'localhost' on cros). 26 27 class CHROMEOS_EXPORT HostResolverImplChromeOS : public net::HostResolverImpl { 28 public: 29 // ChromeOS specific implementation of HostResolver::CreateSystemResolver. 30 // Assumes NetworkHandler has been initialized. 31 // This is expected to be constructed on the same thread that Resolve() is 32 // called from, i.e. the IO thread, which is presumed to differ from the 33 // thread that NetworkStateHandler is called on, i.e. the UI thread. 34 static scoped_ptr<net::HostResolver> CreateSystemResolver( 35 const Options& options, 36 net::NetLog* net_log); 37 38 // Creates a host resolver instance for testing. 39 static scoped_ptr<net::HostResolver> CreateHostResolverForTest( 40 scoped_refptr<base::MessageLoopProxy> network_handler_message_loop, 41 NetworkStateHandler* network_state_handler); 42 43 virtual ~HostResolverImplChromeOS(); 44 45 // HostResolverImpl 46 virtual int Resolve(const RequestInfo& info, 47 net::RequestPriority priority, 48 net::AddressList* addresses, 49 const net::CompletionCallback& callback, 50 RequestHandle* out_req, 51 const net::BoundNetLog& source_net_log) OVERRIDE; 52 53 private: 54 friend class net::HostResolver; 55 class NetworkObserver; 56 57 HostResolverImplChromeOS( 58 scoped_refptr<base::MessageLoopProxy> network_handler_message_loop, 59 NetworkStateHandler* network_state_handler, 60 const Options& options, 61 net::NetLog* net_log); 62 63 void SetIPAddresses(const std::string& ipv4_address, 64 const std::string& ipv6_address); 65 66 bool ResolveLocalIPAddress(const RequestInfo& info, 67 net::AddressList* addresses); 68 69 std::string ipv4_address_; 70 std::string ipv6_address_; 71 base::ThreadChecker thread_checker_; 72 scoped_refptr<base::MessageLoopProxy> network_handler_message_loop_; 73 base::WeakPtrFactory<HostResolverImplChromeOS> weak_ptr_factory_; 74 75 DISALLOW_COPY_AND_ASSIGN(HostResolverImplChromeOS); 76 }; 77 78 } // namespace chromeos 79 80 #endif // CHROMEOS_NETWORK_HOST_RESOLVER_IMPL_CHROMEOS_H_ 81