xref: /openbmc/u-boot/tools/stm32image.c (revision e8f80a5a)
1 // SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause
2 /*
3  * Copyright (C) 2018, STMicroelectronics - All Rights Reserved
4  */
5 
6 #include <image.h>
7 #include "imagetool.h"
8 
9 /* magic ='S' 'T' 'M' 0x32 */
10 #define HEADER_MAGIC be32_to_cpu(0x53544D32)
11 #define VER_MAJOR_IDX	2
12 #define VER_MINOR_IDX	1
13 #define VER_VARIANT_IDX	0
14 #define HEADER_VERSION_V1	0x1
15 /* default option : bit0 => no signature */
16 #define HEADER_DEFAULT_OPTION	(cpu_to_le32(0x00000001))
17 
18 struct stm32_header {
19 	uint32_t magic_number;
20 	uint32_t image_signature[64 / 4];
21 	uint32_t image_checksum;
22 	uint8_t  header_version[4];
23 	uint32_t image_length;
24 	uint32_t image_entry_point;
25 	uint32_t reserved1;
26 	uint32_t load_address;
27 	uint32_t reserved2;
28 	uint32_t version_number;
29 	uint32_t option_flags;
30 	uint32_t ecdsa_algorithm;
31 	uint32_t ecdsa_public_key[64 / 4];
32 	uint32_t padding[84 / 4];
33 };
34 
35 static struct stm32_header stm32image_header;
36 
stm32image_default_header(struct stm32_header * ptr)37 static void stm32image_default_header(struct stm32_header *ptr)
38 {
39 	if (!ptr)
40 		return;
41 
42 	ptr->magic_number = HEADER_MAGIC;
43 	ptr->header_version[VER_MAJOR_IDX] = HEADER_VERSION_V1;
44 	ptr->option_flags = HEADER_DEFAULT_OPTION;
45 	ptr->ecdsa_algorithm = 1;
46 }
47 
stm32image_checksum(void * start,uint32_t len)48 static uint32_t stm32image_checksum(void *start, uint32_t len)
49 {
50 	uint32_t csum = 0;
51 	uint32_t hdr_len = sizeof(struct stm32_header);
52 	uint8_t *p;
53 
54 	if (len < hdr_len)
55 		return 0;
56 
57 	p = start + hdr_len;
58 	len -= hdr_len;
59 
60 	while (len > 0) {
61 		csum += *p;
62 		p++;
63 		len--;
64 	}
65 
66 	return csum;
67 }
68 
stm32image_check_image_types(uint8_t type)69 static int stm32image_check_image_types(uint8_t type)
70 {
71 	if (type == IH_TYPE_STM32IMAGE)
72 		return EXIT_SUCCESS;
73 	return EXIT_FAILURE;
74 }
75 
stm32image_verify_header(unsigned char * ptr,int image_size,struct image_tool_params * params)76 static int stm32image_verify_header(unsigned char *ptr, int image_size,
77 				    struct image_tool_params *params)
78 {
79 	struct stm32_header *stm32hdr = (struct stm32_header *)ptr;
80 	int i;
81 
82 	if (image_size < sizeof(struct stm32_header))
83 		return -1;
84 	if (stm32hdr->magic_number != HEADER_MAGIC)
85 		return -1;
86 	if (stm32hdr->header_version[VER_MAJOR_IDX] != HEADER_VERSION_V1)
87 		return -1;
88 	if (stm32hdr->reserved1 || stm32hdr->reserved2)
89 		return -1;
90 	for (i = 0; i < (sizeof(stm32hdr->padding) / 4); i++) {
91 		if (stm32hdr->padding[i] != 0)
92 			return -1;
93 	}
94 
95 	return 0;
96 }
97 
stm32image_print_header(const void * ptr)98 static void stm32image_print_header(const void *ptr)
99 {
100 	struct stm32_header *stm32hdr = (struct stm32_header *)ptr;
101 
102 	printf("Image Type   : STMicroelectronics STM32 V%d.%d\n",
103 	       stm32hdr->header_version[VER_MAJOR_IDX],
104 	       stm32hdr->header_version[VER_MINOR_IDX]);
105 	printf("Image Size   : %lu bytes\n",
106 	       (unsigned long)le32_to_cpu(stm32hdr->image_length));
107 	printf("Image Load   : 0x%08x\n",
108 	       le32_to_cpu(stm32hdr->load_address));
109 	printf("Entry Point  : 0x%08x\n",
110 	       le32_to_cpu(stm32hdr->image_entry_point));
111 	printf("Checksum     : 0x%08x\n",
112 	       le32_to_cpu(stm32hdr->image_checksum));
113 	printf("Option     : 0x%08x\n",
114 	       le32_to_cpu(stm32hdr->option_flags));
115 }
116 
stm32image_set_header(void * ptr,struct stat * sbuf,int ifd,struct image_tool_params * params)117 static void stm32image_set_header(void *ptr, struct stat *sbuf, int ifd,
118 				  struct image_tool_params *params)
119 {
120 	struct stm32_header *stm32hdr = (struct stm32_header *)ptr;
121 
122 	stm32image_default_header(stm32hdr);
123 
124 	stm32hdr->load_address = cpu_to_le32(params->addr);
125 	stm32hdr->image_entry_point = cpu_to_le32(params->ep);
126 	stm32hdr->image_length = cpu_to_le32((uint32_t)sbuf->st_size -
127 					     sizeof(struct stm32_header));
128 	stm32hdr->image_checksum = stm32image_checksum(ptr, sbuf->st_size);
129 }
130 
131 /*
132  * stm32image parameters
133  */
134 U_BOOT_IMAGE_TYPE(
135 	stm32image,
136 	"STMicroelectronics STM32MP Image support",
137 	sizeof(struct stm32_header),
138 	(void *)&stm32image_header,
139 	NULL,
140 	stm32image_verify_header,
141 	stm32image_print_header,
142 	stm32image_set_header,
143 	NULL,
144 	stm32image_check_image_types,
145 	NULL,
146 	NULL
147 );
148