|
25 | 25 | import os |
26 | 26 |
|
27 | 27 | parser = argparse.ArgumentParser(description='user-space timerlat thread in Python') |
28 | | -parser.add_argument("cpu", help='CPU to run timerlat thread') |
29 | | -parser.add_argument("-p", "--prio", help='FIFO priority') |
30 | | - |
| 28 | +parser.add_argument("cpu", type=int, help='CPU to run timerlat thread') |
| 29 | +parser.add_argument("-p", "--prio", type=int, help='FIFO priority') |
31 | 30 | args = parser.parse_args() |
32 | 31 |
|
33 | 32 | try: |
34 | | - affinity_mask = { int(args.cpu) } |
35 | | -except: |
36 | | - print("Invalid cpu: " + args.cpu) |
37 | | - exit(1) |
38 | | - |
39 | | -try: |
40 | | - os.sched_setaffinity(0, affinity_mask); |
41 | | -except: |
42 | | - print("Error setting affinity") |
43 | | - exit(1) |
| 33 | + affinity_mask = {args.cpu} |
| 34 | + os.sched_setaffinity(0, affinity_mask) |
| 35 | +except Exception as e: |
| 36 | + print(f"Error setting affinity: {e}") |
| 37 | + sys.exit(1) |
44 | 38 |
|
45 | | -if (args.prio): |
| 39 | +if args.prio: |
46 | 40 | try: |
47 | | - param = os.sched_param(int(args.prio)) |
| 41 | + param = os.sched_param(args.prio) |
48 | 42 | os.sched_setscheduler(0, os.SCHED_FIFO, param) |
49 | | - except: |
50 | | - print("Error setting priority") |
51 | | - exit(1) |
| 43 | + except Exception as e: |
| 44 | + print(f"Error setting priority: {e}") |
| 45 | + sys.exit(1) |
52 | 46 |
|
53 | 47 | try: |
54 | | - timerlat_path = "/sys/kernel/tracing/osnoise/per_cpu/cpu" + args.cpu + "/timerlat_fd" |
| 48 | + timerlat_path = f"/sys/kernel/tracing/osnoise/per_cpu/cpu{args.cpu}/timerlat_fd" |
55 | 49 | timerlat_fd = open(timerlat_path, 'r') |
56 | | -except: |
| 50 | +except PermissionError: |
| 51 | + print("Permission denied. Please check your access rights.") |
| 52 | + sys.exit(1) |
| 53 | +except OSError: |
57 | 54 | print("Error opening timerlat fd, did you run timerlat -U?") |
58 | | - exit(1) |
| 55 | + sys.exit(1) |
59 | 56 |
|
60 | 57 | try: |
61 | | - data_fd = open("/dev/full", 'r'); |
62 | | -except: |
63 | | - print("Error opening data fd") |
| 58 | + data_fd = open("/dev/full", 'r') |
| 59 | +except Exception as e: |
| 60 | + print(f"Error opening data fd: {e}") |
| 61 | + sys.exit(1) |
64 | 62 |
|
65 | 63 | while True: |
66 | 64 | try: |
67 | 65 | timerlat_fd.read(1) |
68 | | - data_fd.read(20*1024*1024) |
69 | | - except: |
| 66 | + data_fd.read(20 * 1024 * 1024) |
| 67 | + except KeyboardInterrupt: |
70 | 68 | print("Leaving") |
71 | 69 | break |
| 70 | + except IOError as e: |
| 71 | + print(f"I/O error occurred: {e}") |
| 72 | + break |
| 73 | + except Exception as e: |
| 74 | + print(f"Unexpected error: {e}") |
| 75 | + break |
72 | 76 |
|
73 | 77 | timerlat_fd.close() |
74 | 78 | data_fd.close() |
0 commit comments