Linux Audio

Check our new training course

Loading...
v4.17
   1// SPDX-License-Identifier: GPL-2.0
   2/* pci_sun4v.c: SUN4V specific PCI controller support.
   3 *
   4 * Copyright (C) 2006, 2007, 2008 David S. Miller (davem@davemloft.net)
   5 */
   6
   7#include <linux/kernel.h>
   8#include <linux/types.h>
   9#include <linux/pci.h>
  10#include <linux/init.h>
  11#include <linux/slab.h>
  12#include <linux/interrupt.h>
  13#include <linux/percpu.h>
  14#include <linux/irq.h>
  15#include <linux/msi.h>
  16#include <linux/export.h>
  17#include <linux/log2.h>
  18#include <linux/of_device.h>
  19#include <linux/iommu-common.h>
 
 
  20
  21#include <asm/iommu.h>
  22#include <asm/irq.h>
  23#include <asm/hypervisor.h>
  24#include <asm/prom.h>
  25
  26#include "pci_impl.h"
  27#include "iommu_common.h"
  28#include "kernel.h"
  29
  30#include "pci_sun4v.h"
  31
  32#define DRIVER_NAME	"pci_sun4v"
  33#define PFX		DRIVER_NAME ": "
  34
  35static unsigned long vpci_major;
  36static unsigned long vpci_minor;
  37
  38struct vpci_version {
  39	unsigned long major;
  40	unsigned long minor;
  41};
  42
  43/* Ordered from largest major to lowest */
  44static struct vpci_version vpci_versions[] = {
  45	{ .major = 2, .minor = 0 },
  46	{ .major = 1, .minor = 1 },
  47};
  48
  49static unsigned long vatu_major = 1;
  50static unsigned long vatu_minor = 1;
  51
  52#define PGLIST_NENTS	(PAGE_SIZE / sizeof(u64))
  53
  54struct iommu_batch {
  55	struct device	*dev;		/* Device mapping is for.	*/
  56	unsigned long	prot;		/* IOMMU page protections	*/
  57	unsigned long	entry;		/* Index into IOTSB.		*/
  58	u64		*pglist;	/* List of physical pages	*/
  59	unsigned long	npages;		/* Number of pages in list.	*/
  60};
  61
  62static DEFINE_PER_CPU(struct iommu_batch, iommu_batch);
  63static int iommu_batch_initialized;
  64
  65/* Interrupts must be disabled.  */
  66static inline void iommu_batch_start(struct device *dev, unsigned long prot, unsigned long entry)
  67{
  68	struct iommu_batch *p = this_cpu_ptr(&iommu_batch);
  69
  70	p->dev		= dev;
  71	p->prot		= prot;
  72	p->entry	= entry;
  73	p->npages	= 0;
  74}
  75
 
 
 
 
 
  76/* Interrupts must be disabled.  */
  77static long iommu_batch_flush(struct iommu_batch *p, u64 mask)
  78{
  79	struct pci_pbm_info *pbm = p->dev->archdata.host_controller;
  80	u64 *pglist = p->pglist;
  81	u64 index_count;
  82	unsigned long devhandle = pbm->devhandle;
  83	unsigned long prot = p->prot;
  84	unsigned long entry = p->entry;
  85	unsigned long npages = p->npages;
  86	unsigned long iotsb_num;
  87	unsigned long ret;
  88	long num;
  89
  90	/* VPCI maj=1, min=[0,1] only supports read and write */
  91	if (vpci_major < 2)
  92		prot &= (HV_PCI_MAP_ATTR_READ | HV_PCI_MAP_ATTR_WRITE);
  93
  94	while (npages != 0) {
  95		if (mask <= DMA_BIT_MASK(32)) {
  96			num = pci_sun4v_iommu_map(devhandle,
  97						  HV_PCI_TSBID(0, entry),
  98						  npages,
  99						  prot,
 100						  __pa(pglist));
 101			if (unlikely(num < 0)) {
 102				pr_err_ratelimited("%s: IOMMU map of [%08lx:%08llx:%lx:%lx:%lx] failed with status %ld\n",
 103						   __func__,
 104						   devhandle,
 105						   HV_PCI_TSBID(0, entry),
 106						   npages, prot, __pa(pglist),
 107						   num);
 108				return -1;
 109			}
 110		} else {
 111			index_count = HV_PCI_IOTSB_INDEX_COUNT(npages, entry),
 112			iotsb_num = pbm->iommu->atu->iotsb->iotsb_num;
 113			ret = pci_sun4v_iotsb_map(devhandle,
 114						  iotsb_num,
 115						  index_count,
 116						  prot,
 117						  __pa(pglist),
 118						  &num);
 119			if (unlikely(ret != HV_EOK)) {
 120				pr_err_ratelimited("%s: ATU map of [%08lx:%lx:%llx:%lx:%lx] failed with status %ld\n",
 121						   __func__,
 122						   devhandle, iotsb_num,
 123						   index_count, prot,
 124						   __pa(pglist), ret);
 125				return -1;
 126			}
 127		}
 128		entry += num;
 129		npages -= num;
 130		pglist += num;
 131	}
 132
 133	p->entry = entry;
 134	p->npages = 0;
 135
 136	return 0;
 137}
 138
 139static inline void iommu_batch_new_entry(unsigned long entry, u64 mask)
 140{
 141	struct iommu_batch *p = this_cpu_ptr(&iommu_batch);
 142
 143	if (p->entry + p->npages == entry)
 144		return;
 145	if (p->entry != ~0UL)
 146		iommu_batch_flush(p, mask);
 147	p->entry = entry;
 148}
 149
 150/* Interrupts must be disabled.  */
 151static inline long iommu_batch_add(u64 phys_page, u64 mask)
 152{
 153	struct iommu_batch *p = this_cpu_ptr(&iommu_batch);
 154
 155	BUG_ON(p->npages >= PGLIST_NENTS);
 156
 157	p->pglist[p->npages++] = phys_page;
 158	if (p->npages == PGLIST_NENTS)
 159		return iommu_batch_flush(p, mask);
 160
 161	return 0;
 162}
 163
 164/* Interrupts must be disabled.  */
 165static inline long iommu_batch_end(u64 mask)
 166{
 167	struct iommu_batch *p = this_cpu_ptr(&iommu_batch);
 168
 169	BUG_ON(p->npages >= PGLIST_NENTS);
 170
 171	return iommu_batch_flush(p, mask);
 172}
 173
 174static void *dma_4v_alloc_coherent(struct device *dev, size_t size,
 175				   dma_addr_t *dma_addrp, gfp_t gfp,
 176				   unsigned long attrs)
 177{
 178	u64 mask;
 179	unsigned long flags, order, first_page, npages, n;
 180	unsigned long prot = 0;
 181	struct iommu *iommu;
 182	struct atu *atu;
 183	struct iommu_map_table *tbl;
 184	struct page *page;
 185	void *ret;
 186	long entry;
 187	int nid;
 188
 189	size = IO_PAGE_ALIGN(size);
 190	order = get_order(size);
 191	if (unlikely(order >= MAX_ORDER))
 192		return NULL;
 193
 194	npages = size >> IO_PAGE_SHIFT;
 195
 196	if (attrs & DMA_ATTR_WEAK_ORDERING)
 197		prot = HV_PCI_MAP_ATTR_RELAXED_ORDER;
 198
 199	nid = dev->archdata.numa_node;
 200	page = alloc_pages_node(nid, gfp, order);
 201	if (unlikely(!page))
 202		return NULL;
 203
 204	first_page = (unsigned long) page_address(page);
 205	memset((char *)first_page, 0, PAGE_SIZE << order);
 206
 207	iommu = dev->archdata.iommu;
 208	atu = iommu->atu;
 209
 210	mask = dev->coherent_dma_mask;
 211	if (mask <= DMA_BIT_MASK(32))
 212		tbl = &iommu->tbl;
 213	else
 214		tbl = &atu->tbl;
 215
 216	entry = iommu_tbl_range_alloc(dev, tbl, npages, NULL,
 217				      (unsigned long)(-1), 0);
 218
 219	if (unlikely(entry == IOMMU_ERROR_CODE))
 220		goto range_alloc_fail;
 221
 222	*dma_addrp = (tbl->table_map_base + (entry << IO_PAGE_SHIFT));
 223	ret = (void *) first_page;
 224	first_page = __pa(first_page);
 225
 226	local_irq_save(flags);
 227
 228	iommu_batch_start(dev,
 229			  (HV_PCI_MAP_ATTR_READ | prot |
 230			   HV_PCI_MAP_ATTR_WRITE),
 231			  entry);
 232
 233	for (n = 0; n < npages; n++) {
 234		long err = iommu_batch_add(first_page + (n * PAGE_SIZE), mask);
 235		if (unlikely(err < 0L))
 236			goto iommu_map_fail;
 237	}
 238
 239	if (unlikely(iommu_batch_end(mask) < 0L))
 240		goto iommu_map_fail;
 241
 242	local_irq_restore(flags);
 243
 244	return ret;
 245
 246iommu_map_fail:
 247	local_irq_restore(flags);
 248	iommu_tbl_range_free(tbl, *dma_addrp, npages, IOMMU_ERROR_CODE);
 249
 250range_alloc_fail:
 251	free_pages(first_page, order);
 252	return NULL;
 253}
 254
 255unsigned long dma_4v_iotsb_bind(unsigned long devhandle,
 256				unsigned long iotsb_num,
 257				struct pci_bus *bus_dev)
 258{
 259	struct pci_dev *pdev;
 260	unsigned long err;
 261	unsigned int bus;
 262	unsigned int device;
 263	unsigned int fun;
 264
 265	list_for_each_entry(pdev, &bus_dev->devices, bus_list) {
 266		if (pdev->subordinate) {
 267			/* No need to bind pci bridge */
 268			dma_4v_iotsb_bind(devhandle, iotsb_num,
 269					  pdev->subordinate);
 270		} else {
 271			bus = bus_dev->number;
 272			device = PCI_SLOT(pdev->devfn);
 273			fun = PCI_FUNC(pdev->devfn);
 274			err = pci_sun4v_iotsb_bind(devhandle, iotsb_num,
 275						   HV_PCI_DEVICE_BUILD(bus,
 276								       device,
 277								       fun));
 278
 279			/* If bind fails for one device it is going to fail
 280			 * for rest of the devices because we are sharing
 281			 * IOTSB. So in case of failure simply return with
 282			 * error.
 283			 */
 284			if (err)
 285				return err;
 286		}
 287	}
 288
 289	return 0;
 290}
 291
 292static void dma_4v_iommu_demap(struct device *dev, unsigned long devhandle,
 293			       dma_addr_t dvma, unsigned long iotsb_num,
 294			       unsigned long entry, unsigned long npages)
 295{
 296	unsigned long num, flags;
 297	unsigned long ret;
 298
 299	local_irq_save(flags);
 300	do {
 301		if (dvma <= DMA_BIT_MASK(32)) {
 302			num = pci_sun4v_iommu_demap(devhandle,
 303						    HV_PCI_TSBID(0, entry),
 304						    npages);
 305		} else {
 306			ret = pci_sun4v_iotsb_demap(devhandle, iotsb_num,
 307						    entry, npages, &num);
 308			if (unlikely(ret != HV_EOK)) {
 309				pr_err_ratelimited("pci_iotsb_demap() failed with error: %ld\n",
 310						   ret);
 311			}
 312		}
 313		entry += num;
 314		npages -= num;
 315	} while (npages != 0);
 316	local_irq_restore(flags);
 317}
 318
 319static void dma_4v_free_coherent(struct device *dev, size_t size, void *cpu,
 320				 dma_addr_t dvma, unsigned long attrs)
 321{
 322	struct pci_pbm_info *pbm;
 323	struct iommu *iommu;
 324	struct atu *atu;
 325	struct iommu_map_table *tbl;
 326	unsigned long order, npages, entry;
 327	unsigned long iotsb_num;
 328	u32 devhandle;
 329
 330	npages = IO_PAGE_ALIGN(size) >> IO_PAGE_SHIFT;
 331	iommu = dev->archdata.iommu;
 332	pbm = dev->archdata.host_controller;
 333	atu = iommu->atu;
 334	devhandle = pbm->devhandle;
 335
 336	if (dvma <= DMA_BIT_MASK(32)) {
 337		tbl = &iommu->tbl;
 338		iotsb_num = 0; /* we don't care for legacy iommu */
 339	} else {
 340		tbl = &atu->tbl;
 341		iotsb_num = atu->iotsb->iotsb_num;
 342	}
 343	entry = ((dvma - tbl->table_map_base) >> IO_PAGE_SHIFT);
 344	dma_4v_iommu_demap(dev, devhandle, dvma, iotsb_num, entry, npages);
 345	iommu_tbl_range_free(tbl, dvma, npages, IOMMU_ERROR_CODE);
 346	order = get_order(size);
 347	if (order < 10)
 348		free_pages((unsigned long)cpu, order);
 349}
 350
 351static dma_addr_t dma_4v_map_page(struct device *dev, struct page *page,
 352				  unsigned long offset, size_t sz,
 353				  enum dma_data_direction direction,
 354				  unsigned long attrs)
 355{
 356	struct iommu *iommu;
 357	struct atu *atu;
 358	struct iommu_map_table *tbl;
 359	u64 mask;
 360	unsigned long flags, npages, oaddr;
 361	unsigned long i, base_paddr;
 362	unsigned long prot;
 363	dma_addr_t bus_addr, ret;
 364	long entry;
 365
 366	iommu = dev->archdata.iommu;
 367	atu = iommu->atu;
 368
 369	if (unlikely(direction == DMA_NONE))
 370		goto bad;
 371
 372	oaddr = (unsigned long)(page_address(page) + offset);
 373	npages = IO_PAGE_ALIGN(oaddr + sz) - (oaddr & IO_PAGE_MASK);
 374	npages >>= IO_PAGE_SHIFT;
 375
 376	mask = *dev->dma_mask;
 377	if (mask <= DMA_BIT_MASK(32))
 378		tbl = &iommu->tbl;
 379	else
 380		tbl = &atu->tbl;
 381
 382	entry = iommu_tbl_range_alloc(dev, tbl, npages, NULL,
 383				      (unsigned long)(-1), 0);
 384
 385	if (unlikely(entry == IOMMU_ERROR_CODE))
 386		goto bad;
 387
 388	bus_addr = (tbl->table_map_base + (entry << IO_PAGE_SHIFT));
 389	ret = bus_addr | (oaddr & ~IO_PAGE_MASK);
 390	base_paddr = __pa(oaddr & IO_PAGE_MASK);
 391	prot = HV_PCI_MAP_ATTR_READ;
 392	if (direction != DMA_TO_DEVICE)
 393		prot |= HV_PCI_MAP_ATTR_WRITE;
 394
 395	if (attrs & DMA_ATTR_WEAK_ORDERING)
 396		prot |= HV_PCI_MAP_ATTR_RELAXED_ORDER;
 397
 398	local_irq_save(flags);
 399
 400	iommu_batch_start(dev, prot, entry);
 401
 402	for (i = 0; i < npages; i++, base_paddr += IO_PAGE_SIZE) {
 403		long err = iommu_batch_add(base_paddr, mask);
 404		if (unlikely(err < 0L))
 405			goto iommu_map_fail;
 406	}
 407	if (unlikely(iommu_batch_end(mask) < 0L))
 408		goto iommu_map_fail;
 409
 410	local_irq_restore(flags);
 411
 412	return ret;
 413
 414bad:
 415	if (printk_ratelimit())
 416		WARN_ON(1);
 417	return SPARC_MAPPING_ERROR;
 418
 419iommu_map_fail:
 420	local_irq_restore(flags);
 421	iommu_tbl_range_free(tbl, bus_addr, npages, IOMMU_ERROR_CODE);
 422	return SPARC_MAPPING_ERROR;
 423}
 424
 425static void dma_4v_unmap_page(struct device *dev, dma_addr_t bus_addr,
 426			      size_t sz, enum dma_data_direction direction,
 427			      unsigned long attrs)
 428{
 429	struct pci_pbm_info *pbm;
 430	struct iommu *iommu;
 431	struct atu *atu;
 432	struct iommu_map_table *tbl;
 433	unsigned long npages;
 434	unsigned long iotsb_num;
 435	long entry;
 436	u32 devhandle;
 437
 438	if (unlikely(direction == DMA_NONE)) {
 439		if (printk_ratelimit())
 440			WARN_ON(1);
 441		return;
 442	}
 443
 444	iommu = dev->archdata.iommu;
 445	pbm = dev->archdata.host_controller;
 446	atu = iommu->atu;
 447	devhandle = pbm->devhandle;
 448
 449	npages = IO_PAGE_ALIGN(bus_addr + sz) - (bus_addr & IO_PAGE_MASK);
 450	npages >>= IO_PAGE_SHIFT;
 451	bus_addr &= IO_PAGE_MASK;
 452
 453	if (bus_addr <= DMA_BIT_MASK(32)) {
 454		iotsb_num = 0; /* we don't care for legacy iommu */
 455		tbl = &iommu->tbl;
 456	} else {
 457		iotsb_num = atu->iotsb->iotsb_num;
 458		tbl = &atu->tbl;
 459	}
 460	entry = (bus_addr - tbl->table_map_base) >> IO_PAGE_SHIFT;
 461	dma_4v_iommu_demap(dev, devhandle, bus_addr, iotsb_num, entry, npages);
 462	iommu_tbl_range_free(tbl, bus_addr, npages, IOMMU_ERROR_CODE);
 463}
 464
 465static int dma_4v_map_sg(struct device *dev, struct scatterlist *sglist,
 466			 int nelems, enum dma_data_direction direction,
 467			 unsigned long attrs)
 468{
 469	struct scatterlist *s, *outs, *segstart;
 470	unsigned long flags, handle, prot;
 471	dma_addr_t dma_next = 0, dma_addr;
 472	unsigned int max_seg_size;
 473	unsigned long seg_boundary_size;
 474	int outcount, incount, i;
 475	struct iommu *iommu;
 476	struct atu *atu;
 477	struct iommu_map_table *tbl;
 478	u64 mask;
 479	unsigned long base_shift;
 480	long err;
 481
 482	BUG_ON(direction == DMA_NONE);
 483
 484	iommu = dev->archdata.iommu;
 485	if (nelems == 0 || !iommu)
 486		return 0;
 487	atu = iommu->atu;
 488
 489	prot = HV_PCI_MAP_ATTR_READ;
 490	if (direction != DMA_TO_DEVICE)
 491		prot |= HV_PCI_MAP_ATTR_WRITE;
 492
 493	if (attrs & DMA_ATTR_WEAK_ORDERING)
 494		prot |= HV_PCI_MAP_ATTR_RELAXED_ORDER;
 495
 496	outs = s = segstart = &sglist[0];
 497	outcount = 1;
 498	incount = nelems;
 499	handle = 0;
 500
 501	/* Init first segment length for backout at failure */
 502	outs->dma_length = 0;
 503
 504	local_irq_save(flags);
 505
 506	iommu_batch_start(dev, prot, ~0UL);
 507
 508	max_seg_size = dma_get_max_seg_size(dev);
 509	seg_boundary_size = ALIGN(dma_get_seg_boundary(dev) + 1,
 510				  IO_PAGE_SIZE) >> IO_PAGE_SHIFT;
 511
 512	mask = *dev->dma_mask;
 513	if (mask <= DMA_BIT_MASK(32))
 514		tbl = &iommu->tbl;
 515	else
 516		tbl = &atu->tbl;
 517
 518	base_shift = tbl->table_map_base >> IO_PAGE_SHIFT;
 519
 520	for_each_sg(sglist, s, nelems, i) {
 521		unsigned long paddr, npages, entry, out_entry = 0, slen;
 522
 523		slen = s->length;
 524		/* Sanity check */
 525		if (slen == 0) {
 526			dma_next = 0;
 527			continue;
 528		}
 529		/* Allocate iommu entries for that segment */
 530		paddr = (unsigned long) SG_ENT_PHYS_ADDRESS(s);
 531		npages = iommu_num_pages(paddr, slen, IO_PAGE_SIZE);
 532		entry = iommu_tbl_range_alloc(dev, tbl, npages,
 533					      &handle, (unsigned long)(-1), 0);
 534
 535		/* Handle failure */
 536		if (unlikely(entry == IOMMU_ERROR_CODE)) {
 537			pr_err_ratelimited("iommu_alloc failed, iommu %p paddr %lx npages %lx\n",
 538					   tbl, paddr, npages);
 539			goto iommu_map_failed;
 540		}
 541
 542		iommu_batch_new_entry(entry, mask);
 543
 544		/* Convert entry to a dma_addr_t */
 545		dma_addr = tbl->table_map_base + (entry << IO_PAGE_SHIFT);
 546		dma_addr |= (s->offset & ~IO_PAGE_MASK);
 547
 548		/* Insert into HW table */
 549		paddr &= IO_PAGE_MASK;
 550		while (npages--) {
 551			err = iommu_batch_add(paddr, mask);
 552			if (unlikely(err < 0L))
 553				goto iommu_map_failed;
 554			paddr += IO_PAGE_SIZE;
 555		}
 556
 557		/* If we are in an open segment, try merging */
 558		if (segstart != s) {
 559			/* We cannot merge if:
 560			 * - allocated dma_addr isn't contiguous to previous allocation
 561			 */
 562			if ((dma_addr != dma_next) ||
 563			    (outs->dma_length + s->length > max_seg_size) ||
 564			    (is_span_boundary(out_entry, base_shift,
 565					      seg_boundary_size, outs, s))) {
 566				/* Can't merge: create a new segment */
 567				segstart = s;
 568				outcount++;
 569				outs = sg_next(outs);
 570			} else {
 571				outs->dma_length += s->length;
 572			}
 573		}
 574
 575		if (segstart == s) {
 576			/* This is a new segment, fill entries */
 577			outs->dma_address = dma_addr;
 578			outs->dma_length = slen;
 579			out_entry = entry;
 580		}
 581
 582		/* Calculate next page pointer for contiguous check */
 583		dma_next = dma_addr + slen;
 584	}
 585
 586	err = iommu_batch_end(mask);
 587
 588	if (unlikely(err < 0L))
 589		goto iommu_map_failed;
 590
 591	local_irq_restore(flags);
 592
 593	if (outcount < incount) {
 594		outs = sg_next(outs);
 595		outs->dma_address = SPARC_MAPPING_ERROR;
 596		outs->dma_length = 0;
 597	}
 598
 599	return outcount;
 600
 601iommu_map_failed:
 602	for_each_sg(sglist, s, nelems, i) {
 603		if (s->dma_length != 0) {
 604			unsigned long vaddr, npages;
 605
 606			vaddr = s->dma_address & IO_PAGE_MASK;
 607			npages = iommu_num_pages(s->dma_address, s->dma_length,
 608						 IO_PAGE_SIZE);
 609			iommu_tbl_range_free(tbl, vaddr, npages,
 610					     IOMMU_ERROR_CODE);
 611			/* XXX demap? XXX */
 612			s->dma_address = SPARC_MAPPING_ERROR;
 613			s->dma_length = 0;
 614		}
 615		if (s == outs)
 616			break;
 617	}
 618	local_irq_restore(flags);
 619
 620	return 0;
 621}
 622
 623static void dma_4v_unmap_sg(struct device *dev, struct scatterlist *sglist,
 624			    int nelems, enum dma_data_direction direction,
 625			    unsigned long attrs)
 626{
 627	struct pci_pbm_info *pbm;
 628	struct scatterlist *sg;
 629	struct iommu *iommu;
 630	struct atu *atu;
 631	unsigned long flags, entry;
 632	unsigned long iotsb_num;
 633	u32 devhandle;
 634
 635	BUG_ON(direction == DMA_NONE);
 636
 637	iommu = dev->archdata.iommu;
 638	pbm = dev->archdata.host_controller;
 639	atu = iommu->atu;
 640	devhandle = pbm->devhandle;
 641	
 642	local_irq_save(flags);
 643
 644	sg = sglist;
 645	while (nelems--) {
 646		dma_addr_t dma_handle = sg->dma_address;
 647		unsigned int len = sg->dma_length;
 648		unsigned long npages;
 649		struct iommu_map_table *tbl;
 650		unsigned long shift = IO_PAGE_SHIFT;
 651
 652		if (!len)
 653			break;
 654		npages = iommu_num_pages(dma_handle, len, IO_PAGE_SIZE);
 655
 656		if (dma_handle <= DMA_BIT_MASK(32)) {
 657			iotsb_num = 0; /* we don't care for legacy iommu */
 658			tbl = &iommu->tbl;
 659		} else {
 660			iotsb_num = atu->iotsb->iotsb_num;
 661			tbl = &atu->tbl;
 662		}
 663		entry = ((dma_handle - tbl->table_map_base) >> shift);
 664		dma_4v_iommu_demap(dev, devhandle, dma_handle, iotsb_num,
 665				   entry, npages);
 666		iommu_tbl_range_free(tbl, dma_handle, npages,
 667				     IOMMU_ERROR_CODE);
 668		sg = sg_next(sg);
 669	}
 670
 671	local_irq_restore(flags);
 672}
 673
 674static int dma_4v_supported(struct device *dev, u64 device_mask)
 675{
 676	struct iommu *iommu = dev->archdata.iommu;
 677	u64 dma_addr_mask = iommu->dma_addr_mask;
 678
 679	if (device_mask > DMA_BIT_MASK(32)) {
 680		if (iommu->atu)
 681			dma_addr_mask = iommu->atu->dma_addr_mask;
 682		else
 683			return 0;
 684	}
 685
 686	if ((device_mask & dma_addr_mask) == dma_addr_mask)
 687		return 1;
 688	return pci64_dma_supported(to_pci_dev(dev), device_mask);
 689}
 690
 691static int dma_4v_mapping_error(struct device *dev, dma_addr_t dma_addr)
 692{
 693	return dma_addr == SPARC_MAPPING_ERROR;
 694}
 695
 696static const struct dma_map_ops sun4v_dma_ops = {
 697	.alloc				= dma_4v_alloc_coherent,
 698	.free				= dma_4v_free_coherent,
 699	.map_page			= dma_4v_map_page,
 700	.unmap_page			= dma_4v_unmap_page,
 701	.map_sg				= dma_4v_map_sg,
 702	.unmap_sg			= dma_4v_unmap_sg,
 703	.dma_supported			= dma_4v_supported,
 704	.mapping_error			= dma_4v_mapping_error,
 705};
 706
 707static void pci_sun4v_scan_bus(struct pci_pbm_info *pbm, struct device *parent)
 708{
 709	struct property *prop;
 710	struct device_node *dp;
 711
 712	dp = pbm->op->dev.of_node;
 713	prop = of_find_property(dp, "66mhz-capable", NULL);
 714	pbm->is_66mhz_capable = (prop != NULL);
 715	pbm->pci_bus = pci_scan_one_pbm(pbm, parent);
 716
 717	/* XXX register error interrupt handlers XXX */
 718}
 719
 720static unsigned long probe_existing_entries(struct pci_pbm_info *pbm,
 721					    struct iommu_map_table *iommu)
 722{
 723	struct iommu_pool *pool;
 724	unsigned long i, pool_nr, cnt = 0;
 725	u32 devhandle;
 726
 727	devhandle = pbm->devhandle;
 728	for (pool_nr = 0; pool_nr < iommu->nr_pools; pool_nr++) {
 729		pool = &(iommu->pools[pool_nr]);
 730		for (i = pool->start; i <= pool->end; i++) {
 731			unsigned long ret, io_attrs, ra;
 732
 733			ret = pci_sun4v_iommu_getmap(devhandle,
 734						     HV_PCI_TSBID(0, i),
 735						     &io_attrs, &ra);
 736			if (ret == HV_EOK) {
 737				if (page_in_phys_avail(ra)) {
 738					pci_sun4v_iommu_demap(devhandle,
 739							      HV_PCI_TSBID(0,
 740							      i), 1);
 741				} else {
 742					cnt++;
 743					__set_bit(i, iommu->map);
 744				}
 745			}
 746		}
 747	}
 748	return cnt;
 749}
 750
 751static int pci_sun4v_atu_alloc_iotsb(struct pci_pbm_info *pbm)
 752{
 753	struct atu *atu = pbm->iommu->atu;
 754	struct atu_iotsb *iotsb;
 755	void *table;
 756	u64 table_size;
 757	u64 iotsb_num;
 758	unsigned long order;
 759	unsigned long err;
 760
 761	iotsb = kzalloc(sizeof(*iotsb), GFP_KERNEL);
 762	if (!iotsb) {
 763		err = -ENOMEM;
 764		goto out_err;
 765	}
 766	atu->iotsb = iotsb;
 767
 768	/* calculate size of IOTSB */
 769	table_size = (atu->size / IO_PAGE_SIZE) * 8;
 770	order = get_order(table_size);
 771	table = (void *)__get_free_pages(GFP_KERNEL | __GFP_ZERO, order);
 772	if (!table) {
 773		err = -ENOMEM;
 774		goto table_failed;
 775	}
 776	iotsb->table = table;
 777	iotsb->ra = __pa(table);
 778	iotsb->dvma_size = atu->size;
 779	iotsb->dvma_base = atu->base;
 780	iotsb->table_size = table_size;
 781	iotsb->page_size = IO_PAGE_SIZE;
 782
 783	/* configure and register IOTSB with HV */
 784	err = pci_sun4v_iotsb_conf(pbm->devhandle,
 785				   iotsb->ra,
 786				   iotsb->table_size,
 787				   iotsb->page_size,
 788				   iotsb->dvma_base,
 789				   &iotsb_num);
 790	if (err) {
 791		pr_err(PFX "pci_iotsb_conf failed error: %ld\n", err);
 792		goto iotsb_conf_failed;
 793	}
 794	iotsb->iotsb_num = iotsb_num;
 795
 796	err = dma_4v_iotsb_bind(pbm->devhandle, iotsb_num, pbm->pci_bus);
 797	if (err) {
 798		pr_err(PFX "pci_iotsb_bind failed error: %ld\n", err);
 799		goto iotsb_conf_failed;
 800	}
 801
 802	return 0;
 803
 804iotsb_conf_failed:
 805	free_pages((unsigned long)table, order);
 806table_failed:
 807	kfree(iotsb);
 808out_err:
 809	return err;
 810}
 811
 812static int pci_sun4v_atu_init(struct pci_pbm_info *pbm)
 813{
 814	struct atu *atu = pbm->iommu->atu;
 815	unsigned long err;
 816	const u64 *ranges;
 817	u64 map_size, num_iotte;
 818	u64 dma_mask;
 819	const u32 *page_size;
 820	int len;
 821
 822	ranges = of_get_property(pbm->op->dev.of_node, "iommu-address-ranges",
 823				 &len);
 824	if (!ranges) {
 825		pr_err(PFX "No iommu-address-ranges\n");
 826		return -EINVAL;
 827	}
 828
 829	page_size = of_get_property(pbm->op->dev.of_node, "iommu-pagesizes",
 830				    NULL);
 831	if (!page_size) {
 832		pr_err(PFX "No iommu-pagesizes\n");
 833		return -EINVAL;
 834	}
 835
 836	/* There are 4 iommu-address-ranges supported. Each range is pair of
 837	 * {base, size}. The ranges[0] and ranges[1] are 32bit address space
 838	 * while ranges[2] and ranges[3] are 64bit space.  We want to use 64bit
 839	 * address ranges to support 64bit addressing. Because 'size' for
 840	 * address ranges[2] and ranges[3] are same we can select either of
 841	 * ranges[2] or ranges[3] for mapping. However due to 'size' is too
 842	 * large for OS to allocate IOTSB we are using fix size 32G
 843	 * (ATU_64_SPACE_SIZE) which is more than enough for all PCIe devices
 844	 * to share.
 845	 */
 846	atu->ranges = (struct atu_ranges *)ranges;
 847	atu->base = atu->ranges[3].base;
 848	atu->size = ATU_64_SPACE_SIZE;
 849
 850	/* Create IOTSB */
 851	err = pci_sun4v_atu_alloc_iotsb(pbm);
 852	if (err) {
 853		pr_err(PFX "Error creating ATU IOTSB\n");
 854		return err;
 855	}
 856
 857	/* Create ATU iommu map.
 858	 * One bit represents one iotte in IOTSB table.
 859	 */
 860	dma_mask = (roundup_pow_of_two(atu->size) - 1UL);
 861	num_iotte = atu->size / IO_PAGE_SIZE;
 862	map_size = num_iotte / 8;
 863	atu->tbl.table_map_base = atu->base;
 864	atu->dma_addr_mask = dma_mask;
 865	atu->tbl.map = kzalloc(map_size, GFP_KERNEL);
 866	if (!atu->tbl.map)
 867		return -ENOMEM;
 868
 869	iommu_tbl_pool_init(&atu->tbl, num_iotte, IO_PAGE_SHIFT,
 870			    NULL, false /* no large_pool */,
 871			    0 /* default npools */,
 872			    false /* want span boundary checking */);
 873
 874	return 0;
 875}
 876
 877static int pci_sun4v_iommu_init(struct pci_pbm_info *pbm)
 878{
 879	static const u32 vdma_default[] = { 0x80000000, 0x80000000 };
 880	struct iommu *iommu = pbm->iommu;
 881	unsigned long num_tsb_entries, sz;
 882	u32 dma_mask, dma_offset;
 883	const u32 *vdma;
 884
 885	vdma = of_get_property(pbm->op->dev.of_node, "virtual-dma", NULL);
 886	if (!vdma)
 887		vdma = vdma_default;
 888
 889	if ((vdma[0] | vdma[1]) & ~IO_PAGE_MASK) {
 890		printk(KERN_ERR PFX "Strange virtual-dma[%08x:%08x].\n",
 891		       vdma[0], vdma[1]);
 892		return -EINVAL;
 893	}
 894
 895	dma_mask = (roundup_pow_of_two(vdma[1]) - 1UL);
 896	num_tsb_entries = vdma[1] / IO_PAGE_SIZE;
 897
 898	dma_offset = vdma[0];
 899
 900	/* Setup initial software IOMMU state. */
 901	spin_lock_init(&iommu->lock);
 902	iommu->ctx_lowest_free = 1;
 903	iommu->tbl.table_map_base = dma_offset;
 904	iommu->dma_addr_mask = dma_mask;
 905
 906	/* Allocate and initialize the free area map.  */
 907	sz = (num_tsb_entries + 7) / 8;
 908	sz = (sz + 7UL) & ~7UL;
 909	iommu->tbl.map = kzalloc(sz, GFP_KERNEL);
 910	if (!iommu->tbl.map) {
 911		printk(KERN_ERR PFX "Error, kmalloc(arena.map) failed.\n");
 912		return -ENOMEM;
 913	}
 914	iommu_tbl_pool_init(&iommu->tbl, num_tsb_entries, IO_PAGE_SHIFT,
 915			    NULL, false /* no large_pool */,
 916			    0 /* default npools */,
 917			    false /* want span boundary checking */);
 918	sz = probe_existing_entries(pbm, &iommu->tbl);
 919	if (sz)
 920		printk("%s: Imported %lu TSB entries from OBP\n",
 921		       pbm->name, sz);
 922
 923	return 0;
 924}
 925
 926#ifdef CONFIG_PCI_MSI
 927struct pci_sun4v_msiq_entry {
 928	u64		version_type;
 929#define MSIQ_VERSION_MASK		0xffffffff00000000UL
 930#define MSIQ_VERSION_SHIFT		32
 931#define MSIQ_TYPE_MASK			0x00000000000000ffUL
 932#define MSIQ_TYPE_SHIFT			0
 933#define MSIQ_TYPE_NONE			0x00
 934#define MSIQ_TYPE_MSG			0x01
 935#define MSIQ_TYPE_MSI32			0x02
 936#define MSIQ_TYPE_MSI64			0x03
 937#define MSIQ_TYPE_INTX			0x08
 938#define MSIQ_TYPE_NONE2			0xff
 939
 940	u64		intx_sysino;
 941	u64		reserved1;
 942	u64		stick;
 943	u64		req_id;  /* bus/device/func */
 944#define MSIQ_REQID_BUS_MASK		0xff00UL
 945#define MSIQ_REQID_BUS_SHIFT		8
 946#define MSIQ_REQID_DEVICE_MASK		0x00f8UL
 947#define MSIQ_REQID_DEVICE_SHIFT		3
 948#define MSIQ_REQID_FUNC_MASK		0x0007UL
 949#define MSIQ_REQID_FUNC_SHIFT		0
 950
 951	u64		msi_address;
 952
 953	/* The format of this value is message type dependent.
 954	 * For MSI bits 15:0 are the data from the MSI packet.
 955	 * For MSI-X bits 31:0 are the data from the MSI packet.
 956	 * For MSG, the message code and message routing code where:
 957	 * 	bits 39:32 is the bus/device/fn of the msg target-id
 958	 *	bits 18:16 is the message routing code
 959	 *	bits 7:0 is the message code
 960	 * For INTx the low order 2-bits are:
 961	 *	00 - INTA
 962	 *	01 - INTB
 963	 *	10 - INTC
 964	 *	11 - INTD
 965	 */
 966	u64		msi_data;
 967
 968	u64		reserved2;
 969};
 970
 971static int pci_sun4v_get_head(struct pci_pbm_info *pbm, unsigned long msiqid,
 972			      unsigned long *head)
 973{
 974	unsigned long err, limit;
 975
 976	err = pci_sun4v_msiq_gethead(pbm->devhandle, msiqid, head);
 977	if (unlikely(err))
 978		return -ENXIO;
 979
 980	limit = pbm->msiq_ent_count * sizeof(struct pci_sun4v_msiq_entry);
 981	if (unlikely(*head >= limit))
 982		return -EFBIG;
 983
 984	return 0;
 985}
 986
 987static int pci_sun4v_dequeue_msi(struct pci_pbm_info *pbm,
 988				 unsigned long msiqid, unsigned long *head,
 989				 unsigned long *msi)
 990{
 991	struct pci_sun4v_msiq_entry *ep;
 992	unsigned long err, type;
 993
 994	/* Note: void pointer arithmetic, 'head' is a byte offset  */
 995	ep = (pbm->msi_queues + ((msiqid - pbm->msiq_first) *
 996				 (pbm->msiq_ent_count *
 997				  sizeof(struct pci_sun4v_msiq_entry))) +
 998	      *head);
 999
1000	if ((ep->version_type & MSIQ_TYPE_MASK) == 0)
1001		return 0;
1002
1003	type = (ep->version_type & MSIQ_TYPE_MASK) >> MSIQ_TYPE_SHIFT;
1004	if (unlikely(type != MSIQ_TYPE_MSI32 &&
1005		     type != MSIQ_TYPE_MSI64))
1006		return -EINVAL;
1007
1008	*msi = ep->msi_data;
1009
1010	err = pci_sun4v_msi_setstate(pbm->devhandle,
1011				     ep->msi_data /* msi_num */,
1012				     HV_MSISTATE_IDLE);
1013	if (unlikely(err))
1014		return -ENXIO;
1015
1016	/* Clear the entry.  */
1017	ep->version_type &= ~MSIQ_TYPE_MASK;
1018
1019	(*head) += sizeof(struct pci_sun4v_msiq_entry);
1020	if (*head >=
1021	    (pbm->msiq_ent_count * sizeof(struct pci_sun4v_msiq_entry)))
1022		*head = 0;
1023
1024	return 1;
1025}
1026
1027static int pci_sun4v_set_head(struct pci_pbm_info *pbm, unsigned long msiqid,
1028			      unsigned long head)
1029{
1030	unsigned long err;
1031
1032	err = pci_sun4v_msiq_sethead(pbm->devhandle, msiqid, head);
1033	if (unlikely(err))
1034		return -EINVAL;
1035
1036	return 0;
1037}
1038
1039static int pci_sun4v_msi_setup(struct pci_pbm_info *pbm, unsigned long msiqid,
1040			       unsigned long msi, int is_msi64)
1041{
1042	if (pci_sun4v_msi_setmsiq(pbm->devhandle, msi, msiqid,
1043				  (is_msi64 ?
1044				   HV_MSITYPE_MSI64 : HV_MSITYPE_MSI32)))
1045		return -ENXIO;
1046	if (pci_sun4v_msi_setstate(pbm->devhandle, msi, HV_MSISTATE_IDLE))
1047		return -ENXIO;
1048	if (pci_sun4v_msi_setvalid(pbm->devhandle, msi, HV_MSIVALID_VALID))
1049		return -ENXIO;
1050	return 0;
1051}
1052
1053static int pci_sun4v_msi_teardown(struct pci_pbm_info *pbm, unsigned long msi)
1054{
1055	unsigned long err, msiqid;
1056
1057	err = pci_sun4v_msi_getmsiq(pbm->devhandle, msi, &msiqid);
1058	if (err)
1059		return -ENXIO;
1060
1061	pci_sun4v_msi_setvalid(pbm->devhandle, msi, HV_MSIVALID_INVALID);
1062
1063	return 0;
1064}
1065
1066static int pci_sun4v_msiq_alloc(struct pci_pbm_info *pbm)
1067{
1068	unsigned long q_size, alloc_size, pages, order;
1069	int i;
1070
1071	q_size = pbm->msiq_ent_count * sizeof(struct pci_sun4v_msiq_entry);
1072	alloc_size = (pbm->msiq_num * q_size);
1073	order = get_order(alloc_size);
1074	pages = __get_free_pages(GFP_KERNEL | __GFP_COMP, order);
1075	if (pages == 0UL) {
1076		printk(KERN_ERR "MSI: Cannot allocate MSI queues (o=%lu).\n",
1077		       order);
1078		return -ENOMEM;
1079	}
1080	memset((char *)pages, 0, PAGE_SIZE << order);
1081	pbm->msi_queues = (void *) pages;
1082
1083	for (i = 0; i < pbm->msiq_num; i++) {
1084		unsigned long err, base = __pa(pages + (i * q_size));
1085		unsigned long ret1, ret2;
1086
1087		err = pci_sun4v_msiq_conf(pbm->devhandle,
1088					  pbm->msiq_first + i,
1089					  base, pbm->msiq_ent_count);
1090		if (err) {
1091			printk(KERN_ERR "MSI: msiq register fails (err=%lu)\n",
1092			       err);
1093			goto h_error;
1094		}
1095
1096		err = pci_sun4v_msiq_info(pbm->devhandle,
1097					  pbm->msiq_first + i,
1098					  &ret1, &ret2);
1099		if (err) {
1100			printk(KERN_ERR "MSI: Cannot read msiq (err=%lu)\n",
1101			       err);
1102			goto h_error;
1103		}
1104		if (ret1 != base || ret2 != pbm->msiq_ent_count) {
1105			printk(KERN_ERR "MSI: Bogus qconf "
1106			       "expected[%lx:%x] got[%lx:%lx]\n",
1107			       base, pbm->msiq_ent_count,
1108			       ret1, ret2);
1109			goto h_error;
1110		}
1111	}
1112
1113	return 0;
1114
1115h_error:
1116	free_pages(pages, order);
1117	return -EINVAL;
1118}
1119
1120static void pci_sun4v_msiq_free(struct pci_pbm_info *pbm)
1121{
1122	unsigned long q_size, alloc_size, pages, order;
1123	int i;
1124
1125	for (i = 0; i < pbm->msiq_num; i++) {
1126		unsigned long msiqid = pbm->msiq_first + i;
1127
1128		(void) pci_sun4v_msiq_conf(pbm->devhandle, msiqid, 0UL, 0);
1129	}
1130
1131	q_size = pbm->msiq_ent_count * sizeof(struct pci_sun4v_msiq_entry);
1132	alloc_size = (pbm->msiq_num * q_size);
1133	order = get_order(alloc_size);
1134
1135	pages = (unsigned long) pbm->msi_queues;
1136
1137	free_pages(pages, order);
1138
1139	pbm->msi_queues = NULL;
1140}
1141
1142static int pci_sun4v_msiq_build_irq(struct pci_pbm_info *pbm,
1143				    unsigned long msiqid,
1144				    unsigned long devino)
1145{
1146	unsigned int irq = sun4v_build_irq(pbm->devhandle, devino);
1147
1148	if (!irq)
1149		return -ENOMEM;
1150
1151	if (pci_sun4v_msiq_setvalid(pbm->devhandle, msiqid, HV_MSIQ_VALID))
1152		return -EINVAL;
1153	if (pci_sun4v_msiq_setstate(pbm->devhandle, msiqid, HV_MSIQSTATE_IDLE))
1154		return -EINVAL;
1155
1156	return irq;
1157}
1158
1159static const struct sparc64_msiq_ops pci_sun4v_msiq_ops = {
1160	.get_head	=	pci_sun4v_get_head,
1161	.dequeue_msi	=	pci_sun4v_dequeue_msi,
1162	.set_head	=	pci_sun4v_set_head,
1163	.msi_setup	=	pci_sun4v_msi_setup,
1164	.msi_teardown	=	pci_sun4v_msi_teardown,
1165	.msiq_alloc	=	pci_sun4v_msiq_alloc,
1166	.msiq_free	=	pci_sun4v_msiq_free,
1167	.msiq_build_irq	=	pci_sun4v_msiq_build_irq,
1168};
1169
1170static void pci_sun4v_msi_init(struct pci_pbm_info *pbm)
1171{
1172	sparc64_pbm_msi_init(pbm, &pci_sun4v_msiq_ops);
1173}
1174#else /* CONFIG_PCI_MSI */
1175static void pci_sun4v_msi_init(struct pci_pbm_info *pbm)
1176{
1177}
1178#endif /* !(CONFIG_PCI_MSI) */
1179
1180static int pci_sun4v_pbm_init(struct pci_pbm_info *pbm,
1181			      struct platform_device *op, u32 devhandle)
1182{
1183	struct device_node *dp = op->dev.of_node;
1184	int err;
1185
1186	pbm->numa_node = of_node_to_nid(dp);
1187
1188	pbm->pci_ops = &sun4v_pci_ops;
1189	pbm->config_space_reg_bits = 12;
1190
1191	pbm->index = pci_num_pbms++;
1192
1193	pbm->op = op;
1194
1195	pbm->devhandle = devhandle;
1196
1197	pbm->name = dp->full_name;
1198
1199	printk("%s: SUN4V PCI Bus Module\n", pbm->name);
1200	printk("%s: On NUMA node %d\n", pbm->name, pbm->numa_node);
1201
1202	pci_determine_mem_io_space(pbm);
1203
1204	pci_get_pbm_props(pbm);
1205
1206	err = pci_sun4v_iommu_init(pbm);
1207	if (err)
1208		return err;
1209
1210	pci_sun4v_msi_init(pbm);
1211
1212	pci_sun4v_scan_bus(pbm, &op->dev);
1213
1214	/* if atu_init fails its not complete failure.
1215	 * we can still continue using legacy iommu.
1216	 */
1217	if (pbm->iommu->atu) {
1218		err = pci_sun4v_atu_init(pbm);
1219		if (err) {
1220			kfree(pbm->iommu->atu);
1221			pbm->iommu->atu = NULL;
1222			pr_err(PFX "ATU init failed, err=%d\n", err);
1223		}
1224	}
1225
1226	pbm->next = pci_pbm_root;
1227	pci_pbm_root = pbm;
1228
1229	return 0;
1230}
1231
1232static int pci_sun4v_probe(struct platform_device *op)
1233{
1234	const struct linux_prom64_registers *regs;
1235	static int hvapi_negotiated = 0;
1236	struct pci_pbm_info *pbm;
1237	struct device_node *dp;
1238	struct iommu *iommu;
1239	struct atu *atu;
1240	u32 devhandle;
1241	int i, err = -ENODEV;
1242	static bool hv_atu = true;
1243
1244	dp = op->dev.of_node;
1245
1246	if (!hvapi_negotiated++) {
1247		for (i = 0; i < ARRAY_SIZE(vpci_versions); i++) {
1248			vpci_major = vpci_versions[i].major;
1249			vpci_minor = vpci_versions[i].minor;
1250
1251			err = sun4v_hvapi_register(HV_GRP_PCI, vpci_major,
1252						   &vpci_minor);
1253			if (!err)
1254				break;
1255		}
1256
1257		if (err) {
1258			pr_err(PFX "Could not register hvapi, err=%d\n", err);
1259			return err;
1260		}
1261		pr_info(PFX "Registered hvapi major[%lu] minor[%lu]\n",
1262			vpci_major, vpci_minor);
1263
1264		err = sun4v_hvapi_register(HV_GRP_ATU, vatu_major, &vatu_minor);
1265		if (err) {
1266			/* don't return an error if we fail to register the
1267			 * ATU group, but ATU hcalls won't be available.
1268			 */
1269			hv_atu = false;
1270		} else {
1271			pr_info(PFX "Registered hvapi ATU major[%lu] minor[%lu]\n",
1272				vatu_major, vatu_minor);
1273		}
1274
1275		dma_ops = &sun4v_dma_ops;
1276	}
1277
1278	regs = of_get_property(dp, "reg", NULL);
1279	err = -ENODEV;
1280	if (!regs) {
1281		printk(KERN_ERR PFX "Could not find config registers\n");
1282		goto out_err;
1283	}
1284	devhandle = (regs->phys_addr >> 32UL) & 0x0fffffff;
1285
1286	err = -ENOMEM;
1287	if (!iommu_batch_initialized) {
1288		for_each_possible_cpu(i) {
1289			unsigned long page = get_zeroed_page(GFP_KERNEL);
1290
1291			if (!page)
1292				goto out_err;
1293
1294			per_cpu(iommu_batch, i).pglist = (u64 *) page;
1295		}
1296		iommu_batch_initialized = 1;
1297	}
1298
1299	pbm = kzalloc(sizeof(*pbm), GFP_KERNEL);
1300	if (!pbm) {
1301		printk(KERN_ERR PFX "Could not allocate pci_pbm_info\n");
1302		goto out_err;
1303	}
1304
1305	iommu = kzalloc(sizeof(struct iommu), GFP_KERNEL);
1306	if (!iommu) {
1307		printk(KERN_ERR PFX "Could not allocate pbm iommu\n");
1308		goto out_free_controller;
1309	}
1310
1311	pbm->iommu = iommu;
1312	iommu->atu = NULL;
1313	if (hv_atu) {
1314		atu = kzalloc(sizeof(*atu), GFP_KERNEL);
1315		if (!atu)
1316			pr_err(PFX "Could not allocate atu\n");
1317		else
1318			iommu->atu = atu;
1319	}
1320
1321	err = pci_sun4v_pbm_init(pbm, op, devhandle);
1322	if (err)
1323		goto out_free_iommu;
1324
1325	dev_set_drvdata(&op->dev, pbm);
1326
1327	return 0;
1328
1329out_free_iommu:
1330	kfree(iommu->atu);
1331	kfree(pbm->iommu);
1332
1333out_free_controller:
1334	kfree(pbm);
1335
1336out_err:
1337	return err;
1338}
1339
1340static const struct of_device_id pci_sun4v_match[] = {
1341	{
1342		.name = "pci",
1343		.compatible = "SUNW,sun4v-pci",
1344	},
1345	{},
1346};
1347
1348static struct platform_driver pci_sun4v_driver = {
1349	.driver = {
1350		.name = DRIVER_NAME,
1351		.of_match_table = pci_sun4v_match,
1352	},
1353	.probe		= pci_sun4v_probe,
1354};
1355
1356static int __init pci_sun4v_init(void)
1357{
1358	return platform_driver_register(&pci_sun4v_driver);
1359}
1360
1361subsys_initcall(pci_sun4v_init);
v6.13.7
   1// SPDX-License-Identifier: GPL-2.0
   2/* pci_sun4v.c: SUN4V specific PCI controller support.
   3 *
   4 * Copyright (C) 2006, 2007, 2008 David S. Miller (davem@davemloft.net)
   5 */
   6
   7#include <linux/kernel.h>
   8#include <linux/types.h>
   9#include <linux/pci.h>
  10#include <linux/init.h>
  11#include <linux/slab.h>
  12#include <linux/interrupt.h>
  13#include <linux/percpu.h>
  14#include <linux/irq.h>
  15#include <linux/msi.h>
  16#include <linux/export.h>
  17#include <linux/log2.h>
  18#include <linux/of.h>
  19#include <linux/platform_device.h>
  20#include <linux/dma-map-ops.h>
  21#include <asm/iommu-common.h>
  22
  23#include <asm/iommu.h>
  24#include <asm/irq.h>
  25#include <asm/hypervisor.h>
  26#include <asm/prom.h>
  27
  28#include "pci_impl.h"
  29#include "iommu_common.h"
  30#include "kernel.h"
  31
  32#include "pci_sun4v.h"
  33
  34#define DRIVER_NAME	"pci_sun4v"
  35#define PFX		DRIVER_NAME ": "
  36
  37static unsigned long vpci_major;
  38static unsigned long vpci_minor;
  39
  40struct vpci_version {
  41	unsigned long major;
  42	unsigned long minor;
  43};
  44
  45/* Ordered from largest major to lowest */
  46static struct vpci_version vpci_versions[] = {
  47	{ .major = 2, .minor = 0 },
  48	{ .major = 1, .minor = 1 },
  49};
  50
  51static unsigned long vatu_major = 1;
  52static unsigned long vatu_minor = 1;
  53
  54#define PGLIST_NENTS	(PAGE_SIZE / sizeof(u64))
  55
  56struct iommu_batch {
  57	struct device	*dev;		/* Device mapping is for.	*/
  58	unsigned long	prot;		/* IOMMU page protections	*/
  59	unsigned long	entry;		/* Index into IOTSB.		*/
  60	u64		*pglist;	/* List of physical pages	*/
  61	unsigned long	npages;		/* Number of pages in list.	*/
  62};
  63
  64static DEFINE_PER_CPU(struct iommu_batch, iommu_batch);
  65static int iommu_batch_initialized;
  66
  67/* Interrupts must be disabled.  */
  68static inline void iommu_batch_start(struct device *dev, unsigned long prot, unsigned long entry)
  69{
  70	struct iommu_batch *p = this_cpu_ptr(&iommu_batch);
  71
  72	p->dev		= dev;
  73	p->prot		= prot;
  74	p->entry	= entry;
  75	p->npages	= 0;
  76}
  77
  78static inline bool iommu_use_atu(struct iommu *iommu, u64 mask)
  79{
  80	return iommu->atu && mask > DMA_BIT_MASK(32);
  81}
  82
  83/* Interrupts must be disabled.  */
  84static long iommu_batch_flush(struct iommu_batch *p, u64 mask)
  85{
  86	struct pci_pbm_info *pbm = p->dev->archdata.host_controller;
  87	u64 *pglist = p->pglist;
  88	u64 index_count;
  89	unsigned long devhandle = pbm->devhandle;
  90	unsigned long prot = p->prot;
  91	unsigned long entry = p->entry;
  92	unsigned long npages = p->npages;
  93	unsigned long iotsb_num;
  94	unsigned long ret;
  95	long num;
  96
  97	/* VPCI maj=1, min=[0,1] only supports read and write */
  98	if (vpci_major < 2)
  99		prot &= (HV_PCI_MAP_ATTR_READ | HV_PCI_MAP_ATTR_WRITE);
 100
 101	while (npages != 0) {
 102		if (!iommu_use_atu(pbm->iommu, mask)) {
 103			num = pci_sun4v_iommu_map(devhandle,
 104						  HV_PCI_TSBID(0, entry),
 105						  npages,
 106						  prot,
 107						  __pa(pglist));
 108			if (unlikely(num < 0)) {
 109				pr_err_ratelimited("%s: IOMMU map of [%08lx:%08llx:%lx:%lx:%lx] failed with status %ld\n",
 110						   __func__,
 111						   devhandle,
 112						   HV_PCI_TSBID(0, entry),
 113						   npages, prot, __pa(pglist),
 114						   num);
 115				return -1;
 116			}
 117		} else {
 118			index_count = HV_PCI_IOTSB_INDEX_COUNT(npages, entry),
 119			iotsb_num = pbm->iommu->atu->iotsb->iotsb_num;
 120			ret = pci_sun4v_iotsb_map(devhandle,
 121						  iotsb_num,
 122						  index_count,
 123						  prot,
 124						  __pa(pglist),
 125						  &num);
 126			if (unlikely(ret != HV_EOK)) {
 127				pr_err_ratelimited("%s: ATU map of [%08lx:%lx:%llx:%lx:%lx] failed with status %ld\n",
 128						   __func__,
 129						   devhandle, iotsb_num,
 130						   index_count, prot,
 131						   __pa(pglist), ret);
 132				return -1;
 133			}
 134		}
 135		entry += num;
 136		npages -= num;
 137		pglist += num;
 138	}
 139
 140	p->entry = entry;
 141	p->npages = 0;
 142
 143	return 0;
 144}
 145
 146static inline void iommu_batch_new_entry(unsigned long entry, u64 mask)
 147{
 148	struct iommu_batch *p = this_cpu_ptr(&iommu_batch);
 149
 150	if (p->entry + p->npages == entry)
 151		return;
 152	if (p->entry != ~0UL)
 153		iommu_batch_flush(p, mask);
 154	p->entry = entry;
 155}
 156
 157/* Interrupts must be disabled.  */
 158static inline long iommu_batch_add(u64 phys_page, u64 mask)
 159{
 160	struct iommu_batch *p = this_cpu_ptr(&iommu_batch);
 161
 162	BUG_ON(p->npages >= PGLIST_NENTS);
 163
 164	p->pglist[p->npages++] = phys_page;
 165	if (p->npages == PGLIST_NENTS)
 166		return iommu_batch_flush(p, mask);
 167
 168	return 0;
 169}
 170
 171/* Interrupts must be disabled.  */
 172static inline long iommu_batch_end(u64 mask)
 173{
 174	struct iommu_batch *p = this_cpu_ptr(&iommu_batch);
 175
 176	BUG_ON(p->npages >= PGLIST_NENTS);
 177
 178	return iommu_batch_flush(p, mask);
 179}
 180
 181static void *dma_4v_alloc_coherent(struct device *dev, size_t size,
 182				   dma_addr_t *dma_addrp, gfp_t gfp,
 183				   unsigned long attrs)
 184{
 185	u64 mask;
 186	unsigned long flags, order, first_page, npages, n;
 187	unsigned long prot = 0;
 188	struct iommu *iommu;
 
 189	struct iommu_map_table *tbl;
 190	struct page *page;
 191	void *ret;
 192	long entry;
 193	int nid;
 194
 195	size = IO_PAGE_ALIGN(size);
 196	order = get_order(size);
 197	if (unlikely(order > MAX_PAGE_ORDER))
 198		return NULL;
 199
 200	npages = size >> IO_PAGE_SHIFT;
 201
 202	if (attrs & DMA_ATTR_WEAK_ORDERING)
 203		prot = HV_PCI_MAP_ATTR_RELAXED_ORDER;
 204
 205	nid = dev->archdata.numa_node;
 206	page = alloc_pages_node(nid, gfp, order);
 207	if (unlikely(!page))
 208		return NULL;
 209
 210	first_page = (unsigned long) page_address(page);
 211	memset((char *)first_page, 0, PAGE_SIZE << order);
 212
 213	iommu = dev->archdata.iommu;
 
 
 214	mask = dev->coherent_dma_mask;
 215	if (!iommu_use_atu(iommu, mask))
 216		tbl = &iommu->tbl;
 217	else
 218		tbl = &iommu->atu->tbl;
 219
 220	entry = iommu_tbl_range_alloc(dev, tbl, npages, NULL,
 221				      (unsigned long)(-1), 0);
 222
 223	if (unlikely(entry == IOMMU_ERROR_CODE))
 224		goto range_alloc_fail;
 225
 226	*dma_addrp = (tbl->table_map_base + (entry << IO_PAGE_SHIFT));
 227	ret = (void *) first_page;
 228	first_page = __pa(first_page);
 229
 230	local_irq_save(flags);
 231
 232	iommu_batch_start(dev,
 233			  (HV_PCI_MAP_ATTR_READ | prot |
 234			   HV_PCI_MAP_ATTR_WRITE),
 235			  entry);
 236
 237	for (n = 0; n < npages; n++) {
 238		long err = iommu_batch_add(first_page + (n * PAGE_SIZE), mask);
 239		if (unlikely(err < 0L))
 240			goto iommu_map_fail;
 241	}
 242
 243	if (unlikely(iommu_batch_end(mask) < 0L))
 244		goto iommu_map_fail;
 245
 246	local_irq_restore(flags);
 247
 248	return ret;
 249
 250iommu_map_fail:
 251	local_irq_restore(flags);
 252	iommu_tbl_range_free(tbl, *dma_addrp, npages, IOMMU_ERROR_CODE);
 253
 254range_alloc_fail:
 255	free_pages(first_page, order);
 256	return NULL;
 257}
 258
 259static unsigned long dma_4v_iotsb_bind(unsigned long devhandle,
 260				       unsigned long iotsb_num,
 261				       struct pci_bus *bus_dev)
 262{
 263	struct pci_dev *pdev;
 264	unsigned long err;
 265	unsigned int bus;
 266	unsigned int device;
 267	unsigned int fun;
 268
 269	list_for_each_entry(pdev, &bus_dev->devices, bus_list) {
 270		if (pdev->subordinate) {
 271			/* No need to bind pci bridge */
 272			dma_4v_iotsb_bind(devhandle, iotsb_num,
 273					  pdev->subordinate);
 274		} else {
 275			bus = bus_dev->number;
 276			device = PCI_SLOT(pdev->devfn);
 277			fun = PCI_FUNC(pdev->devfn);
 278			err = pci_sun4v_iotsb_bind(devhandle, iotsb_num,
 279						   HV_PCI_DEVICE_BUILD(bus,
 280								       device,
 281								       fun));
 282
 283			/* If bind fails for one device it is going to fail
 284			 * for rest of the devices because we are sharing
 285			 * IOTSB. So in case of failure simply return with
 286			 * error.
 287			 */
 288			if (err)
 289				return err;
 290		}
 291	}
 292
 293	return 0;
 294}
 295
 296static void dma_4v_iommu_demap(struct device *dev, unsigned long devhandle,
 297			       dma_addr_t dvma, unsigned long iotsb_num,
 298			       unsigned long entry, unsigned long npages)
 299{
 300	unsigned long num, flags;
 301	unsigned long ret;
 302
 303	local_irq_save(flags);
 304	do {
 305		if (dvma <= DMA_BIT_MASK(32)) {
 306			num = pci_sun4v_iommu_demap(devhandle,
 307						    HV_PCI_TSBID(0, entry),
 308						    npages);
 309		} else {
 310			ret = pci_sun4v_iotsb_demap(devhandle, iotsb_num,
 311						    entry, npages, &num);
 312			if (unlikely(ret != HV_EOK)) {
 313				pr_err_ratelimited("pci_iotsb_demap() failed with error: %ld\n",
 314						   ret);
 315			}
 316		}
 317		entry += num;
 318		npages -= num;
 319	} while (npages != 0);
 320	local_irq_restore(flags);
 321}
 322
 323static void dma_4v_free_coherent(struct device *dev, size_t size, void *cpu,
 324				 dma_addr_t dvma, unsigned long attrs)
 325{
 326	struct pci_pbm_info *pbm;
 327	struct iommu *iommu;
 328	struct atu *atu;
 329	struct iommu_map_table *tbl;
 330	unsigned long order, npages, entry;
 331	unsigned long iotsb_num;
 332	u32 devhandle;
 333
 334	npages = IO_PAGE_ALIGN(size) >> IO_PAGE_SHIFT;
 335	iommu = dev->archdata.iommu;
 336	pbm = dev->archdata.host_controller;
 337	atu = iommu->atu;
 338	devhandle = pbm->devhandle;
 339
 340	if (!iommu_use_atu(iommu, dvma)) {
 341		tbl = &iommu->tbl;
 342		iotsb_num = 0; /* we don't care for legacy iommu */
 343	} else {
 344		tbl = &atu->tbl;
 345		iotsb_num = atu->iotsb->iotsb_num;
 346	}
 347	entry = ((dvma - tbl->table_map_base) >> IO_PAGE_SHIFT);
 348	dma_4v_iommu_demap(dev, devhandle, dvma, iotsb_num, entry, npages);
 349	iommu_tbl_range_free(tbl, dvma, npages, IOMMU_ERROR_CODE);
 350	order = get_order(size);
 351	if (order < 10)
 352		free_pages((unsigned long)cpu, order);
 353}
 354
 355static dma_addr_t dma_4v_map_page(struct device *dev, struct page *page,
 356				  unsigned long offset, size_t sz,
 357				  enum dma_data_direction direction,
 358				  unsigned long attrs)
 359{
 360	struct iommu *iommu;
 361	struct atu *atu;
 362	struct iommu_map_table *tbl;
 363	u64 mask;
 364	unsigned long flags, npages, oaddr;
 365	unsigned long i, base_paddr;
 366	unsigned long prot;
 367	dma_addr_t bus_addr, ret;
 368	long entry;
 369
 370	iommu = dev->archdata.iommu;
 371	atu = iommu->atu;
 372
 373	if (unlikely(direction == DMA_NONE))
 374		goto bad;
 375
 376	oaddr = (unsigned long)(page_address(page) + offset);
 377	npages = IO_PAGE_ALIGN(oaddr + sz) - (oaddr & IO_PAGE_MASK);
 378	npages >>= IO_PAGE_SHIFT;
 379
 380	mask = *dev->dma_mask;
 381	if (!iommu_use_atu(iommu, mask))
 382		tbl = &iommu->tbl;
 383	else
 384		tbl = &atu->tbl;
 385
 386	entry = iommu_tbl_range_alloc(dev, tbl, npages, NULL,
 387				      (unsigned long)(-1), 0);
 388
 389	if (unlikely(entry == IOMMU_ERROR_CODE))
 390		goto bad;
 391
 392	bus_addr = (tbl->table_map_base + (entry << IO_PAGE_SHIFT));
 393	ret = bus_addr | (oaddr & ~IO_PAGE_MASK);
 394	base_paddr = __pa(oaddr & IO_PAGE_MASK);
 395	prot = HV_PCI_MAP_ATTR_READ;
 396	if (direction != DMA_TO_DEVICE)
 397		prot |= HV_PCI_MAP_ATTR_WRITE;
 398
 399	if (attrs & DMA_ATTR_WEAK_ORDERING)
 400		prot |= HV_PCI_MAP_ATTR_RELAXED_ORDER;
 401
 402	local_irq_save(flags);
 403
 404	iommu_batch_start(dev, prot, entry);
 405
 406	for (i = 0; i < npages; i++, base_paddr += IO_PAGE_SIZE) {
 407		long err = iommu_batch_add(base_paddr, mask);
 408		if (unlikely(err < 0L))
 409			goto iommu_map_fail;
 410	}
 411	if (unlikely(iommu_batch_end(mask) < 0L))
 412		goto iommu_map_fail;
 413
 414	local_irq_restore(flags);
 415
 416	return ret;
 417
 418bad:
 419	if (printk_ratelimit())
 420		WARN_ON(1);
 421	return DMA_MAPPING_ERROR;
 422
 423iommu_map_fail:
 424	local_irq_restore(flags);
 425	iommu_tbl_range_free(tbl, bus_addr, npages, IOMMU_ERROR_CODE);
 426	return DMA_MAPPING_ERROR;
 427}
 428
 429static void dma_4v_unmap_page(struct device *dev, dma_addr_t bus_addr,
 430			      size_t sz, enum dma_data_direction direction,
 431			      unsigned long attrs)
 432{
 433	struct pci_pbm_info *pbm;
 434	struct iommu *iommu;
 435	struct atu *atu;
 436	struct iommu_map_table *tbl;
 437	unsigned long npages;
 438	unsigned long iotsb_num;
 439	long entry;
 440	u32 devhandle;
 441
 442	if (unlikely(direction == DMA_NONE)) {
 443		if (printk_ratelimit())
 444			WARN_ON(1);
 445		return;
 446	}
 447
 448	iommu = dev->archdata.iommu;
 449	pbm = dev->archdata.host_controller;
 450	atu = iommu->atu;
 451	devhandle = pbm->devhandle;
 452
 453	npages = IO_PAGE_ALIGN(bus_addr + sz) - (bus_addr & IO_PAGE_MASK);
 454	npages >>= IO_PAGE_SHIFT;
 455	bus_addr &= IO_PAGE_MASK;
 456
 457	if (bus_addr <= DMA_BIT_MASK(32)) {
 458		iotsb_num = 0; /* we don't care for legacy iommu */
 459		tbl = &iommu->tbl;
 460	} else {
 461		iotsb_num = atu->iotsb->iotsb_num;
 462		tbl = &atu->tbl;
 463	}
 464	entry = (bus_addr - tbl->table_map_base) >> IO_PAGE_SHIFT;
 465	dma_4v_iommu_demap(dev, devhandle, bus_addr, iotsb_num, entry, npages);
 466	iommu_tbl_range_free(tbl, bus_addr, npages, IOMMU_ERROR_CODE);
 467}
 468
 469static int dma_4v_map_sg(struct device *dev, struct scatterlist *sglist,
 470			 int nelems, enum dma_data_direction direction,
 471			 unsigned long attrs)
 472{
 473	struct scatterlist *s, *outs, *segstart;
 474	unsigned long flags, handle, prot;
 475	dma_addr_t dma_next = 0, dma_addr;
 476	unsigned int max_seg_size;
 477	unsigned long seg_boundary_size;
 478	int outcount, incount, i;
 479	struct iommu *iommu;
 480	struct atu *atu;
 481	struct iommu_map_table *tbl;
 482	u64 mask;
 483	unsigned long base_shift;
 484	long err;
 485
 486	BUG_ON(direction == DMA_NONE);
 487
 488	iommu = dev->archdata.iommu;
 489	if (nelems == 0 || !iommu)
 490		return -EINVAL;
 491	atu = iommu->atu;
 492
 493	prot = HV_PCI_MAP_ATTR_READ;
 494	if (direction != DMA_TO_DEVICE)
 495		prot |= HV_PCI_MAP_ATTR_WRITE;
 496
 497	if (attrs & DMA_ATTR_WEAK_ORDERING)
 498		prot |= HV_PCI_MAP_ATTR_RELAXED_ORDER;
 499
 500	outs = s = segstart = &sglist[0];
 501	outcount = 1;
 502	incount = nelems;
 503	handle = 0;
 504
 505	/* Init first segment length for backout at failure */
 506	outs->dma_length = 0;
 507
 508	local_irq_save(flags);
 509
 510	iommu_batch_start(dev, prot, ~0UL);
 511
 512	max_seg_size = dma_get_max_seg_size(dev);
 513	seg_boundary_size = dma_get_seg_boundary_nr_pages(dev, IO_PAGE_SHIFT);
 
 514
 515	mask = *dev->dma_mask;
 516	if (!iommu_use_atu(iommu, mask))
 517		tbl = &iommu->tbl;
 518	else
 519		tbl = &atu->tbl;
 520
 521	base_shift = tbl->table_map_base >> IO_PAGE_SHIFT;
 522
 523	for_each_sg(sglist, s, nelems, i) {
 524		unsigned long paddr, npages, entry, out_entry = 0, slen;
 525
 526		slen = s->length;
 527		/* Sanity check */
 528		if (slen == 0) {
 529			dma_next = 0;
 530			continue;
 531		}
 532		/* Allocate iommu entries for that segment */
 533		paddr = (unsigned long) SG_ENT_PHYS_ADDRESS(s);
 534		npages = iommu_num_pages(paddr, slen, IO_PAGE_SIZE);
 535		entry = iommu_tbl_range_alloc(dev, tbl, npages,
 536					      &handle, (unsigned long)(-1), 0);
 537
 538		/* Handle failure */
 539		if (unlikely(entry == IOMMU_ERROR_CODE)) {
 540			pr_err_ratelimited("iommu_alloc failed, iommu %p paddr %lx npages %lx\n",
 541					   tbl, paddr, npages);
 542			goto iommu_map_failed;
 543		}
 544
 545		iommu_batch_new_entry(entry, mask);
 546
 547		/* Convert entry to a dma_addr_t */
 548		dma_addr = tbl->table_map_base + (entry << IO_PAGE_SHIFT);
 549		dma_addr |= (s->offset & ~IO_PAGE_MASK);
 550
 551		/* Insert into HW table */
 552		paddr &= IO_PAGE_MASK;
 553		while (npages--) {
 554			err = iommu_batch_add(paddr, mask);
 555			if (unlikely(err < 0L))
 556				goto iommu_map_failed;
 557			paddr += IO_PAGE_SIZE;
 558		}
 559
 560		/* If we are in an open segment, try merging */
 561		if (segstart != s) {
 562			/* We cannot merge if:
 563			 * - allocated dma_addr isn't contiguous to previous allocation
 564			 */
 565			if ((dma_addr != dma_next) ||
 566			    (outs->dma_length + s->length > max_seg_size) ||
 567			    (is_span_boundary(out_entry, base_shift,
 568					      seg_boundary_size, outs, s))) {
 569				/* Can't merge: create a new segment */
 570				segstart = s;
 571				outcount++;
 572				outs = sg_next(outs);
 573			} else {
 574				outs->dma_length += s->length;
 575			}
 576		}
 577
 578		if (segstart == s) {
 579			/* This is a new segment, fill entries */
 580			outs->dma_address = dma_addr;
 581			outs->dma_length = slen;
 582			out_entry = entry;
 583		}
 584
 585		/* Calculate next page pointer for contiguous check */
 586		dma_next = dma_addr + slen;
 587	}
 588
 589	err = iommu_batch_end(mask);
 590
 591	if (unlikely(err < 0L))
 592		goto iommu_map_failed;
 593
 594	local_irq_restore(flags);
 595
 596	if (outcount < incount) {
 597		outs = sg_next(outs);
 
 598		outs->dma_length = 0;
 599	}
 600
 601	return outcount;
 602
 603iommu_map_failed:
 604	for_each_sg(sglist, s, nelems, i) {
 605		if (s->dma_length != 0) {
 606			unsigned long vaddr, npages;
 607
 608			vaddr = s->dma_address & IO_PAGE_MASK;
 609			npages = iommu_num_pages(s->dma_address, s->dma_length,
 610						 IO_PAGE_SIZE);
 611			iommu_tbl_range_free(tbl, vaddr, npages,
 612					     IOMMU_ERROR_CODE);
 613			/* XXX demap? XXX */
 
 614			s->dma_length = 0;
 615		}
 616		if (s == outs)
 617			break;
 618	}
 619	local_irq_restore(flags);
 620
 621	return -EINVAL;
 622}
 623
 624static void dma_4v_unmap_sg(struct device *dev, struct scatterlist *sglist,
 625			    int nelems, enum dma_data_direction direction,
 626			    unsigned long attrs)
 627{
 628	struct pci_pbm_info *pbm;
 629	struct scatterlist *sg;
 630	struct iommu *iommu;
 631	struct atu *atu;
 632	unsigned long flags, entry;
 633	unsigned long iotsb_num;
 634	u32 devhandle;
 635
 636	BUG_ON(direction == DMA_NONE);
 637
 638	iommu = dev->archdata.iommu;
 639	pbm = dev->archdata.host_controller;
 640	atu = iommu->atu;
 641	devhandle = pbm->devhandle;
 642	
 643	local_irq_save(flags);
 644
 645	sg = sglist;
 646	while (nelems--) {
 647		dma_addr_t dma_handle = sg->dma_address;
 648		unsigned int len = sg->dma_length;
 649		unsigned long npages;
 650		struct iommu_map_table *tbl;
 651		unsigned long shift = IO_PAGE_SHIFT;
 652
 653		if (!len)
 654			break;
 655		npages = iommu_num_pages(dma_handle, len, IO_PAGE_SIZE);
 656
 657		if (dma_handle <= DMA_BIT_MASK(32)) {
 658			iotsb_num = 0; /* we don't care for legacy iommu */
 659			tbl = &iommu->tbl;
 660		} else {
 661			iotsb_num = atu->iotsb->iotsb_num;
 662			tbl = &atu->tbl;
 663		}
 664		entry = ((dma_handle - tbl->table_map_base) >> shift);
 665		dma_4v_iommu_demap(dev, devhandle, dma_handle, iotsb_num,
 666				   entry, npages);
 667		iommu_tbl_range_free(tbl, dma_handle, npages,
 668				     IOMMU_ERROR_CODE);
 669		sg = sg_next(sg);
 670	}
 671
 672	local_irq_restore(flags);
 673}
 674
 675static int dma_4v_supported(struct device *dev, u64 device_mask)
 676{
 677	struct iommu *iommu = dev->archdata.iommu;
 
 
 
 
 
 
 
 
 678
 679	if (ali_sound_dma_hack(dev, device_mask))
 680		return 1;
 681	if (device_mask < iommu->dma_addr_mask)
 682		return 0;
 683	return 1;
 
 
 
 684}
 685
 686static const struct dma_map_ops sun4v_dma_ops = {
 687	.alloc				= dma_4v_alloc_coherent,
 688	.free				= dma_4v_free_coherent,
 689	.map_page			= dma_4v_map_page,
 690	.unmap_page			= dma_4v_unmap_page,
 691	.map_sg				= dma_4v_map_sg,
 692	.unmap_sg			= dma_4v_unmap_sg,
 693	.dma_supported			= dma_4v_supported,
 
 694};
 695
 696static void pci_sun4v_scan_bus(struct pci_pbm_info *pbm, struct device *parent)
 697{
 698	struct property *prop;
 699	struct device_node *dp;
 700
 701	dp = pbm->op->dev.of_node;
 702	prop = of_find_property(dp, "66mhz-capable", NULL);
 703	pbm->is_66mhz_capable = (prop != NULL);
 704	pbm->pci_bus = pci_scan_one_pbm(pbm, parent);
 705
 706	/* XXX register error interrupt handlers XXX */
 707}
 708
 709static unsigned long probe_existing_entries(struct pci_pbm_info *pbm,
 710					    struct iommu_map_table *iommu)
 711{
 712	struct iommu_pool *pool;
 713	unsigned long i, pool_nr, cnt = 0;
 714	u32 devhandle;
 715
 716	devhandle = pbm->devhandle;
 717	for (pool_nr = 0; pool_nr < iommu->nr_pools; pool_nr++) {
 718		pool = &(iommu->pools[pool_nr]);
 719		for (i = pool->start; i <= pool->end; i++) {
 720			unsigned long ret, io_attrs, ra;
 721
 722			ret = pci_sun4v_iommu_getmap(devhandle,
 723						     HV_PCI_TSBID(0, i),
 724						     &io_attrs, &ra);
 725			if (ret == HV_EOK) {
 726				if (page_in_phys_avail(ra)) {
 727					pci_sun4v_iommu_demap(devhandle,
 728							      HV_PCI_TSBID(0,
 729							      i), 1);
 730				} else {
 731					cnt++;
 732					__set_bit(i, iommu->map);
 733				}
 734			}
 735		}
 736	}
 737	return cnt;
 738}
 739
 740static int pci_sun4v_atu_alloc_iotsb(struct pci_pbm_info *pbm)
 741{
 742	struct atu *atu = pbm->iommu->atu;
 743	struct atu_iotsb *iotsb;
 744	void *table;
 745	u64 table_size;
 746	u64 iotsb_num;
 747	unsigned long order;
 748	unsigned long err;
 749
 750	iotsb = kzalloc(sizeof(*iotsb), GFP_KERNEL);
 751	if (!iotsb) {
 752		err = -ENOMEM;
 753		goto out_err;
 754	}
 755	atu->iotsb = iotsb;
 756
 757	/* calculate size of IOTSB */
 758	table_size = (atu->size / IO_PAGE_SIZE) * 8;
 759	order = get_order(table_size);
 760	table = (void *)__get_free_pages(GFP_KERNEL | __GFP_ZERO, order);
 761	if (!table) {
 762		err = -ENOMEM;
 763		goto table_failed;
 764	}
 765	iotsb->table = table;
 766	iotsb->ra = __pa(table);
 767	iotsb->dvma_size = atu->size;
 768	iotsb->dvma_base = atu->base;
 769	iotsb->table_size = table_size;
 770	iotsb->page_size = IO_PAGE_SIZE;
 771
 772	/* configure and register IOTSB with HV */
 773	err = pci_sun4v_iotsb_conf(pbm->devhandle,
 774				   iotsb->ra,
 775				   iotsb->table_size,
 776				   iotsb->page_size,
 777				   iotsb->dvma_base,
 778				   &iotsb_num);
 779	if (err) {
 780		pr_err(PFX "pci_iotsb_conf failed error: %ld\n", err);
 781		goto iotsb_conf_failed;
 782	}
 783	iotsb->iotsb_num = iotsb_num;
 784
 785	err = dma_4v_iotsb_bind(pbm->devhandle, iotsb_num, pbm->pci_bus);
 786	if (err) {
 787		pr_err(PFX "pci_iotsb_bind failed error: %ld\n", err);
 788		goto iotsb_conf_failed;
 789	}
 790
 791	return 0;
 792
 793iotsb_conf_failed:
 794	free_pages((unsigned long)table, order);
 795table_failed:
 796	kfree(iotsb);
 797out_err:
 798	return err;
 799}
 800
 801static int pci_sun4v_atu_init(struct pci_pbm_info *pbm)
 802{
 803	struct atu *atu = pbm->iommu->atu;
 804	unsigned long err;
 805	const u64 *ranges;
 806	u64 map_size, num_iotte;
 807	u64 dma_mask;
 808	const u32 *page_size;
 809	int len;
 810
 811	ranges = of_get_property(pbm->op->dev.of_node, "iommu-address-ranges",
 812				 &len);
 813	if (!ranges) {
 814		pr_err(PFX "No iommu-address-ranges\n");
 815		return -EINVAL;
 816	}
 817
 818	page_size = of_get_property(pbm->op->dev.of_node, "iommu-pagesizes",
 819				    NULL);
 820	if (!page_size) {
 821		pr_err(PFX "No iommu-pagesizes\n");
 822		return -EINVAL;
 823	}
 824
 825	/* There are 4 iommu-address-ranges supported. Each range is pair of
 826	 * {base, size}. The ranges[0] and ranges[1] are 32bit address space
 827	 * while ranges[2] and ranges[3] are 64bit space.  We want to use 64bit
 828	 * address ranges to support 64bit addressing. Because 'size' for
 829	 * address ranges[2] and ranges[3] are same we can select either of
 830	 * ranges[2] or ranges[3] for mapping. However due to 'size' is too
 831	 * large for OS to allocate IOTSB we are using fix size 32G
 832	 * (ATU_64_SPACE_SIZE) which is more than enough for all PCIe devices
 833	 * to share.
 834	 */
 835	atu->ranges = (struct atu_ranges *)ranges;
 836	atu->base = atu->ranges[3].base;
 837	atu->size = ATU_64_SPACE_SIZE;
 838
 839	/* Create IOTSB */
 840	err = pci_sun4v_atu_alloc_iotsb(pbm);
 841	if (err) {
 842		pr_err(PFX "Error creating ATU IOTSB\n");
 843		return err;
 844	}
 845
 846	/* Create ATU iommu map.
 847	 * One bit represents one iotte in IOTSB table.
 848	 */
 849	dma_mask = (roundup_pow_of_two(atu->size) - 1UL);
 850	num_iotte = atu->size / IO_PAGE_SIZE;
 851	map_size = num_iotte / 8;
 852	atu->tbl.table_map_base = atu->base;
 853	atu->dma_addr_mask = dma_mask;
 854	atu->tbl.map = kzalloc(map_size, GFP_KERNEL);
 855	if (!atu->tbl.map)
 856		return -ENOMEM;
 857
 858	iommu_tbl_pool_init(&atu->tbl, num_iotte, IO_PAGE_SHIFT,
 859			    NULL, false /* no large_pool */,
 860			    0 /* default npools */,
 861			    false /* want span boundary checking */);
 862
 863	return 0;
 864}
 865
 866static int pci_sun4v_iommu_init(struct pci_pbm_info *pbm)
 867{
 868	static const u32 vdma_default[] = { 0x80000000, 0x80000000 };
 869	struct iommu *iommu = pbm->iommu;
 870	unsigned long num_tsb_entries, sz;
 871	u32 dma_mask, dma_offset;
 872	const u32 *vdma;
 873
 874	vdma = of_get_property(pbm->op->dev.of_node, "virtual-dma", NULL);
 875	if (!vdma)
 876		vdma = vdma_default;
 877
 878	if ((vdma[0] | vdma[1]) & ~IO_PAGE_MASK) {
 879		printk(KERN_ERR PFX "Strange virtual-dma[%08x:%08x].\n",
 880		       vdma[0], vdma[1]);
 881		return -EINVAL;
 882	}
 883
 884	dma_mask = (roundup_pow_of_two(vdma[1]) - 1UL);
 885	num_tsb_entries = vdma[1] / IO_PAGE_SIZE;
 886
 887	dma_offset = vdma[0];
 888
 889	/* Setup initial software IOMMU state. */
 890	spin_lock_init(&iommu->lock);
 891	iommu->ctx_lowest_free = 1;
 892	iommu->tbl.table_map_base = dma_offset;
 893	iommu->dma_addr_mask = dma_mask;
 894
 895	/* Allocate and initialize the free area map.  */
 896	sz = (num_tsb_entries + 7) / 8;
 897	sz = (sz + 7UL) & ~7UL;
 898	iommu->tbl.map = kzalloc(sz, GFP_KERNEL);
 899	if (!iommu->tbl.map) {
 900		printk(KERN_ERR PFX "Error, kmalloc(arena.map) failed.\n");
 901		return -ENOMEM;
 902	}
 903	iommu_tbl_pool_init(&iommu->tbl, num_tsb_entries, IO_PAGE_SHIFT,
 904			    NULL, false /* no large_pool */,
 905			    0 /* default npools */,
 906			    false /* want span boundary checking */);
 907	sz = probe_existing_entries(pbm, &iommu->tbl);
 908	if (sz)
 909		printk("%s: Imported %lu TSB entries from OBP\n",
 910		       pbm->name, sz);
 911
 912	return 0;
 913}
 914
 915#ifdef CONFIG_PCI_MSI
 916struct pci_sun4v_msiq_entry {
 917	u64		version_type;
 918#define MSIQ_VERSION_MASK		0xffffffff00000000UL
 919#define MSIQ_VERSION_SHIFT		32
 920#define MSIQ_TYPE_MASK			0x00000000000000ffUL
 921#define MSIQ_TYPE_SHIFT			0
 922#define MSIQ_TYPE_NONE			0x00
 923#define MSIQ_TYPE_MSG			0x01
 924#define MSIQ_TYPE_MSI32			0x02
 925#define MSIQ_TYPE_MSI64			0x03
 926#define MSIQ_TYPE_INTX			0x08
 927#define MSIQ_TYPE_NONE2			0xff
 928
 929	u64		intx_sysino;
 930	u64		reserved1;
 931	u64		stick;
 932	u64		req_id;  /* bus/device/func */
 933#define MSIQ_REQID_BUS_MASK		0xff00UL
 934#define MSIQ_REQID_BUS_SHIFT		8
 935#define MSIQ_REQID_DEVICE_MASK		0x00f8UL
 936#define MSIQ_REQID_DEVICE_SHIFT		3
 937#define MSIQ_REQID_FUNC_MASK		0x0007UL
 938#define MSIQ_REQID_FUNC_SHIFT		0
 939
 940	u64		msi_address;
 941
 942	/* The format of this value is message type dependent.
 943	 * For MSI bits 15:0 are the data from the MSI packet.
 944	 * For MSI-X bits 31:0 are the data from the MSI packet.
 945	 * For MSG, the message code and message routing code where:
 946	 * 	bits 39:32 is the bus/device/fn of the msg target-id
 947	 *	bits 18:16 is the message routing code
 948	 *	bits 7:0 is the message code
 949	 * For INTx the low order 2-bits are:
 950	 *	00 - INTA
 951	 *	01 - INTB
 952	 *	10 - INTC
 953	 *	11 - INTD
 954	 */
 955	u64		msi_data;
 956
 957	u64		reserved2;
 958};
 959
 960static int pci_sun4v_get_head(struct pci_pbm_info *pbm, unsigned long msiqid,
 961			      unsigned long *head)
 962{
 963	unsigned long err, limit;
 964
 965	err = pci_sun4v_msiq_gethead(pbm->devhandle, msiqid, head);
 966	if (unlikely(err))
 967		return -ENXIO;
 968
 969	limit = pbm->msiq_ent_count * sizeof(struct pci_sun4v_msiq_entry);
 970	if (unlikely(*head >= limit))
 971		return -EFBIG;
 972
 973	return 0;
 974}
 975
 976static int pci_sun4v_dequeue_msi(struct pci_pbm_info *pbm,
 977				 unsigned long msiqid, unsigned long *head,
 978				 unsigned long *msi)
 979{
 980	struct pci_sun4v_msiq_entry *ep;
 981	unsigned long err, type;
 982
 983	/* Note: void pointer arithmetic, 'head' is a byte offset  */
 984	ep = (pbm->msi_queues + ((msiqid - pbm->msiq_first) *
 985				 (pbm->msiq_ent_count *
 986				  sizeof(struct pci_sun4v_msiq_entry))) +
 987	      *head);
 988
 989	if ((ep->version_type & MSIQ_TYPE_MASK) == 0)
 990		return 0;
 991
 992	type = (ep->version_type & MSIQ_TYPE_MASK) >> MSIQ_TYPE_SHIFT;
 993	if (unlikely(type != MSIQ_TYPE_MSI32 &&
 994		     type != MSIQ_TYPE_MSI64))
 995		return -EINVAL;
 996
 997	*msi = ep->msi_data;
 998
 999	err = pci_sun4v_msi_setstate(pbm->devhandle,
1000				     ep->msi_data /* msi_num */,
1001				     HV_MSISTATE_IDLE);
1002	if (unlikely(err))
1003		return -ENXIO;
1004
1005	/* Clear the entry.  */
1006	ep->version_type &= ~MSIQ_TYPE_MASK;
1007
1008	(*head) += sizeof(struct pci_sun4v_msiq_entry);
1009	if (*head >=
1010	    (pbm->msiq_ent_count * sizeof(struct pci_sun4v_msiq_entry)))
1011		*head = 0;
1012
1013	return 1;
1014}
1015
1016static int pci_sun4v_set_head(struct pci_pbm_info *pbm, unsigned long msiqid,
1017			      unsigned long head)
1018{
1019	unsigned long err;
1020
1021	err = pci_sun4v_msiq_sethead(pbm->devhandle, msiqid, head);
1022	if (unlikely(err))
1023		return -EINVAL;
1024
1025	return 0;
1026}
1027
1028static int pci_sun4v_msi_setup(struct pci_pbm_info *pbm, unsigned long msiqid,
1029			       unsigned long msi, int is_msi64)
1030{
1031	if (pci_sun4v_msi_setmsiq(pbm->devhandle, msi, msiqid,
1032				  (is_msi64 ?
1033				   HV_MSITYPE_MSI64 : HV_MSITYPE_MSI32)))
1034		return -ENXIO;
1035	if (pci_sun4v_msi_setstate(pbm->devhandle, msi, HV_MSISTATE_IDLE))
1036		return -ENXIO;
1037	if (pci_sun4v_msi_setvalid(pbm->devhandle, msi, HV_MSIVALID_VALID))
1038		return -ENXIO;
1039	return 0;
1040}
1041
1042static int pci_sun4v_msi_teardown(struct pci_pbm_info *pbm, unsigned long msi)
1043{
1044	unsigned long err, msiqid;
1045
1046	err = pci_sun4v_msi_getmsiq(pbm->devhandle, msi, &msiqid);
1047	if (err)
1048		return -ENXIO;
1049
1050	pci_sun4v_msi_setvalid(pbm->devhandle, msi, HV_MSIVALID_INVALID);
1051
1052	return 0;
1053}
1054
1055static int pci_sun4v_msiq_alloc(struct pci_pbm_info *pbm)
1056{
1057	unsigned long q_size, alloc_size, pages, order;
1058	int i;
1059
1060	q_size = pbm->msiq_ent_count * sizeof(struct pci_sun4v_msiq_entry);
1061	alloc_size = (pbm->msiq_num * q_size);
1062	order = get_order(alloc_size);
1063	pages = __get_free_pages(GFP_KERNEL | __GFP_COMP, order);
1064	if (pages == 0UL) {
1065		printk(KERN_ERR "MSI: Cannot allocate MSI queues (o=%lu).\n",
1066		       order);
1067		return -ENOMEM;
1068	}
1069	memset((char *)pages, 0, PAGE_SIZE << order);
1070	pbm->msi_queues = (void *) pages;
1071
1072	for (i = 0; i < pbm->msiq_num; i++) {
1073		unsigned long err, base = __pa(pages + (i * q_size));
1074		unsigned long ret1, ret2;
1075
1076		err = pci_sun4v_msiq_conf(pbm->devhandle,
1077					  pbm->msiq_first + i,
1078					  base, pbm->msiq_ent_count);
1079		if (err) {
1080			printk(KERN_ERR "MSI: msiq register fails (err=%lu)\n",
1081			       err);
1082			goto h_error;
1083		}
1084
1085		err = pci_sun4v_msiq_info(pbm->devhandle,
1086					  pbm->msiq_first + i,
1087					  &ret1, &ret2);
1088		if (err) {
1089			printk(KERN_ERR "MSI: Cannot read msiq (err=%lu)\n",
1090			       err);
1091			goto h_error;
1092		}
1093		if (ret1 != base || ret2 != pbm->msiq_ent_count) {
1094			printk(KERN_ERR "MSI: Bogus qconf "
1095			       "expected[%lx:%x] got[%lx:%lx]\n",
1096			       base, pbm->msiq_ent_count,
1097			       ret1, ret2);
1098			goto h_error;
1099		}
1100	}
1101
1102	return 0;
1103
1104h_error:
1105	free_pages(pages, order);
1106	return -EINVAL;
1107}
1108
1109static void pci_sun4v_msiq_free(struct pci_pbm_info *pbm)
1110{
1111	unsigned long q_size, alloc_size, pages, order;
1112	int i;
1113
1114	for (i = 0; i < pbm->msiq_num; i++) {
1115		unsigned long msiqid = pbm->msiq_first + i;
1116
1117		(void) pci_sun4v_msiq_conf(pbm->devhandle, msiqid, 0UL, 0);
1118	}
1119
1120	q_size = pbm->msiq_ent_count * sizeof(struct pci_sun4v_msiq_entry);
1121	alloc_size = (pbm->msiq_num * q_size);
1122	order = get_order(alloc_size);
1123
1124	pages = (unsigned long) pbm->msi_queues;
1125
1126	free_pages(pages, order);
1127
1128	pbm->msi_queues = NULL;
1129}
1130
1131static int pci_sun4v_msiq_build_irq(struct pci_pbm_info *pbm,
1132				    unsigned long msiqid,
1133				    unsigned long devino)
1134{
1135	unsigned int irq = sun4v_build_irq(pbm->devhandle, devino);
1136
1137	if (!irq)
1138		return -ENOMEM;
1139
1140	if (pci_sun4v_msiq_setvalid(pbm->devhandle, msiqid, HV_MSIQ_VALID))
1141		return -EINVAL;
1142	if (pci_sun4v_msiq_setstate(pbm->devhandle, msiqid, HV_MSIQSTATE_IDLE))
1143		return -EINVAL;
1144
1145	return irq;
1146}
1147
1148static const struct sparc64_msiq_ops pci_sun4v_msiq_ops = {
1149	.get_head	=	pci_sun4v_get_head,
1150	.dequeue_msi	=	pci_sun4v_dequeue_msi,
1151	.set_head	=	pci_sun4v_set_head,
1152	.msi_setup	=	pci_sun4v_msi_setup,
1153	.msi_teardown	=	pci_sun4v_msi_teardown,
1154	.msiq_alloc	=	pci_sun4v_msiq_alloc,
1155	.msiq_free	=	pci_sun4v_msiq_free,
1156	.msiq_build_irq	=	pci_sun4v_msiq_build_irq,
1157};
1158
1159static void pci_sun4v_msi_init(struct pci_pbm_info *pbm)
1160{
1161	sparc64_pbm_msi_init(pbm, &pci_sun4v_msiq_ops);
1162}
1163#else /* CONFIG_PCI_MSI */
1164static void pci_sun4v_msi_init(struct pci_pbm_info *pbm)
1165{
1166}
1167#endif /* !(CONFIG_PCI_MSI) */
1168
1169static int pci_sun4v_pbm_init(struct pci_pbm_info *pbm,
1170			      struct platform_device *op, u32 devhandle)
1171{
1172	struct device_node *dp = op->dev.of_node;
1173	int err;
1174
1175	pbm->numa_node = of_node_to_nid(dp);
1176
1177	pbm->pci_ops = &sun4v_pci_ops;
1178	pbm->config_space_reg_bits = 12;
1179
1180	pbm->index = pci_num_pbms++;
1181
1182	pbm->op = op;
1183
1184	pbm->devhandle = devhandle;
1185
1186	pbm->name = dp->full_name;
1187
1188	printk("%s: SUN4V PCI Bus Module\n", pbm->name);
1189	printk("%s: On NUMA node %d\n", pbm->name, pbm->numa_node);
1190
1191	pci_determine_mem_io_space(pbm);
1192
1193	pci_get_pbm_props(pbm);
1194
1195	err = pci_sun4v_iommu_init(pbm);
1196	if (err)
1197		return err;
1198
1199	pci_sun4v_msi_init(pbm);
1200
1201	pci_sun4v_scan_bus(pbm, &op->dev);
1202
1203	/* if atu_init fails its not complete failure.
1204	 * we can still continue using legacy iommu.
1205	 */
1206	if (pbm->iommu->atu) {
1207		err = pci_sun4v_atu_init(pbm);
1208		if (err) {
1209			kfree(pbm->iommu->atu);
1210			pbm->iommu->atu = NULL;
1211			pr_err(PFX "ATU init failed, err=%d\n", err);
1212		}
1213	}
1214
1215	pbm->next = pci_pbm_root;
1216	pci_pbm_root = pbm;
1217
1218	return 0;
1219}
1220
1221static int pci_sun4v_probe(struct platform_device *op)
1222{
1223	const struct linux_prom64_registers *regs;
1224	static int hvapi_negotiated = 0;
1225	struct pci_pbm_info *pbm;
1226	struct device_node *dp;
1227	struct iommu *iommu;
1228	struct atu *atu;
1229	u32 devhandle;
1230	int i, err = -ENODEV;
1231	static bool hv_atu = true;
1232
1233	dp = op->dev.of_node;
1234
1235	if (!hvapi_negotiated++) {
1236		for (i = 0; i < ARRAY_SIZE(vpci_versions); i++) {
1237			vpci_major = vpci_versions[i].major;
1238			vpci_minor = vpci_versions[i].minor;
1239
1240			err = sun4v_hvapi_register(HV_GRP_PCI, vpci_major,
1241						   &vpci_minor);
1242			if (!err)
1243				break;
1244		}
1245
1246		if (err) {
1247			pr_err(PFX "Could not register hvapi, err=%d\n", err);
1248			return err;
1249		}
1250		pr_info(PFX "Registered hvapi major[%lu] minor[%lu]\n",
1251			vpci_major, vpci_minor);
1252
1253		err = sun4v_hvapi_register(HV_GRP_ATU, vatu_major, &vatu_minor);
1254		if (err) {
1255			/* don't return an error if we fail to register the
1256			 * ATU group, but ATU hcalls won't be available.
1257			 */
1258			hv_atu = false;
1259		} else {
1260			pr_info(PFX "Registered hvapi ATU major[%lu] minor[%lu]\n",
1261				vatu_major, vatu_minor);
1262		}
1263
1264		dma_ops = &sun4v_dma_ops;
1265	}
1266
1267	regs = of_get_property(dp, "reg", NULL);
1268	err = -ENODEV;
1269	if (!regs) {
1270		printk(KERN_ERR PFX "Could not find config registers\n");
1271		goto out_err;
1272	}
1273	devhandle = (regs->phys_addr >> 32UL) & 0x0fffffff;
1274
1275	err = -ENOMEM;
1276	if (!iommu_batch_initialized) {
1277		for_each_possible_cpu(i) {
1278			unsigned long page = get_zeroed_page(GFP_KERNEL);
1279
1280			if (!page)
1281				goto out_err;
1282
1283			per_cpu(iommu_batch, i).pglist = (u64 *) page;
1284		}
1285		iommu_batch_initialized = 1;
1286	}
1287
1288	pbm = kzalloc(sizeof(*pbm), GFP_KERNEL);
1289	if (!pbm) {
1290		printk(KERN_ERR PFX "Could not allocate pci_pbm_info\n");
1291		goto out_err;
1292	}
1293
1294	iommu = kzalloc(sizeof(struct iommu), GFP_KERNEL);
1295	if (!iommu) {
1296		printk(KERN_ERR PFX "Could not allocate pbm iommu\n");
1297		goto out_free_controller;
1298	}
1299
1300	pbm->iommu = iommu;
1301	iommu->atu = NULL;
1302	if (hv_atu) {
1303		atu = kzalloc(sizeof(*atu), GFP_KERNEL);
1304		if (!atu)
1305			pr_err(PFX "Could not allocate atu\n");
1306		else
1307			iommu->atu = atu;
1308	}
1309
1310	err = pci_sun4v_pbm_init(pbm, op, devhandle);
1311	if (err)
1312		goto out_free_iommu;
1313
1314	dev_set_drvdata(&op->dev, pbm);
1315
1316	return 0;
1317
1318out_free_iommu:
1319	kfree(iommu->atu);
1320	kfree(pbm->iommu);
1321
1322out_free_controller:
1323	kfree(pbm);
1324
1325out_err:
1326	return err;
1327}
1328
1329static const struct of_device_id pci_sun4v_match[] = {
1330	{
1331		.name = "pci",
1332		.compatible = "SUNW,sun4v-pci",
1333	},
1334	{},
1335};
1336
1337static struct platform_driver pci_sun4v_driver = {
1338	.driver = {
1339		.name = DRIVER_NAME,
1340		.of_match_table = pci_sun4v_match,
1341	},
1342	.probe		= pci_sun4v_probe,
1343};
1344
1345static int __init pci_sun4v_init(void)
1346{
1347	return platform_driver_register(&pci_sun4v_driver);
1348}
1349
1350subsys_initcall(pci_sun4v_init);