From d77885a2b52e14139032b952160c7b6152d565d7 Mon Sep 17 00:00:00 2001
From: Lubos Kosco <tarzanek@gmail.com>
Date: Tue, 18 Jun 2024 09:39:26 +0200
Subject: [PATCH] cloud_scripts/listen_on_ports.py simple http mock server

This can be used for quick port testing, to start a simple service.

E.g.

$ python3 listen_on_ports.py -a 10.0.2.15 -p 1111 -p 2222 -p 3333

Starting on address 10.0.2.15
  Starting on port 1111
  Starting on port 3333
  Starting on port 2222
Starting on address localhost
  Starting on port 1111
  Starting on port 3333
  Starting on port 2222
All ports are listening, press ENTER key to stop and exit.

$ curl 10.0.2.15:2222
up
---
 cloud_scripts/README.md          | 27 ++++++++++++++
 cloud_scripts/listen_on_ports.py | 62 ++++++++++++++++++++++++++++++++
 2 files changed, 89 insertions(+)
 create mode 100644 cloud_scripts/README.md
 create mode 100755 cloud_scripts/listen_on_ports.py

diff --git a/cloud_scripts/README.md b/cloud_scripts/README.md
new file mode 100644
index 0000000..f6fe2b8
--- /dev/null
+++ b/cloud_scripts/README.md
@@ -0,0 +1,27 @@
+Various helper scripts
+-------------------------
+
+HTTP Server
+-----------
+Quick HTTP server mock, can listen on multiple ports and addresses
+```
+$ python3 listen_on_ports.py -a 10.0.2.15 -p 1111 -p 2222 -p 3333
+```
+
+```
+$ curl 10.0.2.15:2222
+up
+```
+
+Port Checker
+------------
+Checks for open ports, if they are listening(default timeout 2s, can be changed):
+```
+$ python3 check_ports.py -r 10.0.2.15 -p 2222 -p 1111 -p 3333
+```
+
+alt. use netcat
+```
+$ nc -zv 10.0.2.15 2222
+Connection to 10.0.2.15 2222 port [tcp/*] succeeded!
+```
diff --git a/cloud_scripts/listen_on_ports.py b/cloud_scripts/listen_on_ports.py
new file mode 100755
index 0000000..6ca9716
--- /dev/null
+++ b/cloud_scripts/listen_on_ports.py
@@ -0,0 +1,62 @@
+#!/usr/bin/python3
+
+import argparse
+import socket
+import os, sys
+from threading import Thread
+from socketserver import ThreadingMixIn
+from http.server import HTTPServer, BaseHTTPRequestHandler
+
+class Handler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        self.send_response(200)
+        self.send_header("Content-type", "text/plain")
+        self.end_headers()
+        self.wfile.write(bytes("up", "utf-8"))
+
+class ThreadingHTTPServer(ThreadingMixIn, HTTPServer):
+    daemon_threads = True
+
+def serve_on_port(address,port):
+    server = ThreadingHTTPServer((address,port), Handler)
+    server.serve_forever()
+
+################################################################################
+argp = argparse.ArgumentParser(description = "Open http server on addresses and ports.", formatter_class=argparse.RawDescriptionHelpFormatter,
+                               epilog=
+'''
+Listens on the given set of TCP ports and given addresses with http server. Replies with "up"
+''')
+argp.add_argument('-a','--address', help='addresses list to bind to', default=["localhost"], action='append', dest='address_set')
+argp.add_argument('-p','--port', help='ports to listen on', default=[], action='append', dest='ports_set')
+
+args = argp.parse_args()
+
+# Remove duplicates
+address_set = set(args.address_set)
+ports_set = set(args.ports_set)
+success = True
+
+for addr in address_set:
+    print("Starting on address {}".format(addr))
+
+    for port in ports_set:
+        checked = True
+
+        print("  Starting on port {}".format(port))
+        try:
+            Thread(target=serve_on_port, args=[addr,int(port)]).start()
+        except Exception as inst:
+            print(type(inst))
+            print(inst.args)
+            print(inst)
+            print("  Cannot listen on addr:port: {}:{}".format(addr, port))
+            success = False
+
+if success:
+    input("All ports are listening, press ENTER key to stop and exit.")
+    os._exit(0)
+else:
+    print("Wasn't able to listen on all ports. Exit.")
+    os._exit(1)
+