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

/* Copyright (c) 2010-2011, Code Aurora Forum. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 and
 * only version 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

#define pr_fmt(fmt)	KBUILD_MODNAME ": " fmt

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/io.h>
#include <linux/clk.h>
#include <linux/iommu.h>
#include <linux/interrupt.h>
#include <linux/err.h>
#include <linux/slab.h>

#include "msm_iommu_hw-8xxx.h"
#include "msm_iommu.h"

struct iommu_ctx_iter_data {
	/* input */
	const char *name;

	/* output */
	struct device *dev;
};

static struct platform_device *msm_iommu_root_dev;

static int each_iommu_ctx(struct device *dev, void *data)
{
	struct iommu_ctx_iter_data *res = data;
	struct msm_iommu_ctx_dev *c = dev->platform_data;

	if (!res || !c || !c->name || !res->name)
		return -EINVAL;

	if (!strcmp(res->name, c->name)) {
		res->dev = dev;
		return 1;
	}
	return 0;
}

static int each_iommu(struct device *dev, void *data)
{
	return device_for_each_child(dev, data, each_iommu_ctx);
}

struct device *msm_iommu_get_ctx(const char *ctx_name)
{
	struct iommu_ctx_iter_data r;
	int found;

	if (!msm_iommu_root_dev) {
		pr_err("No root IOMMU device.\n");
		goto fail;
	}

	r.name = ctx_name;
	found = device_for_each_child(&msm_iommu_root_dev->dev, &r, each_iommu);

	if (!found) {
		pr_err("Could not find context <%s>\n", ctx_name);
		goto fail;
	}

	return r.dev;
fail:
	return NULL;
}
EXPORT_SYMBOL(msm_iommu_get_ctx);

static void msm_iommu_reset(void __iomem *base, int ncb)
{
	int ctx;

	SET_RPUE(base, 0);
	SET_RPUEIE(base, 0);
	SET_ESRRESTORE(base, 0);
	SET_TBE(base, 0);
	SET_CR(base, 0);
	SET_SPDMBE(base, 0);
	SET_TESTBUSCR(base, 0);
	SET_TLBRSW(base, 0);
	SET_GLOBAL_TLBIALL(base, 0);
	SET_RPU_ACR(base, 0);
	SET_TLBLKCRWE(base, 1);

	for (ctx = 0; ctx < ncb; ctx++) {
		SET_BPRCOSH(base, ctx, 0);
		SET_BPRCISH(base, ctx, 0);
		SET_BPRCNSH(base, ctx, 0);
		SET_BPSHCFG(base, ctx, 0);
		SET_BPMTCFG(base, ctx, 0);
		SET_ACTLR(base, ctx, 0);
		SET_SCTLR(base, ctx, 0);
		SET_FSRRESTORE(base, ctx, 0);
		SET_TTBR0(base, ctx, 0);
		SET_TTBR1(base, ctx, 0);
		SET_TTBCR(base, ctx, 0);
		SET_BFBCR(base, ctx, 0);
		SET_PAR(base, ctx, 0);
		SET_FAR(base, ctx, 0);
		SET_CTX_TLBIALL(base, ctx, 0);
		SET_TLBFLPTER(base, ctx, 0);
		SET_TLBSLPTER(base, ctx, 0);
		SET_TLBLKCR(base, ctx, 0);
		SET_PRRR(base, ctx, 0);
		SET_NMRR(base, ctx, 0);
		SET_CONTEXTIDR(base, ctx, 0);
	}
}

