[people/ms/u-boot.git] / cmd / armflash.c

/*
 * (C) Copyright 2015
 * Linus Walleij, Linaro
 *
 * Support for ARM Flash Partitions
 *
 * SPDX-License-Identifier:     GPL-2.0+
 */
#include <common.h>
#include <command.h>
#include <console.h>
#include <asm/io.h>

#define MAX_REGIONS 4
#define MAX_IMAGES 32

struct afs_region {
	u32 load_address;
	u32 size;
	u32 offset;
};

struct afs_image {
	flash_info_t *flinfo;
	const char *name;
	u32 version;
	u32 entrypoint;
	u32 attributes;
	u32 region_count;
	struct afs_region regions[MAX_REGIONS];
	ulong flash_mem_start;
	ulong flash_mem_end;
};

static struct afs_image afs_images[MAX_IMAGES];
static int num_afs_images;

static u32 compute_crc(ulong start, u32 len)
{
	u32 sum = 0;
	int i;

	if (len % 4 != 0) {
		printf("bad checksumming\n");
		return 0;
	}

	for (i = 0; i < len; i += 4) {
		u32 val;

		val = readl((void *)start + i);
		if (val > ~sum)
			sum++;
		sum += val;
	}
	return ~sum;
}

static void parse_bank(ulong bank)
{
	int i;
	ulong flstart, flend;
	flash_info_t *info;

	info = &flash_info[bank];
	if (info->flash_id != FLASH_MAN_CFI) {
		printf("Bank %lu: missing or unknown FLASH type\n", bank);
		return;
	}
	if (!info->sector_count) {
		printf("Bank %lu: no FLASH sectors\n", bank);
		return;
	}

	flstart = info->start[0];
	flend = flstart + info->size;

	for (i = 0; i < info->sector_count; ++i) {
		ulong secend;
		u32 foot1, foot2;

		if (ctrlc())
			break;

		if (i == info->sector_count-1)
			secend = flend;
		else
			secend = info->start[i+1];

		/* Check for v1 header */
		foot1 = readl((void *)secend - 0x0c);
		if (foot1 == 0xA0FFFF9FU) {
			struct afs_image *afi = &afs_images[num_afs_images];
			ulong imginfo;

			afi->flinfo = info;
			afi->version = 1;
			afi->flash_mem_start = readl((void *)secend - 0x10);
			afi->flash_mem_end = readl((void *)secend - 0x14);
			afi->attributes = readl((void *)secend - 0x08);
			/* Adjust to even address */
			imginfo = afi->flash_mem_end + afi->flash_mem_end % 4;
			/* Record as a single region */
			afi->region_count = 1;
			afi->regions[0].offset = readl((void *)imginfo + 0x04);
			afi->regions[0].load_address =
				readl((void *)imginfo + 0x08);
			afi->regions[0].size = readl((void *)imginfo + 0x0C);
			afi->entrypoint = readl((void *)imginfo + 0x10);
			afi->name = (const char *)imginfo + 0x14;
			num_afs_images++;
		}

		/* Check for v2 header */
		foot1 = readl((void *)secend - 0x04);
		foot2 = readl((void *)secend - 0x08);
		/* This makes up the string "HSLFTOOF" flash footer */
		if (foot1 == 0x464F4F54U && foot2 == 0x464C5348U) {
			struct afs_image *afi = &afs_images[num_afs_images];
			ulong imginfo;
			u32 block_start, block_end;
			int j;

			afi->flinfo = info;
			afi->version = readl((void *)secend - 0x0c);
			imginfo = secend - 0x30 - readl((void *)secend - 0x10);
			afi->name = (const char *)secend - 0x30;

			afi->entrypoint = readl((void *)imginfo+0x08);
			afi->attributes = readl((void *)imginfo+0x0c);
			afi->region_count = readl((void *)imginfo+0x10);
			block_start = readl((void *)imginfo+0x54);
			block_end = readl((void *)imginfo+0x58);
			afi->flash_mem_start = afi->flinfo->start[block_start];
			afi->flash_mem_end = afi->flinfo->start[block_end];

			/*
			 * Check footer CRC, the algorithm saves the inverse
			 * checksum as part of the summed words, and thus
			 * the result should be zero.
			 */
			if (compute_crc(imginfo + 8, 0x88) != 0) {
				printf("BAD CRC on ARM image info\n");
				printf("(continuing anyway)\n");
			}

			/* Parse regions */
			for (j = 0; j < afi->region_count; j++) {
				afi->regions[j].load_address =
					readl((void *)imginfo+0x14 + j*0x10);
				afi->regions[j].size =
					readl((void *)imginfo+0x18 + j*0x10);
				afi->regions[j].offset =
					readl((void *)imginfo+0x1c + j*0x10);
				/*
				 * At offset 0x20 + j*0x10 there is a region
				 * checksum which seems to be the running
				 * sum + 3, however since we anyway checksum
				 * the entire footer this is skipped over for
				 * checking here.
				 */
			}
			num_afs_images++;
		}
	}
}

