// 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_REALMFUZZER_TESTING_MODULE_H_
#define SRC_SYS_FUZZING_REALMFUZZER_TESTING_MODULE_H_

#include <stddef.h>
#include <stdint.h>
#include <zircon/compiler.h>

#include <memory>

#include "src/lib/fxl/macros.h"
#include "src/sys/fuzzing/common/testing/module.h"
#include "src/sys/fuzzing/realmfuzzer/target/module.h"

namespace fuzzing {

// Wraps a |Module| and automatically provides fake counters and PC tables based on a seed value.
class FakeRealmFuzzerModule final : public FakeModule {
 public:
  // Make a fake module with randomized PCs.
  explicit FakeRealmFuzzerModule(uint32_t seed = 1);

  // Make a fake module with the given PCs.
  explicit FakeRealmFuzzerModule(std::vector<ModulePC>&& pc_table) noexcept;

  FakeRealmFuzzerModule(FakeRealmFuzzerModule&& other) { *this = std::move(other); }
  ~FakeRealmFuzzerModule() override = default;

  FakeRealmFuzzerModule& operator=(FakeRealmFuzzerModule&& other) noexcept;

  const Module& module() const { return *module_; }

  // See corresponding |Module| methods.
  Identifier legacy_id() const { return module_->legacy_id(); }
  const std::string& id() const { return module_->id(); }
  __WARN_UNUSED_RESULT zx_status_t Share(uint64_t target_id, zx::vmo* out) const;
  void Update() { module_->Update(); }
  void Clear() { module_->Clear(); }

 private:
  std::unique_ptr<Module> module_;

  FXL_DISALLOW_COPY_AND_ASSIGN(FakeRealmFuzzerModule);
};

}  // namespace fuzzing

#endif  // SRC_SYS_FUZZING_REALMFUZZER_TESTING_MODULE_H_