src/virtualization/tests/lib/fake_netstack.h

// Copyright 2019 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_VIRTUALIZATION_TESTS_LIB_FAKE_NETSTACK_H_
#define SRC_VIRTUALIZATION_TESTS_LIB_FAKE_NETSTACK_H_

#include <fuchsia/net/cpp/fidl.h>
#include <lib/async-loop/cpp/loop.h>
#include <lib/async/cpp/executor.h>
#include <lib/fpromise/promise.h>
#include <lib/sys/component/cpp/testing/realm_builder.h>

#include <memory>

namespace fake_netstack::internal {
class FakeNetwork;
}

// Implements a fake netstack, providing the fuchsia.net.virtualization.Control
// API.
//
// and allowing packets to be sent and received from devices that attach to the
// fake netstack.
//
// Thread-safe.
class FakeNetstack {
 public:
  FakeNetstack();
  ~FakeNetstack();

  // Send a packet with UDP headers, including the ethernet and IPv6 headers, to the interface with
  // the specified MAC address.
  fpromise::promise<void, zx_status_t> SendUdpPacket(const fuchsia::net::MacAddress& mac_addr,
                                                     std::vector<uint8_t> packet);

  // Send a raw packet to the interface with the specified MAC address.
  fpromise::promise<void, zx_status_t> SendPacket(const fuchsia::net::MacAddress& mac_addr,
                                                  std::vector<uint8_t> packet);

  // Receive a raw packet from the interface with the specified MAC address.
  fpromise::promise<std::vector<uint8_t>, zx_status_t> ReceivePacket(
      const fuchsia::net::MacAddress& mac_addr);

  std::unique_ptr<component_testing::LocalComponentImpl> NewComponent();

 private:
  async::Loop loop_;
  async::Executor executor_;

  // Fakes for fuchsia.net.virtualization.Control.
  std::unique_ptr<fake_netstack::internal::FakeNetwork> network_;
  std::unique_ptr<component_testing::LocalComponentHandles> handles_;
};

#endif  // SRC_VIRTUALIZATION_TESTS_LIB_FAKE_NETSTACK_H_