deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2009 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 | |
| 11 | #include <memory> |
| 12 | #include <string> |
Jonas Olsson | a4d8737 | 2019-07-05 17:08:33 | [diff] [blame] | 13 | |
Mirko Bonadei | 92ea95e | 2017-09-15 04:47:31 | [diff] [blame] | 14 | #include "rtc_base/gunit.h" |
Steve Anton | 10542f2 | 2019-01-11 17:11:00 | [diff] [blame] | 15 | #include "rtc_base/proxy_server.h" |
| 16 | #include "rtc_base/socket_adapters.h" |
| 17 | #include "rtc_base/test_client.h" |
| 18 | #include "rtc_base/test_echo_server.h" |
| 19 | #include "rtc_base/virtual_socket_server.h" |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 20 | |
| 21 | using rtc::Socket; |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 22 | using rtc::SocketAddress; |
| 23 | |
| 24 | static const SocketAddress kSocksProxyIntAddr("1.2.3.4", 1080); |
| 25 | static const SocketAddress kSocksProxyExtAddr("1.2.3.5", 0); |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 26 | static const SocketAddress kBogusProxyIntAddr("1.2.3.4", 999); |
| 27 | |
Niels Möller | 83da552 | 2018-09-26 14:18:38 | [diff] [blame] | 28 | // Sets up a virtual socket server and a SOCKS5 proxy server. |
Mirko Bonadei | 6a489f2 | 2019-04-09 13:11:12 | [diff] [blame] | 29 | class ProxyTest : public ::testing::Test { |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 30 | public: |
deadbeef | 98e186c | 2017-05-17 01:00:06 | [diff] [blame] | 31 | ProxyTest() : ss_(new rtc::VirtualSocketServer()), thread_(ss_.get()) { |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 32 | socks_.reset(new rtc::SocksProxyServer(ss_.get(), kSocksProxyIntAddr, |
| 33 | ss_.get(), kSocksProxyExtAddr)); |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 34 | } |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 35 | |
| 36 | rtc::SocketServer* ss() { return ss_.get(); } |
| 37 | |
| 38 | private: |
| 39 | std::unique_ptr<rtc::SocketServer> ss_; |
nisse | 7eaa4ea | 2017-05-08 12:25:41 | [diff] [blame] | 40 | rtc::AutoSocketServerThread thread_; |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 41 | std::unique_ptr<rtc::SocksProxyServer> socks_; |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 42 | }; |
| 43 | |
| 44 | // Tests whether we can use a SOCKS5 proxy to connect to a server. |
| 45 | TEST_F(ProxyTest, TestSocks5Connect) { |
| 46 | rtc::AsyncSocket* socket = |
| 47 | ss()->CreateAsyncSocket(kSocksProxyIntAddr.family(), SOCK_STREAM); |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 48 | rtc::AsyncSocksProxySocket* proxy_socket = new rtc::AsyncSocksProxySocket( |
| 49 | socket, kSocksProxyIntAddr, "", rtc::CryptString()); |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 50 | // TODO: IPv6-ize these tests when proxy supports IPv6. |
| 51 | |
Danil Chapovalov | 6dbf0e4 | 2018-11-13 17:07:46 | [diff] [blame] | 52 | rtc::TestEchoServer server(rtc::Thread::Current(), |
| 53 | SocketAddress(INADDR_ANY, 0)); |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 54 | |
nisse | 32f2505 | 2017-05-08 08:57:18 | [diff] [blame] | 55 | std::unique_ptr<rtc::AsyncTCPSocket> packet_socket( |
| 56 | rtc::AsyncTCPSocket::Create(proxy_socket, SocketAddress(INADDR_ANY, 0), |
| 57 | server.address())); |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 58 | EXPECT_TRUE(packet_socket != nullptr); |
nisse | 32f2505 | 2017-05-08 08:57:18 | [diff] [blame] | 59 | rtc::TestClient client(std::move(packet_socket)); |
deadbeef | f137e97 | 2017-03-23 22:45:49 | [diff] [blame] | 60 | |
| 61 | EXPECT_EQ(Socket::CS_CONNECTING, proxy_socket->GetState()); |
| 62 | EXPECT_TRUE(client.CheckConnected()); |
| 63 | EXPECT_EQ(Socket::CS_CONNECTED, proxy_socket->GetState()); |
| 64 | EXPECT_EQ(server.address(), client.remote_address()); |
| 65 | client.Send("foo", 3); |
| 66 | EXPECT_TRUE(client.CheckNextPacket("foo", 3, nullptr)); |
| 67 | EXPECT_TRUE(client.CheckNoPacket()); |
| 68 | } |