/*	$NetBSD: drm_pci.c,v 1.6.2.4 2019/04/16 08:10:22 martin Exp $	*/

/*-
 * Copyright (c) 2013 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Taylor R. Campbell.
 *
 * 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 NETBSD FOUNDATION, INC. 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 FOUNDATION 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/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: drm_pci.c,v 1.6.2.4 2019/04/16 08:10:22 martin Exp $");

#include <sys/types.h>
#include <sys/errno.h>
#include <sys/systm.h>

#include <dev/pci/pcivar.h>

#include <drm/drmP.h>

static int	drm_pci_get_irq(struct drm_device *);
static int	drm_pci_irq_install(struct drm_device *,
		    irqreturn_t (*)(void *), int, const char *, void *,
		    struct drm_bus_irq_cookie **);
static void	drm_pci_irq_uninstall(struct drm_device *,
		    struct drm_bus_irq_cookie *);
static const char *
		drm_pci_get_name(struct drm_device *);
static int	drm_pci_set_busid(struct drm_device *, struct drm_master *);
static int	drm_pci_set_unique(struct drm_device *, struct drm_master *,
		    struct drm_unique *);
static int	drm_pci_irq_by_busid(struct drm_device *,
		    struct drm_irq_busid *);

const struct drm_bus drm_pci_bus = {
	.bus_type = DRIVER_BUS_PCI,
	.get_irq = drm_pci_get_irq,
	.irq_install = drm_pci_irq_install,
	.irq_uninstall = drm_pci_irq_uninstall,
	.get_name = drm_pci_get_name,
	.set_busid = drm_pci_set_busid,
	.set_unique = drm_pci_set_unique,
	.irq_by_busid = drm_pci_irq_by_busid,
};

static const struct pci_attach_args *
drm_pci_attach_args(struct drm_device *dev)
{
	return &dev->pdev->pd_pa;
}

int
drm_pci_init(struct drm_driver *driver, struct pci_driver *pdriver __unused)
{

	driver->bus = &drm_pci_bus;
	return 0;
}

void
drm_pci_exit(struct drm_driver *driver __unused,
    struct pci_driver *pdriver __unused)
{
}

int
drm_pci_attach(device_t self, const struct pci_attach_args *pa,
    struct pci_dev *pdev, struct drm_driver *driver, unsigned long cookie,
    struct drm_device **devp)
{
	struct drm_device *dev;
	unsigned int unit;
	int ret;

	/* Ensure the drm agp hooks are installed.  */
	/* XXX errno NetBSD->Linux */
	ret = -drmkms_pci_agp_guarantee_initialized();
	if (ret)
		goto fail0;

	/* Initialize the Linux PCI device descriptor.  */
	linux_pci_dev_init(pdev, self, pa, 0);

	/* Create a DRM device.  */
	dev = drm_dev_alloc(driver, self);
	if (dev == NULL) {
		ret = -ENOMEM;
		goto fail0;
	}

	dev->pdev = pdev;
	pdev->pd_drm_dev = dev;	/* XXX Nouveau kludge.  */

	/* XXX Set the power state to D0?  */

	/* Set up the bus space and bus DMA tags.  */
	dev->bst = pa->pa_memt;
	/* XXX Let the driver say something about 32-bit vs 64-bit DMA?  */
	dev->bus_dmat = (pci_dma64_available(pa)? pa->pa_dmat64 : pa->pa_dmat);
	dev->dmat = dev->bus_dmat;
	dev->dmat_subregion_p = false;

	/* Find all the memory maps.  */
	CTASSERT(PCI_NUM_RESOURCES < (SIZE_MAX / sizeof(dev->bus_maps[0])));
	dev->bus_maps = kmem_zalloc(PCI_NUM_RESOURCES *
	    sizeof(dev->bus_maps[0]), KM_SLEEP);
	dev->bus_nmaps = PCI_NUM_RESOURCES;
	for (unit = 0; unit < PCI_NUM_RESOURCES; unit++) {
		struct drm_bus_map *const bm = &dev->bus_maps[unit];
		const int reg = PCI_BAR(unit);
		const pcireg_t type =
		    pci_mapreg_type(pa->pa_pc, pa->pa_tag, reg);

		/* Reject non-memory mappings.  */
		if ((type & PCI_MAPREG_TYPE_MEM) != PCI_MAPREG_TYPE_MEM) {
			aprint_debug_dev(self, "map %u has non-memory type:"
			    " 0x%"PRIxMAX"\n", unit, (uintmax_t)type);
			continue;
		}

		/*
		 * If it's a 64-bit mapping, don't interpret the second
		 * half of it as another BAR in the next iteration of
		 * the loop -- move on to the next unit.
		 */
		if (PCI_MAPREG_MEM_TYPE(type) == PCI_MAPREG_MEM_TYPE_64BIT)
			unit++;

		/* Inquire about it.  We'll map it in drm_core_ioremap.  */
		if (pci_mapreg_info(pa->pa_pc, pa->pa_tag, reg, type,
			&bm->bm_base, &bm->bm_size, &bm->bm_flags) != 0) {
			aprint_debug_dev(self, "map %u failed\n", unit);
			continue;
		}

		/* Assume since it is a memory mapping it can be linear.  */
		bm->bm_flags |= BUS_SPACE_MAP_LINEAR;
	}

	/* Set up AGP stuff if requested.  */
	if (drm_core_check_feature(dev, DRIVER_USE_AGP)) {
		if (drm_pci_device_is_agp(dev))
			dev->agp = drm_agp_init(dev);
		if (dev->agp)
			dev->agp->agp_mtrr = arch_phys_wc_add(dev->agp->base,
				dev->agp->agp_info.aki_info.ai_aperture_size);
	}

	/* Register the DRM device and do driver-specific initialization.  */
	ret = drm_dev_register(dev, cookie);
	if (ret)
		goto fail1;

	/* Success!  */
	*devp = dev;
	return 0;

