[J-u-boot.git] / tools / imagetool.h

/* SPDX-License-Identifier: GPL-2.0+ */
/*
 * (C) Copyright 2013
 *
 * Written by Guilherme Maciel Ferreira <[email protected]>
 */

#ifndef _IMAGETOOL_H_
#define _IMAGETOOL_H_

#include "os_support.h"
#include <errno.h>
#include <fcntl.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <time.h>
#include <unistd.h>
#include <u-boot/sha1.h>

#include "fdt_host.h"

#define ARRAY_SIZE(x)		(sizeof(x) / sizeof((x)[0]))

#define __ALIGN_MASK(x, mask)	(((x) + (mask)) & ~(mask))
#define ALIGN(x, a)		__ALIGN_MASK((x), (typeof(x))(a) - 1)

#define IH_ARCH_DEFAULT		IH_ARCH_INVALID

/* Information about a file that needs to be placed into the FIT */
struct content_info {
	struct content_info *next;
	int type;		/* File type (IH_TYPE_...) */
	const char *fname;
};

/*
 * This structure defines all such variables those are initialized by
 * mkimage and dumpimage main core and need to be referred by image
 * type specific functions
 */
struct image_tool_params {
	int dflag;
	int eflag;
	int fflag;
	int iflag;
	int lflag;
	int pflag;
	int vflag;
	int xflag;
	int skipcpy;
	int os;
	int arch;
	int type;
	int comp;
	char *dtc;
	unsigned int addr;
	unsigned int ep;
	char *imagename;
	char *imagename2;
	char *datafile;
	char *imagefile;
	char *cmdname;
	const char *outfile;	/* Output filename */
	const char *keydir;	/* Directory holding private keys */
	const char *keydest;	/* Destination .dtb for public key */
	const char *keyfile;	/* Filename of private or public key */
	const char *comment;	/* Comment to add to signature node */
	int require_keys;	/* 1 to mark signing keys as 'required' */
	int file_size;		/* Total size of output file */
	int orig_file_size;	/* Original size for file before padding */
	bool auto_its;		/* Automatically create the .its file */
	int fit_image_type;	/* Image type to put into the FIT */
	char *fit_ramdisk;	/* Ramdisk file to include */
	struct content_info *content_head;	/* List of files to include */
	struct content_info *content_tail;
	bool external_data;	/* Store data outside the FIT */
	bool quiet;		/* Don't output text in normal operation */
	unsigned int external_offset;	/* Add padding to external data */
	int bl_len;		/* Block length in byte for external data */
	const char *engine_id;	/* Engine to use for signing */
	bool reset_timestamp;	/* Reset the timestamp on an existing image */
};

/*
 * image type specific variables and callback functions
 */
struct image_type_params {
	/* name is an identification tag string for added support */
	char *name;
	/*
	 * header size is local to the specific image type to be supported,
	 * mkimage core treats this as number of bytes
	 */
	uint32_t header_size;
	/* Image type header pointer */
	void *hdr;
	/*
	 * There are several arguments that are passed on the command line
	 * and are registered as flags in image_tool_params structure.
	 * This callback function can be used to check the passed arguments
	 * are in-lined with the image type to be supported
	 *
	 * Returns 1 if parameter check is successful
	 */
	int (*check_params) (struct image_tool_params *);
	/*
	 * This function is used by list command (i.e. mkimage -l <filename>)
	 * image type verification code must be put here
	 *
	 * Returns 0 if image header verification is successful
	 * otherwise, returns respective negative error codes
	 */
	int (*verify_header) (unsigned char *, int, struct image_tool_params *);
	/* Prints image information abstracting from image header */
	void (*print_header) (const void *);
	/*
	 * The header or image contents need to be set as per image type to
	 * be generated using this callback function.
	 * further output file post processing (for ex. checksum calculation,
	 * padding bytes etc..) can also be done in this callback function.
	 */
	void (*set_header) (void *, struct stat *, int,
					struct image_tool_params *);
	/*
	 * This function is used by the command to retrieve a component
	 * (sub-image) from the image (i.e. dumpimage -p <position>
	 * -o <component-outfile> <image>). Thus the code to extract a file
	 * from an image must be put here.
	 *
	 * Returns 0 if the file was successfully retrieved from the image,
	 * or a negative value on error.
	 */
	int (*extract_subimage)(void *, struct image_tool_params *);
	/*
	 * Some image generation support for ex (default image type) supports
	 * more than one type_ids, this callback function is used to check
	 * whether input (-T <image_type>) is supported by registered image
	 * generation/list low level code
	 */
	int (*check_image_type) (uint8_t);
	/* This callback function will be executed if fflag is defined */
	int (*fflag_handle) (struct image_tool_params *);
	/*
	 * This callback function will be executed for variable size record
	 * It is expected to build this header in memory and return its length
	 * and a pointer to it by using image_type_params.header_size and
	 * image_type_params.hdr. The return value shall indicate if an
	 * additional padding should be used when copying the data image
	 * by returning the padding length.
	 */
	int (*vrec_header) (struct image_tool_params *,
		struct image_type_params *);
};

