Add timeout to socket connection

This commit is contained in:
Dustin Pianalto 2019-12-16 23:01:04 -09:00
parent fa92f0057b
commit 8a9618c0ed

View File

@ -66,6 +66,7 @@ class RCONConnection:
try: try:
rcon_log.debug(f'Connecting to {self.host}:{self.port}...') rcon_log.debug(f'Connecting to {self.host}:{self.port}...')
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.socket.settimeout(3)
self.socket.connect((self.host, self.port)) self.socket.connect((self.host, self.port))
except TimeoutError as e: except TimeoutError as e:
rcon_log.error(f'Timeout error: {e}') rcon_log.error(f'Timeout error: {e}')