fail2: __unused
	drm_dev_unregister(dev);
fail1:	drm_pci_agp_destroy(dev);
	dev->bus_nmaps = 0;
	kmem_free(dev->bus_maps, PCI_NUM_RESOURCES * sizeof(dev->bus_maps[0]));
	if (dev->dmat_subregion_p)
		bus_dmatag_destroy(dev->dmat);
	drm_dev_unref(dev);
fail0:	return ret;
}

int
drm_pci_detach(struct drm_device *dev, int flags __unused)
{

	/* Do driver-specific detachment and unregister the device.  */
	drm_dev_unregister(dev);

	/* Tear down AGP stuff if necessary.  */
	drm_pci_agp_destroy(dev);

	/* Free the record of available bus space mappings.  */
	dev->bus_nmaps = 0;
	kmem_free(dev->bus_maps, PCI_NUM_RESOURCES * sizeof(dev->bus_maps[0]));

	/* Tear down bus space and bus DMA tags.  */
	if (dev->dmat_subregion_p)
		bus_dmatag_destroy(dev->dmat);

	drm_dev_unref(dev);

	return 0;
}

void
drm_pci_agp_destroy(struct drm_device *dev)
{

	if (dev->agp) {
		arch_phys_wc_del(dev->agp->agp_mtrr);
		drm_agp_clear(dev);
		kfree(dev->agp); /* XXX Should go in drm_agp_clear...  */
		dev->agp = NULL;
	}
}

static int
drm_pci_get_irq(struct drm_device *dev)
{

	/*
	 * Caller expects a nonzero int, and doesn't really use it for
	 * anything, so no need to pci_intr_map here.
	 */
	return dev->pdev->pd_pa.pa_intrpin;
}

static int
drm_pci_irq_install(struct drm_device *dev, irqreturn_t (*handler)(void *),
    int flags, const char *name, void *arg,
    struct drm_bus_irq_cookie **cookiep)
{
	const struct pci_attach_args *const pa = drm_pci_attach_args(dev);
	pci_intr_handle_t ih;
	const char *intrstr;
	void *ih_cookie;
	char intrbuf[PCI_INTRSTR_LEN];

	if (pci_intr_map(pa, &ih))
		return -ENOENT;

	intrstr = pci_intr_string(pa->pa_pc, ih, intrbuf, sizeof(intrbuf));
	ih_cookie = pci_intr_establish(pa->pa_pc, ih, IPL_DRM, handler, arg);
	if (ih_cookie == NULL) {
		aprint_error_dev(dev->dev,
		    "couldn't establish interrupt at %s (%s)\n",
		    intrstr, name);
		return -ENOENT;
	}

	aprint_normal_dev(dev->dev, "interrupting at %s (%s)\n",
	    intrstr, name);
	*cookiep = (struct drm_bus_irq_cookie *)ih_cookie;
	return 0;
}

static void
drm_pci_irq_uninstall(struct drm_device *dev,
    struct drm_bus_irq_cookie *cookie)
{
	const struct pci_attach_args *pa = drm_pci_attach_args(dev);

	pci_intr_disestablish(pa->pa_pc, (void *)cookie);
}

static const char *
drm_pci_get_name(struct drm_device *dev)
{
	return "pci";		/* XXX PCI bus names?  */
}

static int
drm_pci_format_unique(struct drm_device *dev, char *buf, size_t size)
{
	const unsigned int domain = device_unit(device_parent(dev->dev));
	const unsigned int bus = dev->pdev->pd_pa.pa_bus;
	const unsigned int device = dev->pdev->pd_pa.pa_device;
	const unsigned int function = dev->pdev->pd_pa.pa_function;

	return snprintf(buf, size, "pci:%04x:%02x:%02x.%d",
	    domain, bus, device, function);
}

static int
drm_pci_format_devname(struct drm_device *dev, const char *unique,
    char *buf, size_t size)
{

	return snprintf(buf, size, "%s@%s",
	    device_xname(device_parent(dev->dev)),
	    unique);
}

static int
drm_pci_set_busid(struct drm_device *dev, struct drm_master *master)
{
	int n;
	char *buf;

	n = drm_pci_format_unique(dev, NULL, 0);
	if (n < 0)
		return -ENOSPC;	/* XXX */
	if (0xff < n)
		n = 0xff;

	buf = kzalloc(n + 1, GFP_KERNEL);
	(void)drm_pci_format_unique(dev, buf, n + 1);

	if (master->unique)
		kfree(master->unique);
	master->unique = buf;
	master->unique_len = n;
	master->unique_size = n + 1;

	n = drm_pci_format_devname(dev, master->unique, NULL, 0);
	if (n < 0)
		return -ENOSPC;	/* XXX back out? */
	if (0xff < n)
		n = 0xff;

	buf = kzalloc(n + 1, GFP_KERNEL);
	(void)drm_pci_format_devname(dev, master->unique, buf, n + 1);

	if (dev->devname)
		kfree(dev->devname);
	dev->devname = buf;

	return 0;
}

static int
drm_pci_set_unique(struct drm_device *dev, struct drm_master *master,
    struct drm_unique *unique __unused)
{

	/*
	 * XXX This is silly.  We're supposed to reject unique names
	 * that don't match the ones we would generate anyway.  For
	 * expedience, we'll just generate the one we would and ignore
	 * whatever userland threw at us...
	 */
	return drm_pci_set_busid(dev, master);
}

static int
drm_pci_irq_by_busid(struct drm_device *dev, struct drm_irq_busid *busid)
{
	return -ENOSYS;		/* XXX */
}
