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()
