mirror of
https://github.com/polhenarejos/pico-rng.git
synced 2026-04-09 17:25:51 +02:00
78 lines
2.2 KiB
Python
Executable File
78 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import usb.core
|
|
import usb.util
|
|
import os
|
|
import random
|
|
import time
|
|
import argparse
|
|
import sys
|
|
|
|
# Parser stuff
|
|
parser = argparse.ArgumentParser(description="Raspberry Pi Pico Random Number Generator Test Tool")
|
|
parser.add_argument("--performance", action="store_true", help="Performance test the RNG.")
|
|
parser.add_argument("--endless", action="store_true", help="Outputs random bytes endlessly.")
|
|
parser.add_argument("--size", default="100", help="Number of bytes to output.")
|
|
parser.add_argument("--vid", default="0000", help="VID.")
|
|
parser.add_argument("--pid", default="0004", help="PID.")
|
|
args = parser.parse_args()
|
|
|
|
# If this is set, then the /dev/pico_rng file exists
|
|
rng_chardev = None
|
|
|
|
if os.path.exists("/dev/pico_rng"):
|
|
rng_chardev = open("/dev/pico_rng", "rb")
|
|
|
|
# File does not exist, test with usb.core
|
|
if not rng_chardev:
|
|
# Get the device
|
|
rng = usb.core.find(idVendor=int(args.vid, base=16), idProduct=int(args.pid, base=16))
|
|
assert rng is not None
|
|
|
|
# Get the configuration of the device
|
|
cfg = rng.get_active_configuration()
|
|
|
|
# Get the only interface of our device
|
|
intf = cfg.interfaces()[0]
|
|
|
|
# Get the endpoint
|
|
endpt = intf.endpoints()[0]
|
|
|
|
# Time tracking for bits/s
|
|
count = 0
|
|
start_time = (int(time.time()) - 1)
|
|
|
|
def get_data():
|
|
return rng_chardev.read(64) if rng_chardev else endpt.read(64, 500)
|
|
|
|
def get_and_print():
|
|
data = get_data()
|
|
sys.stdout.buffer.write(data)
|
|
|
|
if args.performance:
|
|
while True:
|
|
try:
|
|
from_device = get_data()
|
|
count = count+1
|
|
#print(from_device, end="")
|
|
print("Speed: {0:.2f} KB/s".format((int((count * 64) / (int(time.time()) - start_time))) / 1024 ), end='\r')
|
|
except KeyboardInterrupt:
|
|
exit(0)
|
|
elif args.endless:
|
|
while True:
|
|
try:
|
|
get_and_print()
|
|
except KeyboardInterrupt:
|
|
exit(0)
|
|
except BrokenPipeError:
|
|
exit(0)
|
|
elif args.size:
|
|
size = int(float(args.size))
|
|
for i in range(0,size,64):
|
|
data = get_data()
|
|
sys.stdout.buffer.write(data[:min(size-i,len(data))])
|
|
else:
|
|
from_device = get_data()
|
|
print(from_device)
|
|
sys.stdout.buffer.write(bytearray(from_device))
|