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
|
/* SPDX-License-Identifier: MIT
*
* Copyright (C) 2019 WireGuard LLC. All Rights Reserved.
*/
#ifndef __LEASE_H__
#define __LEASE_H__
#include <stdint.h>
#include <sys/socket.h>
#include <libmnl/libmnl.h>
#include "common.h"
#include "netlink.h"
#define WG_DYNAMIC_LEASE_CHUNKSIZE 256
struct wg_dynamic_lease {
time_t start_real;
time_t start_mono;
uint32_t leasetime; /* in seconds */
struct in_addr ipv4;
struct in6_addr ipv6;
struct in6_addr lladdr;
};
/*
* Initializes internal state, retrieves routes from nlsock and reads leases
* from fname.
*/
void leases_init(char *fname, struct mnl_socket *nlsock);
/*
* Frees everything, closes file.
*/
void leases_free();
/*
* Creates a new lease and returns a pointer to it, or NULL if either
* we ran out of assignable IPs or the requested IP's are already
* taken. Frees currently held lease, if any. Updates allowedips for
* the peer.
*/
struct wg_dynamic_lease *set_lease(const char *devname, wg_key pubkey,
uint32_t leasetime,
const struct in6_addr *lladdr,
const struct in_addr *ipv4,
const struct in6_addr *ipv6);
/*
* Returns all leases belonging to pubkey, or NULL if there are none.
*/
struct wg_dynamic_lease *get_leases(wg_key pubkey);
/*
* Extend the lease to be leasetime seconds long again. Returns true on error,
* or false otherwise.
*/
bool extend_lease(struct wg_dynamic_lease *lease, uint32_t leasetime);
/* Refreshes all leases, meaning expired ones will be removed. Returns the
* amount of seconds until the next lease will expire, or at most INT_MAX/1000.
*/
int leases_refresh(const char *devname);
/*
* Updates allowedips for peer_pubkey on devname, adding what's in
* lease (including lladdr), removing all others.
*/
void update_allowed_ips(const char *devname, wg_key peer_pubkey,
const struct wg_dynamic_lease *lease);
/*
* Updates all pools with information from the mnl socket nlsock.
*/
void leases_update_pools(struct mnl_socket *nlsock);
/*
* Return true if lease is !NULL and has not expired.
*/
bool lease_is_valid(const struct wg_dynamic_lease *lease);
#ifdef DEBUG
char *lease_to_str(const struct wg_dynamic_lease *l);
#endif
#endif
|
