OLD | NEW |
| (Empty) |
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 "remoting/jingle_glue/chromium_socket_factory.h" | |
6 | |
7 #include "base/message_loop/message_loop.h" | |
8 #include "base/run_loop.h" | |
9 #include "testing/gmock/include/gmock/gmock.h" | |
10 #include "testing/gtest/include/gtest/gtest.h" | |
11 #include "third_party/libjingle/source/talk/base/asyncpacketsocket.h" | |
12 #include "third_party/libjingle/source/talk/base/socketaddress.h" | |
13 | |
14 namespace remoting { | |
15 | |
16 class ChromiumSocketFactoryTest : public testing::Test, | |
17 public sigslot::has_slots<> { | |
18 public: | |
19 virtual void SetUp() OVERRIDE { | |
20 socket_factory_.reset(new ChromiumPacketSocketFactory()); | |
21 | |
22 socket_.reset(socket_factory_->CreateUdpSocket( | |
23 talk_base::SocketAddress("127.0.0.1", 0), 0, 0)); | |
24 ASSERT_TRUE(socket_.get() != NULL); | |
25 EXPECT_EQ(socket_->GetState(), talk_base::AsyncPacketSocket::STATE_BOUND); | |
26 socket_->SignalReadPacket.connect( | |
27 this, &ChromiumSocketFactoryTest::OnPacket); | |
28 } | |
29 | |
30 void OnPacket(talk_base::AsyncPacketSocket* socket, | |
31 const char* data, size_t size, | |
32 const talk_base::SocketAddress& address, | |
33 const talk_base::PacketTime& packet_time) { | |
34 EXPECT_EQ(socket, socket_.get()); | |
35 last_packet_.assign(data, data + size); | |
36 last_address_ = address; | |
37 run_loop_.Quit(); | |
38 } | |
39 | |
40 void VerifyCanSendAndReceive(talk_base::AsyncPacketSocket* sender) { | |
41 // UDP packets may be lost, so we have to retry sending it more than once. | |
42 const int kMaxAttempts = 3; | |
43 const base::TimeDelta kAttemptPeriod = base::TimeDelta::FromSeconds(1); | |
44 std::string test_packet("TEST PACKET"); | |
45 int attempts = 0; | |
46 talk_base::PacketOptions options; | |
47 while (last_packet_.empty() && attempts++ < kMaxAttempts) { | |
48 sender->SendTo(test_packet.data(), test_packet.size(), | |
49 socket_->GetLocalAddress(), options); | |
50 message_loop_.PostDelayedTask(FROM_HERE, run_loop_.QuitClosure(), | |
51 kAttemptPeriod); | |
52 run_loop_.Run(); | |
53 } | |
54 EXPECT_EQ(test_packet, last_packet_); | |
55 EXPECT_EQ(sender->GetLocalAddress(), last_address_); | |
56 } | |
57 | |
58 protected: | |
59 base::MessageLoopForIO message_loop_; | |
60 base::RunLoop run_loop_; | |
61 | |
62 scoped_ptr<talk_base::PacketSocketFactory> socket_factory_; | |
63 scoped_ptr<talk_base::AsyncPacketSocket> socket_; | |
64 | |
65 std::string last_packet_; | |
66 talk_base::SocketAddress last_address_; | |
67 }; | |
68 | |
69 TEST_F(ChromiumSocketFactoryTest, SendAndReceive) { | |
70 scoped_ptr<talk_base::AsyncPacketSocket> sending_socket( | |
71 socket_factory_->CreateUdpSocket( | |
72 talk_base::SocketAddress("127.0.0.1", 0), 0, 0)); | |
73 ASSERT_TRUE(sending_socket.get() != NULL); | |
74 EXPECT_EQ(sending_socket->GetState(), | |
75 talk_base::AsyncPacketSocket::STATE_BOUND); | |
76 | |
77 VerifyCanSendAndReceive(sending_socket.get()); | |
78 } | |
79 | |
80 TEST_F(ChromiumSocketFactoryTest, SetOptions) { | |
81 EXPECT_EQ(0, socket_->SetOption(talk_base::Socket::OPT_SNDBUF, 4096)); | |
82 EXPECT_EQ(0, socket_->SetOption(talk_base::Socket::OPT_RCVBUF, 4096)); | |
83 } | |
84 | |
85 TEST_F(ChromiumSocketFactoryTest, PortRange) { | |
86 const int kMinPort = 12400; | |
87 const int kMaxPort = 12410; | |
88 socket_.reset(socket_factory_->CreateUdpSocket( | |
89 talk_base::SocketAddress("127.0.0.1", 0), kMaxPort, kMaxPort)); | |
90 ASSERT_TRUE(socket_.get() != NULL); | |
91 EXPECT_EQ(socket_->GetState(), talk_base::AsyncPacketSocket::STATE_BOUND); | |
92 EXPECT_GE(socket_->GetLocalAddress().port(), kMinPort); | |
93 EXPECT_LE(socket_->GetLocalAddress().port(), kMaxPort); | |
94 } | |
95 | |
96 TEST_F(ChromiumSocketFactoryTest, TransientError) { | |
97 scoped_ptr<talk_base::AsyncPacketSocket> sending_socket( | |
98 socket_factory_->CreateUdpSocket( | |
99 talk_base::SocketAddress("127.0.0.1", 0), 0, 0)); | |
100 std::string test_packet("TEST"); | |
101 | |
102 // Try sending a packet to an IPv6 address from a socket that's bound to an | |
103 // IPv4 address. This send is expected to fail, but the socket should still be | |
104 // functional. | |
105 sending_socket->SendTo(test_packet.data(), test_packet.size(), | |
106 talk_base::SocketAddress("::1", 0), | |
107 talk_base::PacketOptions()); | |
108 | |
109 // Verify that socket is still usable. | |
110 VerifyCanSendAndReceive(sending_socket.get()); | |
111 } | |
112 | |
113 } // namespace remoting | |
OLD | NEW |