[linux-2.6-block.git] / drivers / iommu / shmobile-iommu.c

/*
 * IOMMU for IPMMU/IPMMUI
 * Copyright (C) 2012  Hideki EIRAKU
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 */

#include <linux/dma-mapping.h>
#include <linux/io.h>
#include <linux/iommu.h>
#include <linux/platform_device.h>
#include <linux/sizes.h>
#include <linux/slab.h>
#include <asm/dma-iommu.h>
#include "shmobile-ipmmu.h"

#define L1_SIZE CONFIG_SHMOBILE_IOMMU_L1SIZE
#define L1_LEN (L1_SIZE / 4)
#define L1_ALIGN L1_SIZE
#define L2_SIZE SZ_1K
#define L2_LEN (L2_SIZE / 4)
#define L2_ALIGN L2_SIZE

struct shmobile_iommu_domain_pgtable {
	uint32_t *pgtable;
	dma_addr_t handle;
};

struct shmobile_iommu_archdata {
	struct list_head attached_list;
	struct dma_iommu_mapping *iommu_mapping;
	spinlock_t attach_lock;
	struct shmobile_iommu_domain *attached;
	int num_attached_devices;
	struct shmobile_ipmmu *ipmmu;
};

struct shmobile_iommu_domain {
	struct shmobile_iommu_domain_pgtable l1, l2[L1_LEN];
	spinlock_t map_lock;
	spinlock_t attached_list_lock;
	struct list_head attached_list;
	struct iommu_domain domain;
};

static struct shmobile_iommu_archdata *ipmmu_archdata;
static struct kmem_cache *l1cache, *l2cache;

static struct shmobile_iommu_domain *to_sh_domain(struct iommu_domain *dom)
{
	return container_of(dom, struct shmobile_iommu_domain, domain);
}

static int pgtable_alloc(struct shmobile_iommu_domain_pgtable *pgtable,
			 struct kmem_cache *cache, size_t size)
{
	pgtable->pgtable = kmem_cache_zalloc(cache, GFP_ATOMIC);
	if (!pgtable->pgtable)
		return -ENOMEM;
	pgtable->handle = dma_map_single(NULL, pgtable->pgtable, size,
					 DMA_TO_DEVICE);
	return 0;
}

static void pgtable_free(struct shmobile_iommu_domain_pgtable *pgtable,
			 struct kmem_cache *cache, size_t size)
{
	dma_unmap_single(NULL, pgtable->handle, size, DMA_TO_DEVICE);
	kmem_cache_free(cache, pgtable->pgtable);
}

static uint32_t pgtable_read(struct shmobile_iommu_domain_pgtable *pgtable,
			     unsigned int index)
{
	return pgtable->pgtable[index];
}

static void pgtable_write(struct shmobile_iommu_domain_pgtable *pgtable,
			  unsigned int index, unsigned int count, uint32_t val)
{
	unsigned int i;

	for (i = 0; i < count; i++)
		pgtable->pgtable[index + i] = val;
	dma_sync_single_for_device(NULL, pgtable->handle + index * sizeof(val),
				   sizeof(val) * count, DMA_TO_DEVICE);
}

static struct iommu_domain *shmobile_iommu_domain_alloc(unsigned type)
{
	struct shmobile_iommu_domain *sh_domain;
	int i, ret;

	if (type != IOMMU_DOMAIN_UNMANAGED)
		return NULL;

	sh_domain = kzalloc(sizeof(*sh_domain), GFP_KERNEL);
	if (!sh_domain)
		return NULL;
	ret = pgtable_alloc(&sh_domain->l1, l1cache, L1_SIZE);
	if (ret < 0) {
		kfree(sh_domain);
		return NULL;
	}
	for (i = 0; i < L1_LEN; i++)
		sh_domain->l2[i].pgtable = NULL;
	spin_lock_init(&sh_domain->map_lock);
	spin_lock_init(&sh_domain->attached_list_lock);
	INIT_LIST_HEAD(&sh_domain->attached_list);
	return &sh_domain->domain;
}

