arch/sparc64/kernel/pci_sun4v.c

   1 /* pci_sun4v.c: SUN4V specific PCI controller support.
   2  *
   3  * Copyright (C) 2006 David S. Miller (davem@davemloft.net)
   4  */
   5
   6 #include <linux/kernel.h>
   7 #include <linux/types.h>
   8 #include <linux/pci.h>
   9 #include <linux/init.h>
  10 #include <linux/slab.h>
  11 #include <linux/interrupt.h>
  12 #include <linux/percpu.h>
  13
  14 #include <asm/pbm.h>
  15 #include <asm/iommu.h>
  16 #include <asm/irq.h>
  17 #include <asm/upa.h>
  18 #include <asm/pstate.h>
  19 #include <asm/oplib.h>
  20 #include <asm/hypervisor.h>
  21 #include <asm/prom.h>
  22
  23 #include "pci_impl.h"
  24 #include "iommu_common.h"
  25
  26 #include "pci_sun4v.h"
  27
  28 #define PGLIST_NENTS    (PAGE_SIZE / sizeof(u64))
  29
  30 struct pci_iommu_batch {
  31         struct pci_dev  *pdev;          /* Device mapping is for.       */
  32         unsigned long   prot;           /* IOMMU page protections       */
  33         unsigned long   entry;          /* Index into IOTSB.            */
  34         u64             *pglist;        /* List of physical pages       */
  35         unsigned long   npages;         /* Number of pages in list.     */
  36 };
  37
  38 static DEFINE_PER_CPU(struct pci_iommu_batch, pci_iommu_batch);
  39
  40 /* Interrupts must be disabled.  */
  41 static inline void pci_iommu_batch_start(struct pci_dev *pdev, unsigned long prot, unsigned long entry)
  42 {
  43         struct pci_iommu_batch *p = &__get_cpu_var(pci_iommu_batch);
  44
  45         p->pdev         = pdev;
  46         p->prot         = prot;
  47         p->entry        = entry;
  48         p->npages       = 0;
  49 }
  50
  51 /* Interrupts must be disabled.  */
  52 static long pci_iommu_batch_flush(struct pci_iommu_batch *p)
  53 {
  54         struct pcidev_cookie *pcp = p->pdev->sysdata;
  55         unsigned long devhandle = pcp->pbm->devhandle;
  56         unsigned long prot = p->prot;
  57         unsigned long entry = p->entry;
  58         u64 *pglist = p->pglist;
  59         unsigned long npages = p->npages;
  60
  61         while (npages != 0) {
  62                 long num;
  63
  64                 num = pci_sun4v_iommu_map(devhandle, HV_PCI_TSBID(0, entry),
  65                                           npages, prot, __pa(pglist));
  66                 if (unlikely(num < 0)) {
  67                         if (printk_ratelimit())
  68                                 printk("pci_iommu_batch_flush: IOMMU map of "
  69                                        "[%08lx:%08lx:%lx:%lx:%lx] failed with "
  70                                        "status %ld\n",
  71                                        devhandle, HV_PCI_TSBID(0, entry),
  72                                        npages, prot, __pa(pglist), num);
  73                         return -1;
  74                 }
  75
  76                 entry += num;
  77                 npages -= num;
  78                 pglist += num;
  79         }
  80
  81         p->entry = entry;
  82         p->npages = 0;
  83
  84         return 0;
  85 }
  86
  87 /* Interrupts must be disabled.  */
  88 static inline long pci_iommu_batch_add(u64 phys_page)
  89 {
  90         struct pci_iommu_batch *p = &__get_cpu_var(pci_iommu_batch);
  91
  92         BUG_ON(p->npages >= PGLIST_NENTS);
  93
  94         p->pglist[p->npages++] = phys_page;
  95         if (p->npages == PGLIST_NENTS)
  96                 return pci_iommu_batch_flush(p);
  97
  98         return 0;
  99 }
 100
 101 /* Interrupts must be disabled.  */
 102 static inline long pci_iommu_batch_end(void)
 103 {
 104         struct pci_iommu_batch *p = &__get_cpu_var(pci_iommu_batch);
 105
 106         BUG_ON(p->npages >= PGLIST_NENTS);
 107
 108         return pci_iommu_batch_flush(p);
 109 }
 110
 111 static long pci_arena_alloc(struct pci_iommu_arena *arena, unsigned long npages)
 112 {
 113         unsigned long n, i, start, end, limit;
 114         int pass;
 115
 116         limit = arena->limit;
 117         start = arena->hint;
 118         pass = 0;
 119
 120 again:
 121         n = find_next_zero_bit(arena->map, limit, start);
 122         end = n + npages;
 123         if (unlikely(end >= limit)) {
 124                 if (likely(pass < 1)) {
 125                         limit = start;
 126                         start = 0;
 127                         pass++;
 128                         goto again;
 129                 } else {
 130                         /* Scanned the whole thing, give up. */
 131                         return -1;
 132                 }
 133         }
 134
 135         for (i = n; i < end; i++) {
 136                 if (test_bit(i, arena->map)) {
 137                         start = i + 1;
 138                         goto again;
 139                 }
 140         }
 141
 142         for (i = n; i < end; i++)
 143                 __set_bit(i, arena->map);
 144
 145         arena->hint = end;
 146
 147         return n;
 148 }
 149
 150 static void pci_arena_free(struct pci_iommu_arena *arena, unsigned long base, unsigned long npages)
 151 {
 152         unsigned long i;
 153
 154         for (i = base; i < (base + npages); i++)
 155                 __clear_bit(i, arena->map);
 156 }
 157
 158 static void *pci_4v_alloc_consistent(struct pci_dev *pdev, size_t size, dma_addr_t *dma_addrp, gfp_t gfp)
 159 {
 160         struct pcidev_cookie *pcp;
 161         struct pci_iommu *iommu;
 162         unsigned long flags, order, first_page, npages, n;
 163         void *ret;
 164         long entry;
 165
 166         size = IO_PAGE_ALIGN(size);
 167         order = get_order(size);
 168         if (unlikely(order >= MAX_ORDER))
 169                 return NULL;
 170
 171         npages = size >> IO_PAGE_SHIFT;
 172
 173         first_page = __get_free_pages(gfp, order);
 174         if (unlikely(first_page == 0UL))
 175                 return NULL;
 176
 177         memset((char *)first_page, 0, PAGE_SIZE << order);
 178
 179         pcp = pdev->sysdata;
 180         iommu = pcp->pbm->iommu;
 181
 182         spin_lock_irqsave(&iommu->lock, flags);
 183         entry = pci_arena_alloc(&iommu->arena, npages);
 184         spin_unlock_irqrestore(&iommu->lock, flags);
 185
 186         if (unlikely(entry < 0L))
 187                 goto arena_alloc_fail;
 188
 189         *dma_addrp = (iommu->page_table_map_base +
 190                       (entry << IO_PAGE_SHIFT));
 191         ret = (void *) first_page;
 192         first_page = __pa(first_page);
 193
 194         local_irq_save(flags);
 195
 196         pci_iommu_batch_start(pdev,
 197                               (HV_PCI_MAP_ATTR_READ |
 198                                HV_PCI_MAP_ATTR_WRITE),
 199                               entry);
 200
 201         for (n = 0; n < npages; n++) {
 202                 long err = pci_iommu_batch_add(first_page + (n * PAGE_SIZE));
 203                 if (unlikely(err < 0L))
 204                         goto iommu_map_fail;
 205         }
 206
 207         if (unlikely(pci_iommu_batch_end() < 0L))
 208                 goto iommu_map_fail;
 209
 210         local_irq_restore(flags);
 211
 212         return ret;
 213
 214 iommu_map_fail:
 215         /* Interrupts are disabled.  */
 216         spin_lock(&iommu->lock);
 217         pci_arena_free(&iommu->arena, entry, npages);
 218         spin_unlock_irqrestore(&iommu->lock, flags);
 219
 220 arena_alloc_fail:
 221         free_pages(first_page, order);
 222         return NULL;
 223 }
 224
 225 static void pci_4v_free_consistent(struct pci_dev *pdev, size_t size, void *cpu, dma_addr_t dvma)
 226 {
 227         struct pcidev_cookie *pcp;
 228         struct pci_iommu *iommu;
 229         unsigned long flags, order, npages, entry;
 230         u32 devhandle;
 231
 232         npages = IO_PAGE_ALIGN(size) >> IO_PAGE_SHIFT;
 233         pcp = pdev->sysdata;
 234         iommu = pcp->pbm->iommu;
 235         devhandle = pcp->pbm->devhandle;
 236         entry = ((dvma - iommu->page_table_map_base) >> IO_PAGE_SHIFT);
 237
 238         spin_lock_irqsave(&iommu->lock, flags);
 239
 240         pci_arena_free(&iommu->arena, entry, npages);
 241
 242         do {
 243                 unsigned long num;
 244
 245                 num = pci_sun4v_iommu_demap(devhandle, HV_PCI_TSBID(0, entry),
 246                                             npages);
 247                 entry += num;
 248                 npages -= num;
 249         } while (npages != 0);
 250
 251         spin_unlock_irqrestore(&iommu->lock, flags);
 252
 253         order = get_order(size);
 254         if (order < 10)
 255                 free_pages((unsigned long)cpu, order);
 256 }
 257
 258 static dma_addr_t pci_4v_map_single(struct pci_dev *pdev, void *ptr, size_t sz, int direction)
 259 {
 260         struct pcidev_cookie *pcp;
 261         struct pci_iommu *iommu;
 262         unsigned long flags, npages, oaddr;
 263         unsigned long i, base_paddr;
 264         u32 bus_addr, ret;
 265         unsigned long prot;
 266         long entry;
 267
 268         pcp = pdev->sysdata;
 269         iommu = pcp->pbm->iommu;
 270
 271         if (unlikely(direction == PCI_DMA_NONE))
 272                 goto bad;
 273
 274         oaddr = (unsigned long)ptr;
 275         npages = IO_PAGE_ALIGN(oaddr + sz) - (oaddr & IO_PAGE_MASK);
 276         npages >>= IO_PAGE_SHIFT;
 277
 278         spin_lock_irqsave(&iommu->lock, flags);
 279         entry = pci_arena_alloc(&iommu->arena, npages);
 280         spin_unlock_irqrestore(&iommu->lock, flags);
 281
 282         if (unlikely(entry < 0L))
 283                 goto bad;
 284
 285         bus_addr = (iommu->page_table_map_base +
 286                     (entry << IO_PAGE_SHIFT));
 287         ret = bus_addr | (oaddr & ~IO_PAGE_MASK);
 288         base_paddr = __pa(oaddr & IO_PAGE_MASK);
 289         prot = HV_PCI_MAP_ATTR_READ;
 290         if (direction != PCI_DMA_TODEVICE)
 291                 prot |= HV_PCI_MAP_ATTR_WRITE;
 292
 293         local_irq_save(flags);
 294
 295         pci_iommu_batch_start(pdev, prot, entry);
 296
 297         for (i = 0; i < npages; i++, base_paddr += IO_PAGE_SIZE) {
 298                 long err = pci_iommu_batch_add(base_paddr);
 299                 if (unlikely(err < 0L))
 300                         goto iommu_map_fail;
 301         }
 302         if (unlikely(pci_iommu_batch_end() < 0L))
 303                 goto iommu_map_fail;
 304
 305         local_irq_restore(flags);
 306
 307         return ret;
 308
 309 bad:
 310         if (printk_ratelimit())
 311                 WARN_ON(1);
 312         return PCI_DMA_ERROR_CODE;
 313
 314 iommu_map_fail:
 315         /* Interrupts are disabled.  */
 316         spin_lock(&iommu->lock);
 317         pci_arena_free(&iommu->arena, entry, npages);
 318         spin_unlock_irqrestore(&iommu->lock, flags);
 319
 320         return PCI_DMA_ERROR_CODE;
 321 }
 322
 323 static void pci_4v_unmap_single(struct pci_dev *pdev, dma_addr_t bus_addr, size_t sz, int direction)
 324 {
 325         struct pcidev_cookie *pcp;
 326         struct pci_iommu *iommu;
 327         unsigned long flags, npages;
 328         long entry;
 329         u32 devhandle;
 330
 331         if (unlikely(direction == PCI_DMA_NONE)) {
 332                 if (printk_ratelimit())
 333                         WARN_ON(1);
 334                 return;
 335         }
 336
 337         pcp = pdev->sysdata;
 338         iommu = pcp->pbm->iommu;
 339         devhandle = pcp->pbm->devhandle;
 340
 341         npages = IO_PAGE_ALIGN(bus_addr + sz) - (bus_addr & IO_PAGE_MASK);
 342         npages >>= IO_PAGE_SHIFT;
 343         bus_addr &= IO_PAGE_MASK;
 344
 345         spin_lock_irqsave(&iommu->lock, flags);
 346
 347         entry = (bus_addr - iommu->page_table_map_base) >> IO_PAGE_SHIFT;
 348         pci_arena_free(&iommu->arena, entry, npages);
 349
 350         do {
 351                 unsigned long num;
 352
 353                 num = pci_sun4v_iommu_demap(devhandle, HV_PCI_TSBID(0, entry),
 354                                             npages);
 355                 entry += num;
 356                 npages -= num;
 357         } while (npages != 0);
 358
 359         spin_unlock_irqrestore(&iommu->lock, flags);
 360 }
 361
 362 #define SG_ENT_PHYS_ADDRESS(SG) \
 363         (__pa(page_address((SG)->page)) + (SG)->offset)
 364
 365 static inline long fill_sg(long entry, struct pci_dev *pdev,
 366                            struct scatterlist *sg,
 367                            int nused, int nelems, unsigned long prot)
 368 {
 369         struct scatterlist *dma_sg = sg;
 370         struct scatterlist *sg_end = sg + nelems;
 371         unsigned long flags;
 372         int i;
 373
 374         local_irq_save(flags);
 375
 376         pci_iommu_batch_start(pdev, prot, entry);
 377
 378         for (i = 0; i < nused; i++) {
 379                 unsigned long pteval = ~0UL;
 380                 u32 dma_npages;
 381
 382                 dma_npages = ((dma_sg->dma_address & (IO_PAGE_SIZE - 1UL)) +
 383                               dma_sg->dma_length +
 384                               ((IO_PAGE_SIZE - 1UL))) >> IO_PAGE_SHIFT;
 385                 do {
 386                         unsigned long offset;
 387                         signed int len;
 388
 389                         /* If we are here, we know we have at least one
 390                          * more page to map.  So walk forward until we
 391                          * hit a page crossing, and begin creating new
 392                          * mappings from that spot.
 393                          */
 394                         for (;;) {
 395                                 unsigned long tmp;
 396
 397                                 tmp = SG_ENT_PHYS_ADDRESS(sg);
 398                                 len = sg->length;
 399                                 if (((tmp ^ pteval) >> IO_PAGE_SHIFT) != 0UL) {
 400                                         pteval = tmp & IO_PAGE_MASK;
 401                                         offset = tmp & (IO_PAGE_SIZE - 1UL);
 402                                         break;
 403                                 }
 404                                 if (((tmp ^ (tmp + len - 1UL)) >> IO_PAGE_SHIFT) != 0UL) {
 405                                         pteval = (tmp + IO_PAGE_SIZE) & IO_PAGE_MASK;
 406                                         offset = 0UL;
 407                                         len -= (IO_PAGE_SIZE - (tmp & (IO_PAGE_SIZE - 1UL)));
 408                                         break;
 409                                 }
 410                                 sg++;
 411                         }
 412
 413                         pteval = (pteval & IOPTE_PAGE);
 414                         while (len > 0) {
 415                                 long err;
 416
 417                                 err = pci_iommu_batch_add(pteval);
 418                                 if (unlikely(err < 0L))
 419                                         goto iommu_map_failed;
 420
 421                                 pteval += IO_PAGE_SIZE;
 422                                 len -= (IO_PAGE_SIZE - offset);
 423                                 offset = 0;
 424                                 dma_npages--;
 425                         }
 426
 427                         pteval = (pteval & IOPTE_PAGE) + len;
 428                         sg++;
 429
 430                         /* Skip over any tail mappings we've fully mapped,
 431                          * adjusting pteval along the way.  Stop when we
 432                          * detect a page crossing event.
 433                          */
 434                         while (sg < sg_end &&
 435                                (pteval << (64 - IO_PAGE_SHIFT)) != 0UL &&
 436                                (pteval == SG_ENT_PHYS_ADDRESS(sg)) &&
 437                                ((pteval ^
 438                                  (SG_ENT_PHYS_ADDRESS(sg) + sg->length - 1UL)) >> IO_PAGE_SHIFT) == 0UL) {
 439                                 pteval += sg->length;
 440                                 sg++;
 441                         }
 442                         if ((pteval << (64 - IO_PAGE_SHIFT)) == 0UL)
 443                                 pteval = ~0UL;
 444                 } while (dma_npages != 0);
 445                 dma_sg++;
 446         }
 447
 448         if (unlikely(pci_iommu_batch_end() < 0L))
 449                 goto iommu_map_failed;
 450
 451         local_irq_restore(flags);
 452         return 0;
 453
 454 iommu_map_failed:
 455         local_irq_restore(flags);
 456         return -1L;
 457 }
 458
 459 static int pci_4v_map_sg(struct pci_dev *pdev, struct scatterlist *sglist, int nelems, int direction)
 460 {
 461         struct pcidev_cookie *pcp;
 462         struct pci_iommu *iommu;
 463         unsigned long flags, npages, prot;
 464         u32 dma_base;
 465         struct scatterlist *sgtmp;
 466         long entry, err;
 467         int used;
 468
 469         /* Fast path single entry scatterlists. */
 470         if (nelems == 1) {
 471                 sglist->dma_address =
 472                         pci_4v_map_single(pdev,
 473                                           (page_address(sglist->page) + sglist->offset),
 474                                           sglist->length, direction);
 475                 if (unlikely(sglist->dma_address == PCI_DMA_ERROR_CODE))
 476                         return 0;
 477                 sglist->dma_length = sglist->length;
 478                 return 1;
 479         }
 480
 481         pcp = pdev->sysdata;
 482         iommu = pcp->pbm->iommu;
 483
 484         if (unlikely(direction == PCI_DMA_NONE))
 485                 goto bad;
 486
 487         /* Step 1: Prepare scatter list. */
 488         npages = prepare_sg(sglist, nelems);
 489
 490         /* Step 2: Allocate a cluster and context, if necessary. */
 491         spin_lock_irqsave(&iommu->lock, flags);
 492         entry = pci_arena_alloc(&iommu->arena, npages);
 493         spin_unlock_irqrestore(&iommu->lock, flags);
 494
 495         if (unlikely(entry < 0L))
 496                 goto bad;
 497
 498         dma_base = iommu->page_table_map_base +
 499                 (entry << IO_PAGE_SHIFT);
 500
 501         /* Step 3: Normalize DMA addresses. */
 502         used = nelems;
 503
 504         sgtmp = sglist;
 505         while (used && sgtmp->dma_length) {
 506                 sgtmp->dma_address += dma_base;
 507                 sgtmp++;
 508                 used--;
 509         }
 510         used = nelems - used;
 511
 512         /* Step 4: Create the mappings. */
 513         prot = HV_PCI_MAP_ATTR_READ;
 514         if (direction != PCI_DMA_TODEVICE)
 515                 prot |= HV_PCI_MAP_ATTR_WRITE;
 516
 517         err = fill_sg(entry, pdev, sglist, used, nelems, prot);
 518         if (unlikely(err < 0L))
 519                 goto iommu_map_failed;
 520
 521         return used;
 522
 523 bad:
 524         if (printk_ratelimit())
 525                 WARN_ON(1);
 526         return 0;
 527
 528 iommu_map_failed:
 529         spin_lock_irqsave(&iommu->lock, flags);
 530         pci_arena_free(&iommu->arena, entry, npages);
 531         spin_unlock_irqrestore(&iommu->lock, flags);
 532
 533         return 0;
 534 }
 535
 536 static void pci_4v_unmap_sg(struct pci_dev *pdev, struct scatterlist *sglist, int nelems, int direction)
 537 {
 538         struct pcidev_cookie *pcp;
 539         struct pci_iommu *iommu;
 540         unsigned long flags, i, npages;
 541         long entry;
 542         u32 devhandle, bus_addr;
 543
 544         if (unlikely(direction == PCI_DMA_NONE)) {
 545                 if (printk_ratelimit())
 546                         WARN_ON(1);
 547         }
 548
 549         pcp = pdev->sysdata;
 550         iommu = pcp->pbm->iommu;
 551         devhandle = pcp->pbm->devhandle;
 552
 553         bus_addr = sglist->dma_address & IO_PAGE_MASK;
 554
 555         for (i = 1; i < nelems; i++)
 556                 if (sglist[i].dma_length == 0)
 557                         break;
 558         i--;
 559         npages = (IO_PAGE_ALIGN(sglist[i].dma_address + sglist[i].dma_length) -
 560                   bus_addr) >> IO_PAGE_SHIFT;
 561
 562         entry = ((bus_addr - iommu->page_table_map_base) >> IO_PAGE_SHIFT);
 563
 564         spin_lock_irqsave(&iommu->lock, flags);
 565
 566         pci_arena_free(&iommu->arena, entry, npages);
 567
 568         do {
 569                 unsigned long num;
 570
 571                 num = pci_sun4v_iommu_demap(devhandle, HV_PCI_TSBID(0, entry),
 572                                             npages);
 573                 entry += num;
 574                 npages -= num;
 575         } while (npages != 0);
 576
 577         spin_unlock_irqrestore(&iommu->lock, flags);
 578 }
 579
 580 static void pci_4v_dma_sync_single_for_cpu(struct pci_dev *pdev, dma_addr_t bus_addr, size_t sz, int direction)
 581 {
 582         /* Nothing to do... */
 583 }
 584
 585 static void pci_4v_dma_sync_sg_for_cpu(struct pci_dev *pdev, struct scatterlist *sglist, int nelems, int direction)
 586 {
 587         /* Nothing to do... */
 588 }
 589
 590 struct pci_iommu_ops pci_sun4v_iommu_ops = {
 591         .alloc_consistent               = pci_4v_alloc_consistent,
 592         .free_consistent                = pci_4v_free_consistent,
 593         .map_single                     = pci_4v_map_single,
 594         .unmap_single                   = pci_4v_unmap_single,
 595         .map_sg                         = pci_4v_map_sg,
 596         .unmap_sg                       = pci_4v_unmap_sg,
 597         .dma_sync_single_for_cpu        = pci_4v_dma_sync_single_for_cpu,
 598         .dma_sync_sg_for_cpu            = pci_4v_dma_sync_sg_for_cpu,
 599 };
 600
 601 /* SUN4V PCI configuration space accessors. */
 602
 603 struct pdev_entry {
 604         struct pdev_entry       *next;
 605         u32                     devhandle;
 606         unsigned int            bus;
 607         unsigned int            device;
 608         unsigned int            func;
 609 };
 610
 611 #define PDEV_HTAB_SIZE  16
 612 #define PDEV_HTAB_MASK  (PDEV_HTAB_SIZE - 1)
 613 static struct pdev_entry *pdev_htab[PDEV_HTAB_SIZE];
 614
 615 static inline unsigned int pdev_hashfn(u32 devhandle, unsigned int bus, unsigned int device, unsigned int func)
 616 {
 617         unsigned int val;
 618
 619         val = (devhandle ^ (devhandle >> 4));
 620         val ^= bus;
 621         val ^= device;
 622         val ^= func;
 623
 624         return val & PDEV_HTAB_MASK;
 625 }
 626
 627 static int pdev_htab_add(u32 devhandle, unsigned int bus, unsigned int device, unsigned int func)
 628 {
 629         struct pdev_entry *p = kmalloc(sizeof(*p), GFP_KERNEL);
 630         struct pdev_entry **slot;
 631
 632         if (!p)
 633                 return -ENOMEM;
 634
 635         slot = &pdev_htab[pdev_hashfn(devhandle, bus, device, func)];
 636         p->next = *slot;
 637         *slot = p;
 638
 639         p->devhandle = devhandle;
 640         p->bus = bus;
 641         p->device = device;
 642         p->func = func;
 643
 644         return 0;
 645 }
 646
 647 /* Recursively descend into the OBP device tree, rooted at toplevel_node,
 648  * looking for a PCI device matching bus and devfn.
 649  */
 650 static int obp_find(struct device_node *toplevel_node, unsigned int bus, unsigned int devfn)
 651 {
 652         toplevel_node = toplevel_node->child;
 653
 654         while (toplevel_node != NULL) {
 655                 struct linux_prom_pci_registers *regs;
 656                 struct property *prop;
 657                 int ret;
 658
 659                 ret = obp_find(toplevel_node, bus, devfn);
 660                 if (ret != 0)
 661                         return ret;
 662
 663                 prop = of_find_property(toplevel_node, "reg", NULL);
 664                 if (!prop)
 665                         goto next_sibling;
 666
 667                 regs = prop->value;
 668                 if (((regs->phys_hi >> 16) & 0xff) == bus &&
 669                     ((regs->phys_hi >> 8) & 0xff) == devfn)
 670                         break;
 671
 672         next_sibling:
 673                 toplevel_node = toplevel_node->sibling;
 674         }
 675
 676         return toplevel_node != NULL;
 677 }
 678
 679 static int pdev_htab_populate(struct pci_pbm_info *pbm)
 680 {
 681         u32 devhandle = pbm->devhandle;
 682         unsigned int bus;
 683
 684         for (bus = pbm->pci_first_busno; bus <= pbm->pci_last_busno; bus++) {
 685                 unsigned int devfn;
 686
 687                 for (devfn = 0; devfn < 256; devfn++) {
 688                         unsigned int device = PCI_SLOT(devfn);
 689                         unsigned int func = PCI_FUNC(devfn);
 690
 691                         if (obp_find(pbm->prom_node, bus, devfn)) {
 692                                 int err = pdev_htab_add(devhandle, bus,
 693                                                         device, func);
 694                                 if (err)
 695                                         return err;
 696                         }
 697                 }
 698         }
 699
 700         return 0;
 701 }
 702
 703 static struct pdev_entry *pdev_find(u32 devhandle, unsigned int bus, unsigned int device, unsigned int func)
 704 {
 705         struct pdev_entry *p;
 706
 707         p = pdev_htab[pdev_hashfn(devhandle, bus, device, func)];
 708         while (p) {
 709                 if (p->devhandle == devhandle &&
 710                     p->bus == bus &&
 711                     p->device == device &&
 712                     p->func == func)
 713                         break;
 714
 715                 p = p->next;
 716         }
 717
 718         return p;
 719 }
 720
 721 static inline int pci_sun4v_out_of_range(struct pci_pbm_info *pbm, unsigned int bus, unsigned int device, unsigned int func)
 722 {
 723         if (bus < pbm->pci_first_busno ||
 724             bus > pbm->pci_last_busno)
 725                 return 1;
 726         return pdev_find(pbm->devhandle, bus, device, func) == NULL;
 727 }
 728
 729 static int pci_sun4v_read_pci_cfg(struct pci_bus *bus_dev, unsigned int devfn,
 730                                   int where, int size, u32 *value)
 731 {
 732         struct pci_pbm_info *pbm = bus_dev->sysdata;
 733         u32 devhandle = pbm->devhandle;
 734         unsigned int bus = bus_dev->number;
 735         unsigned int device = PCI_SLOT(devfn);
 736         unsigned int func = PCI_FUNC(devfn);
 737         unsigned long ret;
 738
 739         if (pci_sun4v_out_of_range(pbm, bus, device, func)) {
 740                 ret = ~0UL;
 741         } else {
 742                 ret = pci_sun4v_config_get(devhandle,
 743                                 HV_PCI_DEVICE_BUILD(bus, device, func),
 744                                 where, size);
 745 #if 0
 746                 printk("rcfg: [%x:%x:%x:%d]=[%lx]\n",
 747                        devhandle, HV_PCI_DEVICE_BUILD(bus, device, func),
 748                        where, size, ret);
 749 #endif
 750         }
 751         switch (size) {
 752         case 1:
 753                 *value = ret & 0xff;
 754                 break;
 755         case 2:
 756                 *value = ret & 0xffff;
 757                 break;
 758         case 4:
 759                 *value = ret & 0xffffffff;
 760                 break;
 761         };
 762
 763
 764         return PCIBIOS_SUCCESSFUL;
 765 }
 766
 767 static int pci_sun4v_write_pci_cfg(struct pci_bus *bus_dev, unsigned int devfn,
 768                                    int where, int size, u32 value)
 769 {
 770         struct pci_pbm_info *pbm = bus_dev->sysdata;
 771         u32 devhandle = pbm->devhandle;
 772         unsigned int bus = bus_dev->number;
 773         unsigned int device = PCI_SLOT(devfn);
 774         unsigned int func = PCI_FUNC(devfn);
 775         unsigned long ret;
 776
 777         if (pci_sun4v_out_of_range(pbm, bus, device, func)) {
 778                 /* Do nothing. */
 779         } else {
 780                 ret = pci_sun4v_config_put(devhandle,
 781                                 HV_PCI_DEVICE_BUILD(bus, device, func),
 782                                 where, size, value);
 783 #if 0
 784                 printk("wcfg: [%x:%x:%x:%d] v[%x] == [%lx]\n",
 785                        devhandle, HV_PCI_DEVICE_BUILD(bus, device, func),
 786                        where, size, value, ret);
 787 #endif
 788         }
 789         return PCIBIOS_SUCCESSFUL;
 790 }
 791
 792 static struct pci_ops pci_sun4v_ops = {
 793         .read =         pci_sun4v_read_pci_cfg,
 794         .write =        pci_sun4v_write_pci_cfg,
 795 };
 796
 797
 798 static void pbm_scan_bus(struct pci_controller_info *p,
 799                          struct pci_pbm_info *pbm)
 800 {
 801         struct pcidev_cookie *cookie = kzalloc(sizeof(*cookie), GFP_KERNEL);
 802
 803         if (!cookie) {
 804                 prom_printf("%s: Critical allocation failure.\n", pbm->name);
 805                 prom_halt();
 806         }
 807
 808         /* All we care about is the PBM. */
 809         cookie->pbm = pbm;
 810
 811         pbm->pci_bus = pci_scan_bus(pbm->pci_first_busno, p->pci_ops, pbm);
 812 #if 0
 813         pci_fixup_host_bridge_self(pbm->pci_bus);
 814         pbm->pci_bus->self->sysdata = cookie;
 815 #endif
 816         pci_fill_in_pbm_cookies(pbm->pci_bus, pbm, pbm->prom_node);
 817         pci_record_assignments(pbm, pbm->pci_bus);
 818         pci_assign_unassigned(pbm, pbm->pci_bus);
 819         pci_fixup_irq(pbm, pbm->pci_bus);
 820         pci_determine_66mhz_disposition(pbm, pbm->pci_bus);
 821         pci_setup_busmastering(pbm, pbm->pci_bus);
 822 }
 823
 824 static void pci_sun4v_scan_bus(struct pci_controller_info *p)
 825 {
 826         struct property *prop;
 827         struct device_node *dp;
 828
 829         if ((dp = p->pbm_A.prom_node) != NULL) {
 830                 prop = of_find_property(dp, "66mhz-capable", NULL);
 831                 p->pbm_A.is_66mhz_capable = (prop != NULL);
 832
 833                 pbm_scan_bus(p, &p->pbm_A);
 834         }
 835         if ((dp = p->pbm_B.prom_node) != NULL) {
 836                 prop = of_find_property(dp, "66mhz-capable", NULL);
 837                 p->pbm_B.is_66mhz_capable = (prop != NULL);
 838
 839                 pbm_scan_bus(p, &p->pbm_B);
 840         }
 841
 842         /* XXX register error interrupt handlers XXX */
 843 }
 844
 845 static void pci_sun4v_base_address_update(struct pci_dev *pdev, int resource)
 846 {
 847         struct pcidev_cookie *pcp = pdev->sysdata;
 848         struct pci_pbm_info *pbm = pcp->pbm;
 849         struct resource *res, *root;
 850         u32 reg;
 851         int where, size, is_64bit;
 852
 853         res = &pdev->resource[resource];
 854         if (resource < 6) {
 855                 where = PCI_BASE_ADDRESS_0 + (resource * 4);
 856         } else if (resource == PCI_ROM_RESOURCE) {
 857                 where = pdev->rom_base_reg;
 858         } else {
 859                 /* Somebody might have asked allocation of a non-standard resource */
 860                 return;
 861         }
 862
 863         /* XXX 64-bit MEM handling is not %100 correct... XXX */
 864         is_64bit = 0;
 865         if (res->flags & IORESOURCE_IO)
 866                 root = &pbm->io_space;
 867         else {
 868                 root = &pbm->mem_space;
 869                 if ((res->flags & PCI_BASE_ADDRESS_MEM_TYPE_MASK)
 870                     == PCI_BASE_ADDRESS_MEM_TYPE_64)
 871                         is_64bit = 1;
 872         }
 873
 874         size = res->end - res->start;
 875         pci_read_config_dword(pdev, where, &reg);
 876         reg = ((reg & size) |
 877                (((u32)(res->start - root->start)) & ~size));
 878         if (resource == PCI_ROM_RESOURCE) {
 879                 reg |= PCI_ROM_ADDRESS_ENABLE;
 880                 res->flags |= IORESOURCE_ROM_ENABLE;
 881         }
 882         pci_write_config_dword(pdev, where, reg);
 883
 884         /* This knows that the upper 32-bits of the address
 885          * must be zero.  Our PCI common layer enforces this.
 886          */
 887         if (is_64bit)
 888                 pci_write_config_dword(pdev, where + 4, 0);
 889 }
 890
 891 static void pci_sun4v_resource_adjust(struct pci_dev *pdev,
 892                                       struct resource *res,
 893                                       struct resource *root)
 894 {
 895         res->start += root->start;
 896         res->end += root->start;
 897 }
 898
 899 /* Use ranges property to determine where PCI MEM, I/O, and Config
 900  * space are for this PCI bus module.
 901  */
 902 static void pci_sun4v_determine_mem_io_space(struct pci_pbm_info *pbm)
 903 {
 904         int i, saw_mem, saw_io;
 905
 906         saw_mem = saw_io = 0;
 907         for (i = 0; i < pbm->num_pbm_ranges; i++) {
 908                 struct linux_prom_pci_ranges *pr = &pbm->pbm_ranges[i];
 909                 unsigned long a;
 910                 int type;
 911
 912                 type = (pr->child_phys_hi >> 24) & 0x3;
 913                 a = (((unsigned long)pr->parent_phys_hi << 32UL) |
 914                      ((unsigned long)pr->parent_phys_lo  <<  0UL));
 915
 916                 switch (type) {
 917                 case 1:
 918                         /* 16-bit IO space, 16MB */
 919                         pbm->io_space.start = a;
 920                         pbm->io_space.end = a + ((16UL*1024UL*1024UL) - 1UL);
 921                         pbm->io_space.flags = IORESOURCE_IO;
 922                         saw_io = 1;
 923                         break;
 924
 925                 case 2:
 926                         /* 32-bit MEM space, 2GB */
 927                         pbm->mem_space.start = a;
 928                         pbm->mem_space.end = a + (0x80000000UL - 1UL);
 929                         pbm->mem_space.flags = IORESOURCE_MEM;
 930                         saw_mem = 1;
 931                         break;
 932
 933                 case 3:
 934                         /* XXX 64-bit MEM handling XXX */
 935
 936                 default:
 937                         break;
 938                 };
 939         }
 940
 941         if (!saw_io || !saw_mem) {
 942                 prom_printf("%s: Fatal error, missing %s PBM range.\n",
 943                             pbm->name,
 944                             (!saw_io ? "IO" : "MEM"));
 945                 prom_halt();
 946         }
 947
 948         printk("%s: PCI IO[%lx] MEM[%lx]\n",
 949                pbm->name,
 950                pbm->io_space.start,
 951                pbm->mem_space.start);
 952 }
 953
 954 static void pbm_register_toplevel_resources(struct pci_controller_info *p,
 955                                             struct pci_pbm_info *pbm)
 956 {
 957         pbm->io_space.name = pbm->mem_space.name = pbm->name;
 958
 959         request_resource(&ioport_resource, &pbm->io_space);
 960         request_resource(&iomem_resource, &pbm->mem_space);
 961         pci_register_legacy_regions(&pbm->io_space,
 962                                     &pbm->mem_space);
 963 }
 964
 965 static unsigned long probe_existing_entries(struct pci_pbm_info *pbm,
 966                                             struct pci_iommu *iommu)
 967 {
 968         struct pci_iommu_arena *arena = &iommu->arena;
 969         unsigned long i, cnt = 0;
 970         u32 devhandle;
 971
 972         devhandle = pbm->devhandle;
 973         for (i = 0; i < arena->limit; i++) {
 974                 unsigned long ret, io_attrs, ra;
 975
 976                 ret = pci_sun4v_iommu_getmap(devhandle,
 977                                              HV_PCI_TSBID(0, i),
 978                                              &io_attrs, &ra);
 979                 if (ret == HV_EOK) {
 980                         if (page_in_phys_avail(ra)) {
 981                                 pci_sun4v_iommu_demap(devhandle,
 982                                                       HV_PCI_TSBID(0, i), 1);
 983                         } else {
 984                                 cnt++;
 985                                 __set_bit(i, arena->map);
 986                         }
 987                 }
 988         }
 989
 990         return cnt;
 991 }
 992
 993 static void pci_sun4v_iommu_init(struct pci_pbm_info *pbm)
 994 {
 995         struct pci_iommu *iommu = pbm->iommu;
 996         struct property *prop;
 997         unsigned long num_tsb_entries, sz;
 998         u32 vdma[2], dma_mask, dma_offset;
 999         int tsbsize;
1000
1001         prop = of_find_property(pbm->prom_node, "virtual-dma", NULL);
1002         if (prop) {
1003                 u32 *val = prop->value;
1004
1005                 vdma[0] = val[0];
1006                 vdma[1] = val[1];
1007         } else {
1008                 /* No property, use default values. */
1009                 vdma[0] = 0x80000000;
1010                 vdma[1] = 0x80000000;
1011         }
1012
1013         dma_mask = vdma[0];
1014         switch (vdma[1]) {
1015                 case 0x20000000:
1016                         dma_mask |= 0x1fffffff;
1017                         tsbsize = 64;
1018                         break;
1019
1020                 case 0x40000000:
1021                         dma_mask |= 0x3fffffff;
1022                         tsbsize = 128;
1023                         break;
1024
1025                 case 0x80000000:
1026                         dma_mask |= 0x7fffffff;
1027                         tsbsize = 256;
1028                         break;
1029
1030                 default:
1031                         prom_printf("PCI-SUN4V: strange virtual-dma size.\n");
1032                         prom_halt();
1033         };
1034
1035         tsbsize *= (8 * 1024);
1036
1037         num_tsb_entries = tsbsize / sizeof(iopte_t);
1038
1039         dma_offset = vdma[0];
1040
1041         /* Setup initial software IOMMU state. */
1042         spin_lock_init(&iommu->lock);
1043         iommu->ctx_lowest_free = 1;
1044         iommu->page_table_map_base = dma_offset;
1045         iommu->dma_addr_mask = dma_mask;
1046
1047         /* Allocate and initialize the free area map.  */
1048         sz = num_tsb_entries / 8;
1049         sz = (sz + 7UL) & ~7UL;
1050         iommu->arena.map = kzalloc(sz, GFP_KERNEL);
1051         if (!iommu->arena.map) {
1052                 prom_printf("PCI_IOMMU: Error, kmalloc(arena.map) failed.\n");
1053                 prom_halt();
1054         }
1055         iommu->arena.limit = num_tsb_entries;
1056
1057         sz = probe_existing_entries(pbm, iommu);
1058         if (sz)
1059                 printk("%s: Imported %lu TSB entries from OBP\n",
1060                        pbm->name, sz);
1061 }
1062
1063 static void pci_sun4v_get_bus_range(struct pci_pbm_info *pbm)
1064 {
1065         struct property *prop;
1066         unsigned int *busrange;
1067
1068         prop = of_find_property(pbm->prom_node, "bus-range", NULL);
1069
1070         busrange = prop->value;
1071
1072         pbm->pci_first_busno = busrange[0];
1073         pbm->pci_last_busno = busrange[1];
1074
1075 }
1076
1077 static void pci_sun4v_pbm_init(struct pci_controller_info *p, struct device_node *dp, u32 devhandle)
1078 {
1079         struct pci_pbm_info *pbm;
1080         struct property *prop;
1081         int len, i;
1082
1083         if (devhandle & 0x40)
1084                 pbm = &p->pbm_B;
1085         else
1086                 pbm = &p->pbm_A;
1087
1088         pbm->parent = p;
1089         pbm->prom_node = dp;
1090         pbm->pci_first_slot = 1;
1091
1092         pbm->devhandle = devhandle;
1093
1094         pbm->name = dp->full_name;
1095
1096         printk("%s: SUN4V PCI Bus Module\n", pbm->name);
1097
1098         prop = of_find_property(dp, "ranges", &len);
1099         pbm->pbm_ranges = prop->value;
1100         pbm->num_pbm_ranges =
1101                 (len / sizeof(struct linux_prom_pci_ranges));
1102
1103         /* Mask out the top 8 bits of the ranges, leaving the real
1104          * physical address.
1105          */
1106         for (i = 0; i < pbm->num_pbm_ranges; i++)
1107                 pbm->pbm_ranges[i].parent_phys_hi &= 0x0fffffff;
1108
1109         pci_sun4v_determine_mem_io_space(pbm);
1110         pbm_register_toplevel_resources(p, pbm);
1111
1112         prop = of_find_property(dp, "interrupt-map", &len);
1113         pbm->pbm_intmap = prop->value;
1114         pbm->num_pbm_intmap =
1115                 (len / sizeof(struct linux_prom_pci_intmap));
1116
1117         prop = of_find_property(dp, "interrupt-map-mask", NULL);
1118         pbm->pbm_intmask = prop->value;
1119
1120         pci_sun4v_get_bus_range(pbm);
1121         pci_sun4v_iommu_init(pbm);
1122
1123         pdev_htab_populate(pbm);
1124 }
1125
1126 void sun4v_pci_init(struct device_node *dp, char *model_name)
1127 {
1128         struct pci_controller_info *p;
1129         struct pci_iommu *iommu;
1130         struct property *prop;
1131         struct linux_prom64_registers *regs;
1132         u32 devhandle;
1133         int i;
1134
1135         prop = of_find_property(dp, "reg", NULL);
1136         regs = prop->value;
1137
1138         devhandle = (regs->phys_addr >> 32UL) & 0x0fffffff;
1139
1140         for (p = pci_controller_root; p; p = p->next) {
1141                 struct pci_pbm_info *pbm;
1142
1143                 if (p->pbm_A.prom_node && p->pbm_B.prom_node)
1144                         continue;
1145
1146                 pbm = (p->pbm_A.prom_node ?
1147                        &p->pbm_A :
1148                        &p->pbm_B);
1149
1150                 if (pbm->devhandle == (devhandle ^ 0x40)) {
1151                         pci_sun4v_pbm_init(p, dp, devhandle);
1152                         return;
1153                 }
1154         }
1155
1156         for_each_possible_cpu(i) {
1157                 unsigned long page = get_zeroed_page(GFP_ATOMIC);
1158
1159                 if (!page)
1160                         goto fatal_memory_error;
1161
1162                 per_cpu(pci_iommu_batch, i).pglist = (u64 *) page;
1163         }
1164
1165         p = kzalloc(sizeof(struct pci_controller_info), GFP_ATOMIC);
1166         if (!p)
1167                 goto fatal_memory_error;
1168
1169         iommu = kzalloc(sizeof(struct pci_iommu), GFP_ATOMIC);
1170         if (!iommu)
1171                 goto fatal_memory_error;
1172
1173         p->pbm_A.iommu = iommu;
1174
1175         iommu = kzalloc(sizeof(struct pci_iommu), GFP_ATOMIC);
1176         if (!iommu)
1177                 goto fatal_memory_error;
1178
1179         p->pbm_B.iommu = iommu;
1180
1181         p->next = pci_controller_root;
1182         pci_controller_root = p;
1183
1184         p->index = pci_num_controllers++;
1185         p->pbms_same_domain = 0;
1186
1187         p->scan_bus = pci_sun4v_scan_bus;
1188         p->base_address_update = pci_sun4v_base_address_update;
1189         p->resource_adjust = pci_sun4v_resource_adjust;
1190         p->pci_ops = &pci_sun4v_ops;
1191
1192         /* Like PSYCHO and SCHIZO we have a 2GB aligned area
1193          * for memory space.
1194          */
1195         pci_memspace_mask = 0x7fffffffUL;
1196
1197         pci_sun4v_pbm_init(p, dp, devhandle);
1198         return;
1199
1200 fatal_memory_error:
1201         prom_printf("SUN4V_PCI: Fatal memory allocation error.\n");
1202         prom_halt();
1203 }