/*
 * Copyright 2003 José Fonseca.
 * Copyright 2003 Leif Delgass.
 * All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice (including the next
 * paragraph) shall be included in all copies or substantial portions of the
 * Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
 * AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include <linux/pci.h>
#include <linux/slab.h>
#include <linux/dma-mapping.h>
#include <linux/export.h>
#include <drm/drm_pci.h>
#include <drm/drmP.h>
#include "drm_internal.h"
#include "drm_legacy.h"

#include <sys/bus.h>

static void
drm_pci_busdma_callback(void *arg, bus_dma_segment_t *segs, int nsegs, int error)
{
        drm_dma_handle_t *dmah = arg;

        if (error != 0)
                return;

        KASSERT(nsegs == 1, ("drm_pci_busdma_callback: bad dma segment count"));
        dmah->busaddr = segs[0].ds_addr;
}

/**
 * drm_pci_alloc - Allocate a PCI consistent memory block, for DMA.
 * @dev: DRM device
 * @size: size of block to allocate
 * @align: alignment of block
 *
 * FIXME: This is a needless abstraction of the Linux dma-api and should be
 * removed.
 *
 * Return: A handle to the allocated memory block on success or NULL on
 * failure.
 */
drm_dma_handle_t *drm_pci_alloc(struct drm_device * dev, size_t size, size_t align)
{
#ifdef __DragonFly__
        drm_dma_handle_t *dmah;
        int ret;

        /* Need power-of-two alignment, so fail the allocation if it isn't. */
        if ((align & (align - 1)) != 0) {
                DRM_ERROR("drm_pci_alloc with non-power-of-two alignment %d\n",
                    (int)align);
                return NULL;
        }

        dmah = kmalloc(sizeof(drm_dma_handle_t), M_DRM, M_ZERO | M_NOWAIT);
        if (dmah == NULL)
                return NULL;

        ret = bus_dma_tag_create(
            bus_get_dma_tag(dev->dev->bsddev), /* parent */
            align, 0, /* align, boundary */
            BUS_SPACE_MAXADDR, BUS_SPACE_MAXADDR, /* lowaddr, highaddr */
            size, 1, size, /* maxsize, nsegs, maxsegsize */
            0, /* flags */
            &dmah->tag);
        if (ret != 0) {
                kfree(dmah);
                return NULL;
        }

        ret = bus_dmamem_alloc(dmah->tag, (void **)&dmah->vaddr,
            BUS_DMA_WAITOK | BUS_DMA_ZERO | BUS_DMA_NOCACHE, &dmah->map);
        if (ret != 0) {
                bus_dma_tag_destroy(dmah->tag);
                kfree(dmah);
                return NULL;
        }

        ret = bus_dmamap_load(dmah->tag, dmah->map, dmah->vaddr, size,
            drm_pci_busdma_callback, dmah, BUS_DMA_NOWAIT);
        if (ret != 0) {
                bus_dmamem_free(dmah->tag, dmah->vaddr, dmah->map);
                bus_dma_tag_destroy(dmah->tag);
                kfree(dmah);
                return NULL;
        }
        return dmah;
#else
        drm_dma_handle_t *dmah;
        unsigned long addr;
        size_t sz;

        /* pci_alloc_consistent only guarantees alignment to the smallest
         * PAGE_SIZE order which is greater than or equal to the requested size.
         * Return NULL here for now to make sure nobody tries for larger alignment
         */
        if (align > size)
                return NULL;

        dmah = kmalloc(sizeof(drm_dma_handle_t), M_DRM, GFP_KERNEL);
        if (!dmah)
                return NULL;

        dmah->size = size;
        dmah->vaddr = dma_alloc_coherent(&dev->pdev->dev, size, &dmah->busaddr, GFP_KERNEL);

        if (dmah->vaddr == NULL) {
                kfree(dmah);
                return NULL;
        }

        memset(dmah->vaddr, 0, size);

        /* XXX - Is virt_to_page() legal for consistent mem? */
        /* Reserve */
        for (addr = (unsigned long)dmah->vaddr, sz = size;
             sz > 0; addr += PAGE_SIZE, sz -= PAGE_SIZE) {
#if 0
                SetPageReserved(virt_to_page((void *)addr));
#endif
        }

        return dmah;
#endif
}

EXPORT_SYMBOL(drm_pci_alloc);

/*
 * Free a PCI consistent memory block without freeing its descriptor.
 *
 * This function is for internal use in the Linux-specific DRM core code.
 */
