83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
from ipaddress import IPv4Address
|
|
from router.get_router import get_router_ipv4, get_router_mac
|
|
from ip.ipv6 import link_local
|
|
from subprocess import run
|
|
|
|
|
|
def insert_router_ipv4(insert: [(IPv4Address, str, IPv4Address, int)], interface: str, host: str, port: int, user: str,
|
|
key: str, debug: bool = False, verbose: bool = False):
|
|
"""
|
|
This function insert IPv4 on the router
|
|
|
|
:param insert: The list of IPs, MACs to insert
|
|
:type insert: [(IPv4Address, str, IPv4Address, int)]
|
|
:param interface: The interface of IPs
|
|
:type interface: str
|
|
: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
|
|
:param debug: Disable command on router
|
|
:type debug: bool
|
|
:param verbose: Print each command on router
|
|
:type verbose: bool
|
|
"""
|
|
|
|
ipl = get_router_ipv4(host, port, user, key)
|
|
macl = get_router_mac(host, port, user, key)
|
|
|
|
print("Start insert IPv4 on router")
|
|
for i in insert:
|
|
if i[1]:
|
|
if (i[0] not in ipl) and (i[1] not in macl):
|
|
cmd = ["ssh", "-i", key, "-o", "StrictHostKeyChecking no", f"{user}@{host}", "-p", str(port),
|
|
f"/ip arp add address={i[0]} mac-address={i[1]} interface={interface}"]
|
|
if not debug:
|
|
run(cmd)
|
|
if debug or verbose:
|
|
print(cmd)
|
|
print("Insert IPv4 on router done")
|
|
|
|
|
|
def insert_router_ipv6(insert: [(IPv4Address, str, IPv4Address, int)], ipv6: str, interface: str, host: str, port: int, user: str,
|
|
key: str, debug: bool = False, verbose: bool = False):
|
|
"""
|
|
This function insert IPv6 on the router
|
|
|
|
:param insert: The list of IPs, MACs to insert
|
|
:type insert: [(IPv4Address, str, IPv4Address, int)]
|
|
:param ipv6: The IPV6 template
|
|
:type ipv6: str
|
|
:param interface: The interface of IPs
|
|
:type interface: str
|
|
: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
|
|
:param debug: Disable commit on database
|
|
:type debug: bool
|
|
:param verbose: Print each command on router
|
|
:type verbose: bool
|
|
"""
|
|
|
|
print("Start insert IPv6 on router")
|
|
for i in insert:
|
|
if i[1]:
|
|
ip = ipv6.format(str(i[0]).split(".")[-1])
|
|
gateway = link_local(i[1])
|
|
cmd = ["ssh", "-i", key, "-o", "StrictHostKeyChecking no", f"{user}@{host}", "-p", str(port),
|
|
f"/ipv6 route add dst-address={ip} gateway={gateway}%{interface}"]
|
|
if not debug:
|
|
run(cmd)
|
|
if debug or verbose:
|
|
print(cmd)
|
|
print("Insert IPv6 on router done")
|