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
|
/*
*
* (C) 2013-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 _NETWORK_DISCOVERY_H_
#define _NETWORK_DISCOVERY_H_
#include "ntop_includes.h"
/* ******************************* */
class NetworkDiscovery {
private:
int udp_sock;
NetworkInterface *iface;
pcap_t *pd;
lua_State* mdns_vm;
Mutex m;
struct bpf_program fcode;
bool has_bpf_filter;
u_int32_t wrapsum(u_int32_t sum);
u_int16_t in_cksum(u_int8_t *buf, u_int16_t buf_len, u_int32_t sum);
u_int16_t buildMDNSDiscoveryDatagram(const char *query, u_int32_t sender_ip, u_int8_t *sender_mac,
char *datagram, u_int datagram_len);
void dissectMDNS(u_char *buf, u_int buf_len, char *out, u_int out_len);
inline void setMDNSvm(lua_State* vm) {
m.lock(__FILE__, __LINE__);
mdns_vm = vm;
m.unlock(__FILE__, __LINE__);
}
void discoverDHCP(u_char *mac);
public:
NetworkDiscovery(NetworkInterface *_iface);
~NetworkDiscovery();
void discover(lua_State* vm, u_int timeout);
void arpScan(lua_State* vm);
void queueMDNSResponse(u_int32_t src_ip_nw_byte_order, u_char *buf, u_int buf_len);
void sendArpNetwork(void *data, u_int32_t netp, u_int32_t maskp, u_int32_t sender_ip);
};
#endif /* _NETWORK_DISCOVERY_H_ */
|
