mirror of
https://github.com/EmpireProject/EmPyre
synced 2026-06-08 10:57:53 +00:00
Harmj0y
107 lines
3.6 KiB
Python
107 lines
3.6 KiB
Python
"""
|
|
|
|
Module handling functionality for EmPyre.
|
|
|
|
Right now, just loads up all modules from the
|
|
install path in the common config.
|
|
|
|
"""
|
|
|
|
import sqlite3, fnmatch, os, imp
|
|
import messages
|
|
import helpers
|
|
|
|
class Modules:
|
|
|
|
def __init__(self, MainMenu, args):
|
|
|
|
self.mainMenu = MainMenu
|
|
|
|
# pull the database connection object out of the main menu
|
|
self.conn = self.mainMenu.conn
|
|
|
|
self.args = args
|
|
|
|
# module format:
|
|
# [ ("module/name", instance) ]
|
|
self.modules = {}
|
|
|
|
# pull out the code install path from the database config
|
|
cur = self.conn.cursor()
|
|
cur.execute("SELECT install_path FROM config")
|
|
self.installPath = cur.fetchone()[0]
|
|
cur.close()
|
|
|
|
self.load_modules()
|
|
|
|
def load_modules(self, rootPath=''):
|
|
"""
|
|
Load modules from the install + "/lib/modules/*" path
|
|
"""
|
|
|
|
if rootPath == '':
|
|
rootPath = self.installPath + 'lib/modules/'
|
|
|
|
pattern = '*.py'
|
|
print helpers.color("[*] Loading modules from: %s" %(rootPath))
|
|
|
|
for root, dirs, files in os.walk(rootPath):
|
|
for filename in fnmatch.filter(files, pattern):
|
|
filePath = os.path.join(root, filename)
|
|
|
|
# don't load up the template
|
|
if filename == "template.py" or filename == "jobs-template.py":
|
|
continue
|
|
|
|
# extract just the module name from the full path
|
|
moduleName = filePath.split(rootPath)[-1][0:-3]
|
|
|
|
if rootPath != self.installPath + 'lib/modules/':
|
|
moduleName = "external/%s" %(moduleName)
|
|
|
|
# instantiate the module and save it to the internal cache
|
|
self.modules[moduleName] = imp.load_source(moduleName, filePath).Module(self.mainMenu, [])
|
|
|
|
def reload_module(self, moduleToReload):
|
|
"""
|
|
Reload a specific module from the install + "/lib/modules/*" path
|
|
"""
|
|
|
|
rootPath = self.installPath + 'lib/modules/'
|
|
pattern = '*.py'
|
|
|
|
for root, dirs, files in os.walk(rootPath):
|
|
for filename in fnmatch.filter(files, pattern):
|
|
filePath = os.path.join(root, filename)
|
|
|
|
# don't load up the template
|
|
if filename == "template.py" or filename == "jobs-template.py":
|
|
continue
|
|
|
|
# extract just the module name from the full path
|
|
moduleName = filePath.split("/lib/modules/")[-1][0:-3]
|
|
|
|
# check to make sure we've found the specific module
|
|
if moduleName.lower() == moduleToReload.lower():
|
|
# instantiate the module and save it to the internal cache
|
|
try:
|
|
self.modules[moduleName] = imp.load_source(moduleName, filePath).Module(self.mainMenu, [])
|
|
except Exception as e:
|
|
print helpers.color("[!] Error: Failed to reload module, not instantiate new module!")
|
|
print helpers.color("[!] Module Error:" + str(e), color="yellow")
|
|
|
|
def search_modules(self, searchTerm):
|
|
"""
|
|
Search currently loaded module names and descriptions.
|
|
"""
|
|
|
|
print ""
|
|
|
|
for moduleName, module in self.modules.iteritems():
|
|
if searchTerm.lower() in moduleName.lower() or searchTerm.lower() in ''.join(module.info['Description']).lower():
|
|
messages.display_module_search(moduleName, module)
|
|
|
|
# for comment in module.info['Comments']:
|
|
# if searchTerm.lower() in comment.lower():
|
|
# messages.display_module_search(moduleName, module)
|