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
|
// Copyright 2024-2026, Amlal El Mahrouss (amlal@nekernel.org)
// Licensed under the Apache License, Version 2.0 (see LICENSE file)
// Official repository: https://github.com/nekernel-org/nekernel
#include <BootKit/BootKit.h>
#include <BootKit/BootThread.h>
#include <FirmwareKit/EFI/API.h>
#include <modules/BootNet/BootNet.h>
STATIC EFI_GUID kEfiSimpleProtoGUID = EFI_SIMPLE_NETWORK_PROTOCOL_GUID;
STATIC EFI_SIMPLE_NETWORK_PROTOCOL* kEfiProtocol = nullptr;
STATIC Void bootnet_read_ip_packet(BOOTNET_INTERNET_HEADER inet,
BOOTNET_INTERNET_HEADER** inet_out);
EXTERN_C Int32 BootNetModuleMain(Kernel::HEL::BootInfoHeader* handover) {
fw_init_efi((EfiSystemTable*) handover->f_FirmwareCustomTables[Kernel::HEL::kHandoverTableST]);
Boot::BootTextWriter writer;
writer.Write("BootNet: Init BootNet...\r");
if (BS->LocateProtocol(&kEfiSimpleProtoGUID, nullptr, (VoidPtr*) &kEfiProtocol) != kEfiOk) {
writer.Write("BootNet: Not supported by firmware.\r");
return kEfiFail;
}
BOOTNET_INTERNET_HEADER inet{};
BOOTNET_INTERNET_HEADER* inet_out = nullptr;
SetMem(&inet, 0, sizeof(BOOTNET_INTERNET_HEADER));
writer.Write("BootNet: Downloading kernel...\r");
bootnet_read_ip_packet(inet, &inet_out);
if (!inet_out) {
writer.Write("BootNet: Not a packet, aborting.\r");
return kEfiFail;
}
if (inet_out->NB1 != 'O' || inet_out->NB1 != 'N' || inet_out->NB1 != 'E' ||
inet_out->NB1 != 'T') {
writer.Write("BootNet: Not a packet, aborting.\r");
return kEfiFail;
}
if (inet_out->Length < 1) {
writer.Write("BootNet: No executable attached to the packet, aborting.\r");
return kEfiFail;
}
if (inet_out->Version != kBootNetVersion) {
writer.Write("BootNet: The version clashes, not good.\r");
return kEfiFail;
}
if (!inet_out->ImpliesProgram) {
Boot::BootThread thread(inet_out->Data);
if (thread.IsValid()) {
writer.Write("BootNet: Running NeKernel...\r");
return thread.Start(handover, YES);
}
return kEfiFail;
} else {
constexpr auto kROMSize = 0x200;
if (inet_out->Length > kROMSize) {
writer.Write("BootNet: Not within 512K, won't flash EEPROM.\r");
return kEfiFail;
}
writer.Write("BootNet: Programming the flash...\r");
/// TODO: Program new firmware to EEPROM (if crc and size matches)
const ATTRIBUTE(unused) UIntPtr kEEPROMStartAddress = 0;
const ATTRIBUTE(unused) UInt16 kEEPROMSize = inet_out->Length;
return kEfiFail;
}
return kEfiFail;
}
STATIC Void bootnet_read_ip_packet(BOOTNET_INTERNET_HEADER inet,
BOOTNET_INTERNET_HEADER** inet_out) {
NE_UNUSED(inet);
kEfiProtocol->Start(kEfiProtocol);
UInt32 size_inet = sizeof(inet);
/// Connect to the local BootNet server.
auto len = inet.Length;
/// And receive the handshake packet.
if (kEfiProtocol->Receive(kEfiProtocol, &size_inet, (UInt32*) &len, (VoidPtr) &inet, nullptr,
nullptr, nullptr) == kEfiOk) {
BOOTNET_INTERNET_HEADER* out = nullptr;
BS->AllocatePool(EfiLoaderData, sizeof(BOOTNET_INTERNET_HEADER) + inet.Length, (VoidPtr*) &out);
if (out == nullptr) {
kEfiProtocol->Stop(kEfiProtocol);
kEfiProtocol->Shutdown(kEfiProtocol);
return;
}
SetMem(out, 0, sizeof(BOOTNET_INTERNET_HEADER));
SetMem(out->Data, 0, inet.Length);
size_inet = sizeof(BOOTNET_INTERNET_HEADER);
auto full_size = size_inet + inet.Length;
/// Ask for it again since we know the full_size variable now.
kEfiProtocol->Receive(kEfiProtocol, &size_inet, (UInt32*) &full_size, (VoidPtr) out, nullptr,
nullptr, nullptr);
*inet_out = out;
}
kEfiProtocol->Stop(kEfiProtocol);
kEfiProtocol->Shutdown(kEfiProtocol);
}
|
