1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Copyright (C) 2018 Xilinx, Inc.
4 */
5
6 #include <common.h>
7 #include <log.h>
8 #include <part.h>
9 #include <asm/global_data.h>
10 #include <asm/io.h>
11 #include <asm/arch/hardware.h>
12 #include <asm/arch/sys_proto.h>
13 #include <u-boot/md5.h>
14 #include <zynq_bootimg.h>
15
16 DECLARE_GLOBAL_DATA_PTR;
17
18 #define ZYNQ_IMAGE_PHDR_OFFSET 0x09C
19 #define ZYNQ_IMAGE_FSBL_LEN_OFFSET 0x040
20 #define ZYNQ_PART_HDR_CHKSUM_WORD_COUNT 0x0F
21 #define ZYNQ_PART_HDR_WORD_COUNT 0x10
22 #define ZYNQ_MAXIMUM_IMAGE_WORD_LEN 0x40000000
23 #define MD5_CHECKSUM_SIZE 16
24
25 struct headerarray {
26 u32 fields[16];
27 };
28
29 /*
30 * Check whether the given partition is last partition or not
31 */
zynq_islastpartition(struct headerarray * head)32 static int zynq_islastpartition(struct headerarray *head)
33 {
34 int index;
35
36 debug("%s\n", __func__);
37 if (head->fields[ZYNQ_PART_HDR_CHKSUM_WORD_COUNT] != 0xFFFFFFFF)
38 return -1;
39
40 for (index = 0; index < ZYNQ_PART_HDR_WORD_COUNT - 1; index++) {
41 if (head->fields[index] != 0x0)
42 return -1;
43 }
44
45 return 0;
46 }
47
48 /*
49 * Get the partition count from the partition header
50 */
zynq_get_part_count(struct partition_hdr * part_hdr_info)51 int zynq_get_part_count(struct partition_hdr *part_hdr_info)
52 {
53 u32 count;
54 struct headerarray *hap;
55
56 debug("%s\n", __func__);
57
58 for (count = 0; count < ZYNQ_MAX_PARTITION_NUMBER; count++) {
59 hap = (struct headerarray *)&part_hdr_info[count];
60 if (zynq_islastpartition(hap) != -1)
61 break;
62 }
63
64 return count;
65 }
66
67 /*
68 * Get the partition info of all the partitions available.
69 */
zynq_get_partition_info(u32 image_base_addr,u32 * fsbl_len,struct partition_hdr * part_hdr)70 int zynq_get_partition_info(u32 image_base_addr, u32 *fsbl_len,
71 struct partition_hdr *part_hdr)
72 {
73 u32 parthdroffset;
74
75 *fsbl_len = *((u32 *)(image_base_addr + ZYNQ_IMAGE_FSBL_LEN_OFFSET));
76
77 parthdroffset = *((u32 *)(image_base_addr + ZYNQ_IMAGE_PHDR_OFFSET));
78
79 parthdroffset += image_base_addr;
80
81 memcpy(part_hdr, (u32 *)parthdroffset,
82 (sizeof(struct partition_hdr) * ZYNQ_MAX_PARTITION_NUMBER));
83
84 return 0;
85 }
86
87 /*
88 * Check whether the partition header is valid or not
89 */
zynq_validate_hdr(struct partition_hdr * header)90 int zynq_validate_hdr(struct partition_hdr *header)
91 {
92 struct headerarray *hap;
93 u32 index;
94 u32 checksum;
95
96 debug("%s\n", __func__);
97
98 hap = (struct headerarray *)header;
99
100 for (index = 0; index < ZYNQ_PART_HDR_WORD_COUNT; index++) {
101 if (hap->fields[index])
102 break;
103 }
104 if (index == ZYNQ_PART_HDR_WORD_COUNT)
105 return -1;
106
107 checksum = 0;
108 for (index = 0; index < ZYNQ_PART_HDR_CHKSUM_WORD_COUNT; index++)
109 checksum += hap->fields[index];
110
111 checksum ^= 0xFFFFFFFF;
112
113 if (hap->fields[ZYNQ_PART_HDR_CHKSUM_WORD_COUNT] != checksum) {
114 printf("Error: Checksum 0x%8.8x != 0x%8.8x\n",
115 checksum, hap->fields[ZYNQ_PART_HDR_CHKSUM_WORD_COUNT]);
116 return -1;
117 }
118
119 if (header->imagewordlen > ZYNQ_MAXIMUM_IMAGE_WORD_LEN) {
120 printf("INVALID_PARTITION_LENGTH\n");
121 return -1;
122 }
123
124 return 0;
125 }
126
127 /*
128 * Validate the partition by calculationg the md5 checksum for the
129 * partition and compare with checksum present in checksum offset of
130 * partition
131 */
zynq_validate_partition(u32 start_addr,u32 len,u32 chksum_off)132 int zynq_validate_partition(u32 start_addr, u32 len, u32 chksum_off)
133 {
134 u8 checksum[MD5_CHECKSUM_SIZE];
135 u8 calchecksum[MD5_CHECKSUM_SIZE];
136
137 memcpy(&checksum[0], (u32 *)chksum_off, MD5_CHECKSUM_SIZE);
138
139 md5_wd((u8 *)start_addr, len, &calchecksum[0], 0x10000);
140
141 if (!memcmp(checksum, calchecksum, MD5_CHECKSUM_SIZE))
142 return 0;
143
144 printf("Error: Partition DataChecksum\n");
145 return -1;
146 }
147