static void shmobile_iommu_domain_free(struct iommu_domain *domain)
{
	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
	int i;

	for (i = 0; i < L1_LEN; i++) {
		if (sh_domain->l2[i].pgtable)
			pgtable_free(&sh_domain->l2[i], l2cache, L2_SIZE);
	}
	pgtable_free(&sh_domain->l1, l1cache, L1_SIZE);
	kfree(sh_domain);
}

static int shmobile_iommu_attach_device(struct iommu_domain *domain,
					struct device *dev)
{
	struct shmobile_iommu_archdata *archdata = dev->archdata.iommu;
	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
	int ret = -EBUSY;

	if (!archdata)
		return -ENODEV;
	spin_lock(&sh_domain->attached_list_lock);
	spin_lock(&archdata->attach_lock);
	if (archdata->attached != sh_domain) {
		if (archdata->attached)
			goto err;
		ipmmu_tlb_set(archdata->ipmmu, sh_domain->l1.handle, L1_SIZE,
			      0);
		ipmmu_tlb_flush(archdata->ipmmu);
		archdata->attached = sh_domain;
		archdata->num_attached_devices = 0;
		list_add(&archdata->attached_list, &sh_domain->attached_list);
	}
	archdata->num_attached_devices++;
	ret = 0;
err:
	spin_unlock(&archdata->attach_lock);
	spin_unlock(&sh_domain->attached_list_lock);
	return ret;
}

static void shmobile_iommu_detach_device(struct iommu_domain *domain,
					 struct device *dev)
{
	struct shmobile_iommu_archdata *archdata = dev->archdata.iommu;
	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);

	if (!archdata)
		return;
	spin_lock(&sh_domain->attached_list_lock);
	spin_lock(&archdata->attach_lock);
	archdata->num_attached_devices--;
	if (!archdata->num_attached_devices) {
		ipmmu_tlb_set(archdata->ipmmu, 0, 0, 0);
		ipmmu_tlb_flush(archdata->ipmmu);
		archdata->attached = NULL;
		list_del(&archdata->attached_list);
	}
	spin_unlock(&archdata->attach_lock);
	spin_unlock(&sh_domain->attached_list_lock);
}

static void domain_tlb_flush(struct shmobile_iommu_domain *sh_domain)
{
	struct shmobile_iommu_archdata *archdata;

	spin_lock(&sh_domain->attached_list_lock);
	list_for_each_entry(archdata, &sh_domain->attached_list, attached_list)
		ipmmu_tlb_flush(archdata->ipmmu);
	spin_unlock(&sh_domain->attached_list_lock);
}

static int l2alloc(struct shmobile_iommu_domain *sh_domain,
		   unsigned int l1index)
{
	int ret;

	if (!sh_domain->l2[l1index].pgtable) {
		ret = pgtable_alloc(&sh_domain->l2[l1index], l2cache, L2_SIZE);
		if (ret < 0)
			return ret;
	}
	pgtable_write(&sh_domain->l1, l1index, 1,
		      sh_domain->l2[l1index].handle | 0x1);
	return 0;
}

static void l2realfree(struct shmobile_iommu_domain_pgtable *l2)
{
	if (l2->pgtable)
		pgtable_free(l2, l2cache, L2_SIZE);
}

static void l2free(struct shmobile_iommu_domain *sh_domain,
		   unsigned int l1index,
		   struct shmobile_iommu_domain_pgtable *l2)
{
	pgtable_write(&sh_domain->l1, l1index, 1, 0);
	if (sh_domain->l2[l1index].pgtable) {
		*l2 = sh_domain->l2[l1index];
		sh_domain->l2[l1index].pgtable = NULL;
	}
}

