def keepAlive(): while True: try: # Wait keepAlive period time.sleep(config.timeout / 2) except: break # Connect to master and ask for status print("{time:s} Send KeepAlive".format(time=Utilities.current_time_formatted())) try: conn = messages.Connection(socket.create_connection((config.parent_address, config.parent_port))) connections_manager.add(conn) msg = messages.Status(config.id, [{ "State": "Idle" }]) conn.send(msg) # Inform parent about lower backup registrations and deregistrations for msg in backup_servers.getParentMessages(): conn.send(msg) conn.socket.shutdown(socket.SHUT_WR) # Read all messages from client separated by 0x17 until EOC (write stream closed) for msg in conn: parseMessageBCS(msg) conn.socket.close() connections_manager.remove(conn) except OSError as msg: print("KeepAlive connection error: {:s}".format(str(msg))) print("Assuming main communication server role") # Refresh components timeout, else when assume main cs role all would be thought dead for component in components.active(): component.touch() config.is_backup = False break
def registerBackup(address, port, assumeId=None): print("{time:s} Connecting to {address:s}:{port:d}...".format(time=Utilities.current_time_formatted(), address=address, port=port)) while True: try: conn = messages.Connection(connect((address, port), ('', args.port))) connections_manager.add(conn) break except Exception as msg: print("Trying to bind...") time.sleep(1) print("Send Register message") request = messages.Register(Type="CommunicationServer", SolvableProblems=[], ParallelThreads=0, Id=assumeId) conn.send(request) conn.socket.shutdown(socket.SHUT_WR) for msg in conn: if type(msg) == messages.RegisterResponse: # Has BackupCommunicationServers field -> connect to next backup in list if msg.BackupCommunicationServers: backup = msg.BackupCommunicationServers[0] print("Sent to next BCS at {address:s}:{port:d}".format(address=backup['address'], port=backup['port'])) conn.socket.close() connections_manager.remove(conn) registerBackup(backup['address'], backup['port'], msg.Id) return # Registered at last communication server in list else: print("Registered successfully") config.id = msg.Id config.timeout = msg.Timeout config.parent_address = address config.parent_port = port # Messages with state, sent from higher communication server else: parseMessageBCS(msg) conn.socket.close() connections_manager.remove(conn)
def runServer(): # Open server socket print("Starting server on port {port:d} with timeout {timeout:d}".format(port=args.port, timeout=config.timeout)) server = socket.socket(socket.AF_INET6) server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) while True: try: server.bind(("::", args.port)) break except: print("Trying to bind...") time.sleep(1) server.listen(50) # Main Loop: Listen for incoming connections while True: try: s, addr = server.accept() except: break removeInactiveComponents() conn = messages.Connection(s, timeout=config.timeout) connections_manager.add(conn) threading.Thread(target=handleConnection, args=(conn,)).start()