[mirror_ubuntu-jammy-kernel.git] / drivers / misc / sram.c

/*
 * Generic on-chip SRAM allocation driver
 *
 * Copyright (C) 2012 Philipp Zabel, Pengutronix
 *
 * 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; either version 2
 * of the License, or (at your option) any later version.
 * 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.
 */

#include <linux/clk.h>
#include <linux/delay.h>
#include <linux/genalloc.h>
#include <linux/io.h>
#include <linux/list_sort.h>
#include <linux/of_address.h>
#include <linux/of_device.h>
#include <linux/platform_device.h>
#include <linux/regmap.h>
#include <linux/slab.h>
#include <linux/mfd/syscon.h>
#include <soc/at91/atmel-secumod.h>

#include "sram.h"

#define SRAM_GRANULARITY	32

static ssize_t sram_read(struct file *filp, struct kobject *kobj,
			 struct bin_attribute *attr,
			 char *buf, loff_t pos, size_t count)
{
	struct sram_partition *part;

	part = container_of(attr, struct sram_partition, battr);

	mutex_lock(&part->lock);
	memcpy_fromio(buf, part->base + pos, count);
	mutex_unlock(&part->lock);

	return count;
}

static ssize_t sram_write(struct file *filp, struct kobject *kobj,
			  struct bin_attribute *attr,
			  char *buf, loff_t pos, size_t count)
{
	struct sram_partition *part;

	part = container_of(attr, struct sram_partition, battr);

	mutex_lock(&part->lock);
	memcpy_toio(part->base + pos, buf, count);
	mutex_unlock(&part->lock);

	return count;
}

static int sram_add_pool(struct sram_dev *sram, struct sram_reserve *block,
			 phys_addr_t start, struct sram_partition *part)
{
	int ret;

	part->pool = devm_gen_pool_create(sram->dev, ilog2(SRAM_GRANULARITY),
					  NUMA_NO_NODE, block->label);
	if (IS_ERR(part->pool))
		return PTR_ERR(part->pool);

	ret = gen_pool_add_virt(part->pool, (unsigned long)part->base, start,
				block->size, NUMA_NO_NODE);
	if (ret < 0) {
		dev_err(sram->dev, "failed to register subpool: %d\n", ret);
		return ret;
	}

	return 0;
}

static int sram_add_export(struct sram_dev *sram, struct sram_reserve *block,
			   phys_addr_t start, struct sram_partition *part)
{
	sysfs_bin_attr_init(&part->battr);
	part->battr.attr.name = devm_kasprintf(sram->dev, GFP_KERNEL,
					       "%llx.sram",
					       (unsigned long long)start);
	if (!part->battr.attr.name)
		return -ENOMEM;

	part->battr.attr.mode = S_IRUSR | S_IWUSR;
	part->battr.read = sram_read;
	part->battr.write = sram_write;
	part->battr.size = block->size;

	return device_create_bin_file(sram->dev, &part->battr);
}

static int sram_add_partition(struct sram_dev *sram, struct sram_reserve *block,
			      phys_addr_t start)
{
	int ret;
	struct sram_partition *part = &sram->partition[sram->partitions];

	mutex_init(&part->lock);
	part->base = sram->virt_base + block->start;

	if (block->pool) {
		ret = sram_add_pool(sram, block, start, part);
		if (ret)
			return ret;
	}
	if (block->export) {
		ret = sram_add_export(sram, block, start, part);
		if (ret)
			return ret;
	}
	if (block->protect_exec) {
		ret = sram_check_protect_exec(sram, block, part);
		if (ret)
			return ret;

		ret = sram_add_pool(sram, block, start, part);
		if (ret)
			return ret;

		sram_add_protect_exec(part);
	}

	sram->partitions++;

	return 0;
}

static void sram_free_partitions(struct sram_dev *sram)
{
	struct sram_partition *part;

	if (!sram->partitions)
		return;

	part = &sram->partition[sram->partitions - 1];
	for (; sram->partitions; sram->partitions--, part--) {
		if (part->battr.size)
			device_remove_bin_file(sram->dev, &part->battr);

		if (part->pool &&
		    gen_pool_avail(part->pool) < gen_pool_size(part->pool))
			dev_err(sram->dev, "removed pool while SRAM allocated\n");
	}
}

static int sram_reserve_cmp(void *priv, struct list_head *a,
					struct list_head *b)
{
	struct sram_reserve *ra = list_entry(a, struct sram_reserve, list);
	struct sram_reserve *rb = list_entry(b, struct sram_reserve, list);

