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 #include "extensions/browser/api/dns/dns_api.h" 6 7 #include "base/bind.h" 8 #include "base/values.h" 9 #include "content/public/browser/browser_context.h" 10 #include "content/public/browser/browser_thread.h" 11 #include "content/public/browser/resource_context.h" 12 #include "extensions/browser/api/dns/host_resolver_wrapper.h" 13 #include "extensions/common/api/dns.h" 14 #include "net/base/host_port_pair.h" 15 #include "net/base/net_errors.h" 16 #include "net/base/net_log.h" 17 18 using content::BrowserThread; 19 using extensions::core_api::dns::ResolveCallbackResolveInfo; 20 21 namespace Resolve = extensions::core_api::dns::Resolve; 22 23 namespace extensions { 24 25 DnsResolveFunction::DnsResolveFunction() 26 : resource_context_(NULL), 27 response_(false), 28 request_handle_(new net::HostResolver::RequestHandle()), 29 addresses_(new net::AddressList) {} 30 31 DnsResolveFunction::~DnsResolveFunction() {} 32 33 bool DnsResolveFunction::RunAsync() { 34 scoped_ptr<Resolve::Params> params(Resolve::Params::Create(*args_)); 35 EXTENSION_FUNCTION_VALIDATE(params.get()); 36 37 hostname_ = params->hostname; 38 resource_context_ = browser_context()->GetResourceContext(); 39 40 bool result = BrowserThread::PostTask( 41 BrowserThread::IO, 42 FROM_HERE, 43 base::Bind(&DnsResolveFunction::WorkOnIOThread, this)); 44 DCHECK(result); 45 return true; 46 } 47 48 void DnsResolveFunction::WorkOnIOThread() { 49 DCHECK_CURRENTLY_ON(BrowserThread::IO); 50 51 net::HostResolver* host_resolver = 52 HostResolverWrapper::GetInstance()->GetHostResolver( 53 resource_context_->GetHostResolver()); 54 DCHECK(host_resolver); 55 56 // Yes, we are passing zero as the port. There are some interesting but not 57 // presently relevant reasons why HostResolver asks for the port of the 58 // hostname you'd like to resolve, even though it doesn't use that value in 59 // determining its answer. 60 net::HostPortPair host_port_pair(hostname_, 0); 61 62 net::HostResolver::RequestInfo request_info(host_port_pair); 63 int resolve_result = host_resolver->Resolve( 64 request_info, 65 net::DEFAULT_PRIORITY, 66 addresses_.get(), 67 base::Bind(&DnsResolveFunction::OnLookupFinished, this), 68 request_handle_.get(), 69 net::BoundNetLog()); 70 71 // Balanced in OnLookupFinished. 72 AddRef(); 73 74 if (resolve_result != net::ERR_IO_PENDING) 75 OnLookupFinished(resolve_result); 76 } 77 78 void DnsResolveFunction::RespondOnUIThread() { 79 DCHECK_CURRENTLY_ON(BrowserThread::UI); 80 SendResponse(response_); 81 } 82 83 void DnsResolveFunction::OnLookupFinished(int resolve_result) { 84 scoped_ptr<ResolveCallbackResolveInfo> resolve_info( 85 new ResolveCallbackResolveInfo()); 86 resolve_info->result_code = resolve_result; 87 if (resolve_result == net::OK) { 88 DCHECK(!addresses_->empty()); 89 resolve_info->address.reset( 90 new std::string(addresses_->front().ToStringWithoutPort())); 91 } 92 results_ = Resolve::Results::Create(*resolve_info); 93 response_ = true; 94 95 bool post_task_result = BrowserThread::PostTask( 96 BrowserThread::UI, 97 FROM_HERE, 98 base::Bind(&DnsResolveFunction::RespondOnUIThread, this)); 99 DCHECK(post_task_result); 100 101 Release(); // Added in WorkOnIOThread(). 102 } 103 104 } // namespace extensions 105
