blob: ec767826827834f4bdc374a2a78e74fc6cb87b25 [file] [log] [blame]
henrike@webrtc.orgf0488722014-05-13 18:00:261/*
2 * Copyright 2004 The WebRTC Project Authors. All rights reserved.
3 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Steve Anton10542f22019-01-11 17:11:0011#include "rtc_base/proxy_server.h"
henrike@webrtc.orgf0488722014-05-13 18:00:2612
Yves Gerey3e707812018-11-28 15:47:4913#include <stddef.h>
nisseede5da42017-01-12 13:15:3614
Niels Möllerd9709672019-03-25 07:57:5015#include "absl/memory/memory.h"
Mirko Bonadei92ea95e2017-09-15 04:47:3116#include "rtc_base/checks.h"
Yves Gerey2e00abc2018-10-05 13:39:2417#include "rtc_base/logging.h"
Steve Anton10542f22019-01-11 17:11:0018#include "rtc_base/socket_factory.h"
henrike@webrtc.orgf0488722014-05-13 18:00:2619
20namespace rtc {
21
22// ProxyServer
Yves Gerey665174f2018-06-19 13:03:0523ProxyServer::ProxyServer(SocketFactory* int_factory,
24 const SocketAddress& int_addr,
25 SocketFactory* ext_factory,
26 const SocketAddress& ext_ip)
27 : ext_factory_(ext_factory),
28 ext_ip_(ext_ip.ipaddr(), 0), // strip off port
29 server_socket_(
30 int_factory->CreateAsyncSocket(int_addr.family(), SOCK_STREAM)) {
deadbeef37f5ecf2017-02-27 22:06:4131 RTC_DCHECK(server_socket_.get() != nullptr);
nisseede5da42017-01-12 13:15:3632 RTC_DCHECK(int_addr.family() == AF_INET || int_addr.family() == AF_INET6);
henrike@webrtc.orgf0488722014-05-13 18:00:2633 server_socket_->Bind(int_addr);
34 server_socket_->Listen(5);
35 server_socket_->SignalReadEvent.connect(this, &ProxyServer::OnAcceptEvent);
36}
37
Niels Möllerd9709672019-03-25 07:57:5038ProxyServer::~ProxyServer() = default;
henrike@webrtc.orgf0488722014-05-13 18:00:2639
deadbeefc5d0d952015-07-16 17:22:2140SocketAddress ProxyServer::GetServerAddress() {
41 return server_socket_->GetLocalAddress();
42}
43
henrike@webrtc.orgf0488722014-05-13 18:00:2644void ProxyServer::OnAcceptEvent(AsyncSocket* socket) {
kwibergee89e782017-08-10 00:22:0145 RTC_DCHECK(socket);
46 RTC_DCHECK_EQ(socket, server_socket_.get());
deadbeef37f5ecf2017-02-27 22:06:4147 AsyncSocket* int_socket = socket->Accept(nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:2648 AsyncProxyServerSocket* wrapped_socket = WrapSocket(int_socket);
Yves Gerey665174f2018-06-19 13:03:0549 AsyncSocket* ext_socket =
50 ext_factory_->CreateAsyncSocket(ext_ip_.family(), SOCK_STREAM);
henrike@webrtc.orgf0488722014-05-13 18:00:2651 if (ext_socket) {
52 ext_socket->Bind(ext_ip_);
Niels Möllerd9709672019-03-25 07:57:5053 bindings_.emplace_back(
54 absl::make_unique<ProxyBinding>(wrapped_socket, ext_socket));
henrike@webrtc.orgf0488722014-05-13 18:00:2655 } else {
Mirko Bonadei675513b2017-11-09 10:09:2556 RTC_LOG(LS_ERROR)
57 << "Unable to create external socket on proxy accept event";
henrike@webrtc.orgf0488722014-05-13 18:00:2658 }
59}
60
henrike@webrtc.orgf0488722014-05-13 18:00:2661// ProxyBinding
62ProxyBinding::ProxyBinding(AsyncProxyServerSocket* int_socket,
63 AsyncSocket* ext_socket)
Yves Gerey665174f2018-06-19 13:03:0564 : int_socket_(int_socket),
65 ext_socket_(ext_socket),
66 connected_(false),
67 out_buffer_(kBufferSize),
68 in_buffer_(kBufferSize) {
henrike@webrtc.orgf0488722014-05-13 18:00:2669 int_socket_->SignalConnectRequest.connect(this,
70 &ProxyBinding::OnConnectRequest);
71 int_socket_->SignalReadEvent.connect(this, &ProxyBinding::OnInternalRead);
72 int_socket_->SignalWriteEvent.connect(this, &ProxyBinding::OnInternalWrite);
73 int_socket_->SignalCloseEvent.connect(this, &ProxyBinding::OnInternalClose);
74 ext_socket_->SignalConnectEvent.connect(this,
75 &ProxyBinding::OnExternalConnect);
76 ext_socket_->SignalReadEvent.connect(this, &ProxyBinding::OnExternalRead);
77 ext_socket_->SignalWriteEvent.connect(this, &ProxyBinding::OnExternalWrite);
78 ext_socket_->SignalCloseEvent.connect(this, &ProxyBinding::OnExternalClose);
79}
80
kwiberg@webrtc.org67186fe2015-03-09 22:21:5381ProxyBinding::~ProxyBinding() = default;
82
henrike@webrtc.orgf0488722014-05-13 18:00:2683void ProxyBinding::OnConnectRequest(AsyncProxyServerSocket* socket,
Yves Gerey665174f2018-06-19 13:03:0584 const SocketAddress& addr) {
kwibergee89e782017-08-10 00:22:0185 RTC_DCHECK(!connected_);
86 RTC_DCHECK(ext_socket_);
henrike@webrtc.orgf0488722014-05-13 18:00:2687 ext_socket_->Connect(addr);
88 // TODO: handle errors here
89}
90
91void ProxyBinding::OnInternalRead(AsyncSocket* socket) {
92 Read(int_socket_.get(), &out_buffer_);
93 Write(ext_socket_.get(), &out_buffer_);
94}
95
96void ProxyBinding::OnInternalWrite(AsyncSocket* socket) {
97 Write(int_socket_.get(), &in_buffer_);
98}
99
100void ProxyBinding::OnInternalClose(AsyncSocket* socket, int err) {
101 Destroy();
102}
103
104void ProxyBinding::OnExternalConnect(AsyncSocket* socket) {
deadbeef37f5ecf2017-02-27 22:06:41105 RTC_DCHECK(socket != nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26106 connected_ = true;
107 int_socket_->SendConnectResult(0, socket->GetRemoteAddress());
108}
109
110void ProxyBinding::OnExternalRead(AsyncSocket* socket) {
111 Read(ext_socket_.get(), &in_buffer_);
112 Write(int_socket_.get(), &in_buffer_);
113}
114
115void ProxyBinding::OnExternalWrite(AsyncSocket* socket) {
116 Write(ext_socket_.get(), &out_buffer_);
117}
118
119void ProxyBinding::OnExternalClose(AsyncSocket* socket, int err) {
120 if (!connected_) {
121 int_socket_->SendConnectResult(err, SocketAddress());
122 }
123 Destroy();
124}
125
126void ProxyBinding::Read(AsyncSocket* socket, FifoBuffer* buffer) {
127 // Only read if the buffer is empty.
deadbeef37f5ecf2017-02-27 22:06:41128 RTC_DCHECK(socket != nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26129 size_t size;
130 int read;
131 if (buffer->GetBuffered(&size) && size == 0) {
132 void* p = buffer->GetWriteBuffer(&size);
Stefan Holmer9131efd2016-05-23 16:19:26133 read = socket->Recv(p, size, nullptr);
andresp@webrtc.orgff689be2015-02-12 11:54:26134 buffer->ConsumeWriteBuffer(std::max(read, 0));
henrike@webrtc.orgf0488722014-05-13 18:00:26135 }
136}
137
138void ProxyBinding::Write(AsyncSocket* socket, FifoBuffer* buffer) {
deadbeef37f5ecf2017-02-27 22:06:41139 RTC_DCHECK(socket != nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26140 size_t size;
141 int written;
142 const void* p = buffer->GetReadData(&size);
143 written = socket->Send(p, size);
andresp@webrtc.orgff689be2015-02-12 11:54:26144 buffer->ConsumeReadData(std::max(written, 0));
henrike@webrtc.orgf0488722014-05-13 18:00:26145}
146
147void ProxyBinding::Destroy() {
148 SignalDestroyed(this);
149}
150
deadbeeff137e972017-03-23 22:45:49151AsyncProxyServerSocket* SocksProxyServer::WrapSocket(AsyncSocket* socket) {
152 return new AsyncSocksProxyServerSocket(socket);
153}
154
henrike@webrtc.orgf0488722014-05-13 18:00:26155} // namespace rtc