pep8
This commit is contained in:
Costa Shulyupin
parent
e1c7f3643b
commit
7a987a2bdf
55
srcxray.py
55
srcxray.py
|
|
@ -10,10 +10,15 @@
|
|||
# 2018 Constantine Shulyupin, const@MakeLinux.com
|
||||
#
|
||||
|
||||
import inspect, random, os, sys, collections, subprocess, re
|
||||
import inspect
|
||||
import random
|
||||
import os
|
||||
import sys
|
||||
import collections
|
||||
import subprocess
|
||||
import re
|
||||
|
||||
black_list = ['aligned', '__attribute__', 'unlikely', 'typeof',
|
||||
'u32',
|
||||
black_list = ['aligned', '__attribute__', 'unlikely', 'typeof', 'u32',
|
||||
'PVOP_CALLEE0', 'PVOP_VCALLEE0', 'PVOP_VCALLEE1', 'trace_hardirqs_off']
|
||||
|
||||
level_limit = 7
|
||||
|
|
@ -31,12 +36,15 @@ def print_limited(a):
|
|||
|
||||
|
||||
def log(*args, **kwargs):
|
||||
print(inspect.stack()[1][3], str(*args).rstrip(), file=sys.stderr, **kwargs)
|
||||
print(inspect.stack()[1][3],
|
||||
str(*args).rstrip(), file=sys.stderr, **kwargs)
|
||||
pass
|
||||
|
||||
|
||||
def popen(p):
|
||||
return subprocess.Popen(p, shell=True, stdout=subprocess.PIPE, encoding="utf-8").stdout
|
||||
|
||||
|
||||
def extract_referer(line):
|
||||
line = re.sub(r'__ro_after_init', '', line)
|
||||
line = re.sub(r'FNAME\((\w+)\)', r'\1', line)
|
||||
|
|
@ -47,6 +55,7 @@ def extract_referer(line):
|
|||
if m:
|
||||
return m.group(1)
|
||||
|
||||
|
||||
def extract_referer_test():
|
||||
for a in {
|
||||
"fs=good2()",
|
||||
|
|
@ -59,6 +68,7 @@ def extract_referer_test():
|
|||
"f: a=in bad()"}:
|
||||
print(a, '->', extract_referer(a))
|
||||
|
||||
|
||||
def func_referers_git_grep(name):
|
||||
res = []
|
||||
r = None
|
||||
|
|
@ -73,8 +83,10 @@ def func_referers_git_grep(name):
|
|||
r = extract_referer(line)
|
||||
return res
|
||||
|
||||
|
||||
cscope_warned = False
|
||||
|
||||
|
||||
def func_referers_cscope(name):
|
||||
global cscope_warned
|
||||
if not os.path.isfile('cscope.out'):
|
||||
|
|
@ -82,24 +94,29 @@ def func_referers_cscope(name):
|
|||
print("Recommended: cscope -bkR", file=sys.stderr)
|
||||
cscope_warned = True
|
||||
return []
|
||||
res = [l.split()[1] for l in popen(r'cscope -d -L3 "%s"'%(name)) if not l in black_list]
|
||||
res = [l.split()[1] for l in popen(r'cscope -d -L3 "%s"' %
|
||||
(name)) if not l in black_list]
|
||||
if not res:
|
||||
res = func_referers_git_grep(name)
|
||||
return res
|
||||
|
||||
|
||||
def func_referers_all(name):
|
||||
return set(func_referers_git_grep(name) + func_referers_cscope(name))
|
||||
|
||||
def referers_tree(name, referer=None, printed = None, level = 0):
|
||||
|
||||
def referers_tree(name, referer=None, printed=None, level=0):
|
||||
if not referer:
|
||||
if os.path.isfile('cscope.out'):
|
||||
referer = func_referers_cscope
|
||||
else:
|
||||
print("Using git grep only, recommended to run: cscope -bkR", file=sys.stderr)
|
||||
print("Using git grep only, recommended to run: cscope -bkR",
|
||||
file=sys.stderr)
|
||||
referer = func_referers_git_grep
|
||||
if isinstance(referer, str):
|
||||
referer = eval(referer)
|
||||
if not printed: printed = set()
|
||||
if not printed:
|
||||
printed = set()
|
||||
if name in printed:
|
||||
print_limited(level*'\t' + name + ' ^')
|
||||
return
|
||||
|
|
@ -117,13 +134,15 @@ def referers_tree(name, referer=None, printed = None, level = 0):
|
|||
listed.add(a)
|
||||
return ''
|
||||
|
||||
def call_tree(node, printed = None, level = 0):
|
||||
|
||||
def call_tree(node, printed=None, level=0):
|
||||
if not os.path.isfile('cscope.out'):
|
||||
print("Please run: cscope -bkR", file=sys.stderr)
|
||||
return False
|
||||
if printed == None: printed = set()
|
||||
if printed == None:
|
||||
printed = set()
|
||||
if node in printed:
|
||||
limit= - 1
|
||||
limit = - 1
|
||||
print_limited(level*'\t' + node + ' ^')
|
||||
return
|
||||
else:
|
||||
|
|
@ -133,18 +152,21 @@ def call_tree(node, printed = None, level = 0):
|
|||
print_limited((level + 1)*'\t' + '...')
|
||||
return ''
|
||||
local_printed = set()
|
||||
for line in popen('cscope -d -L2 "%s"'%(node)):
|
||||
for line in popen('cscope -d -L2 "%s"' % (node)):
|
||||
I = line.split()[1]
|
||||
if I in local_printed or I in black_list: continue;
|
||||
if I in local_printed or I in black_list:
|
||||
continue
|
||||
local_printed.add(I)
|
||||
try:
|
||||
call_tree(line.split()[1], printed, level + 1);
|
||||
call_tree(line.split()[1], printed, level + 1)
|
||||
except:
|
||||
pass
|
||||
return ''
|
||||
|
||||
|
||||
me = os.path.basename(sys.argv[0])
|
||||
|
||||
|
||||
def usage():
|
||||
print(me, "referers_tree", "<identifier>")
|
||||
print(me, "call_tree", "<identifier>")
|
||||
|
|
@ -154,6 +176,7 @@ def usage():
|
|||
print(me, "call_tree start_kernel")
|
||||
print(me, "Emergency termination: ^Z, kill %1")
|
||||
|
||||
|
||||
def main():
|
||||
try:
|
||||
ret = False
|
||||
|
|
@ -163,12 +186,14 @@ def main():
|
|||
if '(' in sys.argv[1]:
|
||||
ret = eval(sys.argv[1])
|
||||
else:
|
||||
ret = eval(sys.argv[1] + '(' + ', '.join("'%s'"%(a) for a in sys.argv[2:]) + ')')
|
||||
ret = eval(sys.argv[1] + '(' + ', '.join("'%s'" % (a)
|
||||
for a in sys.argv[2:]) + ')')
|
||||
if type(ret) == type(False) and ret == False:
|
||||
sys.exit(os.EX_CONFIG)
|
||||
print(ret)
|
||||
except KeyboardInterrupt:
|
||||
warning("\nInterrupted")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
|
|||
Loading…
Reference in New Issue