iot-ip-beacon/main.py

#!/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.")