Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-29 19:08:44 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1597 from cortesi/debug

Browse Source 
debug: add a dump of objects in memory
This commit is contained in:
Aldo Cortesi 2016-10-04 17:00:44 +11:00 committed by GitHub
commit 76ae9fdbaa
2 changed files with 23 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
netlib/debug.py
View File

@ -1,5 +1,6 @@
from __future__ import (absolute_import, print_function, division) from __future__ import (absolute_import, print_function, division)
import gc
import os import os
import sys import sys
import threading import threading
@ -37,7 +38,7 @@ def sysinfo():
return "\n".join(data) return "\n".join(data)
def dump_info(signal=None, frame=None, file=sys.stdout): # pragma: no cover def dump_info(signal=None, frame=None, file=sys.stdout, testing=False): # pragma: no cover
print("****************************************************", file=file) print("****************************************************", file=file)
print("Summary", file=file) print("Summary", file=file)
print("=======", file=file) print("=======", file=file)
@ -78,10 +79,26 @@ def dump_info(signal=None, frame=None, file=sys.stdout): # pragma: no cover
for i in bthreads: for i in bthreads:
print(i._threadinfo(), file=file) print(i._threadinfo(), file=file)
print()
print("Memory", file=file)
print("=======", file=file)
gc.collect()
d = {}
for i in gc.get_objects():
t = str(type(i))
if "mitmproxy" in t or "netlib" in t:
d[t] = d.setdefault(t, 0) + 1
itms = list(d.items())
itms.sort(key=lambda x: x[1])
for i in itms[-20:]:
print(i[1], i[0])
print("****************************************************", file=file) print("****************************************************", file=file)
if not testing:
sys.exit(1)
def dump_stacks(signal=None, frame=None, file=sys.stdout):
def dump_stacks(signal=None, frame=None, file=sys.stdout, testing=False):
id2name = dict([(th.ident, th.name) for th in threading.enumerate()]) id2name = dict([(th.ident, th.name) for th in threading.enumerate()])
code = [] code = []
for threadId, stack in sys._current_frames().items(): for threadId, stack in sys._current_frames().items():
@ -95,6 +112,8 @@ def dump_stacks(signal=None, frame=None, file=sys.stdout):
if line: if line:
code.append(" %s" % (line.strip())) code.append(" %s" % (line.strip()))
print("\n".join(code), file=file) print("\n".join(code), file=file)
if not testing:
sys.exit(1)
def register_info_dumpers(): def register_info_dumpers():

4
test/netlib/test_debug.py
View File

@ -6,13 +6,13 @@ from netlib import debug
def test_dump_info(): def test_dump_info():
cs = StringIO() cs = StringIO()
debug.dump_info(None, None, file=cs) debug.dump_info(None, None, file=cs, testing=True)
assert cs.getvalue() assert cs.getvalue()
def test_dump_stacks(): def test_dump_stacks():
cs = StringIO() cs = StringIO()
debug.dump_stacks(None, None, file=cs) debug.dump_stacks(None, None, file=cs, testing=True)
assert cs.getvalue() assert cs.getvalue()