]>
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 <guilherme.maciel.ferreira@gmail.com> | |
f86ed6a8 GMF |
6 | */ |
7 | ||
8 | #include "imagetool.h" | |
9 | ||
0ca6691c GMF |
10 | #include <image.h> |
11 | ||
a93648d1 | 12 | struct image_type_params *imagetool_get_type(int type) |
0ca6691c | 13 | { |
1fddd7b6 AB |
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; | |
0ca6691c | 19 | |
a93648d1 | 20 | for (curr = start; curr != end; curr++) { |
1fddd7b6 AB |
21 | if ((*curr)->check_image_type) { |
22 | if (!(*curr)->check_image_type(type)) | |
23 | return *curr; | |
0ca6691c GMF |
24 | } |
25 | } | |
26 | return NULL; | |
27 | } | |
28 | ||
11f29d44 T |
29 | static 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 | ||
0ca6691c GMF |
35 | int 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; | |
1fddd7b6 AB |
42 | struct image_type_params **curr; |
43 | INIT_SECTION(image_type); | |
0ca6691c | 44 | |
1fddd7b6 AB |
45 | struct image_type_params **start = __start_image_type; |
46 | struct image_type_params **end = __stop_image_type; | |
a93648d1 | 47 | |
11f29d44 T |
48 | if (tparams) |
49 | return imagetool_verify_print_header_by_type(ptr, sbuf, tparams, params); | |
50 | ||
a93648d1 | 51 | for (curr = start; curr != end; curr++) { |
1fddd7b6 AB |
52 | if ((*curr)->verify_header) { |
53 | retval = (*curr)->verify_header((unsigned char *)ptr, | |
0ca6691c GMF |
54 | sbuf->st_size, params); |
55 | ||
56 | if (retval == 0) { | |
57 | /* | |
d32aa3ca | 58 | * Print the image information if verify is |
0ca6691c GMF |
59 | * successful |
60 | */ | |
1fddd7b6 | 61 | if ((*curr)->print_header) { |
bd6e1420 SG |
62 | if (!params->quiet) |
63 | (*curr)->print_header(ptr); | |
0ca6691c GMF |
64 | } else { |
65 | fprintf(stderr, | |
66 | "%s: print_header undefined for %s\n", | |
1fddd7b6 | 67 | params->cmdname, (*curr)->name); |
0ca6691c GMF |
68 | } |
69 | break; | |
70 | } | |
71 | } | |
72 | } | |
73 | ||
74 | return retval; | |
75 | } | |
067d1560 | 76 | |
11f29d44 | 77 | static int imagetool_verify_print_header_by_type( |
d32aa3ca JH |
78 | void *ptr, |
79 | struct stat *sbuf, | |
80 | struct image_type_params *tparams, | |
81 | struct image_tool_params *params) | |
82 | { | |
30705cd8 NH |
83 | int retval = -1; |
84 | ||
85 | if (tparams->verify_header) { | |
86 | retval = tparams->verify_header((unsigned char *)ptr, | |
87 | sbuf->st_size, params); | |
88 | ||
89 | if (retval == 0) { | |
90 | /* | |
91 | * Print the image information if verify is successful | |
92 | */ | |
93 | if (tparams->print_header) { | |
94 | if (!params->quiet) | |
95 | tparams->print_header(ptr); | |
96 | } else { | |
97 | fprintf(stderr, | |
98 | "%s: print_header undefined for %s\n", | |
99 | params->cmdname, tparams->name); | |
100 | } | |
d32aa3ca JH |
101 | } else { |
102 | fprintf(stderr, | |
30705cd8 NH |
103 | "%s: verify_header failed for %s with exit code %d\n", |
104 | params->cmdname, tparams->name, retval); | |
d32aa3ca | 105 | } |
30705cd8 | 106 | |
d32aa3ca | 107 | } else { |
30705cd8 NH |
108 | fprintf(stderr, "%s: print_header undefined for %s\n", |
109 | params->cmdname, tparams->name); | |
d32aa3ca JH |
110 | } |
111 | ||
112 | return retval; | |
113 | } | |
114 | ||
67f946cd | 115 | int imagetool_save_subimage( |
067d1560 GMF |
116 | const char *file_name, |
117 | ulong file_data, | |
118 | ulong file_len) | |
119 | { | |
120 | int dfd; | |
121 | ||
122 | dfd = open(file_name, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, | |
123 | S_IRUSR | S_IWUSR); | |
124 | if (dfd < 0) { | |
125 | fprintf(stderr, "Can't open \"%s\": %s\n", | |
126 | file_name, strerror(errno)); | |
127 | return -1; | |
128 | } | |
129 | ||
130 | if (write(dfd, (void *)file_data, file_len) != (ssize_t)file_len) { | |
131 | fprintf(stderr, "Write error on \"%s\": %s\n", | |
132 | file_name, strerror(errno)); | |
133 | close(dfd); | |
134 | return -1; | |
135 | } | |
136 | ||
137 | close(dfd); | |
138 | ||
139 | return 0; | |
140 | } | |
3837ce65 SG |
141 | |
142 | int imagetool_get_filesize(struct image_tool_params *params, const char *fname) | |
143 | { | |
144 | struct stat sbuf; | |
145 | int fd; | |
146 | ||
147 | fd = open(fname, O_RDONLY | O_BINARY); | |
148 | if (fd < 0) { | |
149 | fprintf(stderr, "%s: Can't open %s: %s\n", | |
150 | params->cmdname, fname, strerror(errno)); | |
151 | return -1; | |
152 | } | |
153 | ||
154 | if (fstat(fd, &sbuf) < 0) { | |
155 | fprintf(stderr, "%s: Can't stat %s: %s\n", | |
156 | params->cmdname, fname, strerror(errno)); | |
b12a81c4 | 157 | close(fd); |
3837ce65 SG |
158 | return -1; |
159 | } | |
160 | close(fd); | |
161 | ||
162 | return sbuf.st_size; | |
163 | } | |
5847084f VC |
164 | |
165 | time_t imagetool_get_source_date( | |
87925df2 | 166 | const char *cmdname, |
5847084f VC |
167 | time_t fallback) |
168 | { | |
169 | char *source_date_epoch = getenv("SOURCE_DATE_EPOCH"); | |
170 | ||
171 | if (source_date_epoch == NULL) | |
172 | return fallback; | |
173 | ||
174 | time_t time = (time_t) strtol(source_date_epoch, NULL, 10); | |
175 | ||
176 | if (gmtime(&time) == NULL) { | |
177 | fprintf(stderr, "%s: SOURCE_DATE_EPOCH is not valid\n", | |
87925df2 | 178 | cmdname); |
5847084f VC |
179 | time = 0; |
180 | } | |
181 | ||
182 | return time; | |
183 | } |