static int shmobile_iommu_map(struct iommu_domain *domain, unsigned long iova,
			      phys_addr_t paddr, size_t size, int prot)
{
	struct shmobile_iommu_domain_pgtable l2 = { .pgtable = NULL };
	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
	unsigned int l1index, l2index;
	int ret;

	l1index = iova >> 20;
	switch (size) {
	case SZ_4K:
		l2index = (iova >> 12) & 0xff;
		spin_lock(&sh_domain->map_lock);
		ret = l2alloc(sh_domain, l1index);
		if (!ret)
			pgtable_write(&sh_domain->l2[l1index], l2index, 1,
				      paddr | 0xff2);
		spin_unlock(&sh_domain->map_lock);
		break;
	case SZ_64K:
		l2index = (iova >> 12) & 0xf0;
		spin_lock(&sh_domain->map_lock);
		ret = l2alloc(sh_domain, l1index);
		if (!ret)
			pgtable_write(&sh_domain->l2[l1index], l2index, 0x10,
				      paddr | 0xff1);
		spin_unlock(&sh_domain->map_lock);
		break;
	case SZ_1M:
		spin_lock(&sh_domain->map_lock);
		l2free(sh_domain, l1index, &l2);
		pgtable_write(&sh_domain->l1, l1index, 1, paddr | 0xc02);
		spin_unlock(&sh_domain->map_lock);
		ret = 0;
		break;
	default:
		ret = -EINVAL;
	}
	if (!ret)
		domain_tlb_flush(sh_domain);
	l2realfree(&l2);
	return ret;
}

static size_t shmobile_iommu_unmap(struct iommu_domain *domain,
				   unsigned long iova, size_t size)
{
	struct shmobile_iommu_domain_pgtable l2 = { .pgtable = NULL };
	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
	unsigned int l1index, l2index;
	uint32_t l2entry = 0;
	size_t ret = 0;

	l1index = iova >> 20;
	if (!(iova & 0xfffff) && size >= SZ_1M) {
		spin_lock(&sh_domain->map_lock);
		l2free(sh_domain, l1index, &l2);
		spin_unlock(&sh_domain->map_lock);
		ret = SZ_1M;
		goto done;
	}
	l2index = (iova >> 12) & 0xff;
	spin_lock(&sh_domain->map_lock);
	if (sh_domain->l2[l1index].pgtable)
		l2entry = pgtable_read(&sh_domain->l2[l1index], l2index);
	switch (l2entry & 3) {
	case 1:
		if (l2index & 0xf)
			break;
		pgtable_write(&sh_domain->l2[l1index], l2index, 0x10, 0);
		ret = SZ_64K;
		break;
	case 2:
		pgtable_write(&sh_domain->l2[l1index], l2index, 1, 0);
		ret = SZ_4K;
		break;
	}
	spin_unlock(&sh_domain->map_lock);
done:
	if (ret)
		domain_tlb_flush(sh_domain);
	l2realfree(&l2);
	return ret;
}

static phys_addr_t shmobile_iommu_iova_to_phys(struct iommu_domain *domain,
					       dma_addr_t iova)
{
	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
	uint32_t l1entry = 0, l2entry = 0;
	unsigned int l1index, l2index;

	l1index = iova >> 20;
	l2index = (iova >> 12) & 0xff;
	spin_lock(&sh_domain->map_lock);
	if (sh_domain->l2[l1index].pgtable)
		l2entry = pgtable_read(&sh_domain->l2[l1index], l2index);
	else
		l1entry = pgtable_read(&sh_domain->l1, l1index);
	spin_unlock(&sh_domain->map_lock);
	switch (l2entry & 3) {
	case 1:
		return (l2entry & ~0xffff) | (iova & 0xffff);
	case 2:
		return (l2entry & ~0xfff) | (iova & 0xfff);
	default:
		if ((l1entry & 3) == 2)
			return (l1entry & ~0xfffff) | (iova & 0xfffff);
		return 0;
	}
}

static int find_dev_name(struct shmobile_ipmmu *ipmmu, const char *dev_name)
{
	unsigned int i, n = ipmmu->num_dev_names;

	for (i = 0; i < n; i++) {
		if (strcmp(ipmmu->dev_names[i], dev_name) == 0)
			return 1;
	}
	return 0;
}