/**
 * imagetool_get_type() - find the image type params for a given image type
 *
 * It scans all registers image type supports
 * checks the input type for each supported image type
 *
 * if successful,
 *     returns respective image_type_params pointer if success
 * if input type_id is not supported by any of image_type_support
 *     returns NULL
 */
struct image_type_params *imagetool_get_type(int type);

/*
 * imagetool_verify_print_header() - verifies the image header
 *
 * Scan registered image types and verify the image_header for each
 * supported image type. If verification is successful, this prints
 * the respective header.
 *
 * @return 0 on success, negative if input image format does not match with
 * any of supported image types
 */
int imagetool_verify_print_header(
	void *ptr,
	struct stat *sbuf,
	struct image_type_params *tparams,
	struct image_tool_params *params);

/*
 * imagetool_verify_print_header_by_type() - verifies the image header
 *
 * Verify the image_header for the image type given by tparams.
 * If verification is successful, this prints the respective header.
 * @ptr: pointer the the image header
 * @sbuf: stat information about the file pointed to by ptr
 * @tparams: image type parameters
 * @params: mkimage parameters
 *
 * @return 0 on success, negative if input image format does not match with
 * the given image type
 */
int imagetool_verify_print_header_by_type(
	void *ptr,
	struct stat *sbuf,
	struct image_type_params *tparams,
	struct image_tool_params *params);

/**
 * imagetool_save_subimage - store data into a file
 * @file_name: name of the destination file
 * @file_data: data to be written
 * @file_len: the amount of data to store
 *
 * imagetool_save_subimage() store file_len bytes of data pointed by file_data
 * into the file name by file_name.
 *
 * returns:
 *     zero in case of success or a negative value if fail.
 */
int imagetool_save_subimage(
	const char *file_name,
	ulong file_data,
	ulong file_len);

/**
 * imagetool_get_filesize() - Utility function to obtain the size of a file
 *
 * This function prints a message if an error occurs, showing the error that
 * was obtained.
 *
 * @params:	mkimage parameters
 * @fname:	filename to check
 * @return size of file, or -ve value on error
 */
int imagetool_get_filesize(struct image_tool_params *params, const char *fname);

/**
 * imagetool_get_source_date() - Get timestamp for build output.
 *
 * Gets a timestamp for embedding it in a build output. If set
 * SOURCE_DATE_EPOCH is used. Else the given fallback value is returned. Prints
 * an error message if SOURCE_DATE_EPOCH contains an invalid value and returns
 * 0.
 *
 * @cmdname:	command name
 * @fallback:	timestamp to use if SOURCE_DATE_EPOCH isn't set
 * @return timestamp based on SOURCE_DATE_EPOCH
 */
time_t imagetool_get_source_date(
	const char *cmdname,
	time_t fallback);

/*
 * There is a c file associated with supported image type low level code
 * for ex. default_image.c, fit_image.c
 */


void pbl_load_uboot(int fd, struct image_tool_params *mparams);
int zynqmpbif_copy_image(int fd, struct image_tool_params *mparams);
int imx8image_copy_image(int fd, struct image_tool_params *mparams);
int imx8mimage_copy_image(int fd, struct image_tool_params *mparams);
int rockchip_copy_image(int fd, struct image_tool_params *mparams);

#define ___cat(a, b) a ## b
#define __cat(a, b) ___cat(a, b)