	return ra->start - rb->start;
}

static int sram_reserve_regions(struct sram_dev *sram, struct resource *res)
{
	struct device_node *np = sram->dev->of_node, *child;
	unsigned long size, cur_start, cur_size;
	struct sram_reserve *rblocks, *block;
	struct list_head reserve_list;
	unsigned int nblocks, exports = 0;
	const char *label;
	int ret = 0;

	INIT_LIST_HEAD(&reserve_list);

	size = resource_size(res);

	/*
	 * We need an additional block to mark the end of the memory region
	 * after the reserved blocks from the dt are processed.
	 */
	nblocks = (np) ? of_get_available_child_count(np) + 1 : 1;
	rblocks = kcalloc(nblocks, sizeof(*rblocks), GFP_KERNEL);
	if (!rblocks)
		return -ENOMEM;

	block = &rblocks[0];
	for_each_available_child_of_node(np, child) {
		struct resource child_res;

		ret = of_address_to_resource(child, 0, &child_res);
		if (ret < 0) {
			dev_err(sram->dev,
				"could not get address for node %pOF\n",
				child);
			goto err_chunks;
		}

		if (child_res.start < res->start || child_res.end > res->end) {
			dev_err(sram->dev,
				"reserved block %pOF outside the sram area\n",
				child);
			ret = -EINVAL;
			goto err_chunks;
		}

		block->start = child_res.start - res->start;
		block->size = resource_size(&child_res);
		list_add_tail(&block->list, &reserve_list);

		if (of_find_property(child, "export", NULL))
			block->export = true;

		if (of_find_property(child, "pool", NULL))
			block->pool = true;

		if (of_find_property(child, "protect-exec", NULL))
			block->protect_exec = true;

		if ((block->export || block->pool || block->protect_exec) &&
		    block->size) {
			exports++;

			label = NULL;
			ret = of_property_read_string(child, "label", &label);
			if (ret && ret != -EINVAL) {
				dev_err(sram->dev,
					"%pOF has invalid label name\n",
					child);
				goto err_chunks;
			}
			if (!label)
				label = child->name;

			block->label = devm_kstrdup(sram->dev,
						    label, GFP_KERNEL);
			if (!block->label) {
				ret = -ENOMEM;
				goto err_chunks;
			}

			dev_dbg(sram->dev, "found %sblock '%s' 0x%x-0x%x\n",
				block->export ? "exported " : "", block->label,
				block->start, block->start + block->size);
		} else {
			dev_dbg(sram->dev, "found reserved block 0x%x-0x%x\n",
				block->start, block->start + block->size);
		}

		block++;
	}
	child = NULL;

	/* the last chunk marks the end of the region */
	rblocks[nblocks - 1].start = size;
	rblocks[nblocks - 1].size = 0;
	list_add_tail(&rblocks[nblocks - 1].list, &reserve_list);

	list_sort(NULL, &reserve_list, sram_reserve_cmp);

	if (exports) {
		sram->partition = devm_kcalloc(sram->dev,
				       exports, sizeof(*sram->partition),
				       GFP_KERNEL);
		if (!sram->partition) {
			ret = -ENOMEM;
			goto err_chunks;
		}
	}

	cur_start = 0;
	list_for_each_entry(block, &reserve_list, list) {
		/* can only happen if sections overlap */
		if (block->start < cur_start) {
			dev_err(sram->dev,
				"block at 0x%x starts after current offset 0x%lx\n",
				block->start, cur_start);
			ret = -EINVAL;
			sram_free_partitions(sram);
			goto err_chunks;
		}

		if ((block->export || block->pool || block->protect_exec) &&
		    block->size) {
			ret = sram_add_partition(sram, block,
						 res->start + block->start);
			if (ret) {
				sram_free_partitions(sram);
				goto err_chunks;
			}
		}

		/* current start is in a reserved block, so continue after it */
		if (block->start == cur_start) {
			cur_start = block->start + block->size;
			continue;
		}

		/*
		 * allocate the space between the current starting
		 * address and the following reserved block, or the
		 * end of the region.
		 */
		cur_size = block->start - cur_start;

		dev_dbg(sram->dev, "adding chunk 0x%lx-0x%lx\n",
			cur_start, cur_start + cur_size);

		ret = gen_pool_add_virt(sram->pool,
				(unsigned long)sram->virt_base + cur_start,
				res->start + cur_start, cur_size, -1);
		if (ret < 0) {
			sram_free_partitions(sram);
			goto err_chunks;
		}

		/* next allocation after this reserved block */
		cur_start = block->start + block->size;
	}

err_chunks:
	of_node_put(child);
	kfree(rblocks);

	return ret;
}

