// Copyright 2019 The Fuchsia Authors
//
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file or at
// https://opensource.org/licenses/MIT

#include <lib/acpi_tables.h>
#include <lib/acpi_tables_test_data.h>
#include <lib/unittest/unittest.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <initializer_list>

bool test_cpus_eve() {
  BEGIN_TEST;
  AcpiTables tables(&acpi_test_data::kEveTableProvider);

  uint32_t cpus = 0;
  EXPECT_EQ(ZX_OK, tables.cpu_count(&cpus));
  EXPECT_EQ(4u, cpus);

  uint32_t ids[4] = {0};
  EXPECT_EQ(ZX_OK, tables.cpu_apic_ids(ids, 4, &cpus));
  ASSERT_EQ(4u, cpus);
  EXPECT_EQ(0u, ids[0]);
  EXPECT_EQ(1u, ids[1]);
  EXPECT_EQ(2u, ids[2]);
  EXPECT_EQ(3u, ids[3]);

  END_TEST;
}

bool test_cpus_z840() {
  BEGIN_TEST;
  AcpiTables tables(&acpi_test_data::kZ840TableProvider);

  // We check the numcpus to ensure basic parsing is working.
  uint32_t cpus = 0;
  EXPECT_EQ(ZX_OK, tables.cpu_count(&cpus));
  EXPECT_EQ(56u, cpus);

  END_TEST;
}

bool test_io() {
  BEGIN_TEST;
  AcpiTables tables(&acpi_test_data::kEveTableProvider);

  uint32_t num_io = 0;
  EXPECT_EQ(ZX_OK, tables.io_apic_count(&num_io));
  EXPECT_EQ(1u, num_io);

  io_apic_descriptor io;
  EXPECT_EQ(ZX_OK, tables.io_apics(&io, 1, &num_io));
  ASSERT_EQ(1u, num_io);
  EXPECT_EQ(2u, io.apic_id);
  EXPECT_EQ(0u, io.global_irq_base);
  EXPECT_EQ(0xFEC00000u, io.paddr);

  END_TEST;
}

bool test_interrupt_source_overrides() {
  BEGIN_TEST;
  AcpiTables tables(&acpi_test_data::kEveTableProvider);

  uint32_t num_overrides = 0;
  EXPECT_EQ(ZX_OK, tables.interrupt_source_overrides_count(&num_overrides));
  EXPECT_EQ(2u, num_overrides);

  io_apic_isa_override overrides[2];
  EXPECT_EQ(ZX_OK, tables.interrupt_source_overrides(overrides, 2, &num_overrides));
  ASSERT_EQ(2u, num_overrides);

  EXPECT_EQ(0u, overrides[0].isa_irq);
  EXPECT_EQ(true, overrides[0].remapped);
  EXPECT_EQ(0, overrides[0].tm);
  EXPECT_EQ(0, overrides[0].pol);
  EXPECT_EQ(2u, overrides[0].global_irq);

  EXPECT_EQ(9u, overrides[1].isa_irq);
  EXPECT_EQ(true, overrides[1].remapped);
  EXPECT_EQ(1, overrides[1].tm);
  EXPECT_EQ(0, overrides[1].pol);
  EXPECT_EQ(9u, overrides[1].global_irq);

  END_TEST;
}

bool test_hpet() {
  BEGIN_TEST;
  AcpiTables tables(&acpi_test_data::kEveTableProvider);

  acpi_hpet_descriptor hpet;
  ASSERT_EQ(ZX_OK, tables.hpet(&hpet));

  EXPECT_EQ(0xFED00000u, hpet.address);
  EXPECT_EQ(false, hpet.port_io);
  EXPECT_EQ(0, hpet.minimum_tick);
  EXPECT_EQ(0, hpet.sequence);

  END_TEST;
}

bool test_numa_z840() {
  BEGIN_TEST;
  AcpiTables tables(&acpi_test_data::kZ840TableProvider);
  size_t domain_counts[2] = {0, 0};
  AcpiNumaDomain domains[2];
  EXPECT_EQ(ZX_OK, tables.VisitCpuNumaPairs([&](const AcpiNumaDomain& region, uint32_t apic_id) {
    domain_counts[region.domain]++;
    domains[region.domain] = region;
  }));

  ASSERT_EQ(28u, domain_counts[0]);
  ASSERT_EQ(28u, domain_counts[1]);
  ASSERT_EQ(1u, domains[0].memory_count);
  ASSERT_EQ(1u, domains[1].memory_count);

  EXPECT_EQ(0u, domains[0].memory[0].base_address);
  EXPECT_EQ(0x1030000000u, domains[0].memory[0].length);

  EXPECT_EQ(0x1030000000u, domains[1].memory[0].base_address);
  EXPECT_EQ(0x1000000000u, domains[1].memory[0].length);
  END_TEST;
}

bool test_numa_2970wx() {
  BEGIN_TEST;
  AcpiTables tables(&acpi_test_data::k2970wxTableProvider);
  size_t domain_counts[4] = {0};
  AcpiNumaDomain domains[4];
  EXPECT_EQ(ZX_OK, tables.VisitCpuNumaPairs([&](const AcpiNumaDomain& region, uint32_t apic_id) {
    domain_counts[region.domain]++;
    domains[region.domain] = region;
  }));

  ASSERT_EQ(12u, domain_counts[0]);
  ASSERT_EQ(12u, domain_counts[1]);
  ASSERT_EQ(12u, domain_counts[2]);
  ASSERT_EQ(12u, domain_counts[3]);
  ASSERT_EQ(3u, domains[0].memory_count);
  ASSERT_EQ(0u, domains[1].memory_count);
  ASSERT_EQ(1u, domains[2].memory_count);
  ASSERT_EQ(0u, domains[3].memory_count);

  EXPECT_EQ(0x0u, domains[0].memory[0].base_address);
  EXPECT_EQ(0xa0000u, domains[0].memory[0].length);
  EXPECT_EQ(0x100000u, domains[0].memory[1].base_address);
  EXPECT_EQ(0x7ff00000u, domains[0].memory[1].length);
  EXPECT_EQ(0x100000000u, domains[0].memory[2].base_address);
  EXPECT_EQ(0x180000000u, domains[0].memory[2].length);

  EXPECT_EQ(0x280000000u, domains[2].memory[0].base_address);
  EXPECT_EQ(0x200000000u, domains[2].memory[0].length);

  END_TEST;
}

bool test_dbg2_parsing() {
  BEGIN_TEST;

  AcpiTables tables(&acpi_test_data::kAtlasDbg2TableProvider);
  AcpiDebugPortDescriptor desc;
  ASSERT_EQ(ZX_OK, tables.debug_port(&desc));
  EXPECT_EQ(0xfe03'4000u, desc.address);

  END_TEST;
}

UNITTEST_START_TESTCASE(acpi_tables_tests)
UNITTEST("Enumerate cpus using pixelbook eve data", test_cpus_eve)
UNITTEST("Enumerate cpus using HP z840 data.", test_cpus_z840)
UNITTEST("Enumerate io_apic_ids.", test_io)
UNITTEST("Enumerate interrupt_source_overrides.", test_interrupt_source_overrides)
UNITTEST("Lookup HPET.", test_hpet)
UNITTEST("Enumerate NUMA regions using HP z840 data.", test_numa_z840)
UNITTEST("Enumerate NUMA regions using Threadripper 2970 data.", test_numa_2970wx)
UNITTEST("Parse DBG2 table entry.", test_dbg2_parsing)
UNITTEST_END_TESTCASE(acpi_tables_tests, "acpi_tables", "Test parsing of acpi tables.")