static int msm_iommu_probe(struct platform_device *pdev)
{
	struct resource *r, *r2;
	struct clk *iommu_clk;
	struct clk *iommu_pclk;
	struct msm_iommu_drvdata *drvdata;
	struct msm_iommu_dev *iommu_dev = pdev->dev.platform_data;
	void __iomem *regs_base;
	resource_size_t	len;
	int ret, irq, par;

	if (pdev->id == -1) {
		msm_iommu_root_dev = pdev;
		return 0;
	}

	drvdata = kzalloc(sizeof(*drvdata), GFP_KERNEL);

	if (!drvdata) {
		ret = -ENOMEM;
		goto fail;
	}

	if (!iommu_dev) {
		ret = -ENODEV;
		goto fail;
	}

	iommu_pclk = clk_get(NULL, "smmu_pclk");
	if (IS_ERR(iommu_pclk)) {
		ret = -ENODEV;
		goto fail;
	}

	ret = clk_prepare_enable(iommu_pclk);
	if (ret)
		goto fail_enable;

	iommu_clk = clk_get(&pdev->dev, "iommu_clk");

	if (!IS_ERR(iommu_clk))	{
		if (clk_get_rate(iommu_clk) == 0)
			clk_set_rate(iommu_clk, 1);

		ret = clk_prepare_enable(iommu_clk);
		if (ret) {
			clk_put(iommu_clk);
			goto fail_pclk;
		}
	} else
		iommu_clk = NULL;

	r = platform_get_resource_byname(pdev, IORESOURCE_MEM, "physbase");

	if (!r) {
		ret = -ENODEV;
		goto fail_clk;
	}

	len = resource_size(r);

	r2 = request_mem_region(r->start, len, r->name);
	if (!r2) {
		pr_err("Could not request memory region: start=%p, len=%d\n",
							(void *) r->start, len);
		ret = -EBUSY;
		goto fail_clk;
	}

	regs_base = ioremap(r2->start, len);

	if (!regs_base) {
		pr_err("Could not ioremap: start=%p, len=%d\n",
			 (void *) r2->start, len);
		ret = -EBUSY;
		goto fail_mem;
	}

	irq = platform_get_irq_byname(pdev, "secure_irq");
	if (irq < 0) {
		ret = -ENODEV;
		goto fail_io;
	}

	msm_iommu_reset(regs_base, iommu_dev->ncb);

	SET_M(regs_base, 0, 1);
	SET_PAR(regs_base, 0, 0);
	SET_V2PCFG(regs_base, 0, 1);
	SET_V2PPR(regs_base, 0, 0);
	par = GET_PAR(regs_base, 0);
	SET_V2PCFG(regs_base, 0, 0);
	SET_M(regs_base, 0, 0);

	if (!par) {
		pr_err("%s: Invalid PAR value detected\n", iommu_dev->name);
		ret = -ENODEV;
		goto fail_io;
	}

	ret = request_irq(irq, msm_iommu_fault_handler, 0,
			"msm_iommu_secure_irpt_handler", drvdata);
	if (ret) {
		pr_err("Request IRQ %d failed with ret=%d\n", irq, ret);
		goto fail_io;
	}


	drvdata->pclk = iommu_pclk;
	drvdata->clk = iommu_clk;
	drvdata->base = regs_base;
	drvdata->irq = irq;
	drvdata->ncb = iommu_dev->ncb;

	pr_info("device %s mapped at %p, irq %d with %d ctx banks\n",
		iommu_dev->name, regs_base, irq, iommu_dev->ncb);

	platform_set_drvdata(pdev, drvdata);

	if (iommu_clk)
		clk_disable(iommu_clk);

	clk_disable(iommu_pclk);

	return 0;
fail_io:
	iounmap(regs_base);
fail_mem:
	release_mem_region(r->start, len);
fail_clk:
	if (iommu_clk) {
		clk_disable(iommu_clk);
		clk_put(iommu_clk);
	}
fail_pclk:
	clk_disable_unprepare(iommu_pclk);
fail_enable:
	clk_put(iommu_pclk);
fail:
	kfree(drvdata);
	return ret;
}

static int msm_iommu_remove(struct platform_device *pdev)
{
	struct msm_iommu_drvdata *drv = NULL;

	drv = platform_get_drvdata(pdev);
	if (drv) {
		if (drv->clk) {
			clk_unprepare(drv->clk);
			clk_put(drv->clk);
		}
		clk_unprepare(drv->pclk);
		clk_put(drv->pclk);
		memset(drv, 0, sizeof(*drv));
		kfree(drv);
	}
	return 0;
}