/* we need some special handling for this host tool running eventually on
 * Darwin. The Mach-O section handling is a bit different than ELF section
 * handling. The differnces in detail are:
 *  a) we have segments which have sections
 *  b) we need a API call to get the respective section symbols */
#if defined(__MACH__)
#include <mach-o/getsect.h>

#define INIT_SECTION(name)  do {					\
		unsigned long name ## _len;				\
		char *__cat(pstart_, name) = getsectdata("__DATA",	\
			#name, &__cat(name, _len));			\
		char *__cat(pstop_, name) = __cat(pstart_, name) +	\
			__cat(name, _len);				\
		__cat(__start_, name) = (void *)__cat(pstart_, name);	\
		__cat(__stop_, name) = (void *)__cat(pstop_, name);	\
	} while (0)
#define SECTION(name)   __attribute__((section("__DATA, " #name)))

struct image_type_params **__start_image_type, **__stop_image_type;
#else
#define INIT_SECTION(name) /* no-op for ELF */
#define SECTION(name)   __attribute__((section(#name)))

/* We construct a table of pointers in an ELF section (pointers generally
 * go unpadded by gcc).  ld creates boundary syms for us. */
extern struct image_type_params *__start_image_type[], *__stop_image_type[];
#endif /* __MACH__ */

#if !defined(__used)
# if __GNUC__ == 3 && __GNUC_MINOR__ < 3
#  define __used			__attribute__((__unused__))
# else
#  define __used			__attribute__((__used__))
# endif
#endif

#define U_BOOT_IMAGE_TYPE( \
		_id, \
		_name, \
		_header_size, \
		_header, \
		_check_params, \
		_verify_header, \
		_print_header, \
		_set_header, \
		_extract_subimage, \
		_check_image_type, \
		_fflag_handle, \
		_vrec_header \
	) \
	static struct image_type_params __cat(image_type_, _id) = \
	{ \
		.name = _name, \
		.header_size = _header_size, \
		.hdr = _header, \
		.check_params = _check_params, \
		.verify_header = _verify_header, \
		.print_header = _print_header, \
		.set_header = _set_header, \
		.extract_subimage = _extract_subimage, \
		.check_image_type = _check_image_type, \
		.fflag_handle = _fflag_handle, \
		.vrec_header = _vrec_header \
	}; \
	static struct image_type_params *SECTION(image_type) __used \
		__cat(image_type_ptr_, _id) = &__cat(image_type_, _id)

#endif /* _IMAGETOOL_H_ */
Commit	Line	Data
83d290c5	1	/* SPDX-License-Identifier: GPL-2.0+ */
f86ed6a8 GMF	2	/*
	3	* (C) Copyright 2013
	4	*
	5	* Written by Guilherme Maciel Ferreira <[email protected]>
f86ed6a8 GMF	6	*/
	7
	8	#ifndef _IMAGETOOL_H_
	9	#define _IMAGETOOL_H_
	10
	11	#include "os_support.h"
	12	#include <errno.h>
	13	#include <fcntl.h>
d2bf1152	14	#include <stdbool.h>
f86ed6a8 GMF	15	#include <stdio.h>
	16	#include <stdlib.h>
	17	#include <string.h>
	18	#include <sys/stat.h>
26e355d1	19	#include <sys/types.h>
f86ed6a8 GMF	20	#include <time.h>
f86ed6a8 GMF	21	#include <unistd.h>
2b9912e6	22	#include <u-boot/sha1.h>
a93648d1	23
f86ed6a8 GMF	24	#include "fdt_host.h"
	25
	26	#define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
	27
c738adb8 KY	28	#define __ALIGN_MASK(x, mask) (((x) + (mask)) & ~(mask))
	29	#define ALIGN(x, a) __ALIGN_MASK((x), (typeof(x))(a) - 1)
	30
f86ed6a8 GMF	31	#define IH_ARCH_DEFAULT IH_ARCH_INVALID
f86ed6a8 GMF	32
fb4cce0f SG	33	/* Information about a file that needs to be placed into the FIT */
	34	struct content_info {
	35	struct content_info *next;
	36	int type; /* File type (IH_TYPE_...) */
	37	const char *fname;
	38	};
	39
