Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(69)

Unified Diff: net/socket/socks_client_socket.cc

Issue 992733002: Remove //net (except for Android test stuff) and sdch (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Created 5 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « net/socket/socks_client_socket.h ('k') | net/socket/socks_client_socket_pool.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: net/socket/socks_client_socket.cc
diff --git a/net/socket/socks_client_socket.cc b/net/socket/socks_client_socket.cc
deleted file mode 100644
index 75eb36a76d73d8a9b6968ea6b31810a2d6effd22..0000000000000000000000000000000000000000
--- a/net/socket/socks_client_socket.cc
+++ /dev/null
@@ -1,461 +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/socks_client_socket.h"
-
-#include "base/basictypes.h"
-#include "base/bind.h"
-#include "base/callback_helpers.h"
-#include "base/compiler_specific.h"
-#include "base/profiler/scoped_tracker.h"
-#include "base/sys_byteorder.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 {
-
-// Every SOCKS server requests a user-id from the client. It is optional
-// and we send an empty string.
-static const char kEmptyUserId[] = "";
-
-// For SOCKS4, the client sends 8 bytes plus the size of the user-id.
-static const unsigned int kWriteHeaderSize = 8;
-
-// For SOCKS4 the server sends 8 bytes for acknowledgement.
-static const unsigned int kReadHeaderSize = 8;
-
-// Server Response codes for SOCKS.
-static const uint8 kServerResponseOk = 0x5A;
-static const uint8 kServerResponseRejected = 0x5B;
-static const uint8 kServerResponseNotReachable = 0x5C;
-static const uint8 kServerResponseMismatchedUserId = 0x5D;
-
-static const uint8 kSOCKSVersion4 = 0x04;
-static const uint8 kSOCKSStreamRequest = 0x01;
-
-// A struct holding the essential details of the SOCKS4 Server Request.
-// The port in the header is stored in network byte order.
-struct SOCKS4ServerRequest {
- uint8 version;
- uint8 command;
- uint16 nw_port;
- uint8 ip[4];
-};
-static_assert(sizeof(SOCKS4ServerRequest) == kWriteHeaderSize,
- "socks4 server request struct has incorrect size");
-
-// A struct holding details of the SOCKS4 Server Response.
-struct SOCKS4ServerResponse {
- uint8 reserved_null;
- uint8 code;
- uint16 port;
- uint8 ip[4];
-};
-static_assert(sizeof(SOCKS4ServerResponse) == kReadHeaderSize,
- "socks4 server response struct has incorrect size");
-
-SOCKSClientSocket::SOCKSClientSocket(
- scoped_ptr<ClientSocketHandle> transport_socket,
- const HostResolver::RequestInfo& req_info,
- RequestPriority priority,
- HostResolver* host_resolver)
- : transport_(transport_socket.Pass()),
- next_state_(STATE_NONE),
- completed_handshake_(false),
- bytes_sent_(0),
- bytes_received_(0),
- was_ever_used_(false),
- host_resolver_(host_resolver),
- host_request_info_(req_info),
- priority_(priority),
- net_log_(transport_->socket()->NetLog()) {}
-
-SOCKSClientSocket::~SOCKSClientSocket() {
- Disconnect();
-}
-
-int SOCKSClientSocket::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;
-
- next_state_ = STATE_RESOLVE_HOST;
-
- net_log_.BeginEvent(NetLog::TYPE_SOCKS_CONNECT);
-
- int rv = DoLoop(OK);
- if (rv == ERR_IO_PENDING) {
- user_callback_ = callback;
- } else {
- net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SOCKS_CONNECT, rv);
- }
- return rv;
-}
-
-void SOCKSClientSocket::Disconnect() {
- completed_handshake_ = false;
- host_resolver_.Cancel();
- 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 SOCKSClientSocket::IsConnected() const {
- return completed_handshake_ && transport_->socket()->IsConnected();
-}
-
-bool SOCKSClientSocket::IsConnectedAndIdle() const {
- return completed_handshake_ && transport_->socket()->IsConnectedAndIdle();
-}
-
-const BoundNetLog& SOCKSClientSocket::NetLog() const {
- return net_log_;
-}
-
-void SOCKSClientSocket::SetSubresourceSpeculation() {
- if (transport_.get() && transport_->socket()) {
- transport_->socket()->SetSubresourceSpeculation();
- } else {
- NOTREACHED();
- }
-}
-
-void SOCKSClientSocket::SetOmniboxSpeculation() {
- if (transport_.get() && transport_->socket()) {
- transport_->socket()->SetOmniboxSpeculation();
- } else {
- NOTREACHED();
- }
-}
-
-bool SOCKSClientSocket::WasEverUsed() const {
- return was_ever_used_;
-}
-
-bool SOCKSClientSocket::UsingTCPFastOpen() const {
- if (transport_.get() && transport_->socket()) {
- return transport_->socket()->UsingTCPFastOpen();
- }
- NOTREACHED();
- return false;
-}
-
-bool SOCKSClientSocket::WasNpnNegotiated() const {
- if (transport_.get() && transport_->socket()) {
- return transport_->socket()->WasNpnNegotiated();
- }
- NOTREACHED();
- return false;
-}
-
-NextProto SOCKSClientSocket::GetNegotiatedProtocol() const {
- if (transport_.get() && transport_->socket()) {
- return transport_->socket()->GetNegotiatedProtocol();
- }
- NOTREACHED();
- return kProtoUnknown;
-}
-
-bool SOCKSClientSocket::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 SOCKSClientSocket::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(&SOCKSClientSocket::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 SOCKSClientSocket::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(&SOCKSClientSocket::OnReadWriteComplete,
- base::Unretained(this), callback));
- if (rv > 0)
- was_ever_used_ = true;
- return rv;
-}
-
-int SOCKSClientSocket::SetReceiveBufferSize(int32 size) {
- return transport_->socket()->SetReceiveBufferSize(size);
-}
-
-int SOCKSClientSocket::SetSendBufferSize(int32 size) {
- return transport_->socket()->SetSendBufferSize(size);
-}
-
-void SOCKSClientSocket::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.
- DVLOG(1) << "Finished setting up SOCKS handshake";
- base::ResetAndReturn(&user_callback_).Run(result);
-}
-
-void SOCKSClientSocket::OnIOComplete(int result) {
- // TODO(vadimt): Remove ScopedTracker below once crbug.com/436634 is fixed.
- tracked_objects::ScopedTracker tracking_profile(
- FROM_HERE_WITH_EXPLICIT_FUNCTION(
- "436634 SOCKSClientSocket::OnIOComplete"));
-
- DCHECK_NE(STATE_NONE, next_state_);
- int rv = DoLoop(result);
- if (rv != ERR_IO_PENDING) {
- net_log_.EndEventWithNetErrorCode(NetLog::TYPE_SOCKS_CONNECT, rv);
- DoCallback(rv);
- }
-}
-
-void SOCKSClientSocket::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 SOCKSClientSocket::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_RESOLVE_HOST:
- DCHECK_EQ(OK, rv);
- rv = DoResolveHost();
- break;
- case STATE_RESOLVE_HOST_COMPLETE:
- rv = DoResolveHostComplete(rv);
- break;
- case STATE_HANDSHAKE_WRITE:
- DCHECK_EQ(OK, rv);
- rv = DoHandshakeWrite();
- break;
- case STATE_HANDSHAKE_WRITE_COMPLETE:
- rv = DoHandshakeWriteComplete(rv);
- break;
- case STATE_HANDSHAKE_READ:
- DCHECK_EQ(OK, rv);
- rv = DoHandshakeRead();
- break;
- case STATE_HANDSHAKE_READ_COMPLETE:
- rv = DoHandshakeReadComplete(rv);
- break;
- default:
- NOTREACHED() << "bad state";
- rv = ERR_UNEXPECTED;
- break;
- }
- } while (rv != ERR_IO_PENDING && next_state_ != STATE_NONE);
- return rv;
-}
-
-int SOCKSClientSocket::DoResolveHost() {
- next_state_ = STATE_RESOLVE_HOST_COMPLETE;
- // SOCKS4 only supports IPv4 addresses, so only try getting the IPv4
- // addresses for the target host.
- host_request_info_.set_address_family(ADDRESS_FAMILY_IPV4);
- return host_resolver_.Resolve(
- host_request_info_,
- priority_,
- &addresses_,
- base::Bind(&SOCKSClientSocket::OnIOComplete, base::Unretained(this)),
- net_log_);
-}
-
-int SOCKSClientSocket::DoResolveHostComplete(int result) {
- if (result != OK) {
- // Resolving the hostname failed; fail the request rather than automatically
- // falling back to SOCKS4a (since it can be confusing to see invalid IP
- // addresses being sent to the SOCKS4 server when it doesn't support 4A.)
- return result;
- }
-
- next_state_ = STATE_HANDSHAKE_WRITE;
- return OK;
-}
-
-// Builds the buffer that is to be sent to the server.
-const std::string SOCKSClientSocket::BuildHandshakeWriteBuffer() const {
- SOCKS4ServerRequest request;
- request.version = kSOCKSVersion4;
- request.command = kSOCKSStreamRequest;
- request.nw_port = base::HostToNet16(host_request_info_.port());
-
- DCHECK(!addresses_.empty());
- const IPEndPoint& endpoint = addresses_.front();
-
- // We disabled IPv6 results when resolving the hostname, so none of the
- // results in the list will be IPv6.
- // TODO(eroman): we only ever use the first address in the list. It would be
- // more robust to try all the IP addresses we have before
- // failing the connect attempt.
- CHECK_EQ(ADDRESS_FAMILY_IPV4, endpoint.GetFamily());
- CHECK_LE(endpoint.address().size(), sizeof(request.ip));
- memcpy(&request.ip, &endpoint.address()[0], endpoint.address().size());
-
- DVLOG(1) << "Resolved Host is : " << endpoint.ToStringWithoutPort();
-
- std::string handshake_data(reinterpret_cast<char*>(&request),
- sizeof(request));
- handshake_data.append(kEmptyUserId, arraysize(kEmptyUserId));
-
- return handshake_data;
-}
-
-// Writes the SOCKS handshake data to the underlying socket connection.
-int SOCKSClientSocket::DoHandshakeWrite() {
- next_state_ = STATE_HANDSHAKE_WRITE_COMPLETE;
-
- if (buffer_.empty()) {
- buffer_ = BuildHandshakeWriteBuffer();
- bytes_sent_ = 0;
- }
-
- int handshake_buf_len = buffer_.size() - bytes_sent_;
- DCHECK_GT(handshake_buf_len, 0);
- 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,
- base::Bind(&SOCKSClientSocket::OnIOComplete, base::Unretained(this)));
-}
-
-int SOCKSClientSocket::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 {
- return ERR_UNEXPECTED;
- }
-
- return OK;
-}
-
-int SOCKSClientSocket::DoHandshakeRead() {
- next_state_ = STATE_HANDSHAKE_READ_COMPLETE;
-
- if (buffer_.empty()) {
- bytes_received_ = 0;
- }
-
- int handshake_buf_len = kReadHeaderSize - bytes_received_;
- handshake_buf_ = new IOBuffer(handshake_buf_len);
- return transport_->socket()->Read(
- handshake_buf_.get(),
- handshake_buf_len,
- base::Bind(&SOCKSClientSocket::OnIOComplete, base::Unretained(this)));
-}
-
-int SOCKSClientSocket::DoHandshakeReadComplete(int result) {
- if (result < 0)
- return result;
-
- // The underlying socket closed unexpectedly.
- if (result == 0)
- return ERR_CONNECTION_CLOSED;
-
- if (bytes_received_ + result > kReadHeaderSize) {
- // TODO(eroman): Describe failure in NetLog.
- return ERR_SOCKS_CONNECTION_FAILED;
- }
-
- buffer_.append(handshake_buf_->data(), result);
- bytes_received_ += result;
- if (bytes_received_ < kReadHeaderSize) {
- next_state_ = STATE_HANDSHAKE_READ;
- return OK;
- }
-
- const SOCKS4ServerResponse* response =
- reinterpret_cast<const SOCKS4ServerResponse*>(buffer_.data());
-
- if (response->reserved_null != 0x00) {
- LOG(ERROR) << "Unknown response from SOCKS server.";
- return ERR_SOCKS_CONNECTION_FAILED;
- }
-
- switch (response->code) {
- case kServerResponseOk:
- completed_handshake_ = true;
- return OK;
- case kServerResponseRejected:
- LOG(ERROR) << "SOCKS request rejected or failed";
- return ERR_SOCKS_CONNECTION_FAILED;
- case kServerResponseNotReachable:
- LOG(ERROR) << "SOCKS request failed because client is not running "
- << "identd (or not reachable from the server)";
- return ERR_SOCKS_CONNECTION_HOST_UNREACHABLE;
- case kServerResponseMismatchedUserId:
- LOG(ERROR) << "SOCKS request failed because client's identd could "
- << "not confirm the user ID string in the request";
- return ERR_SOCKS_CONNECTION_FAILED;
- default:
- LOG(ERROR) << "SOCKS server sent unknown response";
- return ERR_SOCKS_CONNECTION_FAILED;
- }
-
- // Note: we ignore the last 6 bytes as specified by the SOCKS protocol
-}
-
-int SOCKSClientSocket::GetPeerAddress(IPEndPoint* address) const {
- return transport_->socket()->GetPeerAddress(address);
-}
-
-int SOCKSClientSocket::GetLocalAddress(IPEndPoint* address) const {
- return transport_->socket()->GetLocalAddress(address);
-}
-
-} // namespace net
« no previous file with comments | « net/socket/socks_client_socket.h ('k') | net/socket/socks_client_socket_pool.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698