1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * (C) Copyright 2013
3*4882a593Smuzhiyun *
4*4882a593Smuzhiyun * Written by Guilherme Maciel Ferreira <guilherme.maciel.ferreira@gmail.com>
5*4882a593Smuzhiyun *
6*4882a593Smuzhiyun * SPDX-License-Identifier: GPL-2.0+
7*4882a593Smuzhiyun */
8*4882a593Smuzhiyun
9*4882a593Smuzhiyun #include "imagetool.h"
10*4882a593Smuzhiyun
11*4882a593Smuzhiyun #include <image.h>
12*4882a593Smuzhiyun
imagetool_get_type(int type)13*4882a593Smuzhiyun struct image_type_params *imagetool_get_type(int type)
14*4882a593Smuzhiyun {
15*4882a593Smuzhiyun struct image_type_params **curr;
16*4882a593Smuzhiyun INIT_SECTION(image_type);
17*4882a593Smuzhiyun
18*4882a593Smuzhiyun struct image_type_params **start = __start_image_type;
19*4882a593Smuzhiyun struct image_type_params **end = __stop_image_type;
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun for (curr = start; curr != end; curr++) {
22*4882a593Smuzhiyun if ((*curr)->check_image_type) {
23*4882a593Smuzhiyun if (!(*curr)->check_image_type(type))
24*4882a593Smuzhiyun return *curr;
25*4882a593Smuzhiyun }
26*4882a593Smuzhiyun }
27*4882a593Smuzhiyun return NULL;
28*4882a593Smuzhiyun }
29*4882a593Smuzhiyun
imagetool_verify_print_header(void * ptr,struct stat * sbuf,struct image_type_params * tparams,struct image_tool_params * params)30*4882a593Smuzhiyun int imagetool_verify_print_header(
31*4882a593Smuzhiyun void *ptr,
32*4882a593Smuzhiyun struct stat *sbuf,
33*4882a593Smuzhiyun struct image_type_params *tparams,
34*4882a593Smuzhiyun struct image_tool_params *params)
35*4882a593Smuzhiyun {
36*4882a593Smuzhiyun int retval = -1;
37*4882a593Smuzhiyun struct image_type_params **curr;
38*4882a593Smuzhiyun INIT_SECTION(image_type);
39*4882a593Smuzhiyun
40*4882a593Smuzhiyun struct image_type_params **start = __start_image_type;
41*4882a593Smuzhiyun struct image_type_params **end = __stop_image_type;
42*4882a593Smuzhiyun
43*4882a593Smuzhiyun for (curr = start; curr != end; curr++) {
44*4882a593Smuzhiyun if ((*curr)->verify_header) {
45*4882a593Smuzhiyun retval = (*curr)->verify_header((unsigned char *)ptr,
46*4882a593Smuzhiyun sbuf->st_size, params);
47*4882a593Smuzhiyun
48*4882a593Smuzhiyun if (retval == 0) {
49*4882a593Smuzhiyun /*
50*4882a593Smuzhiyun * Print the image information if verify is
51*4882a593Smuzhiyun * successful
52*4882a593Smuzhiyun */
53*4882a593Smuzhiyun if ((*curr)->print_header) {
54*4882a593Smuzhiyun if (!params->quiet)
55*4882a593Smuzhiyun (*curr)->print_header(ptr);
56*4882a593Smuzhiyun } else {
57*4882a593Smuzhiyun fprintf(stderr,
58*4882a593Smuzhiyun "%s: print_header undefined for %s\n",
59*4882a593Smuzhiyun params->cmdname, (*curr)->name);
60*4882a593Smuzhiyun }
61*4882a593Smuzhiyun break;
62*4882a593Smuzhiyun }
63*4882a593Smuzhiyun }
64*4882a593Smuzhiyun }
65*4882a593Smuzhiyun
66*4882a593Smuzhiyun return retval;
67*4882a593Smuzhiyun }
68*4882a593Smuzhiyun
imagetool_save_subimage(const char * file_name,ulong file_data,ulong file_len)69*4882a593Smuzhiyun int imagetool_save_subimage(
70*4882a593Smuzhiyun const char *file_name,
71*4882a593Smuzhiyun ulong file_data,
72*4882a593Smuzhiyun ulong file_len)
73*4882a593Smuzhiyun {
74*4882a593Smuzhiyun int dfd;
75*4882a593Smuzhiyun
76*4882a593Smuzhiyun dfd = open(file_name, O_RDWR | O_CREAT | O_TRUNC | O_BINARY,
77*4882a593Smuzhiyun S_IRUSR | S_IWUSR);
78*4882a593Smuzhiyun if (dfd < 0) {
79*4882a593Smuzhiyun fprintf(stderr, "Can't open \"%s\": %s\n",
80*4882a593Smuzhiyun file_name, strerror(errno));
81*4882a593Smuzhiyun return -1;
82*4882a593Smuzhiyun }
83*4882a593Smuzhiyun
84*4882a593Smuzhiyun if (write(dfd, (void *)file_data, file_len) != (ssize_t)file_len) {
85*4882a593Smuzhiyun fprintf(stderr, "Write error on \"%s\": %s\n",
86*4882a593Smuzhiyun file_name, strerror(errno));
87*4882a593Smuzhiyun close(dfd);
88*4882a593Smuzhiyun return -1;
89*4882a593Smuzhiyun }
90*4882a593Smuzhiyun
91*4882a593Smuzhiyun close(dfd);
92*4882a593Smuzhiyun
93*4882a593Smuzhiyun return 0;
94*4882a593Smuzhiyun }
95*4882a593Smuzhiyun
imagetool_get_filesize(struct image_tool_params * params,const char * fname)96*4882a593Smuzhiyun int imagetool_get_filesize(struct image_tool_params *params, const char *fname)
97*4882a593Smuzhiyun {
98*4882a593Smuzhiyun struct stat sbuf;
99*4882a593Smuzhiyun int fd;
100*4882a593Smuzhiyun
101*4882a593Smuzhiyun fd = open(fname, O_RDONLY | O_BINARY);
102*4882a593Smuzhiyun if (fd < 0) {
103*4882a593Smuzhiyun fprintf(stderr, "%s: Can't open %s: %s\n",
104*4882a593Smuzhiyun params->cmdname, fname, strerror(errno));
105*4882a593Smuzhiyun return -1;
106*4882a593Smuzhiyun }
107*4882a593Smuzhiyun
108*4882a593Smuzhiyun if (fstat(fd, &sbuf) < 0) {
109*4882a593Smuzhiyun fprintf(stderr, "%s: Can't stat %s: %s\n",
110*4882a593Smuzhiyun params->cmdname, fname, strerror(errno));
111*4882a593Smuzhiyun close(fd);
112*4882a593Smuzhiyun return -1;
113*4882a593Smuzhiyun }
114*4882a593Smuzhiyun close(fd);
115*4882a593Smuzhiyun
116*4882a593Smuzhiyun return sbuf.st_size;
117*4882a593Smuzhiyun }
118*4882a593Smuzhiyun
imagetool_get_source_date(struct image_tool_params * params,time_t fallback)119*4882a593Smuzhiyun time_t imagetool_get_source_date(
120*4882a593Smuzhiyun struct image_tool_params *params,
121*4882a593Smuzhiyun time_t fallback)
122*4882a593Smuzhiyun {
123*4882a593Smuzhiyun char *source_date_epoch = getenv("SOURCE_DATE_EPOCH");
124*4882a593Smuzhiyun
125*4882a593Smuzhiyun if (source_date_epoch == NULL)
126*4882a593Smuzhiyun return fallback;
127*4882a593Smuzhiyun
128*4882a593Smuzhiyun time_t time = (time_t) strtol(source_date_epoch, NULL, 10);
129*4882a593Smuzhiyun
130*4882a593Smuzhiyun if (gmtime(&time) == NULL) {
131*4882a593Smuzhiyun fprintf(stderr, "%s: SOURCE_DATE_EPOCH is not valid\n",
132*4882a593Smuzhiyun params->cmdname);
133*4882a593Smuzhiyun time = 0;
134*4882a593Smuzhiyun }
135*4882a593Smuzhiyun
136*4882a593Smuzhiyun return time;
137*4882a593Smuzhiyun }
138