f86ed6a8 GMF	40	/*
	41	* This structure defines all such variables those are initialized by
	42	* mkimage and dumpimage main core and need to be referred by image
	43	* type specific functions
	44	*/
	45	struct image_tool_params {
	46	int dflag;
	47	int eflag;
	48	int fflag;
a804b5ce	49	int iflag;
f86ed6a8	50	int lflag;
a804b5ce	51	int pflag;
f86ed6a8 GMF	52	int vflag;
	53	int xflag;
	54	int skipcpy;
	55	int os;
	56	int arch;
	57	int type;
	58	int comp;
	59	char *dtc;
	60	unsigned int addr;
	61	unsigned int ep;
	62	char *imagename;
	63	char *imagename2;
	64	char *datafile;
	65	char *imagefile;
	66	char *cmdname;
a804b5ce	67	const char outfile; / Output filename */
f86ed6a8 GMF	68	const char keydir; / Directory holding private keys */
f86ed6a8 GMF	69	const char keydest; / Destination .dtb for public key */
36bfcb62	70	const char keyfile; / Filename of private or public key */
f86ed6a8 GMF	71	const char comment; / Comment to add to signature node */
f86ed6a8 GMF	72	int require_keys; /* 1 to mark signing keys as 'required' */
92a655c3	73	int file_size; /* Total size of output file */
1b99e5bb	74	int orig_file_size; /* Original size for file before padding */
8e35bb07	75	bool auto_its; /* Automatically create the .its file */
d505a09c	76	int fit_image_type; /* Image type to put into the FIT */
0f7c6cdc	77	char fit_ramdisk; / Ramdisk file to include */
fb4cce0f SG	78	struct content_info content_head; / List of files to include */
fb4cce0f SG	79	struct content_info *content_tail;
722ebc8f	80	bool external_data; /* Store data outside the FIT */
bd6e1420	81	bool quiet; /* Don't output text in normal operation */
f8f9107d	82	unsigned int external_offset; /* Add padding to external data */
ebfe611b	83	int bl_len; /* Block length in byte for external data */
f1ca1fde	84	const char engine_id; / Engine to use for signing */
152b2462	85	bool reset_timestamp; /* Reset the timestamp on an existing image */
f86ed6a8 GMF	86	};
	87
	88	/*
	89	* image type specific variables and callback functions
	90	*/
	91	struct image_type_params {
	92	/* name is an identification tag string for added support */
	93	char *name;
	94	/*
	95	* header size is local to the specific image type to be supported,
	96	* mkimage core treats this as number of bytes
	97	*/
	98	uint32_t header_size;
	99	/* Image type header pointer */
	100	void *hdr;
	101	/*
	102	* There are several arguments that are passed on the command line
	103	* and are registered as flags in image_tool_params structure.
	104	* This callback function can be used to check the passed arguments
	105	* are in-lined with the image type to be supported
	106	*
	107	* Returns 1 if parameter check is successful
	108	*/
	109	int (check_params) (struct image_tool_params );
	110	/*
	111	* This function is used by list command (i.e. mkimage -l <filename>)
	112	* image type verification code must be put here
	113	*
	114	* Returns 0 if image header verification is successful
	115	* otherwise, returns respective negative error codes
	116	*/
	117	int (verify_header) (unsigned char , int, struct image_tool_params *);
	118	/* Prints image information abstracting from image header */
	119	void (print_header) (const void );
	120	/*
	121	* The header or image contents need to be set as per image type to
	122	* be generated using this callback function.
	123	* further output file post processing (for ex. checksum calculation,
	124	* padding bytes etc..) can also be done in this callback function.
	125	*/
	126	void (set_header) (void , struct stat *, int,
	127	struct image_tool_params *);
a804b5ce	128	/*
67f946cd	129	* This function is used by the command to retrieve a component
e157a111 TH	130	* (sub-image) from the image (i.e. dumpimage -p <position>
	131	* -o <component-outfile> <image>). Thus the code to extract a file
	132	* from an image must be put here.
a804b5ce GMF	133	*
	134	* Returns 0 if the file was successfully retrieved from the image,
	135	* or a negative value on error.
	136	*/
67f946cd	137	int (extract_subimage)(void , struct image_tool_params *);
f86ed6a8 GMF	138	/*
	139	* Some image generation support for ex (default image type) supports
	140	* more than one type_ids, this callback function is used to check
	141	* whether input (-T <image_type>) is supported by registered image
	142	* generation/list low level code
	143	*/
	144	int (*check_image_type) (uint8_t);
	145	/* This callback function will be executed if fflag is defined */
	146	int (fflag_handle) (struct image_tool_params );
	147	/*
	148	* This callback function will be executed for variable size record
	149	* It is expected to build this header in memory and return its length
	150	* and a pointer to it by using image_type_params.header_size and
	151	* image_type_params.hdr. The return value shall indicate if an
	152	* additional padding should be used when copying the data image
	153	* by returning the padding length.
	154	*/
	155	int (vrec_header) (struct image_tool_params ,
	156	struct image_type_params *);
f86ed6a8 GMF	157	};
f86ed6a8 GMF	158
0ca6691c GMF	159	/**
	160	* imagetool_get_type() - find the image type params for a given image type
	161	*
	162	* It scans all registers image type supports
	163	* checks the input type for each supported image type
	164	*
	165	* if successful,
	166	* returns respective image_type_params pointer if success
	167	* if input type_id is not supported by any of image_type_support
	168	* returns NULL
	169	*/
