blob: 41bade3d6ff2095aab03a21945e5ee82c54c115e (
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
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
|
/* -------------------------------------------
Copyright ZKA Technologies
------------------------------------------- */
#include <BootKit/ProgramLoader.hxx>
#include <BootKit/Vendor/Support.hxx>
#include <BootKit/BootKit.hxx>
#include <KernelKit/PEF.hpp>
EXTERN_C
{
#include <string.h>
}
namespace Boot
{
ProgramLoader::ProgramLoader(VoidPtr blob)
: fBlob(blob), fStartAddress(nullptr)
{
// detect the format.
const Char* firstBytes = reinterpret_cast<char*>(fBlob);
BTextWriter writer;
if (!firstBytes)
{
// failed to provide a valid pointer.
return;
}
if (firstBytes[0] == kMagMz0 &&
firstBytes[1] == kMagMz1)
{
// Parse PE32+
fStartAddress = nullptr;
writer.Write("newosldr: MZ executable detected.\r");
}
else if (firstBytes[0] == kPefMagic[0] &&
firstBytes[1] == kPefMagic[1] &&
firstBytes[2] == kPefMagic[2] &&
firstBytes[3] == kPefMagic[3])
{
// Parse Non FAT PEF.
fStartAddress = nullptr;
writer.Write("newosldr: PEF executable detected.\r");
}
else
{
// probably a binary blob.
fStartAddress = fBlob;
}
}
/// @note handover header has to be valid!
Void ProgramLoader::Start(HEL::HandoverInformationHeader* handover)
{
MUST_PASS(handover);
BTextWriter writer;
writer.Write("newosldr: Trying to run: ").Write(fBlobName).Write("\r");
if (!fStartAddress)
{
writer.Write("newosldr: Exec format error.\r");
return;
}
((HEL::HandoverProc)fStartAddress)(handover);
}
const Char* ProgramLoader::GetName()
{
return fBlobName;
}
Void ProgramLoader::SetName(const Char* name)
{
CopyMem(fBlobName, name, StrLen(name));
}
} // namespace Boot
|
