server_ping.py - main - ra0sms/caesar_project

server_ping.py
67 строк2 KB
ra0sms
del ports 5003 and 5004
20 июн 2025, 11:25
20 июн 2025, 11:256783b1d
#!/usr/bin/python3
import socket
import time
from wiringpi import GPIO
import wiringpi
import os

CONN_PIN = 11
UDP_PORT = 5002
TIMEOUT = 1.0
CHECK_INTERVAL = 0.3
MAGIC_PHRASE = b"PING_RESPONSE"

def read_ip_from_file(file_path):
    try:
        with open(file_path, "r") as file:
            ip_address = file.read().strip()
            if not ip_address:
                raise ValueError("IP address not found or empty.")
            return ip_address
    except FileNotFoundError:
        print(f"File {file_path} not found.")
        exit(1)
    except Exception as e:
        print(f"Reading error: {e}")
        exit(1)

def check_udp_availability(ip_address):
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        sock.settimeout(TIMEOUT)
        sock.sendto(b"PING_REQUEST", (ip_address, UDP_PORT))
        try:
            data, addr = sock.recvfrom(1024)
            return data == MAGIC_PHRASE and addr[0] == ip_address
        except socket.timeout:
            return False
        finally:
            sock.close()
    except Exception as e:
        print(f"UDP check error: {e}")
        return False

def monitor_ip(ip_address, check_interval):
    wiringpi.wiringPiSetup()
    wiringpi.pinMode(CONN_PIN, wiringpi.OUTPUT)
    need_ptt_service_reboot = False
    while True:
        if check_udp_availability(ip_address):
            print(f"Client {ip_address} is online")
            wiringpi.digitalWrite(CONN_PIN, GPIO.HIGH)
            if need_ptt_service_reboot:
                os.system("systemctl restart ptt_client.service")
                print(f"ptt_client service rebooted")
                need_ptt_service_reboot = False
        else:
            print(f"Client {ip_address} is offline")
            wiringpi.digitalWrite(CONN_PIN, GPIO.LOW)
            need_ptt_service_reboot = True
        
        time.sleep(check_interval)

if __name__ == "__main__":
    ip_file_path = "server_ip.cfg"
    ip_to_monitor = read_ip_from_file(ip_file_path)
    print(f"Monitoring UDP availability for: {ip_to_monitor}")
    monitor_ip(ip_to_monitor, CHECK_INTERVAL)