a93648d1	170	struct image_type_params *imagetool_get_type(int type);
0ca6691c GMF	171
	172	/*
	173	* imagetool_verify_print_header() - verifies the image header
	174	*
	175	* Scan registered image types and verify the image_header for each
	176	* supported image type. If verification is successful, this prints
	177	* the respective header.
	178	*
	179	* @return 0 on success, negative if input image format does not match with
	180	* any of supported image types
	181	*/
	182	int imagetool_verify_print_header(
	183	void *ptr,
	184	struct stat *sbuf,
d32aa3ca JH	185	struct image_type_params *tparams,
	186	struct image_tool_params *params);
	187
	188	/*
	189	* imagetool_verify_print_header_by_type() - verifies the image header
	190	*
	191	* Verify the image_header for the image type given by tparams.
	192	* If verification is successful, this prints the respective header.
	193	* @ptr: pointer the the image header
	194	* @sbuf: stat information about the file pointed to by ptr
	195	* @tparams: image type parameters
	196	* @params: mkimage parameters
	197	*
	198	* @return 0 on success, negative if input image format does not match with
	199	* the given image type
	200	*/
	201	int imagetool_verify_print_header_by_type(
	202	void *ptr,
	203	struct stat *sbuf,
0ca6691c GMF	204	struct image_type_params *tparams,
	205	struct image_tool_params *params);
	206
067d1560	207	/**
67f946cd	208	* imagetool_save_subimage - store data into a file
067d1560 GMF	209	* @file_name: name of the destination file
	210	* @file_data: data to be written
	211	* @file_len: the amount of data to store
	212	*
67f946cd	213	* imagetool_save_subimage() store file_len bytes of data pointed by file_data
067d1560 GMF	214	* into the file name by file_name.
	215	*
	216	* returns:
	217	* zero in case of success or a negative value if fail.
	218	*/
67f946cd	219	int imagetool_save_subimage(
067d1560 GMF	220	const char *file_name,
	221	ulong file_data,
	222	ulong file_len);
	223
3837ce65 SG	224	/**
	225	* imagetool_get_filesize() - Utility function to obtain the size of a file
	226	*
	227	* This function prints a message if an error occurs, showing the error that
	228	* was obtained.
	229	*
	230	* @params: mkimage parameters
	231	* @fname: filename to check
	232	* @return size of file, or -ve value on error
	233	*/
	234	int imagetool_get_filesize(struct image_tool_params params, const char fname);
	235
