/*-
 * Copyright 1999 Precision Insight, Inc., Cedar Park, Texas.
 * Copyright 2000 VA Linux Systems, Inc., Sunnyvale, California.
 * 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
 * VA LINUX SYSTEMS AND/OR ITS SUPPLIERS 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.
 *
 * Author:
 *    Rickard E. (Rik) Faith <faith@valinux.com>
 *    Gareth Hughes <gareth@valinux.com>
 *
 */

/*
 * Support code for tying the kernel AGP support to DRM drivers and
 * the DRM's AGP ioctls.
 */

#include "drmP.h"

struct drm_agp_mem	*drm_agp_lookup_entry(struct drm_device *, void *);

int
drm_device_is_agp(struct drm_device *dev)
{
	if (dev->driver.device_is_agp != NULL) {
		int ret;
		/*
		 * device_is_agp returns a tristate, 0 = not AGP, 1 = definitely
		 * AGP, 2 = fall back to PCI capability
		 */
		ret = (*dev->driver.device_is_agp)(dev);
		if (ret != DRM_MIGHT_BE_AGP)
			return (ret);
	}

	return (pci_get_capability(dev->pa.pa_pc, dev->pa.pa_tag, PCI_CAP_AGP,
	    NULL, NULL));
}

int
drm_device_is_pcie(struct drm_device *dev)
{
	return (pci_get_capability(dev->pa.pa_pc, dev->pa.pa_tag,
	    PCI_CAP_PCIEXPRESS, NULL, NULL));
}

int
drm_agp_info(struct drm_device * dev, struct drm_agp_info *info)
{
	struct agp_info	*kern;

	if (dev->agp == NULL || !dev->agp->acquired)
		return (EINVAL);

	kern = &dev->agp->info;
#ifndef DRM_NO_AGP
	agp_get_info(dev->agp->agpdev, kern);
#endif
	info->agp_version_major = 1;
	info->agp_version_minor = 0;
	info->mode = kern->ai_mode;
	info->aperture_base = kern->ai_aperture_base;
	info->aperture_size = kern->ai_aperture_size;
	info->memory_allowed = kern->ai_memory_allowed;
	info->memory_used = kern->ai_memory_used;
	info->id_vendor = kern->ai_devid & 0xffff;
	info->id_device = kern->ai_devid >> 16;

	return (0);
}

int
drm_agp_info_ioctl(struct drm_device *dev, void *data,
    struct drm_file *file_priv)
{
	struct drm_agp_info	*info = data;

	return (drm_agp_info(dev, info));
}

int
drm_agp_acquire_ioctl(struct drm_device *dev, void *data,
    struct drm_file *file_priv)
{
	return (drm_agp_acquire(dev));
}

int
drm_agp_acquire(struct drm_device *dev)
{
#ifndef DRM_NO_AGP
	int	retcode;

	if (dev->agp == NULL || dev->agp->acquired)
		return (EINVAL);

	retcode = agp_acquire(dev->agp->agpdev);
	if (retcode)
		return (retcode);

	dev->agp->acquired = 1;
#endif
	return (0);
}

int
drm_agp_release_ioctl(struct drm_device *dev, void *data,
    struct drm_file *file_priv)
{
	return (drm_agp_release(dev));
}

int
drm_agp_release(struct drm_device * dev)
{
#ifndef DRM_NO_AGP
	if (dev->agp == NULL || !dev->agp->acquired)
		return (EINVAL);
	agp_release(dev->agp->agpdev);
	dev->agp->acquired = 0;
#endif
	return (0);
}

int
drm_agp_enable(struct drm_device *dev, drm_agp_mode_t mode)
{
	int	retcode = 0;
#ifndef DRM_NO_AGP
	if (dev->agp == NULL || !dev->agp->acquired)
		return (EINVAL);
	
	dev->agp->mode = mode.mode;
	if ((retcode = agp_enable(dev->agp->agpdev, mode.mode)) == 0)
		dev->agp->enabled = 1;
#endif
	return (retcode);
}

int
drm_agp_enable_ioctl(struct drm_device *dev, void *data,
    struct drm_file *file_priv)
{
	struct drm_agp_mode	*mode = data;

	return (drm_agp_enable(dev, *mode));
}

int
drm_agp_alloc(struct drm_device *dev, struct drm_agp_buffer *request)
{
#ifndef DRM_NO_AGP
	struct drm_agp_mem	*entry;
	void			*handle;
	struct agp_memory_info	 info;
	unsigned long		 pages;
	u_int32_t		 type;

	if (dev->agp == NULL || !dev->agp->acquired)
		return (EINVAL);

	entry = drm_alloc(sizeof(*entry), DRM_MEM_AGPLISTS);
	if (entry == NULL)
		return (ENOMEM);

	pages = (request->size + PAGE_SIZE - 1) / PAGE_SIZE;
	type = (u_int32_t)request->type;

	DRM_UNLOCK();
	handle = drm_agp_allocate_memory(pages, type);
	DRM_LOCK();
	if (handle == NULL) {
		drm_free(entry, sizeof(*entry), DRM_MEM_AGPLISTS);
		return (ENOMEM);
	}
	
	entry->handle = handle;
	entry->bound = 0;
	entry->pages = pages;
	TAILQ_INSERT_HEAD(&dev->agp->memory, entry, link);

	agp_memory_info(dev->agp->agpdev, entry->handle, &info);

	request->handle = (unsigned long)entry->handle;
        request->physical = info.ami_physical;
#endif

	return (0);
}

int
drm_agp_alloc_ioctl(struct drm_device *dev, void *data,
    struct drm_file *file_priv)
{
	struct drm_agp_buffer	*request = data;
	int			 retcode;

