Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(329)

Unified Diff: remoting/protocol/stream_message_pipe_adapter.cc

Issue 1649063003: Add MessagePipe interface. Use it in ChannelDispatcherBase. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@simple_parser
Patch Set: Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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..826630476244093d5a18821710a3ddbc23984055
--- /dev/null
+++ b/remoting/protocol/stream_message_pipe_adapter.cc
@@ -0,0 +1,58 @@
+// 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 "base/callback_helpers.h"
+#include "remoting/base/buffered_socket_writer.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(
+ scoped_ptr<P2PStreamSocket> socket,
+ const ErrorCallback& error_callback)
+ : socket_(std::move(socket)),
+ error_callback_(error_callback),
+ writer_(new BufferedSocketWriter()) {
+ DCHECK(socket_);
+ DCHECK(!error_callback_.is_null());
+
+ writer_->Start(
+ base::Bind(&P2PStreamSocket::Write, base::Unretained(socket_.get())),
+ base::Bind(&StreamMessagePipeAdapter::CloseOnError,
+ base::Unretained(this)));
+}
+
+StreamMessagePipeAdapter::~StreamMessagePipeAdapter() {}
+
+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 (writer_)
+ writer_->Write(SerializeAndFrameMessage(*message), done);
+}
+
+void StreamMessagePipeAdapter::CloseOnError(int error) {
+ // Stop writing on error.
+ writer_.reset();
+
+ if (!error_callback_.is_null())
+ base::ResetAndReturn(&error_callback_).Run(error);
+}
+
+} // namespace protocol
+} // namespace remoting

Powered by Google App Engine
This is Rietveld 408576698