File: test-sgpio-debug.py

package info (click to toggle)
hackrf 2024.02.1-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 36,692 kB
  • sloc: ansic: 56,310; xml: 3,424; perl: 2,730; python: 1,427; makefile: 598; asm: 514; vhdl: 319; sh: 179; awk: 20
file content (87 lines) | stat: -rw-r--r-- 3,350 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#!/usr/bin/python3
import os
import sys
import subprocess
from pathlib import Path

FILENAME = f"/tmp/rx_100kB_{str(os.getpid())}"


def program_device():
    # build new firmware with SGPIO_DEBUG mode enabled
    print("Programming device...")
    fw_dir  = os.getcwd() + "/firmware/hackrf_usb/build"
    del_dir = subprocess.run(["rm", "-rf", "firmware/hackrf_usb/build"])
    mk_dir  = subprocess.run(["mkdir", "firmware/hackrf_usb/build"])
    cmake   = subprocess.run(["cmake", "-D", "SGPIO_DEBUG=1", ".."],
                             cwd=fw_dir, stdout=subprocess.DEVNULL)
    make    = subprocess.run(["make"],
                             cwd=fw_dir, stdout=subprocess.DEVNULL)
    program = subprocess.run(["./ci-scripts/test-firmware-program.sh"],
                             stdout=subprocess.DEVNULL)


def capture():
    shortfall_count = -1
    capture_tries = 0

    while shortfall_count != 0:
        print("Capturing data...")
        rx_100kB = subprocess.run(["host/build/hackrf-tools/src/hackrf_transfer",
                                    "-r", FILENAME, "-d", "RunningFromRAM",
                                    "-n", "50000", "-s", "20000000"],
                                    capture_output=True, encoding="UTF-8")
        print(rx_100kB.stdout)
        print(rx_100kB.stderr)
        print(f"Wrote capture data to file: {FILENAME}")

        debug_state_proc = subprocess.run(["host/build/hackrf-tools/src/hackrf_debug", "--state"],
                                            capture_output=True, encoding="UTF-8")
        print(debug_state_proc.stdout)
        print(debug_state_proc.stderr)
        capture_tries += 1
        debug_state = debug_state_proc.stdout.split("\n")
        shortfalls_line = [s for s in debug_state if s.startswith("Number of shortfalls")]
        shortfall_count = [int(c) for c in shortfalls_line[0].split() if c.isdigit()][0]

        if capture_tries == 10:
            print("Unable to transmit data with 0 shortfalls. " \
            "This is not indicative of a device failure. " \
            "Likely an issue with the testing infrastructure.")
            sys.exit(1)


def check_bytes():
    print(f"Checking length of {FILENAME}")
    rx_data = Path(FILENAME).read_bytes()
    if len(rx_data) != 100000: # file should be 100k bytes when using 50k samples
        print(f"ERROR: Only {str(len(rx_data))} bytes found in file, expected 100k.")
        sys.exit(1)
    else:
        print("Correct file size found.")

    # check that each byte = prev_byte + 1 except at wraparound bounds
    print("Checking bytes...")
    for i in range(1, len(rx_data)):
        if rx_data[i-1] != rx_data[i] - 1:
            if not (rx_data[i] == 0 and rx_data[i-1] == 255):
                print(f"ERROR: Incorrect data value found at location {str(i)} in {FILENAME}:")
                # print up to 5 values starting from at most 1 value before error occurence
                j = -1
                while j < 4:
                    if i + j < len(rx_data) and i + j > -1:
                        print(f"{str(i+j)} : {str(rx_data[i+j])}")
                    j = j + 1
                sys.exit(1)
    print("Successfully validated all bytes in file.\nSGPIO debug test passed.")


def main():
    program_device()
    capture()
    check_bytes()


if __name__ == "__main__":
    main()