188dc4099SStefan Roese // SPDX-License-Identifier: GPL-2.0+
288dc4099SStefan Roese /*
388dc4099SStefan Roese  * Copyright (C) 2018 Stefan Roese <sr@denx.de>
488dc4099SStefan Roese  */
588dc4099SStefan Roese 
688dc4099SStefan Roese #include <common.h>
7*7b3af03fSStefan Roese #include <environment.h>
8b1f51fc2SStefan Roese #include <led.h>
9*7b3af03fSStefan Roese #include <net.h>
10*7b3af03fSStefan Roese #include <spi.h>
11*7b3af03fSStefan Roese #include <spi_flash.h>
12*7b3af03fSStefan Roese #include <uuid.h>
13*7b3af03fSStefan Roese #include <linux/ctype.h>
1448f8e159SStefan Roese #include <linux/io.h>
1548f8e159SStefan Roese 
1648f8e159SStefan Roese #define MT76XX_AGPIO_CFG	0x1000003c
1788dc4099SStefan Roese 
18*7b3af03fSStefan Roese #define FACTORY_DATA_OFFS	0xc0000
19*7b3af03fSStefan Roese #define FACTORY_DATA_SECT_SIZE	0x10000
20*7b3af03fSStefan Roese #if ((CONFIG_ENV_OFFSET_REDUND + CONFIG_ENV_SIZE_REDUND) > FACTORY_DATA_OFFS)
21*7b3af03fSStefan Roese #error "U-Boot image with environment too big (overlapping with factory-data)!"
22*7b3af03fSStefan Roese #endif
23*7b3af03fSStefan Roese #define FACTORY_DATA_USER_OFFS	0x140
24*7b3af03fSStefan Roese #define FACTORY_DATA_SIZE	0x1f0
25*7b3af03fSStefan Roese #define FACTORY_DATA_CRC_LEN	(FACTORY_DATA_SIZE -			\
26*7b3af03fSStefan Roese 				 FACTORY_DATA_USER_OFFS - sizeof(u32))
27*7b3af03fSStefan Roese 
28*7b3af03fSStefan Roese #define FACTORY_DATA_MAGIC	0xCAFEBABE
29*7b3af03fSStefan Roese 
30*7b3af03fSStefan Roese struct factory_data_values {
31*7b3af03fSStefan Roese 	u8 pad_1[4];
32*7b3af03fSStefan Roese 	u8 wifi_mac[6];		/* offs: 0x004: binary value */
33*7b3af03fSStefan Roese 	u8 pad_2[30];
34*7b3af03fSStefan Roese 	u8 eth_mac[6];		/* offs: 0x028: binary value */
35*7b3af03fSStefan Roese 	u8 pad_3[FACTORY_DATA_USER_OFFS - 4 - 6 - 30 - 6];
36*7b3af03fSStefan Roese 	/* User values start here at offset 0x140 */
37*7b3af03fSStefan Roese 	u32 crc;
38*7b3af03fSStefan Roese 	u32 magic;
39*7b3af03fSStefan Roese 	u32 version;
40*7b3af03fSStefan Roese 	char ipr_id[UUID_STR_LEN];	/* UUID as string w/o ending \0 */
41*7b3af03fSStefan Roese 	char hqv_id[UUID_STR_LEN];	/* UUID as string w/o ending \0 */
42*7b3af03fSStefan Roese 	char unielec_id[UUID_STR_LEN];	/* UUID as string w/o ending \0 */
43*7b3af03fSStefan Roese };
44*7b3af03fSStefan Roese 
board_early_init_f(void)4588dc4099SStefan Roese int board_early_init_f(void)
4688dc4099SStefan Roese {
4748f8e159SStefan Roese 	void __iomem *gpio_mode;
4848f8e159SStefan Roese 
4948f8e159SStefan Roese 	/* Configure digital vs analog GPIOs */
5048f8e159SStefan Roese 	gpio_mode = ioremap_nocache(MT76XX_AGPIO_CFG, 0x100);
5148f8e159SStefan Roese 	iowrite32(0x00fe01ff, gpio_mode);
5248f8e159SStefan Roese 
5388dc4099SStefan Roese 	return 0;
5488dc4099SStefan Roese }
55b1f51fc2SStefan Roese 
prepare_uuid_var(const char * fd_ptr,const char * env_var_name,char errorchar)56*7b3af03fSStefan Roese static bool prepare_uuid_var(const char *fd_ptr, const char *env_var_name,
57*7b3af03fSStefan Roese 			     char errorchar)
58*7b3af03fSStefan Roese {
59*7b3af03fSStefan Roese 	char str[UUID_STR_LEN + 1] = { 0 };	/* Enough for UUID stuff */
60*7b3af03fSStefan Roese 	bool env_updated = false;
61*7b3af03fSStefan Roese 	char *env;
62*7b3af03fSStefan Roese 	int i;
63*7b3af03fSStefan Roese 
64*7b3af03fSStefan Roese 	memcpy(str, fd_ptr, UUID_STR_LEN);
65*7b3af03fSStefan Roese 
66*7b3af03fSStefan Roese 	/* Convert non-ascii character to 'X' */
67*7b3af03fSStefan Roese 	for (i = 0; i < UUID_STR_LEN; i++) {
68*7b3af03fSStefan Roese 		if (!(isascii(str[i]) && isprint(str[i])))
69*7b3af03fSStefan Roese 			str[i] = errorchar;
70*7b3af03fSStefan Roese 	}
71*7b3af03fSStefan Roese 
72*7b3af03fSStefan Roese 	env = env_get(env_var_name);
73*7b3af03fSStefan Roese 	if (strcmp(env, str)) {
74*7b3af03fSStefan Roese 		env_set(env_var_name, str);
75*7b3af03fSStefan Roese 		env_updated = true;
76*7b3af03fSStefan Roese 	}
77*7b3af03fSStefan Roese 
78*7b3af03fSStefan Roese 	return env_updated;
79*7b3af03fSStefan Roese }
80*7b3af03fSStefan Roese 
factory_data_env_config(void)81*7b3af03fSStefan Roese static void factory_data_env_config(void)
82*7b3af03fSStefan Roese {
83*7b3af03fSStefan Roese 	struct factory_data_values *fd;
84*7b3af03fSStefan Roese 	struct spi_flash *sf;
85*7b3af03fSStefan Roese 	int env_updated = 0;
86*7b3af03fSStefan Roese 	char str[UUID_STR_LEN + 1];	/* Enough for UUID stuff */
87*7b3af03fSStefan Roese 	char *env;
88*7b3af03fSStefan Roese 	u8 *buf;
89*7b3af03fSStefan Roese 	u32 crc;
90*7b3af03fSStefan Roese 	int ret;
91*7b3af03fSStefan Roese 	u8 *ptr;
92*7b3af03fSStefan Roese 
93*7b3af03fSStefan Roese 	buf = malloc(FACTORY_DATA_SIZE);
94*7b3af03fSStefan Roese 	if (!buf) {
95*7b3af03fSStefan Roese 		printf("F-Data:Unable to allocate buffer\n");
96*7b3af03fSStefan Roese 		return;
97*7b3af03fSStefan Roese 	}
98*7b3af03fSStefan Roese 
99*7b3af03fSStefan Roese 	/*
100*7b3af03fSStefan Roese 	 * Get values from factory-data area in SPI NOR
101*7b3af03fSStefan Roese 	 */
102*7b3af03fSStefan Roese 	sf = spi_flash_probe(CONFIG_SF_DEFAULT_BUS,
103*7b3af03fSStefan Roese 			     CONFIG_SF_DEFAULT_CS,
104*7b3af03fSStefan Roese 			     CONFIG_SF_DEFAULT_SPEED,
105*7b3af03fSStefan Roese 			     CONFIG_SF_DEFAULT_MODE);
106*7b3af03fSStefan Roese 	if (!sf) {
107*7b3af03fSStefan Roese 		printf("F-Data:Unable to access SPI NOR flash\n");
108*7b3af03fSStefan Roese 		goto err_free;
109*7b3af03fSStefan Roese 	}
110*7b3af03fSStefan Roese 
111*7b3af03fSStefan Roese 	ret = spi_flash_read(sf, FACTORY_DATA_OFFS, FACTORY_DATA_SIZE,
112*7b3af03fSStefan Roese 			     (void *)buf);
113*7b3af03fSStefan Roese 	if (ret) {
114*7b3af03fSStefan Roese 		printf("F-Data:Unable to read factory-data from SPI NOR\n");
115*7b3af03fSStefan Roese 		goto err_spi_flash;
116*7b3af03fSStefan Roese 	}
117*7b3af03fSStefan Roese 
118*7b3af03fSStefan Roese 	fd = (struct factory_data_values *)buf;
119*7b3af03fSStefan Roese 
120*7b3af03fSStefan Roese 	if (fd->magic != FACTORY_DATA_MAGIC)
121*7b3af03fSStefan Roese 		printf("F-Data:Magic value not correct\n");
122*7b3af03fSStefan Roese 
123*7b3af03fSStefan Roese 	crc = crc32(0, (u8 *)&fd->magic, FACTORY_DATA_CRC_LEN);
124*7b3af03fSStefan Roese 	if (crc != fd->crc)
125*7b3af03fSStefan Roese 		printf("F-Data:CRC not correct\n");
126*7b3af03fSStefan Roese 	else
127*7b3af03fSStefan Roese 		printf("F-Data:factory-data version %x detected\n",
128*7b3af03fSStefan Roese 		       fd->version);
129*7b3af03fSStefan Roese 
130*7b3af03fSStefan Roese 	/* Handle wifi_mac env variable */
131*7b3af03fSStefan Roese 	ptr = fd->wifi_mac;
132*7b3af03fSStefan Roese 	sprintf(str, "%pM", ptr);
133*7b3af03fSStefan Roese 	if (!is_valid_ethaddr(ptr))
134*7b3af03fSStefan Roese 		printf("F-Data:Invalid MAC addr: wifi_mac %s\n", str);
135*7b3af03fSStefan Roese 
136*7b3af03fSStefan Roese 	env = env_get("wifiaddr");
137*7b3af03fSStefan Roese 	if (strcmp(env, str)) {
138*7b3af03fSStefan Roese 		env_set("wifiaddr", str);
139*7b3af03fSStefan Roese 		env_updated = 1;
140*7b3af03fSStefan Roese 	}
141*7b3af03fSStefan Roese 
142*7b3af03fSStefan Roese 	/* Handle eth_mac env variable */
143*7b3af03fSStefan Roese 	ptr = fd->eth_mac;
144*7b3af03fSStefan Roese 	sprintf(str, "%pM", ptr);
145*7b3af03fSStefan Roese 	if (!is_valid_ethaddr(ptr))
146*7b3af03fSStefan Roese 		printf("F-Data:Invalid MAC addr: eth_mac %s\n", str);
147*7b3af03fSStefan Roese 
148*7b3af03fSStefan Roese 	env = env_get("ethaddr");
149*7b3af03fSStefan Roese 	if (strcmp(env, str)) {
150*7b3af03fSStefan Roese 		env_set("ethaddr", str);
151*7b3af03fSStefan Roese 		env_updated = 1;
152*7b3af03fSStefan Roese 	}
153*7b3af03fSStefan Roese 
154*7b3af03fSStefan Roese 	/* Handle UUID env variables */
155*7b3af03fSStefan Roese 	env_updated |= prepare_uuid_var(fd->ipr_id, "linuxmoduleid", 'X');
156*7b3af03fSStefan Roese 	env_updated |= prepare_uuid_var(fd->hqv_id, "linuxmodulehqvid", '\0');
157*7b3af03fSStefan Roese 	env_updated |= prepare_uuid_var(fd->unielec_id,
158*7b3af03fSStefan Roese 					"linuxmoduleunielecid", '\0');
159*7b3af03fSStefan Roese 
160*7b3af03fSStefan Roese 	/* Check if the environment was updated and needs to get stored */
161*7b3af03fSStefan Roese 	if (env_updated != 0) {
162*7b3af03fSStefan Roese 		printf("F-Data:Values don't match env values -> saving\n");
163*7b3af03fSStefan Roese 		env_save();
164*7b3af03fSStefan Roese 	} else {
165*7b3af03fSStefan Roese 		debug("F-Data:Values match current env values\n");
166*7b3af03fSStefan Roese 	}
167*7b3af03fSStefan Roese 
168*7b3af03fSStefan Roese err_spi_flash:
169*7b3af03fSStefan Roese 	spi_flash_free(sf);
170*7b3af03fSStefan Roese 
171*7b3af03fSStefan Roese err_free:
172*7b3af03fSStefan Roese 	free(buf);
173*7b3af03fSStefan Roese }
174*7b3af03fSStefan Roese 
board_late_init(void)175b1f51fc2SStefan Roese int board_late_init(void)
176b1f51fc2SStefan Roese {
177b1f51fc2SStefan Roese 	if (IS_ENABLED(CONFIG_LED))
178b1f51fc2SStefan Roese 		led_default_state();
179b1f51fc2SStefan Roese 
180*7b3af03fSStefan Roese 	factory_data_env_config();
181*7b3af03fSStefan Roese 
182b1f51fc2SStefan Roese 	return 0;
183b1f51fc2SStefan Roese }
184*7b3af03fSStefan Roese 
copy_or_generate_uuid(char * fd_ptr,const char * env_var_name)185*7b3af03fSStefan Roese static void copy_or_generate_uuid(char *fd_ptr, const char *env_var_name)
186*7b3af03fSStefan Roese {
187*7b3af03fSStefan Roese 	char str[UUID_STR_LEN + 1] = { 0 };	/* Enough for UUID stuff */
188*7b3af03fSStefan Roese 	char *env;
189*7b3af03fSStefan Roese 
190*7b3af03fSStefan Roese 	/* Don't use the UUID dest place, as the \0 char won't fit */
191*7b3af03fSStefan Roese 	env = env_get(env_var_name);
192*7b3af03fSStefan Roese 	if (env)
193*7b3af03fSStefan Roese 		strncpy(str, env, UUID_STR_LEN);
194*7b3af03fSStefan Roese 	else
195*7b3af03fSStefan Roese 		gen_rand_uuid_str(str, UUID_STR_FORMAT_STD);
196*7b3af03fSStefan Roese 
197*7b3af03fSStefan Roese 	memcpy(fd_ptr, str, UUID_STR_LEN);
198*7b3af03fSStefan Roese }
199*7b3af03fSStefan Roese 
200*7b3af03fSStefan Roese /*
201*7b3af03fSStefan Roese  * Helper function to provide some sane factory-data values for testing
202*7b3af03fSStefan Roese  * purpose, when these values are not programmed correctly
203*7b3af03fSStefan Roese  */
do_fd_write(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])204*7b3af03fSStefan Roese int do_fd_write(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
205*7b3af03fSStefan Roese {
206*7b3af03fSStefan Roese 	struct factory_data_values *fd;
207*7b3af03fSStefan Roese 	struct spi_flash *sf;
208*7b3af03fSStefan Roese 	u8 *buf;
209*7b3af03fSStefan Roese 	int ret = CMD_RET_FAILURE;
210*7b3af03fSStefan Roese 
211*7b3af03fSStefan Roese 	buf = malloc(FACTORY_DATA_SECT_SIZE);
212*7b3af03fSStefan Roese 	if (!buf) {
213*7b3af03fSStefan Roese 		printf("F-Data:Unable to allocate buffer\n");
214*7b3af03fSStefan Roese 		return CMD_RET_FAILURE;
215*7b3af03fSStefan Roese 	}
216*7b3af03fSStefan Roese 
217*7b3af03fSStefan Roese 	sf = spi_flash_probe(CONFIG_SF_DEFAULT_BUS,
218*7b3af03fSStefan Roese 			     CONFIG_SF_DEFAULT_CS,
219*7b3af03fSStefan Roese 			     CONFIG_SF_DEFAULT_SPEED,
220*7b3af03fSStefan Roese 			     CONFIG_SF_DEFAULT_MODE);
221*7b3af03fSStefan Roese 	if (!sf) {
222*7b3af03fSStefan Roese 		printf("F-Data:Unable to access SPI NOR flash\n");
223*7b3af03fSStefan Roese 		goto err_free;
224*7b3af03fSStefan Roese 	}
225*7b3af03fSStefan Roese 
226*7b3af03fSStefan Roese 	/* Generate the factory-data struct */
227*7b3af03fSStefan Roese 
228*7b3af03fSStefan Roese 	/* Fist read complete sector into buffer */
229*7b3af03fSStefan Roese 	ret = spi_flash_read(sf, FACTORY_DATA_OFFS, FACTORY_DATA_SECT_SIZE,
230*7b3af03fSStefan Roese 			     (void *)buf);
231*7b3af03fSStefan Roese 	if (ret) {
232*7b3af03fSStefan Roese 		printf("F-Data:spi_flash_read failed (%d)\n", ret);
233*7b3af03fSStefan Roese 		goto err_spi_flash;
234*7b3af03fSStefan Roese 	}
235*7b3af03fSStefan Roese 
236*7b3af03fSStefan Roese 	fd = (struct factory_data_values *)buf;
237*7b3af03fSStefan Roese 	fd->magic = FACTORY_DATA_MAGIC;
238*7b3af03fSStefan Roese 	fd->version = 0x1;
239*7b3af03fSStefan Roese 
240*7b3af03fSStefan Roese 	/* Use existing MAC and UUID values or generate some random ones */
241*7b3af03fSStefan Roese 	if (!eth_env_get_enetaddr("wifiaddr", fd->wifi_mac)) {
242*7b3af03fSStefan Roese 		net_random_ethaddr(fd->wifi_mac);
243*7b3af03fSStefan Roese 		/* to get a different seed value for the MAC address */
244*7b3af03fSStefan Roese 		mdelay(10);
245*7b3af03fSStefan Roese 	}
246*7b3af03fSStefan Roese 
247*7b3af03fSStefan Roese 	if (!eth_env_get_enetaddr("ethaddr", fd->eth_mac))
248*7b3af03fSStefan Roese 		net_random_ethaddr(fd->eth_mac);
249*7b3af03fSStefan Roese 
250*7b3af03fSStefan Roese 	copy_or_generate_uuid(fd->ipr_id, "linuxmoduleid");
251*7b3af03fSStefan Roese 	copy_or_generate_uuid(fd->hqv_id, "linuxmodulehqvid");
252*7b3af03fSStefan Roese 	copy_or_generate_uuid(fd->unielec_id, "linuxmoduleunielecid");
253*7b3af03fSStefan Roese 
254*7b3af03fSStefan Roese 	printf("New factory-data values:\n");
255*7b3af03fSStefan Roese 	printf("wifiaddr=%pM\n", fd->wifi_mac);
256*7b3af03fSStefan Roese 	printf("ethaddr=%pM\n", fd->eth_mac);
257*7b3af03fSStefan Roese 
258*7b3af03fSStefan Roese 	/*
259*7b3af03fSStefan Roese 	 * We don't have the \0 char at the end, so we need to specify the
260*7b3af03fSStefan Roese 	 * length in the printf format instead
261*7b3af03fSStefan Roese 	 */
262*7b3af03fSStefan Roese 	printf("linuxmoduleid=%." __stringify(UUID_STR_LEN) "s\n", fd->ipr_id);
263*7b3af03fSStefan Roese 	printf("linuxmodulehqvid=%." __stringify(UUID_STR_LEN) "s\n",
264*7b3af03fSStefan Roese 	       fd->hqv_id);
265*7b3af03fSStefan Roese 	printf("linuxmoduleunielecid=%." __stringify(UUID_STR_LEN) "s\n",
266*7b3af03fSStefan Roese 	       fd->unielec_id);
267*7b3af03fSStefan Roese 
268*7b3af03fSStefan Roese 	fd->crc = crc32(0, (u8 *)&fd->magic, FACTORY_DATA_CRC_LEN);
269*7b3af03fSStefan Roese 
270*7b3af03fSStefan Roese 	ret = spi_flash_erase(sf, FACTORY_DATA_OFFS, FACTORY_DATA_SECT_SIZE);
271*7b3af03fSStefan Roese 	if (ret) {
272*7b3af03fSStefan Roese 		printf("F-Data:spi_flash_erase failed (%d)\n", ret);
273*7b3af03fSStefan Roese 		goto err_spi_flash;
274*7b3af03fSStefan Roese 	}
275*7b3af03fSStefan Roese 
276*7b3af03fSStefan Roese 	ret = spi_flash_write(sf, FACTORY_DATA_OFFS, FACTORY_DATA_SECT_SIZE,
277*7b3af03fSStefan Roese 			      buf);
278*7b3af03fSStefan Roese 	if (ret) {
279*7b3af03fSStefan Roese 		printf("F-Data:spi_flash_write failed (%d)\n", ret);
280*7b3af03fSStefan Roese 		goto err_spi_flash;
281*7b3af03fSStefan Roese 	}
282*7b3af03fSStefan Roese 
283*7b3af03fSStefan Roese 	printf("F-Data:factory-data values written to SPI NOR flash\n");
284*7b3af03fSStefan Roese 
285*7b3af03fSStefan Roese err_spi_flash:
286*7b3af03fSStefan Roese 	spi_flash_free(sf);
287*7b3af03fSStefan Roese 
288*7b3af03fSStefan Roese err_free:
289*7b3af03fSStefan Roese 	free(buf);
290*7b3af03fSStefan Roese 
291*7b3af03fSStefan Roese 	return ret;
292*7b3af03fSStefan Roese }
293*7b3af03fSStefan Roese 
294*7b3af03fSStefan Roese U_BOOT_CMD(
295*7b3af03fSStefan Roese 	fd_write,	1,	0,	do_fd_write,
296*7b3af03fSStefan Roese 	"Write test factory-data values to SPI NOR",
297*7b3af03fSStefan Roese 	"\n"
298*7b3af03fSStefan Roese );
299