1 // SPDX-License-Identifier: GPL-2.0+ 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. 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 40 int mmap_fdt(const char *cmdname, const char *fname, size_t size_inc, 41 void **blobp, struct stat *sbuf, bool delete_on_error) 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)); 52 goto err; 53 } 54 55 if (fstat(fd, sbuf) < 0) { 56 fprintf(stderr, "%s: Can't stat %s: %s\n", 57 cmdname, fname, strerror(errno)); 58 goto err; 59 } 60 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 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)); 75 goto err; 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); 81 goto err; 82 } 83 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 96 *blobp = ptr; 97 return fd; 98 99 err: 100 if (fd >= 0) 101 close(fd); 102 if (delete_on_error) 103 unlink(fname); 104 105 return -1; 106 } 107