blimp/net/tcp_engine_transport.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 "blimp/net/tcp_engine_transport.h"

#include <memory>
#include <utility>

#include "base/callback.h"
#include "base/callback_helpers.h"
#include "base/location.h"
#include "base/memory/ptr_util.h"
#include "base/threading/thread_task_runner_handle.h"
#include "blimp/net/message_port.h"
#include "blimp/net/tcp_connection.h"
#include "net/log/net_log_source.h"
#include "net/socket/stream_socket.h"
#include "net/socket/tcp_server_socket.h"

namespace blimp {

TCPEngineTransport::TCPEngineTransport(const net::IPEndPoint& address,
                                       net::NetLog* net_log)
    : address_(address), net_log_(net_log), weak_factory_(this) {}

TCPEngineTransport::~TCPEngineTransport() {}

void TCPEngineTransport::Connect(const net::CompletionCallback& callback) {
  DCHECK(!accepted_socket_);
  DCHECK(!callback.is_null());

  if (!server_socket_) {
    server_socket_.reset(
        new net::TCPServerSocket(net_log_, net::NetLogSource()));
    int result = server_socket_->Listen(address_, 5);
    if (result != net::OK) {
      server_socket_.reset();
      base::ThreadTaskRunnerHandle::Get()->PostTask(
          FROM_HERE, base::Bind(callback, result));
      return;
    }
  }

  net::CompletionCallback accept_callback = base::Bind(
      &TCPEngineTransport::OnTCPConnectAccepted, weak_factory_.GetWeakPtr());

  connect_callback_ = callback;
  int result = server_socket_->Accept(&accepted_socket_, accept_callback);
  if (result == net::ERR_IO_PENDING) {
    return;
  }

  base::ThreadTaskRunnerHandle::Get()->PostTask(
      FROM_HERE, base::Bind(&TCPEngineTransport::OnTCPConnectAccepted,
                            weak_factory_.GetWeakPtr(), result));
}

std::unique_ptr<MessagePort> TCPEngineTransport::TakeMessagePort() {
  DCHECK(connect_callback_.is_null());
  DCHECK(accepted_socket_);
  return MessagePort::CreateForStreamSocketWithCompression(
      std::move(accepted_socket_));
}

std::unique_ptr<BlimpConnection> TCPEngineTransport::MakeConnection() {
  return base::MakeUnique<TCPConnection>(TakeMessagePort());
}

const char* TCPEngineTransport::GetName() const {
  return "TCP";
}

void TCPEngineTransport::GetLocalAddress(net::IPEndPoint* address) const {
  DCHECK(server_socket_);
  server_socket_->GetLocalAddress(address);
}

void TCPEngineTransport::OnTCPConnectAccepted(int result) {
  DCHECK_NE(net::ERR_IO_PENDING, result);
  DCHECK(accepted_socket_);
  if (result != net::OK) {
    accepted_socket_.reset();
  }
  base::ResetAndReturn(&connect_callback_).Run(result);
}

}  // namespace blimp