src/developer/debug/shared/buffered_zx_socket.h

// Copyright 2018 The Fuchsia 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 SRC_DEVELOPER_DEBUG_SHARED_BUFFERED_ZX_SOCKET_H_
#define SRC_DEVELOPER_DEBUG_SHARED_BUFFERED_ZX_SOCKET_H_

#include <lib/zx/socket.h>

#include <functional>

#include "src/developer/debug/shared/buffered_stream.h"
#include "src/developer/debug/shared/message_loop.h"
#include "src/developer/debug/shared/socket_watcher.h"

namespace debug {

// An adapter that converts a zx::socket to a StreamBuffer.
class BufferedZxSocket final : public BufferedStream, public SocketWatcher {
 public:
  // Constructs a !IsValid() buffered stream not doing anything.
  BufferedZxSocket();

  // Constructs for the given zx::socket. The socket must be valid and a MessageLoop must already
  // have been set up on the current thread.
  //
  // Start() must be called before stream events will be delivered.
  explicit BufferedZxSocket(zx::socket socket);

  ~BufferedZxSocket() final;

  // BufferedStream implementation.
  bool Start() final;
  bool Stop() final;
  bool IsValid() final { return socket_.is_valid(); }

 private:
  // BufferedStream protected implementation.
  void ResetInternal() final;

  // SocketWatcher implementation.
  void OnSocketReadable(zx_handle_t) override;
  void OnSocketWritable(zx_handle_t) override;
  void OnPeerClosed(zx_handle_t) override;

  // StreamBuffer::Writer implementation.
  size_t ConsumeStreamBufferData(const char* data, size_t len) override;

  zx::socket socket_;
  MessageLoop::WatchHandle watch_handle_;

  FXL_DISALLOW_COPY_AND_ASSIGN(BufferedZxSocket);
};

}  // namespace debug

#endif  // SRC_DEVELOPER_DEBUG_SHARED_BUFFERED_ZX_SOCKET_H_