static void parse_flash(void)
{
	ulong bank;

	/* We have already parsed the images in flash */
	if (num_afs_images > 0)
		return;
	for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank)
		parse_bank(bank);
}

static int load_image(const char * const name, const ulong address)
{
	struct afs_image *afi = NULL;
	int i;

	parse_flash();
	for (i = 0; i < num_afs_images; i++) {
		struct afs_image *tmp = &afs_images[i];

		if (!strcmp(tmp->name, name)) {
			afi = tmp;
			break;
		}
	}
	if (!afi) {
		printf("image \"%s\" not found in flash\n", name);
		return CMD_RET_FAILURE;
	}

	for (i = 0; i < afi->region_count; i++) {
		ulong from, to;

		from = afi->flash_mem_start + afi->regions[i].offset;
		if (address) {
			to = address;
		} else if (afi->regions[i].load_address) {
			to = afi->regions[i].load_address;
		} else {
			printf("no valid load address\n");
			return CMD_RET_FAILURE;
		}

		memcpy((void *)to, (void *)from, afi->regions[i].size);

		printf("loaded region %d from %08lX to %08lX, %08X bytes\n",
		       i,
		       from,
		       to,
		       afi->regions[i].size);
	}
	return CMD_RET_SUCCESS;
}

static void print_images(void)
{
	int i;

	parse_flash();
	for (i = 0; i < num_afs_images; i++) {
		struct afs_image *afi = &afs_images[i];
		int j;

		printf("Image: \"%s\" (v%d):\n", afi->name, afi->version);
		printf("    Entry point: 0x%08X\n", afi->entrypoint);
		printf("    Attributes: 0x%08X: ", afi->attributes);
		if (afi->attributes == 0x01)
			printf("ARM executable");
		if (afi->attributes == 0x08)
			printf("ARM backup");
		printf("\n");
		printf("    Flash mem start: 0x%08lX\n",
		       afi->flash_mem_start);
		printf("    Flash mem end: 0x%08lX\n",
		       afi->flash_mem_end);
		for (j = 0; j < afi->region_count; j++) {
			printf("    region %d\n"
			       "        load address: %08X\n"
			       "        size: %08X\n"
			       "        offset: %08X\n",
			       j,
			       afi->regions[j].load_address,
			       afi->regions[j].size,
			       afi->regions[j].offset);
		}
	}
}

static int exists(const char * const name)
{
	int i;

	parse_flash();
	for (i = 0; i < num_afs_images; i++) {
		struct afs_image *afi = &afs_images[i];

		if (strcmp(afi->name, name) == 0)
			return CMD_RET_SUCCESS;
	}
	return CMD_RET_FAILURE;
}

static int do_afs(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	int ret = CMD_RET_SUCCESS;

	if (argc == 1) {
		print_images();
	} else if (argc == 3 && !strcmp(argv[1], "exists")) {
		ret = exists(argv[2]);
	} else if (argc == 3 && !strcmp(argv[1], "load")) {
		ret = load_image(argv[2], 0x0);
	} else if (argc == 4 && !strcmp(argv[1], "load")) {
		ulong load_addr;

		load_addr = simple_strtoul(argv[3], NULL, 16);
		ret = load_image(argv[2], load_addr);
	} else {
		return CMD_RET_USAGE;
	}

	return ret;
}

U_BOOT_CMD(afs, 4, 0, do_afs, "show AFS partitions",
	   "no arguments\n"
	   "    - list images in flash\n"
	   "exists <image>\n"
	   "    - returns 1 if an image exists, else 0\n"
	   "load <image>\n"
	   "    - load an image to the location indicated in the header\n"
	   "load <image> 0x<address>\n"
	   "    - load an image to the location specified\n");
Commit	Line	Data
4bb66506 LW	1	/*
	2	* (C) Copyright 2015
	3	* Linus Walleij, Linaro
	4	*
	5	* Support for ARM Flash Partitions
	6	*
	7	* SPDX-License-Identifier: GPL-2.0+
	8	*/
	9	#include <common.h>
	10	#include <command.h>
