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