Index: mojo/system/proxy_message_pipe_endpoint.cc |
diff --git a/mojo/system/proxy_message_pipe_endpoint.cc b/mojo/system/proxy_message_pipe_endpoint.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..fce2e437ccdf434835f8a424974f21510808dda0 |
--- /dev/null |
+++ b/mojo/system/proxy_message_pipe_endpoint.cc |
@@ -0,0 +1,144 @@ |
+// Copyright 2013 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 "mojo/system/proxy_message_pipe_endpoint.h" |
+ |
+#include <string.h> |
+ |
+#include "base/logging.h" |
+#include "base/stl_util.h" |
+#include "mojo/system/channel.h" |
+ |
+namespace mojo { |
+namespace system { |
+ |
+ProxyMessagePipeEndpoint::ProxyMessagePipeEndpoint() |
+ : local_id_(MessageInTransit::kInvalidEndpointId), |
+ remote_id_(MessageInTransit::kInvalidEndpointId), |
+ is_open_(true), |
+ is_peer_open_(true) { |
+} |
+ |
+ProxyMessagePipeEndpoint::~ProxyMessagePipeEndpoint() { |
+ DCHECK(!is_running()); |
+ DCHECK(!is_attached()); |
+ AssertConsistentState(); |
+ DCHECK(paused_message_queue_.empty()); |
+} |
+ |
+void ProxyMessagePipeEndpoint::Close() { |
+ DCHECK(is_open_); |
+ is_open_ = false; |
+ |
+ DCHECK(is_attached()); |
+ channel_->DetachMessagePipeEndpoint(local_id_); |
+ channel_ = NULL; |
+ local_id_ = MessageInTransit::kInvalidEndpointId; |
+ remote_id_ = MessageInTransit::kInvalidEndpointId; |
+ |
+ for (std::deque<MessageInTransit*>::iterator it = |
+ paused_message_queue_.begin(); |
+ it != paused_message_queue_.end(); |
+ ++it) { |
+ (*it)->Destroy(); |
+ } |
+ paused_message_queue_.clear(); |
+} |
+ |
+bool ProxyMessagePipeEndpoint::OnPeerClose() { |
+ DCHECK(is_open_); |
+ DCHECK(is_peer_open_); |
+ |
+ is_peer_open_ = false; |
+ if (EnqueueMessage(MessageInTransit::Create( |
+ MessageInTransit::kTypeMessagePipe, |
+ MessageInTransit::kSubtypeMessagePipePeerClosed, |
+ NULL, 0)) != MOJO_RESULT_OK) { |
+ // TODO(vtl): Do something more sensible on error here? |
+ LOG(WARNING) << "Failed to send peer closed control message"; |
+ } |
+ |
+ // Return false -- to indicate that we should be destroyed -- if no messages |
+ // are still enqueued. (Messages may still be enqueued if we're not running |
+ // yet, but our peer was closed.) |
+ return !paused_message_queue_.empty(); |
+} |
+ |
+MojoResult ProxyMessagePipeEndpoint::EnqueueMessage(MessageInTransit* message) { |
+ DCHECK(is_open_); |
+ // If our (local) peer isn't open, we should only be enqueueing our own |
+ // control messages. |
+ DCHECK(is_peer_open_ || |
+ (message->type() == MessageInTransit::kTypeMessagePipe)); |
+ |
+ MojoResult rv = MOJO_RESULT_OK; |
+ |
+ if (is_running()) { |
+ message->set_source_id(local_id_); |
+ message->set_destination_id(remote_id_); |
+ if (!channel_->WriteMessage(message)) |
+ rv = MOJO_RESULT_FAILED_PRECONDITION; |
+ } else { |
+ paused_message_queue_.push_back(message); |
+ } |
+ |
+ return rv; |
+} |
+ |
+void ProxyMessagePipeEndpoint::Attach(scoped_refptr<Channel> channel, |
+ MessageInTransit::EndpointId local_id) { |
+ DCHECK(channel.get()); |
+ DCHECK_NE(local_id, MessageInTransit::kInvalidEndpointId); |
+ |
+ DCHECK(!is_attached()); |
+ |
+ AssertConsistentState(); |
+ channel_ = channel; |
+ local_id_ = local_id; |
+ AssertConsistentState(); |
+} |
+ |
+bool ProxyMessagePipeEndpoint::Run(MessageInTransit::EndpointId remote_id) { |
+ // Assertions about arguments: |
+ DCHECK_NE(remote_id, MessageInTransit::kInvalidEndpointId); |
+ |
+ // Assertions about current state: |
+ DCHECK(is_attached()); |
+ DCHECK(!is_running()); |
+ |
+ AssertConsistentState(); |
+ remote_id_ = remote_id; |
+ AssertConsistentState(); |
+ |
+ MojoResult result = MOJO_RESULT_OK; |
+ for (std::deque<MessageInTransit*>::iterator it = |
+ paused_message_queue_.begin(); |
+ it != paused_message_queue_.end(); |
+ ++it) { |
+ result = EnqueueMessage(*it); |
+ if (result != MOJO_RESULT_OK) { |
+ // TODO(vtl): Do something more sensible on error here? |
+ LOG(WARNING) << "Failed to send message"; |
+ } |
+ } |
+ paused_message_queue_.clear(); |
+ |
+ // If the peer is not open, we should return false since we should be |
+ // destroyed. |
+ return is_peer_open_; |
+} |
+ |
+#ifndef NDEBUG |
+void ProxyMessagePipeEndpoint::AssertConsistentState() const { |
+ if (is_attached()) { |
+ DCHECK_NE(local_id_, MessageInTransit::kInvalidEndpointId); |
+ } else { // Not attached. |
+ DCHECK_EQ(local_id_, MessageInTransit::kInvalidEndpointId); |
+ DCHECK_EQ(remote_id_, MessageInTransit::kInvalidEndpointId); |
+ } |
+} |
+#endif |
+ |
+} // namespace system |
+} // namespace mojo |