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.
5 #include "extensions/browser/api/dns/dns_api.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"
18 using content::BrowserThread;
19 using extensions::core_api::dns::ResolveCallbackResolveInfo;
21 namespace Resolve = extensions::core_api::dns::Resolve;
23 namespace extensions {
25 DnsResolveFunction::DnsResolveFunction()
26 : resource_context_(NULL),
28 request_handle_(new net::HostResolver::RequestHandle()),
29 addresses_(new net::AddressList) {}
31 DnsResolveFunction::~DnsResolveFunction() {}
33 bool DnsResolveFunction::RunAsync() {
34 scoped_ptr<Resolve::Params> params(Resolve::Params::Create(*args_));
35 EXTENSION_FUNCTION_VALIDATE(params.get());
37 hostname_ = params->hostname;
38 resource_context_ = browser_context()->GetResourceContext();
40 bool result = BrowserThread::PostTask(
43 base::Bind(&DnsResolveFunction::WorkOnIOThread, this));
48 void DnsResolveFunction::WorkOnIOThread() {
49 DCHECK_CURRENTLY_ON(BrowserThread::IO);
51 net::HostResolver* host_resolver =
52 HostResolverWrapper::GetInstance()->GetHostResolver(
53 resource_context_->GetHostResolver());
54 DCHECK(host_resolver);
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);
62 net::HostResolver::RequestInfo request_info(host_port_pair);
63 int resolve_result = host_resolver->Resolve(
65 net::DEFAULT_PRIORITY,
67 base::Bind(&DnsResolveFunction::OnLookupFinished, this),
68 request_handle_.get(),
71 // Balanced in OnLookupFinished.
74 if (resolve_result != net::ERR_IO_PENDING)
75 OnLookupFinished(resolve_result);
78 void DnsResolveFunction::RespondOnUIThread() {
79 DCHECK_CURRENTLY_ON(BrowserThread::UI);
80 SendResponse(response_);
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()));
92 results_ = Resolve::Results::Create(*resolve_info);
95 bool post_task_result = BrowserThread::PostTask(
98 base::Bind(&DnsResolveFunction::RespondOnUIThread, this));
99 DCHECK(post_task_result);
101 Release(); // Added in WorkOnIOThread().
104 } // namespace extensions