static int shmobile_iommu_add_device(struct device *dev)
{
	struct shmobile_iommu_archdata *archdata = ipmmu_archdata;
	struct dma_iommu_mapping *mapping;

	if (!find_dev_name(archdata->ipmmu, dev_name(dev)))
		return 0;
	mapping = archdata->iommu_mapping;
	if (!mapping) {
		mapping = arm_iommu_create_mapping(&platform_bus_type, 0,
						   L1_LEN << 20);
		if (IS_ERR(mapping))
			return PTR_ERR(mapping);
		archdata->iommu_mapping = mapping;
	}
	dev->archdata.iommu = archdata;
	if (arm_iommu_attach_device(dev, mapping))
		pr_err("arm_iommu_attach_device failed\n");
	return 0;
}

static const struct iommu_ops shmobile_iommu_ops = {
	.domain_alloc = shmobile_iommu_domain_alloc,
	.domain_free = shmobile_iommu_domain_free,
	.attach_dev = shmobile_iommu_attach_device,
	.detach_dev = shmobile_iommu_detach_device,
	.map = shmobile_iommu_map,
	.unmap = shmobile_iommu_unmap,
	.map_sg = default_iommu_map_sg,
	.iova_to_phys = shmobile_iommu_iova_to_phys,
	.add_device = shmobile_iommu_add_device,
	.pgsize_bitmap = SZ_1M | SZ_64K | SZ_4K,
};

int ipmmu_iommu_init(struct shmobile_ipmmu *ipmmu)
{
	static struct shmobile_iommu_archdata *archdata;

	l1cache = kmem_cache_create("shmobile-iommu-pgtable1", L1_SIZE,
				    L1_ALIGN, SLAB_HWCACHE_ALIGN, NULL);
	if (!l1cache)
		return -ENOMEM;
	l2cache = kmem_cache_create("shmobile-iommu-pgtable2", L2_SIZE,
				    L2_ALIGN, SLAB_HWCACHE_ALIGN, NULL);
	if (!l2cache) {
		kmem_cache_destroy(l1cache);
		return -ENOMEM;
	}
	archdata = kzalloc(sizeof(*archdata), GFP_KERNEL);
	if (!archdata) {
		kmem_cache_destroy(l1cache);
		kmem_cache_destroy(l2cache);
		return -ENOMEM;
	}
	spin_lock_init(&archdata->attach_lock);
	archdata->ipmmu = ipmmu;
	ipmmu_archdata = archdata;
	bus_set_iommu(&platform_bus_type, &shmobile_iommu_ops);
	return 0;
}
Commit	Line	Data
c2c460f7 HE	1	/*
	2	* IOMMU for IPMMU/IPMMUI
	3	* Copyright (C) 2012 Hideki EIRAKU
	4	*
	5	* This program is free software; you can redistribute it and/or modify
	6	* it under the terms of the GNU General Public License as published by
	7	* the Free Software Foundation; version 2 of the License.
	8	*/
	9
	10	#include <linux/dma-mapping.h>
	11	#include <linux/io.h>
	12	#include <linux/iommu.h>
	13	#include <linux/platform_device.h>
	14	#include <linux/sizes.h>
	15	#include <linux/slab.h>
	16	#include <asm/dma-iommu.h>
	17	#include "shmobile-ipmmu.h"
	18
	19	#define L1_SIZE CONFIG_SHMOBILE_IOMMU_L1SIZE
	20	#define L1_LEN (L1_SIZE / 4)
	21	#define L1_ALIGN L1_SIZE
	22	#define L2_SIZE SZ_1K
	23	#define L2_LEN (L2_SIZE / 4)
	24	#define L2_ALIGN L2_SIZE
	25
	26	struct shmobile_iommu_domain_pgtable {
	27	uint32_t *pgtable;
	28	dma_addr_t handle;
	29	};
	30
	31	struct shmobile_iommu_archdata {
	32	struct list_head attached_list;
	33	struct dma_iommu_mapping *iommu_mapping;
	34	spinlock_t attach_lock;
	35	struct shmobile_iommu_domain *attached;
	36	int num_attached_devices;
	37	struct shmobile_ipmmu *ipmmu;
	38	};
	39
	40	struct shmobile_iommu_domain {
	41	struct shmobile_iommu_domain_pgtable l1, l2[L1_LEN];
	42	spinlock_t map_lock;
	43	spinlock_t attached_list_lock;
	44	struct list_head attached_list;
dc8fb227	45	struct iommu_domain domain;
c2c460f7 HE	46	};
	47
	48	static struct shmobile_iommu_archdata *ipmmu_archdata;
	49	static struct kmem_cache l1cache, l2cache;
	50
