]>
git.ipfire.org Git - thirdparty/u-boot.git/blob - tools/mkenvimage.c
1 // SPDX-License-Identifier: GPL-2.0+
3 * (C) Copyright 2011 Free Electrons
4 * David Wagner <david.wagner@free-electrons.com>
6 * Inspired from envcrc.c:
8 * Paolo Scaffardi, AIRVENT SAM s.p.a - RIMINI(ITALY), arsenio@tin.it
19 #include <sys/types.h>
24 #include <u-boot/crc.h>
27 #define CRC_SIZE sizeof(uint32_t)
29 static void usage(const char *exec_name
)
31 fprintf(stderr
, "%s [-h] [-r] [-b] [-p <byte>] -s <environment partition size> -o <output> <input file>\n"
33 "This tool takes a key=value input file (same as would a `printenv' show) and generates the corresponding environment image, ready to be flashed.\n"
35 "\tThe input file is in format:\n"
39 "\tEmpty lines are skipped, and lines with a # in the first\n"
40 "\tcolumn are treated as comments (also skipped).\n"
41 "\t-r : the environment has multiple copies in flash\n"
42 "\t-b : the target is big endian (default is little endian)\n"
43 "\t-p <byte> : fill the image with <byte> bytes instead of 0xff bytes\n"
44 "\t-V : print version information and exit\n"
46 "If the input file is \"-\", data is read from standard input\n",
50 long int xstrtol(const char *s
)
55 tmp
= strtol(s
, NULL
, 0);
60 fprintf(stderr
, "Bad integer format: %s\n", s
);
62 fprintf(stderr
, "Error while parsing %s: %s\n", s
,
68 #define CHUNK_SIZE 4096
70 int main(int argc
, char **argv
)
72 uint32_t crc
, targetendian_crc
;
73 const char *bin_filename
= NULL
;
75 unsigned char *dataptr
, *envptr
;
76 unsigned char *filebuf
= NULL
;
77 unsigned int filesize
= 0, envsize
= 0, datasize
= 0;
80 unsigned char padbyte
= 0xff;
84 int ret
= EXIT_SUCCESS
;
89 prg
= basename(argv
[0]);
91 /* Turn off getopt()'s internal error message */
94 /* Parse the cmdline */
95 while ((option
= getopt(argc
, argv
, ":s:o:rbp:hV")) != -1) {
98 datasize
= xstrtol(optarg
);
101 bin_filename
= strdup(optarg
);
103 fprintf(stderr
, "Can't strdup() the output filename\n");
114 padbyte
= xstrtol(optarg
);
120 printf("%s version %s\n", prg
, PLAIN_VERSION
);
123 fprintf(stderr
, "Missing argument for option -%c\n",
128 fprintf(stderr
, "Wrong option -%c\n", optopt
);
134 /* Check datasize and allocate the data */
136 fprintf(stderr
, "Please specify the size of the environment partition.\n");
141 dataptr
= malloc(datasize
* sizeof(*dataptr
));
143 fprintf(stderr
, "Can't alloc %d bytes for dataptr.\n",
149 * envptr points to the beginning of the actual environment (after the
150 * crc and possible `redundant' byte
152 envsize
= datasize
- (CRC_SIZE
+ redundant
);
153 envptr
= dataptr
+ CRC_SIZE
+ redundant
;
155 /* Pad the environment with the padding byte */
156 memset(envptr
, padbyte
, envsize
);
158 /* Open the input file ... */
159 if (optind
>= argc
|| strcmp(argv
[optind
], "-") == 0) {
160 txt_fd
= STDIN_FILENO
;
162 txt_fd
= open(argv
[optind
], O_RDONLY
);
164 fprintf(stderr
, "Can't open \"%s\": %s\n",
165 argv
[optind
], strerror(errno
));
171 filebuf
= realloc(filebuf
, filesize
+ CHUNK_SIZE
);
173 fprintf(stderr
, "Can't realloc memory for the input file buffer\n");
176 readbytes
= read(txt_fd
, filebuf
+ filesize
, CHUNK_SIZE
);
178 fprintf(stderr
, "Error while reading: %s\n",
182 filesize
+= readbytes
;
183 } while (readbytes
> 0);
185 if (txt_fd
!= STDIN_FILENO
)
188 /* Parse a byte at time until reaching the file OR until the environment fills
189 * up. Check ep against envsize - 1 to allow for extra trailing '\0'. */
190 for (fp
= 0, ep
= 0 ; fp
< filesize
&& ep
< envsize
- 1; fp
++) {
191 if (filebuf
[fp
] == '\n') {
192 if (fp
== 0 || filebuf
[fp
-1] == '\n') {
197 } else if (filebuf
[fp
-1] == '\\') {
199 * Embedded newline in a variable.
201 * The backslash was added to the envptr; rewind
202 * and replace it with a newline
207 /* End of a variable */
210 } else if ((fp
== 0 || filebuf
[fp
-1] == '\n') && filebuf
[fp
] == '#') {
211 /* Comment, skip the line. */
212 while (++fp
< filesize
&& filebuf
[fp
] != '\n')
215 envptr
[ep
++] = filebuf
[fp
];
218 /* If there are more bytes in the file still, it means the env filled up
219 * before parsing the whole file. Eat comments & whitespace here to see if
220 * there was anything meaning full left in the file, and if so, throw a error
222 for( ; fp
< filesize
; fp
++ )
224 if (filebuf
[fp
] == '\n') {
225 if (fp
== 0 || filebuf
[fp
-1] == '\n') {
226 /* Ignore blank lines */
229 } else if ((fp
== 0 || filebuf
[fp
-1] == '\n') && filebuf
[fp
] == '#') {
230 while (++fp
< filesize
&& filebuf
[fp
] != '\n')
233 fprintf(stderr
, "The environment file is too large for the target environment storage\n");
238 * Make sure there is a final '\0'
239 * And do it again on the next byte to mark the end of the environment.
241 if (envptr
[ep
-1] != '\0') {
244 * The text file doesn't have an ending newline. We need to
245 * check the env size again to make sure we have room for two \0
248 fprintf(stderr
, "The environment file is too large for the target environment storage\n");
256 /* Computes the CRC and put it at the beginning of the data */
257 crc
= crc32(0, envptr
, envsize
);
258 targetendian_crc
= bigendian
? cpu_to_be32(crc
) : cpu_to_le32(crc
);
260 memcpy(dataptr
, &targetendian_crc
, sizeof(targetendian_crc
));
262 dataptr
[sizeof(targetendian_crc
)] = 1;
264 if (!bin_filename
|| strcmp(bin_filename
, "-") == 0) {
265 bin_fd
= STDOUT_FILENO
;
267 bin_fd
= creat(bin_filename
, S_IRUSR
| S_IWUSR
| S_IRGRP
|
270 fprintf(stderr
, "Can't open output file \"%s\": %s\n",
271 bin_filename
, strerror(errno
));
276 if (write(bin_fd
, dataptr
, sizeof(*dataptr
) * datasize
) !=
277 sizeof(*dataptr
) * datasize
) {
278 fprintf(stderr
, "write() failed: %s\n", strerror(errno
));