void __drm_legacy_pci_free(struct drm_device * dev, drm_dma_handle_t * dmah)
{
#ifdef __DragonFly__
        if (dmah == NULL)
                return;

        bus_dmamap_unload(dmah->tag, dmah->map);
        bus_dmamem_free(dmah->tag, dmah->vaddr, dmah->map);
        bus_dma_tag_destroy(dmah->tag);
#else
        unsigned long addr;
        size_t sz;

        if (dmah->vaddr) {
                /* XXX - Is virt_to_page() legal for consistent mem? */
                /* Unreserve */
                for (addr = (unsigned long)dmah->vaddr, sz = dmah->size;
                     sz > 0; addr += PAGE_SIZE, sz -= PAGE_SIZE) {
#if 0
                        ClearPageReserved(virt_to_page((void *)addr));
#endif
                }
                dma_free_coherent(&dev->pdev->dev, dmah->size, dmah->vaddr,
                                  dmah->busaddr);
        }
#endif
}

/**
 * drm_pci_free - Free a PCI consistent memory block
 * @dev: DRM device
 * @dmah: handle to memory block
 *
 * FIXME: This is a needless abstraction of the Linux dma-api and should be
 * removed.
 */
void drm_pci_free(struct drm_device * dev, drm_dma_handle_t * dmah)
{
        __drm_legacy_pci_free(dev, dmah);
        kfree(dmah);
}

EXPORT_SYMBOL(drm_pci_free);

#ifdef CONFIG_PCI

static int drm_get_pci_domain(struct drm_device *dev)
{
#ifndef __alpha__
        /* For historical reasons, drm_get_pci_domain() is busticated
         * on most archs and has to remain so for userspace interface
         * < 1.4, except on alpha which was right from the beginning
         */
        if (dev->if_version < 0x10004)
                return 0;
#endif /* __alpha__ */

#if 0
        return pci_domain_nr(dev->pdev->bus);
#else
        return dev->pci_domain;
#endif
}

int drm_pci_set_busid(struct drm_device *dev, struct drm_master *master)
{
        master->unique = kasprintf(GFP_KERNEL, "pci:%04x:%02x:%02x.%d",
                                        drm_get_pci_domain(dev),
                                        dev->pdev->bus->number,
                                        PCI_SLOT(dev->pdev->devfn),
                                        PCI_FUNC(dev->pdev->devfn));
        if (!master->unique)
                return -ENOMEM;

        master->unique_len = strlen(master->unique);
        return 0;
}

static int drm_pci_irq_by_busid(struct drm_device *dev, struct drm_irq_busid *p)
{
        if ((p->busnum >> 8) != drm_get_pci_domain(dev) ||
            (p->busnum & 0xff) != dev->pdev->bus->number ||
            p->devnum != PCI_SLOT(dev->pdev->devfn) || p->funcnum != PCI_FUNC(dev->pdev->devfn))
                return -EINVAL;

        p->irq = dev->pdev->irq;

        DRM_DEBUG("%d:%d:%d => IRQ %d\n", p->busnum, p->devnum, p->funcnum,
                  p->irq);
        return 0;
}

/**
 * drm_irq_by_busid - Get interrupt from bus ID
 * @dev: DRM device
 * @data: IOCTL parameter pointing to a drm_irq_busid structure
 * @file_priv: DRM file private.
 *
 * Finds the PCI device with the specified bus id and gets its IRQ number.
 * This IOCTL is deprecated, and will now return EINVAL for any busid not equal
 * to that of the device that this DRM instance attached to.
 *
 * Return: 0 on success or a negative error code on failure.
 */
int drm_irq_by_busid(struct drm_device *dev, void *data,
                     struct drm_file *file_priv)
{
        struct drm_irq_busid *p = data;

        if (!drm_core_check_feature(dev, DRIVER_LEGACY))
                return -EOPNOTSUPP;

        /* UMS was only ever support on PCI devices. */
        if (WARN_ON(!dev->pdev))
                return -EINVAL;

        if (!drm_core_check_feature(dev, DRIVER_HAVE_IRQ))
                return -EOPNOTSUPP;

        return drm_pci_irq_by_busid(dev, p);
}

#ifdef __DragonFly__
int
drm_getpciinfo(struct drm_device *dev, void *data, struct drm_file *file_priv)
{
        struct drm_pciinfo *info = data;

        info->domain = 0;
        info->bus = dev->pci_bus;
        info->dev = PCI_SLOT(dev->pdev->devfn);
        info->func = PCI_FUNC(dev->pdev->devfn);
        info->vendor_id = dev->pdev->vendor;
        info->device_id = dev->pdev->device;
        info->subvendor_id = dev->pdev->subsystem_vendor;
        info->subdevice_id = dev->pdev->subsystem_device;
        info->revision_id = 0;

        return 0;
}
#endif

