OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/fake_datagram_socket.h" | 5 #include "remoting/protocol/fake_datagram_socket.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/callback_helpers.h" | 8 #include "base/callback_helpers.h" |
9 #include "base/single_thread_task_runner.h" | 9 #include "base/single_thread_task_runner.h" |
10 #include "base/thread_task_runner_handle.h" | 10 #include "base/thread_task_runner_handle.h" |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
44 void FakeDatagramSocket::PairWith(FakeDatagramSocket* peer_socket) { | 44 void FakeDatagramSocket::PairWith(FakeDatagramSocket* peer_socket) { |
45 EXPECT_TRUE(task_runner_->BelongsToCurrentThread()); | 45 EXPECT_TRUE(task_runner_->BelongsToCurrentThread()); |
46 peer_socket_ = peer_socket->GetWeakPtr(); | 46 peer_socket_ = peer_socket->GetWeakPtr(); |
47 peer_socket->peer_socket_ = GetWeakPtr(); | 47 peer_socket->peer_socket_ = GetWeakPtr(); |
48 } | 48 } |
49 | 49 |
50 base::WeakPtr<FakeDatagramSocket> FakeDatagramSocket::GetWeakPtr() { | 50 base::WeakPtr<FakeDatagramSocket> FakeDatagramSocket::GetWeakPtr() { |
51 return weak_factory_.GetWeakPtr(); | 51 return weak_factory_.GetWeakPtr(); |
52 } | 52 } |
53 | 53 |
54 int FakeDatagramSocket::Read(net::IOBuffer* buf, int buf_len, | 54 int FakeDatagramSocket::Recv(const scoped_refptr<net::IOBuffer>& buf, |
| 55 int buf_len, |
55 const net::CompletionCallback& callback) { | 56 const net::CompletionCallback& callback) { |
56 EXPECT_TRUE(task_runner_->BelongsToCurrentThread()); | 57 EXPECT_TRUE(task_runner_->BelongsToCurrentThread()); |
57 if (input_pos_ < static_cast<int>(input_packets_.size())) { | 58 if (input_pos_ < static_cast<int>(input_packets_.size())) { |
58 return CopyReadData(buf, buf_len); | 59 return CopyReadData(buf, buf_len); |
59 } else { | 60 } else { |
60 read_buffer_ = buf; | 61 read_buffer_ = buf; |
61 read_buffer_size_ = buf_len; | 62 read_buffer_size_ = buf_len; |
62 read_callback_ = callback; | 63 read_callback_ = callback; |
63 return net::ERR_IO_PENDING; | 64 return net::ERR_IO_PENDING; |
64 } | 65 } |
65 } | 66 } |
66 | 67 |
67 int FakeDatagramSocket::Write(net::IOBuffer* buf, int buf_len, | 68 int FakeDatagramSocket::Send(const scoped_refptr<net::IOBuffer>& buf, |
68 const net::CompletionCallback& callback) { | 69 int buf_len, |
| 70 const net::CompletionCallback& callback) { |
69 EXPECT_TRUE(task_runner_->BelongsToCurrentThread()); | 71 EXPECT_TRUE(task_runner_->BelongsToCurrentThread()); |
70 written_packets_.push_back(std::string()); | 72 written_packets_.push_back(std::string()); |
71 written_packets_.back().assign(buf->data(), buf->data() + buf_len); | 73 written_packets_.back().assign(buf->data(), buf->data() + buf_len); |
72 | 74 |
73 if (peer_socket_.get()) { | 75 if (peer_socket_.get()) { |
74 task_runner_->PostTask( | 76 task_runner_->PostTask( |
75 FROM_HERE, | 77 FROM_HERE, |
76 base::Bind(&FakeDatagramSocket::AppendInputPacket, | 78 base::Bind(&FakeDatagramSocket::AppendInputPacket, peer_socket_, |
77 peer_socket_, | |
78 std::string(buf->data(), buf->data() + buf_len))); | 79 std::string(buf->data(), buf->data() + buf_len))); |
79 } | 80 } |
80 | 81 |
81 return buf_len; | 82 return buf_len; |
82 } | 83 } |
83 | 84 |
84 int FakeDatagramSocket::SetReceiveBufferSize(int32 size) { | 85 int FakeDatagramSocket::CopyReadData(const scoped_refptr<net::IOBuffer>& buf, |
85 NOTIMPLEMENTED(); | 86 int buf_len) { |
86 return net::ERR_NOT_IMPLEMENTED; | |
87 } | |
88 | |
89 int FakeDatagramSocket::SetSendBufferSize(int32 size) { | |
90 NOTIMPLEMENTED(); | |
91 return net::ERR_NOT_IMPLEMENTED; | |
92 } | |
93 | |
94 int FakeDatagramSocket::CopyReadData(net::IOBuffer* buf, int buf_len) { | |
95 int size = std::min( | 87 int size = std::min( |
96 buf_len, static_cast<int>(input_packets_[input_pos_].size())); | 88 buf_len, static_cast<int>(input_packets_[input_pos_].size())); |
97 memcpy(buf->data(), &(*input_packets_[input_pos_].begin()), size); | 89 memcpy(buf->data(), &(*input_packets_[input_pos_].begin()), size); |
98 ++input_pos_; | 90 ++input_pos_; |
99 return size; | 91 return size; |
100 } | 92 } |
101 | 93 |
102 FakeDatagramChannelFactory::FakeDatagramChannelFactory() | 94 FakeDatagramChannelFactory::FakeDatagramChannelFactory() |
103 : task_runner_(base::ThreadTaskRunnerHandle::Get()), | 95 : task_runner_(base::ThreadTaskRunnerHandle::Get()), |
104 asynchronous_create_(false), | 96 asynchronous_create_(false), |
(...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
160 callback.Run(owned_socket.Pass()); | 152 callback.Run(owned_socket.Pass()); |
161 } | 153 } |
162 | 154 |
163 void FakeDatagramChannelFactory::CancelChannelCreation( | 155 void FakeDatagramChannelFactory::CancelChannelCreation( |
164 const std::string& name) { | 156 const std::string& name) { |
165 channels_.erase(name); | 157 channels_.erase(name); |
166 } | 158 } |
167 | 159 |
168 } // namespace protocol | 160 } // namespace protocol |
169 } // namespace remoting | 161 } // namespace remoting |
OLD | NEW |