]>
git.ipfire.org Git - thirdparty/u-boot.git/blob - tools/mkenvimage.c
2 * (C) Copyright 2011 Free Electrons
3 * David Wagner <david.wagner@free-electrons.com>
5 * Inspired from envcrc.c:
7 * Paolo Scaffardi, AIRVENT SAM s.p.a - RIMINI(ITALY), arsenio@tin.it
9 * See file CREDITS for list of people who contributed to this
12 * This program is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU General Public License as
14 * published by the Free Software Foundation; either version 2 of
15 * the License, or (at your option) any later version.
17 * This program is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU General Public License for more details.
22 * You should have received a copy of the GNU General Public License
23 * along with this program; if not, write to the Free Software
24 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
28 /* We want the GNU version of basename() */
38 #include <sys/types.h>
41 #include <u-boot/crc.h>
44 #define CRC_SIZE sizeof(uint32_t)
46 static void usage(const char *exec_name
)
48 fprintf(stderr
, "%s [-h] [-r] [-b] [-p <byte>] "
49 "-s <environment partition size> -o <output> <input file>\n"
51 "This tool takes a key=value input file (same as would a "
52 "`printenv' show) and generates the corresponding environment "
53 "image, ready to be flashed.\n"
55 "\tThe input file is in format:\n"
59 "\t-r : the environment has multiple copies in flash\n"
60 "\t-b : the target is big endian (default is little endian)\n"
61 "\t-p <byte> : fill the image with <byte> bytes instead of "
63 "\t-V : print version information and exit\n"
65 "If the input file is \"-\", data is read from standard input\n",
69 int main(int argc
, char **argv
)
71 uint32_t crc
, targetendian_crc
;
72 const char *txt_filename
= NULL
, *bin_filename
= NULL
;
74 unsigned char *dataptr
, *envptr
;
75 unsigned char *filebuf
= NULL
;
76 unsigned int filesize
= 0, envsize
= 0, datasize
= 0;
79 unsigned char padbyte
= 0xff;
82 int ret
= EXIT_SUCCESS
;
84 struct stat txt_file_stat
;
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
= strtol(optarg
, NULL
, 0);
101 bin_filename
= strdup(optarg
);
103 fprintf(stderr
, "Can't strdup() the output "
115 padbyte
= strtol(optarg
, NULL
, 0);
121 printf("%s version %s\n", prg
, PLAIN_VERSION
);
124 fprintf(stderr
, "Missing argument for option -%c\n",
129 fprintf(stderr
, "Wrong option -%c\n", optopt
);
135 /* Check datasize and allocate the data */
138 "Please specify the size of the environment "
144 dataptr
= malloc(datasize
* sizeof(*dataptr
));
146 fprintf(stderr
, "Can't alloc dataptr.\n");
151 * envptr points to the beginning of the actual environment (after the
152 * crc and possible `redundant' bit
154 envsize
= datasize
- (CRC_SIZE
+ redundant
);
155 envptr
= dataptr
+ CRC_SIZE
+ redundant
;
157 /* Pad the environment with the padding byte */
158 memset(envptr
, padbyte
, envsize
);
160 /* Open the input file ... */
161 if (optind
>= argc
) {
162 fprintf(stderr
, "Please specify an input filename\n");
166 txt_filename
= argv
[optind
];
167 if (strcmp(txt_filename
, "-") == 0) {
169 int readlen
= sizeof(*envptr
) * 2048;
170 txt_fd
= STDIN_FILENO
;
173 filebuf
= realloc(filebuf
, readlen
);
174 readbytes
= read(txt_fd
, filebuf
+ filesize
, readlen
);
175 filesize
+= readbytes
;
176 } while (readbytes
== readlen
);
179 txt_fd
= open(txt_filename
, O_RDONLY
);
181 fprintf(stderr
, "Can't open \"%s\": %s\n",
182 txt_filename
, strerror(errno
));
185 /* ... and check it */
186 ret
= fstat(txt_fd
, &txt_file_stat
);
188 fprintf(stderr
, "Can't stat() on \"%s\": "
189 "%s\n", txt_filename
, strerror(errno
));
193 filesize
= txt_file_stat
.st_size
;
194 /* Read the raw input file and transform it */
195 filebuf
= malloc(sizeof(*envptr
) * filesize
);
196 ret
= read(txt_fd
, filebuf
, sizeof(*envptr
) * filesize
);
197 if (ret
!= sizeof(*envptr
) * filesize
) {
198 fprintf(stderr
, "Can't read the whole input file\n");
204 * The right test to do is "=>" (not ">") because of the additional
205 * ending \0. See below.
207 if (filesize
>= envsize
) {
208 fprintf(stderr
, "The input file is larger than the "
209 "environment partition size\n");
213 /* Replace newlines separating variables with \0 */
214 for (fp
= 0, ep
= 0 ; fp
< filesize
; fp
++) {
215 if (filebuf
[fp
] == '\n') {
218 * Newline at the beginning of the file ?
222 } else if (filebuf
[fp
-1] == '\\') {
224 * Embedded newline in a variable.
226 * The backslash was added to the envptr ;
227 * rewind and replace it with a newline
232 /* End of a variable */
235 } else if (filebuf
[fp
] == '#') {
236 if (fp
!= 0 && filebuf
[fp
-1] == '\n') {
237 /* This line is a comment, let's skip it */
238 while (fp
< txt_file_stat
.st_size
&& fp
++ &&
239 filebuf
[fp
] != '\n');
241 envptr
[ep
++] = filebuf
[fp
];
244 envptr
[ep
++] = filebuf
[fp
];
248 * Make sure there is a final '\0'
249 * And do it again on the next byte to mark the end of the environment.
251 if (envptr
[ep
-1] != '\0') {
254 * The text file doesn't have an ending newline. We need to
255 * check the env size again to make sure we have room for two \0
258 fprintf(stderr
, "The environment file is too large for "
259 "the target environment storage\n");
267 /* Computes the CRC and put it at the beginning of the data */
268 crc
= crc32(0, envptr
, envsize
);
269 targetendian_crc
= bigendian
? cpu_to_be32(crc
) : cpu_to_le32(crc
);
271 memcpy(dataptr
, &targetendian_crc
, sizeof(uint32_t));
273 bin_fd
= creat(bin_filename
, S_IRUSR
| S_IWUSR
| S_IRGRP
| S_IWGRP
);
275 fprintf(stderr
, "Can't open output file \"%s\": %s\n",
276 bin_filename
, strerror(errno
));
280 if (write(bin_fd
, dataptr
, sizeof(*dataptr
) * datasize
) !=
281 sizeof(*dataptr
) * datasize
) {
282 fprintf(stderr
, "write() failed: %s\n", strerror(errno
));