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