| 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 |