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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
|
/*
*
* (C) 2015-22 - ntop.org
*
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*
*/
#ifndef _HOST_POOLS_H_
#define _HOST_POOLS_H_
#include "ntop_includes.h"
class NetworkInterface;
class Host;
class Mac;
class HostPools {
private:
VLANAddressTree *tree, *tree_shadow;
NetworkInterface *iface;
u_int16_t max_num_pools;
int32_t *num_active_hosts_inline, *num_active_hosts_offline;
int32_t *num_active_l2_devices_inline, *num_active_l2_devices_offline;
HostPoolStats **stats, **stats_shadow;
#ifdef NTOPNG_PRO
bool *children_safe;
bool *forge_global_dns;
u_int8_t *routing_policy_id;
u_int16_t *pool_shaper;
u_int32_t *schedule_bitmap;
bool *enforce_quotas_per_pool_member; /* quotas can be pool-wide or per pool member */
bool *enforce_shapers_per_pool_member;
#endif
inline HostPoolStats* getPoolStats(u_int16_t host_pool_id) {
if((host_pool_id >= max_num_pools) || (!stats))
return NULL;
return stats[host_pool_id];
}
void reloadPoolStats();
static void deleteStats(HostPoolStats ***hps);
void swap(VLANAddressTree *new_trees, HostPoolStats **new_stats);
void loadFromRedis();
inline void incNumMembers(u_int16_t pool_id, int32_t *ctr) const {
if(ctr && pool_id < max_num_pools)
ctr[pool_id]++;
};
inline void decNumMembers(u_int16_t pool_id, int32_t *ctr) const {
if(ctr && pool_id < max_num_pools)
ctr[pool_id]--;
};
public:
HostPools(NetworkInterface *_iface);
virtual ~HostPools();
void dumpToRedis();
void reloadPools();
u_int16_t getPool(Host *h);
u_int16_t getPool(Mac *m);
u_int16_t getPoolByName(const char * const pool_name);
bool findIpPool(IpAddress *ip, VLANid vlan_id,
u_int16_t *found_pool, ndpi_patricia_node_t **found_node);
bool findMacPool(const u_int8_t * const mac, u_int16_t *found_pool);
bool findMacPool(Mac *mac, u_int16_t *found_pool);
void lua(lua_State *vm);
inline int32_t getNumPoolHosts(u_int16_t pool_id) {
if(pool_id >= max_num_pools)
return NO_HOST_POOL_ID;
return num_active_hosts_inline[pool_id] + num_active_hosts_offline[pool_id];
}
inline int32_t getNumPoolL2Devices(u_int16_t pool_id) {
if(pool_id >= max_num_pools)
return NO_HOST_POOL_ID;
return num_active_l2_devices_inline[pool_id] + num_active_l2_devices_offline[pool_id];
}
inline void incNumHosts(u_int16_t pool_id, bool isInlineCall) {
incNumMembers(pool_id, isInlineCall ? num_active_hosts_inline : num_active_hosts_offline);
};
inline void decNumHosts(u_int16_t pool_id, bool isInlineCall) {
decNumMembers(pool_id, isInlineCall ? num_active_hosts_inline : num_active_hosts_offline);
};
inline void incNumL2Devices(u_int16_t pool_id, bool isInlineCall) {
incNumMembers(pool_id, isInlineCall ? num_active_l2_devices_inline : num_active_l2_devices_offline);
};
inline void decNumL2Devices(u_int16_t pool_id, bool isInlineCall) {
decNumMembers(pool_id, isInlineCall ? num_active_l2_devices_inline : num_active_l2_devices_offline);
};
void incPoolNumDroppedFlows(u_int16_t pool_id);
void incPoolStats(u_int32_t when, u_int16_t host_pool_id, u_int16_t ndpi_proto,
ndpi_protocol_category_t category_id, u_int64_t sent_packets, u_int64_t sent_bytes,
u_int64_t rcvd_packets, u_int64_t rcvd_bytes);
void updateStats(const struct timeval *tv);
void luaStats(lua_State *vm);
void luaStats(lua_State *mv, u_int16_t pool_id);
/* To be called on the same thread as incPoolStats */
void checkPoolsStatsReset();
inline bool getProtoStats(u_int16_t host_pool_id, u_int16_t ndpi_proto, u_int64_t *bytes, u_int32_t *duration) {
HostPoolStats *hps;
if (!(hps = getPoolStats(host_pool_id))) return false;
hps->getProtoStats(ndpi_proto, bytes, duration);
return true;
}
inline bool getCategoryStats(u_int16_t host_pool_id, ndpi_protocol_category_t category_id, u_int64_t *bytes, u_int32_t *duration) {
HostPoolStats *hps;
if (!(hps = getPoolStats(host_pool_id))) return false;
hps->getCategoryStats(category_id, bytes, duration);
return true;
}
inline bool getStats(u_int16_t host_pool_id, u_int64_t *bytes, u_int32_t *duration) {
HostPoolStats *hps;
if (!(hps = getPoolStats(host_pool_id))) return false;
hps->getStats(bytes, duration);
return true;
}
void resetPoolsStats(u_int16_t pool_filter);
#ifdef NTOPNG_PRO
inline bool enforceQuotasPerPoolMember(u_int16_t pool_id) {
return(((pool_id != NO_HOST_POOL_ID) && (pool_id < max_num_pools)) ? enforce_quotas_per_pool_member[pool_id] : false);
}
inline bool enforceShapersPerPoolMember(u_int16_t pool_id) {
return(((pool_id != NO_HOST_POOL_ID) && (pool_id < max_num_pools)) ? enforce_shapers_per_pool_member[pool_id] : false);
}
inline u_int16_t getPoolShaper(u_int16_t pool_id) {
return((pool_id < max_num_pools) ? pool_shaper[pool_id] : DEFAULT_SHAPER_ID);
}
inline u_int32_t getPoolSchedule(u_int16_t pool_id) {
return(((pool_id != NO_HOST_POOL_ID) && (pool_id < max_num_pools)) ? schedule_bitmap[pool_id] : DEFAULT_TIME_SCHEDULE);
}
inline bool isChildrenSafePool(u_int16_t pool_id) {
return((pool_id < max_num_pools) ? children_safe[pool_id] : false);
}
inline bool forgeGlobalDns(u_int16_t pool_id) {
return((pool_id < max_num_pools) ? forge_global_dns[pool_id] : false);
}
inline u_int8_t getRoutingPolicy(u_int16_t pool_id) {
return(((pool_id != NO_HOST_POOL_ID) && (pool_id < max_num_pools)) ? routing_policy_id[pool_id] : DEFAULT_ROUTING_TABLE_ID);
}
#endif
};
#endif /* _HOST_POOLS_H_ */
|
