1// SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause 2/* 3 * Copyright (C) 2020, STMicroelectronics - All Rights Reserved 4 */ 5 6#include <common.h> 7#include <blk.h> 8#include <dm.h> 9#include <dfu.h> 10#include <env.h> 11#include <log.h> 12#include <memalign.h> 13#include <misc.h> 14#include <mtd.h> 15#include <mtd_node.h> 16#include <asm/arch/stm32prog.h> 17#include <linux/printk.h> 18 19#define DFU_ALT_BUF_LEN SZ_1K 20 21static void board_get_alt_info_mmc(struct udevice *dev, char *buf) 22{ 23 struct disk_partition info; 24 int p, len, devnum; 25 bool first = true; 26 const char *name; 27 struct mmc *mmc; 28 struct blk_desc *desc; 29 30 mmc = mmc_get_mmc_dev(dev); 31 if (!mmc) 32 return; 33 34 if (mmc_init(mmc)) 35 return; 36 37 desc = mmc_get_blk_desc(mmc); 38 if (!desc) 39 return; 40 41 name = blk_get_uclass_name(desc->uclass_id); 42 devnum = desc->devnum; 43 len = strlen(buf); 44 45 if (buf[0] != '\0') 46 len += snprintf(buf + len, 47 DFU_ALT_BUF_LEN - len, "&"); 48 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, 49 "%s %d=", name, devnum); 50 51 if (IS_MMC(mmc) && mmc->capacity_boot) { 52 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, 53 "%s%d_boot1 raw 0x0 0x%llx mmcpart 1;", 54 name, devnum, mmc->capacity_boot); 55 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, 56 "%s%d_boot2 raw 0x0 0x%llx mmcpart 2", 57 name, devnum, mmc->capacity_boot); 58 first = false; 59 } 60 61 for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) { 62 if (part_get_info(desc, p, &info)) 63 continue; 64 if (!first) 65 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, ";"); 66 first = false; 67 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, 68 "%s%d_%s part %d %d", 69 name, devnum, info.name, devnum, p); 70 } 71} 72 73static void board_get_alt_info_mtd(struct mtd_info *mtd, char *buf) 74{ 75 struct mtd_info *part; 76 const char *name; 77 int len, partnum = 0; 78 79 name = mtd->name; 80 len = strlen(buf); 81 82 if (buf[0] != '\0') 83 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, "&"); 84 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, 85 "mtd %s=", name); 86 87 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, 88 "%s raw 0x0 0x%llx", 89 name, mtd->size); 90 91 list_for_each_entry(part, &mtd->partitions, node) { 92 partnum++; 93 len += snprintf(buf + len, DFU_ALT_BUF_LEN - len, 94 ";%s_%s part %d", 95 name, part->name, partnum); 96 } 97} 98 99void set_dfu_alt_info(char *interface, char *devstr) 100{ 101 struct udevice *dev; 102 struct mtd_info *mtd; 103 104 ALLOC_CACHE_ALIGN_BUFFER(char, buf, DFU_ALT_BUF_LEN); 105 106 if (env_get("dfu_alt_info")) 107 return; 108 109 memset(buf, 0, sizeof(buf)); 110 111 snprintf(buf, DFU_ALT_BUF_LEN, 112 "ram 0=%s", CONFIG_DFU_ALT_RAM0); 113 114 if (CONFIG_IS_ENABLED(MMC)) { 115 if (!uclass_get_device(UCLASS_MMC, 0, &dev)) 116 board_get_alt_info_mmc(dev, buf); 117 118 if (!uclass_get_device(UCLASS_MMC, 1, &dev)) 119 board_get_alt_info_mmc(dev, buf); 120 } 121 122 if (IS_ENABLED(CONFIG_MTD)) { 123 /* probe all MTD devices */ 124 mtd_probe_devices(); 125 126 mtd_for_each_device(mtd) 127 if (!mtd_is_partition(mtd)) 128 board_get_alt_info_mtd(mtd, buf); 129 } 130 131 if (IS_ENABLED(CONFIG_DFU_VIRT)) { 132 /* virtual device id 0 is aligned with stm32mp_dfu_virt.c */ 133 strlcat(buf, "&virt 0=OTP", DFU_ALT_BUF_LEN); 134 135 if (IS_ENABLED(CONFIG_PMIC_STPMIC1)) 136 strlcat(buf, "&virt 1=PMIC", DFU_ALT_BUF_LEN); 137 } 138 139 env_set("dfu_alt_info", buf); 140 puts("DFU alt info setting: done\n"); 141} 142