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 "remoting/protocol/connection_tester.h" | 5 #include "remoting/protocol/connection_tester.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
9 #include "net/base/io_buffer.h" | 9 #include "net/base/io_buffer.h" |
10 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
11 #include "net/socket/stream_socket.h" | 11 #include "net/socket/stream_socket.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
13 | 13 |
14 namespace remoting { | 14 namespace remoting { |
15 namespace protocol { | 15 namespace protocol { |
16 | 16 |
17 StreamConnectionTester::StreamConnectionTester(net::StreamSocket* client_socket, | 17 StreamConnectionTester::StreamConnectionTester(net::StreamSocket* client_socket, |
18 net::StreamSocket* host_socket, | 18 net::StreamSocket* host_socket, |
19 int message_size, | 19 int message_size, |
20 int message_count) | 20 int message_count) |
21 : message_loop_(MessageLoop::current()), | 21 : message_loop_(base::MessageLoop::current()), |
22 host_socket_(host_socket), | 22 host_socket_(host_socket), |
23 client_socket_(client_socket), | 23 client_socket_(client_socket), |
24 message_size_(message_size), | 24 message_size_(message_size), |
25 test_data_size_(message_size * message_count), | 25 test_data_size_(message_size * message_count), |
26 done_(false), | 26 done_(false), |
27 write_errors_(0), | 27 write_errors_(0), |
28 read_errors_(0) { | 28 read_errors_(0) { |
29 } | 29 } |
30 | 30 |
31 StreamConnectionTester::~StreamConnectionTester() { | 31 StreamConnectionTester::~StreamConnectionTester() { |
(...skipping 13 matching lines...) Expand all Loading... |
45 | 45 |
46 output_buffer_->SetOffset(0); | 46 output_buffer_->SetOffset(0); |
47 ASSERT_EQ(test_data_size_, output_buffer_->size()); | 47 ASSERT_EQ(test_data_size_, output_buffer_->size()); |
48 | 48 |
49 EXPECT_EQ(0, memcmp(output_buffer_->data(), | 49 EXPECT_EQ(0, memcmp(output_buffer_->data(), |
50 input_buffer_->StartOfBuffer(), test_data_size_)); | 50 input_buffer_->StartOfBuffer(), test_data_size_)); |
51 } | 51 } |
52 | 52 |
53 void StreamConnectionTester::Done() { | 53 void StreamConnectionTester::Done() { |
54 done_ = true; | 54 done_ = true; |
55 message_loop_->PostTask(FROM_HERE, MessageLoop::QuitClosure()); | 55 message_loop_->PostTask(FROM_HERE, base::MessageLoop::QuitClosure()); |
56 } | 56 } |
57 | 57 |
58 void StreamConnectionTester::InitBuffers() { | 58 void StreamConnectionTester::InitBuffers() { |
59 output_buffer_ = new net::DrainableIOBuffer( | 59 output_buffer_ = new net::DrainableIOBuffer( |
60 new net::IOBuffer(test_data_size_), test_data_size_); | 60 new net::IOBuffer(test_data_size_), test_data_size_); |
61 for (int i = 0; i < test_data_size_; ++i) { | 61 for (int i = 0; i < test_data_size_; ++i) { |
62 output_buffer_->data()[i] = static_cast<char>(i); | 62 output_buffer_->data()[i] = static_cast<char>(i); |
63 } | 63 } |
64 | 64 |
65 input_buffer_ = new net::GrowableIOBuffer(); | 65 input_buffer_ = new net::GrowableIOBuffer(); |
(...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
123 if (input_buffer_->offset() == test_data_size_) | 123 if (input_buffer_->offset() == test_data_size_) |
124 Done(); | 124 Done(); |
125 } | 125 } |
126 } | 126 } |
127 | 127 |
128 DatagramConnectionTester::DatagramConnectionTester(net::Socket* client_socket, | 128 DatagramConnectionTester::DatagramConnectionTester(net::Socket* client_socket, |
129 net::Socket* host_socket, | 129 net::Socket* host_socket, |
130 int message_size, | 130 int message_size, |
131 int message_count, | 131 int message_count, |
132 int delay_ms) | 132 int delay_ms) |
133 : message_loop_(MessageLoop::current()), | 133 : message_loop_(base::MessageLoop::current()), |
134 host_socket_(host_socket), | 134 host_socket_(host_socket), |
135 client_socket_(client_socket), | 135 client_socket_(client_socket), |
136 message_size_(message_size), | 136 message_size_(message_size), |
137 message_count_(message_count), | 137 message_count_(message_count), |
138 delay_ms_(delay_ms), | 138 delay_ms_(delay_ms), |
139 done_(false), | 139 done_(false), |
140 write_errors_(0), | 140 write_errors_(0), |
141 read_errors_(0), | 141 read_errors_(0), |
142 packets_sent_(0), | 142 packets_sent_(0), |
143 packets_received_(0), | 143 packets_received_(0), |
(...skipping 16 matching lines...) Expand all Loading... |
160 EXPECT_EQ(0, bad_packets_received_); | 160 EXPECT_EQ(0, bad_packets_received_); |
161 | 161 |
162 // Verify that we've received at least one packet. | 162 // Verify that we've received at least one packet. |
163 EXPECT_GT(packets_received_, 0); | 163 EXPECT_GT(packets_received_, 0); |
164 LOG(INFO) << "Received " << packets_received_ << " packets out of " | 164 LOG(INFO) << "Received " << packets_received_ << " packets out of " |
165 << message_count_; | 165 << message_count_; |
166 } | 166 } |
167 | 167 |
168 void DatagramConnectionTester::Done() { | 168 void DatagramConnectionTester::Done() { |
169 done_ = true; | 169 done_ = true; |
170 message_loop_->PostTask(FROM_HERE, MessageLoop::QuitClosure()); | 170 message_loop_->PostTask(FROM_HERE, base::MessageLoop::QuitClosure()); |
171 } | 171 } |
172 | 172 |
173 void DatagramConnectionTester::DoWrite() { | 173 void DatagramConnectionTester::DoWrite() { |
174 if (packets_sent_ >= message_count_) { | 174 if (packets_sent_ >= message_count_) { |
175 Done(); | 175 Done(); |
176 return; | 176 return; |
177 } | 177 } |
178 | 178 |
179 scoped_refptr<net::IOBuffer> packet(new net::IOBuffer(message_size_)); | 179 scoped_refptr<net::IOBuffer> packet(new net::IOBuffer(message_size_)); |
180 for (int i = 0; i < message_size_; ++i) { | 180 for (int i = 0; i < message_size_; ++i) { |
(...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
248 if (memcmp(read_buffer_->data(), sent_packets_[packet_id]->data(), | 248 if (memcmp(read_buffer_->data(), sent_packets_[packet_id]->data(), |
249 message_size_) != 0) | 249 message_size_) != 0) |
250 bad_packets_received_++; | 250 bad_packets_received_++; |
251 } | 251 } |
252 } | 252 } |
253 } | 253 } |
254 } | 254 } |
255 | 255 |
256 } // namespace protocol | 256 } // namespace protocol |
257 } // namespace remoting | 257 } // namespace remoting |
OLD | NEW |