2016-09-06 80 views
0

我使用gevent及其Backdoor功能。Gevent后门检查运行代码

这是我的代码的简化版本:

from gevent import backdoor, event 

class App(object): 
    def __init__(self): 
     self.stop_event = event.Event() 
     self.servers = [] 
     self.servers.append(backdoor.BackdoorServer((localhost, 6666)) 
     # There is a tcp stream server using gevent configured too 

    def start(self): 
     for server in self.servers: 
      if not server.started: 
       server.start() 
     self.stop_event.wait() 
     for server in self.servers: 
      if server.started: 
       server.stop() 


def run(*args, **kw): 
    app = App() 
    app.start() 

run方法被称为从entry_points创建console_scripts在我setup.py

我想用我的后门来检查app变量本地在我的run功能。

我连接到后门,并运行命令inspect.stack()。这是我得到:

[ 
(<frame object at 0x7fa49ea0b3a0>, '<console>', 1, '<module>', None, None), 
(<frame object at 0x7fa49ce727f0>, '/usr/lib/python2.7/code.py', 103, 'runcode', ['   exec code in self.locals\n'], 0), 
(<frame object at 0x7fa49ce70250>, '/usr/lib/python2.7/code.py', 87, 'runsource', ['  self.runcode(code)\n'], 0), 
(<frame object at 0x7fa49ced7d38>, '/usr/lib/python2.7/code.py', 265, 'push', ['  more = self.runsource(source, self.filename)\n'], 0), 
(<frame object at 0x3569a40>, '/usr/lib/python2.7/code.py', 243, 'interact', ['     more = self.push(line)\n'], 0), 
(<frame object at 0x7fa49ced39b0>, 'path_to_project/venv/local/lib/python2.7/site-packages/gevent-1.0.2-py2.7-linux-x86_64.egg/gevent/backdoor.py', 75, '_run', ['    console.interact(banner=self.banner)\n'], 0), 
(<frame object at 0x7fa49ced3b90>, 'path_to_project/venv/local/lib/python2.7/site-packages/gevent-1.0.2-py2.7-linux-x86_64.egg/gevent/greenlet.py', 327, 'run', ['    result = self._run(*self.args, **self.kwargs)\n'], 0) 
] 

正如你所看到的,gevent似乎是重写堆栈。我的模块启动程序的run函数不在其中。它停止在执行后门程序的greenlet的运行方法中。

有什么办法可以访问运行的run方法,该方法启动了程序以检查它并访问其当地的app变量?

回答

0

好的,我找到了一个解决方案。 greenlet对象具有parent属性来查找哪个产生了您当前正在查看的对象。然后greenlet有一个属性gr_frame来存储堆栈跟踪。

在我的情况

所以,一旦我连接到后门服务器,这将是类似的东西:

>>> import greenlet, inspect 
>>> greenlet.getcurrent() 
<SocketConsole at 0x7fca7122d190> 

>>> greenlet.getcurrent().parent 
<Hub at 0x7fca71560050 epoll default pending=0 ref=4 fileno=3 resolver=<gevent.resolver_thread.Resolver at 0x7fca7126a510 pool=<ThreadPool at 0x7fca7127fe10 0/1/10>> threadpool=<ThreadPool at 0x7fca7127fe10 0/1/10>> 

>>> greenlet.getcurrent().parent.parent 
<greenlet.greenlet object at 0x7fca7d9be690> 

>>> inspect.getouterframes(greenlet.getcurrent().parent.parent.gr_frame) 
[ 
... 
(<frame object at 0x174ac40>, 'path_to_project/endpoints/app.py', 238, 'run', ['  app.start()\n'], 0), 
... 
] 

>>> inspect.getouterframes(greenlet.getcurrent().parent.parent.gr_frame)[4][0] 
<frame object at 0x174ac40> 

>>> inspect.getargvalues(inspect.getouterframes(greenlet.getcurrent().parent.parent.gr_frame)[4][0]).locals 
{'args': [], 'app': <endpoints.app.App object at 0x7fca71360190>, 'kw': {}} 

>>> inspect.getargvalues(inspect.getouterframes(greenlet.getcurrent().parent.parent.gr_frame)[4][0]).locals['app'] 
<endpoints.app.App object at 0x7fca71360190> 

我有运行功能run内noow访问本地对象app检查值的其他属性和调试我的生活gevent应用程序。