xref: /openbmc/u-boot/tools/rkspi.c (revision d5abcf94)
1 /*
2  * (C) Copyright 2015 Google,  Inc
3  * Written by Simon Glass <sjg@chromium.org>
4  *
5  * SPDX-License-Identifier:	GPL-2.0+
6  *
7  * See README.rockchip for details of the rkspi format
8  */
9 
10 #include "imagetool.h"
11 #include <image.h>
12 #include <rc4.h>
13 #include "mkimage.h"
14 #include "rkcommon.h"
15 
16 enum {
17 	RKSPI_SECT_LEN		= RK_BLK_SIZE * 4,
18 };
19 
20 static char dummy_hdr[RK_IMAGE_HEADER_LEN];
21 
22 static int rkspi_verify_header(unsigned char *buf, int size,
23 			       struct image_tool_params *params)
24 {
25 	return 0;
26 }
27 
28 static void rkspi_print_header(const void *buf)
29 {
30 }
31 
32 static void rkspi_set_header(void *buf, struct stat *sbuf, int ifd,
33 			     struct image_tool_params *params)
34 {
35 	int sector;
36 	unsigned int size;
37 	int ret;
38 
39 	size = params->orig_file_size;
40 	ret = rkcommon_set_header(buf, size, params);
41 	debug("size %x\n", size);
42 	if (ret) {
43 		/* TODO(sjg@chromium.org): This method should return an error */
44 		printf("Warning: SPL image is too large (size %#x) and will not boot\n",
45 		       size);
46 	}
47 
48 	memcpy(buf + RK_SPL_HDR_START, rkcommon_get_spl_hdr(params),
49 	       RK_SPL_HDR_SIZE);
50 
51 	if (rkcommon_need_rc4_spl(params))
52 		rkcommon_rc4_encode_spl(buf, RK_SPL_START - 4,
53 					params->file_size - RK_SPL_START + 4);
54 
55 	/*
56 	 * Spread the image out so we only use the first 2KB of each 4KB
57 	 * region. This is a feature of the SPI format required by the Rockchip
58 	 * boot ROM. Its rationale is unknown.
59 	 */
60 	for (sector = size / RKSPI_SECT_LEN - 1; sector >= 0; sector--) {
61 		debug("sector %u\n", sector);
62 		memmove(buf + sector * RKSPI_SECT_LEN * 2,
63 			buf + sector * RKSPI_SECT_LEN,
64 			RKSPI_SECT_LEN);
65 		memset(buf + sector * RKSPI_SECT_LEN * 2 + RKSPI_SECT_LEN,
66 		       '\0', RKSPI_SECT_LEN);
67 	}
68 }
69 
70 static int rkspi_extract_subimage(void *buf, struct image_tool_params *params)
71 {
72 	return 0;
73 }
74 
75 static int rkspi_check_image_type(uint8_t type)
76 {
77 	if (type == IH_TYPE_RKSPI)
78 		return EXIT_SUCCESS;
79 	else
80 		return EXIT_FAILURE;
81 }
82 
83 /* We pad the file out to a fixed size - this method returns that size */
84 static int rkspi_vrec_header(struct image_tool_params *params,
85 			     struct image_type_params *tparams)
86 {
87 	int pad_size;
88 
89 	pad_size = (rkcommon_get_spl_size(params) + 0x7ff) / 0x800 * 0x800;
90 	params->orig_file_size = pad_size;
91 
92 	/* We will double the image size due to the SPI format */
93 	pad_size *= 2;
94 	pad_size += RK_SPL_HDR_START;
95 	debug("pad_size %x\n", pad_size);
96 
97 	return pad_size - params->file_size;
98 }
99 
100 /*
101  * rk_spi parameters
102  */
103 U_BOOT_IMAGE_TYPE(
104 	rkspi,
105 	"Rockchip SPI Boot Image support",
106 	RK_IMAGE_HEADER_LEN,
107 	dummy_hdr,
108 	rkcommon_check_params,
109 	rkspi_verify_header,
110 	rkspi_print_header,
111 	rkspi_set_header,
112 	rkspi_extract_subimage,
113 	rkspi_check_image_type,
114 	NULL,
115 	rkspi_vrec_header
116 );
117