65 lines
1.5 KiB
Python
65 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
from typing import Optional
|
|
import socket
|
|
import time
|
|
import fcntl
|
|
import struct
|
|
|
|
|
|
def get_ip_address(ifname: str) -> str:
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
return socket.inet_ntoa(fcntl.ioctl(s.fileno(), 0x8915, struct.pack('256s', ifname[:15].encode()))[20:24])
|
|
|
|
|
|
def get_uptime() -> float:
|
|
with open('/proc/uptime', 'r') as f:
|
|
uptime_seconds = float(f.readline().split()[0])
|
|
|
|
return uptime_seconds
|
|
|
|
|
|
def get_hostname() -> str:
|
|
return socket.gethostname()
|
|
|
|
|
|
def get_version() -> str:
|
|
try:
|
|
with open('/version', 'r') as f:
|
|
for line in f:
|
|
if line.lower().startswith("version:"):
|
|
return line.split(":", 2)[1].strip()
|
|
|
|
return "u"
|
|
except FileNotFoundError:
|
|
return "u"
|
|
|
|
|
|
def main():
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_BINDTODEVICE, b"enp3s0")
|
|
|
|
while True:
|
|
|
|
ipaddr = ""
|
|
try:
|
|
ipaddr = get_ip_address("enp3s0")
|
|
except:
|
|
pass
|
|
|
|
message = f"BIRBOX|{ipaddr}|{get_hostname()}|{get_version()}|{int(get_uptime())}"
|
|
|
|
sock.sendto(message.encode("utf-8"), ('<broadcast>', 6969))
|
|
time.sleep(10)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("Starting beacon...")
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
print("Beacon stopped.")
|