1 # SPDX-License-Identifier: GPL-2.0
12 class CmdExitFailure(Exception):
17 def __init__(self, comm, shell=True, fail=True, ns=None, background=False, host=None, timeout=5):
19 comm = f'ip netns exec {ns} ' + comm
27 self.proc = host.cmd(comm)
29 self.proc = subprocess.Popen(comm, shell=shell, stdout=subprocess.PIPE,
30 stderr=subprocess.PIPE)
32 self.process(terminate=False, fail=fail, timeout=timeout)
34 def process(self, terminate=True, fail=None, timeout=5):
40 stdout, stderr = self.proc.communicate(timeout)
41 self.stdout = stdout.decode("utf-8")
42 self.stderr = stderr.decode("utf-8")
43 self.proc.stdout.close()
44 self.proc.stderr.close()
45 self.ret = self.proc.returncode
47 if self.proc.returncode != 0 and fail:
48 if len(stderr) > 0 and stderr[-1] == "\n":
50 raise CmdExitFailure("Command failed: %s\nSTDOUT: %s\nSTDERR: %s" %
51 (self.proc.args, stdout, stderr))
55 def __init__(self, comm, shell=True, fail=None, ns=None, host=None,
57 super().__init__(comm, background=True,
58 shell=shell, fail=fail, ns=ns, host=host)
59 self.terminate = not exit_wait
60 self.check_fail = fail
65 def __exit__(self, ex_type, ex_value, ex_tb):
66 return self.process(terminate=self.terminate, fail=self.check_fail)
69 global_defer_queue = []
73 def __init__(self, func, *args, **kwargs):
74 global global_defer_queue
76 if not callable(func):
77 raise Exception("defer created with un-callable object, did you call the function instead of passing its name?")
83 self._queue = global_defer_queue
84 self._queue.append(self)
89 def __exit__(self, ex_type, ex_value, ex_tb):
93 self.func(*self.args, **self.kwargs)
96 self._queue.remove(self)
103 def tool(name, args, json=None, ns=None, host=None):
108 cmd_obj = cmd(cmd_str, ns=ns, host=host)
110 return _json.loads(cmd_obj.stdout)
114 def ip(args, json=None, ns=None, host=None):
116 args = f'-netns {ns} ' + args
117 return tool('ip', args, json=json, host=host)
120 def ethtool(args, json=None, ns=None, host=None):
121 return tool('ethtool', args, json=json, ns=ns, host=host)
126 Get a random unprivileged port, try to make sure it's not already used.
128 for _ in range(1000):
129 port = random.randint(10000, 65535)
131 with socket.socket(socket.AF_INET6, socket.SOCK_STREAM) as s:
135 if e.errno != errno.EADDRINUSE:
137 raise Exception("Can't find any free unprivileged port")
140 def wait_port_listen(port, proto="tcp", ns=None, host=None, sleep=0.005, deadline=5):
141 end = time.monotonic() + deadline
143 pattern = f":{port:04X} .* "
144 if proto == "tcp": # for tcp protocol additionally check the socket state
146 pattern = re.compile(pattern)
149 data = cmd(f'cat /proc/net/{proto}*', ns=ns, host=host, shell=True).stdout
150 for row in data.split("\n"):
151 if pattern.search(row):
153 if time.monotonic() > end:
154 raise Exception("Waiting for port listen timed out")