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
|
// SPDX-License-Identifier: GPL-2.0-or-later
/*
* Copyright (C) 2022 Felix Fietkau <nbd@nbd.name>
*/
#include "unetd.h"
static const struct wg_ops *wg_get_ops(struct network *net)
{
if (wg_user_ops.check(net))
return &wg_user_ops;
#ifdef linux
return &wg_linux_ops;
#else
return NULL;
#endif
}
int wg_init_network(struct network *net)
{
net->wg.ops = wg_get_ops(net);
if (!net->wg.ops)
return -1;
return net->wg.ops->init(net);
}
void wg_cleanup_network(struct network *net)
{
if (net->wg.ops)
net->wg.ops->cleanup(net);
}
static void
wg_peer_set_connected(struct network *net, struct network_peer *peer, bool val)
{
if (peer->state.connected == val)
return;
peer->state.connected = val;
network_services_peer_update(net, peer);
}
struct network_peer *wg_peer_update_start(struct network *net, const uint8_t *key)
{
struct network_peer *peer;
peer = vlist_find(&net->peers, key, peer, node);
if (!peer || peer->indirect)
return NULL;
peer->state.handshake = false;
peer->state.idle++;
if (peer->state.ping_wait > 0)
peer->state.ping_wait--;
if (peer->state.idle >= 2 * net->net_config.keepalive)
wg_peer_set_connected(net, peer, false);
if (peer->state.idle > net->net_config.keepalive)
network_pex_event(net, peer, PEX_EV_PING);
return peer;
}
void wg_peer_update_done(struct network *net, struct network_peer *peer)
{
if (peer->state.handshake)
network_pex_event(net, peer, PEX_EV_HANDSHAKE);
}
void wg_peer_set_last_handshake(struct network *net, struct network_peer *peer,
uint64_t now, uint64_t sec)
{
peer->state.last_handshake_diff = now - sec;
if (sec == peer->state.last_handshake)
return;
peer->state.handshake = true;
peer->state.last_handshake = sec;
sec = now - sec;
if (sec <= net->net_config.keepalive) {
if (peer->state.idle > sec)
peer->state.idle = sec;
wg_peer_set_connected(net, peer, true);
}
}
void wg_peer_set_rx_bytes(struct network *net, struct network_peer *peer,
uint64_t bytes)
{
int64_t diff = bytes - peer->state.rx_bytes;
peer->state.rx_bytes = bytes;
if (diff > 0) {
peer->state.idle = 0;
wg_peer_set_connected(net, peer, true);
}
}
void wg_peer_set_tx_bytes(struct network *net, struct network_peer *peer,
uint64_t bytes)
{
peer->state.tx_bytes = bytes;
}
void wg_peer_set_endpoint(struct network *net, struct network_peer *peer,
void *data, size_t len)
{
if (len > sizeof(peer->state.endpoint))
return;
if (!memcmp(&peer->state.endpoint, data, len))
return;
memset(&peer->state.endpoint, 0, sizeof(peer->state.endpoint));
memcpy(&peer->state.endpoint, data, len);
network_pex_event(net, peer, PEX_EV_ENDPOINT_CHANGE);
}
|
