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 BUFSIZE			(16*1024)
15e183a174SMasahiro Yamada #define IMG_SIZE		(16*1024)
16e183a174SMasahiro Yamada #define SPL_HEADER_SIZE		16
17e183a174SMasahiro Yamada #define FILE_PERM		(S_IRUSR | S_IWUSR | S_IRGRP \
18e183a174SMasahiro Yamada 				| S_IWGRP | S_IROTH | S_IWOTH)
19e183a174SMasahiro Yamada #define SPL_HEADER		"S5PC210 HEADER  "
20e183a174SMasahiro Yamada /*
21e183a174SMasahiro Yamada * Requirement:
22e183a174SMasahiro Yamada * IROM code reads first 14K bytes from boot device.
23e183a174SMasahiro Yamada * It then calculates the checksum of 14K-4 bytes and compare with data at
24e183a174SMasahiro Yamada * 14K-4 offset.
25e183a174SMasahiro Yamada *
26e183a174SMasahiro Yamada * This function takes two filenames:
27e183a174SMasahiro Yamada * IN  "u-boot-spl.bin" and
28e183a174SMasahiro Yamada * OUT "$(BOARD)-spl.bin as filenames.
29e183a174SMasahiro Yamada * It reads the "u-boot-spl.bin" in 16K buffer.
30e183a174SMasahiro Yamada * It calculates checksum of 14K-4 Bytes and stores at 14K-4 offset in buffer.
31e183a174SMasahiro Yamada * It writes the buffer to "$(BOARD)-spl.bin" file.
32e183a174SMasahiro Yamada */
33e183a174SMasahiro Yamada 
main(int argc,char ** argv)34e183a174SMasahiro Yamada int main(int argc, char **argv)
35e183a174SMasahiro Yamada {
36e183a174SMasahiro Yamada 	int i, len;
37e183a174SMasahiro Yamada 	unsigned char buffer[BUFSIZE] = {0};
38e183a174SMasahiro Yamada 	int ifd, ofd;
39e183a174SMasahiro Yamada 	unsigned int checksum = 0, count;
40e183a174SMasahiro Yamada 
41e183a174SMasahiro Yamada 	if (argc != 3) {
42e183a174SMasahiro Yamada 		printf(" %d Wrong number of arguments\n", argc);
43e183a174SMasahiro Yamada 		exit(EXIT_FAILURE);
44e183a174SMasahiro Yamada 	}
45e183a174SMasahiro Yamada 
46e183a174SMasahiro Yamada 	ifd = open(argv[1], O_RDONLY);
47e183a174SMasahiro Yamada 	if (ifd < 0) {
48e183a174SMasahiro Yamada 		fprintf(stderr, "%s: Can't open %s: %s\n",
49e183a174SMasahiro Yamada 			argv[0], argv[1], strerror(errno));
50e183a174SMasahiro Yamada 		exit(EXIT_FAILURE);
51e183a174SMasahiro Yamada 	}
52e183a174SMasahiro Yamada 
53e183a174SMasahiro Yamada 	ofd = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, FILE_PERM);
54310ae37eSThomas Huth 	if (ofd < 0) {
55e183a174SMasahiro Yamada 		fprintf(stderr, "%s: Can't open %s: %s\n",
56e183a174SMasahiro Yamada 			argv[0], argv[2], strerror(errno));
57e183a174SMasahiro Yamada 		if (ifd)
58e183a174SMasahiro Yamada 			close(ifd);
59e183a174SMasahiro Yamada 		exit(EXIT_FAILURE);
60e183a174SMasahiro Yamada 	}
61e183a174SMasahiro Yamada 
62e183a174SMasahiro Yamada 	len = lseek(ifd, 0, SEEK_END);
63e183a174SMasahiro Yamada 	lseek(ifd, 0, SEEK_SET);
64e183a174SMasahiro Yamada 
65e183a174SMasahiro Yamada 	memcpy(&buffer[0], SPL_HEADER, SPL_HEADER_SIZE);
66e183a174SMasahiro Yamada 
67e183a174SMasahiro Yamada 	count = (len < (IMG_SIZE - SPL_HEADER_SIZE))
68e183a174SMasahiro Yamada 		? len : (IMG_SIZE - SPL_HEADER_SIZE);
69e183a174SMasahiro Yamada 
70e183a174SMasahiro Yamada 	if (read(ifd, buffer + SPL_HEADER_SIZE, count) != count) {
71e183a174SMasahiro Yamada 		fprintf(stderr, "%s: Can't read %s: %s\n",
72e183a174SMasahiro Yamada 			argv[0], argv[1], strerror(errno));
73e183a174SMasahiro Yamada 
74e183a174SMasahiro Yamada 		if (ifd)
75e183a174SMasahiro Yamada 			close(ifd);
76e183a174SMasahiro Yamada 		if (ofd)
77e183a174SMasahiro Yamada 			close(ofd);
78e183a174SMasahiro Yamada 
79e183a174SMasahiro Yamada 		exit(EXIT_FAILURE);
80e183a174SMasahiro Yamada 	}
81e183a174SMasahiro Yamada 
82e183a174SMasahiro Yamada 	for (i = 0; i < IMG_SIZE - SPL_HEADER_SIZE; i++)
83e183a174SMasahiro Yamada 		checksum += buffer[i+16];
84e183a174SMasahiro Yamada 
852e33df80SAndreas Bießmann 	*(unsigned long *)buffer ^= 0x1f;
862e33df80SAndreas Bießmann 	*(unsigned long *)(buffer+4) ^= checksum;
87e183a174SMasahiro Yamada 
88e183a174SMasahiro Yamada 	for (i = 1; i < SPL_HEADER_SIZE; i++)
89e183a174SMasahiro Yamada 		buffer[i] ^= buffer[i-1];
90e183a174SMasahiro Yamada 
91e183a174SMasahiro Yamada 	if (write(ofd, buffer, BUFSIZE) != BUFSIZE) {
92e183a174SMasahiro Yamada 		fprintf(stderr, "%s: Can't write %s: %s\n",
93e183a174SMasahiro Yamada 			argv[0], argv[2], strerror(errno));
94e183a174SMasahiro Yamada 
95e183a174SMasahiro Yamada 		if (ifd)
96e183a174SMasahiro Yamada 			close(ifd);
97e183a174SMasahiro Yamada 		if (ofd)
98e183a174SMasahiro Yamada 			close(ofd);
99e183a174SMasahiro Yamada 
100e183a174SMasahiro Yamada 		exit(EXIT_FAILURE);
101e183a174SMasahiro Yamada 	}
102e183a174SMasahiro Yamada 
103e183a174SMasahiro Yamada 	if (ifd)
104e183a174SMasahiro Yamada 		close(ifd);
105e183a174SMasahiro Yamada 	if (ofd)
106e183a174SMasahiro Yamada 		close(ofd);
107e183a174SMasahiro Yamada 
108e183a174SMasahiro Yamada 	return EXIT_SUCCESS;
109e183a174SMasahiro Yamada }
110