summaryrefslogtreecommitdiffstats
path: root/scripts/trex_show_threads.py
blob: 1824d0738fa106b67836218fff724ac56e44e130 (plain)
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os

def read_task_stats (task_path):

    # files
    status = task_path + "/status"
    stat  = task_path + "/stat"

    stats_dict = {}
    for line in open(status, 'r').readlines():
        name, value = line.split(':', 1)
        stats_dict[name.strip().lower()] = value.strip()

    stat_data = open(stat, 'r').readline().split()

    stats_dict['last_sched_cpu'] = stat_data[-14]

    return stats_dict


def show_threads (pid):
    process_dir = "/proc/{0}/task".format(pid)
    task_paths = ["{0}/{1}".format(process_dir, task) for task in os.listdir(process_dir)]

    
    header = [ 'Task Name', 'PID', 'Allowed CPU', 'Last Sched CPU', 'Asked Ctx Switch', 'Forced Ctx Switch']
    for x in header:
        print('{:^20}'.format(x)),
    print("")

    tasks = []
    for task_path in task_paths:
        task = read_task_stats(task_path)
        tasks.append(task)

    tasks = sorted(tasks, key = lambda x: int(x['cpus_allowed_list']))
    for task in tasks:
        # name
        print("{:<20}".format(task['name'])),
        print("{:^20}".format(task['pid'])),
        print("{:^20}".format(task['cpus_allowed_list'])),
        print("{:^20}".format(task['last_sched_cpu'])),
        print("{:^20}".format(task['voluntary_ctxt_switches'])),
        print("{:^20}".format(task['nonvoluntary_ctxt_switches'])),
        print("")

def isnum (x):
    try:
        int(x)
        return True
    except ValueError:
        return False


def find_trex_pid ():
    procs = [x for x in os.listdir('/proc/') if isnum(x)]
    for proc in procs:
        cmd = open('/proc/{0}/{1}'.format(proc, 'comm')).readline()
        if cmd.startswith('_t-rex-64'):
            return proc

    return None

def main ():
    trex_pid = find_trex_pid()
    if trex_pid is None:
        print("Unable to find Trex PID")
        exit(1)

    print("\nTrex PID on {0}\n".format(trex_pid))
    show_threads(trex_pid)



if __name__ == '__main__':
     main()