/*
 * Copyright 1998-2003 VIA Technologies, Inc. All Rights Reserved.
 * Copyright 2001-2003 S3 Graphics, Inc. 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, sub license,
 * 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 NON-INFRINGEMENT. IN NO EVENT SHALL
 * VIA, S3 GRAPHICS, 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.
 */

#include <linux/config.h>
#include "savage.h"
#include "drmP.h"
#include "savage_drv.h"

#define DRIVER_AUTHOR           "John Zhao, S3 Graphics Inc."

#define DRIVER_NAME             "savage"
#define DRIVER_DESC             "Savage4 Family"
#define DRIVER_DATE             "20011023"

#define DRIVER_MAJOR            1
#define DRIVER_MINOR            0
#define DRIVER_PATCHLEVEL       0

/* Currently Savage4 not implement DMA */
/* mark off by Jiayo Hsu, Oct. 23, 2001*/


#define DRIVER_IOCTLS \
        [DRM_IOCTL_NR(DRM_IOCTL_SAVAGE_ALLOC_CONTINUOUS_MEM)] \
        = {savage_alloc_continuous_mem,1,0},\
        [DRM_IOCTL_NR( DRM_IOCTL_SAVAGE_GET_PHYSICS_ADDRESS)] \
        = {savage_get_physics_address,1,0},\
        [DRM_IOCTL_NR(DRM_IOCTL_SAVAGE_FREE_CONTINUOUS_MEM)]  \
        = {savage_free_cont_mem,1,0}

int savage_alloc_continuous_mem(struct inode *inode, struct file *filp,
                                unsigned int cmd, unsigned long arg)
{
        drm_savage_alloc_cont_mem_t cont_mem;
        unsigned long size, addr;
        void *ret;
        int i;
        mem_map_t *p;
        pgprot_t flags;

        /* add to list */
        drm_file_t *priv = filp->private_data;
        drm_device_t *dev = priv->dev;
        drm_map_t *map;
        drm_map_list_t *list;
        
        dma_addr_t pa;

        if (copy_from_user(&cont_mem, (drm_savage_alloc_cont_mem_t *) arg, sizeof(cont_mem)))
                return -EFAULT;

        /*check the parameters */
        if (cont_mem.size <= 0)
                return -EINVAL;
        if( 0xFFFFFFFFUL / cont_mem.size < cont_mem.type )
                return -EINVAL;
                
        map = DRM(alloc) (sizeof(*map), DRM_MEM_MAPS);
        if (!map)
                return -ENOMEM;

        size = cont_mem.type * cont_mem.size;

        ret = pci_alloc_consistent(/*FIXME*/NULL, size, &pa);
        if (ret == NULL)
                return -ENOMEM;

        /* Set the reserverd flag so that the remap_page_range can map these page */
        for (i = 0, p = virt_to_page(ret); i < size / PAGE_SIZE; i++, p++)
                SetPageReserved(p);

        cont_mem.phyaddress = pa;
        cont_mem.location = DRM_SAVAGE_MEM_LOCATION_PCI;        /* pci only at present */

        /*Map the memory to user space */
        down_write(&current->mm->mmap_sem);
        addr = do_mmap(NULL, 0, size, PROT_READ | PROT_WRITE, MAP_PRIVATE, cont_mem.phyaddress);
        if ((unsigned long)addr > -1024UL)
        {
                up_write(&current->mm->mmap_sem);
                return -EINVAL;
        }
        pgprot_val(flags) = _PAGE_PRESENT | _PAGE_RW | _PAGE_USER;
        if (remap_page_range(addr, cont_mem.phyaddress, size, flags))
        {
                up_write(&current->mm->mmap_sem);
                return -EINVAL;
        }
        up_write(&current->mm->mmap_sem);

        for (i = 0, p = virt_to_page(ret); i < size / PAGE_SIZE; i++, p++)
                ClearPageReserved(p);

        cont_mem.linear = addr;

        /*map list */
        map->handle = ret;      /* to distinguish with other */
        map->offset = cont_mem.phyaddress;
        map->size = size;
        map->mtrr = -1;
        /*map-flags,type?? */

        list = DRM(alloc) (sizeof(*list), DRM_MEM_MAPS);
        if (!list) {
                DRM(free) (map, sizeof(*map), DRM_MEM_MAPS);
                return -EINVAL;
        }
        memset(list, 0, sizeof(*list));
        list->map = map;

        down(&dev->struct_sem);
        list_add(&list->head, &dev->maplist->head);
        up(&dev->struct_sem);

        if (copy_to_user((drm_savage_alloc_cont_mem_t *) arg, &cont_mem, sizeof(cont_mem)))
                return -EFAULT;

#warning "Race at the very least"
        for (i = 0, p = virt_to_page(ret); i < size / PAGE_SIZE; i++, p++)
                atomic_set(&p->count, 1);

        return 1;               /*success */
}