static int atmel_securam_wait(void)
{
	struct regmap *regmap;
	u32 val;

	regmap = syscon_regmap_lookup_by_compatible("atmel,sama5d2-secumod");
	if (IS_ERR(regmap))
		return -ENODEV;

	return regmap_read_poll_timeout(regmap, AT91_SECUMOD_RAMRDY, val,
					val & AT91_SECUMOD_RAMRDY_READY,
					10000, 500000);
}

static const struct of_device_id sram_dt_ids[] = {
	{ .compatible = "mmio-sram" },
	{ .compatible = "atmel,sama5d2-securam", .data = atmel_securam_wait },
	{}
};

static int sram_probe(struct platform_device *pdev)
{
	struct sram_dev *sram;
	struct resource *res;
	size_t size;
	int ret;
	int (*init_func)(void);

	sram = devm_kzalloc(&pdev->dev, sizeof(*sram), GFP_KERNEL);
	if (!sram)
		return -ENOMEM;

	sram->dev = &pdev->dev;

	res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
	if (!res) {
		dev_err(sram->dev, "found no memory resource\n");
		return -EINVAL;
	}

	size = resource_size(res);

	if (!devm_request_mem_region(sram->dev, res->start, size, pdev->name)) {
		dev_err(sram->dev, "could not request region for resource\n");
		return -EBUSY;
	}

	if (of_property_read_bool(pdev->dev.of_node, "no-memory-wc"))
		sram->virt_base = devm_ioremap(sram->dev, res->start, size);
	else
		sram->virt_base = devm_ioremap_wc(sram->dev, res->start, size);
	if (!sram->virt_base)
		return -ENOMEM;

	sram->pool = devm_gen_pool_create(sram->dev, ilog2(SRAM_GRANULARITY),
					  NUMA_NO_NODE, NULL);
	if (IS_ERR(sram->pool))
		return PTR_ERR(sram->pool);

	sram->clk = devm_clk_get(sram->dev, NULL);
	if (IS_ERR(sram->clk))
		sram->clk = NULL;
	else
		clk_prepare_enable(sram->clk);

	ret = sram_reserve_regions(sram, res);
	if (ret)
		goto err_disable_clk;

	platform_set_drvdata(pdev, sram);

	init_func = of_device_get_match_data(&pdev->dev);
	if (init_func) {
		ret = init_func();
		if (ret)
			goto err_free_partitions;
	}

	dev_dbg(sram->dev, "SRAM pool: %zu KiB @ 0x%p\n",
		gen_pool_size(sram->pool) / 1024, sram->virt_base);

	return 0;

err_free_partitions:
	sram_free_partitions(sram);
err_disable_clk:
	if (sram->clk)
		clk_disable_unprepare(sram->clk);

	return ret;
}

static int sram_remove(struct platform_device *pdev)
{
	struct sram_dev *sram = platform_get_drvdata(pdev);

	sram_free_partitions(sram);

	if (gen_pool_avail(sram->pool) < gen_pool_size(sram->pool))
		dev_err(sram->dev, "removed while SRAM allocated\n");

	if (sram->clk)
		clk_disable_unprepare(sram->clk);

	return 0;
}

static struct platform_driver sram_driver = {
	.driver = {
		.name = "sram",
		.of_match_table = sram_dt_ids,
	},
	.probe = sram_probe,
	.remove = sram_remove,
};

static int __init sram_init(void)
{
	return platform_driver_register(&sram_driver);
}

postcore_initcall(sram_init);
Commit	Line	Data
4984c6f5 PZ	1	/*
	2	* Generic on-chip SRAM allocation driver
	3	*
	4	* Copyright (C) 2012 Philipp Zabel, Pengutronix
	5	*
	6	* This program is free software; you can redistribute it and/or
	7	* modify it under the terms of the GNU General Public License
	8	* as published by the Free Software Foundation; either version 2
	9	* of the License, or (at your option) any later version.
	10	* This program is distributed in the hope that it will be useful,
	11	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	13	* GNU General Public License for more details.
	14	*
	15	* You should have received a copy of the GNU General Public License
	16	* along with this program; if not, write to the Free Software
	17	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
	18	* MA 02110-1301, USA.
	19	*/
	20
