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
|
#include <Kernel/Net/Routing.h>
#include <Kernel/Thread.h>
//#define ROUTING_DEBUG
Lockable<HashMap<IPv4Address, MACAddress>>& arp_table()
{
static Lockable<HashMap<IPv4Address, MACAddress>>* the;
if (!the)
the = new Lockable<HashMap<IPv4Address, MACAddress>>;
return *the;
}
bool RoutingDecision::is_zero() const
{
return adapter.is_null() || next_hop.is_zero();
}
RoutingDecision route_to(const IPv4Address& target, const IPv4Address& source)
{
auto target_addr = target.to_u32();
auto source_addr = source.to_u32();
WeakPtr<NetworkAdapter> local_adapter = nullptr;
WeakPtr<NetworkAdapter> gateway_adapter = nullptr;
NetworkAdapter::for_each([source_addr, &target_addr, &local_adapter, &gateway_adapter](auto& adapter) {
auto adapter_addr = adapter.ipv4_address().to_u32();
auto adapter_mask = adapter.ipv4_netmask().to_u32();
if (source_addr != 0 && source_addr != adapter_addr)
return;
if ((target_addr & adapter_mask) == (adapter_addr & adapter_mask))
local_adapter = adapter.make_weak_ptr();
if (adapter.ipv4_gateway().to_u32() != 0)
gateway_adapter = adapter.make_weak_ptr();
});
if (!local_adapter && !gateway_adapter) {
#ifdef ROUTING_DEBUG
kprintf("Routing: Couldn't find a suitable adapter for route to %s\n",
target.to_string().characters());
#endif
return { nullptr, {} };
}
WeakPtr<NetworkAdapter> adapter = nullptr;
IPv4Address next_hop_ip;
if (local_adapter) {
#ifdef ROUTING_DEBUG
kprintf("Routing: Got adapter for route (direct): %s (%s/%s) for %s\n",
local_adapter->name().characters(),
local_adapter->ipv4_address().to_string().characters(),
local_adapter->ipv4_netmask().to_string().characters(),
target.to_string().characters());
#endif
adapter = local_adapter;
next_hop_ip = target;
} else if (gateway_adapter) {
#ifdef ROUTING_DEBUG
kprintf("Routing: Got adapter for route (using gateway %s): %s (%s/%s) for %s\n",
gateway_adapter->ipv4_gateway().to_string().characters(),
gateway_adapter->name().characters(),
gateway_adapter->ipv4_address().to_string().characters(),
gateway_adapter->ipv4_netmask().to_string().characters(),
target.to_string().characters());
#endif
adapter = gateway_adapter;
next_hop_ip = gateway_adapter->ipv4_gateway();
} else {
return { nullptr, {} };
}
{
LOCKER(arp_table().lock());
auto addr = arp_table().resource().get(next_hop_ip);
if (addr.has_value()) {
#ifdef ROUTING_DEBUG
kprintf("Routing: Using cached ARP entry for %s (%s)\n",
next_hop_ip.to_string().characters(),
addr.value().to_string().characters());
#endif
return { adapter, addr.value() };
}
}
#ifdef ROUTING_DEBUG
kprintf("Routing: Sending ARP request via adapter %s for IPv4 address %s\n",
adapter->name().characters(),
next_hop_ip.to_string().characters());
#endif
ARPPacket request;
request.set_operation(ARPOperation::Request);
request.set_target_hardware_address({ 0xff, 0xff, 0xff, 0xff, 0xff, 0xff });
request.set_target_protocol_address(next_hop_ip);
request.set_sender_hardware_address(adapter->mac_address());
request.set_sender_protocol_address(adapter->ipv4_address());
adapter->send({ 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }, request);
(void)current->block_until("Routing (ARP)", [next_hop_ip] {
return arp_table().resource().get(next_hop_ip).has_value();
});
{
LOCKER(arp_table().lock());
auto addr = arp_table().resource().get(next_hop_ip);
if (addr.has_value()) {
#ifdef ROUTING_DEBUG
kprintf("Routing: Got ARP response using adapter %s for %s (%s)\n",
adapter->name().characters(),
next_hop_ip.to_string().characters(),
addr.value().to_string().characters());
#endif
return { adapter, addr.value() };
}
}
#ifdef ROUTING_DEBUG
kprintf("Routing: Couldn't find route using adapter %s for %s\n",
adapter->name().characters(),
target.to_string().characters());
#endif
return { nullptr, {} };
}
|
