Commit | Line | Data |
---|---|---|
1da177e4 | 1 | /* |
1da177e4 LT |
2 | * Read flash partition table from command line |
3 | * | |
a1452a37 DW |
4 | * Copyright © 2002 SYSGO Real-Time Solutions GmbH |
5 | * Copyright © 2002-2010 David Woodhouse <dwmw2@infradead.org> | |
6 | * | |
7 | * This program is free software; you can redistribute it and/or modify | |
8 | * it under the terms of the GNU General Public License as published by | |
9 | * the Free Software Foundation; either version 2 of the License, or | |
10 | * (at your option) any later version. | |
11 | * | |
12 | * This program is distributed in the hope that it will be useful, | |
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
15 | * GNU General Public License for more details. | |
16 | * | |
17 | * You should have received a copy of the GNU General Public License | |
18 | * along with this program; if not, write to the Free Software | |
19 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA | |
1da177e4 LT |
20 | * |
21 | * The format for the command line is as follows: | |
97894cda | 22 | * |
1da177e4 LT |
23 | * mtdparts=<mtddef>[;<mtddef] |
24 | * <mtddef> := <mtd-id>:<partdef>[,<partdef>] | |
ea8b8e27 | 25 | * <partdef> := <size>[@<offset>][<name>][ro][lk] |
1da177e4 LT |
26 | * <mtd-id> := unique name used in mapping driver/device (mtd->name) |
27 | * <size> := standard linux memsize OR "-" to denote all remaining space | |
ea8b8e27 | 28 | * size is automatically truncated at end of device |
29f63f65 | 29 | * if specified or truncated size is 0 the part is skipped |
ea8b8e27 CC |
30 | * <offset> := standard linux memsize |
31 | * if omitted the part will immediately follow the previous part | |
32 | * or 0 if the first part | |
1da177e4 | 33 | * <name> := '(' NAME ')' |
bd6ce5ef | 34 | * NAME will appear in /proc/mtd |
97894cda | 35 | * |
ea8b8e27 CC |
36 | * <size> and <offset> can be specified such that the parts are out of order |
37 | * in physical memory and may even overlap. | |
38 | * | |
39 | * The parts are assigned MTD numbers in the order they are specified in the | |
40 | * command line regardless of their order in physical memory. | |
41 | * | |
1da177e4 | 42 | * Examples: |
97894cda | 43 | * |
1da177e4 LT |
44 | * 1 NOR Flash, with 1 single writable partition: |
45 | * edb7312-nor:- | |
97894cda | 46 | * |
1da177e4 LT |
47 | * 1 NOR Flash with 2 partitions, 1 NAND with one |
48 | * edb7312-nor:256k(ARMboot)ro,-(root);edb7312-nand:-(home) | |
49 | */ | |
50 | ||
51 | #include <linux/kernel.h> | |
52 | #include <linux/slab.h> | |
1da177e4 LT |
53 | #include <linux/mtd/mtd.h> |
54 | #include <linux/mtd/partitions.h> | |
a0e5cc58 | 55 | #include <linux/module.h> |
9e0606fc | 56 | #include <linux/err.h> |
1da177e4 LT |
57 | |
58 | /* error message prefix */ | |
59 | #define ERRP "mtd: " | |
60 | ||
61 | /* debug macro */ | |
62 | #if 0 | |
63 | #define dbg(x) do { printk("DEBUG-CMDLINE-PART: "); printk x; } while(0) | |
64 | #else | |
65 | #define dbg(x) | |
66 | #endif | |
67 | ||
68 | ||
69 | /* special size referring to all the remaining space in a partition */ | |
6f9f59ee HS |
70 | #define SIZE_REMAINING ULLONG_MAX |
71 | #define OFFSET_CONTINUOUS ULLONG_MAX | |
1da177e4 LT |
72 | |
73 | struct cmdline_mtd_partition { | |
74 | struct cmdline_mtd_partition *next; | |
75 | char *mtd_id; | |
76 | int num_parts; | |
77 | struct mtd_partition *parts; | |
78 | }; | |
79 | ||
80 | /* mtdpart_setup() parses into here */ | |
81 | static struct cmdline_mtd_partition *partitions; | |
82 | ||
8abc0d4a | 83 | /* the command line passed to mtdpart_setup() */ |
f5f172dc | 84 | static char *mtdparts; |
1da177e4 | 85 | static char *cmdline; |
aa3c5dc5 | 86 | static int cmdline_parsed; |
1da177e4 LT |
87 | |
88 | /* | |
89 | * Parse one partition definition for an MTD. Since there can be many | |
97894cda | 90 | * comma separated partition definitions, this function calls itself |
1da177e4 LT |
91 | * recursively until no more partition definitions are found. Nice side |
92 | * effect: the memory to keep the mtd_partition structs and the names | |
93 | * is allocated upon the last definition being found. At that point the | |
94 | * syntax has been verified ok. | |
95 | */ | |
97894cda | 96 | static struct mtd_partition * newpart(char *s, |
fac0077c AB |
97 | char **retptr, |
98 | int *num_parts, | |
99 | int this_part, | |
100 | unsigned char **extra_mem_ptr, | |
101 | int extra_mem_size) | |
1da177e4 LT |
102 | { |
103 | struct mtd_partition *parts; | |
6f9f59ee | 104 | unsigned long long size, offset = OFFSET_CONTINUOUS; |
1da177e4 LT |
105 | char *name; |
106 | int name_len; | |
107 | unsigned char *extra_mem; | |
108 | char delim; | |
109 | unsigned int mask_flags; | |
110 | ||
111 | /* fetch the partition size */ | |
fac0077c AB |
112 | if (*s == '-') { |
113 | /* assign all remaining space to this partition */ | |
1da177e4 LT |
114 | size = SIZE_REMAINING; |
115 | s++; | |
fac0077c | 116 | } else { |
1da177e4 | 117 | size = memparse(s, &s); |
d855d23b BN |
118 | if (!size) { |
119 | printk(KERN_ERR ERRP "partition has size 0\n"); | |
9e0606fc | 120 | return ERR_PTR(-EINVAL); |
1da177e4 LT |
121 | } |
122 | } | |
123 | ||
124 | /* fetch partition name and flags */ | |
125 | mask_flags = 0; /* this is going to be a regular partition */ | |
126 | delim = 0; | |
fac0077c AB |
127 | |
128 | /* check for offset */ | |
129 | if (*s == '@') { | |
130 | s++; | |
131 | offset = memparse(s, &s); | |
132 | } | |
133 | ||
134 | /* now look for name */ | |
1da177e4 | 135 | if (*s == '(') |
1da177e4 | 136 | delim = ')'; |
97894cda | 137 | |
fac0077c | 138 | if (delim) { |
1da177e4 LT |
139 | char *p; |
140 | ||
fac0077c | 141 | name = ++s; |
ed262c4f | 142 | p = strchr(name, delim); |
fac0077c | 143 | if (!p) { |
1da177e4 | 144 | printk(KERN_ERR ERRP "no closing %c found in partition name\n", delim); |
9e0606fc | 145 | return ERR_PTR(-EINVAL); |
1da177e4 LT |
146 | } |
147 | name_len = p - name; | |
148 | s = p + 1; | |
fac0077c AB |
149 | } else { |
150 | name = NULL; | |
1da177e4 LT |
151 | name_len = 13; /* Partition_000 */ |
152 | } | |
97894cda | 153 | |
1da177e4 LT |
154 | /* record name length for memory allocation later */ |
155 | extra_mem_size += name_len + 1; | |
156 | ||
fac0077c AB |
157 | /* test for options */ |
158 | if (strncmp(s, "ro", 2) == 0) { | |
1da177e4 LT |
159 | mask_flags |= MTD_WRITEABLE; |
160 | s += 2; | |
fac0077c | 161 | } |
1da177e4 | 162 | |
fac0077c AB |
163 | /* if lk is found do NOT unlock the MTD partition*/ |
164 | if (strncmp(s, "lk", 2) == 0) { | |
e619a75f JT |
165 | mask_flags |= MTD_POWERUP_LOCK; |
166 | s += 2; | |
fac0077c | 167 | } |
e619a75f | 168 | |
1da177e4 | 169 | /* test if more partitions are following */ |
fac0077c AB |
170 | if (*s == ',') { |
171 | if (size == SIZE_REMAINING) { | |
1da177e4 | 172 | printk(KERN_ERR ERRP "no partitions allowed after a fill-up partition\n"); |
9e0606fc | 173 | return ERR_PTR(-EINVAL); |
1da177e4 LT |
174 | } |
175 | /* more partitions follow, parse them */ | |
ed262c4f AB |
176 | parts = newpart(s + 1, &s, num_parts, this_part + 1, |
177 | &extra_mem, extra_mem_size); | |
9e0606fc AB |
178 | if (IS_ERR(parts)) |
179 | return parts; | |
fac0077c AB |
180 | } else { |
181 | /* this is the last partition: allocate space for all */ | |
1da177e4 LT |
182 | int alloc_size; |
183 | ||
184 | *num_parts = this_part + 1; | |
185 | alloc_size = *num_parts * sizeof(struct mtd_partition) + | |
186 | extra_mem_size; | |
fac0077c | 187 | |
95b93a0c | 188 | parts = kzalloc(alloc_size, GFP_KERNEL); |
1da177e4 | 189 | if (!parts) |
9e0606fc | 190 | return ERR_PTR(-ENOMEM); |
1da177e4 LT |
191 | extra_mem = (unsigned char *)(parts + *num_parts); |
192 | } | |
fac0077c | 193 | |
1da177e4 LT |
194 | /* enter this partition (offset will be calculated later if it is zero at this point) */ |
195 | parts[this_part].size = size; | |
196 | parts[this_part].offset = offset; | |
197 | parts[this_part].mask_flags = mask_flags; | |
198 | if (name) | |
1da177e4 | 199 | strlcpy(extra_mem, name, name_len + 1); |
1da177e4 | 200 | else |
1da177e4 | 201 | sprintf(extra_mem, "Partition_%03d", this_part); |
1da177e4 LT |
202 | parts[this_part].name = extra_mem; |
203 | extra_mem += name_len + 1; | |
204 | ||
342ba103 | 205 | dbg(("partition %d: name <%s>, offset %llx, size %llx, mask flags %x\n", |
fac0077c AB |
206 | this_part, parts[this_part].name, parts[this_part].offset, |
207 | parts[this_part].size, parts[this_part].mask_flags)); | |
1da177e4 LT |
208 | |
209 | /* return (updated) pointer to extra_mem memory */ | |
210 | if (extra_mem_ptr) | |
fac0077c | 211 | *extra_mem_ptr = extra_mem; |
1da177e4 LT |
212 | |
213 | /* return (updated) pointer command line string */ | |
214 | *retptr = s; | |
215 | ||
216 | /* return partition table */ | |
217 | return parts; | |
218 | } | |
219 | ||
97894cda TG |
220 | /* |
221 | * Parse the command line. | |
1da177e4 LT |
222 | */ |
223 | static int mtdpart_setup_real(char *s) | |
224 | { | |
225 | cmdline_parsed = 1; | |
226 | ||
227 | for( ; s != NULL; ) | |
228 | { | |
229 | struct cmdline_mtd_partition *this_mtd; | |
230 | struct mtd_partition *parts; | |
fac0077c | 231 | int mtd_id_len, num_parts; |
1da177e4 LT |
232 | char *p, *mtd_id; |
233 | ||
fac0077c AB |
234 | mtd_id = s; |
235 | ||
1da177e4 | 236 | /* fetch <mtd-id> */ |
fac0077c AB |
237 | p = strchr(s, ':'); |
238 | if (!p) { | |
1da177e4 | 239 | printk(KERN_ERR ERRP "no mtd-id\n"); |
9e0606fc | 240 | return -EINVAL; |
1da177e4 LT |
241 | } |
242 | mtd_id_len = p - mtd_id; | |
243 | ||
244 | dbg(("parsing <%s>\n", p+1)); | |
245 | ||
97894cda | 246 | /* |
1da177e4 LT |
247 | * parse one mtd. have it reserve memory for the |
248 | * struct cmdline_mtd_partition and the mtd-id string. | |
249 | */ | |
250 | parts = newpart(p + 1, /* cmdline */ | |
251 | &s, /* out: updated cmdline ptr */ | |
252 | &num_parts, /* out: number of parts */ | |
253 | 0, /* first partition */ | |
254 | (unsigned char**)&this_mtd, /* out: extra mem */ | |
97894cda | 255 | mtd_id_len + 1 + sizeof(*this_mtd) + |
be76c5fb | 256 | sizeof(void*)-1 /*alignment*/); |
fac0077c | 257 | if (IS_ERR(parts)) { |
1da177e4 LT |
258 | /* |
259 | * An error occurred. We're either: | |
260 | * a) out of memory, or | |
261 | * b) in the middle of the partition spec | |
262 | * Either way, this mtd is hosed and we're | |
263 | * unlikely to succeed in parsing any more | |
264 | */ | |
9e0606fc | 265 | return PTR_ERR(parts); |
1da177e4 LT |
266 | } |
267 | ||
be76c5fb | 268 | /* align this_mtd */ |
97894cda | 269 | this_mtd = (struct cmdline_mtd_partition *) |
fac0077c | 270 | ALIGN((unsigned long)this_mtd, sizeof(void *)); |
97894cda | 271 | /* enter results */ |
1da177e4 LT |
272 | this_mtd->parts = parts; |
273 | this_mtd->num_parts = num_parts; | |
274 | this_mtd->mtd_id = (char*)(this_mtd + 1); | |
275 | strlcpy(this_mtd->mtd_id, mtd_id, mtd_id_len + 1); | |
276 | ||
277 | /* link into chain */ | |
97894cda | 278 | this_mtd->next = partitions; |
1da177e4 LT |
279 | partitions = this_mtd; |
280 | ||
97894cda | 281 | dbg(("mtdid=<%s> num_parts=<%d>\n", |
1da177e4 | 282 | this_mtd->mtd_id, this_mtd->num_parts)); |
97894cda | 283 | |
1da177e4 LT |
284 | |
285 | /* EOS - we're done */ | |
286 | if (*s == 0) | |
287 | break; | |
288 | ||
289 | /* does another spec follow? */ | |
fac0077c | 290 | if (*s != ';') { |
1da177e4 | 291 | printk(KERN_ERR ERRP "bad character after partition (%c)\n", *s); |
9e0606fc | 292 | return -EINVAL; |
1da177e4 LT |
293 | } |
294 | s++; | |
295 | } | |
fac0077c | 296 | |
9e0606fc | 297 | return 0; |
1da177e4 LT |
298 | } |
299 | ||
300 | /* | |
301 | * Main function to be called from the MTD mapping driver/device to | |
302 | * obtain the partitioning information. At this point the command line | |
303 | * arguments will actually be parsed and turned to struct mtd_partition | |
304 | * information. It returns partitions for the requested mtd device, or | |
305 | * the first one in the chain if a NULL mtd_id is passed in. | |
306 | */ | |
97894cda | 307 | static int parse_cmdline_partitions(struct mtd_info *master, |
c7975330 DES |
308 | struct mtd_partition **pparts, |
309 | struct mtd_part_parser_data *data) | |
1da177e4 | 310 | { |
6f9f59ee | 311 | unsigned long long offset; |
9e0606fc | 312 | int i, err; |
1da177e4 | 313 | struct cmdline_mtd_partition *part; |
36560d25 | 314 | const char *mtd_id = master->name; |
1da177e4 | 315 | |
1da177e4 | 316 | /* parse command line */ |
9e0606fc AB |
317 | if (!cmdline_parsed) { |
318 | err = mtdpart_setup_real(cmdline); | |
319 | if (err) | |
320 | return err; | |
321 | } | |
1da177e4 | 322 | |
438db5a9 SL |
323 | /* |
324 | * Search for the partition definition matching master->name. | |
325 | * If master->name is not set, stop at first partition definition. | |
326 | */ | |
fac0077c | 327 | for (part = partitions; part; part = part->next) { |
438db5a9 SL |
328 | if ((!mtd_id) || (!strcmp(part->mtd_id, mtd_id))) |
329 | break; | |
330 | } | |
331 | ||
332 | if (!part) | |
333 | return 0; | |
334 | ||
335 | for (i = 0, offset = 0; i < part->num_parts; i++) { | |
336 | if (part->parts[i].offset == OFFSET_CONTINUOUS) | |
337 | part->parts[i].offset = offset; | |
338 | else | |
339 | offset = part->parts[i].offset; | |
340 | ||
341 | if (part->parts[i].size == SIZE_REMAINING) | |
342 | part->parts[i].size = master->size - offset; | |
343 | ||
ebf4f070 CC |
344 | if (offset + part->parts[i].size > master->size) { |
345 | printk(KERN_WARNING ERRP | |
346 | "%s: partitioning exceeds flash size, truncating\n", | |
347 | part->mtd_id); | |
348 | part->parts[i].size = master->size - offset; | |
349 | } | |
350 | offset += part->parts[i].size; | |
351 | ||
438db5a9 SL |
352 | if (part->parts[i].size == 0) { |
353 | printk(KERN_WARNING ERRP | |
354 | "%s: skipping zero sized partition\n", | |
355 | part->mtd_id); | |
356 | part->num_parts--; | |
357 | memmove(&part->parts[i], &part->parts[i + 1], | |
358 | sizeof(*part->parts) * (part->num_parts - i)); | |
e25e0a4d | 359 | i--; |
1da177e4 LT |
360 | } |
361 | } | |
fac0077c | 362 | |
438db5a9 SL |
363 | *pparts = kmemdup(part->parts, sizeof(*part->parts) * part->num_parts, |
364 | GFP_KERNEL); | |
365 | if (!*pparts) | |
366 | return -ENOMEM; | |
367 | ||
368 | return part->num_parts; | |
1da177e4 LT |
369 | } |
370 | ||
371 | ||
97894cda TG |
372 | /* |
373 | * This is the handler for our kernel parameter, called from | |
1da177e4 LT |
374 | * main.c::checksetup(). Note that we can not yet kmalloc() anything, |
375 | * so we only save the commandline for later processing. | |
376 | * | |
377 | * This function needs to be visible for bootloaders. | |
378 | */ | |
f5f172dc | 379 | static int __init mtdpart_setup(char *s) |
1da177e4 LT |
380 | { |
381 | cmdline = s; | |
382 | return 1; | |
383 | } | |
384 | ||
385 | __setup("mtdparts=", mtdpart_setup); | |
386 | ||
387 | static struct mtd_part_parser cmdline_parser = { | |
388 | .owner = THIS_MODULE, | |
389 | .parse_fn = parse_cmdline_partitions, | |
390 | .name = "cmdlinepart", | |
391 | }; | |
392 | ||
393 | static int __init cmdline_parser_init(void) | |
394 | { | |
f5f172dc LR |
395 | if (mtdparts) |
396 | mtdpart_setup(mtdparts); | |
6e14a61d AL |
397 | register_mtd_parser(&cmdline_parser); |
398 | return 0; | |
1da177e4 LT |
399 | } |
400 | ||
422f3890 LR |
401 | static void __exit cmdline_parser_exit(void) |
402 | { | |
403 | deregister_mtd_parser(&cmdline_parser); | |
404 | } | |
405 | ||
1da177e4 | 406 | module_init(cmdline_parser_init); |
422f3890 | 407 | module_exit(cmdline_parser_exit); |
1da177e4 | 408 | |
f5f172dc LR |
409 | MODULE_PARM_DESC(mtdparts, "Partitioning specification"); |
410 | module_param(mtdparts, charp, 0); | |
411 | ||
1da177e4 LT |
412 | MODULE_LICENSE("GPL"); |
413 | MODULE_AUTHOR("Marius Groeger <mag@sysgo.de>"); | |
414 | MODULE_DESCRIPTION("Command line configuration of MTD partitions"); |