dc8fb227 JR	51	static struct shmobile_iommu_domain to_sh_domain(struct iommu_domain dom)
	52	{
	53	return container_of(dom, struct shmobile_iommu_domain, domain);
	54	}
	55
c2c460f7 HE	56	static int pgtable_alloc(struct shmobile_iommu_domain_pgtable *pgtable,
	57	struct kmem_cache *cache, size_t size)
	58	{
	59	pgtable->pgtable = kmem_cache_zalloc(cache, GFP_ATOMIC);
	60	if (!pgtable->pgtable)
	61	return -ENOMEM;
	62	pgtable->handle = dma_map_single(NULL, pgtable->pgtable, size,
	63	DMA_TO_DEVICE);
	64	return 0;
	65	}
	66
	67	static void pgtable_free(struct shmobile_iommu_domain_pgtable *pgtable,
	68	struct kmem_cache *cache, size_t size)
	69	{
	70	dma_unmap_single(NULL, pgtable->handle, size, DMA_TO_DEVICE);
	71	kmem_cache_free(cache, pgtable->pgtable);
	72	}
	73
	74	static uint32_t pgtable_read(struct shmobile_iommu_domain_pgtable *pgtable,
	75	unsigned int index)
	76	{
	77	return pgtable->pgtable[index];
	78	}
	79
	80	static void pgtable_write(struct shmobile_iommu_domain_pgtable *pgtable,
	81	unsigned int index, unsigned int count, uint32_t val)
	82	{
	83	unsigned int i;
	84
	85	for (i = 0; i < count; i++)
	86	pgtable->pgtable[index + i] = val;
	87	dma_sync_single_for_device(NULL, pgtable->handle + index * sizeof(val),
	88	sizeof(val) * count, DMA_TO_DEVICE);
	89	}
	90
