| Index: net/socket/socks5_client_socket.cc
|
| diff --git a/net/socket/socks5_client_socket.cc b/net/socket/socks5_client_socket.cc
|
| deleted file mode 100644
|
| index 6d67913b6e6c256079f224c9872efa63945044e2..0000000000000000000000000000000000000000
|
| --- a/net/socket/socks5_client_socket.cc
|
| +++ /dev/null
|
| @@ -1,506 +0,0 @@
|
| -// Copyright (c) 2012 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/socket/socks5_client_socket.h"
|
| -
|
| -#include "base/basictypes.h"
|
| -#include "base/callback_helpers.h"
|
| -#include "base/compiler_specific.h"
|
| -#include "base/format_macros.h"
|
| -#include "base/strings/string_util.h"
|
| -#include "base/sys_byteorder.h"
|
| -#include "base/trace_event/trace_event.h"
|
| -#include "net/base/io_buffer.h"
|
| -#include "net/base/net_log.h"
|
| -#include "net/base/net_util.h"
|
| -#include "net/socket/client_socket_handle.h"
|
| -
|
| -namespace net {
|
| -
|
| -const unsigned int SOCKS5ClientSocket::kGreetReadHeaderSize = 2;
|
| -const unsigned int SOCKS5ClientSocket::kWriteHeaderSize = 10;
|
| -const unsigned int SOCKS5ClientSocket::kReadHeaderSize = 5;
|
| -const uint8 SOCKS5ClientSocket::kSOCKS5Version = 0x05;
|
| -const uint8 SOCKS5ClientSocket::kTunnelCommand = 0x01;
|
| -const uint8 SOCKS5ClientSocket::kNullByte = 0x00;
|
| -
|
| -static_assert(sizeof(struct in_addr) == 4, "incorrect system size of IPv4");
|
| -static_assert(sizeof(struct in6_addr) == 16, "incorrect system size of IPv6");
|
| -
|
| -SOCKS5ClientSocket::SOCKS5ClientSocket(
|
| - scoped_ptr<ClientSocketHandle> transport_socket,
|
| - const HostResolver::RequestInfo& req_info)
|
| - : io_callback_(base::Bind(&SOCKS5ClientSocket::OnIOComplete,
|
| - base::Unretained(this))),
|
| - transport_(transport_socket.Pass()),
|
| - next_state_(STATE_NONE),
|
| - completed_handshake_(false),
|
| - bytes_sent_(0),
|
| - bytes_received_(0),
|
| - read_header_size(kReadHeaderSize),
|
| - was_ever_used_(false),
|
| - host_request_info_(req_info),
|
| - net_log_(transport_->socket()->NetLog()) {
|
| -}
|
| -
|
| -SOCKS5ClientSocket::~SOCKS5ClientSocket() {
|
| - Disconnect();
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::Connect(const CompletionCallback& callback) {
|
| - DCHECK(transport_.get());
|
| - DCHECK(transport_->socket());
|
| - DCHECK_EQ(STATE_NONE, next_state_);
|
| - DCHECK(user_callback_.is_null());
|
| -
|
| - // If already connected, then just return OK.
|
| - if (completed_handshake_)
|
| - return OK;
|
| -
|
| - net_log_.BeginEvent(NetLog::TYPE_SOCKS5_CONNECT);
|
| -
|
| - next_state_ = STATE_GREET_WRITE;
|
| - buffer_.clear();
|
| -
|
| - int rv = DoLoop(OK);
|
| - if (rv == ERR_IO_PENDING) {
|
| - user_callback_ = callback;
|
| - } else {
|
| - net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SOCKS5_CONNECT, rv);
|
| - }
|
| - return rv;
|
| -}
|
| -
|
| -void SOCKS5ClientSocket::Disconnect() {
|
| - completed_handshake_ = false;
|
| - transport_->socket()->Disconnect();
|
| -
|
| - // Reset other states to make sure they aren't mistakenly used later.
|
| - // These are the states initialized by Connect().
|
| - next_state_ = STATE_NONE;
|
| - user_callback_.Reset();
|
| -}
|
| -
|
| -bool SOCKS5ClientSocket::IsConnected() const {
|
| - return completed_handshake_ && transport_->socket()->IsConnected();
|
| -}
|
| -
|
| -bool SOCKS5ClientSocket::IsConnectedAndIdle() const {
|
| - return completed_handshake_ && transport_->socket()->IsConnectedAndIdle();
|
| -}
|
| -
|
| -const BoundNetLog& SOCKS5ClientSocket::NetLog() const {
|
| - return net_log_;
|
| -}
|
| -
|
| -void SOCKS5ClientSocket::SetSubresourceSpeculation() {
|
| - if (transport_.get() && transport_->socket()) {
|
| - transport_->socket()->SetSubresourceSpeculation();
|
| - } else {
|
| - NOTREACHED();
|
| - }
|
| -}
|
| -
|
| -void SOCKS5ClientSocket::SetOmniboxSpeculation() {
|
| - if (transport_.get() && transport_->socket()) {
|
| - transport_->socket()->SetOmniboxSpeculation();
|
| - } else {
|
| - NOTREACHED();
|
| - }
|
| -}
|
| -
|
| -bool SOCKS5ClientSocket::WasEverUsed() const {
|
| - return was_ever_used_;
|
| -}
|
| -
|
| -bool SOCKS5ClientSocket::UsingTCPFastOpen() const {
|
| - if (transport_.get() && transport_->socket()) {
|
| - return transport_->socket()->UsingTCPFastOpen();
|
| - }
|
| - NOTREACHED();
|
| - return false;
|
| -}
|
| -
|
| -bool SOCKS5ClientSocket::WasNpnNegotiated() const {
|
| - if (transport_.get() && transport_->socket()) {
|
| - return transport_->socket()->WasNpnNegotiated();
|
| - }
|
| - NOTREACHED();
|
| - return false;
|
| -}
|
| -
|
| -NextProto SOCKS5ClientSocket::GetNegotiatedProtocol() const {
|
| - if (transport_.get() && transport_->socket()) {
|
| - return transport_->socket()->GetNegotiatedProtocol();
|
| - }
|
| - NOTREACHED();
|
| - return kProtoUnknown;
|
| -}
|
| -
|
| -bool SOCKS5ClientSocket::GetSSLInfo(SSLInfo* ssl_info) {
|
| - if (transport_.get() && transport_->socket()) {
|
| - return transport_->socket()->GetSSLInfo(ssl_info);
|
| - }
|
| - NOTREACHED();
|
| - return false;
|
| -
|
| -}
|
| -
|
| -// Read is called by the transport layer above to read. This can only be done
|
| -// if the SOCKS handshake is complete.
|
| -int SOCKS5ClientSocket::Read(IOBuffer* buf, int buf_len,
|
| - const CompletionCallback& callback) {
|
| - DCHECK(completed_handshake_);
|
| - DCHECK_EQ(STATE_NONE, next_state_);
|
| - DCHECK(user_callback_.is_null());
|
| - DCHECK(!callback.is_null());
|
| -
|
| - int rv = transport_->socket()->Read(
|
| - buf, buf_len,
|
| - base::Bind(&SOCKS5ClientSocket::OnReadWriteComplete,
|
| - base::Unretained(this), callback));
|
| - if (rv > 0)
|
| - was_ever_used_ = true;
|
| - return rv;
|
| -}
|
| -
|
| -// Write is called by the transport layer. This can only be done if the
|
| -// SOCKS handshake is complete.
|
| -int SOCKS5ClientSocket::Write(IOBuffer* buf, int buf_len,
|
| - const CompletionCallback& callback) {
|
| - DCHECK(completed_handshake_);
|
| - DCHECK_EQ(STATE_NONE, next_state_);
|
| - DCHECK(user_callback_.is_null());
|
| - DCHECK(!callback.is_null());
|
| -
|
| - int rv = transport_->socket()->Write(
|
| - buf, buf_len,
|
| - base::Bind(&SOCKS5ClientSocket::OnReadWriteComplete,
|
| - base::Unretained(this), callback));
|
| - if (rv > 0)
|
| - was_ever_used_ = true;
|
| - return rv;
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::SetReceiveBufferSize(int32 size) {
|
| - return transport_->socket()->SetReceiveBufferSize(size);
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::SetSendBufferSize(int32 size) {
|
| - return transport_->socket()->SetSendBufferSize(size);
|
| -}
|
| -
|
| -void SOCKS5ClientSocket::DoCallback(int result) {
|
| - DCHECK_NE(ERR_IO_PENDING, result);
|
| - DCHECK(!user_callback_.is_null());
|
| -
|
| - // Since Run() may result in Read being called,
|
| - // clear user_callback_ up front.
|
| - base::ResetAndReturn(&user_callback_).Run(result);
|
| -}
|
| -
|
| -void SOCKS5ClientSocket::OnIOComplete(int result) {
|
| - DCHECK_NE(STATE_NONE, next_state_);
|
| - int rv = DoLoop(result);
|
| - if (rv != ERR_IO_PENDING) {
|
| - net_log_.EndEvent(NetLog::TYPE_SOCKS5_CONNECT);
|
| - DoCallback(rv);
|
| - }
|
| -}
|
| -
|
| -void SOCKS5ClientSocket::OnReadWriteComplete(const CompletionCallback& callback,
|
| - int result) {
|
| - DCHECK_NE(ERR_IO_PENDING, result);
|
| - DCHECK(!callback.is_null());
|
| -
|
| - if (result > 0)
|
| - was_ever_used_ = true;
|
| - callback.Run(result);
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::DoLoop(int last_io_result) {
|
| - DCHECK_NE(next_state_, STATE_NONE);
|
| - int rv = last_io_result;
|
| - do {
|
| - State state = next_state_;
|
| - next_state_ = STATE_NONE;
|
| - switch (state) {
|
| - case STATE_GREET_WRITE:
|
| - DCHECK_EQ(OK, rv);
|
| - net_log_.BeginEvent(NetLog::TYPE_SOCKS5_GREET_WRITE);
|
| - rv = DoGreetWrite();
|
| - break;
|
| - case STATE_GREET_WRITE_COMPLETE:
|
| - rv = DoGreetWriteComplete(rv);
|
| - net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SOCKS5_GREET_WRITE, rv);
|
| - break;
|
| - case STATE_GREET_READ:
|
| - DCHECK_EQ(OK, rv);
|
| - net_log_.BeginEvent(NetLog::TYPE_SOCKS5_GREET_READ);
|
| - rv = DoGreetRead();
|
| - break;
|
| - case STATE_GREET_READ_COMPLETE:
|
| - rv = DoGreetReadComplete(rv);
|
| - net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SOCKS5_GREET_READ, rv);
|
| - break;
|
| - case STATE_HANDSHAKE_WRITE:
|
| - DCHECK_EQ(OK, rv);
|
| - net_log_.BeginEvent(NetLog::TYPE_SOCKS5_HANDSHAKE_WRITE);
|
| - rv = DoHandshakeWrite();
|
| - break;
|
| - case STATE_HANDSHAKE_WRITE_COMPLETE:
|
| - rv = DoHandshakeWriteComplete(rv);
|
| - net_log_.EndEventWithNetErrorCode(
|
| - NetLog::TYPE_SOCKS5_HANDSHAKE_WRITE, rv);
|
| - break;
|
| - case STATE_HANDSHAKE_READ:
|
| - DCHECK_EQ(OK, rv);
|
| - net_log_.BeginEvent(NetLog::TYPE_SOCKS5_HANDSHAKE_READ);
|
| - rv = DoHandshakeRead();
|
| - break;
|
| - case STATE_HANDSHAKE_READ_COMPLETE:
|
| - rv = DoHandshakeReadComplete(rv);
|
| - net_log_.EndEventWithNetErrorCode(
|
| - NetLog::TYPE_SOCKS5_HANDSHAKE_READ, rv);
|
| - break;
|
| - default:
|
| - NOTREACHED() << "bad state";
|
| - rv = ERR_UNEXPECTED;
|
| - break;
|
| - }
|
| - } while (rv != ERR_IO_PENDING && next_state_ != STATE_NONE);
|
| - return rv;
|
| -}
|
| -
|
| -const char kSOCKS5GreetWriteData[] = { 0x05, 0x01, 0x00 }; // no authentication
|
| -
|
| -int SOCKS5ClientSocket::DoGreetWrite() {
|
| - // Since we only have 1 byte to send the hostname length in, if the
|
| - // URL has a hostname longer than 255 characters we can't send it.
|
| - if (0xFF < host_request_info_.hostname().size()) {
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKS_HOSTNAME_TOO_BIG);
|
| - return ERR_SOCKS_CONNECTION_FAILED;
|
| - }
|
| -
|
| - if (buffer_.empty()) {
|
| - buffer_ = std::string(kSOCKS5GreetWriteData,
|
| - arraysize(kSOCKS5GreetWriteData));
|
| - bytes_sent_ = 0;
|
| - }
|
| -
|
| - next_state_ = STATE_GREET_WRITE_COMPLETE;
|
| - size_t handshake_buf_len = buffer_.size() - bytes_sent_;
|
| - handshake_buf_ = new IOBuffer(handshake_buf_len);
|
| - memcpy(handshake_buf_->data(), &buffer_.data()[bytes_sent_],
|
| - handshake_buf_len);
|
| - return transport_->socket()
|
| - ->Write(handshake_buf_.get(), handshake_buf_len, io_callback_);
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::DoGreetWriteComplete(int result) {
|
| - if (result < 0)
|
| - return result;
|
| -
|
| - bytes_sent_ += result;
|
| - if (bytes_sent_ == buffer_.size()) {
|
| - buffer_.clear();
|
| - bytes_received_ = 0;
|
| - next_state_ = STATE_GREET_READ;
|
| - } else {
|
| - next_state_ = STATE_GREET_WRITE;
|
| - }
|
| - return OK;
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::DoGreetRead() {
|
| - next_state_ = STATE_GREET_READ_COMPLETE;
|
| - size_t handshake_buf_len = kGreetReadHeaderSize - bytes_received_;
|
| - handshake_buf_ = new IOBuffer(handshake_buf_len);
|
| - return transport_->socket()
|
| - ->Read(handshake_buf_.get(), handshake_buf_len, io_callback_);
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::DoGreetReadComplete(int result) {
|
| - if (result < 0)
|
| - return result;
|
| -
|
| - if (result == 0) {
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKS_UNEXPECTEDLY_CLOSED_DURING_GREETING);
|
| - return ERR_SOCKS_CONNECTION_FAILED;
|
| - }
|
| -
|
| - bytes_received_ += result;
|
| - buffer_.append(handshake_buf_->data(), result);
|
| - if (bytes_received_ < kGreetReadHeaderSize) {
|
| - next_state_ = STATE_GREET_READ;
|
| - return OK;
|
| - }
|
| -
|
| - // Got the greet data.
|
| - if (buffer_[0] != kSOCKS5Version) {
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKS_UNEXPECTED_VERSION,
|
| - NetLog::IntegerCallback("version", buffer_[0]));
|
| - return ERR_SOCKS_CONNECTION_FAILED;
|
| - }
|
| - if (buffer_[1] != 0x00) {
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKS_UNEXPECTED_AUTH,
|
| - NetLog::IntegerCallback("method", buffer_[1]));
|
| - return ERR_SOCKS_CONNECTION_FAILED;
|
| - }
|
| -
|
| - buffer_.clear();
|
| - next_state_ = STATE_HANDSHAKE_WRITE;
|
| - return OK;
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::BuildHandshakeWriteBuffer(std::string* handshake)
|
| - const {
|
| - DCHECK(handshake->empty());
|
| -
|
| - handshake->push_back(kSOCKS5Version);
|
| - handshake->push_back(kTunnelCommand); // Connect command
|
| - handshake->push_back(kNullByte); // Reserved null
|
| -
|
| - handshake->push_back(kEndPointDomain); // The type of the address.
|
| -
|
| - DCHECK_GE(static_cast<size_t>(0xFF), host_request_info_.hostname().size());
|
| -
|
| - // First add the size of the hostname, followed by the hostname.
|
| - handshake->push_back(static_cast<unsigned char>(
|
| - host_request_info_.hostname().size()));
|
| - handshake->append(host_request_info_.hostname());
|
| -
|
| - uint16 nw_port = base::HostToNet16(host_request_info_.port());
|
| - handshake->append(reinterpret_cast<char*>(&nw_port), sizeof(nw_port));
|
| - return OK;
|
| -}
|
| -
|
| -// Writes the SOCKS handshake data to the underlying socket connection.
|
| -int SOCKS5ClientSocket::DoHandshakeWrite() {
|
| - next_state_ = STATE_HANDSHAKE_WRITE_COMPLETE;
|
| -
|
| - if (buffer_.empty()) {
|
| - int rv = BuildHandshakeWriteBuffer(&buffer_);
|
| - if (rv != OK)
|
| - return rv;
|
| - bytes_sent_ = 0;
|
| - }
|
| -
|
| - int handshake_buf_len = buffer_.size() - bytes_sent_;
|
| - DCHECK_LT(0, handshake_buf_len);
|
| - handshake_buf_ = new IOBuffer(handshake_buf_len);
|
| - memcpy(handshake_buf_->data(), &buffer_[bytes_sent_],
|
| - handshake_buf_len);
|
| - return transport_->socket()
|
| - ->Write(handshake_buf_.get(), handshake_buf_len, io_callback_);
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::DoHandshakeWriteComplete(int result) {
|
| - if (result < 0)
|
| - return result;
|
| -
|
| - // We ignore the case when result is 0, since the underlying Write
|
| - // may return spurious writes while waiting on the socket.
|
| -
|
| - bytes_sent_ += result;
|
| - if (bytes_sent_ == buffer_.size()) {
|
| - next_state_ = STATE_HANDSHAKE_READ;
|
| - buffer_.clear();
|
| - } else if (bytes_sent_ < buffer_.size()) {
|
| - next_state_ = STATE_HANDSHAKE_WRITE;
|
| - } else {
|
| - NOTREACHED();
|
| - }
|
| -
|
| - return OK;
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::DoHandshakeRead() {
|
| - next_state_ = STATE_HANDSHAKE_READ_COMPLETE;
|
| -
|
| - if (buffer_.empty()) {
|
| - bytes_received_ = 0;
|
| - read_header_size = kReadHeaderSize;
|
| - }
|
| -
|
| - int handshake_buf_len = read_header_size - bytes_received_;
|
| - handshake_buf_ = new IOBuffer(handshake_buf_len);
|
| - return transport_->socket()
|
| - ->Read(handshake_buf_.get(), handshake_buf_len, io_callback_);
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::DoHandshakeReadComplete(int result) {
|
| - if (result < 0)
|
| - return result;
|
| -
|
| - // The underlying socket closed unexpectedly.
|
| - if (result == 0) {
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKS_UNEXPECTEDLY_CLOSED_DURING_HANDSHAKE);
|
| - return ERR_SOCKS_CONNECTION_FAILED;
|
| - }
|
| -
|
| - buffer_.append(handshake_buf_->data(), result);
|
| - bytes_received_ += result;
|
| -
|
| - // When the first few bytes are read, check how many more are required
|
| - // and accordingly increase them
|
| - if (bytes_received_ == kReadHeaderSize) {
|
| - if (buffer_[0] != kSOCKS5Version || buffer_[2] != kNullByte) {
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKS_UNEXPECTED_VERSION,
|
| - NetLog::IntegerCallback("version", buffer_[0]));
|
| - return ERR_SOCKS_CONNECTION_FAILED;
|
| - }
|
| - if (buffer_[1] != 0x00) {
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKS_SERVER_ERROR,
|
| - NetLog::IntegerCallback("error_code", buffer_[1]));
|
| - return ERR_SOCKS_CONNECTION_FAILED;
|
| - }
|
| -
|
| - // We check the type of IP/Domain the server returns and accordingly
|
| - // increase the size of the response. For domains, we need to read the
|
| - // size of the domain, so the initial request size is upto the domain
|
| - // size. Since for IPv4/IPv6 the size is fixed and hence no 'size' is
|
| - // read, we substract 1 byte from the additional request size.
|
| - SocksEndPointAddressType address_type =
|
| - static_cast<SocksEndPointAddressType>(buffer_[3]);
|
| - if (address_type == kEndPointDomain)
|
| - read_header_size += static_cast<uint8>(buffer_[4]);
|
| - else if (address_type == kEndPointResolvedIPv4)
|
| - read_header_size += sizeof(struct in_addr) - 1;
|
| - else if (address_type == kEndPointResolvedIPv6)
|
| - read_header_size += sizeof(struct in6_addr) - 1;
|
| - else {
|
| - net_log_.AddEvent(NetLog::TYPE_SOCKS_UNKNOWN_ADDRESS_TYPE,
|
| - NetLog::IntegerCallback("address_type", buffer_[3]));
|
| - return ERR_SOCKS_CONNECTION_FAILED;
|
| - }
|
| -
|
| - read_header_size += 2; // for the port.
|
| - next_state_ = STATE_HANDSHAKE_READ;
|
| - return OK;
|
| - }
|
| -
|
| - // When the final bytes are read, setup handshake. We ignore the rest
|
| - // of the response since they represent the SOCKSv5 endpoint and have
|
| - // no use when doing a tunnel connection.
|
| - if (bytes_received_ == read_header_size) {
|
| - completed_handshake_ = true;
|
| - buffer_.clear();
|
| - next_state_ = STATE_NONE;
|
| - return OK;
|
| - }
|
| -
|
| - next_state_ = STATE_HANDSHAKE_READ;
|
| - return OK;
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::GetPeerAddress(IPEndPoint* address) const {
|
| - return transport_->socket()->GetPeerAddress(address);
|
| -}
|
| -
|
| -int SOCKS5ClientSocket::GetLocalAddress(IPEndPoint* address) const {
|
| - return transport_->socket()->GetLocalAddress(address);
|
| -}
|
| -
|
| -} // namespace net
|
|
|