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