dc8fb227	91	static struct iommu_domain *shmobile_iommu_domain_alloc(unsigned type)
c2c460f7 HE	92	{
	93	struct shmobile_iommu_domain *sh_domain;
	94	int i, ret;
	95
dc8fb227 JR	96	if (type != IOMMU_DOMAIN_UNMANAGED)
	97	return NULL;
	98
	99	sh_domain = kzalloc(sizeof(*sh_domain), GFP_KERNEL);
c2c460f7	100	if (!sh_domain)
dc8fb227	101	return NULL;
c2c460f7 HE	102	ret = pgtable_alloc(&sh_domain->l1, l1cache, L1_SIZE);
	103	if (ret < 0) {
	104	kfree(sh_domain);
dc8fb227	105	return NULL;
c2c460f7 HE	106	}
	107	for (i = 0; i < L1_LEN; i++)
	108	sh_domain->l2[i].pgtable = NULL;
	109	spin_lock_init(&sh_domain->map_lock);
	110	spin_lock_init(&sh_domain->attached_list_lock);
	111	INIT_LIST_HEAD(&sh_domain->attached_list);
dc8fb227	112	return &sh_domain->domain;
c2c460f7 HE	113	}
c2c460f7 HE	114
dc8fb227	115	static void shmobile_iommu_domain_free(struct iommu_domain *domain)
c2c460f7	116	{
dc8fb227	117	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
c2c460f7 HE	118	int i;
	119
	120	for (i = 0; i < L1_LEN; i++) {
	121	if (sh_domain->l2[i].pgtable)
	122	pgtable_free(&sh_domain->l2[i], l2cache, L2_SIZE);
	123	}
	124	pgtable_free(&sh_domain->l1, l1cache, L1_SIZE);
	125	kfree(sh_domain);
c2c460f7 HE	126	}
	127
	128	static int shmobile_iommu_attach_device(struct iommu_domain *domain,
	129	struct device *dev)
	130	{
	131	struct shmobile_iommu_archdata *archdata = dev->archdata.iommu;
dc8fb227	132	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
c2c460f7 HE	133	int ret = -EBUSY;
	134
	135	if (!archdata)
	136	return -ENODEV;
	137	spin_lock(&sh_domain->attached_list_lock);
	138	spin_lock(&archdata->attach_lock);
	139	if (archdata->attached != sh_domain) {
	140	if (archdata->attached)
	141	goto err;
	142	ipmmu_tlb_set(archdata->ipmmu, sh_domain->l1.handle, L1_SIZE,
	143	0);
	144	ipmmu_tlb_flush(archdata->ipmmu);
	145	archdata->attached = sh_domain;
	146	archdata->num_attached_devices = 0;
	147	list_add(&archdata->attached_list, &sh_domain->attached_list);
	148	}
	149	archdata->num_attached_devices++;
	150	ret = 0;
	151	err:
	152	spin_unlock(&archdata->attach_lock);
	153	spin_unlock(&sh_domain->attached_list_lock);
	154	return ret;
	155	}
	156
	157	static void shmobile_iommu_detach_device(struct iommu_domain *domain,
	158	struct device *dev)
	159	{
	160	struct shmobile_iommu_archdata *archdata = dev->archdata.iommu;
dc8fb227	161	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
c2c460f7 HE	162
	163	if (!archdata)
	164	return;
	165	spin_lock(&sh_domain->attached_list_lock);
	166	spin_lock(&archdata->attach_lock);
	167	archdata->num_attached_devices--;
	168	if (!archdata->num_attached_devices) {
	169	ipmmu_tlb_set(archdata->ipmmu, 0, 0, 0);
	170	ipmmu_tlb_flush(archdata->ipmmu);
	171	archdata->attached = NULL;
	172	list_del(&archdata->attached_list);
	173	}
	174	spin_unlock(&archdata->attach_lock);
	175	spin_unlock(&sh_domain->attached_list_lock);
	176	}
	177
	178	static void domain_tlb_flush(struct shmobile_iommu_domain *sh_domain)
	179	{
	180	struct shmobile_iommu_archdata *archdata;
	181
	182	spin_lock(&sh_domain->attached_list_lock);
	183	list_for_each_entry(archdata, &sh_domain->attached_list, attached_list)
	184	ipmmu_tlb_flush(archdata->ipmmu);
	185	spin_unlock(&sh_domain->attached_list_lock);
	186	}
	187
	188	static int l2alloc(struct shmobile_iommu_domain *sh_domain,
	189	unsigned int l1index)
	190	{
	191	int ret;
	192
	193	if (!sh_domain->l2[l1index].pgtable) {
	194	ret = pgtable_alloc(&sh_domain->l2[l1index], l2cache, L2_SIZE);
	195	if (ret < 0)
	196	return ret;
	197	}
	198	pgtable_write(&sh_domain->l1, l1index, 1,
	199	sh_domain->l2[l1index].handle \| 0x1);
	200	return 0;
	201	}
	202
	203	static void l2realfree(struct shmobile_iommu_domain_pgtable *l2)
	204	{
	205	if (l2->pgtable)
	206	pgtable_free(l2, l2cache, L2_SIZE);
	207	}
	208
	209	static void l2free(struct shmobile_iommu_domain *sh_domain,
	210	unsigned int l1index,
	211	struct shmobile_iommu_domain_pgtable *l2)
	212	{
	213	pgtable_write(&sh_domain->l1, l1index, 1, 0);
	214	if (sh_domain->l2[l1index].pgtable) {
	215	*l2 = sh_domain->l2[l1index];
	216	sh_domain->l2[l1index].pgtable = NULL;
	217	}
	218	}
	219
	220	static int shmobile_iommu_map(struct iommu_domain *domain, unsigned long iova,
	221	phys_addr_t paddr, size_t size, int prot)
	222	{
	223	struct shmobile_iommu_domain_pgtable l2 = { .pgtable = NULL };
dc8fb227	224	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
c2c460f7 HE	225	unsigned int l1index, l2index;
	226	int ret;
	227
	228	l1index = iova >> 20;
	229	switch (size) {
	230	case SZ_4K:
	231	l2index = (iova >> 12) & 0xff;
	232	spin_lock(&sh_domain->map_lock);
	233	ret = l2alloc(sh_domain, l1index);
	234	if (!ret)
	235	pgtable_write(&sh_domain->l2[l1index], l2index, 1,
	236	paddr \| 0xff2);
	237	spin_unlock(&sh_domain->map_lock);
	238	break;
	239	case SZ_64K:
	240	l2index = (iova >> 12) & 0xf0;
	241	spin_lock(&sh_domain->map_lock);
	242	ret = l2alloc(sh_domain, l1index);
	243	if (!ret)
	244	pgtable_write(&sh_domain->l2[l1index], l2index, 0x10,
	245	paddr \| 0xff1);
	246	spin_unlock(&sh_domain->map_lock);
	247	break;
	248	case SZ_1M:
	249	spin_lock(&sh_domain->map_lock);
	250	l2free(sh_domain, l1index, &l2);
	251	pgtable_write(&sh_domain->l1, l1index, 1, paddr \| 0xc02);
	252	spin_unlock(&sh_domain->map_lock);
	253	ret = 0;
	254	break;
	255	default:
	256	ret = -EINVAL;
	257	}
	258	if (!ret)
	259	domain_tlb_flush(sh_domain);
	260	l2realfree(&l2);
	261	return ret;
	262	}
	263
	264	static size_t shmobile_iommu_unmap(struct iommu_domain *domain,
	265	unsigned long iova, size_t size)
	266	{
	267	struct shmobile_iommu_domain_pgtable l2 = { .pgtable = NULL };
dc8fb227	268	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
c2c460f7 HE	269	unsigned int l1index, l2index;
	270	uint32_t l2entry = 0;
	271	size_t ret = 0;
	272
	273	l1index = iova >> 20;
	274	if (!(iova & 0xfffff) && size >= SZ_1M) {
	275	spin_lock(&sh_domain->map_lock);
	276	l2free(sh_domain, l1index, &l2);
	277	spin_unlock(&sh_domain->map_lock);
	278	ret = SZ_1M;
	279	goto done;
	280	}
	281	l2index = (iova >> 12) & 0xff;
	282	spin_lock(&sh_domain->map_lock);
	283	if (sh_domain->l2[l1index].pgtable)
	284	l2entry = pgtable_read(&sh_domain->l2[l1index], l2index);
	285	switch (l2entry & 3) {
	286	case 1:
	287	if (l2index & 0xf)
	288	break;
	289	pgtable_write(&sh_domain->l2[l1index], l2index, 0x10, 0);
	290	ret = SZ_64K;
	291	break;
	292	case 2:
	293	pgtable_write(&sh_domain->l2[l1index], l2index, 1, 0);
	294	ret = SZ_4K;
	295	break;
	296	}
	297	spin_unlock(&sh_domain->map_lock);
	298	done:
	299	if (ret)
	300	domain_tlb_flush(sh_domain);
	301	l2realfree(&l2);
	302	return ret;
	303	}
	304
	305	static phys_addr_t shmobile_iommu_iova_to_phys(struct iommu_domain *domain,
bb5547ac	306	dma_addr_t iova)
c2c460f7	307	{
dc8fb227	308	struct shmobile_iommu_domain *sh_domain = to_sh_domain(domain);
c2c460f7 HE	309	uint32_t l1entry = 0, l2entry = 0;
	310	unsigned int l1index, l2index;
	311
	312	l1index = iova >> 20;
	313	l2index = (iova >> 12) & 0xff;
	314	spin_lock(&sh_domain->map_lock);
	315	if (sh_domain->l2[l1index].pgtable)
	316	l2entry = pgtable_read(&sh_domain->l2[l1index], l2index);
	317	else
	318	l1entry = pgtable_read(&sh_domain->l1, l1index);
	319	spin_unlock(&sh_domain->map_lock);
	320	switch (l2entry & 3) {
	321	case 1:
	322	return (l2entry & ~0xffff) \| (iova & 0xffff);
	323	case 2:
	324	return (l2entry & ~0xfff) \| (iova & 0xfff);
	325	default:
	326	if ((l1entry & 3) == 2)
	327	return (l1entry & ~0xfffff) \| (iova & 0xfffff);
	328	return 0;
	329	}
	330	}
	331
	332	static int find_dev_name(struct shmobile_ipmmu ipmmu, const char dev_name)
	333	{
	334	unsigned int i, n = ipmmu->num_dev_names;
	335
	336	for (i = 0; i < n; i++) {
	337	if (strcmp(ipmmu->dev_names[i], dev_name) == 0)
	338	return 1;
	339	}
	340	return 0;
	341	}
	342
	343	static int shmobile_iommu_add_device(struct device *dev)
	344	{
	345	struct shmobile_iommu_archdata *archdata = ipmmu_archdata;
	346	struct dma_iommu_mapping *mapping;
	347
	348	if (!find_dev_name(archdata->ipmmu, dev_name(dev)))
	349	return 0;
	350	mapping = archdata->iommu_mapping;
	351	if (!mapping) {
	352	mapping = arm_iommu_create_mapping(&platform_bus_type, 0,
68efd7d2	353	L1_LEN << 20);
c2c460f7 HE	354	if (IS_ERR(mapping))
	355	return PTR_ERR(mapping);
	356	archdata->iommu_mapping = mapping;
	357	}
	358	dev->archdata.iommu = archdata;
	359	if (arm_iommu_attach_device(dev, mapping))
	360	pr_err("arm_iommu_attach_device failed\n");
	361	return 0;
	362	}
	363
