scifijunkie/pwnagotchi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: don't allow plugins to make the main process crash

Browse Source
This commit is contained in:
Simone Margaritelli 2019-10-09 12:01:44 +02:00
parent 8af5604034
commit fac4b5c460
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pwnagotchi/plugins/__init__.py
View File

@ -1,6 +1,7 @@
import os import os
import glob import glob
import importlib, importlib.util import importlib, importlib.util
import logging
default_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "default") default_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "default")
loaded = {} loaded = {}
@ -13,10 +14,13 @@ def dummy_callback():
def on(event_name, *args, **kwargs): def on(event_name, *args, **kwargs):
global loaded global loaded
cb_name = 'on_%s' % event_name cb_name = 'on_%s' % event_name
for _, plugin in loaded.items(): for plugin_name, plugin in loaded.items():
if cb_name in plugin.__dict__: if cb_name in plugin.__dict__:
# print("calling %s %s(%s)" %(cb_name, args, kwargs)) # print("calling %s %s(%s)" %(cb_name, args, kwargs))
plugin.__dict__[cb_name](*args, **kwargs) try:
plugin.__dict__[cb_name](*args, **kwargs)
except Exception as e:
logging.error("error while running %s.%s : %s" % (plugin_name, cb_name, e))
def load_from_file(filename): def load_from_file(filename):