1
2
3
4
5
6
7#include <linux/module.h>
8#include <linux/pci.h>
9#include <linux/init.h>
10#include <linux/agp_backend.h>
11#include <linux/page-flags.h>
12#include <linux/mm.h>
13#include <linux/jiffies.h>
14#include "agp.h"
15
16
17#define NVIDIA_0_APSIZE 0x80
18#define NVIDIA_1_WBC 0xf0
19#define NVIDIA_2_GARTCTRL 0xd0
20#define NVIDIA_2_APBASE 0xd8
21#define NVIDIA_2_APLIMIT 0xdc
22#define NVIDIA_2_ATTBASE(i) (0xe0 + (i) * 4)
23#define NVIDIA_3_APBASE 0x50
24#define NVIDIA_3_APLIMIT 0x54
25
26
27static struct _nvidia_private {
28 struct pci_dev *dev_1;
29 struct pci_dev *dev_2;
30 struct pci_dev *dev_3;
31 volatile u32 __iomem *aperture;
32 int num_active_entries;
33 off_t pg_offset;
34 u32 wbc_mask;
35} nvidia_private;
36
37
38static int nvidia_fetch_size(void)
39{
40 int i;
41 u8 size_value;
42 struct aper_size_info_8 *values;
43
44 pci_read_config_byte(agp_bridge->dev, NVIDIA_0_APSIZE, &size_value);
45 size_value &= 0x0f;
46 values = A_SIZE_8(agp_bridge->driver->aperture_sizes);
47
48 for (i = 0; i < agp_bridge->driver->num_aperture_sizes; i++) {
49 if (size_value == values[i].size_value) {
50 agp_bridge->previous_size =
51 agp_bridge->current_size = (void *) (values + i);
52 agp_bridge->aperture_size_idx = i;
53 return values[i].size;
54 }
55 }
56
57 return 0;
58}
59
60#define SYSCFG 0xC0010010
61#define IORR_BASE0 0xC0010016
62#define IORR_MASK0 0xC0010017
63#define AMD_K7_NUM_IORR 2
64
65static int nvidia_init_iorr(u32 base, u32 size)
66{
67 u32 base_hi, base_lo;
68 u32 mask_hi, mask_lo;
69 u32 sys_hi, sys_lo;
70 u32 iorr_addr, free_iorr_addr;
71
72
73
74 free_iorr_addr = AMD_K7_NUM_IORR;
75 for (iorr_addr = 0; iorr_addr < AMD_K7_NUM_IORR; iorr_addr++) {
76 rdmsr(IORR_BASE0 + 2 * iorr_addr, base_lo, base_hi);
77 rdmsr(IORR_MASK0 + 2 * iorr_addr, mask_lo, mask_hi);
78
79 if ((base_lo & 0xfffff000) == (base & 0xfffff000))
80 break;
81
82 if ((mask_lo & 0x00000800) == 0)
83 free_iorr_addr = iorr_addr;
84 }
85
86 if (iorr_addr >= AMD_K7_NUM_IORR) {
87 iorr_addr = free_iorr_addr;
88 if (iorr_addr >= AMD_K7_NUM_IORR)
89 return -EINVAL;
90 }
91 base_hi = 0x0;
92 base_lo = (base & ~0xfff) | 0x18;
93 mask_hi = 0xf;
94 mask_lo = ((~(size - 1)) & 0xfffff000) | 0x800;
95 wrmsr(IORR_BASE0 + 2 * iorr_addr, base_lo, base_hi);
96 wrmsr(IORR_MASK0 + 2 * iorr_addr, mask_lo, mask_hi);
97
98 rdmsr(SYSCFG, sys_lo, sys_hi);
99 sys_lo |= 0x00100000;
100 wrmsr(SYSCFG, sys_lo, sys_hi);
101
102 return 0;
103}
104
105static int nvidia_configure(void)
106{
107 int i, rc, num_dirs;
108 u32 apbase, aplimit;
109 phys_addr_t apbase_phys;
110 struct aper_size_info_8 *current_size;
111 u32 temp;
112
113 current_size = A_SIZE_8(agp_bridge->current_size);
114
115
116 pci_write_config_byte(agp_bridge->dev, NVIDIA_0_APSIZE,
117 current_size->size_value);
118
119
120 apbase = pci_bus_address(agp_bridge->dev, AGP_APERTURE_BAR);
121 agp_bridge->gart_bus_addr = apbase;
122 aplimit = apbase + (current_size->size * 1024 * 1024) - 1;
123 pci_write_config_dword(nvidia_private.dev_2, NVIDIA_2_APBASE, apbase);
124 pci_write_config_dword(nvidia_private.dev_2, NVIDIA_2_APLIMIT, aplimit);
125 pci_write_config_dword(nvidia_private.dev_3, NVIDIA_3_APBASE, apbase);
126 pci_write_config_dword(nvidia_private.dev_3, NVIDIA_3_APLIMIT, aplimit);
127 if (0 != (rc = nvidia_init_iorr(apbase, current_size->size * 1024 * 1024)))
128 return rc;
129
130
131 num_dirs = current_size->size / 64;
132 nvidia_private.num_active_entries = current_size->num_entries;
133 nvidia_private.pg_offset = 0;
134 if (num_dirs == 0) {
135 num_dirs = 1;
136 nvidia_private.num_active_entries /= (64 / current_size->size);
137 nvidia_private.pg_offset = (apbase & (64 * 1024 * 1024 - 1) &
138 ~(current_size->size * 1024 * 1024 - 1)) / PAGE_SIZE;
139 }
140
141
142 for (i = 0; i < 8; i++) {
143 pci_write_config_dword(nvidia_private.dev_2, NVIDIA_2_ATTBASE(i),
144 (agp_bridge->gatt_bus_addr + (i % num_dirs) * 64 * 1024) | 1);
145 }
146
147
148 pci_read_config_dword(nvidia_private.dev_2, NVIDIA_2_GARTCTRL, &temp);
149 pci_write_config_dword(nvidia_private.dev_2, NVIDIA_2_GARTCTRL, temp | 0x11);
150
151
152 pci_read_config_dword(agp_bridge->dev, NVIDIA_0_APSIZE, &temp);
153 pci_write_config_dword(agp_bridge->dev, NVIDIA_0_APSIZE, temp | 0x100);
154
155
156 apbase_phys = pci_resource_start(agp_bridge->dev, AGP_APERTURE_BAR);
157 nvidia_private.aperture =
158 (volatile u32 __iomem *) ioremap(apbase_phys, 33 * PAGE_SIZE);
159
160 if (!nvidia_private.aperture)
161 return -ENOMEM;
162
163 return 0;
164}
165
166static void nvidia_cleanup(void)
167{
168 struct aper_size_info_8 *previous_size;
169 u32 temp;
170
171
172 pci_read_config_dword(agp_bridge->dev, NVIDIA_0_APSIZE, &temp);
173 pci_write_config_dword(agp_bridge->dev, NVIDIA_0_APSIZE, temp & ~(0x100));
174
175
176 pci_read_config_dword(nvidia_private.dev_2, NVIDIA_2_GARTCTRL, &temp);
177 pci_write_config_dword(nvidia_private.dev_2, NVIDIA_2_GARTCTRL, temp & ~(0x11));
178
179
180 iounmap((void __iomem *) nvidia_private.aperture);
181
182
183 previous_size = A_SIZE_8(agp_bridge->previous_size);
184 pci_write_config_byte(agp_bridge->dev, NVIDIA_0_APSIZE,
185 previous_size->size_value);
186
187
188 nvidia_init_iorr(agp_bridge->gart_bus_addr,
189 previous_size->size * 1024 * 1024);
190}
191
192
193
194
195
196
197
198
199extern int agp_memory_reserved;
200
201static int nvidia_insert_memory(struct agp_memory *mem, off_t pg_start, int type)
202{
203 int i, j;
204 int mask_type;
205
206 mask_type = agp_generic_type_to_mask_type(mem->bridge, type);
207 if (mask_type != 0 || type != mem->type)
208 return -EINVAL;
209
210 if (mem->page_count == 0)
211 return 0;
212
213 if ((pg_start + mem->page_count) >
214 (nvidia_private.num_active_entries - agp_memory_reserved/PAGE_SIZE))
215 return -EINVAL;
216
217 for (j = pg_start; j < (pg_start + mem->page_count); j++) {
218 if (!PGE_EMPTY(agp_bridge, readl(agp_bridge->gatt_table+nvidia_private.pg_offset+j)))
219 return -EBUSY;
220 }
221
222 if (!mem->is_flushed) {
223 global_cache_flush();
224 mem->is_flushed = true;
225 }
226 for (i = 0, j = pg_start; i < mem->page_count; i++, j++) {
227 writel(agp_bridge->driver->mask_memory(agp_bridge,
228 page_to_phys(mem->pages[i]), mask_type),
229 agp_bridge->gatt_table+nvidia_private.pg_offset+j);
230 }
231
232
233 readl(agp_bridge->gatt_table+nvidia_private.pg_offset+j - 1);
234
235 agp_bridge->driver->tlb_flush(mem);
236 return 0;
237}
238
239
240static int nvidia_remove_memory(struct agp_memory *mem, off_t pg_start, int type)
241{
242 int i;
243
244 int mask_type;
245
246 mask_type = agp_generic_type_to_mask_type(mem->bridge, type);
247 if (mask_type != 0 || type != mem->type)
248 return -EINVAL;
249
250 if (mem->page_count == 0)
251 return 0;
252
253 for (i = pg_start; i < (mem->page_count + pg_start); i++)
254 writel(agp_bridge->scratch_page, agp_bridge->gatt_table+nvidia_private.pg_offset+i);
255
256 agp_bridge->driver->tlb_flush(mem);
257 return 0;
258}
259
260
261static void nvidia_tlbflush(struct agp_memory *mem)
262{
263 unsigned long end;
264 u32 wbc_reg, temp;
265 int i;
266
267
268 if (nvidia_private.wbc_mask) {
269 pci_read_config_dword(nvidia_private.dev_1, NVIDIA_1_WBC, &wbc_reg);
270 wbc_reg |= nvidia_private.wbc_mask;
271 pci_write_config_dword(nvidia_private.dev_1, NVIDIA_1_WBC, wbc_reg);
272
273 end = jiffies + 3*HZ;
274 do {
275 pci_read_config_dword(nvidia_private.dev_1,
276 NVIDIA_1_WBC, &wbc_reg);
277 if (time_before_eq(end, jiffies)) {
278 printk(KERN_ERR PFX
279 "TLB flush took more than 3 seconds.\n");
280 }
281 } while (wbc_reg & nvidia_private.wbc_mask);
282 }
283
284
285 for (i = 0; i < 32 + 1; i++)
286 temp = readl(nvidia_private.aperture+(i * PAGE_SIZE / sizeof(u32)));
287 for (i = 0; i < 32 + 1; i++)
288 temp = readl(nvidia_private.aperture+(i * PAGE_SIZE / sizeof(u32)));
289}
290
291
292static const struct aper_size_info_8 nvidia_generic_sizes[5] =
293{
294 {512, 131072, 7, 0},
295 {256, 65536, 6, 8},
296 {128, 32768, 5, 12},
297 {64, 16384, 4, 14},
298
299 {32, 16384, 4, 15}
300};
301
302
303static const struct gatt_mask nvidia_generic_masks[] =
304{
305 { .mask = 1, .type = 0}
306};
307
308
309static const struct agp_bridge_driver nvidia_driver = {
310 .owner = THIS_MODULE,
311 .aperture_sizes = nvidia_generic_sizes,
312 .size_type = U8_APER_SIZE,
313 .num_aperture_sizes = 5,
314 .needs_scratch_page = true,
315 .configure = nvidia_configure,
316 .fetch_size = nvidia_fetch_size,
317 .cleanup = nvidia_cleanup,
318 .tlb_flush = nvidia_tlbflush,
319 .mask_memory = agp_generic_mask_memory,
320 .masks = nvidia_generic_masks,
321 .agp_enable = agp_generic_enable,
322 .cache_flush = global_cache_flush,
323 .create_gatt_table = agp_generic_create_gatt_table,
324 .free_gatt_table = agp_generic_free_gatt_table,
325 .insert_memory = nvidia_insert_memory,
326 .remove_memory = nvidia_remove_memory,
327 .alloc_by_type = agp_generic_alloc_by_type,
328 .free_by_type = agp_generic_free_by_type,
329 .agp_alloc_page = agp_generic_alloc_page,
330 .agp_alloc_pages = agp_generic_alloc_pages,
331 .agp_destroy_page = agp_generic_destroy_page,
332 .agp_destroy_pages = agp_generic_destroy_pages,
333 .agp_type_to_mask_type = agp_generic_type_to_mask_type,
334};
335
336static int agp_nvidia_probe(struct pci_dev *pdev,
337 const struct pci_device_id *ent)
338{
339 struct agp_bridge_data *bridge;
340 u8 cap_ptr;
341
342 nvidia_private.dev_1 =
343 pci_get_bus_and_slot((unsigned int)pdev->bus->number, PCI_DEVFN(0, 1));
344 nvidia_private.dev_2 =
345 pci_get_bus_and_slot((unsigned int)pdev->bus->number, PCI_DEVFN(0, 2));
346 nvidia_private.dev_3 =
347 pci_get_bus_and_slot((unsigned int)pdev->bus->number, PCI_DEVFN(30, 0));
348
349 if (!nvidia_private.dev_1 || !nvidia_private.dev_2 || !nvidia_private.dev_3) {
350 printk(KERN_INFO PFX "Detected an NVIDIA nForce/nForce2 "
351 "chipset, but could not find the secondary devices.\n");
352 return -ENODEV;
353 }
354
355 cap_ptr = pci_find_capability(pdev, PCI_CAP_ID_AGP);
356 if (!cap_ptr)
357 return -ENODEV;
358
359 switch (pdev->device) {
360 case PCI_DEVICE_ID_NVIDIA_NFORCE:
361 printk(KERN_INFO PFX "Detected NVIDIA nForce chipset\n");
362 nvidia_private.wbc_mask = 0x00010000;
363 break;
364 case PCI_DEVICE_ID_NVIDIA_NFORCE2:
365 printk(KERN_INFO PFX "Detected NVIDIA nForce2 chipset\n");
366 nvidia_private.wbc_mask = 0x80000000;
367 break;
368 default:
369 printk(KERN_ERR PFX "Unsupported NVIDIA chipset (device id: %04x)\n",
370 pdev->device);
371 return -ENODEV;
372 }
373
374 bridge = agp_alloc_bridge();
375 if (!bridge)
376 return -ENOMEM;
377
378 bridge->driver = &nvidia_driver;
379 bridge->dev_private_data = &nvidia_private,
380 bridge->dev = pdev;
381 bridge->capndx = cap_ptr;
382
383
384 pci_read_config_dword(pdev,
385 bridge->capndx+PCI_AGP_STATUS,
386 &bridge->mode);
387
388 pci_set_drvdata(pdev, bridge);
389 return agp_add_bridge(bridge);
390}
391
392static void agp_nvidia_remove(struct pci_dev *pdev)
393{
394 struct agp_bridge_data *bridge = pci_get_drvdata(pdev);
395
396 agp_remove_bridge(bridge);
397 agp_put_bridge(bridge);
398}
399
400#ifdef CONFIG_PM
401static int agp_nvidia_suspend(struct pci_dev *pdev, pm_message_t state)
402{
403 pci_save_state(pdev);
404 pci_set_power_state(pdev, PCI_D3hot);
405
406 return 0;
407}
408
409static int agp_nvidia_resume(struct pci_dev *pdev)
410{
411
412 pci_set_power_state(pdev, PCI_D0);
413 pci_restore_state(pdev);
414
415
416 nvidia_configure();
417
418 return 0;
419}
420#endif
421
422
423static struct pci_device_id agp_nvidia_pci_table[] = {
424 {
425 .class = (PCI_CLASS_BRIDGE_HOST << 8),
426 .class_mask = ~0,
427 .vendor = PCI_VENDOR_ID_NVIDIA,
428 .device = PCI_DEVICE_ID_NVIDIA_NFORCE,
429 .subvendor = PCI_ANY_ID,
430 .subdevice = PCI_ANY_ID,
431 },
432 {
433 .class = (PCI_CLASS_BRIDGE_HOST << 8),
434 .class_mask = ~0,
435 .vendor = PCI_VENDOR_ID_NVIDIA,
436 .device = PCI_DEVICE_ID_NVIDIA_NFORCE2,
437 .subvendor = PCI_ANY_ID,
438 .subdevice = PCI_ANY_ID,
439 },
440 { }
441};
442
443MODULE_DEVICE_TABLE(pci, agp_nvidia_pci_table);
444
445static struct pci_driver agp_nvidia_pci_driver = {
446 .name = "agpgart-nvidia",
447 .id_table = agp_nvidia_pci_table,
448 .probe = agp_nvidia_probe,
449 .remove = agp_nvidia_remove,
450#ifdef CONFIG_PM
451 .suspend = agp_nvidia_suspend,
452 .resume = agp_nvidia_resume,
453#endif
454};
455
456static int __init agp_nvidia_init(void)
457{
458 if (agp_off)
459 return -EINVAL;
460 return pci_register_driver(&agp_nvidia_pci_driver);
461}
462
463static void __exit agp_nvidia_cleanup(void)
464{
465 pci_unregister_driver(&agp_nvidia_pci_driver);
466 pci_dev_put(nvidia_private.dev_1);
467 pci_dev_put(nvidia_private.dev_2);
468 pci_dev_put(nvidia_private.dev_3);
469}
470
471module_init(agp_nvidia_init);
472module_exit(agp_nvidia_cleanup);
473
474MODULE_LICENSE("GPL and additional rights");
475MODULE_AUTHOR("NVIDIA Corporation");
476
477