int savage_get_physics_address(struct inode *inode, struct file *filp,
                               unsigned int cmd, unsigned long arg)
{

        drm_savage_get_physcis_address_t req;
        unsigned long buf;
        pgd_t *pgd;
        pmd_t *pmd;
        pte_t *pte;
        struct mm_struct *mm;

        if (copy_from_user(&req, (drm_savage_get_physcis_address_t *) arg, sizeof(req)))
                return -EFAULT;
        buf = req.v_address;

#warning "FIXME: need to redo logic for this"
        /*What kind of virtual address ? */
        if (buf >= (unsigned long) high_memory)
                mm = &init_mm;
        else
                mm = current->mm;

        spin_lock(&mm->page_table_lock);

        pgd = pgd_offset(mm, buf);
        pmd = pmd_offset(pgd, buf);
        pte = pte_offset(pmd, buf);

        if (!pte_present(*pte))
        {
                spin_unlock(&mm->page_table_lock);
                return -EINVAL;
        }
        req.p_address = ((pte_val(*pte) & PAGE_MASK) | (buf & (PAGE_SIZE - 1)));
        spin_unlock(&mm->page_table_lock);

        if (copy_to_user((drm_savage_get_physcis_address_t *) arg, &req, sizeof(req)))
                return -EFAULT;
        return 1;
}

/*free the continuous memory*/
int savage_free_cont_mem(struct inode *inode, struct file *filp,
                         unsigned int cmd, unsigned long arg)
{
        drm_savage_alloc_cont_mem_t cont_mem;
        unsigned long size;

        /*map  list */
        drm_file_t *priv = filp->private_data;
        drm_device_t *dev = priv->dev;
        drm_map_t *map;
        struct list_head *list;
        drm_map_list_t *r_list = NULL;

        if (copy_from_user(&cont_mem, (drm_savage_alloc_cont_mem_t *) arg, sizeof(cont_mem)))
                return -EFAULT;
#warning "fix size overflow check"
        size = cont_mem.type * cont_mem.size;
        if (size <= 0)
                return -EINVAL;

        /* find the map in the list */
        list_for_each(list, &dev->maplist->head) {
                r_list = (drm_map_list_t *) list;

                if (r_list->map && r_list->map->offset == cont_mem.phyaddress)
                        break;
        }
        /*find none */
        if (list == (&dev->maplist->head)) {
                up(&dev->struct_sem);
                return -EINVAL;
        }
        map = r_list->map;
        list_del(list);
        DRM(free) (list, sizeof(*list), DRM_MEM_MAPS);

        /*unmap the user space */
        if (do_munmap(current->mm, cont_mem.linear, size) != 0)
                return -EFAULT;
        /*free the page */
        pci_free_consistent(NULL, size, map->handle, cont_mem.phyaddress);

        return 1;
}


#include "drm_agpsupport.h"
#include "drm_auth.h"
#include "drm_bufs.h"
#include "drm_context.h"
#include "drm_dma.h"
#include "drm_drawable.h"
#include "drm_drv.h"
#include "drm_fops.h"
#include "drm_init.h"
#include "drm_ioctl.h"
#include "drm_lock.h"
#include "drm_memory.h"
#include "drm_proc.h"
#include "drm_vm.h"
#include "drm_stub.h"