Index: remoting/protocol/stream_message_pipe_adapter.cc |
diff --git a/remoting/protocol/stream_message_pipe_adapter.cc b/remoting/protocol/stream_message_pipe_adapter.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..773a4043e1b9daae2baac48c2ebf862bc0d87f5e |
--- /dev/null |
+++ b/remoting/protocol/stream_message_pipe_adapter.cc |
@@ -0,0 +1,57 @@ |
+// Copyright 2016 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "remoting/protocol/stream_message_pipe_adapter.h" |
+ |
+#include <utility> |
+ |
+#include "base/bind.h" |
+#include "remoting/base/compound_buffer.h" |
+#include "remoting/protocol/message_serialization.h" |
+#include "remoting/protocol/p2p_stream_socket.h" |
+ |
+namespace remoting { |
+namespace protocol { |
+ |
+StreamMessagePipeAdapter::StreamMessagePipeAdapter() {} |
+StreamMessagePipeAdapter::~StreamMessagePipeAdapter() {} |
+ |
+void StreamMessagePipeAdapter::Initialize(scoped_ptr<P2PStreamSocket> socket, |
+ const ErrorCallback& error_callback) { |
+ DCHECK(socket); |
+ DCHECK(!error_callback.is_null()); |
+ |
+ socket_ = std::move(socket); |
+ error_callback_ = error_callback; |
+ |
+ writer_.Start( |
+ base::Bind(&P2PStreamSocket::Write, base::Unretained(socket_.get())), |
+ base::Bind(&StreamMessagePipeAdapter::CloseOnError, |
+ base::Unretained(this))); |
+} |
+ |
+void StreamMessagePipeAdapter::StartReceiving( |
+ const MessageReceivedCallback& callback) { |
+ reader_.StartReading(socket_.get(), callback, |
+ base::Bind(&StreamMessagePipeAdapter::CloseOnError, |
+ base::Unretained(this))); |
+} |
+ |
+void StreamMessagePipeAdapter::Send(google::protobuf::MessageLite* message, |
+ const base::Closure& done) { |
+ if (closed_) |
+ return; |
Jamie
2016/02/01 22:48:20
Call the error callback here? Otherwise the caller
Sergey Ulanov
2016/02/02 21:16:42
I think just dropping the callback here is the rig
Jamie
2016/02/02 22:43:10
Based on off-line discussion, it sounds like there
Sergey Ulanov
2016/02/03 18:45:26
Done.
|
+ |
+ writer_.Write(SerializeAndFrameMessage(*message), done); |
+} |
+ |
+void StreamMessagePipeAdapter::CloseOnError(int error) { |
+ // Stop reading after any error. |
+ closed_ = true; |
+ |
+ error_callback_.Run(error); |
+} |
+ |
+} // namespace protocol |
+} // namespace remoting |