1// SPDX-License-Identifier: GPL-2.0+
2/*
3 * (C) Copyright 2013
4 *
5 * Written by Guilherme Maciel Ferreira <guilherme.maciel.ferreira@gmail.com>
6 */
7
8#include "imagetool.h"
9
10#include <image.h>
11
12struct image_type_params *imagetool_get_type(int type)
13{
14	struct image_type_params **curr;
15	INIT_SECTION(image_type);
16
17	struct image_type_params **start = __start_image_type;
18	struct image_type_params **end = __stop_image_type;
19
20	for (curr = start; curr != end; curr++) {
21		if ((*curr)->check_image_type) {
22			if (!(*curr)->check_image_type(type))
23				return *curr;
24		}
25	}
26	return NULL;
27}
28
29static int imagetool_verify_print_header_by_type(
30	void *ptr,
31	struct stat *sbuf,
32	struct image_type_params *tparams,
33	struct image_tool_params *params);
34
35int imagetool_verify_print_header(
36	void *ptr,
37	struct stat *sbuf,
38	struct image_type_params *tparams,
39	struct image_tool_params *params)
40{
41	int retval = -1;
42	struct image_type_params **curr;
43	INIT_SECTION(image_type);
44
45	struct image_type_params **start = __start_image_type;
46	struct image_type_params **end = __stop_image_type;
47
48	if (tparams)
49		return imagetool_verify_print_header_by_type(ptr, sbuf, tparams, params);
50
51	for (curr = start; curr != end; curr++) {
52		/*
53		 * Basically every data file can be guessed / verified as gpimage,
54		 * so skip autodetection of data file as gpimage as it does not work.
55		 */
56		if ((*curr)->check_image_type && (*curr)->check_image_type(IH_TYPE_GPIMAGE) == 0)
57			continue;
58		if ((*curr)->verify_header) {
59			retval = (*curr)->verify_header((unsigned char *)ptr,
60						     sbuf->st_size, params);
61
62			if (retval == 0) {
63				/*
64				 * Print the image information if verify is
65				 * successful
66				 */
67				if ((*curr)->print_header) {
68					if (!params->quiet)
69						(*curr)->print_header(ptr, params);
70				} else {
71					fprintf(stderr,
72						"%s: print_header undefined for %s\n",
73						params->cmdname, (*curr)->name);
74				}
75				break;
76			}
77		}
78	}
79
80	if (retval != 0) {
81		fprintf(stderr, "%s: cannot detect image type\n",
82			params->cmdname);
83	}
84
85	return retval;
86}
87
88static int imagetool_verify_print_header_by_type(
89	void *ptr,
90	struct stat *sbuf,
91	struct image_type_params *tparams,
92	struct image_tool_params *params)
93{
94	int retval = -1;
95
96	if (tparams->verify_header) {
97		retval = tparams->verify_header((unsigned char *)ptr,
98						sbuf->st_size, params);
99
100		if (retval == 0) {
101			/*
102			 * Print the image information if verify is successful
103			 */
104			if (tparams->print_header) {
105				if (!params->quiet)
106					tparams->print_header(ptr, params);
107			} else {
108				fprintf(stderr,
109					"%s: print_header undefined for %s\n",
110					params->cmdname, tparams->name);
111			}
112		} else {
113			fprintf(stderr,
114				"%s: verify_header failed for %s with exit code %d\n",
115				params->cmdname, tparams->name, retval);
116		}
117
118	} else {
119		fprintf(stderr, "%s: verify_header undefined for %s\n",
120			params->cmdname, tparams->name);
121	}
122
123	return retval;
124}
125
126int imagetool_save_subimage(
127	const char *file_name,
128	ulong file_data,
129	ulong file_len)
130{
131	int dfd;
132
133	dfd = open(file_name, O_RDWR | O_CREAT | O_TRUNC | O_BINARY,
134		   S_IRUSR | S_IWUSR);
135	if (dfd < 0) {
136		fprintf(stderr, "Can't open \"%s\": %s\n",
137			file_name, strerror(errno));
138		return -1;
139	}
140
141	if (write(dfd, (void *)file_data, file_len) != (ssize_t)file_len) {
142		fprintf(stderr, "Write error on \"%s\": %s\n",
143			file_name, strerror(errno));
144		close(dfd);
145		return -1;
146	}
147
148	close(dfd);
149
150	return 0;
151}
152
153int imagetool_get_filesize(struct image_tool_params *params, const char *fname)
154{
155	struct stat sbuf;
156	int fd;
157
158	fd = open(fname, O_RDONLY | O_BINARY);
159	if (fd < 0) {
160		fprintf(stderr, "%s: Can't open %s: %s\n",
161			params->cmdname, fname, strerror(errno));
162		return -1;
163	}
164
165	if (fstat(fd, &sbuf) < 0) {
166		fprintf(stderr, "%s: Can't stat %s: %s\n",
167			params->cmdname, fname, strerror(errno));
168		close(fd);
169		return -1;
170	}
171	close(fd);
172
173	return sbuf.st_size;
174}
175
176time_t imagetool_get_source_date(
177	 const char *cmdname,
178	 time_t fallback)
179{
180	char *source_date_epoch = getenv("SOURCE_DATE_EPOCH");
181
182	if (source_date_epoch == NULL)
183		return fallback;
184
185	time_t time = (time_t) strtol(source_date_epoch, NULL, 10);
186
187	if (gmtime(&time) == NULL) {
188		fprintf(stderr, "%s: SOURCE_DATE_EPOCH is not valid\n",
189			cmdname);
190		time = 0;
191	}
192
193	return time;
194}
195