// Copyright (c) 2012 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include "base/memory/ref_counted.h" #include "base/strings/stringprintf.h" #include "chrome/browser/extensions/extension_apitest.h" #include "chrome/browser/extensions/extension_service.h" #include "chrome/browser/ui/browser.h" #include "chrome/common/chrome_paths.h" #include "chrome/test/base/in_process_browser_test.h" #include "extensions/browser/api/dns/host_resolver_wrapper.h" #include "extensions/browser/api/dns/mock_host_resolver_creator.h" #include "extensions/browser/api/socket/socket_api.h" #include "extensions/test/extension_test_message_listener.h" #include "extensions/test/result_catcher.h" #include "net/dns/mock_host_resolver.h" #include "net/test/spawned_test_server/spawned_test_server.h" using extensions::Extension; using extensions::ResultCatcher; namespace { const std::string kHostname = "127.0.0.1"; const int kPort = 8888; class SocketApiTest : public ExtensionApiTest { public: SocketApiTest() : resolver_event_(true, false), resolver_creator_( new extensions::MockHostResolverCreator()) { } void SetUpOnMainThread() override { extensions::HostResolverWrapper::GetInstance()->SetHostResolverForTesting( resolver_creator_->CreateMockHostResolver()); } void TearDownOnMainThread() override { extensions::HostResolverWrapper::GetInstance()-> SetHostResolverForTesting(NULL); resolver_creator_->DeleteMockHostResolver(); } private: base::WaitableEvent resolver_event_; // The MockHostResolver asserts that it's used on the same thread on which // it's created, which is actually a stronger rule than its real counterpart. // But that's fine; it's good practice. scoped_refptr resolver_creator_; }; } // namespace IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketUDPExtension) { scoped_ptr test_server( new net::SpawnedTestServer( net::SpawnedTestServer::TYPE_UDP_ECHO, net::SpawnedTestServer::kLocalhost, base::FilePath(FILE_PATH_LITERAL("net/data")))); EXPECT_TRUE(test_server->Start()); net::HostPortPair host_port_pair = test_server->host_port_pair(); int port = host_port_pair.port(); ASSERT_GT(port, 0); // Test that sendTo() is properly resolving hostnames. host_port_pair.set_host("LOCALhost"); ResultCatcher catcher; catcher.RestrictToBrowserContext(browser()->profile()); ExtensionTestMessageListener listener("info_please", true); ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); EXPECT_TRUE(listener.WaitUntilSatisfied()); listener.Reply( base::StringPrintf("udp:%s:%d", host_port_pair.host().c_str(), port)); EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); } IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPExtension) { scoped_ptr test_server( new net::SpawnedTestServer( net::SpawnedTestServer::TYPE_TCP_ECHO, net::SpawnedTestServer::kLocalhost, base::FilePath(FILE_PATH_LITERAL("net/data")))); EXPECT_TRUE(test_server->Start()); net::HostPortPair host_port_pair = test_server->host_port_pair(); int port = host_port_pair.port(); ASSERT_GT(port, 0); // Test that connect() is properly resolving hostnames. host_port_pair.set_host("lOcAlHoSt"); ResultCatcher catcher; catcher.RestrictToBrowserContext(browser()->profile()); ExtensionTestMessageListener listener("info_please", true); ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); EXPECT_TRUE(listener.WaitUntilSatisfied()); listener.Reply( base::StringPrintf("tcp:%s:%d", host_port_pair.host().c_str(), port)); EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); } IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPServerExtension) { ResultCatcher catcher; catcher.RestrictToBrowserContext(browser()->profile()); ExtensionTestMessageListener listener("info_please", true); ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); EXPECT_TRUE(listener.WaitUntilSatisfied()); listener.Reply( base::StringPrintf("tcp_server:%s:%d", kHostname.c_str(), kPort)); EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); } IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPServerUnbindOnUnload) { ResultCatcher catcher; const Extension* extension = LoadExtension(test_data_dir_.AppendASCII("socket/unload")); ASSERT_TRUE(extension); EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); UnloadExtension(extension->id()); ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/unload"))) << message_; EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); } IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketMulticast) { ResultCatcher catcher; catcher.RestrictToBrowserContext(browser()->profile()); ExtensionTestMessageListener listener("info_please", true); ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); EXPECT_TRUE(listener.WaitUntilSatisfied()); listener.Reply( base::StringPrintf("multicast:%s:%d", kHostname.c_str(), kPort)); EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); }