]>
Commit | Line | Data |
---|---|---|
1 | /** @file\r | |
2 | FFS file access utilities.\r | |
3 | \r | |
4 | Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved.<BR>\r | |
5 | SPDX-License-Identifier: BSD-2-Clause-Patent\r | |
6 | \r | |
7 | **/\r | |
8 | \r | |
9 | #include "DxeMain.h"\r | |
10 | #include "FwVolDriver.h"\r | |
11 | \r | |
12 | /**\r | |
13 | Get the FFS file state by checking the highest bit set in the header's state field.\r | |
14 | \r | |
15 | @param ErasePolarity Erase polarity attribute of the firmware volume\r | |
16 | @param FfsHeader Points to the FFS file header\r | |
17 | \r | |
18 | @return FFS File state\r | |
19 | \r | |
20 | **/\r | |
21 | EFI_FFS_FILE_STATE\r | |
22 | GetFileState (\r | |
23 | IN UINT8 ErasePolarity,\r | |
24 | IN EFI_FFS_FILE_HEADER *FfsHeader\r | |
25 | )\r | |
26 | {\r | |
27 | EFI_FFS_FILE_STATE FileState;\r | |
28 | UINT8 HighestBit;\r | |
29 | \r | |
30 | FileState = FfsHeader->State;\r | |
31 | \r | |
32 | if (ErasePolarity != 0) {\r | |
33 | FileState = (EFI_FFS_FILE_STATE) ~FileState;\r | |
34 | }\r | |
35 | \r | |
36 | HighestBit = 0x80;\r | |
37 | while (HighestBit != 0 && ((HighestBit & FileState) == 0)) {\r | |
38 | HighestBit >>= 1;\r | |
39 | }\r | |
40 | \r | |
41 | return (EFI_FFS_FILE_STATE)HighestBit;\r | |
42 | }\r | |
43 | \r | |
44 | /**\r | |
45 | Check if a block of buffer is erased.\r | |
46 | \r | |
47 | @param ErasePolarity Erase polarity attribute of the firmware volume\r | |
48 | @param InBuffer The buffer to be checked\r | |
49 | @param BufferSize Size of the buffer in bytes\r | |
50 | \r | |
51 | @retval TRUE The block of buffer is erased\r | |
52 | @retval FALSE The block of buffer is not erased\r | |
53 | \r | |
54 | **/\r | |
55 | BOOLEAN\r | |
56 | IsBufferErased (\r | |
57 | IN UINT8 ErasePolarity,\r | |
58 | IN VOID *InBuffer,\r | |
59 | IN UINTN BufferSize\r | |
60 | )\r | |
61 | {\r | |
62 | UINTN Count;\r | |
63 | UINT8 EraseByte;\r | |
64 | UINT8 *Buffer;\r | |
65 | \r | |
66 | if (ErasePolarity == 1) {\r | |
67 | EraseByte = 0xFF;\r | |
68 | } else {\r | |
69 | EraseByte = 0;\r | |
70 | }\r | |
71 | \r | |
72 | Buffer = InBuffer;\r | |
73 | for (Count = 0; Count < BufferSize; Count++) {\r | |
74 | if (Buffer[Count] != EraseByte) {\r | |
75 | return FALSE;\r | |
76 | }\r | |
77 | }\r | |
78 | \r | |
79 | return TRUE;\r | |
80 | }\r | |
81 | \r | |
82 | /**\r | |
83 | Verify checksum of the firmware volume header.\r | |
84 | \r | |
85 | @param FvHeader Points to the firmware volume header to be checked\r | |
86 | \r | |
87 | @retval TRUE Checksum verification passed\r | |
88 | @retval FALSE Checksum verification failed\r | |
89 | \r | |
90 | **/\r | |
91 | BOOLEAN\r | |
92 | VerifyFvHeaderChecksum (\r | |
93 | IN EFI_FIRMWARE_VOLUME_HEADER *FvHeader\r | |
94 | )\r | |
95 | {\r | |
96 | UINT16 Checksum;\r | |
97 | \r | |
98 | Checksum = CalculateSum16 ((UINT16 *)FvHeader, FvHeader->HeaderLength);\r | |
99 | \r | |
100 | if (Checksum == 0) {\r | |
101 | return TRUE;\r | |
102 | } else {\r | |
103 | return FALSE;\r | |
104 | }\r | |
105 | }\r | |
106 | \r | |
107 | /**\r | |
108 | Verify checksum of the FFS file header.\r | |
109 | \r | |
110 | @param FfsHeader Points to the FFS file header to be checked\r | |
111 | \r | |
112 | @retval TRUE Checksum verification passed\r | |
113 | @retval FALSE Checksum verification failed\r | |
114 | \r | |
115 | **/\r | |
116 | BOOLEAN\r | |
117 | VerifyHeaderChecksum (\r | |
118 | IN EFI_FFS_FILE_HEADER *FfsHeader\r | |
119 | )\r | |
120 | {\r | |
121 | UINT8 HeaderChecksum;\r | |
122 | \r | |
123 | if (IS_FFS_FILE2 (FfsHeader)) {\r | |
124 | HeaderChecksum = CalculateSum8 ((UINT8 *)FfsHeader, sizeof (EFI_FFS_FILE_HEADER2));\r | |
125 | } else {\r | |
126 | HeaderChecksum = CalculateSum8 ((UINT8 *)FfsHeader, sizeof (EFI_FFS_FILE_HEADER));\r | |
127 | }\r | |
128 | \r | |
129 | HeaderChecksum = (UINT8)(HeaderChecksum - FfsHeader->State - FfsHeader->IntegrityCheck.Checksum.File);\r | |
130 | \r | |
131 | if (HeaderChecksum == 0) {\r | |
132 | return TRUE;\r | |
133 | } else {\r | |
134 | return FALSE;\r | |
135 | }\r | |
136 | }\r | |
137 | \r | |
138 | /**\r | |
139 | Check if it's a valid FFS file header.\r | |
140 | \r | |
141 | @param ErasePolarity Erase polarity attribute of the firmware volume\r | |
142 | @param FfsHeader Points to the FFS file header to be checked\r | |
143 | @param FileState FFS file state to be returned\r | |
144 | \r | |
145 | @retval TRUE Valid FFS file header\r | |
146 | @retval FALSE Invalid FFS file header\r | |
147 | \r | |
148 | **/\r | |
149 | BOOLEAN\r | |
150 | IsValidFfsHeader (\r | |
151 | IN UINT8 ErasePolarity,\r | |
152 | IN EFI_FFS_FILE_HEADER *FfsHeader,\r | |
153 | OUT EFI_FFS_FILE_STATE *FileState\r | |
154 | )\r | |
155 | {\r | |
156 | *FileState = GetFileState (ErasePolarity, FfsHeader);\r | |
157 | \r | |
158 | switch (*FileState) {\r | |
159 | case EFI_FILE_HEADER_VALID:\r | |
160 | case EFI_FILE_DATA_VALID:\r | |
161 | case EFI_FILE_MARKED_FOR_UPDATE:\r | |
162 | case EFI_FILE_DELETED:\r | |
163 | //\r | |
164 | // Here we need to verify header checksum\r | |
165 | //\r | |
166 | return VerifyHeaderChecksum (FfsHeader);\r | |
167 | \r | |
168 | case EFI_FILE_HEADER_CONSTRUCTION:\r | |
169 | case EFI_FILE_HEADER_INVALID:\r | |
170 | default:\r | |
171 | return FALSE;\r | |
172 | }\r | |
173 | }\r | |
174 | \r | |
175 | /**\r | |
176 | Check if it's a valid FFS file.\r | |
177 | Here we are sure that it has a valid FFS file header since we must call IsValidFfsHeader() first.\r | |
178 | \r | |
179 | @param ErasePolarity Erase polarity attribute of the firmware volume\r | |
180 | @param FfsHeader Points to the FFS file to be checked\r | |
181 | \r | |
182 | @retval TRUE Valid FFS file\r | |
183 | @retval FALSE Invalid FFS file\r | |
184 | \r | |
185 | **/\r | |
186 | BOOLEAN\r | |
187 | IsValidFfsFile (\r | |
188 | IN UINT8 ErasePolarity,\r | |
189 | IN EFI_FFS_FILE_HEADER *FfsHeader\r | |
190 | )\r | |
191 | {\r | |
192 | EFI_FFS_FILE_STATE FileState;\r | |
193 | UINT8 DataCheckSum;\r | |
194 | \r | |
195 | FileState = GetFileState (ErasePolarity, FfsHeader);\r | |
196 | switch (FileState) {\r | |
197 | case EFI_FILE_DELETED:\r | |
198 | case EFI_FILE_DATA_VALID:\r | |
199 | case EFI_FILE_MARKED_FOR_UPDATE:\r | |
200 | DataCheckSum = FFS_FIXED_CHECKSUM;\r | |
201 | if ((FfsHeader->Attributes & FFS_ATTRIB_CHECKSUM) == FFS_ATTRIB_CHECKSUM) {\r | |
202 | if (IS_FFS_FILE2 (FfsHeader)) {\r | |
203 | DataCheckSum = CalculateCheckSum8 ((CONST UINT8 *)FfsHeader + sizeof (EFI_FFS_FILE_HEADER2), FFS_FILE2_SIZE (FfsHeader) - sizeof (EFI_FFS_FILE_HEADER2));\r | |
204 | } else {\r | |
205 | DataCheckSum = CalculateCheckSum8 ((CONST UINT8 *)FfsHeader + sizeof (EFI_FFS_FILE_HEADER), FFS_FILE_SIZE (FfsHeader) - sizeof (EFI_FFS_FILE_HEADER));\r | |
206 | }\r | |
207 | }\r | |
208 | \r | |
209 | if (FfsHeader->IntegrityCheck.Checksum.File == DataCheckSum) {\r | |
210 | return TRUE;\r | |
211 | }\r | |
212 | \r | |
213 | default:\r | |
214 | return FALSE;\r | |
215 | }\r | |
216 | }\r |