From: Christian Heller Date: Sun, 15 Nov 2020 23:49:37 +0000 (+0100) Subject: Add basic flood protection to server. X-Git-Url: https://plomlompom.com/repos/%7B%7Bdb.prefix%7D%7D/%7B%7B%20web_path%20%7D%7D/blog?a=commitdiff_plain;h=baab532ba5125527bdc7061864f7cc7db8d7abc3;p=plomrogue2 Add basic flood protection to server. --- diff --git a/plomrogue/io.py b/plomrogue/io.py index 8e0f9bf..68f580b 100644 --- a/plomrogue/io.py +++ b/plomrogue/io.py @@ -14,10 +14,21 @@ class GameIO(): self.servers = [] def loop(self, q): - """Handle commands coming through queue q, run game, send results back.""" + """Handle commands coming through queue q, run game, send results back. + + As basic flood protection, Only accepts one command per connection per + 1/100 of a second. + + """ + import time + potential_flooders = {} while True: try: command, connection_id = q.get(timeout=0.001) + if connection_id in potential_flooders: + if int(time.time() * 100) == potential_flooders[connection_id]: + continue + potential_flooders[connection_id] = int(time.time() * 100) self.handle_input(connection_id, command) except queue.Empty: self.game.run_tick()