1*002181f5SRafał Miłecki // SPDX-License-Identifier: GPL-2.0-only 2*002181f5SRafał Miłecki /* 3*002181f5SRafał Miłecki * Copyright © 2022 Rafał Miłecki <rafal@milecki.pl> 4*002181f5SRafał Miłecki */ 5*002181f5SRafał Miłecki 6*002181f5SRafał Miłecki #include <linux/module.h> 7*002181f5SRafał Miłecki #include <linux/kernel.h> 8*002181f5SRafał Miłecki #include <linux/slab.h> 9*002181f5SRafał Miłecki #include <linux/mtd/mtd.h> 10*002181f5SRafał Miłecki #include <linux/mtd/partitions.h> 11*002181f5SRafał Miłecki 12*002181f5SRafał Miłecki #define BRCM_U_BOOT_MAX_OFFSET 0x200000 13*002181f5SRafał Miłecki #define BRCM_U_BOOT_STEP 0x1000 14*002181f5SRafał Miłecki 15*002181f5SRafał Miłecki #define BRCM_U_BOOT_MAX_PARTS 2 16*002181f5SRafał Miłecki 17*002181f5SRafał Miłecki #define BRCM_U_BOOT_MAGIC 0x75456e76 /* uEnv */ 18*002181f5SRafał Miłecki 19*002181f5SRafał Miłecki struct brcm_u_boot_header { 20*002181f5SRafał Miłecki __le32 magic; 21*002181f5SRafał Miłecki __le32 length; 22*002181f5SRafał Miłecki } __packed; 23*002181f5SRafał Miłecki 24*002181f5SRafał Miłecki static const char *names[BRCM_U_BOOT_MAX_PARTS] = { 25*002181f5SRafał Miłecki "u-boot-env", 26*002181f5SRafał Miłecki "u-boot-env-backup", 27*002181f5SRafał Miłecki }; 28*002181f5SRafał Miłecki 29*002181f5SRafał Miłecki static int brcm_u_boot_parse(struct mtd_info *mtd, 30*002181f5SRafał Miłecki const struct mtd_partition **pparts, 31*002181f5SRafał Miłecki struct mtd_part_parser_data *data) 32*002181f5SRafał Miłecki { 33*002181f5SRafał Miłecki struct brcm_u_boot_header header; 34*002181f5SRafał Miłecki struct mtd_partition *parts; 35*002181f5SRafał Miłecki size_t bytes_read; 36*002181f5SRafał Miłecki size_t offset; 37*002181f5SRafał Miłecki int err; 38*002181f5SRafał Miłecki int i = 0; 39*002181f5SRafał Miłecki 40*002181f5SRafał Miłecki parts = kcalloc(BRCM_U_BOOT_MAX_PARTS, sizeof(*parts), GFP_KERNEL); 41*002181f5SRafał Miłecki if (!parts) 42*002181f5SRafał Miłecki return -ENOMEM; 43*002181f5SRafał Miłecki 44*002181f5SRafał Miłecki for (offset = 0; 45*002181f5SRafał Miłecki offset < min_t(size_t, mtd->size, BRCM_U_BOOT_MAX_OFFSET); 46*002181f5SRafał Miłecki offset += BRCM_U_BOOT_STEP) { 47*002181f5SRafał Miłecki err = mtd_read(mtd, offset, sizeof(header), &bytes_read, (uint8_t *)&header); 48*002181f5SRafał Miłecki if (err && !mtd_is_bitflip(err)) { 49*002181f5SRafał Miłecki pr_err("Failed to read from %s at 0x%zx: %d\n", mtd->name, offset, err); 50*002181f5SRafał Miłecki continue; 51*002181f5SRafał Miłecki } 52*002181f5SRafał Miłecki 53*002181f5SRafał Miłecki if (le32_to_cpu(header.magic) != BRCM_U_BOOT_MAGIC) 54*002181f5SRafał Miłecki continue; 55*002181f5SRafał Miłecki 56*002181f5SRafał Miłecki parts[i].name = names[i]; 57*002181f5SRafał Miłecki parts[i].offset = offset; 58*002181f5SRafał Miłecki parts[i].size = sizeof(header) + le32_to_cpu(header.length); 59*002181f5SRafał Miłecki i++; 60*002181f5SRafał Miłecki pr_info("offset:0x%zx magic:0x%08x BINGO\n", offset, header.magic); 61*002181f5SRafał Miłecki 62*002181f5SRafał Miłecki if (i == BRCM_U_BOOT_MAX_PARTS) 63*002181f5SRafał Miłecki break; 64*002181f5SRafał Miłecki } 65*002181f5SRafał Miłecki 66*002181f5SRafał Miłecki *pparts = parts; 67*002181f5SRafał Miłecki 68*002181f5SRafał Miłecki return i; 69*002181f5SRafał Miłecki }; 70*002181f5SRafał Miłecki 71*002181f5SRafał Miłecki static const struct of_device_id brcm_u_boot_of_match_table[] = { 72*002181f5SRafał Miłecki { .compatible = "brcm,u-boot" }, 73*002181f5SRafał Miłecki {}, 74*002181f5SRafał Miłecki }; 75*002181f5SRafał Miłecki MODULE_DEVICE_TABLE(of, brcm_u_boot_of_match_table); 76*002181f5SRafał Miłecki 77*002181f5SRafał Miłecki static struct mtd_part_parser brcm_u_boot_mtd_parser = { 78*002181f5SRafał Miłecki .parse_fn = brcm_u_boot_parse, 79*002181f5SRafał Miłecki .name = "brcm_u-boot", 80*002181f5SRafał Miłecki .of_match_table = brcm_u_boot_of_match_table, 81*002181f5SRafał Miłecki }; 82*002181f5SRafał Miłecki module_mtd_part_parser(brcm_u_boot_mtd_parser); 83*002181f5SRafał Miłecki 84*002181f5SRafał Miłecki MODULE_LICENSE("GPL"); 85