mojo/edk/system/ports_message.h

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

#ifndef MOJO_EDK_SYSTEM_PORTS_MESSAGE_H__
#define MOJO_EDK_SYSTEM_PORTS_MESSAGE_H__

#include <memory>
#include <utility>

#include "mojo/edk/embedder/platform_handle_vector.h"
#include "mojo/edk/system/channel.h"
#include "mojo/edk/system/ports/message.h"

namespace mojo {
namespace edk {

class NodeController;

class PortsMessage : public ports::Message {
 public:
  static std::unique_ptr<PortsMessage> NewUserMessage(size_t num_payload_bytes,
                                                      size_t num_ports,
                                                      size_t num_handles);

  ~PortsMessage() override;

  size_t num_handles() const { return channel_message_->num_handles(); }
  bool has_handles() const { return channel_message_->has_handles(); }

  void SetHandles(ScopedPlatformHandleVectorPtr handles) {
    channel_message_->SetHandles(std::move(handles));
  }

  ScopedPlatformHandleVectorPtr TakeHandles() {
    return channel_message_->TakeHandles();
  }

  Channel::MessagePtr TakeChannelMessage() {
    return std::move(channel_message_);
  }

 private:
  friend class NodeController;

  // Construct a new user PortsMessage backed by a new Channel::Message.
  PortsMessage(size_t num_payload_bytes, size_t num_ports, size_t num_handles);

  // Construct a new PortsMessage backed by a Channel::Message. If
  // |channel_message| is null, a new one is allocated internally.
  PortsMessage(size_t num_header_bytes,
               size_t num_payload_bytes,
               size_t num_ports_bytes,
               Channel::MessagePtr channel_message);

  Channel::MessagePtr channel_message_;
};

}  // namespace edk
}  // namespace mojo

#endif  // MOJO_EDK_SYSTEM_PORTS_MESSAGE_H__