1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
/*
* xen/common/efi/pe.c
*
* PE executable header parser.
*
* Derived from https://github.com/systemd/systemd/blob/master/src/boot/efi/pe.c
* commit 07d5ed536ec0a76b08229c7a80b910cb9acaf6b1
*
* Copyright (C) 2015 Kay Sievers <kay@vrfy.org>
* Copyright (C) 2020 Trammell Hudson <hudson@trmm.net>
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*/
#include "efi.h"
struct DosFileHeader {
UINT8 Magic[2];
UINT16 LastSize;
UINT16 nBlocks;
UINT16 nReloc;
UINT16 HdrSize;
UINT16 MinAlloc;
UINT16 MaxAlloc;
UINT16 ss;
UINT16 sp;
UINT16 Checksum;
UINT16 ip;
UINT16 cs;
UINT16 RelocPos;
UINT16 nOverlay;
UINT16 reserved[4];
UINT16 OEMId;
UINT16 OEMInfo;
UINT16 reserved2[10];
UINT32 ExeHeader;
};
#if defined(__arm__) || defined (__aarch64__)
#define PE_HEADER_MACHINE 0xaa64
#elif defined(__x86_64__)
#define PE_HEADER_MACHINE 0x8664
#else
#error "Unknown architecture"
#endif
struct PeFileHeader {
UINT16 Machine;
UINT16 NumberOfSections;
UINT32 TimeDateStamp;
UINT32 PointerToSymbolTable;
UINT32 NumberOfSymbols;
UINT16 SizeOfOptionalHeader;
UINT16 Characteristics;
};
struct PeHeader {
UINT8 Magic[4];
struct PeFileHeader FileHeader;
};
struct PeSectionHeader {
CHAR8 Name[8];
UINT32 VirtualSize;
UINT32 VirtualAddress;
UINT32 SizeOfRawData;
UINT32 PointerToRawData;
UINT32 PointerToRelocations;
UINT32 PointerToLinenumbers;
UINT16 NumberOfRelocations;
UINT16 NumberOfLinenumbers;
UINT32 Characteristics;
};
static bool __init pe_name_compare(const struct PeSectionHeader *sect,
const CHAR16 *name)
{
size_t i;
if ( sect->Name[0] != '.' )
return false;
for ( i = 1; i < sizeof(sect->Name); i++ )
{
const char c = sect->Name[i];
if ( c != name[i - 1] )
return false;
if ( c == '\0' )
return true;
}
return name[i - 1] == L'\0';
}
const void *__init pe_find_section(const void *image, const UINTN image_size,
const CHAR16 *section_name, UINTN *size_out)
{
const struct DosFileHeader *dos = image;
const struct PeHeader *pe;
const struct PeSectionHeader *sect;
UINTN offset, i;
if ( image_size < sizeof(*dos) ||
memcmp(dos->Magic, "MZ", 2) != 0 )
return NULL;
offset = dos->ExeHeader;
pe = image + offset;
offset += sizeof(*pe);
if ( image_size < offset ||
memcmp(pe->Magic, "PE\0\0", 4) != 0 )
return NULL;
if ( pe->FileHeader.Machine != PE_HEADER_MACHINE )
return NULL;
offset += pe->FileHeader.SizeOfOptionalHeader;
for ( i = 0; i < pe->FileHeader.NumberOfSections; i++ )
{
sect = image + offset;
if ( image_size < offset + sizeof(*sect) )
return NULL;
if ( !pe_name_compare(sect, section_name) )
{
offset += sizeof(*sect);
continue;
}
if ( image_size < sect->VirtualSize + sect->VirtualAddress )
blexit(L"PE invalid section size + address");
if ( size_out )
*size_out = sect->VirtualSize;
return image + sect->VirtualAddress;
}
return NULL;
}
|
