blob: 855a096d0ab0f2579a07bcc60d662a9d8eaf2a22 [file] [log] [blame]
Guilherme Maciel Ferreiraf86ed6a2013-12-01 12:43:10 -07001/*
2 * (C) Copyright 2013
3 *
4 * Written by Guilherme Maciel Ferreira <guilherme.maciel.ferreira@gmail.com>
5 *
6 * SPDX-License-Identifier: GPL-2.0+
7 */
8
9#include "imagetool.h"
10
Guilherme Maciel Ferreira0ca66912015-01-15 02:48:05 -020011#include <image.h>
12
Guilherme Maciel Ferreiraa93648d2015-01-15 02:48:07 -020013struct image_type_params *imagetool_get_type(int type)
Guilherme Maciel Ferreira0ca66912015-01-15 02:48:05 -020014{
Andreas Bießmann1fddd7b2015-02-09 00:06:10 +010015 struct image_type_params **curr;
16 INIT_SECTION(image_type);
17
18 struct image_type_params **start = __start_image_type;
19 struct image_type_params **end = __stop_image_type;
Guilherme Maciel Ferreira0ca66912015-01-15 02:48:05 -020020
Guilherme Maciel Ferreiraa93648d2015-01-15 02:48:07 -020021 for (curr = start; curr != end; curr++) {
Andreas Bießmann1fddd7b2015-02-09 00:06:10 +010022 if ((*curr)->check_image_type) {
23 if (!(*curr)->check_image_type(type))
24 return *curr;
Guilherme Maciel Ferreira0ca66912015-01-15 02:48:05 -020025 }
26 }
27 return NULL;
28}
29
30int imagetool_verify_print_header(
31 void *ptr,
32 struct stat *sbuf,
33 struct image_type_params *tparams,
34 struct image_tool_params *params)
35{
36 int retval = -1;
Andreas Bießmann1fddd7b2015-02-09 00:06:10 +010037 struct image_type_params **curr;
38 INIT_SECTION(image_type);
Guilherme Maciel Ferreira0ca66912015-01-15 02:48:05 -020039
Andreas Bießmann1fddd7b2015-02-09 00:06:10 +010040 struct image_type_params **start = __start_image_type;
41 struct image_type_params **end = __stop_image_type;
Guilherme Maciel Ferreiraa93648d2015-01-15 02:48:07 -020042
43 for (curr = start; curr != end; curr++) {
Andreas Bießmann1fddd7b2015-02-09 00:06:10 +010044 if ((*curr)->verify_header) {
45 retval = (*curr)->verify_header((unsigned char *)ptr,
Guilherme Maciel Ferreira0ca66912015-01-15 02:48:05 -020046 sbuf->st_size, params);
47
48 if (retval == 0) {
49 /*
50 * Print the image information if verify is
51 * successful
52 */
Andreas Bießmann1fddd7b2015-02-09 00:06:10 +010053 if ((*curr)->print_header) {
Simon Glassbd6e1422016-05-01 13:55:38 -060054 if (!params->quiet)
55 (*curr)->print_header(ptr);
Guilherme Maciel Ferreira0ca66912015-01-15 02:48:05 -020056 } else {
57 fprintf(stderr,
58 "%s: print_header undefined for %s\n",
Andreas Bießmann1fddd7b2015-02-09 00:06:10 +010059 params->cmdname, (*curr)->name);
Guilherme Maciel Ferreira0ca66912015-01-15 02:48:05 -020060 }
61 break;
62 }
63 }
64 }
65
66 return retval;
67}
Guilherme Maciel Ferreira067d1562015-01-15 02:48:06 -020068
Guilherme Maciel Ferreira67f946c2015-01-15 02:54:41 -020069int imagetool_save_subimage(
Guilherme Maciel Ferreira067d1562015-01-15 02:48:06 -020070 const char *file_name,
71 ulong file_data,
72 ulong file_len)
73{
74 int dfd;
75
76 dfd = open(file_name, O_RDWR | O_CREAT | O_TRUNC | O_BINARY,
77 S_IRUSR | S_IWUSR);
78 if (dfd < 0) {
79 fprintf(stderr, "Can't open \"%s\": %s\n",
80 file_name, strerror(errno));
81 return -1;
82 }
83
84 if (write(dfd, (void *)file_data, file_len) != (ssize_t)file_len) {
85 fprintf(stderr, "Write error on \"%s\": %s\n",
86 file_name, strerror(errno));
87 close(dfd);
88 return -1;
89 }
90
91 close(dfd);
92
93 return 0;
94}
Simon Glass3837ce62016-02-22 22:55:49 -070095
96int imagetool_get_filesize(struct image_tool_params *params, const char *fname)
97{
98 struct stat sbuf;
99 int fd;
100
101 fd = open(fname, O_RDONLY | O_BINARY);
102 if (fd < 0) {
103 fprintf(stderr, "%s: Can't open %s: %s\n",
104 params->cmdname, fname, strerror(errno));
105 return -1;
106 }
107
108 if (fstat(fd, &sbuf) < 0) {
109 fprintf(stderr, "%s: Can't stat %s: %s\n",
110 params->cmdname, fname, strerror(errno));
Simon Glassb12a81c2016-03-16 07:45:37 -0600111 close(fd);
Simon Glass3837ce62016-02-22 22:55:49 -0700112 return -1;
113 }
114 close(fd);
115
116 return sbuf.st_size;
117}
Vagrant Cascadian58470842016-06-16 12:28:40 -0700118
119time_t imagetool_get_source_date(
120 struct image_tool_params *params,
121 time_t fallback)
122{
123 char *source_date_epoch = getenv("SOURCE_DATE_EPOCH");
124
125 if (source_date_epoch == NULL)
126 return fallback;
127
128 time_t time = (time_t) strtol(source_date_epoch, NULL, 10);
129
130 if (gmtime(&time) == NULL) {
131 fprintf(stderr, "%s: SOURCE_DATE_EPOCH is not valid\n",
132 params->cmdname);
133 time = 0;
134 }
135
136 return time;
137}