static int msm_iommu_ctx_probe(struct platform_device *pdev)
{
	struct msm_iommu_ctx_dev *c = pdev->dev.platform_data;
	struct msm_iommu_drvdata *drvdata;
	struct msm_iommu_ctx_drvdata *ctx_drvdata;
	int i, ret;

	if (!c || !pdev->dev.parent)
		return -EINVAL;

	drvdata = dev_get_drvdata(pdev->dev.parent);
	if (!drvdata)
		return -ENODEV;

	ctx_drvdata = kzalloc(sizeof(*ctx_drvdata), GFP_KERNEL);
	if (!ctx_drvdata)
		return -ENOMEM;

	ctx_drvdata->num = c->num;
	ctx_drvdata->pdev = pdev;

	INIT_LIST_HEAD(&ctx_drvdata->attached_elm);
	platform_set_drvdata(pdev, ctx_drvdata);

	ret = clk_prepare_enable(drvdata->pclk);
	if (ret)
		goto fail;

	if (drvdata->clk) {
		ret = clk_prepare_enable(drvdata->clk);
		if (ret) {
			clk_disable_unprepare(drvdata->pclk);
			goto fail;
		}
	}

	/* Program the M2V tables for this context */
	for (i = 0; i < MAX_NUM_MIDS; i++) {
		int mid = c->mids[i];
		if (mid == -1)
			break;

		SET_M2VCBR_N(drvdata->base, mid, 0);
		SET_CBACR_N(drvdata->base, c->num, 0);

		/* Set VMID = 0 */
		SET_VMID(drvdata->base, mid, 0);

		/* Set the context number for that MID to this context */
		SET_CBNDX(drvdata->base, mid, c->num);

		/* Set MID associated with this context bank to 0*/
		SET_CBVMID(drvdata->base, c->num, 0);

		/* Set the ASID for TLB tagging for this context */
		SET_CONTEXTIDR_ASID(drvdata->base, c->num, c->num);

		/* Set security bit override to be Non-secure */
		SET_NSCFG(drvdata->base, mid, 3);
	}

	if (drvdata->clk)
		clk_disable(drvdata->clk);
	clk_disable(drvdata->pclk);

	dev_info(&pdev->dev, "context %s using bank %d\n", c->name, c->num);
	return 0;
fail:
	kfree(ctx_drvdata);
	return ret;
}

static int msm_iommu_ctx_remove(struct platform_device *pdev)
{
	struct msm_iommu_ctx_drvdata *drv = NULL;
	drv = platform_get_drvdata(pdev);
	if (drv) {
		memset(drv, 0, sizeof(struct msm_iommu_ctx_drvdata));
		kfree(drv);
	}
	return 0;
}

static struct platform_driver msm_iommu_driver = {
	.driver = {
		.name	= "msm_iommu",
	},
	.probe		= msm_iommu_probe,
	.remove		= msm_iommu_remove,
};

static struct platform_driver msm_iommu_ctx_driver = {
	.driver = {
		.name	= "msm_iommu_ctx",
	},
	.probe		= msm_iommu_ctx_probe,
	.remove		= msm_iommu_ctx_remove,
};

static int __init msm_iommu_driver_init(void)
{
	int ret;
	ret = platform_driver_register(&msm_iommu_driver);
	if (ret != 0) {
		pr_err("Failed to register IOMMU driver\n");
		goto error;
	}

	ret = platform_driver_register(&msm_iommu_ctx_driver);
	if (ret != 0) {
		platform_driver_unregister(&msm_iommu_driver);
		pr_err("Failed to register IOMMU context driver\n");
		goto error;
	}

error:
	return ret;
}

static void __exit msm_iommu_driver_exit(void)
{
	platform_driver_unregister(&msm_iommu_ctx_driver);
	platform_driver_unregister(&msm_iommu_driver);
}

subsys_initcall(msm_iommu_driver_init);
module_exit(msm_iommu_driver_exit);

