add support for upstream port and add garage example script

1 files changed, 11 insertions, 3 deletions

diff --git a/net.py b/net.py
index bc20e4d..8d0a248 100644
--- a/net.py
+++ b/net.py
@@ -1,4 +1,5 @@
 import subprocess
+import os
 
 _netns = ["ip", "netns"]
 def run_netns(*cmd):
@@ -8,9 +9,16 @@ def run_netns(*cmd):
     return process
 
 class ns:
-    def list():
-        process = run_netns("list")
-        return [ns for ns in process.stdout.split("\n") if ns.startswith("testnet-")]
+    def name_unconfined():
+        if "unconfined" not in ns.list(True):
+            run_netns("attach", "unconfined", str(os.getpid()))
+
+    def list(include_unconfined = False):
+        try:
+            nss = os.listdir("/var/run/netns")
+            return [ns for ns in nss if ns.startswith("testnet-") or include_unconfined and ns == "unconfined"]
+        except FileNotFoundError:
+            return []
 
     def forget(name):
         run_netns("del", name)