remoting/protocol/fake_message_pipe_wrapper.h

// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef REMOTING_PROTOCOL_FAKE_MESSAGE_PIPE_WRAPPER_H_
#define REMOTING_PROTOCOL_FAKE_MESSAGE_PIPE_WRAPPER_H_

#include <memory>

#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "remoting/protocol/message_pipe.h"

namespace google {
namespace protobuf {
class MessageLite;
}  // namespace protobuf
}  // namespace google

namespace remoting::protocol {

class FakeMessagePipe;

// This class should not be used explicitly: use FakeMessagePipe::Wrap().
class FakeMessagePipeWrapper final : public MessagePipe {
 public:
  // |pipe| must outlive this instance.
  explicit FakeMessagePipeWrapper(FakeMessagePipe* pipe);
  ~FakeMessagePipeWrapper() override;

  // MessagePipe implementation.
  void Start(EventHandler* event_handler) override;
  void Send(google::protobuf::MessageLite* message,
            base::OnceClosure done) override;

  void Receive(std::unique_ptr<CompoundBuffer> message);
  void OpenPipe();
  void ClosePipe();

  base::WeakPtr<FakeMessagePipeWrapper> GetWeakPtr();

 private:
  const raw_ptr<FakeMessagePipe> pipe_;
  base::WeakPtrFactory<FakeMessagePipeWrapper> weak_factory_{this};
};

}  // namespace remoting::protocol

#endif  // REMOTING_PROTOCOL_FAKE_MESSAGE_PIPE_WRAPPER_H_