xref: /linux/tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Util.py (revision 194605d45dcb511983caca699d81855693b25fe0)
1 # Util.py - Python extension for perf script, miscellaneous utility code
2 #
3 # Copyright (C) 2010 by Tom Zanussi <tzanussi@gmail.com>
4 #
5 # This software may be distributed under the terms of the GNU General
6 # Public License ("GPL") version 2 as published by the Free Software
7 # Foundation.
8 from __future__ import print_function
9 
10 import errno, os
11 
12 FUTEX_WAIT = 0
13 FUTEX_WAKE = 1
14 FUTEX_PRIVATE_FLAG = 128
15 FUTEX_CLOCK_REALTIME = 256
16 FUTEX_CMD_MASK = ~(FUTEX_PRIVATE_FLAG | FUTEX_CLOCK_REALTIME)
17 
18 NSECS_PER_SEC    = 1000000000
19 
20 def avg(total, n):
21     return total / n
22 
23 def nsecs(secs, nsecs):
24     return secs * NSECS_PER_SEC + nsecs
25 
26 def nsecs_secs(nsecs):
27     return nsecs / NSECS_PER_SEC
28 
29 def nsecs_nsecs(nsecs):
30     return nsecs % NSECS_PER_SEC
31 
32 def nsecs_str(nsecs):
33     str = "%5u.%09u" % (nsecs_secs(nsecs), nsecs_nsecs(nsecs)),
34     return str
35 
36 def add_stats(dict, key, value):
37 	if key not in dict:
38 		dict[key] = (value, value, value, 1)
39 	else:
40 		min, max, avg, count = dict[key]
41 		if value < min:
42 			min = value
43 		if value > max:
44 			max = value
45 		avg = (avg + value) / 2
46 		dict[key] = (min, max, avg, count + 1)
47 
48 def clear_term():
49     print("\x1b[H\x1b[2J")
50 
51 audit_package_warned = False
52 
53 try:
54 	import audit
55 	machine_to_id = {
56 		'x86_64': audit.MACH_86_64,
57 		'aarch64': audit.MACH_AARCH64,
58 		'alpha'	: audit.MACH_ALPHA,
59 		'ia64'	: audit.MACH_IA64,
60 		'ppc'	: audit.MACH_PPC,
61 		'ppc64'	: audit.MACH_PPC64,
62 		'ppc64le' : audit.MACH_PPC64LE,
63 		's390'	: audit.MACH_S390,
64 		's390x'	: audit.MACH_S390X,
65 		'i386'	: audit.MACH_X86,
66 		'i586'	: audit.MACH_X86,
67 		'i686'	: audit.MACH_X86,
68 	}
69 	try:
70 		machine_to_id['armeb'] = audit.MACH_ARMEB
71 	except:
72 		pass
73 	machine_id = machine_to_id[os.uname()[4]]
74 except:
75 	if not audit_package_warned:
76 		audit_package_warned = True
77 		print("Install the python-audit package to get syscall names.\n"
78                     "For example:\n  # apt-get install python3-audit (Ubuntu)"
79                     "\n  # yum install python3-audit (Fedora)"
80                     "\n  etc.\n")
81 
82 def syscall_name(id):
83 	try:
84 		return audit.audit_syscall_to_name(id, machine_id)
85 	except:
86 		return str(id)
87 
88 def strerror(nr):
89 	try:
90 		return errno.errorcode[abs(nr)]
91 	except:
92 		return "Unknown %d errno" % nr
93