1 files changed, 63 insertions, 0 deletions
diff --git a/.ci/utils/httpd.py b/.ci/utils/httpd.py
new file mode 100644
index 000000000..53741566e
--- /dev/null
+++ b/.ci/utils/httpd.py
@@ -0,0 +1,63 @@
+import http.server
+import time
+import os
+import pyroute2
+import socket
+import socketserver
+import threading
+from functools import partial
+from pathlib import Path
+
+class ThreadingHTTPServer(socketserver.ThreadingMixIn, http.server.HTTPServer):
+    pass
+
+class HTTPServer:
+    def __init__(self, path, remote_ip):
+        self.path = path
+        self.port = None
+        self.old_path = None
+        self.httpd = None
+
+        with pyroute2.IPRoute() as ipr:
+            r = ipr.route('get', dst=remote_ip)
+            for attr in r[0]['attrs']:
+                if attr[0] == 'RTA_PREFSRC':
+                    self.ip = attr[1]
+        with socket.socket() as s:
+            s.bind(('', 0))
+            self.port = s.getsockname()[1]
+
+    def get_url(self, filename):
+        path = Path(self.path) / filename
+        assert path.exists()
+        return f"http://{self.ip}:{self.port}/{filename}"
+
+    def __enter__(self):
+        def start_server():
+            Handler = http.server.SimpleHTTPRequestHandler
+            self.httpd = ThreadingHTTPServer(("", self.port), Handler)
+            self.httpd.serve_forever()
+
+        # Kind of annoying, but to work with older pythons where
+        # SimpleHTTPRequestHandler doesn't take a directory parameter but only
+        # serves the current directory:
+        self.old_path = os.getcwd()
+        os.chdir(self.path)
+
+        self.thread = threading.Thread(target=start_server)
+        self.thread.start()
+        return self
+
+    def __exit__(self, type, value, exc):
+        if self.httpd is not None:
+            self.httpd.shutdown()
+            self.httpd.server_close()
+        if self.old_path is not None:
+            os.chdir(self.old_path)
+
+if __name__ == '__main__':
+    with HTTPServer("/tmp", "127.0.0.1") as server:
+        print("server ip", server.ip)
+        print("server port", server.port)
+        time.sleep(300)
+