50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
|
from subprocess import run, PIPE
|
||
|
|
from ipaddress import IPv4Address
|
||
|
|
import re
|
||
|
|
|
||
|
|
rip = re.compile(r"[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}") # IP regex
|
||
|
|
rmac = re.compile(r"(?:[A-F]|[0-9]){1,3}:(?:[A-F]|[0-9]){1,3}:(?:[A-F]|[0-9]){1,3}:(?:[A-F]|[0-9]){1,3}:(?:[A-F]|[0-9])"
|
||
|
|
r"{1,3}:(?:[A-F]|[0-9]){1,3}") # MAC regex
|
||
|
|
|
||
|
|
|
||
|
|
def get_router_ipv4(host: str, port: int, user: str, key: str) -> [IPv4Address]:
|
||
|
|
"""
|
||
|
|
Gets IPv4 list of the router
|
||
|
|
|
||
|
|
:param host: The SSH host of the router
|
||
|
|
:type host: str
|
||
|
|
:param port: The SSH port of the router
|
||
|
|
:type port: int
|
||
|
|
:param user: The SSH user of the router
|
||
|
|
:type port: str
|
||
|
|
:param key: The SSH key of the router
|
||
|
|
:type key: str
|
||
|
|
:return: List of IPv4 in the router
|
||
|
|
:rtype: [IPv4Address]
|
||
|
|
"""
|
||
|
|
|
||
|
|
out = run(["ssh", "-i", key, "-o", "StrictHostKeyChecking no", f"{user}@{host}", "-p", str(port), "/ip arp print"],
|
||
|
|
stdout=PIPE).stdout.decode()
|
||
|
|
return [IPv4Address(i) for i in rip.findall(out)]
|
||
|
|
|
||
|
|
|
||
|
|
def get_router_mac(host: str, port: int, user: str, key: str) -> [str]:
|
||
|
|
"""
|
||
|
|
Gets MAC list of the router
|
||
|
|
|
||
|
|
:param host: The SSH host of the router
|
||
|
|
:type host: str
|
||
|
|
:param port: The SSH port of the router
|
||
|
|
:type port: int
|
||
|
|
:param user: The SSH user of the router
|
||
|
|
:type port: str
|
||
|
|
:param key: The SSH key of the router
|
||
|
|
:type key: str
|
||
|
|
:return: List of MAC in the router
|
||
|
|
:rtype: [str]
|
||
|
|
"""
|
||
|
|
|
||
|
|
out = run(["ssh", "-i", key, "-o", "StrictHostKeyChecking no", f"{user}@{host}", "-p", port, "/ip arp print"],
|
||
|
|
stdout=PIPE).stdout.decode()
|
||
|
|
return rmac.findall(out)
|