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