70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
from scapy.layers.l2 import ARP, Ether
|
|
from scapy.sendrecv import srp
|
|
import netifaces
|
|
from netaddr import IPNetwork
|
|
|
|
|
|
class NetworkScanner:
|
|
"""Иницилазицая сетевого сканера"""
|
|
|
|
def __init__(self, interface=None):
|
|
"""Инициализация параметров интерфейсов"""
|
|
self.interface = (
|
|
interface or netifaces.gateways()["default"][netifaces.AF_INET][1]
|
|
)
|
|
self.ip_info = netifaces.ifaddresses(self.interface).get(netifaces.AF_INET, [])
|
|
self.networks = [
|
|
IPNetwork(f"{ip['addr']}/{ip['netmask']}") for ip in self.ip_info
|
|
]
|
|
self.local_mac = netifaces.ifaddresses(self.interface)[netifaces.AF_LINK][0][
|
|
"addr"
|
|
]
|
|
|
|
def _create_arp_packet(self, target_ip):
|
|
"""Создание пакета ARP для отправки"""
|
|
arp = ARP(pdst=target_ip)
|
|
ether = Ether(dst="ff:ff:ff:ff:ff:ff")
|
|
return ether / arp
|
|
|
|
def scan_network(self, attempts=3, timeout=2):
|
|
"""Подготовка к сканированию сети"""
|
|
all_clients = []
|
|
|
|
for network in self.networks:
|
|
target_ip = f"{network.network}/{network.prefixlen}"
|
|
packet = self._create_arp_packet(target_ip)
|
|
clients = []
|
|
|
|
for _ in range(attempts):
|
|
result = srp(packet, timeout=timeout, iface=self.interface, verbose=0)[
|
|
0
|
|
]
|
|
for _, received in result:
|
|
client = {"ip": received.psrc, "mac": received.hwsrc}
|
|
if client not in clients: # Избегаем дублирования клиентов
|
|
clients.append(client)
|
|
|
|
all_clients.extend(clients)
|
|
|
|
# Добавляет все локальные IP-адреса интерфейса как клиенты
|
|
for ip_info in self.ip_info:
|
|
local_client = {"ip": ip_info["addr"], "mac": self.local_mac}
|
|
if local_client not in all_clients:
|
|
all_clients.append(local_client)
|
|
|
|
# Удаление дубликатов
|
|
unique_clients = {tuple(client.items()) for client in all_clients}
|
|
return [dict(client) for client in unique_clients]
|
|
|
|
def print_clients(self, clients):
|
|
"""Лист адресов клиентов"""
|
|
print("Доступные устройства в сети:")
|
|
print("IP" + " " * 18 + "MAC")
|
|
for client in sorted(clients, key=lambda x: x["ip"]):
|
|
print(f"{client['ip']:<16} {client['mac']}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
scanner = NetworkScanner()
|
|
clients = scanner.scan_network()
|
|
scanner.print_clients(clients)
|