blob: 83b10ddb3b94df4adf1390dc02206f92569bfdb9 (
plain)
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
|
#pragma once
/* SPDX-License-Identifier: Unlicense
*/
#include "coff.h"
#include "data_buffer.h"
namespace COFF {
class Image {
const DataBuffer _data;
char *const _error;
const COFF::FileHeader _file_header;
bool _has_optional_header;
const COFF::OptionalHeader _optional_header;
public:
explicit Image(DataBuffer&&);
~Image();
constexpr bool IsValid() const { return _error == nullptr; }
constexpr const DataBuffer &Data() const { return _data; };
constexpr const char *Error() const { return _error; }
constexpr const COFF::FileHeader &FileHeader() const { return _file_header; }
constexpr bool HasOptionalHeader() const { return _has_optional_header; }
constexpr const COFF::OptionalHeader &OptionalHeader() const { return _optional_header; }
constexpr const COFF::SectionHeader GetSectionHeader(unsigned index) const
{
if (index > _file_header.nsections) {
return SectionHeader{};
}
const size_t offset = COFF::kFileHeaderSize + _file_header.optional_header_nbytes +
COFF::kSectionHeaderSize * index;
if (offset + kSectionHeaderSize > _data.buffer_size) {
return SectionHeader{};
}
return SectionHeader::FromBytes(_data.buffer + offset);
}
constexpr DataView GetSectionDataView(const COFF::SectionHeader &header) const
{
return DataView{_data.buffer + header.section_offset, header.size};
}
};
}
|
