summaryrefslogtreecommitdiffstats
path: root/drivers/mtd/partition.c
blob: df2eb40bf80d33141c24447703fec19cd1e8456d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
#include <common.h>
#include <errno.h>
#include <malloc.h>
#include <linux/err.h>
#include <linux/mtd/mtd.h>

struct mtd_part {
	struct mtd_info mtd;
	struct mtd_info *master;
	uint64_t offset;
	struct list_head list;
};

#define PART(x)  ((struct mtd_part *)(x))

static int mtd_part_read(struct mtd_info *mtd, loff_t from, size_t len,
                size_t *retlen, u_char *buf)
{
	struct mtd_part *part = PART(mtd);
	struct mtd_ecc_stats stats;
	int res;

	stats = part->master->ecc_stats;

	if (from >= mtd->size)
		len = 0;
	else if (from + len > mtd->size)
		len = mtd->size - from;
	res = part->master->read(part->master, from + part->offset,
				len, retlen, buf);
	return res;
}

static int mtd_part_write(struct mtd_info *mtd, loff_t to, size_t len,
                size_t *retlen, const u_char *buf)
{
	struct mtd_part *part = PART(mtd);

	if (!(mtd->flags & MTD_WRITEABLE))
		return -EROFS;
	if (to >= mtd->size)
		len = 0;
	else if (to + len > mtd->size)
		len = mtd->size - to;
	return part->master->write(part->master, to + part->offset,
					len, retlen, buf);
}

static int mtd_part_erase(struct mtd_info *mtd, struct erase_info *instr)
{
	struct mtd_part *part = PART(mtd);
	int ret;

	if (!(mtd->flags & MTD_WRITEABLE))
		return -EROFS;
	if (instr->addr >= mtd->size)
		return -EINVAL;
	instr->addr += part->offset;
	ret = part->master->erase(part->master, instr);
	if (ret) {
		if (instr->fail_addr != 0xffffffff)
			instr->fail_addr -= part->offset;
		instr->addr -= part->offset;
	}
	return ret;
}

static int mtd_part_block_isbad(struct mtd_info *mtd, loff_t ofs)
{
	struct mtd_part *part = PART(mtd);
	if (ofs >= mtd->size)
		return -EINVAL;
	ofs += part->offset;
	return part->master->block_isbad(part->master, ofs);
}

static int mtd_part_block_markbad(struct mtd_info *mtd, loff_t ofs)
{
	struct mtd_part *part = PART(mtd);
	int res;

	if (!(mtd->flags & MTD_WRITEABLE))
		return -EROFS;
	if (ofs >= mtd->size)
		return -EINVAL;
	ofs += part->offset;
	res = part->master->block_markbad(part->master, ofs);
	if (!res)
		mtd->ecc_stats.badblocks++;
	return res;
}

struct mtd_info *mtd_add_partition(struct mtd_info *mtd, off_t offset, size_t size,
		unsigned long flags, const char *name)
{
	struct mtd_part *slave;
	struct mtd_info *slave_mtd;
	int start = 0, end = 0, i;

	slave = xzalloc(sizeof(*slave));
	slave_mtd = &slave->mtd;

	memcpy(slave_mtd, mtd, sizeof(*slave));

	/*
	 * find the number of eraseregions the partition includes.
	 * Do not bother to create the mtd_erase_region_infos as
	 * ubi is only interested in its number. UBI does not
	 * yet support multiple erase regions.
	 */
	for (i = mtd->numeraseregions - 1; i >= 0; i--) {
		struct mtd_erase_region_info *region = &mtd->eraseregions[i];
		if (offset >= region->offset &&
		    offset < region->offset + region->erasesize * region->numblocks)
			start = i;
		if (offset + size >= region->offset &&
		    offset + size <= region->offset + region->erasesize * region->numblocks)
			end = i;
	}

	slave_mtd->numeraseregions = end - start;

	slave_mtd->read = mtd_part_read;
	slave_mtd->write = mtd_part_write;
	slave_mtd->erase = mtd_part_erase;
	slave_mtd->block_isbad = mtd->block_isbad ? mtd_part_block_isbad : NULL;
	slave_mtd->block_markbad = mtd->block_markbad ? mtd_part_block_markbad : NULL;
	slave_mtd->size = size;
	slave_mtd->name = strdup(name);

	slave->offset = offset;
	slave->master = mtd;

	return slave_mtd;
}

void mtd_del_partition(struct mtd_info *mtd)
{
	struct mtd_part *part = PART(mtd);

	free(mtd->name);
	free(part);
}