/* SPDX-License-Identifier: GPL-2.0-only */

/*
 * Quirks for AMD IOMMU
 *
 * Copyright (C) 2019 Kai-Heng Feng <kai.heng.feng@canonical.com>
 */

#ifdef CONFIG_DMI
#include <linux/dmi.h>

#include "amd_iommu.h"

#define IVHD_SPECIAL_IOAPIC             1

struct ivrs_quirk_entry {
        u8 id;
        u32 devid;
};

enum {
        DELL_INSPIRON_7375 = 0,
        DELL_LATITUDE_5495,
        LENOVO_IDEAPAD_330S_15ARR,
};

static const struct ivrs_quirk_entry ivrs_ioapic_quirks[][3] __initconst = {
        /* ivrs_ioapic[4]=00:14.0 ivrs_ioapic[5]=00:00.2 */
        [DELL_INSPIRON_7375] = {
                { .id = 4, .devid = 0xa0 },
                { .id = 5, .devid = 0x2 },
                {}
        },
        /* ivrs_ioapic[4]=00:14.0 */
        [DELL_LATITUDE_5495] = {
                { .id = 4, .devid = 0xa0 },
                {}
        },
        /* ivrs_ioapic[32]=00:14.0 */
        [LENOVO_IDEAPAD_330S_15ARR] = {
                { .id = 32, .devid = 0xa0 },
                {}
        },
        {}
};

static int __init ivrs_ioapic_quirk_cb(const struct dmi_system_id *d)
{
        const struct ivrs_quirk_entry *i;

        for (i = d->driver_data; i->id != 0 && i->devid != 0; i++)
                add_special_device(IVHD_SPECIAL_IOAPIC, i->id, (u32 *)&i->devid, 0);

        return 0;
}

static const struct dmi_system_id ivrs_quirks[] __initconst = {
        {
                .callback = ivrs_ioapic_quirk_cb,
                .ident = "Dell Inspiron 7375",
                .matches = {
                        DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
                        DMI_MATCH(DMI_PRODUCT_NAME, "Inspiron 7375"),
                },
                .driver_data = (void *)&ivrs_ioapic_quirks[DELL_INSPIRON_7375],
        },
        {
                .callback = ivrs_ioapic_quirk_cb,
                .ident = "Dell Latitude 5495",
                .matches = {
                        DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
                        DMI_MATCH(DMI_PRODUCT_NAME, "Latitude 5495"),
                },
                .driver_data = (void *)&ivrs_ioapic_quirks[DELL_LATITUDE_5495],
        },
        {
                /*
                 * Acer Aspire A315-41 requires the very same workaround as
                 * Dell Latitude 5495
                 */
                .callback = ivrs_ioapic_quirk_cb,
                .ident = "Acer Aspire A315-41",
                .matches = {
                        DMI_MATCH(DMI_SYS_VENDOR, "Acer"),
                        DMI_MATCH(DMI_PRODUCT_NAME, "Aspire A315-41"),
                },
                .driver_data = (void *)&ivrs_ioapic_quirks[DELL_LATITUDE_5495],
        },
        {
                .callback = ivrs_ioapic_quirk_cb,
                .ident = "Lenovo ideapad 330S-15ARR",
                .matches = {
                        DMI_MATCH(DMI_SYS_VENDOR, "LENOVO"),
                        DMI_MATCH(DMI_PRODUCT_NAME, "81FB"),
                },
                .driver_data = (void *)&ivrs_ioapic_quirks[LENOVO_IDEAPAD_330S_15ARR],
        },
        {}
};

void __init amd_iommu_apply_ivrs_quirks(void)
{
        dmi_check_system(ivrs_quirks);
}
#endif