mirror of
https://github.com/kernelkit/infix.git
synced 2026-07-30 12:33:02 +02:00
Mattias Walström
35a909edf0
This is now possible for SSH/RESTCONF/NETCONF, most useful with SSH and adding/modify users.
136 lines
4.4 KiB
Python
136 lines
4.4 KiB
Python
import argparse
|
|
import os
|
|
import pydot
|
|
import shlex
|
|
import sys
|
|
import random
|
|
import inspect
|
|
|
|
from . import neigh, netconf, restconf, ssh, tap, topology
|
|
|
|
|
|
class NullEnv:
|
|
def attr(self, _, default=None):
|
|
return default
|
|
|
|
|
|
ENV = NullEnv()
|
|
|
|
|
|
class ArgumentParser(argparse.ArgumentParser):
|
|
def __init__(self, top):
|
|
super().__init__()
|
|
|
|
self.add_argument("-d", "--debug", default=False, action="store_true")
|
|
self.add_argument("-l", "--logical-topology", dest="ltop", default=top)
|
|
self.add_argument("-p", "--package", default=None)
|
|
self.add_argument("-y", "--yangdir", default=None)
|
|
self.add_argument("-t", "--transport", default=None)
|
|
self.add_argument("ptop", nargs=1, metavar="topology")
|
|
|
|
|
|
class Env(object):
|
|
def __init__(self, ltop=None, argv=sys.argv[1::], environ=os.environ):
|
|
if "INFAMY_ARGS" in environ:
|
|
argv = shlex.split(environ["INFAMY_ARGS"]) + argv
|
|
|
|
self.args = ArgumentParser(ltop).parse_args(argv)
|
|
|
|
pdot = pydot.graph_from_dot_file(self.args.ptop[0])[0]
|
|
self.ptop = topology.Topology(pdot)
|
|
|
|
self.ltop = None
|
|
if self.args.ltop != False:
|
|
if self.args.ltop is None:
|
|
stack = inspect.stack()
|
|
caller_frame = stack[1]
|
|
top_path = caller_frame.filename
|
|
top_path = os.path.join(os.path.dirname(top_path), "topology.dot")
|
|
else:
|
|
top_path = self.args.ltop
|
|
|
|
ldot = pydot.graph_from_dot_file(top_path)[0]
|
|
self.ltop = topology.Topology(ldot)
|
|
if not self.ltop.map_to(self.ptop):
|
|
raise tap.TestSkip()
|
|
|
|
print(repr(self.ltop))
|
|
|
|
global ENV
|
|
ENV = self
|
|
|
|
def attr(self, name, default=None):
|
|
val = self.ptop.get_attr("ix_" + name)
|
|
if val is None:
|
|
return default
|
|
|
|
try:
|
|
return int(val, 0)
|
|
except ValueError:
|
|
pass
|
|
|
|
return val
|
|
|
|
def get_password(self, node):
|
|
return self.ptop.get_password(node)
|
|
|
|
def attach(self, node, port, protocol=None, test_reset=True, username = None, password = None):
|
|
"""Attach to node on port using protocol."""
|
|
|
|
name = node
|
|
if self.ltop:
|
|
mapping = self.ltop.mapping[node]
|
|
node, port = self.ltop.xlate(node, port)
|
|
else:
|
|
mapping = None
|
|
|
|
# Test protocol always highest prio, followed by command line,
|
|
# then environment (detected from defconfig), lastly random.
|
|
if protocol is None:
|
|
if self.args.transport is not None:
|
|
protocol = self.args.transport
|
|
else:
|
|
hseed = os.environ.get('PYTHONHASHSEED', 0)
|
|
random.seed(f"{sys.argv[0]}-{hseed}")
|
|
protocol = random.choice(["netconf", "restconf"])
|
|
|
|
if password is None:
|
|
password = self.get_password(node)
|
|
if username is None:
|
|
username = "admin"
|
|
|
|
ctrl = self.ptop.get_ctrl()
|
|
cport, _ = self.ptop.get_mgmt_link(ctrl, node)
|
|
|
|
print(f"Probing {node} on port {cport} for IPv6LL mgmt address ...")
|
|
mgmtip = neigh.ll6ping(cport)
|
|
if not mgmtip:
|
|
raise Exception(f"Failed, cannot find mgmt IP for {node}")
|
|
|
|
if protocol == "netconf":
|
|
dev = netconf.Device(name,
|
|
location=netconf.Location(cport, mgmtip,
|
|
username,password),
|
|
mapping=mapping,
|
|
yangdir=self.args.yangdir)
|
|
if test_reset:
|
|
dev.test_reset()
|
|
return dev
|
|
|
|
if protocol == "ssh":
|
|
return ssh.Device(name, ssh.Location(mgmtip, username, password))
|
|
|
|
if protocol == "restconf":
|
|
dev = restconf.Device(name,
|
|
location=restconf.Location(cport,
|
|
mgmtip,
|
|
username,
|
|
password),
|
|
mapping=mapping,
|
|
yangdir=self.args.yangdir)
|
|
if test_reset:
|
|
dev.test_reset()
|
|
return dev
|
|
|
|
raise Exception(f"Unsupported management procotol \"{protocol}\"")
|