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