]>
Commit | Line | Data |
---|---|---|
83d290c5 | 1 | // SPDX-License-Identifier: GPL-2.0+ |
6bf4ca07 HS |
2 | /* |
3 | * (C) Copyright 2014 | |
4 | * DENX Software Engineering | |
5 | * Heiko Schocher <hs@denx.de> | |
6 | * | |
7 | * (C) Copyright 2008 Semihalf | |
8 | * | |
9 | * (C) Copyright 2000-2004 | |
10 | * DENX Software Engineering | |
11 | * Wolfgang Denk, wd@denx.de | |
12 | * | |
13 | * Updated-by: Prafulla Wadaskar <prafulla@marvell.com> | |
14 | * FIT image specific code abstracted from mkimage.c | |
15 | * some functions added to address abstraction | |
16 | * | |
17 | * All rights reserved. | |
6bf4ca07 HS |
18 | */ |
19 | ||
20 | #include "imagetool.h" | |
21 | #include "mkimage.h" | |
22 | #include "fit_common.h" | |
23 | #include <image.h> | |
24 | #include <u-boot/crc.h> | |
25 | ||
26 | int fit_verify_header(unsigned char *ptr, int image_size, | |
27 | struct image_tool_params *params) | |
28 | { | |
29 | return fdt_check_header(ptr); | |
30 | } | |
31 | ||
32 | int fit_check_image_types(uint8_t type) | |
33 | { | |
34 | if (type == IH_TYPE_FLATDT) | |
35 | return EXIT_SUCCESS; | |
36 | else | |
37 | return EXIT_FAILURE; | |
38 | } | |
39 | ||
a9468115 SG |
40 | int mmap_fdt(const char *cmdname, const char *fname, size_t size_inc, |
41 | void **blobp, struct stat *sbuf, bool delete_on_error) | |
6bf4ca07 HS |
42 | { |
43 | void *ptr; | |
44 | int fd; | |
45 | ||
46 | /* Load FIT blob into memory (we need to write hashes/signatures) */ | |
47 | fd = open(fname, O_RDWR | O_BINARY); | |
48 | ||
49 | if (fd < 0) { | |
50 | fprintf(stderr, "%s: Can't open %s: %s\n", | |
51 | cmdname, fname, strerror(errno)); | |
ef0af64b | 52 | goto err; |
6bf4ca07 HS |
53 | } |
54 | ||
55 | if (fstat(fd, sbuf) < 0) { | |
56 | fprintf(stderr, "%s: Can't stat %s: %s\n", | |
57 | cmdname, fname, strerror(errno)); | |
ef0af64b | 58 | goto err; |
6bf4ca07 HS |
59 | } |
60 | ||
a9468115 SG |
61 | if (size_inc) { |
62 | sbuf->st_size += size_inc; | |
63 | if (ftruncate(fd, sbuf->st_size)) { | |
64 | fprintf(stderr, "%s: Can't expand %s: %s\n", | |
65 | cmdname, fname, strerror(errno)); | |
66 | goto err; | |
67 | } | |
68 | } | |
69 | ||
6bf4ca07 HS |
70 | errno = 0; |
71 | ptr = mmap(0, sbuf->st_size, PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0); | |
72 | if ((ptr == MAP_FAILED) || (errno != 0)) { | |
73 | fprintf(stderr, "%s: Can't read %s: %s\n", | |
74 | cmdname, fname, strerror(errno)); | |
ef0af64b | 75 | goto err; |
6bf4ca07 HS |
76 | } |
77 | ||
78 | /* check if ptr has a valid blob */ | |
79 | if (fdt_check_header(ptr)) { | |
80 | fprintf(stderr, "%s: Invalid FIT blob\n", cmdname); | |
ef0af64b | 81 | goto err; |
6bf4ca07 HS |
82 | } |
83 | ||
a9468115 SG |
84 | /* expand if needed */ |
85 | if (size_inc) { | |
86 | int ret; | |
87 | ||
88 | ret = fdt_open_into(ptr, ptr, sbuf->st_size); | |
89 | if (ret) { | |
90 | fprintf(stderr, "%s: Cannot expand FDT: %s\n", | |
91 | cmdname, fdt_strerror(ret)); | |
92 | goto err; | |
93 | } | |
94 | } | |
95 | ||
6bf4ca07 HS |
96 | *blobp = ptr; |
97 | return fd; | |
ef0af64b SG |
98 | |
99 | err: | |
100 | if (fd >= 0) | |
101 | close(fd); | |
102 | if (delete_on_error) | |
103 | unlink(fname); | |
104 | ||
105 | return -1; | |
6bf4ca07 | 106 | } |