mirror of
https://git.alemi.dev/pc-monitor.git
synced 2024-11-22 15:34:49 +01:00
52 lines
1.5 KiB
Python
Executable file
52 lines
1.5 KiB
Python
Executable file
#!/usr/bin/env python
|
|
import sys
|
|
import struct
|
|
from time import sleep
|
|
|
|
import serial
|
|
import psutil
|
|
|
|
def cpu_load_serial_driver(device:str, retry_interval:float=5.0):
|
|
while True:
|
|
try:
|
|
port = serial.Serial(device, baudrate=57600)
|
|
avg_usage_to_serial(port)
|
|
except serial.SerialException as e:
|
|
print(f"[!] Could not connect to device: {str(e)}", file=sys.stderr)
|
|
else:
|
|
port.close()
|
|
sleep(retry_interval)
|
|
|
|
def avg_usage_to_serial(port:serial.Serial):
|
|
net = psutil.net_io_counters()
|
|
net_tx = net.bytes_sent
|
|
net_rx = net.bytes_recv
|
|
port.write(struct.pack("BB", 0, 0))
|
|
port.flush()
|
|
while True:
|
|
# Map float [0:100] to int [0:255], square it to put more values in the lower end, where led is more sensible
|
|
load = [ int(((x/100) **2) * 255) for x in psutil.cpu_percent(0.05, percpu=True) ] # mypy whines but percpu returns a list
|
|
try:
|
|
port.write(struct.pack("BBBBBB", 1, 4, *load))
|
|
port.flush()
|
|
except serial.SerialException as e:
|
|
print(f"[!] Failed writing payload to device: {str(e)}", file=sys.stderr)
|
|
break
|
|
net = psutil.net_io_counters()
|
|
try:
|
|
port.write(struct.pack("BBBB", 2, 2, int(net.bytes_sent > net_tx), int(net.bytes_recv > net_rx)))
|
|
port.flush()
|
|
except serial.SerialException as e:
|
|
print(f"[!] Failed writing payload to device: {str(e)}", file=sys.stderr)
|
|
break
|
|
net_rx = net.bytes_recv
|
|
net_tx = net.bytes_sent
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 2:
|
|
print("[!] No device specified")
|
|
exit(-1)
|
|
|
|
cpu_load_serial_driver(sys.argv[1])
|
|
|