src/sys/fuzzing/common/testing/registrar.h

// Copyright 2021 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_SYS_FUZZING_COMMON_TESTING_REGISTRAR_H_
#define SRC_SYS_FUZZING_COMMON_TESTING_REGISTRAR_H_

#include <fuchsia/fuzzer/cpp/fidl.h>
#include <lib/fidl/cpp/binding.h>
#include <lib/fidl/cpp/interface_handle.h>

#include "src/lib/fxl/macros.h"
#include "src/sys/fuzzing/common/async-deque.h"
#include "src/sys/fuzzing/common/async-types.h"

namespace fuzzing {

using ::fuchsia::fuzzer::Controller;
using ::fuchsia::fuzzer::ControllerProvider;
using ::fuchsia::fuzzer::Registrar;

// Alias this type to improve readability.
using ControllerProviderHandle = fidl::InterfaceHandle<ControllerProvider>;

// Dummy URL that can be used with tests that involve the |FakeRegistrar|.
extern const char* kFakeFuzzerUrl;

// |FakeRegistrar| runs a simple implementation of |fuchsia.fuzzer.Registrar| locally.
class FakeRegistrar final : public Registrar {
 public:
  explicit FakeRegistrar(ExecutorPtr executor);
  ~FakeRegistrar() override = default;

  fidl::InterfaceHandle<Registrar> NewBinding();

  // FIDL methods.
  // TODO(https://fxbug.dev/42056592): Change the type of |url| when a proper FIDL URL type is available.
  void Register(std::string url, ControllerProviderHandle provider,
                RegisterCallback callback) override;

  // Returns a promise to return the next |ControllerProvider| handle sent to this object via the
  // |fuchsia.fuzzer.Registrar/Register| FIDL method.
  Promise<ControllerProviderHandle> TakeProvider();

 private:
  fidl::Binding<Registrar> binding_;
  ExecutorPtr executor_;
  AsyncSender<ControllerProviderHandle> sender_;
  AsyncReceiver<ControllerProviderHandle> receiver_;

  FXL_DISALLOW_COPY_ASSIGN_AND_MOVE(FakeRegistrar);
};

}  // namespace fuzzing

#endif  // SRC_SYS_FUZZING_COMMON_TESTING_REGISTRAR_H_