1 /*- 2 * Copyright (c) 2003 Matthew N. Dodd <winter@jurai.net> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27 #include <sys/cdefs.h> 28 __FBSDID("$FreeBSD$"); 29 30 /* 31 * Written using information gleaned from the 32 * NVIDIA nForce/nForce2 AGPGART Linux Kernel Patch. 33 */ 34 35 #include "opt_bus.h" 36 37 #include <sys/param.h> 38 #include <sys/systm.h> 39 #include <sys/malloc.h> 40 #include <sys/kernel.h> 41 #include <sys/module.h> 42 #include <sys/bus.h> 43 #include <sys/lock.h> 44 45 #if __FreeBSD_version < 500000 46 #include "opt_pci.h" 47 #endif 48 49 #if __FreeBSD_version > 500000 50 #include <sys/mutex.h> 51 #include <sys/proc.h> 52 #endif 53 54 #include <dev/pci/pcivar.h> 55 #include <dev/pci/pcireg.h> 56 #include <pci/agppriv.h> 57 #include <pci/agpreg.h> 58 59 #include <vm/vm.h> 60 #include <vm/vm_object.h> 61 #include <vm/pmap.h> 62 63 #include <machine/bus.h> 64 #include <machine/resource.h> 65 #include <sys/rman.h> 66 67 #define NVIDIA_VENDORID 0x10de 68 #define NVIDIA_DEVICEID_NFORCE 0x01a4 69 #define NVIDIA_DEVICEID_NFORCE2 0x01e0 70 71 struct agp_nvidia_softc { 72 struct agp_softc agp; 73 u_int32_t initial_aperture; /* aperture size at startup */ 74 struct agp_gatt * gatt; 75 76 device_t dev; /* AGP Controller */ 77 device_t mc1_dev; /* Memory Controller */ 78 device_t mc2_dev; /* Memory Controller */ 79 device_t bdev; /* Bridge */ 80 81 u_int32_t wbc_mask; 82 int num_dirs; 83 int num_active_entries; 84 off_t pg_offset; 85 }; 86 87 static const char *agp_nvidia_match(device_t dev); 88 static int agp_nvidia_probe(device_t); 89 static int agp_nvidia_attach(device_t); 90 static int agp_nvidia_detach(device_t); 91 static u_int32_t agp_nvidia_get_aperture(device_t); 92 static int agp_nvidia_set_aperture(device_t, u_int32_t); 93 static int agp_nvidia_bind_page(device_t, int, vm_offset_t); 94 static int agp_nvidia_unbind_page(device_t, int); 95 96 static int nvidia_init_iorr(u_int32_t, u_int32_t); 97 98 static const char * 99 agp_nvidia_match (device_t dev) 100 { 101 if (pci_get_class(dev) != PCIC_BRIDGE || 102 pci_get_subclass(dev) != PCIS_BRIDGE_HOST || 103 pci_get_vendor(dev) != NVIDIA_VENDORID) 104 return (NULL); 105 106 switch (pci_get_device(dev)) { 107 case NVIDIA_DEVICEID_NFORCE: 108 return ("NVIDIA nForce AGP Controller"); 109 case NVIDIA_DEVICEID_NFORCE2: 110 return ("NVIDIA nForce2 AGP Controller"); 111 } 112 return (NULL); 113 } 114 115 static int 116 agp_nvidia_probe (device_t dev) 117 { 118 const char *desc; 119 120 if (resource_disabled("agp", device_get_unit(dev))) 121 return (ENXIO); 122 desc = agp_nvidia_match(dev); 123 if (desc) { 124 device_set_desc(dev, desc); 125 return (BUS_PROBE_DEFAULT); 126 } 127 return (ENXIO); 128 } 129 130 static int 131 agp_nvidia_attach (device_t dev) 132 { 133 struct agp_nvidia_softc *sc = device_get_softc(dev); 134 struct agp_gatt *gatt; 135 u_int32_t apbase; 136 u_int32_t aplimit; 137 u_int32_t temp; 138 int size; 139 int i; 140 int error; 141 142 switch (pci_get_device(dev)) { 143 case NVIDIA_DEVICEID_NFORCE: 144 sc->wbc_mask = 0x00010000; 145 break; 146 case NVIDIA_DEVICEID_NFORCE2: 147 sc->wbc_mask = 0x80000000; 148 break; 149 default: 150 device_printf(dev, "Bad chip id\n"); 151 return (ENODEV); 152 } 153 154 /* AGP Controller */ 155 sc->dev = dev; 156 157 /* Memory Controller 1 */ 158 sc->mc1_dev = pci_find_bsf(pci_get_bus(dev), 0, 1); 159 if (sc->mc1_dev == NULL) { 160 device_printf(dev, 161 "Unable to find NVIDIA Memory Controller 1.\n"); 162 return (ENODEV); 163 } 164 165 /* Memory Controller 2 */ 166 sc->mc2_dev = pci_find_bsf(pci_get_bus(dev), 0, 2); 167 if (sc->mc2_dev == NULL) { 168 device_printf(dev, 169 "Unable to find NVIDIA Memory Controller 2.\n"); 170 return (ENODEV); 171 } 172 173 /* AGP Host to PCI Bridge */ 174 sc->bdev = pci_find_bsf(pci_get_bus(dev), 30, 0); 175 if (sc->bdev == NULL) { 176 device_printf(dev, 177 "Unable to find NVIDIA AGP Host to PCI Bridge.\n"); 178 return (ENODEV); 179 } 180 181 error = agp_generic_attach(dev); 182 if (error) 183 return (error); 184 185 sc->initial_aperture = AGP_GET_APERTURE(dev); 186 187 for (;;) { 188 gatt = agp_alloc_gatt(dev); 189 if (gatt) 190 break; 191 /* 192 * Probably contigmalloc failure. Try reducing the 193 * aperture so that the gatt size reduces. 194 */ 195 if (AGP_SET_APERTURE(dev, AGP_GET_APERTURE(dev) / 2)) 196 goto fail; 197 } 198 sc->gatt = gatt; 199 200 apbase = rman_get_start(sc->agp.as_aperture); 201 aplimit = apbase + AGP_GET_APERTURE(dev) - 1; 202 pci_write_config(sc->mc2_dev, AGP_NVIDIA_2_APBASE, apbase, 4); 203 pci_write_config(sc->mc2_dev, AGP_NVIDIA_2_APLIMIT, aplimit, 4); 204 pci_write_config(sc->bdev, AGP_NVIDIA_3_APBASE, apbase, 4); 205 pci_write_config(sc->bdev, AGP_NVIDIA_3_APLIMIT, aplimit, 4); 206 207 error = nvidia_init_iorr(apbase, AGP_GET_APERTURE(dev)); 208 if (error) { 209 device_printf(dev, "Failed to setup IORRs\n"); 210 goto fail; 211 } 212 213 /* directory size is 64k */ 214 size = AGP_GET_APERTURE(dev) / 1024 / 1024; 215 sc->num_dirs = size / 64; 216 sc->num_active_entries = (size == 32) ? 16384 : ((size * 1024) / 4); 217 sc->pg_offset = 0; 218 if (sc->num_dirs == 0) { 219 sc->num_dirs = 1; 220 sc->num_active_entries /= (64 / size); 221 sc->pg_offset = (apbase & (64 * 1024 * 1024 - 1) & 222 ~(AGP_GET_APERTURE(dev) - 1)) / PAGE_SIZE; 223 } 224 225 /* (G)ATT Base Address */ 226 for (i = 0; i < 8; i++) { 227 pci_write_config(sc->mc2_dev, AGP_NVIDIA_2_ATTBASE(i), 228 (sc->gatt->ag_physical + 229 (i % sc->num_dirs) * 64 * 1024) | 1, 4); 230 } 231 232 /* GTLB Control */ 233 temp = pci_read_config(sc->mc2_dev, AGP_NVIDIA_2_GARTCTRL, 4); 234 pci_write_config(sc->mc2_dev, AGP_NVIDIA_2_GARTCTRL, temp | 0x11, 4); 235 236 /* GART Control */ 237 temp = pci_read_config(sc->dev, AGP_NVIDIA_0_APSIZE, 4); 238 pci_write_config(sc->dev, AGP_NVIDIA_0_APSIZE, temp | 0x100, 4); 239 240 return (0); 241 fail: 242 agp_generic_detach(dev); 243 return (ENOMEM); 244 } 245 246 static int 247 agp_nvidia_detach (device_t dev) 248 { 249 struct agp_nvidia_softc *sc = device_get_softc(dev); 250 int error; 251 u_int32_t temp; 252 253 error = agp_generic_detach(dev); 254 if (error) 255 return (error); 256 257 /* GART Control */ 258 temp = pci_read_config(sc->dev, AGP_NVIDIA_0_APSIZE, 4); 259 pci_write_config(sc->dev, AGP_NVIDIA_0_APSIZE, temp & ~(0x100), 4); 260 261 /* GTLB Control */ 262 temp = pci_read_config(sc->mc2_dev, AGP_NVIDIA_2_GARTCTRL, 4); 263 pci_write_config(sc->mc2_dev, AGP_NVIDIA_2_GARTCTRL, temp & ~(0x11), 4); 264 265 /* Put the aperture back the way it started. */ 266 AGP_SET_APERTURE(dev, sc->initial_aperture); 267 268 /* restore iorr for previous aperture size */ 269 nvidia_init_iorr(rman_get_start(sc->agp.as_aperture), 270 sc->initial_aperture); 271 272 agp_free_gatt(sc->gatt); 273 274 return (0); 275 } 276 277 static u_int32_t 278 agp_nvidia_get_aperture(device_t dev) 279 { 280 switch (pci_read_config(dev, AGP_NVIDIA_0_APSIZE, 1) & 0x0f) { 281 case 0: return (512 * 1024 * 1024); break; 282 case 8: return (256 * 1024 * 1024); break; 283 case 12: return (128 * 1024 * 1024); break; 284 case 14: return (64 * 1024 * 1024); break; 285 case 15: return (32 * 1024 * 1024); break; 286 default: 287 device_printf(dev, "Invalid aperture setting 0x%x", 288 pci_read_config(dev, AGP_NVIDIA_0_APSIZE, 1)); 289 return 0; 290 } 291 } 292 293 static int 294 agp_nvidia_set_aperture(device_t dev, u_int32_t aperture) 295 { 296 u_int8_t val; 297 u_int8_t key; 298 299 switch (aperture) { 300 case (512 * 1024 * 1024): key = 0; break; 301 case (256 * 1024 * 1024): key = 8; break; 302 case (128 * 1024 * 1024): key = 12; break; 303 case (64 * 1024 * 1024): key = 14; break; 304 case (32 * 1024 * 1024): key = 15; break; 305 default: 306 device_printf(dev, "Invalid aperture size (%dMb)\n", 307 aperture / 1024 / 1024); 308 return (EINVAL); 309 } 310 val = pci_read_config(dev, AGP_NVIDIA_0_APSIZE, 1); 311 pci_write_config(dev, AGP_NVIDIA_0_APSIZE, ((val & ~0x0f) | key), 1); 312 313 return (0); 314 } 315 316 static int 317 agp_nvidia_bind_page(device_t dev, int offset, vm_offset_t physical) 318 { 319 struct agp_nvidia_softc *sc = device_get_softc(dev); 320 u_int32_t index; 321 322 if (offset < 0 || offset >= (sc->gatt->ag_entries << AGP_PAGE_SHIFT)) 323 return (EINVAL); 324 325 index = (sc->pg_offset + offset) >> AGP_PAGE_SHIFT; 326 sc->gatt->ag_virtual[index] = physical | 1; 327 328 return (0); 329 } 330 331 static int 332 agp_nvidia_unbind_page(device_t dev, int offset) 333 { 334 struct agp_nvidia_softc *sc = device_get_softc(dev); 335 u_int32_t index; 336 337 if (offset < 0 || offset >= (sc->gatt->ag_entries << AGP_PAGE_SHIFT)) 338 return (EINVAL); 339 340 index = (sc->pg_offset + offset) >> AGP_PAGE_SHIFT; 341 sc->gatt->ag_virtual[index] = 0; 342 343 return (0); 344 } 345 346 static int 347 agp_nvidia_flush_tlb (device_t dev, int offset) 348 { 349 struct agp_nvidia_softc *sc; 350 u_int32_t wbc_reg, temp; 351 volatile u_int32_t *ag_virtual; 352 int i; 353 354 sc = (struct agp_nvidia_softc *)device_get_softc(dev); 355 356 if (sc->wbc_mask) { 357 wbc_reg = pci_read_config(sc->mc1_dev, AGP_NVIDIA_1_WBC, 4); 358 wbc_reg |= sc->wbc_mask; 359 pci_write_config(sc->mc1_dev, AGP_NVIDIA_1_WBC, wbc_reg, 4); 360 361 /* Wait no more than 3 seconds. */ 362 for (i = 0; i < 3000; i++) { 363 wbc_reg = pci_read_config(sc->mc1_dev, 364 AGP_NVIDIA_1_WBC, 4); 365 if ((sc->wbc_mask & wbc_reg) == 0) 366 break; 367 else 368 DELAY(1000); 369 } 370 if (i == 3000) 371 device_printf(dev, 372 "TLB flush took more than 3 seconds.\n"); 373 } 374 375 ag_virtual = (volatile u_int32_t *)sc->gatt->ag_virtual; 376 377 /* Flush TLB entries. */ 378 for(i = 0; i < 32 + 1; i++) 379 temp = ag_virtual[i * PAGE_SIZE / sizeof(u_int32_t)]; 380 for(i = 0; i < 32 + 1; i++) 381 temp = ag_virtual[i * PAGE_SIZE / sizeof(u_int32_t)]; 382 383 return (0); 384 } 385 386 #define SYSCFG 0xC0010010 387 #define IORR_BASE0 0xC0010016 388 #define IORR_MASK0 0xC0010017 389 #define AMD_K7_NUM_IORR 2 390 391 static int 392 nvidia_init_iorr(u_int32_t addr, u_int32_t size) 393 { 394 quad_t base, mask, sys; 395 u_int32_t iorr_addr, free_iorr_addr; 396 397 /* Find the iorr that is already used for the addr */ 398 /* If not found, determine the uppermost available iorr */ 399 free_iorr_addr = AMD_K7_NUM_IORR; 400 for(iorr_addr = 0; iorr_addr < AMD_K7_NUM_IORR; iorr_addr++) { 401 base = rdmsr(IORR_BASE0 + 2 * iorr_addr); 402 mask = rdmsr(IORR_MASK0 + 2 * iorr_addr); 403 404 if ((base & 0xfffff000ULL) == (addr & 0xfffff000)) 405 break; 406 407 if ((mask & 0x00000800ULL) == 0) 408 free_iorr_addr = iorr_addr; 409 } 410 411 if (iorr_addr >= AMD_K7_NUM_IORR) { 412 iorr_addr = free_iorr_addr; 413 if (iorr_addr >= AMD_K7_NUM_IORR) 414 return (EINVAL); 415 } 416 417 base = (addr & ~0xfff) | 0x18; 418 mask = (0xfULL << 32) | ((~(size - 1)) & 0xfffff000) | 0x800; 419 wrmsr(IORR_BASE0 + 2 * iorr_addr, base); 420 wrmsr(IORR_MASK0 + 2 * iorr_addr, mask); 421 422 sys = rdmsr(SYSCFG); 423 sys |= 0x00100000ULL; 424 wrmsr(SYSCFG, sys); 425 426 return (0); 427 } 428 429 static device_method_t agp_nvidia_methods[] = { 430 /* Device interface */ 431 DEVMETHOD(device_probe, agp_nvidia_probe), 432 DEVMETHOD(device_attach, agp_nvidia_attach), 433 DEVMETHOD(device_detach, agp_nvidia_detach), 434 DEVMETHOD(device_shutdown, bus_generic_shutdown), 435 DEVMETHOD(device_suspend, bus_generic_suspend), 436 DEVMETHOD(device_resume, bus_generic_resume), 437 438 /* AGP interface */ 439 DEVMETHOD(agp_get_aperture, agp_nvidia_get_aperture), 440 DEVMETHOD(agp_set_aperture, agp_nvidia_set_aperture), 441 DEVMETHOD(agp_bind_page, agp_nvidia_bind_page), 442 DEVMETHOD(agp_unbind_page, agp_nvidia_unbind_page), 443 DEVMETHOD(agp_flush_tlb, agp_nvidia_flush_tlb), 444 445 DEVMETHOD(agp_enable, agp_generic_enable), 446 DEVMETHOD(agp_alloc_memory, agp_generic_alloc_memory), 447 DEVMETHOD(agp_free_memory, agp_generic_free_memory), 448 DEVMETHOD(agp_bind_memory, agp_generic_bind_memory), 449 DEVMETHOD(agp_unbind_memory, agp_generic_unbind_memory), 450 451 { 0, 0 } 452 }; 453 454 static driver_t agp_nvidia_driver = { 455 "agp", 456 agp_nvidia_methods, 457 sizeof(struct agp_nvidia_softc), 458 }; 459 460 static devclass_t agp_devclass; 461 462 DRIVER_MODULE(agp_nvidia, hostb, agp_nvidia_driver, agp_devclass, 0, 0); 463 MODULE_DEPEND(agp_nvidia, agp, 1, 1, 1); 464 MODULE_DEPEND(agp_nvidia, pci, 1, 1, 1); 465