1 /* SPDX-License-Identifier: GPL-2.0+ */
2 /*
3  * (C) Copyright 2008
4  * Marvell Semiconductor <www.marvell.com>
5  * Written-by: Prafulla Wadaskar <prafulla@marvell.com>
6  */
7 
8 #ifndef _KWBIMAGE_H_
9 #define _KWBIMAGE_H_
10 
11 #include <compiler.h>
12 #include <stdint.h>
13 
14 #ifdef __GNUC__
15 #define __packed __attribute((packed))
16 #else
17 #define __packed
18 #endif
19 
20 #define KWBIMAGE_MAX_CONFIG	((0x1dc - 0x20)/sizeof(struct reg_config))
21 #define MAX_TEMPBUF_LEN		32
22 
23 /* NAND ECC Mode */
24 #define IBR_HDR_ECC_DEFAULT		0x00
25 #define IBR_HDR_ECC_FORCED_HAMMING	0x01
26 #define IBR_HDR_ECC_FORCED_RS		0x02
27 #define IBR_HDR_ECC_DISABLED		0x03
28 
29 /* Boot Type - block ID */
30 #define IBR_HDR_I2C_ID			0x4D
31 #define IBR_HDR_SPI_ID			0x5A
32 #define IBR_HDR_NAND_ID			0x8B
33 #define IBR_HDR_SATA_ID			0x78
34 #define IBR_HDR_PEX_ID			0x9C
35 #define IBR_HDR_UART_ID			0x69
36 #define IBR_HDR_SDIO_ID			0xAE
37 #define IBR_DEF_ATTRIB			0x00
38 
39 /* Structure of the main header, version 0 (Kirkwood, Dove) */
40 struct main_hdr_v0 {
41 	uint8_t  blockid;		/* 0x0       */
42 	uint8_t  nandeccmode;		/* 0x1       */
43 	uint16_t nandpagesize;		/* 0x2-0x3   */
44 	uint32_t blocksize;		/* 0x4-0x7   */
45 	uint32_t rsvd1;			/* 0x8-0xB   */
46 	uint32_t srcaddr;		/* 0xC-0xF   */
47 	uint32_t destaddr;		/* 0x10-0x13 */
48 	uint32_t execaddr;		/* 0x14-0x17 */
49 	uint8_t  satapiomode;		/* 0x18      */
50 	uint8_t  rsvd3;			/* 0x19      */
51 	uint16_t ddrinitdelay;		/* 0x1A-0x1B */
52 	uint16_t rsvd2;			/* 0x1C-0x1D */
53 	uint8_t  ext;			/* 0x1E      */
54 	uint8_t  checksum;		/* 0x1F      */
55 } __packed;
56 
57 struct ext_hdr_v0_reg {
58 	uint32_t raddr;
59 	uint32_t rdata;
60 } __packed;
61 
62 #define EXT_HDR_V0_REG_COUNT ((0x1dc - 0x20) / sizeof(struct ext_hdr_v0_reg))
63 
64 struct ext_hdr_v0 {
65 	uint32_t              offset;
66 	uint8_t               reserved[0x20 - sizeof(uint32_t)];
67 	struct ext_hdr_v0_reg rcfg[EXT_HDR_V0_REG_COUNT];
68 	uint8_t               reserved2[7];
69 	uint8_t               checksum;
70 } __packed;
71 
72 /* Structure of the main header, version 1 (Armada 370/XP/375/38x/39x) */
73 struct main_hdr_v1 {
74 	uint8_t  blockid;               /* 0x0       */
75 	uint8_t  flags;                 /* 0x1       */
76 	uint16_t nandpagesize;          /* 0x2-0x3   */
77 	uint32_t blocksize;             /* 0x4-0x7   */
78 	uint8_t  version;               /* 0x8       */
79 	uint8_t  headersz_msb;          /* 0x9       */
80 	uint16_t headersz_lsb;          /* 0xA-0xB   */
81 	uint32_t srcaddr;               /* 0xC-0xF   */
82 	uint32_t destaddr;              /* 0x10-0x13 */
83 	uint32_t execaddr;              /* 0x14-0x17 */
84 	uint8_t  options;               /* 0x18      */
85 	uint8_t  nandblocksize;         /* 0x19      */
86 	uint8_t  nandbadblklocation;    /* 0x1A      */
87 	uint8_t  reserved4;             /* 0x1B      */
88 	uint16_t reserved5;             /* 0x1C-0x1D */
89 	uint8_t  ext;                   /* 0x1E      */
90 	uint8_t  checksum;              /* 0x1F      */
91 } __packed;
92 
93 /*
94  * Main header options
95  */
96 #define MAIN_HDR_V1_OPT_BAUD_DEFAULT	0
97 #define MAIN_HDR_V1_OPT_BAUD_2400	0x1
98 #define MAIN_HDR_V1_OPT_BAUD_4800	0x2
99 #define MAIN_HDR_V1_OPT_BAUD_9600	0x3
100 #define MAIN_HDR_V1_OPT_BAUD_19200	0x4
101 #define MAIN_HDR_V1_OPT_BAUD_38400	0x5
102 #define MAIN_HDR_V1_OPT_BAUD_57600	0x6
103 #define MAIN_HDR_V1_OPT_BAUD_115200	0x7
104 
105 /*
106  * Header for the optional headers, version 1 (Armada 370/XP/375/38x/39x)
107  */
108 struct opt_hdr_v1 {
109 	uint8_t  headertype;
110 	uint8_t  headersz_msb;
111 	uint16_t headersz_lsb;
112 	char     data[0];
113 } __packed;
114 
115 /*
116  * Public Key data in DER format
117  */
118 struct pubkey_der_v1 {
119 	uint8_t key[524];
120 } __packed;
121 
122 /*
123  * Signature (RSA 2048)
124  */
125 struct sig_v1 {
126 	uint8_t sig[256];
127 } __packed;
128 
129 /*
130  * Structure of secure header (Armada XP/375/38x/39x)
131  */
132 struct secure_hdr_v1 {
133 	uint8_t  headertype;		/* 0x0 */
134 	uint8_t  headersz_msb;		/* 0x1 */
135 	uint16_t headersz_lsb;		/* 0x2 - 0x3 */
136 	uint32_t reserved1;		/* 0x4 - 0x7 */
137 	struct pubkey_der_v1 kak;	/* 0x8 - 0x213 */
138 	uint8_t  jtag_delay;		/* 0x214 */
139 	uint8_t  reserved2;		/* 0x215 */
140 	uint16_t reserved3;		/* 0x216 - 0x217 */
141 	uint32_t boxid;			/* 0x218 - 0x21B */
142 	uint32_t flashid;		/* 0x21C - 0x21F */
143 	struct sig_v1 hdrsig;		/* 0x220 - 0x31F */
144 	struct sig_v1 imgsig;		/* 0x320 - 0x41F */
145 	struct pubkey_der_v1 csk[16];	/* 0x420 - 0x24DF */
146 	struct sig_v1 csksig;		/* 0x24E0 - 0x25DF */
147 	uint8_t  next;			/* 0x25E0 */
148 	uint8_t  reserved4;		/* 0x25E1 */
149 	uint16_t reserved5;		/* 0x25E2 - 0x25E3 */
150 } __packed;
151 
152 /*
153  * Structure of register set
154  */
155 struct register_set_hdr_v1 {
156 	uint8_t  headertype;		/* 0x0 */
157 	uint8_t  headersz_msb;		/* 0x1 */
158 	uint16_t headersz_lsb;		/* 0x2 - 0x3 */
159 	union {
160 		struct {
161 			uint32_t address;	/* 0x4+8*N - 0x7+8*N */
162 			uint32_t value;		/* 0x8+8*N - 0xB+8*N */
163 		} __packed entry;
164 		struct {
165 			uint8_t  next;		/* 0xC+8*N */
166 			uint8_t  delay;		/* 0xD+8*N */
167 			uint16_t reserved;	/* 0xE+8*N - 0xF+8*N */
168 		} __packed last_entry;
169 	} data[];
170 } __packed;
171 
172 /*
173  * Value 0 in register_set_hdr_v1 delay field is special.
174  * Instead of delay it setup SDRAM Controller.
175  */
176 #define REGISTER_SET_HDR_OPT_DELAY_SDRAM_SETUP 0
177 #define REGISTER_SET_HDR_OPT_DELAY_MS(val) ((val) ?: 1)
178 
179 /*
180  * Various values for the opt_hdr_v1->headertype field, describing the
181  * different types of optional headers. The "secure" header contains
182  * informations related to secure boot (encryption keys, etc.). The
183  * "binary" header contains ARM binary code to be executed prior to
184  * executing the main payload (usually the bootloader). This is
185  * typically used to execute DDR3 training code. The "register" header
186  * allows to describe a set of (address, value) tuples that are
187  * generally used to configure the DRAM controller.
188  */
189 #define OPT_HDR_V1_SECURE_TYPE   0x1
190 #define OPT_HDR_V1_BINARY_TYPE   0x2
191 #define OPT_HDR_V1_REGISTER_TYPE 0x3
192 
193 enum kwbimage_cmd {
194 	CMD_INVALID,
195 	CMD_BOOT_FROM,
196 	CMD_NAND_ECC_MODE,
197 	CMD_NAND_PAGE_SIZE,
198 	CMD_SATA_PIO_MODE,
199 	CMD_DDR_INIT_DELAY,
200 	CMD_DATA
201 };
202 
203 enum kwbimage_cmd_types {
204 	CFG_INVALID = -1,
205 	CFG_COMMAND,
206 	CFG_DATA0,
207 	CFG_DATA1
208 };
209 
210 /*
211  * functions
212  */
213 void init_kwb_image_type (void);
214 
215 /*
216  * Byte 8 of the image header contains the version number. In the v0
217  * header, byte 8 was reserved, and always set to 0. In the v1 header,
218  * byte 8 has been changed to a proper field, set to 1.
219  */
kwbimage_version(const void * header)220 static inline unsigned int kwbimage_version(const void *header)
221 {
222 	const unsigned char *ptr = header;
223 	return ptr[8];
224 }
225 
kwbheader_size(const void * header)226 static inline size_t kwbheader_size(const void *header)
227 {
228 	if (kwbimage_version(header) == 0) {
229 		const struct main_hdr_v0 *hdr = header;
230 
231 		return sizeof(*hdr) +
232 		       (hdr->ext & 0x1) ? sizeof(struct ext_hdr_v0) : 0;
233 	} else {
234 		const struct main_hdr_v1 *hdr = header;
235 
236 		return (hdr->headersz_msb << 16) |
237 		       le16_to_cpu(hdr->headersz_lsb);
238 	}
239 }
240 
kwbheader_size_for_csum(const void * header)241 static inline size_t kwbheader_size_for_csum(const void *header)
242 {
243 	if (kwbimage_version(header) == 0)
244 		return sizeof(struct main_hdr_v0);
245 	else
246 		return kwbheader_size(header);
247 }
248 
opt_hdr_v1_size(const struct opt_hdr_v1 * ohdr)249 static inline uint32_t opt_hdr_v1_size(const struct opt_hdr_v1 *ohdr)
250 {
251 	return (ohdr->headersz_msb << 16) | le16_to_cpu(ohdr->headersz_lsb);
252 }
253 
opt_hdr_v1_valid_size(const struct opt_hdr_v1 * ohdr,const void * mhdr_end)254 static inline int opt_hdr_v1_valid_size(const struct opt_hdr_v1 *ohdr,
255 					const void *mhdr_end)
256 {
257 	uint32_t ohdr_size;
258 
259 	if ((void *)(ohdr + 1) > mhdr_end)
260 		return 0;
261 
262 	ohdr_size = opt_hdr_v1_size(ohdr);
263 	if (ohdr_size < 8 || (void *)((uint8_t *)ohdr + ohdr_size) > mhdr_end)
264 		return 0;
265 
266 	return 1;
267 }
268 
opt_hdr_v1_first(void * img)269 static inline struct opt_hdr_v1 *opt_hdr_v1_first(void *img) {
270 	struct main_hdr_v1 *mhdr;
271 
272 	if (kwbimage_version(img) != 1)
273 		return NULL;
274 
275 	mhdr = img;
276 	if (mhdr->ext & 0x1)
277 		return (struct opt_hdr_v1 *)(mhdr + 1);
278 	else
279 		return NULL;
280 }
281 
opt_hdr_v1_ext(struct opt_hdr_v1 * cur)282 static inline uint8_t *opt_hdr_v1_ext(struct opt_hdr_v1 *cur)
283 {
284 	uint32_t size = opt_hdr_v1_size(cur);
285 
286 	return (uint8_t *)cur + size - 4;
287 }
288 
_opt_hdr_v1_next(struct opt_hdr_v1 * cur)289 static inline struct opt_hdr_v1 *_opt_hdr_v1_next(struct opt_hdr_v1 *cur)
290 {
291 	return (struct opt_hdr_v1 *)((uint8_t *)cur + opt_hdr_v1_size(cur));
292 }
293 
opt_hdr_v1_next(struct opt_hdr_v1 * cur)294 static inline struct opt_hdr_v1 *opt_hdr_v1_next(struct opt_hdr_v1 *cur)
295 {
296 	if (*opt_hdr_v1_ext(cur) & 0x1)
297 		return _opt_hdr_v1_next(cur);
298 	else
299 		return NULL;
300 }
301 
302 #define for_each_opt_hdr_v1(ohdr, img)		\
303 	for ((ohdr) = opt_hdr_v1_first((img));	\
304 	     (ohdr) != NULL;			\
305 	     (ohdr) = opt_hdr_v1_next((ohdr)))
306 
307 #endif /* _KWBIMAGE_H_ */
308