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
|
/* $Id: IPv4Pool.h $ */
/** @file
* DHCP server - a pool of IPv4 addresses
*/
/*
* Copyright (C) 2017-2019 Oracle Corporation
*
* This file is part of VirtualBox Open Source Edition (OSE), as
* available from http://www.virtualbox.org. This file is free software;
* you can redistribute it and/or modify it under the terms of the GNU
* General Public License (GPL) as published by the Free Software
* Foundation, in version 2 as it comes in the "COPYING" file of the
* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
*/
#ifndef VBOX_INCLUDED_SRC_Dhcpd_IPv4Pool_h
#define VBOX_INCLUDED_SRC_Dhcpd_IPv4Pool_h
#ifndef RT_WITHOUT_PRAGMA_ONCE
# pragma once
#endif
#include <iprt/asm.h>
#include <iprt/stdint.h>
#include <iprt/net.h>
#include <set>
typedef uint32_t ip_haddr_t; /* in host order */
/*
* A range of IPv4 addresses (in host order).
*/
struct IPv4Range
{
ip_haddr_t FirstAddr;
ip_haddr_t LastAddr; /* inclusive */
IPv4Range()
: FirstAddr(), LastAddr() {}
explicit IPv4Range(ip_haddr_t aSingleAddr)
: FirstAddr(aSingleAddr), LastAddr(aSingleAddr) {}
IPv4Range(ip_haddr_t aFirstAddr, ip_haddr_t aLastAddr)
: FirstAddr(aFirstAddr), LastAddr(aLastAddr) {}
explicit IPv4Range(RTNETADDRIPV4 aSingleAddr)
: FirstAddr(RT_N2H_U32(aSingleAddr.u)), LastAddr(RT_N2H_U32(aSingleAddr.u)) {}
IPv4Range(RTNETADDRIPV4 aFirstAddr, RTNETADDRIPV4 aLastAddr)
: FirstAddr(RT_N2H_U32(aFirstAddr.u)), LastAddr(RT_N2H_U32(aLastAddr.u)) {}
bool isValid() const
{
return FirstAddr <= LastAddr;
}
bool contains(ip_haddr_t addr) const
{
return FirstAddr <= addr && addr <= LastAddr;
}
bool contains(RTNETADDRIPV4 addr) const
{
return contains(RT_N2H_U32(addr.u));
}
bool contains(const IPv4Range &range) const
{
return range.isValid() && FirstAddr <= range.FirstAddr && range.LastAddr <= LastAddr;
}
};
inline bool operator==(const IPv4Range &l, const IPv4Range &r)
{
return l.FirstAddr == r.FirstAddr && l.LastAddr == r.LastAddr;
}
inline bool operator<(const IPv4Range &l, const IPv4Range &r)
{
return l.LastAddr < r.FirstAddr;
}
class IPv4Pool
{
typedef std::set<IPv4Range> set_t;
typedef set_t::iterator it_t;
IPv4Range m_range;
set_t m_pool;
public:
IPv4Pool() {}
int init(const IPv4Range &aRange);
int init(RTNETADDRIPV4 aFirstAddr, RTNETADDRIPV4 aLastAddr);
bool contains(RTNETADDRIPV4 addr) const
{ return m_range.contains(addr); }
int insert(const IPv4Range &range);
#if 0
int insert(ip_haddr_t single)
{ return insert(IPv4Range(single)); }
#endif
int insert(ip_haddr_t first, ip_haddr_t last)
{ return insert(IPv4Range(first, last)); }
int insert(RTNETADDRIPV4 single)
{ return insert(IPv4Range(single)); }
int insert(RTNETADDRIPV4 first, RTNETADDRIPV4 last)
{ return insert(IPv4Range(first, last)); }
RTNETADDRIPV4 allocate();
bool allocate(RTNETADDRIPV4);
};
#endif /* !VBOX_INCLUDED_SRC_Dhcpd_IPv4Pool_h */
|
