mirror of
https://github.com/polhenarejos/pico-rng.git
synced 2026-04-28 09:43:22 +02:00
45 lines
1.3 KiB
Python
Executable File
45 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import usb.core
|
|
import usb.util
|
|
import random
|
|
import time
|
|
import argparse
|
|
|
|
# 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.")
|
|
args = parser.parse_args()
|
|
|
|
# Get the device
|
|
rng = usb.core.find(idVendor=0x0000, idProduct=0x0004)
|
|
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 endpoints
|
|
endpts = intf.endpoints()
|
|
|
|
# Get the IN or Host RCV Device TX endpoint
|
|
endpts_in = endpts[0] if endpts[0].bEndpointAddress == usb.util.ENDPOINT_IN else endpts[1]
|
|
|
|
# Time tracking for bits/s
|
|
count = 0
|
|
start_time = (int(time.time()) - 1)
|
|
|
|
if args.performance:
|
|
while True:
|
|
try:
|
|
from_device = endpts_in.read(endpts_in.wMaxPacketSize, 500)
|
|
count = count+1
|
|
print(":".join("{:02x}".format(b) for b in from_device), end="")
|
|
print(" KBps {0:.2f}".format((int((count * 64) / (int(time.time()) - start_time))) / 1024 ))
|
|
except KeyboardInterrupt:
|
|
exit(0)
|
|
else:
|
|
print(":".join("{:02x}".format(b) for b in endpts_in.read(endpts_in.wMaxPacketSize, 500)))
|