4984c6f5	21	#include <linux/clk.h>
2ae2e288	22	#include <linux/delay.h>
98ce2d27	23	#include <linux/genalloc.h>
4984c6f5	24	#include <linux/io.h>
2da19688	25	#include <linux/list_sort.h>
98ce2d27	26	#include <linux/of_address.h>
2ae2e288	27	#include <linux/of_device.h>
4984c6f5	28	#include <linux/platform_device.h>
2ae2e288	29	#include <linux/regmap.h>
4984c6f5	30	#include <linux/slab.h>
2ae2e288 AB	31	#include <linux/mfd/syscon.h>
2ae2e288 AB	32	#include <soc/at91/atmel-secumod.h>
4984c6f5	33
cdd1737c	34	#include "sram.h"
665d82fb	35
cdd1737c	36	#define SRAM_GRANULARITY 32
2da19688	37
b4c3fcb3 VZ	38	static ssize_t sram_read(struct file filp, struct kobject kobj,
	39	struct bin_attribute *attr,
	40	char *buf, loff_t pos, size_t count)
	41	{
	42	struct sram_partition *part;
	43
	44	part = container_of(attr, struct sram_partition, battr);
	45
	46	mutex_lock(&part->lock);
525d12f2	47	memcpy_fromio(buf, part->base + pos, count);
b4c3fcb3 VZ	48	mutex_unlock(&part->lock);
	49
	50	return count;
	51	}
	52
	53	static ssize_t sram_write(struct file filp, struct kobject kobj,
	54	struct bin_attribute *attr,
	55	char *buf, loff_t pos, size_t count)
	56	{
	57	struct sram_partition *part;
	58
	59	part = container_of(attr, struct sram_partition, battr);
	60
	61	mutex_lock(&part->lock);
525d12f2	62	memcpy_toio(part->base + pos, buf, count);
b4c3fcb3 VZ	63	mutex_unlock(&part->lock);
	64
	65	return count;
	66	}
	67
	68	static int sram_add_pool(struct sram_dev sram, struct sram_reserve block,
	69	phys_addr_t start, struct sram_partition *part)
	70	{
	71	int ret;
	72
	73	part->pool = devm_gen_pool_create(sram->dev, ilog2(SRAM_GRANULARITY),
	74	NUMA_NO_NODE, block->label);
	75	if (IS_ERR(part->pool))
	76	return PTR_ERR(part->pool);
	77
	78	ret = gen_pool_add_virt(part->pool, (unsigned long)part->base, start,
	79	block->size, NUMA_NO_NODE);
	80	if (ret < 0) {
	81	dev_err(sram->dev, "failed to register subpool: %d\n", ret);
	82	return ret;
	83	}
	84
	85	return 0;
	86	}
	87
	88	static int sram_add_export(struct sram_dev sram, struct sram_reserve block,
	89	phys_addr_t start, struct sram_partition *part)
	90	{
	91	sysfs_bin_attr_init(&part->battr);
	92	part->battr.attr.name = devm_kasprintf(sram->dev, GFP_KERNEL,
	93	"%llx.sram",
	94	(unsigned long long)start);
	95	if (!part->battr.attr.name)
	96	return -ENOMEM;
	97
	98	part->battr.attr.mode = S_IRUSR \| S_IWUSR;
	99	part->battr.read = sram_read;
	100	part->battr.write = sram_write;
	101	part->battr.size = block->size;
	102
	103	return device_create_bin_file(sram->dev, &part->battr);
	104	}
	105
	106	static int sram_add_partition(struct sram_dev sram, struct sram_reserve block,
	107	phys_addr_t start)
	108	{
	109	int ret;
	110	struct sram_partition *part = &sram->partition[sram->partitions];
	111
	112	mutex_init(&part->lock);
	113	part->base = sram->virt_base + block->start;
	114
	115	if (block->pool) {
	116	ret = sram_add_pool(sram, block, start, part);
	117	if (ret)
	118	return ret;
	119	}
	120	if (block->export) {
	121	ret = sram_add_export(sram, block, start, part);
	122	if (ret)
	123	return ret;
	124	}
37afff0d DG	125	if (block->protect_exec) {
	126	ret = sram_check_protect_exec(sram, block, part);
	127	if (ret)
	128	return ret;
	129
	130	ret = sram_add_pool(sram, block, start, part);
	131	if (ret)
	132	return ret;
	133
	134	sram_add_protect_exec(part);
	135	}
	136
b4c3fcb3 VZ	137	sram->partitions++;
	138
	139	return 0;
	140	}
	141
	142	static void sram_free_partitions(struct sram_dev *sram)
	143	{
	144	struct sram_partition *part;
	145
	146	if (!sram->partitions)
	147	return;
	148
	149	part = &sram->partition[sram->partitions - 1];
	150	for (; sram->partitions; sram->partitions--, part--) {
	151	if (part->battr.size)
	152	device_remove_bin_file(sram->dev, &part->battr);
	153
	154	if (part->pool &&
	155	gen_pool_avail(part->pool) < gen_pool_size(part->pool))
	156	dev_err(sram->dev, "removed pool while SRAM allocated\n");
	157	}
	158	}
	159
2da19688 HS	160	static int sram_reserve_cmp(void priv, struct list_head a,
	161	struct list_head *b)
	162	{
	163	struct sram_reserve *ra = list_entry(a, struct sram_reserve, list);
	164	struct sram_reserve *rb = list_entry(b, struct sram_reserve, list);
	165
	166	return ra->start - rb->start;
	167	}
	168
a0a5be0b	169	static int sram_reserve_regions(struct sram_dev sram, struct resource res)
4984c6f5	170	{
a0a5be0b	171	struct device_node np = sram->dev->of_node, child;
2da19688 HS	172	unsigned long size, cur_start, cur_size;
	173	struct sram_reserve rblocks, block;
	174	struct list_head reserve_list;
b4c3fcb3 VZ	175	unsigned int nblocks, exports = 0;
b4c3fcb3 VZ	176	const char *label;
a0a5be0b	177	int ret = 0;
4984c6f5	178
2da19688 HS	179	INIT_LIST_HEAD(&reserve_list);
2da19688 HS	180
f3cbfa5d	181	size = resource_size(res);
4984c6f5	182
2da19688 HS	183	/*
	184	* We need an additional block to mark the end of the memory region
	185	* after the reserved blocks from the dt are processed.
	186	*/
	187	nblocks = (np) ? of_get_available_child_count(np) + 1 : 1;
6396bb22	188	rblocks = kcalloc(nblocks, sizeof(*rblocks), GFP_KERNEL);
ee895ccd VZ	189	if (!rblocks)
ee895ccd VZ	190	return -ENOMEM;
4984c6f5	191
2da19688 HS	192	block = &rblocks[0];
	193	for_each_available_child_of_node(np, child) {
	194	struct resource child_res;
	195
	196	ret = of_address_to_resource(child, 0, &child_res);
	197	if (ret < 0) {
665d82fb	198	dev_err(sram->dev,
34d0eb50 RH	199	"could not get address for node %pOF\n",
34d0eb50 RH	200	child);
2da19688 HS	201	goto err_chunks;
	202	}
	203
	204	if (child_res.start < res->start \|\| child_res.end > res->end) {
665d82fb	205	dev_err(sram->dev,
34d0eb50 RH	206	"reserved block %pOF outside the sram area\n",
34d0eb50 RH	207	child);
2da19688 HS	208	ret = -EINVAL;
	209	goto err_chunks;
	210	}
	211
	212	block->start = child_res.start - res->start;
	213	block->size = resource_size(&child_res);
	214	list_add_tail(&block->list, &reserve_list);
	215
b4c3fcb3 VZ	216	if (of_find_property(child, "export", NULL))
	217	block->export = true;
	218
	219	if (of_find_property(child, "pool", NULL))
	220	block->pool = true;
	221
37afff0d DG	222	if (of_find_property(child, "protect-exec", NULL))
	223	block->protect_exec = true;
	224
	225	if ((block->export \|\| block->pool \|\| block->protect_exec) &&
	226	block->size) {
b4c3fcb3 VZ	227	exports++;
	228
	229	label = NULL;
	230	ret = of_property_read_string(child, "label", &label);
	231	if (ret && ret != -EINVAL) {
	232	dev_err(sram->dev,
34d0eb50 RH	233	"%pOF has invalid label name\n",
34d0eb50 RH	234	child);
b4c3fcb3 VZ	235	goto err_chunks;
	236	}
	237	if (!label)
	238	label = child->name;
	239
	240	block->label = devm_kstrdup(sram->dev,
	241	label, GFP_KERNEL);
ddc5c9a3 PB	242	if (!block->label) {
ddc5c9a3 PB	243	ret = -ENOMEM;
b4c3fcb3	244	goto err_chunks;
ddc5c9a3	245	}
b4c3fcb3 VZ	246
	247	dev_dbg(sram->dev, "found %sblock '%s' 0x%x-0x%x\n",
	248	block->export ? "exported " : "", block->label,
	249	block->start, block->start + block->size);
	250	} else {
	251	dev_dbg(sram->dev, "found reserved block 0x%x-0x%x\n",
	252	block->start, block->start + block->size);
	253	}
2da19688 HS	254
	255	block++;
	256	}
b4c3fcb3	257	child = NULL;
2da19688 HS	258
	259	/* the last chunk marks the end of the region */
	260	rblocks[nblocks - 1].start = size;
	261	rblocks[nblocks - 1].size = 0;
	262	list_add_tail(&rblocks[nblocks - 1].list, &reserve_list);
	263
	264	list_sort(NULL, &reserve_list, sram_reserve_cmp);
	265
b4c3fcb3	266	if (exports) {
a86854d0 KC	267	sram->partition = devm_kcalloc(sram->dev,
a86854d0 KC	268	exports, sizeof(*sram->partition),
b4c3fcb3 VZ	269	GFP_KERNEL);
	270	if (!sram->partition) {
	271	ret = -ENOMEM;
	272	goto err_chunks;
	273	}
	274	}
2da19688	275
b4c3fcb3	276	cur_start = 0;
2da19688 HS	277	list_for_each_entry(block, &reserve_list, list) {
	278	/* can only happen if sections overlap */
	279	if (block->start < cur_start) {
665d82fb	280	dev_err(sram->dev,
2da19688 HS	281	"block at 0x%x starts after current offset 0x%lx\n",
	282	block->start, cur_start);
	283	ret = -EINVAL;
b4c3fcb3	284	sram_free_partitions(sram);
2da19688 HS	285	goto err_chunks;
	286	}
	287
37afff0d DG	288	if ((block->export \|\| block->pool \|\| block->protect_exec) &&
37afff0d DG	289	block->size) {
b4c3fcb3 VZ	290	ret = sram_add_partition(sram, block,
	291	res->start + block->start);
	292	if (ret) {
	293	sram_free_partitions(sram);
	294	goto err_chunks;
	295	}
	296	}
	297
2da19688 HS	298	/* current start is in a reserved block, so continue after it */
	299	if (block->start == cur_start) {
	300	cur_start = block->start + block->size;
	301	continue;
	302	}
	303
	304	/*
	305	* allocate the space between the current starting
	306	* address and the following reserved block, or the
	307	* end of the region.
	308	*/
	309	cur_size = block->start - cur_start;
	310
665d82fb	311	dev_dbg(sram->dev, "adding chunk 0x%lx-0x%lx\n",
2da19688	312	cur_start, cur_start + cur_size);
665d82fb	313
2da19688	314	ret = gen_pool_add_virt(sram->pool,
665d82fb	315	(unsigned long)sram->virt_base + cur_start,
2da19688	316	res->start + cur_start, cur_size, -1);
b4c3fcb3 VZ	317	if (ret < 0) {
b4c3fcb3 VZ	318	sram_free_partitions(sram);
2da19688	319	goto err_chunks;
b4c3fcb3	320	}
2da19688 HS	321
	322	/* next allocation after this reserved block */
	323	cur_start = block->start + block->size;
	324	}
	325
3104389e	326	err_chunks:
3104389e	327	of_node_put(child);
2da19688 HS	328	kfree(rblocks);
2da19688 HS	329
a0a5be0b VZ	330	return ret;
	331	}
	332
2ae2e288 AB	333	static int atmel_securam_wait(void)
	334	{
	335	struct regmap *regmap;
	336	u32 val;
	337
	338	regmap = syscon_regmap_lookup_by_compatible("atmel,sama5d2-secumod");
	339	if (IS_ERR(regmap))
	340	return -ENODEV;
	341
	342	return regmap_read_poll_timeout(regmap, AT91_SECUMOD_RAMRDY, val,
	343	val & AT91_SECUMOD_RAMRDY_READY,
	344	10000, 500000);
	345	}
	346
2ae2e288 AB	347	static const struct of_device_id sram_dt_ids[] = {
	348	{ .compatible = "mmio-sram" },
	349	{ .compatible = "atmel,sama5d2-securam", .data = atmel_securam_wait },
	350	{}
	351	};
2ae2e288	352
a0a5be0b VZ	353	static int sram_probe(struct platform_device *pdev)
	354	{
	355	struct sram_dev *sram;
	356	struct resource *res;
	357	size_t size;
	358	int ret;
2ae2e288	359	int (*init_func)(void);
a0a5be0b VZ	360
	361	sram = devm_kzalloc(&pdev->dev, sizeof(*sram), GFP_KERNEL);
	362	if (!sram)
	363	return -ENOMEM;
	364
	365	sram->dev = &pdev->dev;
	366
	367	res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
	368	if (!res) {
	369	dev_err(sram->dev, "found no memory resource\n");
	370	return -EINVAL;
	371	}
	372
	373	size = resource_size(res);
	374
	375	if (!devm_request_mem_region(sram->dev, res->start, size, pdev->name)) {
	376	dev_err(sram->dev, "could not request region for resource\n");
	377	return -EBUSY;
	378	}
	379
eb43e023 MW	380	if (of_property_read_bool(pdev->dev.of_node, "no-memory-wc"))
	381	sram->virt_base = devm_ioremap(sram->dev, res->start, size);
	382	else
	383	sram->virt_base = devm_ioremap_wc(sram->dev, res->start, size);
d449d69d VZ	384	if (!sram->virt_base)
d449d69d VZ	385	return -ENOMEM;
a0a5be0b	386
73858173 VZ	387	sram->pool = devm_gen_pool_create(sram->dev, ilog2(SRAM_GRANULARITY),
	388	NUMA_NO_NODE, NULL);
	389	if (IS_ERR(sram->pool))
	390	return PTR_ERR(sram->pool);
a0a5be0b	391
665d82fb	392	sram->clk = devm_clk_get(sram->dev, NULL);
ee895ccd VZ	393	if (IS_ERR(sram->clk))
	394	sram->clk = NULL;
	395	else
	396	clk_prepare_enable(sram->clk);
	397
d5b9653d JH	398	ret = sram_reserve_regions(sram, res);
	399	if (ret)
	400	goto err_disable_clk;
	401
4984c6f5 PZ	402	platform_set_drvdata(pdev, sram);
4984c6f5 PZ	403
2ae2e288 AB	404	init_func = of_device_get_match_data(&pdev->dev);
	405	if (init_func) {
	406	ret = init_func();
	407	if (ret)
d5b9653d	408	goto err_free_partitions;
2ae2e288 AB	409	}
2ae2e288 AB	410
665d82fb VZ	411	dev_dbg(sram->dev, "SRAM pool: %zu KiB @ 0x%p\n",
665d82fb VZ	412	gen_pool_size(sram->pool) / 1024, sram->virt_base);
4984c6f5 PZ	413
4984c6f5 PZ	414	return 0;
f294d009	415
d5b9653d JH	416	err_free_partitions:
d5b9653d JH	417	sram_free_partitions(sram);
f294d009 JH	418	err_disable_clk:
	419	if (sram->clk)
	420	clk_disable_unprepare(sram->clk);
f294d009 JH	421
f294d009 JH	422	return ret;
4984c6f5 PZ	423	}
	424
	425	static int sram_remove(struct platform_device *pdev)
	426	{
	427	struct sram_dev *sram = platform_get_drvdata(pdev);
	428
b4c3fcb3 VZ	429	sram_free_partitions(sram);
b4c3fcb3 VZ	430
4984c6f5	431	if (gen_pool_avail(sram->pool) < gen_pool_size(sram->pool))
665d82fb	432	dev_err(sram->dev, "removed while SRAM allocated\n");
4984c6f5	433
4984c6f5 PZ	434	if (sram->clk)
	435	clk_disable_unprepare(sram->clk);
	436
	437	return 0;
	438	}
	439
4984c6f5 PZ	440	static struct platform_driver sram_driver = {
	441	.driver = {
	442	.name = "sram",
2aa488a6	443	.of_match_table = sram_dt_ids,
4984c6f5 PZ	444	},
	445	.probe = sram_probe,
	446	.remove = sram_remove,
	447	};
	448
	449	static int __init sram_init(void)
	450	{
	451	return platform_driver_register(&sram_driver);
	452	}
	453
	454	postcore_initcall(sram_init);