chrome/test/chromedriver/net/pipe_handler.h

// Copyright (c) 2020 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 CHROME_TEST_CHROMEDRIVER_NET_PIPE_HANDLER_H_
#define CHROME_TEST_CHROMEDRIVER_NET_PIPE_HANDLER_H_

#include <string>

#include "base/compiler_specific.h"
#include "base/memory/ref_counted.h"
#include "chrome/test/chromedriver/net/websocket.h"

namespace net {
class DrainableIOBuffer;
}

class WebSocketListener;

class PipeHandler {
 public:
  PipeHandler(WebSocketListener* listener, int write_fd, int read_fd);
  virtual ~PipeHandler();

  // Sends the given message and returns true on success.
  bool Send(const std::string& message);

 private:
  void WriteIntoPipe();

  void Read();
  void Close();

  WebSocketListener* listener_;
  int write_fd_;
  int read_fd_;

  scoped_refptr<net::DrainableIOBuffer> write_buffer_;
  scoped_refptr<net::DrainableIOBuffer> read_buffer_;
  std::string pending_write_;

  DISALLOW_COPY_AND_ASSIGN(PipeHandler);
};

#endif  // CHROME_TEST_CHROMEDRIVER_NET_PIPE_HANDLER_H_