2016-09-06 39 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應用程序。