Merge branch 'u-boot/master' into 'u-boot-arm/master'
[platform/kernel/u-boot.git] / common / cmd_mtdparts.c
1 /*
2  * (C) Copyright 2002
3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4  *
5  * (C) Copyright 2002
6  * Robert Schwebel, Pengutronix, <r.schwebel@pengutronix.de>
7  *
8  * (C) Copyright 2003
9  * Kai-Uwe Bloem, Auerswald GmbH & Co KG, <linux-development@auerswald.de>
10  *
11  * (C) Copyright 2005
12  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
13  *
14  *   Added support for reading flash partition table from environment.
15  *   Parsing routines are based on driver/mtd/cmdline.c from the linux 2.4
16  *   kernel tree.
17  *
18  * (C) Copyright 2008
19  * Harald Welte, OpenMoko, Inc., Harald Welte <laforge@openmoko.org>
20  *
21  *   $Id: cmdlinepart.c,v 1.17 2004/11/26 11:18:47 lavinen Exp $
22  *   Copyright 2002 SYSGO Real-Time Solutions GmbH
23  *
24  * SPDX-License-Identifier:     GPL-2.0+
25  */
26
27 /*
28  * Three environment variables are used by the parsing routines:
29  *
30  * 'partition' - keeps current partition identifier
31  *
32  * partition  := <part-id>
33  * <part-id>  := <dev-id>,part_num
34  *
35  *
36  * 'mtdids' - linux kernel mtd device id <-> u-boot device id mapping
37  *
38  * mtdids=<idmap>[,<idmap>,...]
39  *
40  * <idmap>    := <dev-id>=<mtd-id>
41  * <dev-id>   := 'nand'|'nor'|'onenand'<dev-num>
42  * <dev-num>  := mtd device number, 0...
43  * <mtd-id>   := unique device tag used by linux kernel to find mtd device (mtd->name)
44  *
45  *
46  * 'mtdparts' - partition list
47  *
48  * mtdparts=mtdparts=<mtd-def>[;<mtd-def>...]
49  *
50  * <mtd-def>  := <mtd-id>:<part-def>[,<part-def>...]
51  * <mtd-id>   := unique device tag used by linux kernel to find mtd device (mtd->name)
52  * <part-def> := <size>[@<offset>][<name>][<ro-flag>]
53  * <size>     := standard linux memsize OR '-' to denote all remaining space
54  * <offset>   := partition start offset within the device
55  * <name>     := '(' NAME ')'
56  * <ro-flag>  := when set to 'ro' makes partition read-only (not used, passed to kernel)
57  *
58  * Notes:
59  * - each <mtd-id> used in mtdparts must albo exist in 'mtddis' mapping
60  * - if the above variables are not set defaults for a given target are used
61  *
62  * Examples:
63  *
64  * 1 NOR Flash, with 1 single writable partition:
65  * mtdids=nor0=edb7312-nor
66  * mtdparts=mtdparts=edb7312-nor:-
67  *
68  * 1 NOR Flash with 2 partitions, 1 NAND with one
69  * mtdids=nor0=edb7312-nor,nand0=edb7312-nand
70  * mtdparts=mtdparts=edb7312-nor:256k(ARMboot)ro,-(root);edb7312-nand:-(home)
71  *
72  */
73
74 #include <common.h>
75 #include <command.h>
76 #include <malloc.h>
77 #include <jffs2/load_kernel.h>
78 #include <linux/list.h>
79 #include <linux/ctype.h>
80 #include <linux/err.h>
81 #include <linux/mtd/mtd.h>
82
83 #if defined(CONFIG_CMD_NAND)
84 #include <linux/mtd/nand.h>
85 #include <nand.h>
86 #endif
87
88 #if defined(CONFIG_CMD_ONENAND)
89 #include <linux/mtd/onenand.h>
90 #include <onenand_uboot.h>
91 #endif
92
93 DECLARE_GLOBAL_DATA_PTR;
94
95 /* special size referring to all the remaining space in a partition */
96 #define SIZE_REMAINING          (~0llu)
97
98 /* special offset value, it is used when not provided by user
99  *
100  * this value is used temporarily during parsing, later such offests
101  * are recalculated */
102 #define OFFSET_NOT_SPECIFIED    (~0llu)
103
104 /* minimum partition size */
105 #define MIN_PART_SIZE           4096
106
107 /* this flag needs to be set in part_info struct mask_flags
108  * field for read-only partitions */
109 #define MTD_WRITEABLE_CMD               1
110
111 /* default values for mtdids and mtdparts variables */
112 #if defined(MTDIDS_DEFAULT)
113 static const char *const mtdids_default = MTDIDS_DEFAULT;
114 #else
115 static const char *const mtdids_default = NULL;
116 #endif
117
118 #if defined(MTDPARTS_DEFAULT)
119 static const char *const mtdparts_default = MTDPARTS_DEFAULT;
120 #else
121 static const char *const mtdparts_default = NULL;
122 #endif
123
124 /* copies of last seen 'mtdids', 'mtdparts' and 'partition' env variables */
125 #define MTDIDS_MAXLEN           128
126 #define MTDPARTS_MAXLEN         512
127 #define PARTITION_MAXLEN        16
128 static char last_ids[MTDIDS_MAXLEN];
129 static char last_parts[MTDPARTS_MAXLEN];
130 static char last_partition[PARTITION_MAXLEN];
131
132 /* low level jffs2 cache cleaning routine */
133 extern void jffs2_free_cache(struct part_info *part);
134
135 /* mtdids mapping list, filled by parse_ids() */
136 static struct list_head mtdids;
137
138 /* device/partition list, parse_cmdline() parses into here */
139 static struct list_head devices;
140
141 /* current active device and partition number */
142 struct mtd_device *current_mtd_dev = NULL;
143 u8 current_mtd_partnum = 0;
144
145 static struct part_info* mtd_part_info(struct mtd_device *dev, unsigned int part_num);
146
147 /* command line only routines */
148 static struct mtdids* id_find_by_mtd_id(const char *mtd_id, unsigned int mtd_id_len);
149 static int device_del(struct mtd_device *dev);
150
151 /**
152  * Parses a string into a number.  The number stored at ptr is
153  * potentially suffixed with K (for kilobytes, or 1024 bytes),
154  * M (for megabytes, or 1048576 bytes), or G (for gigabytes, or
155  * 1073741824).  If the number is suffixed with K, M, or G, then
156  * the return value is the number multiplied by one kilobyte, one
157  * megabyte, or one gigabyte, respectively.
158  *
159  * @param ptr where parse begins
160  * @param retptr output pointer to next char after parse completes (output)
161  * @return resulting unsigned int
162  */
163 static u64 memsize_parse (const char *const ptr, const char **retptr)
164 {
165         u64 ret = simple_strtoull(ptr, (char **)retptr, 0);
166
167         switch (**retptr) {
168                 case 'G':
169                 case 'g':
170                         ret <<= 10;
171                 case 'M':
172                 case 'm':
173                         ret <<= 10;
174                 case 'K':
175                 case 'k':
176                         ret <<= 10;
177                         (*retptr)++;
178                 default:
179                         break;
180         }
181
182         return ret;
183 }
184
185 /**
186  * Format string describing supplied size. This routine does the opposite job
187  * to memsize_parse(). Size in bytes is converted to string and if possible
188  * shortened by using k (kilobytes), m (megabytes) or g (gigabytes) suffix.
189  *
190  * Note, that this routine does not check for buffer overflow, it's the caller
191  * who must assure enough space.
192  *
193  * @param buf output buffer
194  * @param size size to be converted to string
195  */
196 static void memsize_format(char *buf, u64 size)
197 {
198 #define SIZE_GB ((u32)1024*1024*1024)
199 #define SIZE_MB ((u32)1024*1024)
200 #define SIZE_KB ((u32)1024)
201
202         if ((size % SIZE_GB) == 0)
203                 sprintf(buf, "%llug", size/SIZE_GB);
204         else if ((size % SIZE_MB) == 0)
205                 sprintf(buf, "%llum", size/SIZE_MB);
206         else if (size % SIZE_KB == 0)
207                 sprintf(buf, "%lluk", size/SIZE_KB);
208         else
209                 sprintf(buf, "%llu", size);
210 }
211
212 /**
213  * This routine does global indexing of all partitions. Resulting index for
214  * current partition is saved in 'mtddevnum'. Current partition name in
215  * 'mtddevname'.
216  */
217 static void index_partitions(void)
218 {
219         u16 mtddevnum;
220         struct part_info *part;
221         struct list_head *dentry;
222         struct mtd_device *dev;
223
224         debug("--- index partitions ---\n");
225
226         if (current_mtd_dev) {
227                 mtddevnum = 0;
228                 list_for_each(dentry, &devices) {
229                         dev = list_entry(dentry, struct mtd_device, link);
230                         if (dev == current_mtd_dev) {
231                                 mtddevnum += current_mtd_partnum;
232                                 setenv_ulong("mtddevnum", mtddevnum);
233                                 break;
234                         }
235                         mtddevnum += dev->num_parts;
236                 }
237
238                 part = mtd_part_info(current_mtd_dev, current_mtd_partnum);
239                 setenv("mtddevname", part->name);
240
241                 debug("=> mtddevnum %d,\n=> mtddevname %s\n", mtddevnum, part->name);
242         } else {
243                 setenv("mtddevnum", NULL);
244                 setenv("mtddevname", NULL);
245
246                 debug("=> mtddevnum NULL\n=> mtddevname NULL\n");
247         }
248 }
249
250 /**
251  * Save current device and partition in environment variable 'partition'.
252  */
253 static void current_save(void)
254 {
255         char buf[16];
256
257         debug("--- current_save ---\n");
258
259         if (current_mtd_dev) {
260                 sprintf(buf, "%s%d,%d", MTD_DEV_TYPE(current_mtd_dev->id->type),
261                                         current_mtd_dev->id->num, current_mtd_partnum);
262
263                 setenv("partition", buf);
264                 strncpy(last_partition, buf, 16);
265
266                 debug("=> partition %s\n", buf);
267         } else {
268                 setenv("partition", NULL);
269                 last_partition[0] = '\0';
270
271                 debug("=> partition NULL\n");
272         }
273         index_partitions();
274 }
275
276
277 /**
278  * Produce a mtd_info given a type and num.
279  *
280  * @param type mtd type
281  * @param num mtd number
282  * @param mtd a pointer to an mtd_info instance (output)
283  * @return 0 if device is valid, 1 otherwise
284  */
285 static int get_mtd_info(u8 type, u8 num, struct mtd_info **mtd)
286 {
287         char mtd_dev[16];
288
289         sprintf(mtd_dev, "%s%d", MTD_DEV_TYPE(type), num);
290         *mtd = get_mtd_device_nm(mtd_dev);
291         if (IS_ERR(*mtd)) {
292                 printf("Device %s not found!\n", mtd_dev);
293                 return 1;
294         }
295         put_mtd_device(*mtd);
296
297         return 0;
298 }
299
300 /**
301  * Performs sanity check for supplied flash partition.
302  * Table of existing MTD flash devices is searched and partition device
303  * is located. Alignment with the granularity of nand erasesize is verified.
304  *
305  * @param id of the parent device
306  * @param part partition to validate
307  * @return 0 if partition is valid, 1 otherwise
308  */
309 static int part_validate_eraseblock(struct mtdids *id, struct part_info *part)
310 {
311         struct mtd_info *mtd = NULL;
312         int i, j;
313         ulong start;
314         u64 offset, size;
315
316         if (get_mtd_info(id->type, id->num, &mtd))
317                 return 1;
318
319         part->sector_size = mtd->erasesize;
320
321         if (!mtd->numeraseregions) {
322                 /*
323                  * Only one eraseregion (NAND, OneNAND or uniform NOR),
324                  * checking for alignment is easy here
325                  */
326                 offset = part->offset;
327                 if (do_div(offset, mtd->erasesize)) {
328                         printf("%s%d: partition (%s) start offset"
329                                "alignment incorrect\n",
330                                MTD_DEV_TYPE(id->type), id->num, part->name);
331                         return 1;
332                 }
333
334                 size = part->size;
335                 if (do_div(size, mtd->erasesize)) {
336                         printf("%s%d: partition (%s) size alignment incorrect\n",
337                                MTD_DEV_TYPE(id->type), id->num, part->name);
338                         return 1;
339                 }
340         } else {
341                 /*
342                  * Multiple eraseregions (non-uniform NOR),
343                  * checking for alignment is more complex here
344                  */
345
346                 /* Check start alignment */
347                 for (i = 0; i < mtd->numeraseregions; i++) {
348                         start = mtd->eraseregions[i].offset;
349                         for (j = 0; j < mtd->eraseregions[i].numblocks; j++) {
350                                 if (part->offset == start)
351                                         goto start_ok;
352                                 start += mtd->eraseregions[i].erasesize;
353                         }
354                 }
355
356                 printf("%s%d: partition (%s) start offset alignment incorrect\n",
357                        MTD_DEV_TYPE(id->type), id->num, part->name);
358                 return 1;
359
360         start_ok:
361
362                 /* Check end/size alignment */
363                 for (i = 0; i < mtd->numeraseregions; i++) {
364                         start = mtd->eraseregions[i].offset;
365                         for (j = 0; j < mtd->eraseregions[i].numblocks; j++) {
366                                 if ((part->offset + part->size) == start)
367                                         goto end_ok;
368                                 start += mtd->eraseregions[i].erasesize;
369                         }
370                 }
371                 /* Check last sector alignment */
372                 if ((part->offset + part->size) == start)
373                         goto end_ok;
374
375                 printf("%s%d: partition (%s) size alignment incorrect\n",
376                        MTD_DEV_TYPE(id->type), id->num, part->name);
377                 return 1;
378
379         end_ok:
380                 return 0;
381         }
382
383         return 0;
384 }
385
386
387 /**
388  * Performs sanity check for supplied partition. Offset and size are
389  * verified to be within valid range. Partition type is checked and
390  * part_validate_eraseblock() is called with the argument of part.
391  *
392  * @param id of the parent device
393  * @param part partition to validate
394  * @return 0 if partition is valid, 1 otherwise
395  */
396 static int part_validate(struct mtdids *id, struct part_info *part)
397 {
398         if (part->size == SIZE_REMAINING)
399                 part->size = id->size - part->offset;
400
401         if (part->offset > id->size) {
402                 printf("%s: offset %08llx beyond flash size %08llx\n",
403                                 id->mtd_id, part->offset, id->size);
404                 return 1;
405         }
406
407         if ((part->offset + part->size) <= part->offset) {
408                 printf("%s%d: partition (%s) size too big\n",
409                                 MTD_DEV_TYPE(id->type), id->num, part->name);
410                 return 1;
411         }
412
413         if (part->offset + part->size > id->size) {
414                 printf("%s: partitioning exceeds flash size\n", id->mtd_id);
415                 return 1;
416         }
417
418         /*
419          * Now we need to check if the partition starts and ends on
420          * sector (eraseblock) regions
421          */
422         return part_validate_eraseblock(id, part);
423 }
424
425 /**
426  * Delete selected partition from the partition list of the specified device.
427  *
428  * @param dev device to delete partition from
429  * @param part partition to delete
430  * @return 0 on success, 1 otherwise
431  */
432 static int part_del(struct mtd_device *dev, struct part_info *part)
433 {
434         u8 current_save_needed = 0;
435
436         /* if there is only one partition, remove whole device */
437         if (dev->num_parts == 1)
438                 return device_del(dev);
439
440         /* otherwise just delete this partition */
441
442         if (dev == current_mtd_dev) {
443                 /* we are modyfing partitions for the current device,
444                  * update current */
445                 struct part_info *curr_pi;
446                 curr_pi = mtd_part_info(current_mtd_dev, current_mtd_partnum);
447
448                 if (curr_pi) {
449                         if (curr_pi == part) {
450                                 printf("current partition deleted, resetting current to 0\n");
451                                 current_mtd_partnum = 0;
452                         } else if (part->offset <= curr_pi->offset) {
453                                 current_mtd_partnum--;
454                         }
455                         current_save_needed = 1;
456                 }
457         }
458
459         list_del(&part->link);
460         free(part);
461         dev->num_parts--;
462
463         if (current_save_needed > 0)
464                 current_save();
465         else
466                 index_partitions();
467
468         return 0;
469 }
470
471 /**
472  * Delete all partitions from parts head list, free memory.
473  *
474  * @param head list of partitions to delete
475  */
476 static void part_delall(struct list_head *head)
477 {
478         struct list_head *entry, *n;
479         struct part_info *part_tmp;
480
481         /* clean tmp_list and free allocated memory */
482         list_for_each_safe(entry, n, head) {
483                 part_tmp = list_entry(entry, struct part_info, link);
484
485                 list_del(entry);
486                 free(part_tmp);
487         }
488 }
489
490 /**
491  * Add new partition to the supplied partition list. Make sure partitions are
492  * sorted by offset in ascending order.
493  *
494  * @param head list this partition is to be added to
495  * @param new partition to be added
496  */
497 static int part_sort_add(struct mtd_device *dev, struct part_info *part)
498 {
499         struct list_head *entry;
500         struct part_info *new_pi, *curr_pi;
501
502         /* link partition to parrent dev */
503         part->dev = dev;
504
505         if (list_empty(&dev->parts)) {
506                 debug("part_sort_add: list empty\n");
507                 list_add(&part->link, &dev->parts);
508                 dev->num_parts++;
509                 index_partitions();
510                 return 0;
511         }
512
513         new_pi = list_entry(&part->link, struct part_info, link);
514
515         /* get current partition info if we are updating current device */
516         curr_pi = NULL;
517         if (dev == current_mtd_dev)
518                 curr_pi = mtd_part_info(current_mtd_dev, current_mtd_partnum);
519
520         list_for_each(entry, &dev->parts) {
521                 struct part_info *pi;
522
523                 pi = list_entry(entry, struct part_info, link);
524
525                 /* be compliant with kernel cmdline, allow only one partition at offset zero */
526                 if ((new_pi->offset == pi->offset) && (pi->offset == 0)) {
527                         printf("cannot add second partition at offset 0\n");
528                         return 1;
529                 }
530
531                 if (new_pi->offset <= pi->offset) {
532                         list_add_tail(&part->link, entry);
533                         dev->num_parts++;
534
535                         if (curr_pi && (pi->offset <= curr_pi->offset)) {
536                                 /* we are modyfing partitions for the current
537                                  * device, update current */
538                                 current_mtd_partnum++;
539                                 current_save();
540                         } else {
541                                 index_partitions();
542                         }
543                         return 0;
544                 }
545         }
546
547         list_add_tail(&part->link, &dev->parts);
548         dev->num_parts++;
549         index_partitions();
550         return 0;
551 }
552
553 /**
554  * Add provided partition to the partition list of a given device.
555  *
556  * @param dev device to which partition is added
557  * @param part partition to be added
558  * @return 0 on success, 1 otherwise
559  */
560 static int part_add(struct mtd_device *dev, struct part_info *part)
561 {
562         /* verify alignment and size */
563         if (part_validate(dev->id, part) != 0)
564                 return 1;
565
566         /* partition is ok, add it to the list */
567         if (part_sort_add(dev, part) != 0)
568                 return 1;
569
570         return 0;
571 }
572
573 /**
574  * Parse one partition definition, allocate memory and return pointer to this
575  * location in retpart.
576  *
577  * @param partdef pointer to the partition definition string i.e. <part-def>
578  * @param ret output pointer to next char after parse completes (output)
579  * @param retpart pointer to the allocated partition (output)
580  * @return 0 on success, 1 otherwise
581  */
582 static int part_parse(const char *const partdef, const char **ret, struct part_info **retpart)
583 {
584         struct part_info *part;
585         u64 size;
586         u64 offset;
587         const char *name;
588         int name_len;
589         unsigned int mask_flags;
590         const char *p;
591
592         p = partdef;
593         *retpart = NULL;
594         *ret = NULL;
595
596         /* fetch the partition size */
597         if (*p == '-') {
598                 /* assign all remaining space to this partition */
599                 debug("'-': remaining size assigned\n");
600                 size = SIZE_REMAINING;
601                 p++;
602         } else {
603                 size = memsize_parse(p, &p);
604                 if (size < MIN_PART_SIZE) {
605                         printf("partition size too small (%llx)\n", size);
606                         return 1;
607                 }
608         }
609
610         /* check for offset */
611         offset = OFFSET_NOT_SPECIFIED;
612         if (*p == '@') {
613                 p++;
614                 offset = memsize_parse(p, &p);
615         }
616
617         /* now look for the name */
618         if (*p == '(') {
619                 name = ++p;
620                 if ((p = strchr(name, ')')) == NULL) {
621                         printf("no closing ) found in partition name\n");
622                         return 1;
623                 }
624                 name_len = p - name + 1;
625                 if ((name_len - 1) == 0) {
626                         printf("empty partition name\n");
627                         return 1;
628                 }
629                 p++;
630         } else {
631                 /* 0x00000000@0x00000000 */
632                 name_len = 22;
633                 name = NULL;
634         }
635
636         /* test for options */
637         mask_flags = 0;
638         if (strncmp(p, "ro", 2) == 0) {
639                 mask_flags |= MTD_WRITEABLE_CMD;
640                 p += 2;
641         }
642
643         /* check for next partition definition */
644         if (*p == ',') {
645                 if (size == SIZE_REMAINING) {
646                         *ret = NULL;
647                         printf("no partitions allowed after a fill-up partition\n");
648                         return 1;
649                 }
650                 *ret = ++p;
651         } else if ((*p == ';') || (*p == '\0')) {
652                 *ret = p;
653         } else {
654                 printf("unexpected character '%c' at the end of partition\n", *p);
655                 *ret = NULL;
656                 return 1;
657         }
658
659         /*  allocate memory */
660         part = (struct part_info *)malloc(sizeof(struct part_info) + name_len);
661         if (!part) {
662                 printf("out of memory\n");
663                 return 1;
664         }
665         memset(part, 0, sizeof(struct part_info) + name_len);
666         part->size = size;
667         part->offset = offset;
668         part->mask_flags = mask_flags;
669         part->name = (char *)(part + 1);
670
671         if (name) {
672                 /* copy user provided name */
673                 strncpy(part->name, name, name_len - 1);
674                 part->auto_name = 0;
675         } else {
676                 /* auto generated name in form of size@offset */
677                 sprintf(part->name, "0x%08llx@0x%08llx", size, offset);
678                 part->auto_name = 1;
679         }
680
681         part->name[name_len - 1] = '\0';
682         INIT_LIST_HEAD(&part->link);
683
684         debug("+ partition: name %-22s size 0x%08llx offset 0x%08llx mask flags %d\n",
685                         part->name, part->size,
686                         part->offset, part->mask_flags);
687
688         *retpart = part;
689         return 0;
690 }
691
692 /**
693  * Check device number to be within valid range for given device type.
694  *
695  * @param type mtd type
696  * @param num mtd number
697  * @param size a pointer to the size of the mtd device (output)
698  * @return 0 if device is valid, 1 otherwise
699  */
700 static int mtd_device_validate(u8 type, u8 num, u64 *size)
701 {
702         struct mtd_info *mtd = NULL;
703
704         if (get_mtd_info(type, num, &mtd))
705                 return 1;
706
707         *size = mtd->size;
708
709         return 0;
710 }
711
712 /**
713  * Delete all mtd devices from a supplied devices list, free memory allocated for
714  * each device and delete all device partitions.
715  *
716  * @return 0 on success, 1 otherwise
717  */
718 static int device_delall(struct list_head *head)
719 {
720         struct list_head *entry, *n;
721         struct mtd_device *dev_tmp;
722
723         /* clean devices list */
724         list_for_each_safe(entry, n, head) {
725                 dev_tmp = list_entry(entry, struct mtd_device, link);
726                 list_del(entry);
727                 part_delall(&dev_tmp->parts);
728                 free(dev_tmp);
729         }
730         INIT_LIST_HEAD(&devices);
731
732         return 0;
733 }
734
735 /**
736  * If provided device exists it's partitions are deleted, device is removed
737  * from device list and device memory is freed.
738  *
739  * @param dev device to be deleted
740  * @return 0 on success, 1 otherwise
741  */
742 static int device_del(struct mtd_device *dev)
743 {
744         part_delall(&dev->parts);
745         list_del(&dev->link);
746         free(dev);
747
748         if (dev == current_mtd_dev) {
749                 /* we just deleted current device */
750                 if (list_empty(&devices)) {
751                         current_mtd_dev = NULL;
752                 } else {
753                         /* reset first partition from first dev from the
754                          * devices list as current */
755                         current_mtd_dev = list_entry(devices.next, struct mtd_device, link);
756                         current_mtd_partnum = 0;
757                 }
758                 current_save();
759                 return 0;
760         }
761
762         index_partitions();
763         return 0;
764 }
765
766 /**
767  * Search global device list and return pointer to the device of type and num
768  * specified.
769  *
770  * @param type device type
771  * @param num device number
772  * @return NULL if requested device does not exist
773  */
774 struct mtd_device *device_find(u8 type, u8 num)
775 {
776         struct list_head *entry;
777         struct mtd_device *dev_tmp;
778
779         list_for_each(entry, &devices) {
780                 dev_tmp = list_entry(entry, struct mtd_device, link);
781
782                 if ((dev_tmp->id->type == type) && (dev_tmp->id->num == num))
783                         return dev_tmp;
784         }
785
786         return NULL;
787 }
788
789 /**
790  * Add specified device to the global device list.
791  *
792  * @param dev device to be added
793  */
794 static void device_add(struct mtd_device *dev)
795 {
796         u8 current_save_needed = 0;
797
798         if (list_empty(&devices)) {
799                 current_mtd_dev = dev;
800                 current_mtd_partnum = 0;
801                 current_save_needed = 1;
802         }
803
804         list_add_tail(&dev->link, &devices);
805
806         if (current_save_needed > 0)
807                 current_save();
808         else
809                 index_partitions();
810 }
811
812 /**
813  * Parse device type, name and mtd-id. If syntax is ok allocate memory and
814  * return pointer to the device structure.
815  *
816  * @param mtd_dev pointer to the device definition string i.e. <mtd-dev>
817  * @param ret output pointer to next char after parse completes (output)
818  * @param retdev pointer to the allocated device (output)
819  * @return 0 on success, 1 otherwise
820  */
821 static int device_parse(const char *const mtd_dev, const char **ret, struct mtd_device **retdev)
822 {
823         struct mtd_device *dev;
824         struct part_info *part;
825         struct mtdids *id;
826         const char *mtd_id;
827         unsigned int mtd_id_len;
828         const char *p;
829         const char *pend;
830         LIST_HEAD(tmp_list);
831         struct list_head *entry, *n;
832         u16 num_parts;
833         u64 offset;
834         int err = 1;
835
836         debug("===device_parse===\n");
837
838         assert(retdev);
839         *retdev = NULL;
840
841         if (ret)
842                 *ret = NULL;
843
844         /* fetch <mtd-id> */
845         mtd_id = p = mtd_dev;
846         if (!(p = strchr(mtd_id, ':'))) {
847                 printf("no <mtd-id> identifier\n");
848                 return 1;
849         }
850         mtd_id_len = p - mtd_id + 1;
851         p++;
852
853         /* verify if we have a valid device specified */
854         if ((id = id_find_by_mtd_id(mtd_id, mtd_id_len - 1)) == NULL) {
855                 printf("invalid mtd device '%.*s'\n", mtd_id_len - 1, mtd_id);
856                 return 1;
857         }
858
859 #ifdef DEBUG
860         pend = strchr(p, ';');
861 #endif
862         debug("dev type = %d (%s), dev num = %d, mtd-id = %s\n",
863                         id->type, MTD_DEV_TYPE(id->type),
864                         id->num, id->mtd_id);
865         debug("parsing partitions %.*s\n", (int)(pend ? pend - p : strlen(p)), p);
866
867
868         /* parse partitions */
869         num_parts = 0;
870
871         offset = 0;
872         if ((dev = device_find(id->type, id->num)) != NULL) {
873                 /* if device already exists start at the end of the last partition */
874                 part = list_entry(dev->parts.prev, struct part_info, link);
875                 offset = part->offset + part->size;
876         }
877
878         while (p && (*p != '\0') && (*p != ';')) {
879                 err = 1;
880                 if ((part_parse(p, &p, &part) != 0) || (!part))
881                         break;
882
883                 /* calculate offset when not specified */
884                 if (part->offset == OFFSET_NOT_SPECIFIED)
885                         part->offset = offset;
886                 else
887                         offset = part->offset;
888
889                 /* verify alignment and size */
890                 if (part_validate(id, part) != 0)
891                         break;
892
893                 offset += part->size;
894
895                 /* partition is ok, add it to the list */
896                 list_add_tail(&part->link, &tmp_list);
897                 num_parts++;
898                 err = 0;
899         }
900         if (err == 1) {
901                 part_delall(&tmp_list);
902                 return 1;
903         }
904
905         if (num_parts == 0) {
906                 printf("no partitions for device %s%d (%s)\n",
907                                 MTD_DEV_TYPE(id->type), id->num, id->mtd_id);
908                 return 1;
909         }
910
911         debug("\ntotal partitions: %d\n", num_parts);
912
913         /* check for next device presence */
914         if (p) {
915                 if (*p == ';') {
916                         if (ret)
917                                 *ret = ++p;
918                 } else if (*p == '\0') {
919                         if (ret)
920                                 *ret = p;
921                 } else {
922                         printf("unexpected character '%c' at the end of device\n", *p);
923                         if (ret)
924                                 *ret = NULL;
925                         return 1;
926                 }
927         }
928
929         /* allocate memory for mtd_device structure */
930         if ((dev = (struct mtd_device *)malloc(sizeof(struct mtd_device))) == NULL) {
931                 printf("out of memory\n");
932                 return 1;
933         }
934         memset(dev, 0, sizeof(struct mtd_device));
935         dev->id = id;
936         dev->num_parts = 0; /* part_sort_add increments num_parts */
937         INIT_LIST_HEAD(&dev->parts);
938         INIT_LIST_HEAD(&dev->link);
939
940         /* move partitions from tmp_list to dev->parts */
941         list_for_each_safe(entry, n, &tmp_list) {
942                 part = list_entry(entry, struct part_info, link);
943                 list_del(entry);
944                 if (part_sort_add(dev, part) != 0) {
945                         device_del(dev);
946                         return 1;
947                 }
948         }
949
950         *retdev = dev;
951
952         debug("===\n\n");
953         return 0;
954 }
955
956 /**
957  * Initialize global device list.
958  *
959  * @return 0 on success, 1 otherwise
960  */
961 static int mtd_devices_init(void)
962 {
963         last_parts[0] = '\0';
964         current_mtd_dev = NULL;
965         current_save();
966
967         return device_delall(&devices);
968 }
969
970 /*
971  * Search global mtdids list and find id of requested type and number.
972  *
973  * @return pointer to the id if it exists, NULL otherwise
974  */
975 static struct mtdids* id_find(u8 type, u8 num)
976 {
977         struct list_head *entry;
978         struct mtdids *id;
979
980         list_for_each(entry, &mtdids) {
981                 id = list_entry(entry, struct mtdids, link);
982
983                 if ((id->type == type) && (id->num == num))
984                         return id;
985         }
986
987         return NULL;
988 }
989
990 /**
991  * Search global mtdids list and find id of a requested mtd_id.
992  *
993  * Note: first argument is not null terminated.
994  *
995  * @param mtd_id string containing requested mtd_id
996  * @param mtd_id_len length of supplied mtd_id
997  * @return pointer to the id if it exists, NULL otherwise
998  */
999 static struct mtdids* id_find_by_mtd_id(const char *mtd_id, unsigned int mtd_id_len)
1000 {
1001         struct list_head *entry;
1002         struct mtdids *id;
1003
1004         debug("--- id_find_by_mtd_id: '%.*s' (len = %d)\n",
1005                         mtd_id_len, mtd_id, mtd_id_len);
1006
1007         list_for_each(entry, &mtdids) {
1008                 id = list_entry(entry, struct mtdids, link);
1009
1010                 debug("entry: '%s' (len = %zu)\n",
1011                                 id->mtd_id, strlen(id->mtd_id));
1012
1013                 if (mtd_id_len != strlen(id->mtd_id))
1014                         continue;
1015                 if (strncmp(id->mtd_id, mtd_id, mtd_id_len) == 0)
1016                         return id;
1017         }
1018
1019         return NULL;
1020 }
1021
1022 /**
1023  * Parse device id string <dev-id> := 'nand'|'nor'|'onenand'<dev-num>,
1024  * return device type and number.
1025  *
1026  * @param id string describing device id
1027  * @param ret_id output pointer to next char after parse completes (output)
1028  * @param dev_type parsed device type (output)
1029  * @param dev_num parsed device number (output)
1030  * @return 0 on success, 1 otherwise
1031  */
1032 int mtd_id_parse(const char *id, const char **ret_id, u8 *dev_type,
1033                  u8 *dev_num)
1034 {
1035         const char *p = id;
1036
1037         *dev_type = 0;
1038         if (strncmp(p, "nand", 4) == 0) {
1039                 *dev_type = MTD_DEV_TYPE_NAND;
1040                 p += 4;
1041         } else if (strncmp(p, "nor", 3) == 0) {
1042                 *dev_type = MTD_DEV_TYPE_NOR;
1043                 p += 3;
1044         } else if (strncmp(p, "onenand", 7) == 0) {
1045                 *dev_type = MTD_DEV_TYPE_ONENAND;
1046                 p += 7;
1047         } else {
1048                 printf("incorrect device type in %s\n", id);
1049                 return 1;
1050         }
1051
1052         if (!isdigit(*p)) {
1053                 printf("incorrect device number in %s\n", id);
1054                 return 1;
1055         }
1056
1057         *dev_num = simple_strtoul(p, (char **)&p, 0);
1058         if (ret_id)
1059                 *ret_id = p;
1060         return 0;
1061 }
1062
1063 /**
1064  * Process all devices and generate corresponding mtdparts string describing
1065  * all partitions on all devices.
1066  *
1067  * @param buf output buffer holding generated mtdparts string (output)
1068  * @param buflen buffer size
1069  * @return 0 on success, 1 otherwise
1070  */
1071 static int generate_mtdparts(char *buf, u32 buflen)
1072 {
1073         struct list_head *pentry, *dentry;
1074         struct mtd_device *dev;
1075         struct part_info *part, *prev_part;
1076         char *p = buf;
1077         char tmpbuf[32];
1078         u64 size, offset;
1079         u32 len, part_cnt;
1080         u32 maxlen = buflen - 1;
1081
1082         debug("--- generate_mtdparts ---\n");
1083
1084         if (list_empty(&devices)) {
1085                 buf[0] = '\0';
1086                 return 0;
1087         }
1088
1089         sprintf(p, "mtdparts=");
1090         p += 9;
1091
1092         list_for_each(dentry, &devices) {
1093                 dev = list_entry(dentry, struct mtd_device, link);
1094
1095                 /* copy mtd_id */
1096                 len = strlen(dev->id->mtd_id) + 1;
1097                 if (len > maxlen)
1098                         goto cleanup;
1099                 memcpy(p, dev->id->mtd_id, len - 1);
1100                 p += len - 1;
1101                 *(p++) = ':';
1102                 maxlen -= len;
1103
1104                 /* format partitions */
1105                 prev_part = NULL;
1106                 part_cnt = 0;
1107                 list_for_each(pentry, &dev->parts) {
1108                         part = list_entry(pentry, struct part_info, link);
1109                         size = part->size;
1110                         offset = part->offset;
1111                         part_cnt++;
1112
1113                         /* partition size */
1114                         memsize_format(tmpbuf, size);
1115                         len = strlen(tmpbuf);
1116                         if (len > maxlen)
1117                                 goto cleanup;
1118                         memcpy(p, tmpbuf, len);
1119                         p += len;
1120                         maxlen -= len;
1121
1122
1123                         /* add offset only when there is a gap between
1124                          * partitions */
1125                         if ((!prev_part && (offset != 0)) ||
1126                                         (prev_part && ((prev_part->offset + prev_part->size) != part->offset))) {
1127
1128                                 memsize_format(tmpbuf, offset);
1129                                 len = strlen(tmpbuf) + 1;
1130                                 if (len > maxlen)
1131                                         goto cleanup;
1132                                 *(p++) = '@';
1133                                 memcpy(p, tmpbuf, len - 1);
1134                                 p += len - 1;
1135                                 maxlen -= len;
1136                         }
1137
1138                         /* copy name only if user supplied */
1139                         if(!part->auto_name) {
1140                                 len = strlen(part->name) + 2;
1141                                 if (len > maxlen)
1142                                         goto cleanup;
1143
1144                                 *(p++) = '(';
1145                                 memcpy(p, part->name, len - 2);
1146                                 p += len - 2;
1147                                 *(p++) = ')';
1148                                 maxlen -= len;
1149                         }
1150
1151                         /* ro mask flag */
1152                         if (part->mask_flags && MTD_WRITEABLE_CMD) {
1153                                 len = 2;
1154                                 if (len > maxlen)
1155                                         goto cleanup;
1156                                 *(p++) = 'r';
1157                                 *(p++) = 'o';
1158                                 maxlen -= 2;
1159                         }
1160
1161                         /* print ',' separator if there are other partitions
1162                          * following */
1163                         if (dev->num_parts > part_cnt) {
1164                                 if (1 > maxlen)
1165                                         goto cleanup;
1166                                 *(p++) = ',';
1167                                 maxlen--;
1168                         }
1169                         prev_part = part;
1170                 }
1171                 /* print ';' separator if there are other devices following */
1172                 if (dentry->next != &devices) {
1173                         if (1 > maxlen)
1174                                 goto cleanup;
1175                         *(p++) = ';';
1176                         maxlen--;
1177                 }
1178         }
1179
1180         /* we still have at least one char left, as we decremented maxlen at
1181          * the begining */
1182         *p = '\0';
1183
1184         return 0;
1185
1186 cleanup:
1187         last_parts[0] = '\0';
1188         return 1;
1189 }
1190
1191 /**
1192  * Call generate_mtdparts to process all devices and generate corresponding
1193  * mtdparts string, save it in mtdparts environment variable.
1194  *
1195  * @param buf output buffer holding generated mtdparts string (output)
1196  * @param buflen buffer size
1197  * @return 0 on success, 1 otherwise
1198  */
1199 static int generate_mtdparts_save(char *buf, u32 buflen)
1200 {
1201         int ret;
1202
1203         ret = generate_mtdparts(buf, buflen);
1204
1205         if ((buf[0] != '\0') && (ret == 0))
1206                 setenv("mtdparts", buf);
1207         else
1208                 setenv("mtdparts", NULL);
1209
1210         return ret;
1211 }
1212
1213 #if defined(CONFIG_CMD_MTDPARTS_SHOW_NET_SIZES)
1214 /**
1215  * Get the net size (w/o bad blocks) of the given partition.
1216  *
1217  * @param mtd the mtd info
1218  * @param part the partition
1219  * @return the calculated net size of this partition
1220  */
1221 static uint64_t net_part_size(struct mtd_info *mtd, struct part_info *part)
1222 {
1223         uint64_t i, net_size = 0;
1224
1225         if (!mtd->block_isbad)
1226                 return part->size;
1227
1228         for (i = 0; i < part->size; i += mtd->erasesize) {
1229                 if (!mtd->block_isbad(mtd, part->offset + i))
1230                         net_size += mtd->erasesize;
1231         }
1232
1233         return net_size;
1234 }
1235 #endif
1236
1237 static void print_partition_table(void)
1238 {
1239         struct list_head *dentry, *pentry;
1240         struct part_info *part;
1241         struct mtd_device *dev;
1242         int part_num;
1243
1244         list_for_each(dentry, &devices) {
1245                 dev = list_entry(dentry, struct mtd_device, link);
1246                 /* list partitions for given device */
1247                 part_num = 0;
1248 #if defined(CONFIG_CMD_MTDPARTS_SHOW_NET_SIZES)
1249                 struct mtd_info *mtd;
1250
1251                 if (get_mtd_info(dev->id->type, dev->id->num, &mtd))
1252                         return;
1253
1254                 printf("\ndevice %s%d <%s>, # parts = %d\n",
1255                                 MTD_DEV_TYPE(dev->id->type), dev->id->num,
1256                                 dev->id->mtd_id, dev->num_parts);
1257                 printf(" #: name\t\tsize\t\tnet size\toffset\t\tmask_flags\n");
1258
1259                 list_for_each(pentry, &dev->parts) {
1260                         u32 net_size;
1261                         char *size_note;
1262
1263                         part = list_entry(pentry, struct part_info, link);
1264                         net_size = net_part_size(mtd, part);
1265                         size_note = part->size == net_size ? " " : " (!)";
1266                         printf("%2d: %-20s0x%08x\t0x%08x%s\t0x%08x\t%d\n",
1267                                         part_num, part->name, part->size,
1268                                         net_size, size_note, part->offset,
1269                                         part->mask_flags);
1270 #else /* !defined(CONFIG_CMD_MTDPARTS_SHOW_NET_SIZES) */
1271                 printf("\ndevice %s%d <%s>, # parts = %d\n",
1272                                 MTD_DEV_TYPE(dev->id->type), dev->id->num,
1273                                 dev->id->mtd_id, dev->num_parts);
1274                 printf(" #: name\t\tsize\t\toffset\t\tmask_flags\n");
1275
1276                 list_for_each(pentry, &dev->parts) {
1277                         part = list_entry(pentry, struct part_info, link);
1278                         printf("%2d: %-20s0x%08llx\t0x%08llx\t%d\n",
1279                                         part_num, part->name, part->size,
1280                                         part->offset, part->mask_flags);
1281 #endif /* defined(CONFIG_CMD_MTDPARTS_SHOW_NET_SIZES) */
1282                         part_num++;
1283                 }
1284         }
1285
1286         if (list_empty(&devices))
1287                 printf("no partitions defined\n");
1288 }
1289
1290 /**
1291  * Format and print out a partition list for each device from global device
1292  * list.
1293  */
1294 static void list_partitions(void)
1295 {
1296         struct part_info *part;
1297
1298         debug("\n---list_partitions---\n");
1299         print_partition_table();
1300
1301         /* current_mtd_dev is not NULL only when we have non empty device list */
1302         if (current_mtd_dev) {
1303                 part = mtd_part_info(current_mtd_dev, current_mtd_partnum);
1304                 if (part) {
1305                         printf("\nactive partition: %s%d,%d - (%s) 0x%08llx @ 0x%08llx\n",
1306                                         MTD_DEV_TYPE(current_mtd_dev->id->type),
1307                                         current_mtd_dev->id->num, current_mtd_partnum,
1308                                         part->name, part->size, part->offset);
1309                 } else {
1310                         printf("could not get current partition info\n\n");
1311                 }
1312         }
1313
1314         printf("\ndefaults:\n");
1315         printf("mtdids  : %s\n",
1316                 mtdids_default ? mtdids_default : "none");
1317         /*
1318          * Using printf() here results in printbuffer overflow
1319          * if default mtdparts string is greater than console
1320          * printbuffer. Use puts() to prevent system crashes.
1321          */
1322         puts("mtdparts: ");
1323         puts(mtdparts_default ? mtdparts_default : "none");
1324         puts("\n");
1325 }
1326
1327 /**
1328  * Given partition identifier in form of <dev_type><dev_num>,<part_num> find
1329  * corresponding device and verify partition number.
1330  *
1331  * @param id string describing device and partition or partition name
1332  * @param dev pointer to the requested device (output)
1333  * @param part_num verified partition number (output)
1334  * @param part pointer to requested partition (output)
1335  * @return 0 on success, 1 otherwise
1336  */
1337 int find_dev_and_part(const char *id, struct mtd_device **dev,
1338                 u8 *part_num, struct part_info **part)
1339 {
1340         struct list_head *dentry, *pentry;
1341         u8 type, dnum, pnum;
1342         const char *p;
1343
1344         debug("--- find_dev_and_part ---\nid = %s\n", id);
1345
1346         list_for_each(dentry, &devices) {
1347                 *part_num = 0;
1348                 *dev = list_entry(dentry, struct mtd_device, link);
1349                 list_for_each(pentry, &(*dev)->parts) {
1350                         *part = list_entry(pentry, struct part_info, link);
1351                         if (strcmp((*part)->name, id) == 0)
1352                                 return 0;
1353                         (*part_num)++;
1354                 }
1355         }
1356
1357         p = id;
1358         *dev = NULL;
1359         *part = NULL;
1360         *part_num = 0;
1361
1362         if (mtd_id_parse(p, &p, &type, &dnum) != 0)
1363                 return 1;
1364
1365         if ((*p++ != ',') || (*p == '\0')) {
1366                 printf("no partition number specified\n");
1367                 return 1;
1368         }
1369         pnum = simple_strtoul(p, (char **)&p, 0);
1370         if (*p != '\0') {
1371                 printf("unexpected trailing character '%c'\n", *p);
1372                 return 1;
1373         }
1374
1375         if ((*dev = device_find(type, dnum)) == NULL) {
1376                 printf("no such device %s%d\n", MTD_DEV_TYPE(type), dnum);
1377                 return 1;
1378         }
1379
1380         if ((*part = mtd_part_info(*dev, pnum)) == NULL) {
1381                 printf("no such partition\n");
1382                 *dev = NULL;
1383                 return 1;
1384         }
1385
1386         *part_num = pnum;
1387
1388         return 0;
1389 }
1390
1391 /**
1392  * Find and delete partition. For partition id format see find_dev_and_part().
1393  *
1394  * @param id string describing device and partition
1395  * @return 0 on success, 1 otherwise
1396  */
1397 static int delete_partition(const char *id)
1398 {
1399         u8 pnum;
1400         struct mtd_device *dev;
1401         struct part_info *part;
1402
1403         if (find_dev_and_part(id, &dev, &pnum, &part) == 0) {
1404
1405                 debug("delete_partition: device = %s%d, partition %d = (%s) 0x%08llx@0x%08llx\n",
1406                                 MTD_DEV_TYPE(dev->id->type), dev->id->num, pnum,
1407                                 part->name, part->size, part->offset);
1408
1409                 if (part_del(dev, part) != 0)
1410                         return 1;
1411
1412                 if (generate_mtdparts_save(last_parts, MTDPARTS_MAXLEN) != 0) {
1413                         printf("generated mtdparts too long, resetting to null\n");
1414                         return 1;
1415                 }
1416                 return 0;
1417         }
1418
1419         printf("partition %s not found\n", id);
1420         return 1;
1421 }
1422
1423 #if defined(CONFIG_CMD_MTDPARTS_SPREAD)
1424 /**
1425  * Increase the size of the given partition so that it's net size is at least
1426  * as large as the size member and such that the next partition would start on a
1427  * good block if it were adjacent to this partition.
1428  *
1429  * @param mtd the mtd device
1430  * @param part the partition
1431  * @param next_offset pointer to the offset of the next partition after this
1432  *                    partition's size has been modified (output)
1433  */
1434 static void spread_partition(struct mtd_info *mtd, struct part_info *part,
1435                              uint64_t *next_offset)
1436 {
1437         uint64_t net_size, padding_size = 0;
1438         int truncated;
1439
1440         mtd_get_len_incl_bad(mtd, part->offset, part->size, &net_size,
1441                              &truncated);
1442
1443         /*
1444          * Absorb bad blocks immediately following this
1445          * partition also into the partition, such that
1446          * the next partition starts with a good block.
1447          */
1448         if (!truncated) {
1449                 mtd_get_len_incl_bad(mtd, part->offset + net_size,
1450                                      mtd->erasesize, &padding_size, &truncated);
1451                 if (truncated)
1452                         padding_size = 0;
1453                 else
1454                         padding_size -= mtd->erasesize;
1455         }
1456
1457         if (truncated) {
1458                 printf("truncated partition %s to %lld bytes\n", part->name,
1459                        (uint64_t) net_size + padding_size);
1460         }
1461
1462         part->size = net_size + padding_size;
1463         *next_offset = part->offset + part->size;
1464 }
1465
1466 /**
1467  * Adjust all of the partition sizes, such that all partitions are at least
1468  * as big as their mtdparts environment variable sizes and they each start
1469  * on a good block.
1470  *
1471  * @return 0 on success, 1 otherwise
1472  */
1473 static int spread_partitions(void)
1474 {
1475         struct list_head *dentry, *pentry;
1476         struct mtd_device *dev;
1477         struct part_info *part;
1478         struct mtd_info *mtd;
1479         int part_num;
1480         uint64_t cur_offs;
1481
1482         list_for_each(dentry, &devices) {
1483                 dev = list_entry(dentry, struct mtd_device, link);
1484
1485                 if (get_mtd_info(dev->id->type, dev->id->num, &mtd))
1486                         return 1;
1487
1488                 part_num = 0;
1489                 cur_offs = 0;
1490                 list_for_each(pentry, &dev->parts) {
1491                         part = list_entry(pentry, struct part_info, link);
1492
1493                         debug("spread_partitions: device = %s%d, partition %d ="
1494                                 " (%s) 0x%08x@0x%08x\n",
1495                                 MTD_DEV_TYPE(dev->id->type), dev->id->num,
1496                                 part_num, part->name, part->size,
1497                                 part->offset);
1498
1499                         if (cur_offs > part->offset)
1500                                 part->offset = cur_offs;
1501
1502                         spread_partition(mtd, part, &cur_offs);
1503
1504                         part_num++;
1505                 }
1506         }
1507
1508         index_partitions();
1509
1510         if (generate_mtdparts_save(last_parts, MTDPARTS_MAXLEN) != 0) {
1511                 printf("generated mtdparts too long, resetting to null\n");
1512                 return 1;
1513         }
1514         return 0;
1515 }
1516 #endif /* CONFIG_CMD_MTDPARTS_SPREAD */
1517
1518 /**
1519  * Accept character string describing mtd partitions and call device_parse()
1520  * for each entry. Add created devices to the global devices list.
1521  *
1522  * @param mtdparts string specifing mtd partitions
1523  * @return 0 on success, 1 otherwise
1524  */
1525 static int parse_mtdparts(const char *const mtdparts)
1526 {
1527         const char *p = mtdparts;
1528         struct mtd_device *dev;
1529         int err = 1;
1530         char tmp_parts[MTDPARTS_MAXLEN];
1531
1532         debug("\n---parse_mtdparts---\nmtdparts = %s\n\n", p);
1533
1534         /* delete all devices and partitions */
1535         if (mtd_devices_init() != 0) {
1536                 printf("could not initialise device list\n");
1537                 return err;
1538         }
1539
1540         /* re-read 'mtdparts' variable, mtd_devices_init may be updating env */
1541         if (gd->flags & GD_FLG_ENV_READY) {
1542                 p = getenv("mtdparts");
1543         } else {
1544                 p = tmp_parts;
1545                 getenv_f("mtdparts", tmp_parts, MTDPARTS_MAXLEN);
1546         }
1547
1548         if (strncmp(p, "mtdparts=", 9) != 0) {
1549                 printf("mtdparts variable doesn't start with 'mtdparts='\n");
1550                 return err;
1551         }
1552         p += 9;
1553
1554         while (p && (*p != '\0')) {
1555                 err = 1;
1556                 if ((device_parse(p, &p, &dev) != 0) || (!dev))
1557                         break;
1558
1559                 debug("+ device: %s\t%d\t%s\n", MTD_DEV_TYPE(dev->id->type),
1560                                 dev->id->num, dev->id->mtd_id);
1561
1562                 /* check if parsed device is already on the list */
1563                 if (device_find(dev->id->type, dev->id->num) != NULL) {
1564                         printf("device %s%d redefined, please correct mtdparts variable\n",
1565                                         MTD_DEV_TYPE(dev->id->type), dev->id->num);
1566                         break;
1567                 }
1568
1569                 list_add_tail(&dev->link, &devices);
1570                 err = 0;
1571         }
1572         if (err == 1) {
1573                 device_delall(&devices);
1574                 return 1;
1575         }
1576
1577         return 0;
1578 }
1579
1580 /**
1581  * Parse provided string describing mtdids mapping (see file header for mtdids
1582  * variable format). Allocate memory for each entry and add all found entries
1583  * to the global mtdids list.
1584  *
1585  * @param ids mapping string
1586  * @return 0 on success, 1 otherwise
1587  */
1588 static int parse_mtdids(const char *const ids)
1589 {
1590         const char *p = ids;
1591         const char *mtd_id;
1592         int mtd_id_len;
1593         struct mtdids *id;
1594         struct list_head *entry, *n;
1595         struct mtdids *id_tmp;
1596         u8 type, num;
1597         u64 size;
1598         int ret = 1;
1599
1600         debug("\n---parse_mtdids---\nmtdids = %s\n\n", ids);
1601
1602         /* clean global mtdids list */
1603         list_for_each_safe(entry, n, &mtdids) {
1604                 id_tmp = list_entry(entry, struct mtdids, link);
1605                 debug("mtdids del: %d %d\n", id_tmp->type, id_tmp->num);
1606                 list_del(entry);
1607                 free(id_tmp);
1608         }
1609         last_ids[0] = '\0';
1610         INIT_LIST_HEAD(&mtdids);
1611
1612         while(p && (*p != '\0')) {
1613
1614                 ret = 1;
1615                 /* parse 'nor'|'nand'|'onenand'<dev-num> */
1616                 if (mtd_id_parse(p, &p, &type, &num) != 0)
1617                         break;
1618
1619                 if (*p != '=') {
1620                         printf("mtdids: incorrect <dev-num>\n");
1621                         break;
1622                 }
1623                 p++;
1624
1625                 /* check if requested device exists */
1626                 if (mtd_device_validate(type, num, &size) != 0)
1627                         return 1;
1628
1629                 /* locate <mtd-id> */
1630                 mtd_id = p;
1631                 if ((p = strchr(mtd_id, ',')) != NULL) {
1632                         mtd_id_len = p - mtd_id + 1;
1633                         p++;
1634                 } else {
1635                         mtd_id_len = strlen(mtd_id) + 1;
1636                 }
1637                 if (mtd_id_len == 0) {
1638                         printf("mtdids: no <mtd-id> identifier\n");
1639                         break;
1640                 }
1641
1642                 /* check if this id is already on the list */
1643                 int double_entry = 0;
1644                 list_for_each(entry, &mtdids) {
1645                         id_tmp = list_entry(entry, struct mtdids, link);
1646                         if ((id_tmp->type == type) && (id_tmp->num == num)) {
1647                                 double_entry = 1;
1648                                 break;
1649                         }
1650                 }
1651                 if (double_entry) {
1652                         printf("device id %s%d redefined, please correct mtdids variable\n",
1653                                         MTD_DEV_TYPE(type), num);
1654                         break;
1655                 }
1656
1657                 /* allocate mtdids structure */
1658                 if (!(id = (struct mtdids *)malloc(sizeof(struct mtdids) + mtd_id_len))) {
1659                         printf("out of memory\n");
1660                         break;
1661                 }
1662                 memset(id, 0, sizeof(struct mtdids) + mtd_id_len);
1663                 id->num = num;
1664                 id->type = type;
1665                 id->size = size;
1666                 id->mtd_id = (char *)(id + 1);
1667                 strncpy(id->mtd_id, mtd_id, mtd_id_len - 1);
1668                 id->mtd_id[mtd_id_len - 1] = '\0';
1669                 INIT_LIST_HEAD(&id->link);
1670
1671                 debug("+ id %s%d\t%16lld bytes\t%s\n",
1672                                 MTD_DEV_TYPE(id->type), id->num,
1673                                 id->size, id->mtd_id);
1674
1675                 list_add_tail(&id->link, &mtdids);
1676                 ret = 0;
1677         }
1678         if (ret == 1) {
1679                 /* clean mtdids list and free allocated memory */
1680                 list_for_each_safe(entry, n, &mtdids) {
1681                         id_tmp = list_entry(entry, struct mtdids, link);
1682                         list_del(entry);
1683                         free(id_tmp);
1684                 }
1685                 return 1;
1686         }
1687
1688         return 0;
1689 }
1690
1691 /**
1692  * Parse and initialize global mtdids mapping and create global
1693  * device/partition list.
1694  *
1695  * @return 0 on success, 1 otherwise
1696  */
1697 int mtdparts_init(void)
1698 {
1699         static int initialized = 0;
1700         const char *ids, *parts;
1701         const char *current_partition;
1702         int ids_changed;
1703         char tmp_ep[PARTITION_MAXLEN];
1704         char tmp_parts[MTDPARTS_MAXLEN];
1705
1706         debug("\n---mtdparts_init---\n");
1707         if (!initialized) {
1708                 INIT_LIST_HEAD(&mtdids);
1709                 INIT_LIST_HEAD(&devices);
1710                 memset(last_ids, 0, MTDIDS_MAXLEN);
1711                 memset(last_parts, 0, MTDPARTS_MAXLEN);
1712                 memset(last_partition, 0, PARTITION_MAXLEN);
1713                 initialized = 1;
1714         }
1715
1716         /* get variables */
1717         ids = getenv("mtdids");
1718         /*
1719          * The mtdparts variable tends to be long. If we need to access it
1720          * before the env is relocated, then we need to use our own stack
1721          * buffer.  gd->env_buf will be too small.
1722          */
1723         if (gd->flags & GD_FLG_ENV_READY) {
1724                 parts = getenv("mtdparts");
1725         } else {
1726                 parts = tmp_parts;
1727                 getenv_f("mtdparts", tmp_parts, MTDPARTS_MAXLEN);
1728         }
1729         current_partition = getenv("partition");
1730
1731         /* save it for later parsing, cannot rely on current partition pointer
1732          * as 'partition' variable may be updated during init */
1733         tmp_ep[0] = '\0';
1734         if (current_partition)
1735                 strncpy(tmp_ep, current_partition, PARTITION_MAXLEN);
1736
1737         debug("last_ids  : %s\n", last_ids);
1738         debug("env_ids   : %s\n", ids);
1739         debug("last_parts: %s\n", last_parts);
1740         debug("env_parts : %s\n\n", parts);
1741
1742         debug("last_partition : %s\n", last_partition);
1743         debug("env_partition  : %s\n", current_partition);
1744
1745         /* if mtdids varible is empty try to use defaults */
1746         if (!ids) {
1747                 if (mtdids_default) {
1748                         debug("mtdids variable not defined, using default\n");
1749                         ids = mtdids_default;
1750                         setenv("mtdids", (char *)ids);
1751                 } else {
1752                         printf("mtdids not defined, no default present\n");
1753                         return 1;
1754                 }
1755         }
1756         if (strlen(ids) > MTDIDS_MAXLEN - 1) {
1757                 printf("mtdids too long (> %d)\n", MTDIDS_MAXLEN);
1758                 return 1;
1759         }
1760
1761         /* do no try to use defaults when mtdparts variable is not defined,
1762          * just check the length */
1763         if (!parts)
1764                 printf("mtdparts variable not set, see 'help mtdparts'\n");
1765
1766         if (parts && (strlen(parts) > MTDPARTS_MAXLEN - 1)) {
1767                 printf("mtdparts too long (> %d)\n", MTDPARTS_MAXLEN);
1768                 return 1;
1769         }
1770
1771         /* check if we have already parsed those mtdids */
1772         if ((last_ids[0] != '\0') && (strcmp(last_ids, ids) == 0)) {
1773                 ids_changed = 0;
1774         } else {
1775                 ids_changed = 1;
1776
1777                 if (parse_mtdids(ids) != 0) {
1778                         mtd_devices_init();
1779                         return 1;
1780                 }
1781
1782                 /* ok it's good, save new ids */
1783                 strncpy(last_ids, ids, MTDIDS_MAXLEN);
1784         }
1785
1786         /* parse partitions if either mtdparts or mtdids were updated */
1787         if (parts && ((last_parts[0] == '\0') || ((strcmp(last_parts, parts) != 0)) || ids_changed)) {
1788                 if (parse_mtdparts(parts) != 0)
1789                         return 1;
1790
1791                 if (list_empty(&devices)) {
1792                         printf("mtdparts_init: no valid partitions\n");
1793                         return 1;
1794                 }
1795
1796                 /* ok it's good, save new parts */
1797                 strncpy(last_parts, parts, MTDPARTS_MAXLEN);
1798
1799                 /* reset first partition from first dev from the list as current */
1800                 current_mtd_dev = list_entry(devices.next, struct mtd_device, link);
1801                 current_mtd_partnum = 0;
1802                 current_save();
1803
1804                 debug("mtdparts_init: current_mtd_dev  = %s%d, current_mtd_partnum = %d\n",
1805                                 MTD_DEV_TYPE(current_mtd_dev->id->type),
1806                                 current_mtd_dev->id->num, current_mtd_partnum);
1807         }
1808
1809         /* mtdparts variable was reset to NULL, delete all devices/partitions */
1810         if (!parts && (last_parts[0] != '\0'))
1811                 return mtd_devices_init();
1812
1813         /* do not process current partition if mtdparts variable is null */
1814         if (!parts)
1815                 return 0;
1816
1817         /* is current partition set in environment? if so, use it */
1818         if ((tmp_ep[0] != '\0') && (strcmp(tmp_ep, last_partition) != 0)) {
1819                 struct part_info *p;
1820                 struct mtd_device *cdev;
1821                 u8 pnum;
1822
1823                 debug("--- getting current partition: %s\n", tmp_ep);
1824
1825                 if (find_dev_and_part(tmp_ep, &cdev, &pnum, &p) == 0) {
1826                         current_mtd_dev = cdev;
1827                         current_mtd_partnum = pnum;
1828                         current_save();
1829                 }
1830         } else if (getenv("partition") == NULL) {
1831                 debug("no partition variable set, setting...\n");
1832                 current_save();
1833         }
1834
1835         return 0;
1836 }
1837
1838 /**
1839  * Return pointer to the partition of a requested number from a requested
1840  * device.
1841  *
1842  * @param dev device that is to be searched for a partition
1843  * @param part_num requested partition number
1844  * @return pointer to the part_info, NULL otherwise
1845  */
1846 static struct part_info* mtd_part_info(struct mtd_device *dev, unsigned int part_num)
1847 {
1848         struct list_head *entry;
1849         struct part_info *part;
1850         int num;
1851
1852         if (!dev)
1853                 return NULL;
1854
1855         debug("\n--- mtd_part_info: partition number %d for device %s%d (%s)\n",
1856                         part_num, MTD_DEV_TYPE(dev->id->type),
1857                         dev->id->num, dev->id->mtd_id);
1858
1859         if (part_num >= dev->num_parts) {
1860                 printf("invalid partition number %d for device %s%d (%s)\n",
1861                                 part_num, MTD_DEV_TYPE(dev->id->type),
1862                                 dev->id->num, dev->id->mtd_id);
1863                 return NULL;
1864         }
1865
1866         /* locate partition number, return it */
1867         num = 0;
1868         list_for_each(entry, &dev->parts) {
1869                 part = list_entry(entry, struct part_info, link);
1870
1871                 if (part_num == num++) {
1872                         return part;
1873                 }
1874         }
1875
1876         return NULL;
1877 }
1878
1879 /***************************************************/
1880 /* U-boot commands                                 */
1881 /***************************************************/
1882 /* command line only */
1883 /**
1884  * Routine implementing u-boot chpart command. Sets new current partition based
1885  * on the user supplied partition id. For partition id format see find_dev_and_part().
1886  *
1887  * @param cmdtp command internal data
1888  * @param flag command flag
1889  * @param argc number of arguments supplied to the command
1890  * @param argv arguments list
1891  * @return 0 on success, 1 otherwise
1892  */
1893 static int do_chpart(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
1894 {
1895 /* command line only */
1896         struct mtd_device *dev;
1897         struct part_info *part;
1898         u8 pnum;
1899
1900         if (mtdparts_init() !=0)
1901                 return 1;
1902
1903         if (argc < 2) {
1904                 printf("no partition id specified\n");
1905                 return 1;
1906         }
1907
1908         if (find_dev_and_part(argv[1], &dev, &pnum, &part) != 0)
1909                 return 1;
1910
1911         current_mtd_dev = dev;
1912         current_mtd_partnum = pnum;
1913         current_save();
1914
1915         printf("partition changed to %s%d,%d\n",
1916                         MTD_DEV_TYPE(dev->id->type), dev->id->num, pnum);
1917
1918         return 0;
1919 }
1920
1921 /**
1922  * Routine implementing u-boot mtdparts command. Initialize/update default global
1923  * partition list and process user partition request (list, add, del).
1924  *
1925  * @param cmdtp command internal data
1926  * @param flag command flag
1927  * @param argc number of arguments supplied to the command
1928  * @param argv arguments list
1929  * @return 0 on success, 1 otherwise
1930  */
1931 static int do_mtdparts(cmd_tbl_t *cmdtp, int flag, int argc,
1932                        char * const argv[])
1933 {
1934         if (argc == 2) {
1935                 if (strcmp(argv[1], "default") == 0) {
1936                         setenv("mtdids", (char *)mtdids_default);
1937                         setenv("mtdparts", (char *)mtdparts_default);
1938                         setenv("partition", NULL);
1939
1940                         mtdparts_init();
1941                         return 0;
1942                 } else if (strcmp(argv[1], "delall") == 0) {
1943                         /* this may be the first run, initialize lists if needed */
1944                         mtdparts_init();
1945
1946                         setenv("mtdparts", NULL);
1947
1948                         /* mtd_devices_init() calls current_save() */
1949                         return mtd_devices_init();
1950                 }
1951         }
1952
1953         /* make sure we are in sync with env variables */
1954         if (mtdparts_init() != 0)
1955                 return 1;
1956
1957         if (argc == 1) {
1958                 list_partitions();
1959                 return 0;
1960         }
1961
1962         /* mtdparts add <mtd-dev> <size>[@<offset>] <name> [ro] */
1963         if (((argc == 5) || (argc == 6)) && (strncmp(argv[1], "add", 3) == 0)) {
1964 #define PART_ADD_DESC_MAXLEN 64
1965                 char tmpbuf[PART_ADD_DESC_MAXLEN];
1966 #if defined(CONFIG_CMD_MTDPARTS_SPREAD)
1967                 struct mtd_info *mtd;
1968                 uint64_t next_offset;
1969 #endif
1970                 u8 type, num, len;
1971                 struct mtd_device *dev;
1972                 struct mtd_device *dev_tmp;
1973                 struct mtdids *id;
1974                 struct part_info *p;
1975
1976                 if (mtd_id_parse(argv[2], NULL, &type, &num) != 0)
1977                         return 1;
1978
1979                 if ((id = id_find(type, num)) == NULL) {
1980                         printf("no such device %s defined in mtdids variable\n", argv[2]);
1981                         return 1;
1982                 }
1983
1984                 len = strlen(id->mtd_id) + 1;   /* 'mtd_id:' */
1985                 len += strlen(argv[3]);         /* size@offset */
1986                 len += strlen(argv[4]) + 2;     /* '(' name ')' */
1987                 if (argv[5] && (strlen(argv[5]) == 2))
1988                         len += 2;               /* 'ro' */
1989
1990                 if (len >= PART_ADD_DESC_MAXLEN) {
1991                         printf("too long partition description\n");
1992                         return 1;
1993                 }
1994                 sprintf(tmpbuf, "%s:%s(%s)%s",
1995                                 id->mtd_id, argv[3], argv[4], argv[5] ? argv[5] : "");
1996                 debug("add tmpbuf: %s\n", tmpbuf);
1997
1998                 if ((device_parse(tmpbuf, NULL, &dev) != 0) || (!dev))
1999                         return 1;
2000
2001                 debug("+ %s\t%d\t%s\n", MTD_DEV_TYPE(dev->id->type),
2002                                 dev->id->num, dev->id->mtd_id);
2003
2004                 p = list_entry(dev->parts.next, struct part_info, link);
2005
2006 #if defined(CONFIG_CMD_MTDPARTS_SPREAD)
2007                 if (get_mtd_info(dev->id->type, dev->id->num, &mtd))
2008                         return 1;
2009
2010                 if (!strcmp(&argv[1][3], ".spread")) {
2011                         spread_partition(mtd, p, &next_offset);
2012                         debug("increased %s to %d bytes\n", p->name, p->size);
2013                 }
2014 #endif
2015
2016                 dev_tmp = device_find(dev->id->type, dev->id->num);
2017                 if (dev_tmp == NULL) {
2018                         device_add(dev);
2019                 } else if (part_add(dev_tmp, p) != 0) {
2020                         /* merge new partition with existing ones*/
2021                         device_del(dev);
2022                         return 1;
2023                 }
2024
2025                 if (generate_mtdparts_save(last_parts, MTDPARTS_MAXLEN) != 0) {
2026                         printf("generated mtdparts too long, resetting to null\n");
2027                         return 1;
2028                 }
2029
2030                 return 0;
2031         }
2032
2033         /* mtdparts del part-id */
2034         if ((argc == 3) && (strcmp(argv[1], "del") == 0)) {
2035                 debug("del: part-id = %s\n", argv[2]);
2036
2037                 return delete_partition(argv[2]);
2038         }
2039
2040 #if defined(CONFIG_CMD_MTDPARTS_SPREAD)
2041         if ((argc == 2) && (strcmp(argv[1], "spread") == 0))
2042                 return spread_partitions();
2043 #endif /* CONFIG_CMD_MTDPARTS_SPREAD */
2044
2045         return CMD_RET_USAGE;
2046 }
2047
2048 /***************************************************/
2049 U_BOOT_CMD(
2050         chpart, 2,      0,      do_chpart,
2051         "change active partition",
2052         "part-id\n"
2053         "    - change active partition (e.g. part-id = nand0,1)"
2054 );
2055
2056 #ifdef CONFIG_SYS_LONGHELP
2057 static char mtdparts_help_text[] =
2058         "\n"
2059         "    - list partition table\n"
2060         "mtdparts delall\n"
2061         "    - delete all partitions\n"
2062         "mtdparts del part-id\n"
2063         "    - delete partition (e.g. part-id = nand0,1)\n"
2064         "mtdparts add <mtd-dev> <size>[@<offset>] [<name>] [ro]\n"
2065         "    - add partition\n"
2066 #if defined(CONFIG_CMD_MTDPARTS_SPREAD)
2067         "mtdparts add.spread <mtd-dev> <size>[@<offset>] [<name>] [ro]\n"
2068         "    - add partition, padding size by skipping bad blocks\n"
2069 #endif
2070         "mtdparts default\n"
2071         "    - reset partition table to defaults\n"
2072 #if defined(CONFIG_CMD_MTDPARTS_SPREAD)
2073         "mtdparts spread\n"
2074         "    - adjust the sizes of the partitions so they are\n"
2075         "      at least as big as the mtdparts variable specifies\n"
2076         "      and they each start on a good block\n\n"
2077 #else
2078         "\n"
2079 #endif /* CONFIG_CMD_MTDPARTS_SPREAD */
2080         "-----\n\n"
2081         "this command uses three environment variables:\n\n"
2082         "'partition' - keeps current partition identifier\n\n"
2083         "partition  := <part-id>\n"
2084         "<part-id>  := <dev-id>,part_num\n\n"
2085         "'mtdids' - linux kernel mtd device id <-> u-boot device id mapping\n\n"
2086         "mtdids=<idmap>[,<idmap>,...]\n\n"
2087         "<idmap>    := <dev-id>=<mtd-id>\n"
2088         "<dev-id>   := 'nand'|'nor'|'onenand'<dev-num>\n"
2089         "<dev-num>  := mtd device number, 0...\n"
2090         "<mtd-id>   := unique device tag used by linux kernel to find mtd device (mtd->name)\n\n"
2091         "'mtdparts' - partition list\n\n"
2092         "mtdparts=mtdparts=<mtd-def>[;<mtd-def>...]\n\n"
2093         "<mtd-def>  := <mtd-id>:<part-def>[,<part-def>...]\n"
2094         "<mtd-id>   := unique device tag used by linux kernel to find mtd device (mtd->name)\n"
2095         "<part-def> := <size>[@<offset>][<name>][<ro-flag>]\n"
2096         "<size>     := standard linux memsize OR '-' to denote all remaining space\n"
2097         "<offset>   := partition start offset within the device\n"
2098         "<name>     := '(' NAME ')'\n"
2099         "<ro-flag>  := when set to 'ro' makes partition read-only (not used, passed to kernel)";
2100 #endif
2101
2102 U_BOOT_CMD(
2103         mtdparts,       6,      0,      do_mtdparts,
2104         "define flash/nand partitions", mtdparts_help_text
2105 );
2106 /***************************************************/