MODULE_LICENSE("GPL v2");
MODULE_AUTHOR("Stepan Moskovchenko <stepanm@codeaurora.org>");
Commit	Line	Data
b61401ad	1	/* Copyright (c) 2010-2011, Code Aurora Forum. All rights reserved.
c6a5951e SM	2	*
	3	* This program is free software; you can redistribute it and/or modify
	4	* it under the terms of the GNU General Public License version 2 and
	5	* only version 2 as published by the Free Software Foundation.
	6	*
	7	* This program is distributed in the hope that it will be useful,
	8	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	9	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	10	* GNU General Public License for more details.
	11	*
	12	* You should have received a copy of the GNU General Public License
	13	* along with this program; if not, write to the Free Software
	14	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
	15	* 02110-1301, USA.
	16	*/
	17
	18	#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
	19
	20	#include <linux/kernel.h>
	21	#include <linux/module.h>
	22	#include <linux/platform_device.h>
	23	#include <linux/io.h>
	24	#include <linux/clk.h>
	25	#include <linux/iommu.h>
	26	#include <linux/interrupt.h>
	27	#include <linux/err.h>
	28	#include <linux/slab.h>
	29
0b559df5 SB	30	#include "msm_iommu_hw-8xxx.h"
0b559df5 SB	31	#include "msm_iommu.h"
c6a5951e SM	32
	33	struct iommu_ctx_iter_data {
	34	/* input */
	35	const char *name;
	36
	37	/* output */
	38	struct device *dev;
	39	};
	40
	41	static struct platform_device *msm_iommu_root_dev;
	42
	43	static int each_iommu_ctx(struct device dev, void data)
	44	{
	45	struct iommu_ctx_iter_data *res = data;
	46	struct msm_iommu_ctx_dev *c = dev->platform_data;
	47
	48	if (!res \|\| !c \|\| !c->name \|\| !res->name)
	49	return -EINVAL;
	50
	51	if (!strcmp(res->name, c->name)) {
	52	res->dev = dev;
	53	return 1;
	54	}
	55	return 0;
	56	}
	57
	58	static int each_iommu(struct device dev, void data)
	59	{
	60	return device_for_each_child(dev, data, each_iommu_ctx);
	61	}
	62
	63	struct device msm_iommu_get_ctx(const char ctx_name)
	64	{
	65	struct iommu_ctx_iter_data r;
	66	int found;
	67
	68	if (!msm_iommu_root_dev) {
	69	pr_err("No root IOMMU device.\n");
	70	goto fail;
	71	}
	72
	73	r.name = ctx_name;
	74	found = device_for_each_child(&msm_iommu_root_dev->dev, &r, each_iommu);
	75
	76	if (!found) {
	77	pr_err("Could not find context <%s>\n", ctx_name);
	78	goto fail;
	79	}
	80
	81	return r.dev;
	82	fail:
	83	return NULL;
	84	}
	85	EXPORT_SYMBOL(msm_iommu_get_ctx);
	86
