zircon/system/utest/core/interrupt/fixture.h

// Copyright 2020 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 ZIRCON_SYSTEM_UTEST_CORE_INTERRUPT_FIXTURE_H_
#define ZIRCON_SYSTEM_UTEST_CORE_INTERRUPT_FIXTURE_H_

#include <lib/standalone-test/standalone.h>
#include <lib/zx/bti.h>
#include <lib/zx/iommu.h>
#include <lib/zx/msi.h>
#include <lib/zx/resource.h>
#include <lib/zx/thread.h>
#include <zircon/syscalls/iommu.h>

#include <zxtest/zxtest.h>

namespace {

class ResourceFixture : public zxtest::Test {
 public:
  void SetUp() override {
    root_resource_ = standalone::GetRootResource();
    irq_resource_ = standalone::GetIrqResource();

    zx_iommu_desc_dummy_t desc = {};
    ASSERT_OK(
        zx::iommu::create(*root_resource_, ZX_IOMMU_TYPE_DUMMY, &desc, sizeof(desc), &iommu_));
    ASSERT_OK(zx::bti::create(iommu_, 0, 0xdeadbeef, &bti_));
  }

  bool MsiTestsSupported() {
    zx::msi msi;
    return !(zx::msi::allocate(*root_resource_, 1, &msi) == ZX_ERR_NOT_SUPPORTED);
  }

 protected:
  zx::unowned_bti bti() { return bti_.borrow(); }
  zx::unowned_resource& root_resource() { return root_resource_; }
  zx::unowned_resource& irq_resource() { return irq_resource_; }
  zx::unowned_iommu iommu() { return iommu_.borrow(); }

 private:
  zx::unowned_resource root_resource_;
  zx::unowned_resource irq_resource_;
  zx::iommu iommu_;
  zx::bti bti_;
};

}  // namespace

#endif  // ZIRCON_SYSTEM_UTEST_CORE_INTERRUPT_FIXTURE_H_