ntopng/include/NetworkDiscovery.h
2026-03-15 10:50:50 +01:00

64 lines
2 KiB
C++

/*
*
* (C) 2013-26 - 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 debug_mode;
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_ */