89 lines
2.2 KiB
Python
89 lines
2.2 KiB
Python
from time import sleep
|
|
|
|
from scapy.all import ICMP, IP, Dot1Q, Ether, sendp, AsyncSniffer, wrpcap
|
|
from tqdm import tqdm
|
|
|
|
a = AsyncSniffer(iface="enp5s0d1", filter="icmp")
|
|
a.start()
|
|
for i in tqdm(range(0, 4095)):
|
|
sendp(
|
|
Ether(dst="00:02:c9:27:10:73", src="00:f0:cb:ef:e0:3b")
|
|
/ Dot1Q(vlan=i)
|
|
/ IP(dst="10.0.45.45", src="10.0.45.5")
|
|
/ ICMP(id=i),
|
|
iface="enp2s0",
|
|
verbose=False,
|
|
)
|
|
print("Wrapping up and waiting for last packets")
|
|
sleep(1)
|
|
plist = a.stop()
|
|
|
|
seen = [False for i in range(4095)]
|
|
icmpid = [False for i in range(4095)]
|
|
|
|
def pprint(seen):
|
|
last = None
|
|
s = ""
|
|
for k, v in enumerate(seen):
|
|
if v and last is None:
|
|
last = k
|
|
s += str(k)
|
|
if not v and last is not None:
|
|
if k-1 == last:
|
|
s += ","
|
|
else:
|
|
s += f"-{k-1},"
|
|
last = None
|
|
if last is not None and last < k:
|
|
s += f"-{k}"
|
|
return s + "\n"
|
|
|
|
|
|
for p in plist:
|
|
if Dot1Q in p:
|
|
vlan = int(p[Dot1Q].vlan)
|
|
seen[vlan] = True
|
|
if ICMP in p and p[ICMP].id == vlan:
|
|
icmpid[vlan] = True
|
|
|
|
print("Icmp correspondance")
|
|
print(pprint(icmpid))
|
|
print("Seen vlans")
|
|
print(pprint(seen))
|
|
|
|
print("Collecting :")
|
|
import subprocess
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
import shutil
|
|
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") + "-test"
|
|
log_dir = Path("./logs") / timestamp
|
|
log_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(log_dir / "seen_vlans.txt", "w") as f:
|
|
f.write(pprint(seen))
|
|
|
|
with open(log_dir / "seen_icmp_ids.txt", "w") as f:
|
|
f.write(pprint(icmpid))
|
|
|
|
shutil.copy(Path("./script.py"), log_dir / "script.py")
|
|
|
|
print(" - [x] script.py")
|
|
|
|
with open(log_dir / "egress.pcap", "wb") as file:
|
|
wrpcap(file, plist)
|
|
|
|
print(" - [x] pcap")
|
|
|
|
cmd_config = ["ssh", "-o", "StrictHostKeyChecking=no", "root@192.168.42.6", "cli show config"]
|
|
cmd_rsi = ["ssh", "-o", "StrictHostKeyChecking=no", "root@192.168.42.6", "cli request support information"]
|
|
|
|
result = subprocess.run(cmd_config, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
|
|
|
|
with open(log_dir / "config.txt", "w") as f:
|
|
f.write(result.stdout)
|
|
|
|
print(" - [x] config")
|
|
|
|
print("Finished, saved to", log_dir)
|