24b852a7	11	#include <console.h>
4bb66506 LW	12	#include <asm/io.h>
	13
	14	#define MAX_REGIONS 4
	15	#define MAX_IMAGES 32
	16
	17	struct afs_region {
	18	u32 load_address;
	19	u32 size;
	20	u32 offset;
	21	};
	22
	23	struct afs_image {
	24	flash_info_t *flinfo;
	25	const char *name;
	26	u32 version;
	27	u32 entrypoint;
	28	u32 attributes;
	29	u32 region_count;
	30	struct afs_region regions[MAX_REGIONS];
	31	ulong flash_mem_start;
	32	ulong flash_mem_end;
	33	};
	34
	35	static struct afs_image afs_images[MAX_IMAGES];
	36	static int num_afs_images;
	37
	38	static u32 compute_crc(ulong start, u32 len)
	39	{
	40	u32 sum = 0;
	41	int i;
	42
	43	if (len % 4 != 0) {
	44	printf("bad checksumming\n");
	45	return 0;
	46	}
	47
	48	for (i = 0; i < len; i += 4) {
	49	u32 val;
	50
	51	val = readl((void *)start + i);
	52	if (val > ~sum)
	53	sum++;
	54	sum += val;
	55	}
	56	return ~sum;
	57	}
	58
	59	static void parse_bank(ulong bank)
	60	{
	61	int i;
	62	ulong flstart, flend;
	63	flash_info_t *info;
	64
	65	info = &flash_info[bank];
	66	if (info->flash_id != FLASH_MAN_CFI) {
	67	printf("Bank %lu: missing or unknown FLASH type\n", bank);
	68	return;
	69	}
	70	if (!info->sector_count) {
	71	printf("Bank %lu: no FLASH sectors\n", bank);
	72	return;
	73	}
	74
	75	flstart = info->start[0];
76	flend = flstart + info->size;
77
78	for (i = 0; i < info->sector_count; ++i) {
79	ulong secend;
80	u32 foot1, foot2;
81
82	if (ctrlc())
83	break;
84
85	if (i == info->sector_count-1)
86	secend = flend;
87	else
88	secend = info->start[i+1];
89
90	/* Check for v1 header */
91	foot1 = readl((void *)secend - 0x0c);
92	if (foot1 == 0xA0FFFF9FU) {
93	struct afs_image *afi = &afs_images[num_afs_images];
94	ulong imginfo;
95
96	afi->flinfo = info;
97	afi->version = 1;
98	afi->flash_mem_start = readl((void *)secend - 0x10);
99	afi->flash_mem_end = readl((void *)secend - 0x14);
100	afi->attributes = readl((void *)secend - 0x08);
101	/* Adjust to even address */
102	imginfo = afi->flash_mem_end + afi->flash_mem_end % 4;
103	/* Record as a single region */
104	afi->region_count = 1;
105	afi->regions[0].offset = readl((void *)imginfo + 0x04);
106	afi->regions[0].load_address =
107	readl((void *)imginfo + 0x08);
108	afi->regions[0].size = readl((void *)imginfo + 0x0C);
109	afi->entrypoint = readl((void *)imginfo + 0x10);
110	afi->name = (const char *)imginfo + 0x14;
111	num_afs_images++;
112	}
113
114	/* Check for v2 header */
115	foot1 = readl((void *)secend - 0x04);
116	foot2 = readl((void *)secend - 0x08);
117	/* This makes up the string "HSLFTOOF" flash footer */
118	if (foot1 == 0x464F4F54U && foot2 == 0x464C5348U) {
119	struct afs_image *afi = &afs_images[num_afs_images];
120	ulong imginfo;
121	u32 block_start, block_end;
122	int j;
123
124	afi->flinfo = info;
125	afi->version = readl((void *)secend - 0x0c);
126	imginfo = secend - 0x30 - readl((void *)secend - 0x10);
127	afi->name = (const char *)secend - 0x30;
128
129	afi->entrypoint = readl((void *)imginfo+0x08);
130	afi->attributes = readl((void *)imginfo+0x0c);
131	afi->region_count = readl((void *)imginfo+0x10);
132	block_start = readl((void *)imginfo+0x54);
133	block_end = readl((void *)imginfo+0x58);
134	afi->flash_mem_start = afi->flinfo->start[block_start];
135	afi->flash_mem_end = afi->flinfo->start[block_end];
136
137	/*
138	* Check footer CRC, the algorithm saves the inverse
139	* checksum as part of the summed words, and thus
140	* the result should be zero.
141	*/
142	if (compute_crc(imginfo + 8, 0x88) != 0) {
143	printf("BAD CRC on ARM image info\n");
144	printf("(continuing anyway)\n");
145	}
146
147	/* Parse regions */
148	for (j = 0; j < afi->region_count; j++) {
149	afi->regions[j].load_address =
150	readl((void )imginfo+0x14 + j0x10);
151	afi->regions[j].size =
152	readl((void )imginfo+0x18 + j0x10);
153	afi->regions[j].offset =
154	readl((void )imginfo+0x1c + j0x10);
155	/*
156	* At offset 0x20 + j*0x10 there is a region
157	* checksum which seems to be the running
158	* sum + 3, however since we anyway checksum
159	* the entire footer this is skipped over for
160	* checking here.
161	*/
162	}
163	num_afs_images++;
164	}
165	}
166	}
167
168	static void parse_flash(void)
169	{
170	ulong bank;
171
172	/* We have already parsed the images in flash */
173	if (num_afs_images > 0)
174	return;
175	for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank)
176	parse_bank(bank);
177	}
178
6607d397	179	static int load_image(const char * const name, const ulong address)
4bb66506 LW	180	{
	181	struct afs_image *afi = NULL;
	182	int i;
	183
	184	parse_flash();
	185	for (i = 0; i < num_afs_images; i++) {
	186	struct afs_image *tmp = &afs_images[i];
	187
	188	if (!strcmp(tmp->name, name)) {
	189	afi = tmp;
	190	break;
	191	}
	192	}
	193	if (!afi) {
	194	printf("image \"%s\" not found in flash\n", name);
6607d397	195	return CMD_RET_FAILURE;
4bb66506 LW	196	}
	197
	198	for (i = 0; i < afi->region_count; i++) {
	199	ulong from, to;
	200
	201	from = afi->flash_mem_start + afi->regions[i].offset;
	202	if (address) {
	203	to = address;
	204	} else if (afi->regions[i].load_address) {
	205	to = afi->regions[i].load_address;
	206	} else {
	207	printf("no valid load address\n");
6607d397	208	return CMD_RET_FAILURE;
4bb66506 LW	209	}
	210
	211	memcpy((void )to, (void )from, afi->regions[i].size);
	212
	213	printf("loaded region %d from %08lX to %08lX, %08X bytes\n",
	214	i,
	215	from,
	216	to,
	217	afi->regions[i].size);
	218	}
6607d397	219	return CMD_RET_SUCCESS;
4bb66506 LW	220	}
	221
	222	static void print_images(void)
	223	{
	224	int i;
	225
	226	parse_flash();
	227	for (i = 0; i < num_afs_images; i++) {
	228	struct afs_image *afi = &afs_images[i];
	229	int j;
	230
	231	printf("Image: \"%s\" (v%d):\n", afi->name, afi->version);
	232	printf(" Entry point: 0x%08X\n", afi->entrypoint);
	233	printf(" Attributes: 0x%08X: ", afi->attributes);
	234	if (afi->attributes == 0x01)
	235	printf("ARM executable");
	236	if (afi->attributes == 0x08)
	237	printf("ARM backup");
	238	printf("\n");
	239	printf(" Flash mem start: 0x%08lX\n",
	240	afi->flash_mem_start);
	241	printf(" Flash mem end: 0x%08lX\n",
	242	afi->flash_mem_end);
	243	for (j = 0; j < afi->region_count; j++) {
	244	printf(" region %d\n"
	245	" load address: %08X\n"
	246	" size: %08X\n"
	247	" offset: %08X\n",
	248	j,
	249	afi->regions[j].load_address,
	250	afi->regions[j].size,
	251	afi->regions[j].offset);
	252	}
	253	}
	254	}
	255
