| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #include "webrtc/base/proxyserver.h" | 11 #include "webrtc/base/proxyserver.h" |
| 12 | 12 |
| 13 #include <algorithm> | 13 #include <algorithm> |
| 14 |
| 15 #include "webrtc/base/checks.h" |
| 14 #include "webrtc/base/socketfactory.h" | 16 #include "webrtc/base/socketfactory.h" |
| 15 | 17 |
| 16 namespace rtc { | 18 namespace rtc { |
| 17 | 19 |
| 18 // ProxyServer | 20 // ProxyServer |
| 19 ProxyServer::ProxyServer( | 21 ProxyServer::ProxyServer( |
| 20 SocketFactory* int_factory, const SocketAddress& int_addr, | 22 SocketFactory* int_factory, const SocketAddress& int_addr, |
| 21 SocketFactory* ext_factory, const SocketAddress& ext_ip) | 23 SocketFactory* ext_factory, const SocketAddress& ext_ip) |
| 22 : ext_factory_(ext_factory), ext_ip_(ext_ip.ipaddr(), 0), // strip off port | 24 : ext_factory_(ext_factory), ext_ip_(ext_ip.ipaddr(), 0), // strip off port |
| 23 server_socket_(int_factory->CreateAsyncSocket(int_addr.family(), | 25 server_socket_(int_factory->CreateAsyncSocket(int_addr.family(), |
| 24 SOCK_STREAM)) { | 26 SOCK_STREAM)) { |
| 25 ASSERT(server_socket_.get() != NULL); | 27 RTC_DCHECK(server_socket_.get() != NULL); |
| 26 ASSERT(int_addr.family() == AF_INET || int_addr.family() == AF_INET6); | 28 RTC_DCHECK(int_addr.family() == AF_INET || int_addr.family() == AF_INET6); |
| 27 server_socket_->Bind(int_addr); | 29 server_socket_->Bind(int_addr); |
| 28 server_socket_->Listen(5); | 30 server_socket_->Listen(5); |
| 29 server_socket_->SignalReadEvent.connect(this, &ProxyServer::OnAcceptEvent); | 31 server_socket_->SignalReadEvent.connect(this, &ProxyServer::OnAcceptEvent); |
| 30 } | 32 } |
| 31 | 33 |
| 32 ProxyServer::~ProxyServer() { | 34 ProxyServer::~ProxyServer() { |
| 33 for (BindingList::iterator it = bindings_.begin(); | 35 for (BindingList::iterator it = bindings_.begin(); |
| 34 it != bindings_.end(); ++it) { | 36 it != bindings_.end(); ++it) { |
| 35 delete (*it); | 37 delete (*it); |
| 36 } | 38 } |
| 37 } | 39 } |
| 38 | 40 |
| 39 SocketAddress ProxyServer::GetServerAddress() { | 41 SocketAddress ProxyServer::GetServerAddress() { |
| 40 return server_socket_->GetLocalAddress(); | 42 return server_socket_->GetLocalAddress(); |
| 41 } | 43 } |
| 42 | 44 |
| 43 void ProxyServer::OnAcceptEvent(AsyncSocket* socket) { | 45 void ProxyServer::OnAcceptEvent(AsyncSocket* socket) { |
| 44 ASSERT(socket != NULL && socket == server_socket_.get()); | 46 RTC_DCHECK(socket != NULL && socket == server_socket_.get()); |
| 45 AsyncSocket* int_socket = socket->Accept(NULL); | 47 AsyncSocket* int_socket = socket->Accept(NULL); |
| 46 AsyncProxyServerSocket* wrapped_socket = WrapSocket(int_socket); | 48 AsyncProxyServerSocket* wrapped_socket = WrapSocket(int_socket); |
| 47 AsyncSocket* ext_socket = ext_factory_->CreateAsyncSocket(ext_ip_.family(), | 49 AsyncSocket* ext_socket = ext_factory_->CreateAsyncSocket(ext_ip_.family(), |
| 48 SOCK_STREAM); | 50 SOCK_STREAM); |
| 49 if (ext_socket) { | 51 if (ext_socket) { |
| 50 ext_socket->Bind(ext_ip_); | 52 ext_socket->Bind(ext_ip_); |
| 51 bindings_.push_back(new ProxyBinding(wrapped_socket, ext_socket)); | 53 bindings_.push_back(new ProxyBinding(wrapped_socket, ext_socket)); |
| 52 } else { | 54 } else { |
| 53 LOG(LS_ERROR) << "Unable to create external socket on proxy accept event"; | 55 LOG(LS_ERROR) << "Unable to create external socket on proxy accept event"; |
| 54 } | 56 } |
| (...skipping 20 matching lines...) Expand all Loading... |
| 75 &ProxyBinding::OnExternalConnect); | 77 &ProxyBinding::OnExternalConnect); |
| 76 ext_socket_->SignalReadEvent.connect(this, &ProxyBinding::OnExternalRead); | 78 ext_socket_->SignalReadEvent.connect(this, &ProxyBinding::OnExternalRead); |
| 77 ext_socket_->SignalWriteEvent.connect(this, &ProxyBinding::OnExternalWrite); | 79 ext_socket_->SignalWriteEvent.connect(this, &ProxyBinding::OnExternalWrite); |
| 78 ext_socket_->SignalCloseEvent.connect(this, &ProxyBinding::OnExternalClose); | 80 ext_socket_->SignalCloseEvent.connect(this, &ProxyBinding::OnExternalClose); |
| 79 } | 81 } |
| 80 | 82 |
| 81 ProxyBinding::~ProxyBinding() = default; | 83 ProxyBinding::~ProxyBinding() = default; |
| 82 | 84 |
| 83 void ProxyBinding::OnConnectRequest(AsyncProxyServerSocket* socket, | 85 void ProxyBinding::OnConnectRequest(AsyncProxyServerSocket* socket, |
| 84 const SocketAddress& addr) { | 86 const SocketAddress& addr) { |
| 85 ASSERT(!connected_ && ext_socket_.get() != NULL); | 87 RTC_DCHECK(!connected_ && ext_socket_.get() != NULL); |
| 86 ext_socket_->Connect(addr); | 88 ext_socket_->Connect(addr); |
| 87 // TODO: handle errors here | 89 // TODO: handle errors here |
| 88 } | 90 } |
| 89 | 91 |
| 90 void ProxyBinding::OnInternalRead(AsyncSocket* socket) { | 92 void ProxyBinding::OnInternalRead(AsyncSocket* socket) { |
| 91 Read(int_socket_.get(), &out_buffer_); | 93 Read(int_socket_.get(), &out_buffer_); |
| 92 Write(ext_socket_.get(), &out_buffer_); | 94 Write(ext_socket_.get(), &out_buffer_); |
| 93 } | 95 } |
| 94 | 96 |
| 95 void ProxyBinding::OnInternalWrite(AsyncSocket* socket) { | 97 void ProxyBinding::OnInternalWrite(AsyncSocket* socket) { |
| 96 Write(int_socket_.get(), &in_buffer_); | 98 Write(int_socket_.get(), &in_buffer_); |
| 97 } | 99 } |
| 98 | 100 |
| 99 void ProxyBinding::OnInternalClose(AsyncSocket* socket, int err) { | 101 void ProxyBinding::OnInternalClose(AsyncSocket* socket, int err) { |
| 100 Destroy(); | 102 Destroy(); |
| 101 } | 103 } |
| 102 | 104 |
| 103 void ProxyBinding::OnExternalConnect(AsyncSocket* socket) { | 105 void ProxyBinding::OnExternalConnect(AsyncSocket* socket) { |
| 104 ASSERT(socket != NULL); | 106 RTC_DCHECK(socket != NULL); |
| 105 connected_ = true; | 107 connected_ = true; |
| 106 int_socket_->SendConnectResult(0, socket->GetRemoteAddress()); | 108 int_socket_->SendConnectResult(0, socket->GetRemoteAddress()); |
| 107 } | 109 } |
| 108 | 110 |
| 109 void ProxyBinding::OnExternalRead(AsyncSocket* socket) { | 111 void ProxyBinding::OnExternalRead(AsyncSocket* socket) { |
| 110 Read(ext_socket_.get(), &in_buffer_); | 112 Read(ext_socket_.get(), &in_buffer_); |
| 111 Write(int_socket_.get(), &in_buffer_); | 113 Write(int_socket_.get(), &in_buffer_); |
| 112 } | 114 } |
| 113 | 115 |
| 114 void ProxyBinding::OnExternalWrite(AsyncSocket* socket) { | 116 void ProxyBinding::OnExternalWrite(AsyncSocket* socket) { |
| 115 Write(ext_socket_.get(), &out_buffer_); | 117 Write(ext_socket_.get(), &out_buffer_); |
| 116 } | 118 } |
| 117 | 119 |
| 118 void ProxyBinding::OnExternalClose(AsyncSocket* socket, int err) { | 120 void ProxyBinding::OnExternalClose(AsyncSocket* socket, int err) { |
| 119 if (!connected_) { | 121 if (!connected_) { |
| 120 int_socket_->SendConnectResult(err, SocketAddress()); | 122 int_socket_->SendConnectResult(err, SocketAddress()); |
| 121 } | 123 } |
| 122 Destroy(); | 124 Destroy(); |
| 123 } | 125 } |
| 124 | 126 |
| 125 void ProxyBinding::Read(AsyncSocket* socket, FifoBuffer* buffer) { | 127 void ProxyBinding::Read(AsyncSocket* socket, FifoBuffer* buffer) { |
| 126 // Only read if the buffer is empty. | 128 // Only read if the buffer is empty. |
| 127 ASSERT(socket != NULL); | 129 RTC_DCHECK(socket != NULL); |
| 128 size_t size; | 130 size_t size; |
| 129 int read; | 131 int read; |
| 130 if (buffer->GetBuffered(&size) && size == 0) { | 132 if (buffer->GetBuffered(&size) && size == 0) { |
| 131 void* p = buffer->GetWriteBuffer(&size); | 133 void* p = buffer->GetWriteBuffer(&size); |
| 132 read = socket->Recv(p, size, nullptr); | 134 read = socket->Recv(p, size, nullptr); |
| 133 buffer->ConsumeWriteBuffer(std::max(read, 0)); | 135 buffer->ConsumeWriteBuffer(std::max(read, 0)); |
| 134 } | 136 } |
| 135 } | 137 } |
| 136 | 138 |
| 137 void ProxyBinding::Write(AsyncSocket* socket, FifoBuffer* buffer) { | 139 void ProxyBinding::Write(AsyncSocket* socket, FifoBuffer* buffer) { |
| 138 ASSERT(socket != NULL); | 140 RTC_DCHECK(socket != NULL); |
| 139 size_t size; | 141 size_t size; |
| 140 int written; | 142 int written; |
| 141 const void* p = buffer->GetReadData(&size); | 143 const void* p = buffer->GetReadData(&size); |
| 142 written = socket->Send(p, size); | 144 written = socket->Send(p, size); |
| 143 buffer->ConsumeReadData(std::max(written, 0)); | 145 buffer->ConsumeReadData(std::max(written, 0)); |
| 144 } | 146 } |
| 145 | 147 |
| 146 void ProxyBinding::Destroy() { | 148 void ProxyBinding::Destroy() { |
| 147 SignalDestroyed(this); | 149 SignalDestroyed(this); |
| 148 } | 150 } |
| 149 | 151 |
| 150 AsyncProxyServerSocket* SocksProxyServer::WrapSocket(AsyncSocket* socket) { | 152 AsyncProxyServerSocket* SocksProxyServer::WrapSocket(AsyncSocket* socket) { |
| 151 return new AsyncSocksProxyServerSocket(socket); | 153 return new AsyncSocksProxyServerSocket(socket); |
| 152 } | 154 } |
| 153 | 155 |
| 154 } // namespace rtc | 156 } // namespace rtc |
| OLD | NEW |