// SPDX-License-Identifier: GPL-2.0+
// Copyright 2019 IBM Corp.
#include <linux/module.h>
#include "ocxl_internal.h"

/*
 * Any opencapi device which wants to use this 'generic' driver should
 * use the 0x062B device ID. Vendors should define the subsystem
 * vendor/device ID to help differentiate devices.
 */
static const struct pci_device_id ocxl_pci_tbl[] = {
        { PCI_DEVICE(PCI_VENDOR_ID_IBM, 0x062B), },
        { }
};
MODULE_DEVICE_TABLE(pci, ocxl_pci_tbl);

static int ocxl_probe(struct pci_dev *dev, const struct pci_device_id *id)
{
        int rc;
        struct ocxl_afu *afu, *tmp;
        struct ocxl_fn *fn;
        struct list_head *afu_list;

        fn = ocxl_function_open(dev);
        if (IS_ERR(fn))
                return PTR_ERR(fn);

        pci_set_drvdata(dev, fn);

        afu_list = ocxl_function_afu_list(fn);

        list_for_each_entry_safe(afu, tmp, afu_list, list) {
                // Cleanup handled within ocxl_file_register_afu()
                rc = ocxl_file_register_afu(afu);
                if (rc) {
                        dev_err(&dev->dev, "Failed to register AFU '%s' index %d",
                                        afu->config.name, afu->config.idx);
                }
        }

        return 0;
}

static void ocxl_remove(struct pci_dev *dev)
{
        struct ocxl_fn *fn;
        struct ocxl_afu *afu;
        struct list_head *afu_list;

        fn = pci_get_drvdata(dev);
        afu_list = ocxl_function_afu_list(fn);

        list_for_each_entry(afu, afu_list, list) {
                ocxl_file_unregister_afu(afu);
        }

        ocxl_function_close(fn);
}

struct pci_driver ocxl_pci_driver = {
        .name = "ocxl",
        .id_table = ocxl_pci_tbl,
        .probe = ocxl_probe,
        .remove = ocxl_remove,
        .shutdown = ocxl_remove,
};