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