1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/memory/ref_counted.h"
6 #include "base/strings/stringprintf.h"
7 #include "chrome/browser/extensions/api/dns/mock_host_resolver_creator.h"
8 #include "chrome/browser/extensions/extension_apitest.h"
9 #include "chrome/browser/extensions/extension_function_test_utils.h"
10 #include "chrome/browser/extensions/extension_service.h"
11 #include "chrome/browser/extensions/extension_test_message_listener.h"
12 #include "chrome/browser/ui/browser.h"
13 #include "chrome/common/chrome_paths.h"
14 #include "chrome/test/base/in_process_browser_test.h"
15 #include "chrome/test/base/ui_test_utils.h"
16 #include "extensions/browser/api/dns/host_resolver_wrapper.h"
17 #include "extensions/browser/api/socket/socket_api.h"
18 #include "net/dns/mock_host_resolver.h"
19 #include "net/test/spawned_test_server/spawned_test_server.h"
20
21 using extensions::Extension;
22
23 namespace utils = extension_function_test_utils;
24
25 namespace {
26
27 const std::string kHostname = "127.0.0.1";
28 const int kPort = 8888;
29
30 class SocketApiTest : public ExtensionApiTest {
31 public:
SocketApiTest()32 SocketApiTest() : resolver_event_(true, false),
33 resolver_creator_(
34 new extensions::MockHostResolverCreator()) {
35 }
36
SetUpOnMainThread()37 virtual void SetUpOnMainThread() OVERRIDE {
38 extensions::HostResolverWrapper::GetInstance()->SetHostResolverForTesting(
39 resolver_creator_->CreateMockHostResolver());
40 }
41
CleanUpOnMainThread()42 virtual void CleanUpOnMainThread() OVERRIDE {
43 extensions::HostResolverWrapper::GetInstance()->
44 SetHostResolverForTesting(NULL);
45 resolver_creator_->DeleteMockHostResolver();
46 }
47
48 private:
49 base::WaitableEvent resolver_event_;
50
51 // The MockHostResolver asserts that it's used on the same thread on which
52 // it's created, which is actually a stronger rule than its real counterpart.
53 // But that's fine; it's good practice.
54 scoped_refptr<extensions::MockHostResolverCreator> resolver_creator_;
55 };
56
57 } // namespace
58
IN_PROC_BROWSER_TEST_F(SocketApiTest,SocketUDPCreateGood)59 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketUDPCreateGood) {
60 scoped_refptr<extensions::SocketCreateFunction> socket_create_function(
61 new extensions::SocketCreateFunction());
62 scoped_refptr<Extension> empty_extension(utils::CreateEmptyExtension());
63
64 socket_create_function->set_extension(empty_extension.get());
65 socket_create_function->set_has_callback(true);
66
67 scoped_ptr<base::Value> result(utils::RunFunctionAndReturnSingleResult(
68 socket_create_function.get(), "[\"udp\"]", browser(), utils::NONE));
69 ASSERT_EQ(base::Value::TYPE_DICTIONARY, result->GetType());
70 base::DictionaryValue *value =
71 static_cast<base::DictionaryValue*>(result.get());
72 int socket_id = -1;
73 EXPECT_TRUE(value->GetInteger("socketId", &socket_id));
74 EXPECT_GT(socket_id, 0);
75 }
76
IN_PROC_BROWSER_TEST_F(SocketApiTest,SocketTCPCreateGood)77 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPCreateGood) {
78 scoped_refptr<extensions::SocketCreateFunction> socket_create_function(
79 new extensions::SocketCreateFunction());
80 scoped_refptr<Extension> empty_extension(utils::CreateEmptyExtension());
81
82 socket_create_function->set_extension(empty_extension.get());
83 socket_create_function->set_has_callback(true);
84
85 scoped_ptr<base::Value> result(utils::RunFunctionAndReturnSingleResult(
86 socket_create_function.get(), "[\"tcp\"]", browser(), utils::NONE));
87 ASSERT_EQ(base::Value::TYPE_DICTIONARY, result->GetType());
88 base::DictionaryValue *value =
89 static_cast<base::DictionaryValue*>(result.get());
90 int socket_id = -1;
91 EXPECT_TRUE(value->GetInteger("socketId", &socket_id));
92 ASSERT_GT(socket_id, 0);
93 }
94
IN_PROC_BROWSER_TEST_F(SocketApiTest,GetNetworkList)95 IN_PROC_BROWSER_TEST_F(SocketApiTest, GetNetworkList) {
96 scoped_refptr<extensions::SocketGetNetworkListFunction> socket_function(
97 new extensions::SocketGetNetworkListFunction());
98 scoped_refptr<Extension> empty_extension(utils::CreateEmptyExtension());
99
100 socket_function->set_extension(empty_extension.get());
101 socket_function->set_has_callback(true);
102
103 scoped_ptr<base::Value> result(utils::RunFunctionAndReturnSingleResult(
104 socket_function.get(), "[]", browser(), utils::NONE));
105 ASSERT_EQ(base::Value::TYPE_LIST, result->GetType());
106
107 // If we're invoking socket tests, all we can confirm is that we have at
108 // least one address, but not what it is.
109 base::ListValue *value = static_cast<base::ListValue*>(result.get());
110 ASSERT_GT(value->GetSize(), 0U);
111 }
112
IN_PROC_BROWSER_TEST_F(SocketApiTest,SocketUDPExtension)113 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketUDPExtension) {
114 scoped_ptr<net::SpawnedTestServer> test_server(
115 new net::SpawnedTestServer(
116 net::SpawnedTestServer::TYPE_UDP_ECHO,
117 net::SpawnedTestServer::kLocalhost,
118 base::FilePath(FILE_PATH_LITERAL("net/data"))));
119 EXPECT_TRUE(test_server->Start());
120
121 net::HostPortPair host_port_pair = test_server->host_port_pair();
122 int port = host_port_pair.port();
123 ASSERT_GT(port, 0);
124
125 // Test that sendTo() is properly resolving hostnames.
126 host_port_pair.set_host("LOCALhost");
127
128 ResultCatcher catcher;
129 catcher.RestrictToProfile(browser()->profile());
130
131 ExtensionTestMessageListener listener("info_please", true);
132
133 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api")));
134 EXPECT_TRUE(listener.WaitUntilSatisfied());
135 listener.Reply(
136 base::StringPrintf("udp:%s:%d", host_port_pair.host().c_str(), port));
137
138 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
139 }
140
IN_PROC_BROWSER_TEST_F(SocketApiTest,SocketTCPExtension)141 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPExtension) {
142 scoped_ptr<net::SpawnedTestServer> test_server(
143 new net::SpawnedTestServer(
144 net::SpawnedTestServer::TYPE_TCP_ECHO,
145 net::SpawnedTestServer::kLocalhost,
146 base::FilePath(FILE_PATH_LITERAL("net/data"))));
147 EXPECT_TRUE(test_server->Start());
148
149 net::HostPortPair host_port_pair = test_server->host_port_pair();
150 int port = host_port_pair.port();
151 ASSERT_GT(port, 0);
152
153 // Test that connect() is properly resolving hostnames.
154 host_port_pair.set_host("lOcAlHoSt");
155
156 ResultCatcher catcher;
157 catcher.RestrictToProfile(browser()->profile());
158
159 ExtensionTestMessageListener listener("info_please", true);
160
161 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api")));
162 EXPECT_TRUE(listener.WaitUntilSatisfied());
163 listener.Reply(
164 base::StringPrintf("tcp:%s:%d", host_port_pair.host().c_str(), port));
165
166 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
167 }
168
IN_PROC_BROWSER_TEST_F(SocketApiTest,SocketTCPServerExtension)169 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPServerExtension) {
170 ResultCatcher catcher;
171 catcher.RestrictToProfile(browser()->profile());
172 ExtensionTestMessageListener listener("info_please", true);
173 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api")));
174 EXPECT_TRUE(listener.WaitUntilSatisfied());
175 listener.Reply(
176 base::StringPrintf("tcp_server:%s:%d", kHostname.c_str(), kPort));
177
178 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
179 }
180
IN_PROC_BROWSER_TEST_F(SocketApiTest,SocketTCPServerUnbindOnUnload)181 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPServerUnbindOnUnload) {
182 ResultCatcher catcher;
183 const Extension* extension =
184 LoadExtension(test_data_dir_.AppendASCII("socket/unload"));
185 ASSERT_TRUE(extension);
186 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
187
188 UnloadExtension(extension->id());
189
190 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/unload")))
191 << message_;
192 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
193 }
194
IN_PROC_BROWSER_TEST_F(SocketApiTest,SocketMulticast)195 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketMulticast) {
196 ResultCatcher catcher;
197 catcher.RestrictToProfile(browser()->profile());
198 ExtensionTestMessageListener listener("info_please", true);
199 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api")));
200 EXPECT_TRUE(listener.WaitUntilSatisfied());
201 listener.Reply(
202 base::StringPrintf("multicast:%s:%d", kHostname.c_str(), kPort));
203
204 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
205 }
206