// SPDX-License-Identifier: GPL-2.0-or-later
/*
 *  FM801 gameport driver for Linux
 *
 *  Copyright (c) by Takashi Iwai <tiwai@suse.de>
 */

#include <asm/io.h>
#include <linux/delay.h>
#include <linux/errno.h>
#include <linux/ioport.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/pci.h>
#include <linux/slab.h>
#include <linux/gameport.h>

#define PCI_VENDOR_ID_FORTEMEDIA        0x1319
#define PCI_DEVICE_ID_FM801_GP  0x0802

#define HAVE_COOKED

struct fm801_gp {
        struct gameport *gameport;
        struct resource *res_port;
};

#ifdef HAVE_COOKED
static int fm801_gp_cooked_read(struct gameport *gameport, int *axes, int *buttons)
{
        unsigned short w;

        w = inw(gameport->io + 2);
        *buttons = (~w >> 14) & 0x03;
        axes[0] = (w == 0xffff) ? -1 : ((w & 0x1fff) << 5);
        w = inw(gameport->io + 4);
        axes[1] = (w == 0xffff) ? -1 : ((w & 0x1fff) << 5);
        w = inw(gameport->io + 6);
        *buttons |= ((~w >> 14) & 0x03) << 2;
        axes[2] = (w == 0xffff) ? -1 : ((w & 0x1fff) << 5);
        w = inw(gameport->io + 8);
        axes[3] = (w == 0xffff) ? -1 : ((w & 0x1fff) << 5);
        outw(0xff, gameport->io); /* reset */

        return 0;
}
#endif

static int fm801_gp_open(struct gameport *gameport, int mode)
{
        switch (mode) {
#ifdef HAVE_COOKED
        case GAMEPORT_MODE_COOKED:
                return 0;
#endif
        case GAMEPORT_MODE_RAW:
                return 0;
        default:
                return -1;
        }

        return 0;
}

static int fm801_gp_probe(struct pci_dev *pci, const struct pci_device_id *id)
{
        struct fm801_gp *gp;
        struct gameport *port;
        int error;

        gp = kzalloc_obj(*gp);
        port = gameport_allocate_port();
        if (!gp || !port) {
                printk(KERN_ERR "fm801-gp: Memory allocation failed\n");
                error = -ENOMEM;
                goto err_out_free;
        }

        error = pci_enable_device(pci);
        if (error)
                goto err_out_free;

        port->open = fm801_gp_open;
#ifdef HAVE_COOKED
        port->cooked_read = fm801_gp_cooked_read;
#endif
        gameport_set_name(port, "FM801");
        gameport_set_phys(port, "pci%s/gameport0", pci_name(pci));
        port->dev.parent = &pci->dev;
        port->io = pci_resource_start(pci, 0);

        gp->gameport = port;
        gp->res_port = request_region(port->io, 0x10, "FM801 GP");
        if (!gp->res_port) {
                printk(KERN_DEBUG "fm801-gp: unable to grab region 0x%x-0x%x\n",
                        port->io, port->io + 0x0f);
                error = -EBUSY;
                goto err_out_disable_dev;
        }

        pci_set_drvdata(pci, gp);

        outb(0x60, port->io + 0x0d); /* enable joystick 1 and 2 */
        gameport_register_port(port);

        return 0;

 err_out_disable_dev:
        pci_disable_device(pci);
 err_out_free:
        gameport_free_port(port);
        kfree(gp);
        return error;
}

static void fm801_gp_remove(struct pci_dev *pci)
{
        struct fm801_gp *gp = pci_get_drvdata(pci);

        gameport_unregister_port(gp->gameport);
        release_resource(gp->res_port);
        kfree(gp);

        pci_disable_device(pci);
}

static const struct pci_device_id fm801_gp_id_table[] = {
        { PCI_VDEVICE(FORTEMEDIA, PCI_DEVICE_ID_FM801_GP) },
        { }
};
MODULE_DEVICE_TABLE(pci, fm801_gp_id_table);

static struct pci_driver fm801_gp_driver = {
        .name =         "FM801_gameport",
        .id_table =     fm801_gp_id_table,
        .probe =        fm801_gp_probe,
        .remove =       fm801_gp_remove,
};

module_pci_driver(fm801_gp_driver);

MODULE_DESCRIPTION("FM801 gameport driver");
MODULE_AUTHOR("Takashi Iwai <tiwai@suse.de>");
MODULE_LICENSE("GPL");