1*83d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+ 2e183a174SMasahiro Yamada /* 3e183a174SMasahiro Yamada * Copyright (C) 2011 Samsung Electronics 4e183a174SMasahiro Yamada */ 5e183a174SMasahiro Yamada 6e183a174SMasahiro Yamada #include <stdio.h> 7e183a174SMasahiro Yamada #include <stdlib.h> 8e183a174SMasahiro Yamada #include <unistd.h> 9e183a174SMasahiro Yamada #include <fcntl.h> 10e183a174SMasahiro Yamada #include <errno.h> 11e183a174SMasahiro Yamada #include <string.h> 12e183a174SMasahiro Yamada #include <sys/stat.h> 13e183a174SMasahiro Yamada 14e183a174SMasahiro Yamada #define CHECKSUM_OFFSET (14*1024-4) 15e183a174SMasahiro Yamada #define BUFSIZE (16*1024) 16e183a174SMasahiro Yamada #define FILE_PERM (S_IRUSR | S_IWUSR | S_IRGRP \ 17e183a174SMasahiro Yamada | S_IWGRP | S_IROTH | S_IWOTH) 18e183a174SMasahiro Yamada /* 19e183a174SMasahiro Yamada * Requirement: 20e183a174SMasahiro Yamada * IROM code reads first 14K bytes from boot device. 21e183a174SMasahiro Yamada * It then calculates the checksum of 14K-4 bytes and compare with data at 22e183a174SMasahiro Yamada * 14K-4 offset. 23e183a174SMasahiro Yamada * 24e183a174SMasahiro Yamada * This function takes two filenames: 25e183a174SMasahiro Yamada * IN "u-boot-spl.bin" and 26e183a174SMasahiro Yamada * OUT "u-boot-mmc-spl.bin" as filenames. 27e183a174SMasahiro Yamada * It reads the "u-boot-spl.bin" in 16K buffer. 28e183a174SMasahiro Yamada * It calculates checksum of 14K-4 Bytes and stores at 14K-4 offset in buffer. 29e183a174SMasahiro Yamada * It writes the buffer to "u-boot-mmc-spl.bin" file. 30e183a174SMasahiro Yamada */ 31e183a174SMasahiro Yamada 32e183a174SMasahiro Yamada int main(int argc, char **argv) 33e183a174SMasahiro Yamada { 34e183a174SMasahiro Yamada int i, len; 35e183a174SMasahiro Yamada unsigned char buffer[BUFSIZE] = {0}; 36e183a174SMasahiro Yamada int ifd, ofd; 37e183a174SMasahiro Yamada unsigned int checksum = 0, count; 38e183a174SMasahiro Yamada 39e183a174SMasahiro Yamada if (argc != 3) { 40e183a174SMasahiro Yamada printf(" %d Wrong number of arguments\n", argc); 41e183a174SMasahiro Yamada exit(EXIT_FAILURE); 42e183a174SMasahiro Yamada } 43e183a174SMasahiro Yamada 44e183a174SMasahiro Yamada ifd = open(argv[1], O_RDONLY); 45e183a174SMasahiro Yamada if (ifd < 0) { 46e183a174SMasahiro Yamada fprintf(stderr, "%s: Can't open %s: %s\n", 47e183a174SMasahiro Yamada argv[0], argv[1], strerror(errno)); 48e183a174SMasahiro Yamada exit(EXIT_FAILURE); 49e183a174SMasahiro Yamada } 50e183a174SMasahiro Yamada 51e183a174SMasahiro Yamada ofd = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, FILE_PERM); 52310ae37eSThomas Huth if (ofd < 0) { 53e183a174SMasahiro Yamada fprintf(stderr, "%s: Can't open %s: %s\n", 54e183a174SMasahiro Yamada argv[0], argv[2], strerror(errno)); 55e183a174SMasahiro Yamada if (ifd) 56e183a174SMasahiro Yamada close(ifd); 57e183a174SMasahiro Yamada exit(EXIT_FAILURE); 58e183a174SMasahiro Yamada } 59e183a174SMasahiro Yamada 60e183a174SMasahiro Yamada len = lseek(ifd, 0, SEEK_END); 61e183a174SMasahiro Yamada lseek(ifd, 0, SEEK_SET); 62e183a174SMasahiro Yamada 63e183a174SMasahiro Yamada count = (len < CHECKSUM_OFFSET) ? len : CHECKSUM_OFFSET; 64e183a174SMasahiro Yamada 65e183a174SMasahiro Yamada if (read(ifd, buffer, count) != count) { 66e183a174SMasahiro Yamada fprintf(stderr, "%s: Can't read %s: %s\n", 67e183a174SMasahiro Yamada argv[0], argv[1], strerror(errno)); 68e183a174SMasahiro Yamada 69e183a174SMasahiro Yamada if (ifd) 70e183a174SMasahiro Yamada close(ifd); 71e183a174SMasahiro Yamada if (ofd) 72e183a174SMasahiro Yamada close(ofd); 73e183a174SMasahiro Yamada 74e183a174SMasahiro Yamada exit(EXIT_FAILURE); 75e183a174SMasahiro Yamada } 76e183a174SMasahiro Yamada 77e183a174SMasahiro Yamada for (i = 0, checksum = 0; i < CHECKSUM_OFFSET; i++) 78e183a174SMasahiro Yamada checksum += buffer[i]; 79e183a174SMasahiro Yamada 80e183a174SMasahiro Yamada memcpy(&buffer[CHECKSUM_OFFSET], &checksum, sizeof(checksum)); 81e183a174SMasahiro Yamada 82e183a174SMasahiro Yamada if (write(ofd, buffer, BUFSIZE) != BUFSIZE) { 83e183a174SMasahiro Yamada fprintf(stderr, "%s: Can't write %s: %s\n", 84e183a174SMasahiro Yamada argv[0], argv[2], strerror(errno)); 85e183a174SMasahiro Yamada 86e183a174SMasahiro Yamada if (ifd) 87e183a174SMasahiro Yamada close(ifd); 88e183a174SMasahiro Yamada if (ofd) 89e183a174SMasahiro Yamada close(ofd); 90e183a174SMasahiro Yamada 91e183a174SMasahiro Yamada exit(EXIT_FAILURE); 92e183a174SMasahiro Yamada } 93e183a174SMasahiro Yamada 94e183a174SMasahiro Yamada if (ifd) 95e183a174SMasahiro Yamada close(ifd); 96e183a174SMasahiro Yamada if (ofd) 97e183a174SMasahiro Yamada close(ofd); 98e183a174SMasahiro Yamada 99e183a174SMasahiro Yamada return EXIT_SUCCESS; 100e183a174SMasahiro Yamada } 101