Chromium Code Reviews| 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) {} |
|
alexeypa (please no reviews)
2013/04/29 17:17:41
nit: the closing bracket should got to a separate
xhwang
2013/04/30 00:02:36
Done.
| |
| 29 } | |
| 30 | 29 |
| 31 StreamConnectionTester::~StreamConnectionTester() { | 30 StreamConnectionTester::~StreamConnectionTester() { |
| 32 } | 31 } |
| 33 | 32 |
| 34 void StreamConnectionTester::Start() { | 33 void StreamConnectionTester::Start() { |
| 35 InitBuffers(); | 34 InitBuffers(); |
| 36 DoRead(); | 35 DoRead(); |
| 37 DoWrite(); | 36 DoWrite(); |
| 38 } | 37 } |
| 39 | 38 |
| 40 void StreamConnectionTester::CheckResults() { | 39 void StreamConnectionTester::CheckResults() { |
| 41 EXPECT_EQ(0, write_errors_); | 40 EXPECT_EQ(0, write_errors_); |
| 42 EXPECT_EQ(0, read_errors_); | 41 EXPECT_EQ(0, read_errors_); |
| 43 | 42 |
| 44 ASSERT_EQ(test_data_size_, input_buffer_->offset()); | 43 ASSERT_EQ(test_data_size_, input_buffer_->offset()); |
| 45 | 44 |
| 46 output_buffer_->SetOffset(0); | 45 output_buffer_->SetOffset(0); |
| 47 ASSERT_EQ(test_data_size_, output_buffer_->size()); | 46 ASSERT_EQ(test_data_size_, output_buffer_->size()); |
| 48 | 47 |
| 49 EXPECT_EQ(0, memcmp(output_buffer_->data(), | 48 EXPECT_EQ(0, memcmp(output_buffer_->data(), |
| 50 input_buffer_->StartOfBuffer(), test_data_size_)); | 49 input_buffer_->StartOfBuffer(), test_data_size_)); |
| 51 } | 50 } |
| 52 | 51 |
| 53 void StreamConnectionTester::Done() { | 52 void StreamConnectionTester::Done() { |
| 54 done_ = true; | 53 done_ = true; |
| 55 message_loop_->PostTask(FROM_HERE, MessageLoop::QuitClosure()); | 54 message_loop_->PostTask(FROM_HERE, base::MessageLoop::QuitClosure()); |
| 56 } | 55 } |
| 57 | 56 |
| 58 void StreamConnectionTester::InitBuffers() { | 57 void StreamConnectionTester::InitBuffers() { |
| 59 output_buffer_ = new net::DrainableIOBuffer( | 58 output_buffer_ = new net::DrainableIOBuffer( |
| 60 new net::IOBuffer(test_data_size_), test_data_size_); | 59 new net::IOBuffer(test_data_size_), test_data_size_); |
| 61 for (int i = 0; i < test_data_size_; ++i) { | 60 for (int i = 0; i < test_data_size_; ++i) { |
| 62 output_buffer_->data()[i] = static_cast<char>(i); | 61 output_buffer_->data()[i] = static_cast<char>(i); |
| 63 } | 62 } |
| 64 | 63 |
| 65 input_buffer_ = new net::GrowableIOBuffer(); | 64 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_) | 122 if (input_buffer_->offset() == test_data_size_) |
| 124 Done(); | 123 Done(); |
| 125 } | 124 } |
| 126 } | 125 } |
| 127 | 126 |
| 128 DatagramConnectionTester::DatagramConnectionTester(net::Socket* client_socket, | 127 DatagramConnectionTester::DatagramConnectionTester(net::Socket* client_socket, |
| 129 net::Socket* host_socket, | 128 net::Socket* host_socket, |
| 130 int message_size, | 129 int message_size, |
| 131 int message_count, | 130 int message_count, |
| 132 int delay_ms) | 131 int delay_ms) |
| 133 : message_loop_(MessageLoop::current()), | 132 : message_loop_(base::MessageLoop::current()), |
| 134 host_socket_(host_socket), | 133 host_socket_(host_socket), |
| 135 client_socket_(client_socket), | 134 client_socket_(client_socket), |
| 136 message_size_(message_size), | 135 message_size_(message_size), |
| 137 message_count_(message_count), | 136 message_count_(message_count), |
| 138 delay_ms_(delay_ms), | 137 delay_ms_(delay_ms), |
| 139 done_(false), | 138 done_(false), |
| 140 write_errors_(0), | 139 write_errors_(0), |
| 141 read_errors_(0), | 140 read_errors_(0), |
| 142 packets_sent_(0), | 141 packets_sent_(0), |
| 143 packets_received_(0), | 142 packets_received_(0), |
| (...skipping 16 matching lines...) Expand all Loading... | |
| 160 EXPECT_EQ(0, bad_packets_received_); | 159 EXPECT_EQ(0, bad_packets_received_); |
| 161 | 160 |
| 162 // Verify that we've received at least one packet. | 161 // Verify that we've received at least one packet. |
| 163 EXPECT_GT(packets_received_, 0); | 162 EXPECT_GT(packets_received_, 0); |
| 164 LOG(INFO) << "Received " << packets_received_ << " packets out of " | 163 LOG(INFO) << "Received " << packets_received_ << " packets out of " |
| 165 << message_count_; | 164 << message_count_; |
| 166 } | 165 } |
| 167 | 166 |
| 168 void DatagramConnectionTester::Done() { | 167 void DatagramConnectionTester::Done() { |
| 169 done_ = true; | 168 done_ = true; |
| 170 message_loop_->PostTask(FROM_HERE, MessageLoop::QuitClosure()); | 169 message_loop_->PostTask(FROM_HERE, base::MessageLoop::QuitClosure()); |
| 171 } | 170 } |
| 172 | 171 |
| 173 void DatagramConnectionTester::DoWrite() { | 172 void DatagramConnectionTester::DoWrite() { |
| 174 if (packets_sent_ >= message_count_) { | 173 if (packets_sent_ >= message_count_) { |
| 175 Done(); | 174 Done(); |
| 176 return; | 175 return; |
| 177 } | 176 } |
| 178 | 177 |
| 179 scoped_refptr<net::IOBuffer> packet(new net::IOBuffer(message_size_)); | 178 scoped_refptr<net::IOBuffer> packet(new net::IOBuffer(message_size_)); |
| 180 for (int i = 0; i < message_size_; ++i) { | 179 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(), | 247 if (memcmp(read_buffer_->data(), sent_packets_[packet_id]->data(), |
| 249 message_size_) != 0) | 248 message_size_) != 0) |
| 250 bad_packets_received_++; | 249 bad_packets_received_++; |
| 251 } | 250 } |
| 252 } | 251 } |
| 253 } | 252 } |
| 254 } | 253 } |
| 255 | 254 |
| 256 } // namespace protocol | 255 } // namespace protocol |
| 257 } // namespace remoting | 256 } // namespace remoting |
| OLD | NEW |