OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/extensions/api/socket/udp_socket.h" | 5 #include "chrome/browser/extensions/api/socket/udp_socket.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
10 #include "base/message_loop.h" | 10 #include "base/message_loop.h" |
(...skipping 21 matching lines...) Expand all Loading... |
32 } | 32 } |
33 | 33 |
34 static const char test_message[] = "$$TESTMESSAGETESTMESSAGETESTMESSAGETEST$$"; | 34 static const char test_message[] = "$$TESTMESSAGETESTMESSAGETESTMESSAGETEST$$"; |
35 static const int test_message_length = ARRAYSIZE_UNSAFE(test_message); | 35 static const int test_message_length = ARRAYSIZE_UNSAFE(test_message); |
36 | 36 |
37 static void OnSendCompleted(int result) { | 37 static void OnSendCompleted(int result) { |
38 EXPECT_EQ(test_message_length, result); | 38 EXPECT_EQ(test_message_length, result); |
39 } | 39 } |
40 | 40 |
41 TEST(UDPSocketUnitTest, TestUDPSocketRecvFrom) { | 41 TEST(UDPSocketUnitTest, TestUDPSocketRecvFrom) { |
42 MessageLoopForIO io_loop; // For RecvFrom to do its threaded work. | 42 base::MessageLoopForIO io_loop; // For RecvFrom to do its threaded work. |
43 UDPSocket socket("abcdefghijklmnopqrst"); | 43 UDPSocket socket("abcdefghijklmnopqrst"); |
44 | 44 |
45 // Confirm that we can call two RecvFroms in quick succession without | 45 // Confirm that we can call two RecvFroms in quick succession without |
46 // triggering crbug.com/146606. | 46 // triggering crbug.com/146606. |
47 socket.Connect("127.0.0.1", 40000, base::Bind(&OnConnected)); | 47 socket.Connect("127.0.0.1", 40000, base::Bind(&OnConnected)); |
48 socket.RecvFrom(4096, base::Bind(&OnCompleted)); | 48 socket.RecvFrom(4096, base::Bind(&OnCompleted)); |
49 socket.RecvFrom(4096, base::Bind(&OnCompleted)); | 49 socket.RecvFrom(4096, base::Bind(&OnCompleted)); |
50 } | 50 } |
51 | 51 |
52 TEST(UDPSocketUnitTest, TestUDPMulticastJoinGroup) { | 52 TEST(UDPSocketUnitTest, TestUDPMulticastJoinGroup) { |
(...skipping 21 matching lines...) Expand all Loading... |
74 } | 74 } |
75 | 75 |
76 TEST(UDPSocketUnitTest, TestUDPMulticastLoopbackMode) { | 76 TEST(UDPSocketUnitTest, TestUDPMulticastLoopbackMode) { |
77 const char* kGroup = "237.132.100.17"; | 77 const char* kGroup = "237.132.100.17"; |
78 UDPSocket socket("abcdefghijklmnopqrst"); | 78 UDPSocket socket("abcdefghijklmnopqrst"); |
79 EXPECT_EQ(0, socket.SetMulticastLoopbackMode(false)); | 79 EXPECT_EQ(0, socket.SetMulticastLoopbackMode(false)); |
80 socket.Connect(kGroup, 13333, base::Bind(&OnConnected)); | 80 socket.Connect(kGroup, 13333, base::Bind(&OnConnected)); |
81 } | 81 } |
82 | 82 |
83 static void QuitMessageLoop() { | 83 static void QuitMessageLoop() { |
84 MessageLoopForIO::current()->QuitNow(); | 84 base::MessageLoopForIO::current()->QuitNow(); |
85 } | 85 } |
86 | 86 |
87 // Send a test multicast packet every second. | 87 // Send a test multicast packet every second. |
88 // Once the target socket received the packet, the message loop will exit. | 88 // Once the target socket received the packet, the message loop will exit. |
89 static void SendMulticastPacket(UDPSocket* src, int result) { | 89 static void SendMulticastPacket(UDPSocket* src, int result) { |
90 if (result == 0) { | 90 if (result == 0) { |
91 scoped_refptr<net::IOBuffer> data = new net::WrappedIOBuffer(test_message); | 91 scoped_refptr<net::IOBuffer> data = new net::WrappedIOBuffer(test_message); |
92 src->Write(data, test_message_length, base::Bind(&OnSendCompleted)); | 92 src->Write(data, test_message_length, base::Bind(&OnSendCompleted)); |
93 MessageLoopForIO::current()->PostDelayedTask(FROM_HERE, | 93 base::MessageLoopForIO::current()->PostDelayedTask(FROM_HERE, |
94 base::Bind(&SendMulticastPacket, src, result), | 94 base::Bind(&SendMulticastPacket, src, result), |
95 base::TimeDelta::FromSeconds(1)); | 95 base::TimeDelta::FromSeconds(1)); |
96 } else { | 96 } else { |
97 QuitMessageLoop(); | 97 QuitMessageLoop(); |
98 FAIL() << "Failed to connect to multicast address. Error code: " << result; | 98 FAIL() << "Failed to connect to multicast address. Error code: " << result; |
99 } | 99 } |
100 } | 100 } |
101 | 101 |
102 static void OnMulticastReadCompleted(bool *packet_received, | 102 static void OnMulticastReadCompleted(bool *packet_received, |
103 int count, | 103 int count, |
104 scoped_refptr<net::IOBuffer> io_buffer) { | 104 scoped_refptr<net::IOBuffer> io_buffer) { |
105 EXPECT_EQ(test_message_length, count); | 105 EXPECT_EQ(test_message_length, count); |
106 EXPECT_EQ(0, strncmp(io_buffer->data(), test_message, test_message_length)); | 106 EXPECT_EQ(0, strncmp(io_buffer->data(), test_message, test_message_length)); |
107 *packet_received = true; | 107 *packet_received = true; |
108 QuitMessageLoop(); | 108 QuitMessageLoop(); |
109 } | 109 } |
110 | 110 |
111 TEST(UDPSocketUnitTest, TestUDPMulticastRecv) { | 111 TEST(UDPSocketUnitTest, TestUDPMulticastRecv) { |
112 const int kPort = 9999; | 112 const int kPort = 9999; |
113 const char* const kGroup = "237.132.100.17"; | 113 const char* const kGroup = "237.132.100.17"; |
114 bool packet_received = false; | 114 bool packet_received = false; |
115 MessageLoopForIO io_loop; // For Read to do its threaded work. | 115 base::MessageLoopForIO io_loop; // For Read to do its threaded work. |
116 UDPSocket dest("abcdefghijklmnopqrst"); | 116 UDPSocket dest("abcdefghijklmnopqrst"); |
117 UDPSocket src("abcdefghijklmnopqrst"); | 117 UDPSocket src("abcdefghijklmnopqrst"); |
118 | 118 |
119 // Receiver | 119 // Receiver |
120 EXPECT_EQ(0, dest.Bind("0.0.0.0", kPort)); | 120 EXPECT_EQ(0, dest.Bind("0.0.0.0", kPort)); |
121 EXPECT_EQ(0, dest.JoinGroup(kGroup)); | 121 EXPECT_EQ(0, dest.JoinGroup(kGroup)); |
122 dest.Read(1024, base::Bind(&OnMulticastReadCompleted, &packet_received)); | 122 dest.Read(1024, base::Bind(&OnMulticastReadCompleted, &packet_received)); |
123 | 123 |
124 // Sender | 124 // Sender |
125 EXPECT_EQ(0, src.SetMulticastTimeToLive(0)); | 125 EXPECT_EQ(0, src.SetMulticastTimeToLive(0)); |
126 src.Connect(kGroup, kPort, base::Bind(&SendMulticastPacket, &src)); | 126 src.Connect(kGroup, kPort, base::Bind(&SendMulticastPacket, &src)); |
127 | 127 |
128 // If not received within the test action timeout, quit the message loop. | 128 // If not received within the test action timeout, quit the message loop. |
129 io_loop.PostDelayedTask(FROM_HERE, | 129 io_loop.PostDelayedTask(FROM_HERE, |
130 base::Bind(&QuitMessageLoop), | 130 base::Bind(&QuitMessageLoop), |
131 TestTimeouts::action_timeout()); | 131 TestTimeouts::action_timeout()); |
132 | 132 |
133 io_loop.Run(); | 133 io_loop.Run(); |
134 | 134 |
135 EXPECT_TRUE(packet_received) << "Failed to receive from multicast address"; | 135 EXPECT_TRUE(packet_received) << "Failed to receive from multicast address"; |
136 } | 136 } |
137 | 137 |
138 } // namespace extensions | 138 } // namespace extensions |
OLD | NEW |