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

fix: prevent user contributed plugins to crash the main process while loading

Browse Source
This commit is contained in:
Simone Margaritelli 2019-10-24 17:23:33 +02:00
parent 30e3898f3c
commit 61b957ac77
1 changed files with 12 additions and 8 deletions

20
pwnagotchi/plugins/__init__.py

@ -35,14 +35,18 @@ def load_from_file(filename):
def load_from_path(path, enabled=()): def load_from_path(path, enabled=()):
global loaded global loaded
for filename in glob.glob(os.path.join(path, "*.py")): for filename in glob.glob(os.path.join(path, "*.py")):
name, plugin = load_from_file(filename) try:
if name in loaded: name, plugin = load_from_file(filename)
raise Exception("plugin %s already loaded from %s" % (name, plugin.__file__)) if name in loaded:
elif name not in enabled: raise Exception("plugin %s already loaded from %s" % (name, plugin.__file__))
# print("plugin %s is not enabled" % name) elif name not in enabled:
pass # print("plugin %s is not enabled" % name)
else: pass
loaded[name] = plugin else:
loaded[name] = plugin
except Exception as e:
logging.warning("error while loading %s: %s" % (filename, e))
logging.debug(e, exc_info=True)
return loaded return loaded