arc/network/adb_proxy.h

// Copyright 2019 The Chromium OS 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 ARC_NETWORK_ADB_PROXY_H_
#define ARC_NETWORK_ADB_PROXY_H_

#include <deque>
#include <memory>

#include <base/memory/weak_ptr.h>
#include <brillo/daemons/daemon.h>
#include <base/message_loop/message_loop.h>

#include "arc/network/socket.h"
#include "arc/network/socket_forwarder.h"

namespace arc_networkd {

// Subprocess for proxying ADB traffic.
class AdbProxy : public brillo::Daemon, public base::MessageLoopForIO::Watcher {
 public:
  AdbProxy();
  virtual ~AdbProxy();

 protected:
  // Starts the server on the |src_port| over TCP.
  int OnInit() override;

  // Watcher callback for accepting new connections.
  void OnFileCanReadWithoutBlocking(int fd) override;
  void OnFileCanWriteWithoutBlocking(int fd) override {}

 private:
  // Detects when the guest OS is turned up and torn down and cleans up any
  // existing forwarders and connections.
  bool OnSignal(const struct signalfd_siginfo& info);

  // Attempts to establish a connection to ADB at well-known destinations.
  std::unique_ptr<Socket> Connect() const;

  std::unique_ptr<Socket> src_;
  std::deque<std::unique_ptr<SocketForwarder>> fwd_;
  base::MessageLoopForIO::FileDescriptorWatcher src_watcher_;

  base::WeakPtrFactory<AdbProxy> weak_factory_{this};
  DISALLOW_COPY_AND_ASSIGN(AdbProxy);
};

}  // namespace arc_networkd

#endif  // ARC_NETWORK_ADB_PROXY_H_