net/proxy/mojo_proxy_resolver_impl.cc

   1 // Copyright 2015 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 "net/proxy/mojo_proxy_resolver_impl.h"
   6
   7 #include "base/stl_util.h"
   8 #include "mojo/common/url_type_converters.h"
   9 #include "net/base/net_errors.h"
  10 #include "net/log/net_log.h"
  11 #include "net/proxy/mojo_proxy_type_converters.h"
  12 #include "net/proxy/proxy_info.h"
  13 #include "net/proxy/proxy_resolver_script_data.h"
  14
  15 namespace net {
  16
  17 class MojoProxyResolverImpl::Job {
  18  public:
  19   Job(interfaces::ProxyResolverRequestClientPtr client,
  20       MojoProxyResolverImpl* resolver,
  21       const GURL& url);
  22   ~Job();
  23
  24   void Start();
  25
  26   net::ProxyResolver::RequestHandle request_handle() { return request_handle_; }
  27
  28  private:
  29   // Mojo error handler. This is invoked in response to the client
  30   // disconnecting, indicating cancellation.
  31   void OnConnectionError();
  32
  33   void GetProxyDone(int error);
  34
  35   MojoProxyResolverImpl* resolver_;
  36
  37   interfaces::ProxyResolverRequestClientPtr client_;
  38   ProxyInfo result_;
  39   GURL url_;
  40   net::ProxyResolver::RequestHandle request_handle_;
  41   bool done_;
  42
  43   DISALLOW_COPY_AND_ASSIGN(Job);
  44 };
  45
  46 MojoProxyResolverImpl::MojoProxyResolverImpl(
  47     scoped_ptr<net::ProxyResolver> resolver)
  48     : resolver_(resolver.Pass()) {
  49 }
  50
  51 MojoProxyResolverImpl::~MojoProxyResolverImpl() {
  52   STLDeleteElements(&resolve_jobs_);
  53 }
  54
  55 void MojoProxyResolverImpl::GetProxyForUrl(
  56     const mojo::String& url,
  57     interfaces::ProxyResolverRequestClientPtr client) {
  58   DVLOG(1) << "GetProxyForUrl(" << url << ")";
  59   Job* job = new Job(client.Pass(), this, url.To<GURL>());
  60   bool inserted = resolve_jobs_.insert(job).second;
  61   DCHECK(inserted);
  62   job->Start();
  63 }
  64
  65 void MojoProxyResolverImpl::DeleteJob(Job* job) {
  66   if (job->request_handle())
  67     request_handle_to_job_.erase(job->request_handle());
  68
  69   size_t num_erased = resolve_jobs_.erase(job);
  70   DCHECK(num_erased);
  71   delete job;
  72 }
  73
  74 MojoProxyResolverImpl::Job::Job(
  75     interfaces::ProxyResolverRequestClientPtr client,
  76     MojoProxyResolverImpl* resolver,
  77     const GURL& url)
  78     : resolver_(resolver),
  79       client_(client.Pass()),
  80       url_(url),
  81       request_handle_(nullptr),
  82       done_(false) {
  83 }
  84
  85 MojoProxyResolverImpl::Job::~Job() {
  86   if (request_handle_ && !done_)
  87     resolver_->resolver_->CancelRequest(request_handle_);
  88 }
  89
  90 void MojoProxyResolverImpl::Job::Start() {
  91   int result = resolver_->resolver_->GetProxyForURL(
  92       url_, &result_, base::Bind(&Job::GetProxyDone, base::Unretained(this)),
  93       &request_handle_, BoundNetLog());
  94   if (result != ERR_IO_PENDING) {
  95     GetProxyDone(result);
  96     return;
  97   }
  98   client_.set_connection_error_handler(base::Bind(
  99       &MojoProxyResolverImpl::Job::OnConnectionError, base::Unretained(this)));
 100   resolver_->request_handle_to_job_.insert(
 101       std::make_pair(request_handle_, this));
 102 }
 103
 104 void MojoProxyResolverImpl::Job::GetProxyDone(int error) {
 105   done_ = true;
 106   DVLOG(1) << "GetProxyForUrl(" << url_ << ") finished with error " << error
 107            << ". " << result_.proxy_list().size() << " Proxies returned:";
 108   for (const auto& proxy : result_.proxy_list().GetAll()) {
 109     DVLOG(1) << proxy.ToURI();
 110   }
 111   mojo::Array<interfaces::ProxyServerPtr> result;
 112   if (error == OK) {
 113     result = mojo::Array<interfaces::ProxyServerPtr>::From(
 114         result_.proxy_list().GetAll());
 115   }
 116   client_->ReportResult(error, result.Pass());
 117   resolver_->DeleteJob(this);
 118 }
 119
 120 void MojoProxyResolverImpl::Job::OnConnectionError() {
 121   resolver_->DeleteJob(this);
 122 }
 123
 124 }  // namespace net