1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-03-09 15:49:59 +00:00

ST: Support show coroutines.

This commit is contained in:
winlin 2020-12-10 11:36:24 +08:00
parent c7c6d8778a
commit 9caeb606bf

View file

@ -1,30 +1,70 @@
import gdb, sys, traceback; import gdb, traceback;
'''
Usage:
nn_coroutines
nn_coroutines 1000
'''
class NnCouroutines(gdb.Command): class NnCouroutines(gdb.Command):
def __init__(self): def __init__(self):
super(NnCouroutines, self).__init__('nn_coroutines', gdb.COMMAND_DATA) super(NnCouroutines, self).__init__('nn_coroutines', gdb.COMMAND_DATA)
# https://sourceware.org/gdb/current/onlinedocs/gdb/Python-API.html#Python-API
def invoke(self, arg, from_tty): def invoke(self, arg, from_tty):
nn_interval=1000
nn_coroutines = 1
# Parse args.
args = arg.split(' ')
if args[0] != '':
nn_interval = int(args[0])
try: try:
if len(arg) > 0: pnext = prev = pthis = gdb.parse_and_eval('&_st_this_thread->tlink').__str__()
prev = pthis = arg print('interval: %s, first: %s, args(%s): %s'%(nn_interval, pthis, len(args), args))
print('start from %s'%(arg))
else:
prev = pthis = gdb.parse_and_eval('&_st_this_thread->tlink').__str__()
print('start from &_st_this_thread->tlink')
print('first is: %s'%(pthis))
nn_coroutines, pnext = 1, pthis
while True: while True:
v = gdb.parse_and_eval('((_st_clist_t*)%s)->next'%pnext) v = gdb.parse_and_eval('((_st_clist_t*)%s)->next'%pnext)
prev = pnext = str(v.__str__()).split(' ')[0] prev = pnext = str(v.__str__()).split(' ')[0]
if pnext == pthis: if pnext == pthis:
break break
nn_coroutines += 1 nn_coroutines += 1
if len(arg) > 0 or (nn_coroutines%1000) == 0: if (nn_coroutines%nn_interval) == 0:
print('next is %s, total %s'%(pnext, nn_coroutines)) print('next is %s, total %s'%(pnext, nn_coroutines))
print('total coroutines: %s'%(nn_coroutines))
except: except:
print('Error: prev=%s, this=%s, next=%s, v=%s'%(prev, pthis, pnext, v)) print('Error: prev=%s, this=%s, next=%s, v=%s'%(prev, pthis, pnext, v))
traceback.print_exc() traceback.print_exc()
# Result.
print('total coroutines: %s'%(nn_coroutines))
NnCouroutines() NnCouroutines()
'''
Usage:
show_coroutines
'''
class ShowCouroutines(gdb.Command):
def __init__(self):
super(ShowCouroutines, self).__init__('show_coroutines', gdb.COMMAND_DATA)
# https://sourceware.org/gdb/current/onlinedocs/gdb/Python-API.html#Python-API
def invoke(self, arg, from_tty):
offset = gdb.parse_and_eval('(int)(&(((_st_thread_t*)(0))->tlink))').__str__()
_st_this_thread = gdb.parse_and_eval('_st_this_thread').__str__()
pnext = prev = pthis = gdb.parse_and_eval('&_st_this_thread->tlink').__str__()
this_thread2 = gdb.parse_and_eval('(_st_thread_t*)(%s - %s)'%(pthis, offset)).__str__()
#print('offset=%s, _st_this_thread=%s, pthis-offset=%s'%(offset, _st_this_thread, this_thread2))
try:
while True:
trd = gdb.parse_and_eval('(_st_thread_t*)(%s - %s)'%(pnext, offset)).__str__()
jmpbuf = gdb.parse_and_eval('((_st_thread_t*)%s)->context.__jmpbuf'%(trd)).__str__().split(', ')
rbp, rsp, crip = int(jmpbuf[1]), int(jmpbuf[6]), None
if rbp > 0 and rsp > 0:
crip = gdb.execute('x/2xa %s'%(rbp), to_string=True).split('\t')[2].strip()
print('thread: %s, caller: %s'%(trd, crip))
v = gdb.parse_and_eval('((_st_clist_t*)%s)->next'%pnext)
prev = pnext = str(v.__str__()).split(' ')[0]
if pnext == pthis:
break
except:
print('Error: prev=%s, this=%s, next=%s'%(prev, pthis, pnext))
traceback.print_exc()
ShowCouroutines()