/**
 * drm_get_pci_dev - Register a PCI device with the DRM subsystem
 * @pdev: PCI device
 * @ent: entry from the PCI ID table that matches @pdev
 * @driver: DRM device driver
 *
 * Attempt to gets inter module "drm" information. If we are first
 * then register the character device and inter module information.
 * Try and register, if we fail to register, backout previous work.
 *
 * NOTE: This function is deprecated, please use drm_dev_alloc() and
 * drm_dev_register() instead and remove your &drm_driver.load callback.
 *
 * Return: 0 on success or a negative error code on failure.
 */
int drm_get_pci_dev(struct pci_dev *pdev, const struct pci_device_id *ent,
                    struct drm_driver *driver)
{
        struct drm_device *dev;
        int ret;

        DRM_DEBUG("\n");

        dev = drm_dev_alloc(driver, &pdev->dev);
        if (IS_ERR(dev))
                return PTR_ERR(dev);

#if 0
        ret = pci_enable_device(pdev);
        if (ret)
                goto err_free;
#endif

        dev->pdev = pdev;
#ifdef __alpha__
        dev->hose = pdev->sysdata;
#endif

        if (drm_core_check_feature(dev, DRIVER_MODESET))
                pci_set_drvdata(pdev, dev);

#if 0
        drm_pci_agp_init(dev);
#endif

        ret = drm_dev_register(dev, ent->driver_data);
        if (ret)
                goto err_agp;

        /* No locking needed since shadow-attach is single-threaded since it may
         * only be called from the per-driver module init hook. */
        if (drm_core_check_feature(dev, DRIVER_LEGACY))
                list_add_tail(&dev->legacy_dev_list, &driver->legacy_dev_list);

        return 0;

err_agp:
#if 0
        drm_pci_agp_destroy(dev);
        pci_disable_device(pdev);
err_free:
        drm_dev_unref(dev);
#endif
        return ret;
}
EXPORT_SYMBOL(drm_get_pci_dev);

/**
 * drm_legacy_pci_init - shadow-attach a legacy DRM PCI driver
 * @driver: DRM device driver
 * @pdriver: PCI device driver
 *
 * This is only used by legacy dri1 drivers and deprecated.
 *
 * Return: 0 on success or a negative error code on failure.
 */
int drm_legacy_pci_init(struct drm_driver *driver, struct pci_driver *pdriver)
{
#if 0
        struct pci_dev *pdev = NULL;
        const struct pci_device_id *pid;
        int i;
#endif

        DRM_DEBUG("\n");

        if (WARN_ON(!(driver->driver_features & DRIVER_LEGACY)))
                return -EINVAL;

#if 0
        /* If not using KMS, fall back to stealth mode manual scanning. */
        INIT_LIST_HEAD(&driver->legacy_dev_list);
        for (i = 0; pdriver->id_table[i].vendor != 0; i++) {
                pid = &pdriver->id_table[i];

                /* Loop around setting up a DRM device for each PCI device
                 * matching our ID and device class.  If we had the internal
                 * function that pci_get_subsys and pci_get_class used, we'd
                 * be able to just pass pid in instead of doing a two-stage
                 * thing.
                 */
                pdev = NULL;
                while ((pdev =
                        pci_get_subsys(pid->vendor, pid->device, pid->subvendor,
                                       pid->subdevice, pdev)) != NULL) {
                        if ((pdev->class & pid->class_mask) != pid->class)
                                continue;

                        /* stealth mode requires a manual probe */
                        pci_dev_get(pdev);
                        drm_get_pci_dev(pdev, pid, driver);
                }
        }
#endif
        return 0;
}
EXPORT_SYMBOL(drm_legacy_pci_init);

#else

void drm_pci_agp_destroy(struct drm_device *dev) {}

int drm_irq_by_busid(struct drm_device *dev, void *data,
                     struct drm_file *file_priv)
{
        return -EINVAL;
}
#endif

/**
 * drm_legacy_pci_exit - unregister shadow-attach legacy DRM driver
 * @driver: DRM device driver
 * @pdriver: PCI device driver
 *
 * Unregister a DRM driver shadow-attached through drm_legacy_pci_init(). This
 * is deprecated and only used by dri1 drivers.
 */
void drm_legacy_pci_exit(struct drm_driver *driver, struct pci_driver *pdriver)
{
        struct drm_device *dev, *tmp;
        DRM_DEBUG("\n");

        if (!(driver->driver_features & DRIVER_LEGACY)) {
                WARN_ON(1);
        } else {
                list_for_each_entry_safe(dev, tmp, &driver->legacy_dev_list,
                                         legacy_dev_list) {
                        list_del(&dev->legacy_dev_list);
#if 0
                        drm_put_dev(dev);
#endif
                }
        }
        DRM_INFO("Module unloaded\n");
}
EXPORT_SYMBOL(drm_legacy_pci_exit);