	DRM_LOCK();
	retcode = drm_agp_alloc(dev, request);
	DRM_UNLOCK();
	return (retcode);
}

struct drm_agp_mem *
drm_agp_lookup_entry(struct drm_device *dev, void *handle)
{
	struct drm_agp_mem	*entry;

	TAILQ_FOREACH(entry, &dev->agp->memory, link) {
		if (entry->handle == handle)
			break;
	}
	return (entry);
}

int
drm_agp_unbind(struct drm_device *dev, struct drm_agp_binding *request)
{
	struct drm_agp_mem	*entry;
	int			 retcode;

	if (dev->agp == NULL || !dev->agp->acquired)
		return (EINVAL);

	entry = drm_agp_lookup_entry(dev, (void *)request->handle);
	if (entry == NULL || !entry->bound)
		return (EINVAL);

	DRM_UNLOCK();
	retcode = drm_agp_unbind_memory(entry->handle);
	DRM_LOCK();

	if (retcode == 0)
		entry->bound = 0;

	return (retcode);
}

int
drm_agp_unbind_ioctl(struct drm_device *dev, void *data,
    struct drm_file *file_priv)
{
	struct drm_agp_binding	*request = data;
	int			 retcode;

	DRM_LOCK();
	retcode = drm_agp_unbind(dev, request);
	DRM_UNLOCK();

	return (retcode);
}

int
drm_agp_bind(struct drm_device *dev, struct drm_agp_binding *request)
{
	struct drm_agp_mem	*entry;
	int			 retcode, page;
	
	if (dev->agp == NULL || !dev->agp->acquired)
		return (EINVAL);

	DRM_DEBUG("agp_bind, page_size=%x\n", PAGE_SIZE);

	entry = drm_agp_lookup_entry(dev, (void *)request->handle);
	if (entry == NULL || entry->bound)
		return (EINVAL);

	page = (request->offset + PAGE_SIZE - 1) / PAGE_SIZE;

	DRM_UNLOCK();
	retcode = drm_agp_bind_memory(entry->handle, page);
	DRM_LOCK();
	if (retcode == 0)
		entry->bound = dev->agp->base + (page << PAGE_SHIFT);

	return (retcode);
}

int
drm_agp_bind_ioctl(struct drm_device *dev, void *data,
    struct drm_file *file_priv)
{
	struct drm_agp_binding	*request = data;
	int			 retcode;

	DRM_LOCK();
	retcode = drm_agp_bind(dev, request);
	DRM_UNLOCK();

	return (retcode);
}

int
drm_agp_free(struct drm_device *dev, struct drm_agp_buffer *request)
{
	struct drm_agp_mem	*entry;
	
	if (dev->agp == NULL || !dev->agp->acquired)
		return (EINVAL);

	entry = drm_agp_lookup_entry(dev, (void*)request->handle);
	if (entry == NULL)
		return (EINVAL);
   
	TAILQ_REMOVE(&dev->agp->memory, entry, link);

	DRM_UNLOCK();
	if (entry->bound)
		drm_agp_unbind_memory(entry->handle);
	drm_agp_free_memory(entry->handle);
	DRM_LOCK();

	drm_free(entry, sizeof(*entry), DRM_MEM_AGPLISTS);

	return (0);

}

int
drm_agp_free_ioctl(struct drm_device *dev, void *data,
    struct drm_file *file_priv)
{
	struct drm_agp_buffer	*request = data;
	int			 retcode;

	DRM_LOCK();
	retcode = drm_agp_free(dev, request);
	DRM_UNLOCK();

	return (retcode);
}

drm_agp_head_t *
drm_agp_init(void)
{
#ifndef DRM_NO_AGP
	struct device	*agpdev;
	drm_agp_head_t	*head = NULL;
	int		 agp_available = 1;
   
	agpdev = DRM_AGP_FIND_DEVICE();
	if (agpdev == NULL)
		agp_available = 0;

	DRM_DEBUG("agp_available = %d\n", agp_available);

	if (agp_available) {
		head = drm_calloc(1, sizeof(*head), DRM_MEM_AGPLISTS);
		if (head == NULL)
			return (NULL);
		head->agpdev = agpdev;
		agp_get_info(agpdev, &head->info);
		head->base = head->info.ai_aperture_base;
		TAILQ_INIT(&head->memory);
	}
	return (head);
#else
	return (NULL);
#endif
}

void *
drm_agp_allocate_memory(size_t pages, u32 type)
{
	struct device	*agpdev;

	agpdev = DRM_AGP_FIND_DEVICE();
	if (agpdev == NULL)
		return (NULL);

	return (agp_alloc_memory(agpdev, type, pages << AGP_PAGE_SHIFT));
}

int
drm_agp_free_memory(void *handle)
{
	struct device	*agpdev;

	agpdev = DRM_AGP_FIND_DEVICE();
	if (agpdev == NULL || handle == NULL)
		return (0);

	agp_free_memory(agpdev, handle);
	return (1);
}

int
drm_agp_bind_memory(void *handle, off_t start)
{
#ifndef DRM_NO_AGP
	struct device	*agpdev;

	agpdev = DRM_AGP_FIND_DEVICE();
	if (agpdev == NULL || handle == NULL)
		return (EINVAL);

	return (agp_bind_memory(agpdev, handle, start * PAGE_SIZE));
#else
	return (0);
#endif
}

int
drm_agp_unbind_memory(void *handle)
{
#ifndef DRM_NO_AGP
	struct device	*agpdev;

	agpdev = DRM_AGP_FIND_DEVICE();
	if (agpdev == NULL || handle == NULL)
		return (EINVAL);

	return (agp_unbind_memory(agpdev, handle));
#else
	return (0);
#endif
}