5847084f VC	236	/**
	237	* imagetool_get_source_date() - Get timestamp for build output.
	238	*
	239	* Gets a timestamp for embedding it in a build output. If set
	240	* SOURCE_DATE_EPOCH is used. Else the given fallback value is returned. Prints
	241	* an error message if SOURCE_DATE_EPOCH contains an invalid value and returns
	242	* 0.
	243	*
87925df2	244	* @cmdname: command name
5847084f VC	245	* @fallback: timestamp to use if SOURCE_DATE_EPOCH isn't set
	246	* @return timestamp based on SOURCE_DATE_EPOCH
	247	*/
	248	time_t imagetool_get_source_date(
87925df2	249	const char *cmdname,
5847084f VC	250	time_t fallback);
5847084f VC	251
f86ed6a8 GMF	252	/*
	253	* There is a c file associated with supported image type low level code
	254	* for ex. default_image.c, fit_image.c
f86ed6a8	255	*/
a93648d1	256
f86ed6a8 GMF	257
f86ed6a8 GMF	258	void pbl_load_uboot(int fd, struct image_tool_params *mparams);
6915dcf3	259	int zynqmpbif_copy_image(int fd, struct image_tool_params *mparams);
a2b96ece	260	int imx8image_copy_image(int fd, struct image_tool_params *mparams);
6609c266	261	int imx8mimage_copy_image(int fd, struct image_tool_params *mparams);
eea6cd8d	262	int rockchip_copy_image(int fd, struct image_tool_params *mparams);
f86ed6a8	263
1fddd7b6 AB	264	#define ___cat(a, b) a ## b
	265	#define __cat(a, b) ___cat(a, b)
	266
	267	/* we need some special handling for this host tool running eventually on
	268	* Darwin. The Mach-O section handling is a bit different than ELF section
	269	* handling. The differnces in detail are:
	270	* a) we have segments which have sections
	271	* b) we need a API call to get the respective section symbols */
	272	#if defined(__MACH__)
	273	#include <mach-o/getsect.h>
	274
	275	#define INIT_SECTION(name) do { \
	276	unsigned long name ## _len; \
3b142045	277	char *__cat(pstart_, name) = getsectdata("__DATA", \
1fddd7b6 AB	278	#name, &__cat(name, _len)); \
	279	char *__cat(pstop_, name) = __cat(pstart_, name) + \
	280	__cat(name, _len); \
	281	__cat(__start_, name) = (void *)__cat(pstart_, name); \
	282	__cat(__stop_, name) = (void *)__cat(pstop_, name); \
	283	} while (0)
3b142045	284	#define SECTION(name) __attribute__((section("__DATA, " #name)))
1fddd7b6 AB	285
	286	struct image_type_params __start_image_type, __stop_image_type;
	287	#else
	288	#define INIT_SECTION(name) /* no-op for ELF */
	289	#define SECTION(name) __attribute__((section(#name)))
	290
	291	/* We construct a table of pointers in an ELF section (pointers generally
	292	* go unpadded by gcc). ld creates boundary syms for us. */
	293	extern struct image_type_params __start_image_type[], __stop_image_type[];
	294	#endif /* __MACH__ */
	295
	296	#if !defined(__used)
	297	# if __GNUC__ == 3 && __GNUC_MINOR__ < 3
	298	# define __used __attribute__((__unused__))
	299	# else
	300	# define __used __attribute__((__used__))
	301	# endif
	302	#endif
	303
a93648d1 GMF	304	#define U_BOOT_IMAGE_TYPE( \
	305	_id, \
	306	_name, \
	307	_header_size, \
	308	_header, \
	309	_check_params, \
	310	_verify_header, \
	311	_print_header, \
	312	_set_header, \
67f946cd	313	_extract_subimage, \
a93648d1 GMF	314	_check_image_type, \
	315	_fflag_handle, \
	316	_vrec_header \
	317	) \
1fddd7b6 AB	318	static struct image_type_params __cat(image_type_, _id) = \
1fddd7b6 AB	319	{ \
a93648d1 GMF	320	.name = _name, \
	321	.header_size = _header_size, \
	322	.hdr = _header, \
	323	.check_params = _check_params, \
	324	.verify_header = _verify_header, \
	325	.print_header = _print_header, \
	326	.set_header = _set_header, \
67f946cd	327	.extract_subimage = _extract_subimage, \
a93648d1 GMF	328	.check_image_type = _check_image_type, \
	329	.fflag_handle = _fflag_handle, \
	330	.vrec_header = _vrec_header \
1fddd7b6 AB	331	}; \
	332	static struct image_type_params *SECTION(image_type) __used \
	333	__cat(image_type_ptr_, _id) = &__cat(image_type_, _id)
a93648d1	334
f86ed6a8	335	#endif /* _IMAGETOOL_H_ */