a43d8c10	87	static void msm_iommu_reset(void __iomem *base, int ncb)
c6a5951e	88	{
a43d8c10	89	int ctx;
c6a5951e SM	90
	91	SET_RPUE(base, 0);
	92	SET_RPUEIE(base, 0);
	93	SET_ESRRESTORE(base, 0);
	94	SET_TBE(base, 0);
	95	SET_CR(base, 0);
	96	SET_SPDMBE(base, 0);
	97	SET_TESTBUSCR(base, 0);
	98	SET_TLBRSW(base, 0);
	99	SET_GLOBAL_TLBIALL(base, 0);
	100	SET_RPU_ACR(base, 0);
	101	SET_TLBLKCRWE(base, 1);
c6a5951e SM	102
	103	for (ctx = 0; ctx < ncb; ctx++) {
	104	SET_BPRCOSH(base, ctx, 0);
	105	SET_BPRCISH(base, ctx, 0);
	106	SET_BPRCNSH(base, ctx, 0);
	107	SET_BPSHCFG(base, ctx, 0);
	108	SET_BPMTCFG(base, ctx, 0);
	109	SET_ACTLR(base, ctx, 0);
	110	SET_SCTLR(base, ctx, 0);
	111	SET_FSRRESTORE(base, ctx, 0);
	112	SET_TTBR0(base, ctx, 0);
	113	SET_TTBR1(base, ctx, 0);
	114	SET_TTBCR(base, ctx, 0);
	115	SET_BFBCR(base, ctx, 0);
	116	SET_PAR(base, ctx, 0);
	117	SET_FAR(base, ctx, 0);
	118	SET_CTX_TLBIALL(base, ctx, 0);
	119	SET_TLBFLPTER(base, ctx, 0);
	120	SET_TLBSLPTER(base, ctx, 0);
	121	SET_TLBLKCR(base, ctx, 0);
	122	SET_PRRR(base, ctx, 0);
	123	SET_NMRR(base, ctx, 0);
	124	SET_CONTEXTIDR(base, ctx, 0);
	125	}
	126	}
	127
	128	static int msm_iommu_probe(struct platform_device *pdev)
	129	{
a86c44d4	130	struct resource r, r2;
c6a5951e	131	struct clk *iommu_clk;
b61401ad	132	struct clk *iommu_pclk;
c6a5951e SM	133	struct msm_iommu_drvdata *drvdata;
	134	struct msm_iommu_dev *iommu_dev = pdev->dev.platform_data;
	135	void __iomem *regs_base;
	136	resource_size_t len;
a43d8c10	137	int ret, irq, par;
c6a5951e	138
b61401ad SM	139	if (pdev->id == -1) {
	140	msm_iommu_root_dev = pdev;
	141	return 0;
	142	}
c6a5951e	143
b61401ad	144	drvdata = kzalloc(sizeof(*drvdata), GFP_KERNEL);
c6a5951e	145
b61401ad SM	146	if (!drvdata) {
	147	ret = -ENOMEM;
	148	goto fail;
	149	}
	150
	151	if (!iommu_dev) {
	152	ret = -ENODEV;
	153	goto fail;
	154	}
c6a5951e	155
b61401ad SM	156	iommu_pclk = clk_get(NULL, "smmu_pclk");
	157	if (IS_ERR(iommu_pclk)) {
	158	ret = -ENODEV;
	159	goto fail;
	160	}
	161
1107b171	162	ret = clk_prepare_enable(iommu_pclk);
b61401ad SM	163	if (ret)
	164	goto fail_enable;
	165
	166	iommu_clk = clk_get(&pdev->dev, "iommu_clk");
	167
	168	if (!IS_ERR(iommu_clk)) {
	169	if (clk_get_rate(iommu_clk) == 0)
8d214153	170	clk_set_rate(iommu_clk, 1);
b61401ad	171
1107b171	172	ret = clk_prepare_enable(iommu_clk);
b61401ad	173	if (ret) {
c6a5951e	174	clk_put(iommu_clk);
b61401ad	175	goto fail_pclk;
c6a5951e	176	}
b61401ad SM	177	} else
b61401ad SM	178	iommu_clk = NULL;
c6a5951e	179
b61401ad	180	r = platform_get_resource_byname(pdev, IORESOURCE_MEM, "physbase");
c6a5951e	181
b61401ad SM	182	if (!r) {
	183	ret = -ENODEV;
	184	goto fail_clk;
	185	}
c6a5951e	186
b61401ad	187	len = resource_size(r);
c6a5951e	188
b61401ad SM	189	r2 = request_mem_region(r->start, len, r->name);
	190	if (!r2) {
	191	pr_err("Could not request memory region: start=%p, len=%d\n",
	192	(void *) r->start, len);
	193	ret = -EBUSY;
	194	goto fail_clk;
	195	}
c6a5951e	196
b61401ad	197	regs_base = ioremap(r2->start, len);
c6a5951e	198
b61401ad SM	199	if (!regs_base) {
	200	pr_err("Could not ioremap: start=%p, len=%d\n",
	201	(void *) r2->start, len);
	202	ret = -EBUSY;
	203	goto fail_mem;
	204	}
	205
	206	irq = platform_get_irq_byname(pdev, "secure_irq");
	207	if (irq < 0) {
	208	ret = -ENODEV;
	209	goto fail_io;
	210	}
c6a5951e	211
a43d8c10	212	msm_iommu_reset(regs_base, iommu_dev->ncb);
c6a5951e	213
a43d8c10 SM	214	SET_M(regs_base, 0, 1);
	215	SET_PAR(regs_base, 0, 0);
	216	SET_V2PCFG(regs_base, 0, 1);
	217	SET_V2PPR(regs_base, 0, 0);
	218	par = GET_PAR(regs_base, 0);
	219	SET_V2PCFG(regs_base, 0, 0);
	220	SET_M(regs_base, 0, 0);
	221
	222	if (!par) {
	223	pr_err("%s: Invalid PAR value detected\n", iommu_dev->name);
b61401ad SM	224	ret = -ENODEV;
	225	goto fail_io;
	226	}
c6a5951e	227
b61401ad SM	228	ret = request_irq(irq, msm_iommu_fault_handler, 0,
	229	"msm_iommu_secure_irpt_handler", drvdata);
	230	if (ret) {
	231	pr_err("Request IRQ %d failed with ret=%d\n", irq, ret);
	232	goto fail_io;
	233	}
c6a5951e	234
a43d8c10	235
b61401ad SM	236	drvdata->pclk = iommu_pclk;
	237	drvdata->clk = iommu_clk;
	238	drvdata->base = regs_base;
	239	drvdata->irq = irq;
a43d8c10	240	drvdata->ncb = iommu_dev->ncb;
c6a5951e	241
b61401ad	242	pr_info("device %s mapped at %p, irq %d with %d ctx banks\n",
a43d8c10	243	iommu_dev->name, regs_base, irq, iommu_dev->ncb);
c6a5951e	244
b61401ad	245	platform_set_drvdata(pdev, drvdata);
c6a5951e	246
b61401ad SM	247	if (iommu_clk)
	248	clk_disable(iommu_clk);
	249
	250	clk_disable(iommu_pclk);
c6a5951e	251
b61401ad	252	return 0;
a86c44d4 VK	253	fail_io:
	254	iounmap(regs_base);
	255	fail_mem:
	256	release_mem_region(r->start, len);
b61401ad SM	257	fail_clk:
	258	if (iommu_clk) {
	259	clk_disable(iommu_clk);
	260	clk_put(iommu_clk);
	261	}
	262	fail_pclk:
1107b171	263	clk_disable_unprepare(iommu_pclk);
b61401ad SM	264	fail_enable:
b61401ad SM	265	clk_put(iommu_pclk);
c6a5951e SM	266	fail:
	267	kfree(drvdata);
	268	return ret;
	269	}
	270
	271	static int msm_iommu_remove(struct platform_device *pdev)
	272	{
	273	struct msm_iommu_drvdata *drv = NULL;
	274
	275	drv = platform_get_drvdata(pdev);
	276	if (drv) {
1107b171 SB	277	if (drv->clk) {
1107b171 SB	278	clk_unprepare(drv->clk);
b61401ad	279	clk_put(drv->clk);
1107b171 SB	280	}
1107b171 SB	281	clk_unprepare(drv->pclk);
b61401ad SM	282	clk_put(drv->pclk);
b61401ad SM	283	memset(drv, 0, sizeof(*drv));
c6a5951e	284	kfree(drv);
c6a5951e SM	285	}
	286	return 0;
	287	}
	288
	289	static int msm_iommu_ctx_probe(struct platform_device *pdev)
	290	{
	291	struct msm_iommu_ctx_dev *c = pdev->dev.platform_data;
	292	struct msm_iommu_drvdata *drvdata;
34e3a58c	293	struct msm_iommu_ctx_drvdata *ctx_drvdata;
b61401ad	294	int i, ret;
c6a5951e	295
34e3a58c LC	296	if (!c \|\| !pdev->dev.parent)
34e3a58c LC	297	return -EINVAL;
c6a5951e	298
34e3a58c LC	299	drvdata = dev_get_drvdata(pdev->dev.parent);
	300	if (!drvdata)
	301	return -ENODEV;
c6a5951e SM	302
c6a5951e SM	303	ctx_drvdata = kzalloc(sizeof(*ctx_drvdata), GFP_KERNEL);
34e3a58c LC	304	if (!ctx_drvdata)
	305	return -ENOMEM;
	306
c6a5951e SM	307	ctx_drvdata->num = c->num;
	308	ctx_drvdata->pdev = pdev;
	309
	310	INIT_LIST_HEAD(&ctx_drvdata->attached_elm);
	311	platform_set_drvdata(pdev, ctx_drvdata);
	312
1107b171	313	ret = clk_prepare_enable(drvdata->pclk);
b61401ad SM	314	if (ret)
	315	goto fail;
	316
	317	if (drvdata->clk) {
1107b171	318	ret = clk_prepare_enable(drvdata->clk);
b61401ad	319	if (ret) {
1107b171	320	clk_disable_unprepare(drvdata->pclk);
b61401ad SM	321	goto fail;
	322	}
	323	}
	324
c6a5951e SM	325	/* Program the M2V tables for this context */
	326	for (i = 0; i < MAX_NUM_MIDS; i++) {
	327	int mid = c->mids[i];
	328	if (mid == -1)
	329	break;
	330
	331	SET_M2VCBR_N(drvdata->base, mid, 0);
	332	SET_CBACR_N(drvdata->base, c->num, 0);
	333
2e8c8ba9 SM	334	/* Set VMID = 0 */
2e8c8ba9 SM	335	SET_VMID(drvdata->base, mid, 0);
c6a5951e SM	336
	337	/* Set the context number for that MID to this context */
	338	SET_CBNDX(drvdata->base, mid, c->num);
	339
2e8c8ba9 SM	340	/* Set MID associated with this context bank to 0*/
	341	SET_CBVMID(drvdata->base, c->num, 0);
	342
	343	/* Set the ASID for TLB tagging for this context */
	344	SET_CONTEXTIDR_ASID(drvdata->base, c->num, c->num);
c6a5951e SM	345
	346	/* Set security bit override to be Non-secure */
	347	SET_NSCFG(drvdata->base, mid, 3);
	348	}
	349
b61401ad SM	350	if (drvdata->clk)
	351	clk_disable(drvdata->clk);
	352	clk_disable(drvdata->pclk);
c6a5951e	353
b61401ad	354	dev_info(&pdev->dev, "context %s using bank %d\n", c->name, c->num);
c6a5951e SM	355	return 0;
	356	fail:
	357	kfree(ctx_drvdata);
	358	return ret;
	359	}
	360
	361	static int msm_iommu_ctx_remove(struct platform_device *pdev)
	362	{
	363	struct msm_iommu_ctx_drvdata *drv = NULL;
	364	drv = platform_get_drvdata(pdev);
	365	if (drv) {
	366	memset(drv, 0, sizeof(struct msm_iommu_ctx_drvdata));
	367	kfree(drv);
c6a5951e SM	368	}
	369	return 0;
	370	}
	371
	372	static struct platform_driver msm_iommu_driver = {
	373	.driver = {
	374	.name = "msm_iommu",
	375	},
	376	.probe = msm_iommu_probe,
	377	.remove = msm_iommu_remove,
	378	};
	379
	380	static struct platform_driver msm_iommu_ctx_driver = {
	381	.driver = {
	382	.name = "msm_iommu_ctx",
	383	},
	384	.probe = msm_iommu_ctx_probe,
	385	.remove = msm_iommu_ctx_remove,
	386	};
	387
516cbc79	388	static int __init msm_iommu_driver_init(void)
c6a5951e SM	389	{
	390	int ret;
	391	ret = platform_driver_register(&msm_iommu_driver);
	392	if (ret != 0) {
	393	pr_err("Failed to register IOMMU driver\n");
	394	goto error;
	395	}
	396
	397	ret = platform_driver_register(&msm_iommu_ctx_driver);
	398	if (ret != 0) {
34e3a58c	399	platform_driver_unregister(&msm_iommu_driver);
c6a5951e SM	400	pr_err("Failed to register IOMMU context driver\n");
	401	goto error;
	402	}
	403
	404	error:
	405	return ret;
	406	}
	407
516cbc79	408	static void __exit msm_iommu_driver_exit(void)
c6a5951e SM	409	{
	410	platform_driver_unregister(&msm_iommu_ctx_driver);
	411	platform_driver_unregister(&msm_iommu_driver);
	412	}
	413
	414	subsys_initcall(msm_iommu_driver_init);
	415	module_exit(msm_iommu_driver_exit);
	416
	417	MODULE_LICENSE("GPL v2");
	418	MODULE_AUTHOR("Stepan Moskovchenko <stepanm@codeaurora.org>");