From 5d0fbc06c689974fffcb2c5fa786d651ab20eea2 Mon Sep 17 00:00:00 2001
From: Arthur Lu <learthurgo@gmail.com>
Date: Thu, 14 Oct 2021 22:52:15 +0000
Subject: [PATCH] possible fix: UDP broadcast, not connecing from outside LAN

Signed-off-by: Arthur Lu <learthurgo@gmail.com>

Former-commit-id: 531c8f80f36d2170bef582fbb6625377b58a7b2c
---
 src/cli/client.py      | 19 +++++++++++++++++
 src/cli/superscript.py | 46 ++++++++++++------------------------------
 2 files changed, 32 insertions(+), 33 deletions(-)
 create mode 100644 src/cli/client.py

diff --git a/src/cli/client.py b/src/cli/client.py
new file mode 100644
index 0000000..e3e7241
--- /dev/null
+++ b/src/cli/client.py
@@ -0,0 +1,19 @@
+import socket
+
+client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP) # UDP
+
+# Enable port reusage so we will be able to run multiple clients and servers on single (host, port). 
+# Do not use socket.SO_REUSEADDR except you using linux(kernel<3.9): goto https://stackoverflow.com/questions/14388706/how-do-so-reuseaddr-and-so-reuseport-differ for more information.
+# For linux hosts all sockets that want to share the same address and port combination must belong to processes that share the same effective user ID!
+# So, on linux(kernel>=3.9) you have to run multiple servers and clients under one user to share the same (host, port).
+# Thanks to @stevenreddie
+client.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
+
+# Enable broadcasting mode
+client.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
+
+client.connect(("", 5678))
+while True:
+    # Thanks @seym45 for a fix
+    data, addr = client.recvfrom(1024)
+    print("received message: %s"%data)
\ No newline at end of file
diff --git a/src/cli/superscript.py b/src/cli/superscript.py
index 3dd42cc..36432cb 100644
--- a/src/cli/superscript.py
+++ b/src/cli/superscript.py
@@ -151,18 +151,16 @@ __all__ = [
 
 # imports:
 
-import asyncio
 import json
 import math
 from multiprocessing import Pool, freeze_support
 import os
 import pymongo
+import socket
 import sys
-import threading
 import time
 import traceback
 import warnings
-import websockets
 
 from interface import splash, log, ERR, INF, stdout, stderr
 from data import get_previous_time, pull_new_tba_matches, set_current_time, load_match, push_match, load_pit, push_pit, get_database_config, set_database_config, check_new_database_matches
@@ -541,44 +539,26 @@ def start(pid_path, verbose = False, profile = False, debug = False):
 		sys.exit(exit_code)
 
 	else:
-
+		
 		f = open('errorlog.log', 'w+')
 		with daemon.DaemonContext(
-			working_directory=os.getcwd(),
-			pidfile=pidfile.TimeoutPIDLockFile(pid_path),
-			stderr=f
+			working_directory = os.getcwd(),
+			pidfile = pidfile.TimeoutPIDLockFile(pid_path),
+			stderr = f
 			):
 
-			async def handler(client, path):
-				clients.append(client)
-				while True:
-					try:
-						pong_waiter = await client.ping()
-						await pong_waiter
-						time.sleep(3)
-					except Exception as e:
-						clients.remove(client)
-						break
+			server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
+			server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
+			server.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
+			server.settimeout(0.2)
 
-			async def send_one(client, data):
-				await client.send(data)
-				
 			def send(target, level, message, code = 0):
-				message_clients = clients.copy()
-				for client in message_clients:
-					try:
-						asyncio.run(send_one(client, message))
-					except:
-						pass
-
-			clients = []
-			start_server = websockets.serve(handler, "0.0.0.0", 5678)
-
-			asyncio.get_event_loop().run_until_complete(start_server)
-			threading.Thread(target = asyncio.get_event_loop().run_forever).start()
+				server.sendto(bytes(message, 'utf-8'), ('<broadcast>', 5678))
 
 			exit_code = main(send)
-			sys.exit(exit_code)
+			server.close()
+			f.close()
+			sys.exit()
 
 def stop(pid_path):
 	try: