]>
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 <[email protected]> | |
6 | * | |
7 | * (C) Copyright 2008 Semihalf | |
8 | * | |
9 | * (C) Copyright 2000-2004 | |
10 | * DENX Software Engineering | |
11 | * Wolfgang Denk, [email protected] | |
12 | * | |
13 | * Updated-by: Prafulla Wadaskar <[email protected]> | |
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 | { | |
d32aa3ca JH |
29 | if (fdt_check_header(ptr) != EXIT_SUCCESS || !fit_check_format(ptr)) |
30 | return EXIT_FAILURE; | |
31 | ||
32 | return EXIT_SUCCESS; | |
6bf4ca07 HS |
33 | } |
34 | ||
35 | int fit_check_image_types(uint8_t type) | |
36 | { | |
37 | if (type == IH_TYPE_FLATDT) | |
38 | return EXIT_SUCCESS; | |
39 | else | |
40 | return EXIT_FAILURE; | |
41 | } | |
42 | ||
a9468115 SG |
43 | int mmap_fdt(const char *cmdname, const char *fname, size_t size_inc, |
44 | void **blobp, struct stat *sbuf, bool delete_on_error) | |
6bf4ca07 HS |
45 | { |
46 | void *ptr; | |
47 | int fd; | |
48 | ||
49 | /* Load FIT blob into memory (we need to write hashes/signatures) */ | |
50 | fd = open(fname, O_RDWR | O_BINARY); | |
51 | ||
52 | if (fd < 0) { | |
53 | fprintf(stderr, "%s: Can't open %s: %s\n", | |
54 | cmdname, fname, strerror(errno)); | |
ef0af64b | 55 | goto err; |
6bf4ca07 HS |
56 | } |
57 | ||
58 | if (fstat(fd, sbuf) < 0) { | |
59 | fprintf(stderr, "%s: Can't stat %s: %s\n", | |
60 | cmdname, fname, strerror(errno)); | |
ef0af64b | 61 | goto err; |
6bf4ca07 HS |
62 | } |
63 | ||
a9468115 SG |
64 | if (size_inc) { |
65 | sbuf->st_size += size_inc; | |
66 | if (ftruncate(fd, sbuf->st_size)) { | |
67 | fprintf(stderr, "%s: Can't expand %s: %s\n", | |
68 | cmdname, fname, strerror(errno)); | |
69 | goto err; | |
70 | } | |
71 | } | |
72 | ||
6bf4ca07 HS |
73 | errno = 0; |
74 | ptr = mmap(0, sbuf->st_size, PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0); | |
75 | if ((ptr == MAP_FAILED) || (errno != 0)) { | |
76 | fprintf(stderr, "%s: Can't read %s: %s\n", | |
77 | cmdname, fname, strerror(errno)); | |
ef0af64b | 78 | goto err; |
6bf4ca07 HS |
79 | } |
80 | ||
81 | /* check if ptr has a valid blob */ | |
82 | if (fdt_check_header(ptr)) { | |
83 | fprintf(stderr, "%s: Invalid FIT blob\n", cmdname); | |
ef0af64b | 84 | goto err; |
6bf4ca07 HS |
85 | } |
86 | ||
a9468115 SG |
87 | /* expand if needed */ |
88 | if (size_inc) { | |
89 | int ret; | |
90 | ||
91 | ret = fdt_open_into(ptr, ptr, sbuf->st_size); | |
92 | if (ret) { | |
93 | fprintf(stderr, "%s: Cannot expand FDT: %s\n", | |
94 | cmdname, fdt_strerror(ret)); | |
95 | goto err; | |
96 | } | |
97 | } | |
98 | ||
6bf4ca07 HS |
99 | *blobp = ptr; |
100 | return fd; | |
ef0af64b SG |
101 | |
102 | err: | |
103 | if (fd >= 0) | |
104 | close(fd); | |
105 | if (delete_on_error) | |
106 | unlink(fname); | |
107 | ||
108 | return -1; | |
6bf4ca07 | 109 | } |