1a9717fb RH	256	static int exists(const char * const name)
	257	{
	258	int i;
	259
	260	parse_flash();
	261	for (i = 0; i < num_afs_images; i++) {
	262	struct afs_image *afi = &afs_images[i];
	263
	264	if (strcmp(afi->name, name) == 0)
	265	return CMD_RET_SUCCESS;
	266	}
	267	return CMD_RET_FAILURE;
	268	}
	269
4bb66506 LW	270	static int do_afs(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
4bb66506 LW	271	{
1a9717fb RH	272	int ret = CMD_RET_SUCCESS;
1a9717fb RH	273
4bb66506 LW	274	if (argc == 1) {
4bb66506 LW	275	print_images();
1a9717fb RH	276	} else if (argc == 3 && !strcmp(argv[1], "exists")) {
1a9717fb RH	277	ret = exists(argv[2]);
4bb66506	278	} else if (argc == 3 && !strcmp(argv[1], "load")) {
6607d397	279	ret = load_image(argv[2], 0x0);
4bb66506 LW	280	} else if (argc == 4 && !strcmp(argv[1], "load")) {
	281	ulong load_addr;
	282
	283	load_addr = simple_strtoul(argv[3], NULL, 16);
6607d397	284	ret = load_image(argv[2], load_addr);
4bb66506 LW	285	} else {
	286	return CMD_RET_USAGE;
	287	}
	288
1a9717fb	289	return ret;
4bb66506 LW	290	}
	291
	292	U_BOOT_CMD(afs, 4, 0, do_afs, "show AFS partitions",
	293	"no arguments\n"
	294	" - list images in flash\n"
1a9717fb RH	295	"exists <image>\n"
1a9717fb RH	296	" - returns 1 if an image exists, else 0\n"
4bb66506 LW	297	"load <image>\n"
	298	" - load an image to the location indicated in the header\n"
	299	"load <image> 0x<address>\n"
	300	" - load an image to the location specified\n");