mtd: Make UBI detection more robust
When we want to detect if a mtd device contains an UBI image then testing the first block is not enough since it can always happen that UBI has just erased the block before the power failed during last boot. Since UBI only ever erases one block at a time and directly writes the ec header to it afterwards, it shouldn't be necessary to scan the whole device for UBI data. Scan the first 64 blocks. The first non-empty block then must contain UBI data, if instead we find foreign data we assume that no UBI is on that mtd device. Signed-off-by: Sascha Hauer <s.hauer@pengutronix.de>
This commit is contained in:
parent
6f58e5cac9
commit
0d68468c96
|
@ -18,6 +18,7 @@
|
|||
#include <common.h>
|
||||
#include <linux/mtd/nand.h>
|
||||
#include <linux/mtd/mtd.h>
|
||||
#include <mtd/mtd-peb.h>
|
||||
#include <mtd/ubi-user.h>
|
||||
#include <cmdlinepart.h>
|
||||
#include <filetype.h>
|
||||
|
@ -622,9 +623,11 @@ static int mtd_detect(struct device_d *dev)
|
|||
struct mtd_info *mtd = container_of(dev, struct mtd_info, class_dev);
|
||||
int bufsize = 512;
|
||||
void *buf;
|
||||
int ret;
|
||||
int ret = 0, i;
|
||||
enum filetype filetype;
|
||||
size_t retlen;
|
||||
int npebs = mtd_div_by_eb(mtd->size, mtd);
|
||||
|
||||
npebs = max(npebs, 64);
|
||||
|
||||
/*
|
||||
* Do not try to attach an UBI device if this device has partitions
|
||||
|
@ -636,17 +639,27 @@ static int mtd_detect(struct device_d *dev)
|
|||
|
||||
buf = xmalloc(bufsize);
|
||||
|
||||
ret = mtd_read(mtd, 0, bufsize, &retlen, buf);
|
||||
if (ret)
|
||||
goto out;
|
||||
for (i = 0; i < npebs; i++) {
|
||||
if (mtd_peb_is_bad(mtd, i))
|
||||
continue;
|
||||
|
||||
filetype = file_detect_type(buf, bufsize);
|
||||
if (filetype == filetype_ubi) {
|
||||
ret = ubi_attach_mtd_dev(mtd, UBI_DEV_NUM_AUTO, 0, 20);
|
||||
if (ret == -EEXIST)
|
||||
ret = 0;
|
||||
ret = mtd_peb_read(mtd, buf, i, 0, 512);
|
||||
if (ret)
|
||||
continue;
|
||||
|
||||
if (mtd_buf_all_ff(buf, 512))
|
||||
continue;
|
||||
|
||||
filetype = file_detect_type(buf, bufsize);
|
||||
if (filetype == filetype_ubi) {
|
||||
ret = ubi_attach_mtd_dev(mtd, UBI_DEV_NUM_AUTO, 0, 20);
|
||||
if (ret == -EEXIST)
|
||||
ret = 0;
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
out:
|
||||
|
||||
free(buf);
|
||||
|
||||
return ret;
|
||||
|
|
Loading…
Reference in New Issue