blob: 4dba0dd8f34be4c62c843e53d3963e7008741fed [file] [log] [blame]
henrike@webrtc.org47be73b2014-05-13 18:00:261/*
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
jbauch1286d0e2016-04-26 10:13:2211#include <memory>
henrike@webrtc.org47be73b2014-05-13 18:00:2612#include <string>
13#include "webrtc/base/autodetectproxy.h"
14#include "webrtc/base/gunit.h"
15#include "webrtc/base/httpserver.h"
16#include "webrtc/base/proxyserver.h"
17#include "webrtc/base/socketadapters.h"
18#include "webrtc/base/testclient.h"
19#include "webrtc/base/testechoserver.h"
20#include "webrtc/base/virtualsocketserver.h"
21
22using rtc::Socket;
23using rtc::Thread;
24using rtc::SocketAddress;
25
26static const SocketAddress kSocksProxyIntAddr("1.2.3.4", 1080);
27static const SocketAddress kSocksProxyExtAddr("1.2.3.5", 0);
28static const SocketAddress kHttpsProxyIntAddr("1.2.3.4", 443);
29static const SocketAddress kHttpsProxyExtAddr("1.2.3.5", 0);
30static const SocketAddress kBogusProxyIntAddr("1.2.3.4", 999);
31
32// Used to run a proxy detect on the current thread. Otherwise we would need
33// to make both threads share the same VirtualSocketServer.
34class AutoDetectProxyRunner : public rtc::AutoDetectProxy {
35 public:
36 explicit AutoDetectProxyRunner(const std::string& agent)
37 : AutoDetectProxy(agent) {}
38 void Run() {
39 DoWork();
40 Thread::Current()->Restart(); // needed to reset the messagequeue
41 }
42};
43
44// Sets up a virtual socket server and HTTPS/SOCKS5 proxy servers.
45class ProxyTest : public testing::Test {
46 public:
47 ProxyTest() : ss_(new rtc::VirtualSocketServer(NULL)) {
48 Thread::Current()->set_socketserver(ss_.get());
49 socks_.reset(new rtc::SocksProxyServer(
50 ss_.get(), kSocksProxyIntAddr, ss_.get(), kSocksProxyExtAddr));
51 https_.reset(new rtc::HttpListenServer());
52 https_->Listen(kHttpsProxyIntAddr);
53 }
54 ~ProxyTest() {
55 Thread::Current()->set_socketserver(NULL);
56 }
57
58 rtc::SocketServer* ss() { return ss_.get(); }
59
60 rtc::ProxyType DetectProxyType(const SocketAddress& address) {
61 rtc::ProxyType type;
62 AutoDetectProxyRunner* detect = new AutoDetectProxyRunner("unittest/1.0");
63 detect->set_proxy(address);
64 detect->Run(); // blocks until done
65 type = detect->proxy().type;
66 detect->Destroy(false);
67 return type;
68 }
69
70 private:
jbauch1286d0e2016-04-26 10:13:2271 std::unique_ptr<rtc::SocketServer> ss_;
72 std::unique_ptr<rtc::SocksProxyServer> socks_;
henrike@webrtc.org47be73b2014-05-13 18:00:2673 // TODO: Make this a real HTTPS proxy server.
jbauch1286d0e2016-04-26 10:13:2274 std::unique_ptr<rtc::HttpListenServer> https_;
henrike@webrtc.org47be73b2014-05-13 18:00:2675};
76
77// Tests whether we can use a SOCKS5 proxy to connect to a server.
henrike@webrtc.orgd0892c42014-10-09 22:08:1578TEST_F(ProxyTest, TestSocks5Connect) {
henrike@webrtc.org47be73b2014-05-13 18:00:2679 rtc::AsyncSocket* socket =
80 ss()->CreateAsyncSocket(kSocksProxyIntAddr.family(), SOCK_STREAM);
81 rtc::AsyncSocksProxySocket* proxy_socket =
82 new rtc::AsyncSocksProxySocket(socket, kSocksProxyIntAddr,
83 "", rtc::CryptString());
84 // TODO: IPv6-ize these tests when proxy supports IPv6.
85
86 rtc::TestEchoServer server(Thread::Current(),
87 SocketAddress(INADDR_ANY, 0));
88
89 rtc::AsyncTCPSocket* packet_socket = rtc::AsyncTCPSocket::Create(
90 proxy_socket, SocketAddress(INADDR_ANY, 0), server.address());
91 EXPECT_TRUE(packet_socket != NULL);
92 rtc::TestClient client(packet_socket);
93
94 EXPECT_EQ(Socket::CS_CONNECTING, proxy_socket->GetState());
95 EXPECT_TRUE(client.CheckConnected());
96 EXPECT_EQ(Socket::CS_CONNECTED, proxy_socket->GetState());
97 EXPECT_EQ(server.address(), client.remote_address());
98 client.Send("foo", 3);
99 EXPECT_TRUE(client.CheckNextPacket("foo", 3, NULL));
100 EXPECT_TRUE(client.CheckNoPacket());
101}
102
103/*
104// Tests whether we can use a HTTPS proxy to connect to a server.
105TEST_F(ProxyTest, TestHttpsConnect) {
106 AsyncSocket* socket = ss()->CreateAsyncSocket(SOCK_STREAM);
107 AsyncHttpsProxySocket* proxy_socket = new AsyncHttpsProxySocket(
108 socket, "unittest/1.0", kHttpsProxyIntAddress, "", CryptString());
109 TestClient client(new AsyncTCPSocket(proxy_socket));
110 TestEchoServer server(Thread::Current(), SocketAddress());
111
112 EXPECT_TRUE(client.Connect(server.address()));
113 EXPECT_TRUE(client.CheckConnected());
114 EXPECT_EQ(server.address(), client.remote_address());
115 client.Send("foo", 3);
116 EXPECT_TRUE(client.CheckNextPacket("foo", 3, NULL));
117 EXPECT_TRUE(client.CheckNoPacket());
118}
119*/
120
121// Tests whether we can autodetect a SOCKS5 proxy.
henrike@webrtc.orgd0892c42014-10-09 22:08:15122TEST_F(ProxyTest, TestAutoDetectSocks5) {
henrike@webrtc.org47be73b2014-05-13 18:00:26123 EXPECT_EQ(rtc::PROXY_SOCKS5, DetectProxyType(kSocksProxyIntAddr));
124}
125
126/*
127// Tests whether we can autodetect a HTTPS proxy.
128TEST_F(ProxyTest, TestAutoDetectHttps) {
129 EXPECT_EQ(rtc::PROXY_HTTPS, DetectProxyType(kHttpsProxyIntAddr));
130}
131*/
132
133// Tests whether we fail properly for no proxy.
henrike@webrtc.orgd0892c42014-10-09 22:08:15134TEST_F(ProxyTest, TestAutoDetectBogus) {
henrike@webrtc.org47be73b2014-05-13 18:00:26135 EXPECT_EQ(rtc::PROXY_UNKNOWN, DetectProxyType(kBogusProxyIntAddr));
136}