/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2024 Arm Ltd
 * Copyright (c) 2022 The FreeBSD Foundation
 *
 * Portions of this software were developed by Andrew Turner under sponsorship
 * from the FreeBSD Foundation.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/bus.h>
#include <sys/kernel.h>
#include <sys/lock.h>
#include <sys/module.h>
#include <sys/mutex.h>

#include <contrib/dev/acpica/include/acpi.h>
#include <dev/acpica/acpivar.h>

#include <arm64/spe/arm_spe_dev.h>

static device_identify_t arm_spe_acpi_identify;
static device_probe_t arm_spe_acpi_probe;

static device_method_t arm_spe_acpi_methods[] = {
        /* Device interface */
        DEVMETHOD(device_identify,      arm_spe_acpi_identify),
        DEVMETHOD(device_probe,         arm_spe_acpi_probe),

        DEVMETHOD_END,
};

DEFINE_CLASS_1(spe, arm_spe_acpi_driver, arm_spe_acpi_methods,
    sizeof(struct arm_spe_softc), arm_spe_driver);

DRIVER_MODULE(spe, acpi, arm_spe_acpi_driver, 0, 0);

struct madt_data {
        u_int irq;
        bool found;
        bool valid;
};

static void
madt_handler(ACPI_SUBTABLE_HEADER *entry, void *arg)
{
        ACPI_MADT_GENERIC_INTERRUPT *intr;
        struct madt_data *madt_data;
        u_int irq;

        madt_data = (struct madt_data *)arg;

        /* Exit early if we are have decided to not attach */
        if (!madt_data->valid)
                return;

        switch(entry->Type) {
        case ACPI_MADT_TYPE_GENERIC_INTERRUPT:
                intr = (ACPI_MADT_GENERIC_INTERRUPT *)entry;
                irq = intr->SpeInterrupt;

                if (irq == 0) {
                        madt_data->valid = false;
                } else if (!madt_data->found) {
                        madt_data->found = true;
                        madt_data->irq = irq;
                } else if (madt_data->irq != irq) {
                        madt_data->valid = false;
                }
                break;

        default:
                break;
        }
}

static void
arm_spe_acpi_identify(driver_t *driver, device_t parent)
{
        struct madt_data madt_data;
        ACPI_TABLE_MADT *madt;
        device_t dev;
        vm_paddr_t physaddr;

        physaddr = acpi_find_table(ACPI_SIG_MADT);
        if (physaddr == 0)
                return;

        madt = acpi_map_table(physaddr, ACPI_SIG_MADT);
        if (madt == NULL) {
                device_printf(parent, "spe: Unable to map the MADT\n");
                return;
        }

        madt_data.irq = 0;
        madt_data.found = false;
        madt_data.valid = true;

        acpi_walk_subtables(madt + 1, (char *)madt + madt->Header.Length,
            madt_handler, &madt_data);

        if (!madt_data.found || !madt_data.valid)
                goto out;

        MPASS(madt_data.irq != 0);

        dev = BUS_ADD_CHILD(parent, 0, "spe", -1);
        if (dev == NULL) {
                device_printf(parent, "add spe child failed\n");
                goto out;
        }

        BUS_SET_RESOURCE(parent, dev, SYS_RES_IRQ, 0, madt_data.irq, 1);

out:
        acpi_unmap_table(madt);
}

static int
arm_spe_acpi_probe(device_t dev)
{
        device_set_desc(dev, "ARM Statistical Profiling Extension");
        return (BUS_PROBE_NOWILDCARD);
}