| 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 | 
|  |