mirror of
				git://git.openwrt.org/openwrt/openwrt.git
				synced 2025-11-03 14:34:27 -05:00 
			
		
		
		
	Rebase local patches on top of quarterly timed release, allowing to drop numerous patches which have been accepted upstream since the release of U-Boot 2023.07.02. Signed-off-by: Daniel Golle <daniel@makrotopia.org>
		
			
				
	
	
		
			131 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Diff
		
	
	
	
	
	
			
		
		
	
	
			131 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Diff
		
	
	
	
	
	
--- a/cmd/bootm.c
 | 
						|
+++ b/cmd/bootm.c
 | 
						|
@@ -245,6 +245,67 @@ U_BOOT_CMD(
 | 
						|
 /* iminfo - print header info for a requested image */
 | 
						|
 /*******************************************************************/
 | 
						|
 #if defined(CONFIG_CMD_IMI)
 | 
						|
+#if defined(CONFIG_FIT)
 | 
						|
+#define SECTOR_SHIFT 9
 | 
						|
+static int image_totalsize(struct cmd_tbl *cmdtp, int flag, int argc,
 | 
						|
+			   char *const argv[], short int in_blocks)
 | 
						|
+{
 | 
						|
+	ulong addr;
 | 
						|
+	void *fit;
 | 
						|
+	int bsize, tsize;
 | 
						|
+	char buf[16];
 | 
						|
+
 | 
						|
+	if (argc >= 2)
 | 
						|
+		addr = simple_strtoul(argv[1], NULL, 16);
 | 
						|
+	else
 | 
						|
+		addr = image_load_addr;
 | 
						|
+
 | 
						|
+	fit = (void *)map_sysmem(addr, 0);
 | 
						|
+	tsize = fit_get_totalsize(fit);
 | 
						|
+	unmap_sysmem(fit);
 | 
						|
+	if (tsize == 0)
 | 
						|
+		return 1;
 | 
						|
+
 | 
						|
+	bsize = (tsize >> SECTOR_SHIFT) + ((tsize & ((1 << SECTOR_SHIFT) - 1))?1:0);
 | 
						|
+
 | 
						|
+	if (!in_blocks)
 | 
						|
+		snprintf(buf, sizeof(buf), "%x", tsize);
 | 
						|
+	else
 | 
						|
+		snprintf(buf, sizeof(buf), "%x", bsize);
 | 
						|
+
 | 
						|
+	if (argc >= 3)
 | 
						|
+		return env_set(argv[2], buf);
 | 
						|
+	else
 | 
						|
+		printf("%s\n", buf);
 | 
						|
+
 | 
						|
+	return 0;
 | 
						|
+}
 | 
						|
+
 | 
						|
+static int do_imsz(struct cmd_tbl *cmdtp, int flag, int argc,
 | 
						|
+		     char *const argv[])
 | 
						|
+{
 | 
						|
+	return image_totalsize(cmdtp, flag, argc, argv, 0);
 | 
						|
+}
 | 
						|
+
 | 
						|
+static int do_imszb(struct cmd_tbl *cmdtp, int flag, int argc,
 | 
						|
+		     char *const argv[])
 | 
						|
+{
 | 
						|
+	return image_totalsize(cmdtp, flag, argc, argv, 1);
 | 
						|
+}
 | 
						|
+
 | 
						|
+U_BOOT_CMD(
 | 
						|
+	imsz,	CONFIG_SYS_MAXARGS,	1,	do_imsz,
 | 
						|
+	"get image total size (in bytes)",
 | 
						|
+	"addr [maxhdrlen] [varname]\n"
 | 
						|
+);
 | 
						|
+
 | 
						|
+U_BOOT_CMD(
 | 
						|
+	imszb,	CONFIG_SYS_MAXARGS,	1,	do_imszb,
 | 
						|
+	"get image total size (in blocks)",
 | 
						|
+	"addr [maxhdrlen] [varname]\n"
 | 
						|
+);
 | 
						|
+
 | 
						|
+#endif
 | 
						|
 static int do_iminfo(struct cmd_tbl *cmdtp, int flag, int argc,
 | 
						|
 		     char *const argv[])
 | 
						|
 {
 | 
						|
--- a/boot/image-fit.c
 | 
						|
+++ b/boot/image-fit.c
 | 
						|
@@ -2051,6 +2051,47 @@ static const char *fit_get_image_type_pr
 | 
						|
 	return "unknown";
 | 
						|
 }
 | 
						|
 
 | 
						|
+size_t fit_get_totalsize(const void *fit)
 | 
						|
+{
 | 
						|
+	int ret, ndepth, noffset, images_noffset;
 | 
						|
+	size_t data_size, hdrsize, img_total, max_size = 0;
 | 
						|
+	const void *data;
 | 
						|
+
 | 
						|
+	ret = fdt_check_header(fit);
 | 
						|
+	if (ret) {
 | 
						|
+		debug("Wrong FIT format: not a flattened device tree (err=%d)\n",
 | 
						|
+			  ret);
 | 
						|
+		return 0;
 | 
						|
+	}
 | 
						|
+
 | 
						|
+	hdrsize = fdt_totalsize(fit);
 | 
						|
+
 | 
						|
+	/* take care of simple FIT with internal images */
 | 
						|
+	max_size = hdrsize;
 | 
						|
+
 | 
						|
+	images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
 | 
						|
+	if (images_noffset < 0)
 | 
						|
+		goto out;
 | 
						|
+
 | 
						|
+	for (ndepth = 0,
 | 
						|
+	     noffset = fdt_next_node(fit, images_noffset, &ndepth);
 | 
						|
+	     (noffset >= 0) && (ndepth > 0);
 | 
						|
+	     noffset = fdt_next_node(fit, noffset, &ndepth)) {
 | 
						|
+		if (ndepth == 1) {
 | 
						|
+			ret = fit_image_get_data_and_size(fit, noffset, &data, &data_size);
 | 
						|
+			if (ret)
 | 
						|
+				goto out;
 | 
						|
+
 | 
						|
+			img_total = data_size + (data - fit);
 | 
						|
+
 | 
						|
+			max_size = (max_size > img_total) ? max_size : img_total;
 | 
						|
+		}
 | 
						|
+	}
 | 
						|
+
 | 
						|
+out:
 | 
						|
+	return max_size;
 | 
						|
+}
 | 
						|
+
 | 
						|
 int fit_image_load(struct bootm_headers *images, ulong addr,
 | 
						|
 		   const char **fit_unamep, const char **fit_uname_configp,
 | 
						|
 		   int arch, int ph_type, int bootstage_id,
 | 
						|
--- a/include/image.h
 | 
						|
+++ b/include/image.h
 | 
						|
@@ -1049,6 +1049,7 @@ int fit_parse_subimage(const char *spec,
 | 
						|
 		ulong *addr, const char **image_name);
 | 
						|
 
 | 
						|
 int fit_get_subimage_count(const void *fit, int images_noffset);
 | 
						|
+size_t fit_get_totalsize(const void *fit);
 | 
						|
 void fit_print_contents(const void *fit);
 | 
						|
 void fit_image_print(const void *fit, int noffset, const char *p);
 | 
						|
 
 |