Loading...
1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Copyright 2011 The Chromium Authors, All Rights Reserved.
4 * Copyright 2008 Jon Loeliger, Freescale Semiconductor, Inc.
5 *
6 * util_is_printable_string contributed by
7 * Pantelis Antoniou <pantelis.antoniou AT gmail.com>
8 */
9
10#include <ctype.h>
11#include <stdio.h>
12#include <stdlib.h>
13#include <stdarg.h>
14#include <string.h>
15#include <assert.h>
16#include <inttypes.h>
17
18#include <errno.h>
19#include <fcntl.h>
20#include <unistd.h>
21
22#include "libfdt.h"
23#include "util.h"
24#include "version_gen.h"
25
26char *xstrdup(const char *s)
27{
28 int len = strlen(s) + 1;
29 char *d = xmalloc(len);
30
31 memcpy(d, s, len);
32
33 return d;
34}
35
36char *xstrndup(const char *s, size_t n)
37{
38 size_t len = strnlen(s, n) + 1;
39 char *d = xmalloc(len);
40
41 memcpy(d, s, len - 1);
42 d[len - 1] = '\0';
43
44 return d;
45}
46
47int xavsprintf_append(char **strp, const char *fmt, va_list ap)
48{
49 int n, size = 0; /* start with 128 bytes */
50 char *p;
51 va_list ap_copy;
52
53 p = *strp;
54 if (p)
55 size = strlen(p);
56
57 va_copy(ap_copy, ap);
58 n = vsnprintf(NULL, 0, fmt, ap_copy) + 1;
59 va_end(ap_copy);
60
61 p = xrealloc(p, size + n);
62
63 n = vsnprintf(p + size, n, fmt, ap);
64
65 *strp = p;
66 return strlen(p);
67}
68
69int xasprintf_append(char **strp, const char *fmt, ...)
70{
71 int n;
72 va_list ap;
73
74 va_start(ap, fmt);
75 n = xavsprintf_append(strp, fmt, ap);
76 va_end(ap);
77
78 return n;
79}
80
81int xasprintf(char **strp, const char *fmt, ...)
82{
83 int n;
84 va_list ap;
85
86 *strp = NULL;
87
88 va_start(ap, fmt);
89 n = xavsprintf_append(strp, fmt, ap);
90 va_end(ap);
91
92 return n;
93}
94
95char *join_path(const char *path, const char *name)
96{
97 int lenp = strlen(path);
98 int lenn = strlen(name);
99 int len;
100 int needslash = 1;
101 char *str;
102
103 len = lenp + lenn + 2;
104 if ((lenp > 0) && (path[lenp-1] == '/')) {
105 needslash = 0;
106 len--;
107 }
108
109 str = xmalloc(len);
110 memcpy(str, path, lenp);
111 if (needslash) {
112 str[lenp] = '/';
113 lenp++;
114 }
115 memcpy(str+lenp, name, lenn+1);
116 return str;
117}
118
119bool util_is_printable_string(const void *data, int len)
120{
121 const char *s = data;
122 const char *ss, *se;
123
124 /* zero length is not */
125 if (len == 0)
126 return 0;
127
128 /* must terminate with zero */
129 if (s[len - 1] != '\0')
130 return 0;
131
132 se = s + len;
133
134 while (s < se) {
135 ss = s;
136 while (s < se && *s && isprint((unsigned char)*s))
137 s++;
138
139 /* not zero, or not done yet */
140 if (*s != '\0' || s == ss)
141 return 0;
142
143 s++;
144 }
145
146 return 1;
147}
148
149/*
150 * Parse a octal encoded character starting at index i in string s. The
151 * resulting character will be returned and the index i will be updated to
152 * point at the character directly after the end of the encoding, this may be
153 * the '\0' terminator of the string.
154 */
155static char get_oct_char(const char *s, int *i)
156{
157 char x[4];
158 char *endx;
159 long val;
160
161 x[3] = '\0';
162 strncpy(x, s + *i, 3);
163
164 val = strtol(x, &endx, 8);
165
166 assert(endx > x);
167
168 (*i) += endx - x;
169 return val;
170}
171
172/*
173 * Parse a hexadecimal encoded character starting at index i in string s. The
174 * resulting character will be returned and the index i will be updated to
175 * point at the character directly after the end of the encoding, this may be
176 * the '\0' terminator of the string.
177 */
178static char get_hex_char(const char *s, int *i)
179{
180 char x[3];
181 char *endx;
182 long val;
183
184 x[2] = '\0';
185 strncpy(x, s + *i, 2);
186
187 val = strtol(x, &endx, 16);
188 if (!(endx > x))
189 die("\\x used with no following hex digits\n");
190
191 (*i) += endx - x;
192 return val;
193}
194
195char get_escape_char(const char *s, int *i)
196{
197 char c = s[*i];
198 int j = *i + 1;
199 char val;
200
201 switch (c) {
202 case 'a':
203 val = '\a';
204 break;
205 case 'b':
206 val = '\b';
207 break;
208 case 't':
209 val = '\t';
210 break;
211 case 'n':
212 val = '\n';
213 break;
214 case 'v':
215 val = '\v';
216 break;
217 case 'f':
218 val = '\f';
219 break;
220 case 'r':
221 val = '\r';
222 break;
223 case '0':
224 case '1':
225 case '2':
226 case '3':
227 case '4':
228 case '5':
229 case '6':
230 case '7':
231 j--; /* need to re-read the first digit as
232 * part of the octal value */
233 val = get_oct_char(s, &j);
234 break;
235 case 'x':
236 val = get_hex_char(s, &j);
237 break;
238 default:
239 val = c;
240 }
241
242 (*i) = j;
243 return val;
244}
245
246int utilfdt_read_err(const char *filename, char **buffp, size_t *len)
247{
248 int fd = 0; /* assume stdin */
249 char *buf = NULL;
250 size_t bufsize = 1024, offset = 0;
251 int ret = 0;
252
253 *buffp = NULL;
254 if (strcmp(filename, "-") != 0) {
255 fd = open(filename, O_RDONLY);
256 if (fd < 0)
257 return errno;
258 }
259
260 /* Loop until we have read everything */
261 buf = xmalloc(bufsize);
262 do {
263 /* Expand the buffer to hold the next chunk */
264 if (offset == bufsize) {
265 bufsize *= 2;
266 buf = xrealloc(buf, bufsize);
267 }
268
269 ret = read(fd, &buf[offset], bufsize - offset);
270 if (ret < 0) {
271 ret = errno;
272 break;
273 }
274 offset += ret;
275 } while (ret != 0);
276
277 /* Clean up, including closing stdin; return errno on error */
278 close(fd);
279 if (ret)
280 free(buf);
281 else
282 *buffp = buf;
283 if (len)
284 *len = bufsize;
285 return ret;
286}
287
288char *utilfdt_read(const char *filename, size_t *len)
289{
290 char *buff;
291 int ret = utilfdt_read_err(filename, &buff, len);
292
293 if (ret) {
294 fprintf(stderr, "Couldn't open blob from '%s': %s\n", filename,
295 strerror(ret));
296 return NULL;
297 }
298 /* Successful read */
299 return buff;
300}
301
302int utilfdt_write_err(const char *filename, const void *blob)
303{
304 int fd = 1; /* assume stdout */
305 int totalsize;
306 int offset;
307 int ret = 0;
308 const char *ptr = blob;
309
310 if (strcmp(filename, "-") != 0) {
311 fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0666);
312 if (fd < 0)
313 return errno;
314 }
315
316 totalsize = fdt_totalsize(blob);
317 offset = 0;
318
319 while (offset < totalsize) {
320 ret = write(fd, ptr + offset, totalsize - offset);
321 if (ret < 0) {
322 ret = -errno;
323 break;
324 }
325 offset += ret;
326 }
327 /* Close the file/stdin; return errno on error */
328 if (fd != 1)
329 close(fd);
330 return ret < 0 ? -ret : 0;
331}
332
333
334int utilfdt_write(const char *filename, const void *blob)
335{
336 int ret = utilfdt_write_err(filename, blob);
337
338 if (ret) {
339 fprintf(stderr, "Couldn't write blob to '%s': %s\n", filename,
340 strerror(ret));
341 }
342 return ret ? -1 : 0;
343}
344
345int utilfdt_decode_type(const char *fmt, int *type, int *size)
346{
347 int qualifier = 0;
348
349 if (!*fmt)
350 return -1;
351
352 /* get the conversion qualifier */
353 *size = -1;
354 if (strchr("hlLb", *fmt)) {
355 qualifier = *fmt++;
356 if (qualifier == *fmt) {
357 switch (*fmt++) {
358/* TODO: case 'l': qualifier = 'L'; break;*/
359 case 'h':
360 qualifier = 'b';
361 break;
362 }
363 }
364 }
365
366 /* we should now have a type */
367 if ((*fmt == '\0') || !strchr("iuxsr", *fmt))
368 return -1;
369
370 /* convert qualifier (bhL) to byte size */
371 if (*fmt != 's' && *fmt != 'r')
372 *size = qualifier == 'b' ? 1 :
373 qualifier == 'h' ? 2 :
374 qualifier == 'l' ? 4 : -1;
375 *type = *fmt++;
376
377 /* that should be it! */
378 if (*fmt)
379 return -1;
380 return 0;
381}
382
383void utilfdt_print_data(const char *data, int len)
384{
385 int i;
386 const char *s;
387
388 /* no data, don't print */
389 if (len == 0)
390 return;
391
392 if (util_is_printable_string(data, len)) {
393 printf(" = ");
394
395 s = data;
396 do {
397 printf("\"%s\"", s);
398 s += strlen(s) + 1;
399 if (s < data + len)
400 printf(", ");
401 } while (s < data + len);
402
403 } else if ((len % 4) == 0) {
404 const fdt32_t *cell = (const fdt32_t *)data;
405
406 printf(" = <");
407 for (i = 0, len /= 4; i < len; i++)
408 printf("0x%08" PRIx32 "%s", fdt32_to_cpu(cell[i]),
409 i < (len - 1) ? " " : "");
410 printf(">");
411 } else {
412 const unsigned char *p = (const unsigned char *)data;
413 printf(" = [");
414 for (i = 0; i < len; i++)
415 printf("%02x%s", *p++, i < len - 1 ? " " : "");
416 printf("]");
417 }
418}
419
420void NORETURN util_version(void)
421{
422 printf("Version: %s\n", DTC_VERSION);
423 exit(0);
424}
425
426void NORETURN util_usage(const char *errmsg, const char *synopsis,
427 const char *short_opts,
428 struct option const long_opts[],
429 const char * const opts_help[])
430{
431 FILE *fp = errmsg ? stderr : stdout;
432 const char a_arg[] = "<arg>";
433 size_t a_arg_len = strlen(a_arg) + 1;
434 size_t i;
435 int optlen;
436
437 fprintf(fp,
438 "Usage: %s\n"
439 "\n"
440 "Options: -[%s]\n", synopsis, short_opts);
441
442 /* prescan the --long opt length to auto-align */
443 optlen = 0;
444 for (i = 0; long_opts[i].name; ++i) {
445 /* +1 is for space between --opt and help text */
446 int l = strlen(long_opts[i].name) + 1;
447 if (long_opts[i].has_arg == a_argument)
448 l += a_arg_len;
449 if (optlen < l)
450 optlen = l;
451 }
452
453 for (i = 0; long_opts[i].name; ++i) {
454 /* helps when adding new applets or options */
455 assert(opts_help[i] != NULL);
456
457 /* first output the short flag if it has one */
458 if (long_opts[i].val > '~')
459 fprintf(fp, " ");
460 else
461 fprintf(fp, " -%c, ", long_opts[i].val);
462
463 /* then the long flag */
464 if (long_opts[i].has_arg == no_argument)
465 fprintf(fp, "--%-*s", optlen, long_opts[i].name);
466 else
467 fprintf(fp, "--%s %s%*s", long_opts[i].name, a_arg,
468 (int)(optlen - strlen(long_opts[i].name) - a_arg_len), "");
469
470 /* finally the help text */
471 fprintf(fp, "%s\n", opts_help[i]);
472 }
473
474 if (errmsg) {
475 fprintf(fp, "\nError: %s\n", errmsg);
476 exit(EXIT_FAILURE);
477 } else
478 exit(EXIT_SUCCESS);
479}
1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Copyright 2011 The Chromium Authors, All Rights Reserved.
4 * Copyright 2008 Jon Loeliger, Freescale Semiconductor, Inc.
5 *
6 * util_is_printable_string contributed by
7 * Pantelis Antoniou <pantelis.antoniou AT gmail.com>
8 */
9
10#include <ctype.h>
11#include <stdio.h>
12#include <stdlib.h>
13#include <stdarg.h>
14#include <string.h>
15#include <assert.h>
16#include <inttypes.h>
17
18#include <errno.h>
19#include <fcntl.h>
20#include <unistd.h>
21
22#include "libfdt.h"
23#include "util.h"
24#include "version_gen.h"
25
26char *xstrdup(const char *s)
27{
28 int len = strlen(s) + 1;
29 char *d = xmalloc(len);
30
31 memcpy(d, s, len);
32
33 return d;
34}
35
36int xavsprintf_append(char **strp, const char *fmt, va_list ap)
37{
38 int n, size = 0; /* start with 128 bytes */
39 char *p;
40 va_list ap_copy;
41
42 p = *strp;
43 if (p)
44 size = strlen(p);
45
46 va_copy(ap_copy, ap);
47 n = vsnprintf(NULL, 0, fmt, ap_copy) + 1;
48 va_end(ap_copy);
49
50 p = xrealloc(p, size + n);
51
52 n = vsnprintf(p + size, n, fmt, ap);
53
54 *strp = p;
55 return strlen(p);
56}
57
58int xasprintf_append(char **strp, const char *fmt, ...)
59{
60 int n;
61 va_list ap;
62
63 va_start(ap, fmt);
64 n = xavsprintf_append(strp, fmt, ap);
65 va_end(ap);
66
67 return n;
68}
69
70int xasprintf(char **strp, const char *fmt, ...)
71{
72 int n;
73 va_list ap;
74
75 *strp = NULL;
76
77 va_start(ap, fmt);
78 n = xavsprintf_append(strp, fmt, ap);
79 va_end(ap);
80
81 return n;
82}
83
84char *join_path(const char *path, const char *name)
85{
86 int lenp = strlen(path);
87 int lenn = strlen(name);
88 int len;
89 int needslash = 1;
90 char *str;
91
92 len = lenp + lenn + 2;
93 if ((lenp > 0) && (path[lenp-1] == '/')) {
94 needslash = 0;
95 len--;
96 }
97
98 str = xmalloc(len);
99 memcpy(str, path, lenp);
100 if (needslash) {
101 str[lenp] = '/';
102 lenp++;
103 }
104 memcpy(str+lenp, name, lenn+1);
105 return str;
106}
107
108bool util_is_printable_string(const void *data, int len)
109{
110 const char *s = data;
111 const char *ss, *se;
112
113 /* zero length is not */
114 if (len == 0)
115 return 0;
116
117 /* must terminate with zero */
118 if (s[len - 1] != '\0')
119 return 0;
120
121 se = s + len;
122
123 while (s < se) {
124 ss = s;
125 while (s < se && *s && isprint((unsigned char)*s))
126 s++;
127
128 /* not zero, or not done yet */
129 if (*s != '\0' || s == ss)
130 return 0;
131
132 s++;
133 }
134
135 return 1;
136}
137
138/*
139 * Parse a octal encoded character starting at index i in string s. The
140 * resulting character will be returned and the index i will be updated to
141 * point at the character directly after the end of the encoding, this may be
142 * the '\0' terminator of the string.
143 */
144static char get_oct_char(const char *s, int *i)
145{
146 char x[4];
147 char *endx;
148 long val;
149
150 x[3] = '\0';
151 strncpy(x, s + *i, 3);
152
153 val = strtol(x, &endx, 8);
154
155 assert(endx > x);
156
157 (*i) += endx - x;
158 return val;
159}
160
161/*
162 * Parse a hexadecimal encoded character starting at index i in string s. The
163 * resulting character will be returned and the index i will be updated to
164 * point at the character directly after the end of the encoding, this may be
165 * the '\0' terminator of the string.
166 */
167static char get_hex_char(const char *s, int *i)
168{
169 char x[3];
170 char *endx;
171 long val;
172
173 x[2] = '\0';
174 strncpy(x, s + *i, 2);
175
176 val = strtol(x, &endx, 16);
177 if (!(endx > x))
178 die("\\x used with no following hex digits\n");
179
180 (*i) += endx - x;
181 return val;
182}
183
184char get_escape_char(const char *s, int *i)
185{
186 char c = s[*i];
187 int j = *i + 1;
188 char val;
189
190 switch (c) {
191 case 'a':
192 val = '\a';
193 break;
194 case 'b':
195 val = '\b';
196 break;
197 case 't':
198 val = '\t';
199 break;
200 case 'n':
201 val = '\n';
202 break;
203 case 'v':
204 val = '\v';
205 break;
206 case 'f':
207 val = '\f';
208 break;
209 case 'r':
210 val = '\r';
211 break;
212 case '0':
213 case '1':
214 case '2':
215 case '3':
216 case '4':
217 case '5':
218 case '6':
219 case '7':
220 j--; /* need to re-read the first digit as
221 * part of the octal value */
222 val = get_oct_char(s, &j);
223 break;
224 case 'x':
225 val = get_hex_char(s, &j);
226 break;
227 default:
228 val = c;
229 }
230
231 (*i) = j;
232 return val;
233}
234
235int utilfdt_read_err(const char *filename, char **buffp, size_t *len)
236{
237 int fd = 0; /* assume stdin */
238 char *buf = NULL;
239 size_t bufsize = 1024, offset = 0;
240 int ret = 0;
241
242 *buffp = NULL;
243 if (strcmp(filename, "-") != 0) {
244 fd = open(filename, O_RDONLY);
245 if (fd < 0)
246 return errno;
247 }
248
249 /* Loop until we have read everything */
250 buf = xmalloc(bufsize);
251 do {
252 /* Expand the buffer to hold the next chunk */
253 if (offset == bufsize) {
254 bufsize *= 2;
255 buf = xrealloc(buf, bufsize);
256 }
257
258 ret = read(fd, &buf[offset], bufsize - offset);
259 if (ret < 0) {
260 ret = errno;
261 break;
262 }
263 offset += ret;
264 } while (ret != 0);
265
266 /* Clean up, including closing stdin; return errno on error */
267 close(fd);
268 if (ret)
269 free(buf);
270 else
271 *buffp = buf;
272 if (len)
273 *len = bufsize;
274 return ret;
275}
276
277char *utilfdt_read(const char *filename, size_t *len)
278{
279 char *buff;
280 int ret = utilfdt_read_err(filename, &buff, len);
281
282 if (ret) {
283 fprintf(stderr, "Couldn't open blob from '%s': %s\n", filename,
284 strerror(ret));
285 return NULL;
286 }
287 /* Successful read */
288 return buff;
289}
290
291int utilfdt_write_err(const char *filename, const void *blob)
292{
293 int fd = 1; /* assume stdout */
294 int totalsize;
295 int offset;
296 int ret = 0;
297 const char *ptr = blob;
298
299 if (strcmp(filename, "-") != 0) {
300 fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0666);
301 if (fd < 0)
302 return errno;
303 }
304
305 totalsize = fdt_totalsize(blob);
306 offset = 0;
307
308 while (offset < totalsize) {
309 ret = write(fd, ptr + offset, totalsize - offset);
310 if (ret < 0) {
311 ret = -errno;
312 break;
313 }
314 offset += ret;
315 }
316 /* Close the file/stdin; return errno on error */
317 if (fd != 1)
318 close(fd);
319 return ret < 0 ? -ret : 0;
320}
321
322
323int utilfdt_write(const char *filename, const void *blob)
324{
325 int ret = utilfdt_write_err(filename, blob);
326
327 if (ret) {
328 fprintf(stderr, "Couldn't write blob to '%s': %s\n", filename,
329 strerror(ret));
330 }
331 return ret ? -1 : 0;
332}
333
334int utilfdt_decode_type(const char *fmt, int *type, int *size)
335{
336 int qualifier = 0;
337
338 if (!*fmt)
339 return -1;
340
341 /* get the conversion qualifier */
342 *size = -1;
343 if (strchr("hlLb", *fmt)) {
344 qualifier = *fmt++;
345 if (qualifier == *fmt) {
346 switch (*fmt++) {
347/* TODO: case 'l': qualifier = 'L'; break;*/
348 case 'h':
349 qualifier = 'b';
350 break;
351 }
352 }
353 }
354
355 /* we should now have a type */
356 if ((*fmt == '\0') || !strchr("iuxs", *fmt))
357 return -1;
358
359 /* convert qualifier (bhL) to byte size */
360 if (*fmt != 's')
361 *size = qualifier == 'b' ? 1 :
362 qualifier == 'h' ? 2 :
363 qualifier == 'l' ? 4 : -1;
364 *type = *fmt++;
365
366 /* that should be it! */
367 if (*fmt)
368 return -1;
369 return 0;
370}
371
372void utilfdt_print_data(const char *data, int len)
373{
374 int i;
375 const char *s;
376
377 /* no data, don't print */
378 if (len == 0)
379 return;
380
381 if (util_is_printable_string(data, len)) {
382 printf(" = ");
383
384 s = data;
385 do {
386 printf("\"%s\"", s);
387 s += strlen(s) + 1;
388 if (s < data + len)
389 printf(", ");
390 } while (s < data + len);
391
392 } else if ((len % 4) == 0) {
393 const fdt32_t *cell = (const fdt32_t *)data;
394
395 printf(" = <");
396 for (i = 0, len /= 4; i < len; i++)
397 printf("0x%08" PRIx32 "%s", fdt32_to_cpu(cell[i]),
398 i < (len - 1) ? " " : "");
399 printf(">");
400 } else {
401 const unsigned char *p = (const unsigned char *)data;
402 printf(" = [");
403 for (i = 0; i < len; i++)
404 printf("%02x%s", *p++, i < len - 1 ? " " : "");
405 printf("]");
406 }
407}
408
409void NORETURN util_version(void)
410{
411 printf("Version: %s\n", DTC_VERSION);
412 exit(0);
413}
414
415void NORETURN util_usage(const char *errmsg, const char *synopsis,
416 const char *short_opts,
417 struct option const long_opts[],
418 const char * const opts_help[])
419{
420 FILE *fp = errmsg ? stderr : stdout;
421 const char a_arg[] = "<arg>";
422 size_t a_arg_len = strlen(a_arg) + 1;
423 size_t i;
424 int optlen;
425
426 fprintf(fp,
427 "Usage: %s\n"
428 "\n"
429 "Options: -[%s]\n", synopsis, short_opts);
430
431 /* prescan the --long opt length to auto-align */
432 optlen = 0;
433 for (i = 0; long_opts[i].name; ++i) {
434 /* +1 is for space between --opt and help text */
435 int l = strlen(long_opts[i].name) + 1;
436 if (long_opts[i].has_arg == a_argument)
437 l += a_arg_len;
438 if (optlen < l)
439 optlen = l;
440 }
441
442 for (i = 0; long_opts[i].name; ++i) {
443 /* helps when adding new applets or options */
444 assert(opts_help[i] != NULL);
445
446 /* first output the short flag if it has one */
447 if (long_opts[i].val > '~')
448 fprintf(fp, " ");
449 else
450 fprintf(fp, " -%c, ", long_opts[i].val);
451
452 /* then the long flag */
453 if (long_opts[i].has_arg == no_argument)
454 fprintf(fp, "--%-*s", optlen, long_opts[i].name);
455 else
456 fprintf(fp, "--%s %s%*s", long_opts[i].name, a_arg,
457 (int)(optlen - strlen(long_opts[i].name) - a_arg_len), "");
458
459 /* finally the help text */
460 fprintf(fp, "%s\n", opts_help[i]);
461 }
462
463 if (errmsg) {
464 fprintf(fp, "\nError: %s\n", errmsg);
465 exit(EXIT_FAILURE);
466 } else
467 exit(EXIT_SUCCESS);
468}