b22f6434	364	static const struct iommu_ops shmobile_iommu_ops = {
dc8fb227 JR	365	.domain_alloc = shmobile_iommu_domain_alloc,
dc8fb227 JR	366	.domain_free = shmobile_iommu_domain_free,
c2c460f7 HE	367	.attach_dev = shmobile_iommu_attach_device,
	368	.detach_dev = shmobile_iommu_detach_device,
	369	.map = shmobile_iommu_map,
	370	.unmap = shmobile_iommu_unmap,
315786eb	371	.map_sg = default_iommu_map_sg,
c2c460f7 HE	372	.iova_to_phys = shmobile_iommu_iova_to_phys,
	373	.add_device = shmobile_iommu_add_device,
	374	.pgsize_bitmap = SZ_1M \| SZ_64K \| SZ_4K,
	375	};
	376
	377	int ipmmu_iommu_init(struct shmobile_ipmmu *ipmmu)
	378	{
	379	static struct shmobile_iommu_archdata *archdata;
	380
	381	l1cache = kmem_cache_create("shmobile-iommu-pgtable1", L1_SIZE,
	382	L1_ALIGN, SLAB_HWCACHE_ALIGN, NULL);
	383	if (!l1cache)
	384	return -ENOMEM;
	385	l2cache = kmem_cache_create("shmobile-iommu-pgtable2", L2_SIZE,
	386	L2_ALIGN, SLAB_HWCACHE_ALIGN, NULL);
	387	if (!l2cache) {
	388	kmem_cache_destroy(l1cache);
	389	return -ENOMEM;
	390	}
dc89f797	391	archdata = kzalloc(sizeof(*archdata), GFP_KERNEL);
c2c460f7 HE	392	if (!archdata) {
	393	kmem_cache_destroy(l1cache);
	394	kmem_cache_destroy(l2cache);
	395	return -ENOMEM;
	396	}
	397	spin_lock_init(&archdata->attach_lock);
c2c460f7 HE	398	archdata->ipmmu = ipmmu;
	399	ipmmu_archdata = archdata;
	400	bus_set_iommu(&platform_bus_type, &shmobile_iommu_ops);
	401	return 0;
	402	}