| 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/message_loop.h" | 8 #include "base/message_loop/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 "remoting/protocol/p2p_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(P2PStreamSocket* client_socket, |
| 18 net::StreamSocket* host_socket, | 18 P2PStreamSocket* host_socket, |
| 19 int message_size, | 19 int message_size, |
| 20 int message_count) | 20 int message_count) |
| 21 : message_loop_(base::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) { |
| (...skipping 91 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 120 read_errors_++; | 120 read_errors_++; |
| 121 Done(); | 121 Done(); |
| 122 } else if (result > 0) { | 122 } else if (result > 0) { |
| 123 // Allocate memory for the next read. | 123 // Allocate memory for the next read. |
| 124 input_buffer_->set_offset(input_buffer_->offset() + result); | 124 input_buffer_->set_offset(input_buffer_->offset() + result); |
| 125 if (input_buffer_->offset() == test_data_size_) | 125 if (input_buffer_->offset() == test_data_size_) |
| 126 Done(); | 126 Done(); |
| 127 } | 127 } |
| 128 } | 128 } |
| 129 | 129 |
| 130 DatagramConnectionTester::DatagramConnectionTester(net::Socket* client_socket, | 130 DatagramConnectionTester::DatagramConnectionTester( |
| 131 net::Socket* host_socket, | 131 P2PDatagramSocket* client_socket, |
| 132 int message_size, | 132 P2PDatagramSocket* host_socket, |
| 133 int message_count, | 133 int message_size, |
| 134 int delay_ms) | 134 int message_count, |
| 135 int delay_ms) |
| 135 : message_loop_(base::MessageLoop::current()), | 136 : message_loop_(base::MessageLoop::current()), |
| 136 host_socket_(host_socket), | 137 host_socket_(host_socket), |
| 137 client_socket_(client_socket), | 138 client_socket_(client_socket), |
| 138 message_size_(message_size), | 139 message_size_(message_size), |
| 139 message_count_(message_count), | 140 message_count_(message_count), |
| 140 delay_ms_(delay_ms), | 141 delay_ms_(delay_ms), |
| 141 done_(false), | 142 done_(false), |
| 142 write_errors_(0), | 143 write_errors_(0), |
| 143 read_errors_(0), | 144 read_errors_(0), |
| 144 packets_sent_(0), | 145 packets_sent_(0), |
| (...skipping 107 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 252 if (memcmp(read_buffer_->data(), sent_packets_[packet_id]->data(), | 253 if (memcmp(read_buffer_->data(), sent_packets_[packet_id]->data(), |
| 253 message_size_) != 0) | 254 message_size_) != 0) |
| 254 bad_packets_received_++; | 255 bad_packets_received_++; |
| 255 } | 256 } |
| 256 } | 257 } |
| 257 } | 258 } |
| 258 } | 259 } |
| 259 | 260 |
| 260 } // namespace protocol | 261 } // namespace protocol |
| 261 } // namespace remoting | 262 } // namespace remoting |
| OLD | NEW |