net/proxy/mojo_proxy_resolver_impl.cc

// Copyright 2015 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/proxy/mojo_proxy_resolver_impl.h"

#include "base/stl_util.h"
#include "mojo/common/common_type_converters.h"
#include "net/base/net_errors.h"
#include "net/base/net_log.h"
#include "net/proxy/mojo_type_converters.h"
#include "net/proxy/proxy_info.h"
#include "net/proxy/proxy_resolver.h"
#include "net/proxy/proxy_resolver_script_data.h"

namespace net {

class MojoProxyResolverImpl::Job : public mojo::ErrorHandler {
 public:
  Job(interfaces::ProxyResolverRequestClientPtr client,
      MojoProxyResolverImpl* resolver,
      const GURL& url);
  ~Job() override;

  void Start();

 private:
  // mojo::ErrorHandler override.
  // This is invoked in response to the client disconnecting, indicating
  // cancellation.
  void OnConnectionError() override;

  void GetProxyDone(int error);

  MojoProxyResolverImpl* resolver_;

  interfaces::ProxyResolverRequestClientPtr client_;
  ProxyInfo result_;
  GURL url_;
  net::ProxyResolver::RequestHandle request_handle_;

  DISALLOW_COPY_AND_ASSIGN(Job);
};

MojoProxyResolverImpl::MojoProxyResolverImpl(
    scoped_ptr<net::ProxyResolver> resolver)
    : resolver_(resolver.Pass()) {
  DCHECK(resolver_->expects_pac_bytes());
}

MojoProxyResolverImpl::~MojoProxyResolverImpl() {
  if (!set_pac_script_requests_.empty())
    resolver_->CancelSetPacScript();
  STLDeleteElements(&resolve_jobs_);
}

void MojoProxyResolverImpl::SetPacScript(
    const mojo::String& data,
    const mojo::Callback<void(int32_t)>& callback) {
  DVLOG(1) << "SetPacScript(" << data << ")";
  set_pac_script_requests_.push(
      SetPacScriptRequest(ProxyResolverScriptData::FromUTF8(data), callback));
  if (set_pac_script_requests_.size() == 1)
    StartSetPacScript();
}

void MojoProxyResolverImpl::GetProxyForUrl(
    const mojo::String& url,
    interfaces::ProxyResolverRequestClientPtr client) {
  DVLOG(1) << "GetProxyForUrl(" << url << ")";
  Job* job = new Job(client.Pass(), this, url.To<GURL>());
  bool inserted = resolve_jobs_.insert(job).second;
  DCHECK(inserted);
  job->Start();
}

void MojoProxyResolverImpl::DeleteJob(Job* job) {
  size_t num_erased = resolve_jobs_.erase(job);
  DCHECK(num_erased);
  delete job;
}

void MojoProxyResolverImpl::StartSetPacScript() {
  DCHECK(!set_pac_script_requests_.empty());
  int result = resolver_->SetPacScript(
      set_pac_script_requests_.front().script_data,
      base::Bind(&MojoProxyResolverImpl::SetPacScriptDone,
                 base::Unretained(this)));
  if (result != ERR_IO_PENDING)
    SetPacScriptDone(result);
}

void MojoProxyResolverImpl::SetPacScriptDone(int result) {
  DVLOG(1) << "SetPacScript finished with error " << result;
  DCHECK(!set_pac_script_requests_.empty());
  set_pac_script_requests_.front().callback.Run(result);
  set_pac_script_requests_.pop();
  if (!set_pac_script_requests_.empty())
    StartSetPacScript();
}

MojoProxyResolverImpl::Job::Job(
    interfaces::ProxyResolverRequestClientPtr client,
    MojoProxyResolverImpl* resolver,
    const GURL& url)
    : resolver_(resolver),
      client_(client.Pass()),
      url_(url),
      request_handle_(nullptr) {
}

MojoProxyResolverImpl::Job::~Job() {
  if (request_handle_)
    resolver_->resolver_->CancelRequest(request_handle_);
}

void MojoProxyResolverImpl::Job::Start() {
  int result = resolver_->resolver_->GetProxyForURL(
      url_, &result_, base::Bind(&Job::GetProxyDone, base::Unretained(this)),
      &request_handle_, BoundNetLog());
  if (result != ERR_IO_PENDING) {
    GetProxyDone(result);
    return;
  }
  client_.set_error_handler(this);
}

void MojoProxyResolverImpl::Job::GetProxyDone(int error) {
  request_handle_ = nullptr;
  DVLOG(1) << "GetProxyForUrl(" << url_ << ") finished with error " << error
           << ". " << result_.proxy_list().size() << " Proxies returned:";
  for (const auto& proxy : result_.proxy_list().GetAll()) {
    DVLOG(1) << proxy.ToURI();
  }
  mojo::Array<interfaces::ProxyServerPtr> result;
  if (error == OK) {
    result = mojo::Array<interfaces::ProxyServerPtr>::From(
        result_.proxy_list().GetAll());
  }
  client_->ReportResult(error, result.Pass());
  resolver_->DeleteJob(this);
}

void MojoProxyResolverImpl::Job::OnConnectionError() {
  resolver_->DeleteJob(this);
}

MojoProxyResolverImpl::SetPacScriptRequest::SetPacScriptRequest(
    const scoped_refptr<ProxyResolverScriptData>& script_data,
    const mojo::Callback<void(int32_t)>& callback)
    : script_data(script_data), callback(callback) {
}

MojoProxyResolverImpl::SetPacScriptRequest::~SetPacScriptRequest() = default;

}  // namespace net