drivers/char/agp/sworks-agp.c - linux - Git at Google

 /*
  * Serverworks AGPGART routines.
  */

 #include <linux/module.h>
 #include <linux/pci.h>
 #include <linux/init.h>
 #include <linux/string.h>
 #include <linux/slab.h>
 #include <linux/jiffies.h>
 #include <linux/agp_backend.h>
 #include <asm/set_memory.h>
 #include "agp.h"

 #define SVWRKS_COMMAND		0x04
 #define SVWRKS_APSIZE		0x10
 #define SVWRKS_MMBASE		0x14
 #define SVWRKS_CACHING		0x4b
 #define SVWRKS_AGP_ENABLE	0x60
 #define SVWRKS_FEATURE		0x68

 #define SVWRKS_SIZE_MASK	0xfe000000

 /* Memory mapped registers */
 #define SVWRKS_GART_CACHE	0x02
 #define SVWRKS_GATTBASE		0x04
 #define SVWRKS_TLBFLUSH		0x10
 #define SVWRKS_POSTFLUSH	0x14
 #define SVWRKS_DIRFLUSH		0x0c


 struct serverworks_page_map {
 	unsigned long *real;
 	unsigned long __iomem *remapped;
 };

 static struct _serverworks_private {
 	struct pci_dev *svrwrks_dev;	/* device one */
 	volatile u8 __iomem *registers;
 	struct serverworks_page_map **gatt_pages;
 	int num_tables;
 	struct serverworks_page_map scratch_dir;

 	int gart_addr_ofs;
 	int mm_addr_ofs;
 } serverworks_private;

 static int serverworks_create_page_map(struct serverworks_page_map *page_map)
 {
 	int i;

 	page_map->real = (unsigned long *) __get_free_page(GFP_KERNEL);
 	if (page_map->real == NULL) {
 		return -ENOMEM;
 	}

 	set_memory_uc((unsigned long)page_map->real, 1);
 	page_map->remapped = page_map->real;

 	for (i = 0; i < PAGE_SIZE / sizeof(unsigned long); i++)
 		writel(agp_bridge->scratch_page, page_map->remapped+i);
 		/* Red Pen: Everyone else does pci posting flush here */

 	return 0;
 }

 static void serverworks_free_page_map(struct serverworks_page_map *page_map)
 {
 	set_memory_wb((unsigned long)page_map->real, 1);
 	free_page((unsigned long) page_map->real);
 }

 static void serverworks_free_gatt_pages(void)
 {
 	int i;
 	struct serverworks_page_map **tables;
 	struct serverworks_page_map *entry;

 	tables = serverworks_private.gatt_pages;
 	for (i = 0; i < serverworks_private.num_tables; i++) {
 		entry = tables[i];
 		if (entry != NULL) {
 			if (entry->real != NULL) {
 				serverworks_free_page_map(entry);
 			}
 			kfree(entry);
 		}
 	}
 	kfree(tables);
 }

 static int serverworks_create_gatt_pages(int nr_tables)
 {
 	struct serverworks_page_map **tables;
 	struct serverworks_page_map *entry;
 	int retval = 0;
 	int i;

 	tables = kcalloc(nr_tables + 1, sizeof(struct serverworks_page_map *),
 			 GFP_KERNEL);
 	if (tables == NULL)
 		return -ENOMEM;

 	for (i = 0; i < nr_tables; i++) {
 		entry = kzalloc(sizeof(struct serverworks_page_map), GFP_KERNEL);
 		if (entry == NULL) {
 			retval = -ENOMEM;
 			break;
 		}
 		tables[i] = entry;
 		retval = serverworks_create_page_map(entry);
 		if (retval != 0) break;
 	}
 	serverworks_private.num_tables = nr_tables;
 	serverworks_private.gatt_pages = tables;

 	if (retval != 0) serverworks_free_gatt_pages();

 	return retval;
 }

 #define SVRWRKS_GET_GATT(addr) (serverworks_private.gatt_pages[\
 	GET_PAGE_DIR_IDX(addr)]->remapped)

 #ifndef GET_PAGE_DIR_OFF
 #define GET_PAGE_DIR_OFF(addr) (addr >> 22)
 #endif

 #ifndef GET_PAGE_DIR_IDX
 #define GET_PAGE_DIR_IDX(addr) (GET_PAGE_DIR_OFF(addr) - \
 	GET_PAGE_DIR_OFF(agp_bridge->gart_bus_addr))
 #endif

 #ifndef GET_GATT_OFF
 #define GET_GATT_OFF(addr) ((addr & 0x003ff000) >> 12)
 #endif

 static int serverworks_create_gatt_table(struct agp_bridge_data *bridge)
 {
 	struct aper_size_info_lvl2 *value;
 	struct serverworks_page_map page_dir;
 	int retval;
 	u32 temp;
 	int i;

 	value = A_SIZE_LVL2(agp_bridge->current_size);
 	retval = serverworks_create_page_map(&page_dir);
 	if (retval != 0) {
 		return retval;
 	}
 	retval = serverworks_create_page_map(&serverworks_private.scratch_dir);
 	if (retval != 0) {
 		serverworks_free_page_map(&page_dir);
 		return retval;
 	}
 	/* Create a fake scratch directory */
 	for (i = 0; i < 1024; i++) {
 		writel(agp_bridge->scratch_page, serverworks_private.scratch_dir.remapped+i);
 		writel(virt_to_phys(serverworks_private.scratch_dir.real) | 1, page_dir.remapped+i);
 	}

 	retval = serverworks_create_gatt_pages(value->num_entries / 1024);
 	if (retval != 0) {
 		serverworks_free_page_map(&page_dir);
 		serverworks_free_page_map(&serverworks_private.scratch_dir);
 		return retval;
 	}

 	agp_bridge->gatt_table_real = (u32 *)page_dir.real;
 	agp_bridge->gatt_table = (u32 __iomem *)page_dir.remapped;
 	agp_bridge->gatt_bus_addr = virt_to_phys(page_dir.real);

 	/* Get the address for the gart region.
 	 * This is a bus address even on the alpha, b/c its
 	 * used to program the agp master not the cpu
 	 */

 	pci_read_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,&temp);
 	agp_bridge->gart_bus_addr = (temp & PCI_BASE_ADDRESS_MEM_MASK);

 	/* Calculate the agp offset */
 	for (i = 0; i < value->num_entries / 1024; i++)
 		writel(virt_to_phys(serverworks_private.gatt_pages[i]->real)|1, page_dir.remapped+i);

 	return 0;
 }

 static int serverworks_free_gatt_table(struct agp_bridge_data *bridge)
 {
 	struct serverworks_page_map page_dir;

 	page_dir.real = (unsigned long *)agp_bridge->gatt_table_real;
 	page_dir.remapped = (unsigned long __iomem *)agp_bridge->gatt_table;

 	serverworks_free_gatt_pages();
 	serverworks_free_page_map(&page_dir);
 	serverworks_free_page_map(&serverworks_private.scratch_dir);
 	return 0;
 }

 static int serverworks_fetch_size(void)
 {
 	int i;
 	u32 temp;
 	u32 temp2;
 	struct aper_size_info_lvl2 *values;

 	values = A_SIZE_LVL2(agp_bridge->driver->aperture_sizes);
 	pci_read_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,&temp);
 	pci_write_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,
 					SVWRKS_SIZE_MASK);
 	pci_read_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,&temp2);
 	pci_write_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,temp);
 	temp2 &= SVWRKS_SIZE_MASK;

 	for (i = 0; i < agp_bridge->driver->num_aperture_sizes; i++) {
 		if (temp2 == values[i].size_value) {
 			agp_bridge->previous_size =
 			    agp_bridge->current_size = (void *) (values + i);

 			agp_bridge->aperture_size_idx = i;
 			return values[i].size;
 		}
 	}

 	return 0;
 }

 /*
  * This routine could be implemented by taking the addresses
  * written to the GATT, and flushing them individually.  However
  * currently it just flushes the whole table.  Which is probably
  * more efficient, since agp_memory blocks can be a large number of
  * entries.
  */
 static void serverworks_tlbflush(struct agp_memory *temp)
 {
 	unsigned long timeout;

 	writeb(1, serverworks_private.registers+SVWRKS_POSTFLUSH);
 	timeout = jiffies + 3*HZ;
 	while (readb(serverworks_private.registers+SVWRKS_POSTFLUSH) == 1) {
 		cpu_relax();
 		if (time_after(jiffies, timeout)) {
 			dev_err(&serverworks_private.svrwrks_dev->dev,
 				"TLB post flush took more than 3 seconds\n");
 			break;
 		}
 	}

 	writel(1, serverworks_private.registers+SVWRKS_DIRFLUSH);
 	timeout = jiffies + 3*HZ;
 	while (readl(serverworks_private.registers+SVWRKS_DIRFLUSH) == 1) {
 		cpu_relax();
 		if (time_after(jiffies, timeout)) {
 			dev_err(&serverworks_private.svrwrks_dev->dev,
 				"TLB Dir flush took more than 3 seconds\n");
 			break;
 		}
 	}
 }

 static int serverworks_configure(void)
 {
 	struct aper_size_info_lvl2 *current_size;
 	u32 temp;
 	u8 enable_reg;
 	u16 cap_reg;

 	current_size = A_SIZE_LVL2(agp_bridge->current_size);

 	/* Get the memory mapped registers */
 	pci_read_config_dword(agp_bridge->dev, serverworks_private.mm_addr_ofs, &temp);
 	temp = (temp & PCI_BASE_ADDRESS_MEM_MASK);
 	serverworks_private.registers = (volatile u8 __iomem *) ioremap(temp, 4096);
 	if (!serverworks_private.registers) {
 		dev_err(&agp_bridge->dev->dev, "can't ioremap(%#x)\n", temp);
 		return -ENOMEM;
 	}

 	writeb(0xA, serverworks_private.registers+SVWRKS_GART_CACHE);
 	readb(serverworks_private.registers+SVWRKS_GART_CACHE);	/* PCI Posting. */

 	writel(agp_bridge->gatt_bus_addr, serverworks_private.registers+SVWRKS_GATTBASE);
 	readl(serverworks_private.registers+SVWRKS_GATTBASE);	/* PCI Posting. */

 	cap_reg = readw(serverworks_private.registers+SVWRKS_COMMAND);
 	cap_reg &= ~0x0007;
 	cap_reg |= 0x4;
 	writew(cap_reg, serverworks_private.registers+SVWRKS_COMMAND);
 	readw(serverworks_private.registers+SVWRKS_COMMAND);

 	pci_read_config_byte(serverworks_private.svrwrks_dev,SVWRKS_AGP_ENABLE, &enable_reg);
 	enable_reg |= 0x1; /* Agp Enable bit */
 	pci_write_config_byte(serverworks_private.svrwrks_dev,SVWRKS_AGP_ENABLE, enable_reg);
 	serverworks_tlbflush(NULL);

 	agp_bridge->capndx = pci_find_capability(serverworks_private.svrwrks_dev, PCI_CAP_ID_AGP);

 	/* Fill in the mode register */
 	pci_read_config_dword(serverworks_private.svrwrks_dev,
 			      agp_bridge->capndx+PCI_AGP_STATUS, &agp_bridge->mode);

 	pci_read_config_byte(agp_bridge->dev, SVWRKS_CACHING, &enable_reg);
 	enable_reg &= ~0x3;
 	pci_write_config_byte(agp_bridge->dev, SVWRKS_CACHING, enable_reg);

 	pci_read_config_byte(agp_bridge->dev, SVWRKS_FEATURE, &enable_reg);
 	enable_reg |= (1<<6);
 	pci_write_config_byte(agp_bridge->dev,SVWRKS_FEATURE, enable_reg);

 	return 0;
 }

 static void serverworks_cleanup(void)
 {
 	iounmap((void __iomem *) serverworks_private.registers);
 }

 static int serverworks_insert_memory(struct agp_memory *mem,
 			     off_t pg_start, int type)
 {
 	int i, j, num_entries;
 	unsigned long __iomem *cur_gatt;
 	unsigned long addr;

 	num_entries = A_SIZE_LVL2(agp_bridge->current_size)->num_entries;

 	if (type != 0 || mem->type != 0) {
 		return -EINVAL;
 	}
 	if ((pg_start + mem->page_count) > num_entries) {
 		return -EINVAL;
 	}

 	j = pg_start;
 	while (j < (pg_start + mem->page_count)) {
 		addr = (j * PAGE_SIZE) + agp_bridge->gart_bus_addr;
 		cur_gatt = SVRWRKS_GET_GATT(addr);
 		if (!PGE_EMPTY(agp_bridge, readl(cur_gatt+GET_GATT_OFF(addr))))
 			return -EBUSY;
 		j++;
 	}

 	if (!mem->is_flushed) {
 		global_cache_flush();
 		mem->is_flushed = true;
 	}

 	for (i = 0, j = pg_start; i < mem->page_count; i++, j++) {
 		addr = (j * PAGE_SIZE) + agp_bridge->gart_bus_addr;
 		cur_gatt = SVRWRKS_GET_GATT(addr);
 		writel(agp_bridge->driver->mask_memory(agp_bridge,
 				page_to_phys(mem->pages[i]), mem->type),
 		       cur_gatt+GET_GATT_OFF(addr));
 	}
 	serverworks_tlbflush(mem);
 	return 0;
 }

 static int serverworks_remove_memory(struct agp_memory *mem, off_t pg_start,
 			     int type)
 {
 	int i;
 	unsigned long __iomem *cur_gatt;
 	unsigned long addr;

 	if (type != 0 || mem->type != 0) {
 		return -EINVAL;
 	}

 	global_cache_flush();
 	serverworks_tlbflush(mem);

 	for (i = pg_start; i < (mem->page_count + pg_start); i++) {
 		addr = (i * PAGE_SIZE) + agp_bridge->gart_bus_addr;
 		cur_gatt = SVRWRKS_GET_GATT(addr);
 		writel(agp_bridge->scratch_page, cur_gatt+GET_GATT_OFF(addr));
 	}

 	serverworks_tlbflush(mem);
 	return 0;
 }

 static const struct gatt_mask serverworks_masks[] =
 {
 	{.mask = 1, .type = 0}
 };

 static const struct aper_size_info_lvl2 serverworks_sizes[7] =
 {
 	{2048, 524288, 0x80000000},
 	{1024, 262144, 0xc0000000},
 	{512, 131072, 0xe0000000},
 	{256, 65536, 0xf0000000},
 	{128, 32768, 0xf8000000},
 	{64, 16384, 0xfc000000},
 	{32, 8192, 0xfe000000}
 };

 static void serverworks_agp_enable(struct agp_bridge_data *bridge, u32 mode)
 {
 	u32 command;

 	pci_read_config_dword(serverworks_private.svrwrks_dev,
 			      bridge->capndx + PCI_AGP_STATUS,
 			      &command);

 	command = agp_collect_device_status(bridge, mode, command);

 	command &= ~0x10;	/* disable FW */
 	command &= ~0x08;

 	command |= 0x100;

 	pci_write_config_dword(serverworks_private.svrwrks_dev,
 			       bridge->capndx + PCI_AGP_COMMAND,
 			       command);

 	agp_device_command(command, false);
 }

 static const struct agp_bridge_driver sworks_driver = {
 	.owner			= THIS_MODULE,
 	.aperture_sizes		= serverworks_sizes,
 	.size_type		= LVL2_APER_SIZE,
 	.num_aperture_sizes	= 7,
 	.configure		= serverworks_configure,
 	.fetch_size		= serverworks_fetch_size,
 	.cleanup		= serverworks_cleanup,
 	.tlb_flush		= serverworks_tlbflush,
 	.mask_memory		= agp_generic_mask_memory,
 	.masks			= serverworks_masks,
 	.agp_enable		= serverworks_agp_enable,
 	.cache_flush		= global_cache_flush,
 	.create_gatt_table	= serverworks_create_gatt_table,
 	.free_gatt_table	= serverworks_free_gatt_table,
 	.insert_memory		= serverworks_insert_memory,
 	.remove_memory		= serverworks_remove_memory,
 	.alloc_by_type		= agp_generic_alloc_by_type,
 	.free_by_type		= agp_generic_free_by_type,
 	.agp_alloc_page		= agp_generic_alloc_page,
 	.agp_alloc_pages	= agp_generic_alloc_pages,
 	.agp_destroy_page	= agp_generic_destroy_page,
 	.agp_destroy_pages	= agp_generic_destroy_pages,
 	.agp_type_to_mask_type  = agp_generic_type_to_mask_type,
 };

 static int agp_serverworks_probe(struct pci_dev *pdev,
 				 const struct pci_device_id *ent)
 {
 	struct agp_bridge_data *bridge;
 	struct pci_dev *bridge_dev;
 	u32 temp, temp2;
 	u8 cap_ptr = 0;

 	cap_ptr = pci_find_capability(pdev, PCI_CAP_ID_AGP);

 	switch (pdev->device) {
 	case 0x0006:
 		dev_err(&pdev->dev, "ServerWorks CNB20HE is unsupported due to lack of documentation\n");
 		return -ENODEV;

 	case PCI_DEVICE_ID_SERVERWORKS_HE:
 	case PCI_DEVICE_ID_SERVERWORKS_LE:
 	case 0x0007:
 		break;

 	default:
 		if (cap_ptr)
 			dev_err(&pdev->dev, "unsupported Serverworks chipset "
 				"[%04x/%04x]\n", pdev->vendor, pdev->device);
 		return -ENODEV;
 	}

 	/* Everything is on func 1 here so we are hardcoding function one */
 	bridge_dev = pci_get_domain_bus_and_slot(pci_domain_nr(pdev->bus),
 			(unsigned int)pdev->bus->number,
 			PCI_DEVFN(0, 1));
 	if (!bridge_dev) {
 		dev_info(&pdev->dev, "can't find secondary device\n");
 		return -ENODEV;
 	}

 	serverworks_private.svrwrks_dev = bridge_dev;
 	serverworks_private.gart_addr_ofs = 0x10;

 	pci_read_config_dword(pdev, SVWRKS_APSIZE, &temp);
 	if (temp & PCI_BASE_ADDRESS_MEM_TYPE_64) {
 		pci_read_config_dword(pdev, SVWRKS_APSIZE + 4, &temp2);
 		if (temp2 != 0) {
 			dev_info(&pdev->dev, "64 bit aperture address, "
 				 "but top bits are not zero; disabling AGP\n");
 			return -ENODEV;
 		}
 		serverworks_private.mm_addr_ofs = 0x18;
 	} else
 		serverworks_private.mm_addr_ofs = 0x14;

 	pci_read_config_dword(pdev, serverworks_private.mm_addr_ofs, &temp);
 	if (temp & PCI_BASE_ADDRESS_MEM_TYPE_64) {
 		pci_read_config_dword(pdev,
 				serverworks_private.mm_addr_ofs + 4, &temp2);
 		if (temp2 != 0) {
 			dev_info(&pdev->dev, "64 bit MMIO address, but top "
 				 "bits are not zero; disabling AGP\n");
 			return -ENODEV;
 		}
 	}

 	bridge = agp_alloc_bridge();
 	if (!bridge)
 		return -ENOMEM;

 	bridge->driver = &sworks_driver;
 	bridge->dev_private_data = &serverworks_private;
 	bridge->dev = pci_dev_get(pdev);

 	pci_set_drvdata(pdev, bridge);
 	return agp_add_bridge(bridge);
 }

 static void agp_serverworks_remove(struct pci_dev *pdev)
 {
 	struct agp_bridge_data *bridge = pci_get_drvdata(pdev);

 	pci_dev_put(bridge->dev);
 	agp_remove_bridge(bridge);
 	agp_put_bridge(bridge);
 	pci_dev_put(serverworks_private.svrwrks_dev);
 	serverworks_private.svrwrks_dev = NULL;
 }

 static struct pci_device_id agp_serverworks_pci_table[] = {
 	{
 	.class		= (PCI_CLASS_BRIDGE_HOST << 8),
 	.class_mask	= ~0,
 	.vendor		= PCI_VENDOR_ID_SERVERWORKS,
 	.device		= PCI_ANY_ID,
 	.subvendor	= PCI_ANY_ID,
 	.subdevice	= PCI_ANY_ID,
 	},
 	{ }
 };

 MODULE_DEVICE_TABLE(pci, agp_serverworks_pci_table);

 static struct pci_driver agp_serverworks_pci_driver = {
 	.name		= "agpgart-serverworks",
 	.id_table	= agp_serverworks_pci_table,
 	.probe		= agp_serverworks_probe,
 	.remove		= agp_serverworks_remove,
 };

 static int __init agp_serverworks_init(void)
 {
 	if (agp_off)
 		return -EINVAL;
 	return pci_register_driver(&agp_serverworks_pci_driver);
 }

 static void __exit agp_serverworks_cleanup(void)
 {
 	pci_unregister_driver(&agp_serverworks_pci_driver);
 }

 module_init(agp_serverworks_init);
 module_exit(agp_serverworks_cleanup);

 MODULE_LICENSE("GPL and additional rights");
	/*
	* Serverworks AGPGART routines.
	*/

	#include <linux/module.h>
	#include <linux/pci.h>
	#include <linux/init.h>
	#include <linux/string.h>
	#include <linux/slab.h>
	#include <linux/jiffies.h>
	#include <linux/agp_backend.h>
	#include <asm/set_memory.h>
	#include "agp.h"

	#define SVWRKS_COMMAND 0x04
	#define SVWRKS_APSIZE 0x10
	#define SVWRKS_MMBASE 0x14
	#define SVWRKS_CACHING 0x4b
	#define SVWRKS_AGP_ENABLE 0x60
	#define SVWRKS_FEATURE 0x68

	#define SVWRKS_SIZE_MASK 0xfe000000

	/* Memory mapped registers */
	#define SVWRKS_GART_CACHE 0x02
	#define SVWRKS_GATTBASE 0x04
	#define SVWRKS_TLBFLUSH 0x10
	#define SVWRKS_POSTFLUSH 0x14
	#define SVWRKS_DIRFLUSH 0x0c


	struct serverworks_page_map {
	unsigned long *real;
	unsigned long __iomem *remapped;
	};

	static struct _serverworks_private {
	struct pci_dev svrwrks_dev; / device one */
	volatile u8 __iomem *registers;
	struct serverworks_page_map **gatt_pages;
	int num_tables;
	struct serverworks_page_map scratch_dir;

	int gart_addr_ofs;
	int mm_addr_ofs;
	} serverworks_private;

	static int serverworks_create_page_map(struct serverworks_page_map *page_map)
	{
	int i;

	page_map->real = (unsigned long *) __get_free_page(GFP_KERNEL);
	if (page_map->real == NULL) {
	return -ENOMEM;
	}

	set_memory_uc((unsigned long)page_map->real, 1);
	page_map->remapped = page_map->real;

	for (i = 0; i < PAGE_SIZE / sizeof(unsigned long); i++)
	writel(agp_bridge->scratch_page, page_map->remapped+i);
	/* Red Pen: Everyone else does pci posting flush here */

	return 0;
	}

	static void serverworks_free_page_map(struct serverworks_page_map *page_map)
	{
	set_memory_wb((unsigned long)page_map->real, 1);
	free_page((unsigned long) page_map->real);
	}

	static void serverworks_free_gatt_pages(void)
	{
	int i;
	struct serverworks_page_map **tables;
	struct serverworks_page_map *entry;

	tables = serverworks_private.gatt_pages;
	for (i = 0; i < serverworks_private.num_tables; i++) {
	entry = tables[i];
	if (entry != NULL) {
	if (entry->real != NULL) {
	serverworks_free_page_map(entry);
	}
	kfree(entry);
	}
	}
	kfree(tables);
	}

	static int serverworks_create_gatt_pages(int nr_tables)
	{
	struct serverworks_page_map **tables;
	struct serverworks_page_map *entry;
	int retval = 0;
	int i;

	tables = kcalloc(nr_tables + 1, sizeof(struct serverworks_page_map *),
	GFP_KERNEL);
	if (tables == NULL)
	return -ENOMEM;

	for (i = 0; i < nr_tables; i++) {
	entry = kzalloc(sizeof(struct serverworks_page_map), GFP_KERNEL);
	if (entry == NULL) {
	retval = -ENOMEM;
	break;
	}
	tables[i] = entry;
	retval = serverworks_create_page_map(entry);
	if (retval != 0) break;
	}
	serverworks_private.num_tables = nr_tables;
	serverworks_private.gatt_pages = tables;

	if (retval != 0) serverworks_free_gatt_pages();

	return retval;
	}

	#define SVRWRKS_GET_GATT(addr) (serverworks_private.gatt_pages[\
	GET_PAGE_DIR_IDX(addr)]->remapped)

	#ifndef GET_PAGE_DIR_OFF
	#define GET_PAGE_DIR_OFF(addr) (addr >> 22)
	#endif

	#ifndef GET_PAGE_DIR_IDX
	#define GET_PAGE_DIR_IDX(addr) (GET_PAGE_DIR_OFF(addr) - \
	GET_PAGE_DIR_OFF(agp_bridge->gart_bus_addr))
	#endif

	#ifndef GET_GATT_OFF
	#define GET_GATT_OFF(addr) ((addr & 0x003ff000) >> 12)
	#endif

	static int serverworks_create_gatt_table(struct agp_bridge_data *bridge)
	{
	struct aper_size_info_lvl2 *value;
	struct serverworks_page_map page_dir;
	int retval;
	u32 temp;
	int i;

	value = A_SIZE_LVL2(agp_bridge->current_size);
	retval = serverworks_create_page_map(&page_dir);
	if (retval != 0) {
	return retval;
	}
	retval = serverworks_create_page_map(&serverworks_private.scratch_dir);
	if (retval != 0) {
	serverworks_free_page_map(&page_dir);
	return retval;
	}
	/* Create a fake scratch directory */
	for (i = 0; i < 1024; i++) {
	writel(agp_bridge->scratch_page, serverworks_private.scratch_dir.remapped+i);
	writel(virt_to_phys(serverworks_private.scratch_dir.real) \| 1, page_dir.remapped+i);
	}

	retval = serverworks_create_gatt_pages(value->num_entries / 1024);
	if (retval != 0) {
	serverworks_free_page_map(&page_dir);
	serverworks_free_page_map(&serverworks_private.scratch_dir);
	return retval;
	}

	agp_bridge->gatt_table_real = (u32 *)page_dir.real;
	agp_bridge->gatt_table = (u32 __iomem *)page_dir.remapped;
	agp_bridge->gatt_bus_addr = virt_to_phys(page_dir.real);

	/* Get the address for the gart region.
	* This is a bus address even on the alpha, b/c its
	* used to program the agp master not the cpu
	*/

	pci_read_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,&temp);
	agp_bridge->gart_bus_addr = (temp & PCI_BASE_ADDRESS_MEM_MASK);

	/* Calculate the agp offset */
	for (i = 0; i < value->num_entries / 1024; i++)
	writel(virt_to_phys(serverworks_private.gatt_pages[i]->real)\|1, page_dir.remapped+i);

	return 0;
	}

	static int serverworks_free_gatt_table(struct agp_bridge_data *bridge)
	{
	struct serverworks_page_map page_dir;

	page_dir.real = (unsigned long *)agp_bridge->gatt_table_real;
	page_dir.remapped = (unsigned long __iomem *)agp_bridge->gatt_table;

	serverworks_free_gatt_pages();
	serverworks_free_page_map(&page_dir);
	serverworks_free_page_map(&serverworks_private.scratch_dir);
	return 0;
	}

	static int serverworks_fetch_size(void)
	{
	int i;
	u32 temp;
	u32 temp2;
	struct aper_size_info_lvl2 *values;

	values = A_SIZE_LVL2(agp_bridge->driver->aperture_sizes);
	pci_read_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,&temp);
	pci_write_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,
	SVWRKS_SIZE_MASK);
	pci_read_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,&temp2);
	pci_write_config_dword(agp_bridge->dev,serverworks_private.gart_addr_ofs,temp);
	temp2 &= SVWRKS_SIZE_MASK;

	for (i = 0; i < agp_bridge->driver->num_aperture_sizes; i++) {
	if (temp2 == values[i].size_value) {
	agp_bridge->previous_size =
	agp_bridge->current_size = (void *) (values + i);

	agp_bridge->aperture_size_idx = i;
	return values[i].size;
	}
	}

	return 0;
	}

	/*
	* This routine could be implemented by taking the addresses
	* written to the GATT, and flushing them individually. However
	* currently it just flushes the whole table. Which is probably
	* more efficient, since agp_memory blocks can be a large number of
	* entries.
	*/
	static void serverworks_tlbflush(struct agp_memory *temp)
	{
	unsigned long timeout;

	writeb(1, serverworks_private.registers+SVWRKS_POSTFLUSH);
	timeout = jiffies + 3*HZ;
	while (readb(serverworks_private.registers+SVWRKS_POSTFLUSH) == 1) {
	cpu_relax();
	if (time_after(jiffies, timeout)) {
	dev_err(&serverworks_private.svrwrks_dev->dev,
	"TLB post flush took more than 3 seconds\n");
	break;
	}
	}

	writel(1, serverworks_private.registers+SVWRKS_DIRFLUSH);
	timeout = jiffies + 3*HZ;
	while (readl(serverworks_private.registers+SVWRKS_DIRFLUSH) == 1) {
	cpu_relax();
	if (time_after(jiffies, timeout)) {
	dev_err(&serverworks_private.svrwrks_dev->dev,
	"TLB Dir flush took more than 3 seconds\n");
	break;
	}
	}
	}

	static int serverworks_configure(void)
	{
	struct aper_size_info_lvl2 *current_size;
	u32 temp;
	u8 enable_reg;
	u16 cap_reg;

	current_size = A_SIZE_LVL2(agp_bridge->current_size);

	/* Get the memory mapped registers */
	pci_read_config_dword(agp_bridge->dev, serverworks_private.mm_addr_ofs, &temp);
	temp = (temp & PCI_BASE_ADDRESS_MEM_MASK);
	serverworks_private.registers = (volatile u8 __iomem *) ioremap(temp, 4096);
	if (!serverworks_private.registers) {
	dev_err(&agp_bridge->dev->dev, "can't ioremap(%#x)\n", temp);
	return -ENOMEM;
	}

	writeb(0xA, serverworks_private.registers+SVWRKS_GART_CACHE);
	readb(serverworks_private.registers+SVWRKS_GART_CACHE); /* PCI Posting. */

	writel(agp_bridge->gatt_bus_addr, serverworks_private.registers+SVWRKS_GATTBASE);
	readl(serverworks_private.registers+SVWRKS_GATTBASE); /* PCI Posting. */

	cap_reg = readw(serverworks_private.registers+SVWRKS_COMMAND);
	cap_reg &= ~0x0007;
	cap_reg \|= 0x4;
	writew(cap_reg, serverworks_private.registers+SVWRKS_COMMAND);
	readw(serverworks_private.registers+SVWRKS_COMMAND);

	pci_read_config_byte(serverworks_private.svrwrks_dev,SVWRKS_AGP_ENABLE, &enable_reg);
	enable_reg \|= 0x1; /* Agp Enable bit */
	pci_write_config_byte(serverworks_private.svrwrks_dev,SVWRKS_AGP_ENABLE, enable_reg);
	serverworks_tlbflush(NULL);

	agp_bridge->capndx = pci_find_capability(serverworks_private.svrwrks_dev, PCI_CAP_ID_AGP);

	/* Fill in the mode register */
	pci_read_config_dword(serverworks_private.svrwrks_dev,
	agp_bridge->capndx+PCI_AGP_STATUS, &agp_bridge->mode);

	pci_read_config_byte(agp_bridge->dev, SVWRKS_CACHING, &enable_reg);
	enable_reg &= ~0x3;
	pci_write_config_byte(agp_bridge->dev, SVWRKS_CACHING, enable_reg);

	pci_read_config_byte(agp_bridge->dev, SVWRKS_FEATURE, &enable_reg);
	enable_reg \|= (1<<6);
	pci_write_config_byte(agp_bridge->dev,SVWRKS_FEATURE, enable_reg);

	return 0;
	}

	static void serverworks_cleanup(void)
	{
	iounmap((void __iomem *) serverworks_private.registers);
	}

	static int serverworks_insert_memory(struct agp_memory *mem,
	off_t pg_start, int type)
	{
	int i, j, num_entries;
	unsigned long __iomem *cur_gatt;
	unsigned long addr;

	num_entries = A_SIZE_LVL2(agp_bridge->current_size)->num_entries;

	if (type != 0 \|\| mem->type != 0) {
	return -EINVAL;
	}
	if ((pg_start + mem->page_count) > num_entries) {
	return -EINVAL;
	}

	j = pg_start;
	while (j < (pg_start + mem->page_count)) {
	addr = (j * PAGE_SIZE) + agp_bridge->gart_bus_addr;
	cur_gatt = SVRWRKS_GET_GATT(addr);
	if (!PGE_EMPTY(agp_bridge, readl(cur_gatt+GET_GATT_OFF(addr))))
	return -EBUSY;
	j++;
	}

	if (!mem->is_flushed) {
	global_cache_flush();
	mem->is_flushed = true;
	}

	for (i = 0, j = pg_start; i < mem->page_count; i++, j++) {
	addr = (j * PAGE_SIZE) + agp_bridge->gart_bus_addr;
	cur_gatt = SVRWRKS_GET_GATT(addr);
	writel(agp_bridge->driver->mask_memory(agp_bridge,
	page_to_phys(mem->pages[i]), mem->type),
	cur_gatt+GET_GATT_OFF(addr));
	}
	serverworks_tlbflush(mem);
	return 0;
	}

	static int serverworks_remove_memory(struct agp_memory *mem, off_t pg_start,
	int type)
	{
	int i;
	unsigned long __iomem *cur_gatt;
	unsigned long addr;

	if (type != 0 \|\| mem->type != 0) {
	return -EINVAL;
	}

	global_cache_flush();
	serverworks_tlbflush(mem);

	for (i = pg_start; i < (mem->page_count + pg_start); i++) {
	addr = (i * PAGE_SIZE) + agp_bridge->gart_bus_addr;
	cur_gatt = SVRWRKS_GET_GATT(addr);
	writel(agp_bridge->scratch_page, cur_gatt+GET_GATT_OFF(addr));
	}

	serverworks_tlbflush(mem);
	return 0;
	}

	static const struct gatt_mask serverworks_masks[] =
	{
	{.mask = 1, .type = 0}
	};

	static const struct aper_size_info_lvl2 serverworks_sizes[7] =
	{
	{2048, 524288, 0x80000000},
	{1024, 262144, 0xc0000000},
	{512, 131072, 0xe0000000},
	{256, 65536, 0xf0000000},
	{128, 32768, 0xf8000000},
	{64, 16384, 0xfc000000},
	{32, 8192, 0xfe000000}
	};

	static void serverworks_agp_enable(struct agp_bridge_data *bridge, u32 mode)
	{
	u32 command;

	pci_read_config_dword(serverworks_private.svrwrks_dev,
	bridge->capndx + PCI_AGP_STATUS,
	&command);

	command = agp_collect_device_status(bridge, mode, command);

	command &= ~0x10; /* disable FW */
	command &= ~0x08;

	command \|= 0x100;

	pci_write_config_dword(serverworks_private.svrwrks_dev,
	bridge->capndx + PCI_AGP_COMMAND,
	command);

	agp_device_command(command, false);
	}

	static const struct agp_bridge_driver sworks_driver = {
	.owner = THIS_MODULE,
	.aperture_sizes = serverworks_sizes,
	.size_type = LVL2_APER_SIZE,
	.num_aperture_sizes = 7,
	.configure = serverworks_configure,
	.fetch_size = serverworks_fetch_size,
	.cleanup = serverworks_cleanup,
	.tlb_flush = serverworks_tlbflush,
	.mask_memory = agp_generic_mask_memory,
	.masks = serverworks_masks,
	.agp_enable = serverworks_agp_enable,
	.cache_flush = global_cache_flush,
	.create_gatt_table = serverworks_create_gatt_table,
	.free_gatt_table = serverworks_free_gatt_table,
	.insert_memory = serverworks_insert_memory,
	.remove_memory = serverworks_remove_memory,
	.alloc_by_type = agp_generic_alloc_by_type,
	.free_by_type = agp_generic_free_by_type,
	.agp_alloc_page = agp_generic_alloc_page,
	.agp_alloc_pages = agp_generic_alloc_pages,
	.agp_destroy_page = agp_generic_destroy_page,
	.agp_destroy_pages = agp_generic_destroy_pages,
	.agp_type_to_mask_type = agp_generic_type_to_mask_type,
	};

	static int agp_serverworks_probe(struct pci_dev *pdev,
	const struct pci_device_id *ent)
	{
	struct agp_bridge_data *bridge;
	struct pci_dev *bridge_dev;
	u32 temp, temp2;
	u8 cap_ptr = 0;

	cap_ptr = pci_find_capability(pdev, PCI_CAP_ID_AGP);

	switch (pdev->device) {
	case 0x0006:
	dev_err(&pdev->dev, "ServerWorks CNB20HE is unsupported due to lack of documentation\n");
	return -ENODEV;

	case PCI_DEVICE_ID_SERVERWORKS_HE:
	case PCI_DEVICE_ID_SERVERWORKS_LE:
	case 0x0007:
	break;

	default:
	if (cap_ptr)
	dev_err(&pdev->dev, "unsupported Serverworks chipset "
	"[%04x/%04x]\n", pdev->vendor, pdev->device);
	return -ENODEV;
	}

	/* Everything is on func 1 here so we are hardcoding function one */
	bridge_dev = pci_get_domain_bus_and_slot(pci_domain_nr(pdev->bus),
	(unsigned int)pdev->bus->number,
	PCI_DEVFN(0, 1));
	if (!bridge_dev) {
	dev_info(&pdev->dev, "can't find secondary device\n");
	return -ENODEV;
	}

	serverworks_private.svrwrks_dev = bridge_dev;
	serverworks_private.gart_addr_ofs = 0x10;

	pci_read_config_dword(pdev, SVWRKS_APSIZE, &temp);
	if (temp & PCI_BASE_ADDRESS_MEM_TYPE_64) {
	pci_read_config_dword(pdev, SVWRKS_APSIZE + 4, &temp2);
	if (temp2 != 0) {
	dev_info(&pdev->dev, "64 bit aperture address, "
	"but top bits are not zero; disabling AGP\n");
	return -ENODEV;
	}
	serverworks_private.mm_addr_ofs = 0x18;
	} else
	serverworks_private.mm_addr_ofs = 0x14;

	pci_read_config_dword(pdev, serverworks_private.mm_addr_ofs, &temp);
	if (temp & PCI_BASE_ADDRESS_MEM_TYPE_64) {
	pci_read_config_dword(pdev,
	serverworks_private.mm_addr_ofs + 4, &temp2);
	if (temp2 != 0) {
	dev_info(&pdev->dev, "64 bit MMIO address, but top "
	"bits are not zero; disabling AGP\n");
	return -ENODEV;
	}
	}

	bridge = agp_alloc_bridge();
	if (!bridge)
	return -ENOMEM;

	bridge->driver = &sworks_driver;
	bridge->dev_private_data = &serverworks_private;
	bridge->dev = pci_dev_get(pdev);

	pci_set_drvdata(pdev, bridge);
	return agp_add_bridge(bridge);
	}

	static void agp_serverworks_remove(struct pci_dev *pdev)
	{
	struct agp_bridge_data *bridge = pci_get_drvdata(pdev);

	pci_dev_put(bridge->dev);
	agp_remove_bridge(bridge);
	agp_put_bridge(bridge);
	pci_dev_put(serverworks_private.svrwrks_dev);
	serverworks_private.svrwrks_dev = NULL;
	}

	static struct pci_device_id agp_serverworks_pci_table[] = {
	{
	.class = (PCI_CLASS_BRIDGE_HOST << 8),
	.class_mask = ~0,
	.vendor = PCI_VENDOR_ID_SERVERWORKS,
	.device = PCI_ANY_ID,
	.subvendor = PCI_ANY_ID,
	.subdevice = PCI_ANY_ID,
	},
	{ }
	};

	MODULE_DEVICE_TABLE(pci, agp_serverworks_pci_table);

	static struct pci_driver agp_serverworks_pci_driver = {
	.name = "agpgart-serverworks",
	.id_table = agp_serverworks_pci_table,
	.probe = agp_serverworks_probe,
	.remove = agp_serverworks_remove,
	};

	static int __init agp_serverworks_init(void)
	{
	if (agp_off)
	return -EINVAL;
	return pci_register_driver(&agp_serverworks_pci_driver);
	}

	static void __exit agp_serverworks_cleanup(void)
	{
	pci_unregister_driver(&agp_serverworks_pci_driver);
	}

	module_init(agp_serverworks_init);
	module_exit(agp_serverworks_cleanup);

	MODULE_LICENSE("GPL and additional rights");