def __init__(self, parent):
QTreeWidget.__init__(self, parent)
self.name = "Modules"
self.tm = TaskManager()
self.loader = loader.loader()
self.setupUi(self)
self.initTreeModule()
def loadModules(self,
modulesPaths,
displayOutput=None,
defaultConfig=None):
modulesPaths = self.modulesLocalPath(modulesPaths)
self.loader.do_load(modulesPaths, displayOutput, reload=False)
if defaultConfig is not None:
for module in defaultConfig:
TaskManager().ppModules.add(module)
flags = []
arguments = {}
if defaultConfig[module].has_key("flags"):
flags = defaultConfig[module]["flags"]
if defaultConfig[module].has_key("arguments"):
arguments = defaultConfig[module]["arguments"]
TaskManager().ppModules.add(module, arguments, flags)
def __init__(self, parent=None, log=''):
QTextEdit.__init__(self, parent)
self.setupUi(self)
self.name = self.windowTitle()
self.hookTermSize()
Console.__init__(self, sigstp=False)
self.completion = completion.Completion(self)
taskmanager = TaskManager()
self.vfs = vfs.vfs()
self.log = log or ''
if parent is None:
self.eofKey = Qt.Key_D
else:
self.eofKey = None
self.line = QString()
self.lines = []
self.point = 0
self.more = 0
self.reading = 0
self.pointer = 0
self.cursor_pos = 0
self.fgcolor = QColor("white")
self.selcolor = QColor("green")
self.preloop()
self.redirect = RedirectIO()
self.sig = "SSputtext(QString)"
self.connect(self, SIGNAL(self.sig), self.puttext)
self.connect(self, SIGNAL("Sputtext(QString)"), self.puttext)
self.redirect.addparent(self, ["dff.ui.gui.widget.shell", "dff.ui.console.console", "dff.ui.console.completion", "dff.ui.console.line_to_arguments", "dff.api.taskmanager.taskmanager", "dff.api.taskmanager.scheduler", "dff.api.taskmanager.processus"], True)
self.writePrompt()
def validateModule(self):
# get values
args = {}
try:
for argname, lmanager in self.valueArgs.iteritems():
if lmanager.isEnabled():
arg = self.conf.argumentByName(argname)
if arg.inputType() == Argument.Empty:
params = True
else:
params = lmanager.get(argname)
args[argname] = params
genargs = self.conf.generate(args)
self.taskmanager = TaskManager()
self.taskmanager.add(str(self.nameModule), genargs,
["thread", "gui"])
self.accept()
except RuntimeError:
err_type, err_value, err_traceback = sys.exc_info()
err_trace = traceback.format_tb(err_traceback)
err_typeval = traceback.format_exception_only(err_type, err_value)
terr = QString()
detailerr = QString()
for err in err_trace:
detailerr.append(err)
for errw in err_typeval:
terr.append(errw)
detailerr.append(err)
self.messageBox(terr, detailerr)
return
def __init__(self):
Script.__init__(self, "batch")
self.tm = TaskManager()
self.DEBUG = False
self.VERBOSITY = 0
self.lp = LineParser(self.DEBUG, self.VERBOSITY - 1)
self.cm = ConfigManager.Get()
def __init__(self, arguments):
UI.__init__(self, arguments)
self.taskManager = TaskManager()
self.reportManager = ReportManager()
self.registryManager = ModuleProcessusManager().get("winreg")
self.evtxManager = ModuleProcessusManager().get("evtx")
self.sqliteManager = ModuleProcessusManager().get('SqliteDB')
self.root = vfs().getnode("/")
def __init__(self, parent=None):
QTreeWidget.__init__(self, parent)
sizePolicy = QSizePolicy(QSizePolicy.Preferred,
QSizePolicy.MinimumExpanding)
self.setSizePolicy(sizePolicy)
self.loader = loader()
self.taskmanager = TaskManager()
moduleMap = self.ppModuleMap()
self.header().hide()
self.populate(moduleMap)
self.connect(self, SIGNAL("itemClicked(QTreeWidgetItem*, int)"),
self.moduleClicked)
def __init__(self, selection, listmodel):
super(QWidget, self).__init__()
self.setupUi(self)
self.processusManager = ProcessusManager()
self.loader = loader.loader()
self.lmodules = self.loader.modules
self.taskmanager = TaskManager()
self.mainwindow = QApplication.instance().mainWindow
self.createActions()
self.checkedSelection = selection
self.selection = None
self.model = listmodel
self.bookManager = BookmarkManager(self.model)
def __init__(self, selector, vnode):
QWidget.__init__(self)
EventHandler.__init__(self)
self.vnode = vnode.value()
self.filesize = self.vnode.size()
self.tm = TaskManager()
self.selector = selector
self.setLayout(QVBoxLayout())
self.factor = 1
self.parsetime = 0
self.time = time.time()
self.starttime = time.time()
self.createStartOffset()
self.createButtons()
self.createStateInfo()
def execModule(name, type, nameArg, listNode):
tm = TaskManager()
configs = ConfigManager.Get()
mconf = configs.configByName(str(name))
arg = mconf.argumentByName(nameArg)
if isinstance(listNode, Node):
args = mconf.generate({arg.name(): listNode})
tm.add(str(name), args, ["thread", "gui"])
else:
if arg.inputType() == Argument.List:
args = mconf.generate({arg.name(): listNode})
tm.add(str(name), args, ["thread", "gui"])
else:
for i in listNode:
args = mconf.generate({arg.name(): i})
tm.add(str(name), args, ["thread", "gui"])
def __init__(self, selection, listmodel):
super(QWidget, self).__init__()
self.__iconView = False
self.setupUi(self)
self.processusManager = ProcessusManager()
self.loader = loader.loader()
self.lmodules = self.loader.modules
self.taskmanager = TaskManager()
self.mainwindow = QApplication.instance().mainWindow
self.createActions()
self.checkedSelection = selection
self.selection = None
self.model = listmodel
self.bookManager = BookmarkManager(self.model)
#self.document = QtWebKit.QWebView()
#self.document.loadFinished.connect(self.__print)
self.__corrupt = base64.b64encode(
str(QtGui.QImage(":file_broken.png").bits()))
self.__printer = QtGui.QPrinter(QtGui.QPrinter.ScreenResolution)
self.__printer.setOutputFormat(QtGui.QPrinter.PdfFormat)
self.__printer.setPaperSize(QtGui.QPrinter.A4)
self.__printer.setFullPage(True)
def __init__(self, parent=None):
QTreeWidget.__init__(self, parent)
self.taskManager = TaskManager()
self.setColumnCount(3)
self.setHeaderLabels(["Root", "Items", "Progress"])
self.header().resizeSection(0, 1000)
self.jobItemMap = {}
self.analyseItem = {}
ppsched.registerDisplay(self)
ppsched.processingQueue.registerDisplay(self.rootNodes_s,
self.nodeProcessed_s)
ppsched.processusQueue.registerDisplay(self.setModule_s,
self.moduleProcessed_s)
ppsched.analyseQueue.registerDisplay(self.setAnalyse_s,
self.analyseProcessed_s)
self.connect(self, SIGNAL("rootNodes"), self.rootNodes)
self.connect(self, SIGNAL("nodeProcessed"), self.nodeProcessed)
self.connect(self, SIGNAL("setModule"), self.setModule)
self.connect(self, SIGNAL("moduleProcessed"), self.moduleProcessed)
self.connect(self, SIGNAL("setAnalyse"), self.setAnalyse)
self.connect(self, SIGNAL("analyseProcessed"), self.analyseProcessed)
self.connect(self, SIGNAL("info"), self.info_s)
def __init__(self, parent=None):
QWidget.__init__(self, parent)
self.taskManager = TaskManager()
self.name = "Scanner"
self.setWindowIcon(QIcon(":scan"))
self.state = False
self.label = QLabel()
self.label.setWordWrap(True)
self.tree = PostProcessJobsTree(self)
layout = QVBoxLayout()
layout.setSpacing(0)
layout.setContentsMargins(0, 0, 0, 0)
layout.addWidget(self.label)
layout.addWidget(self.tree)
self.setLayout(layout)
ppsched.registerState(self)
self.startTime = 0
self.lastDuration = 0
self.connect(self, SIGNAL("sigAskWait"), self.sigAskWait,
Qt.BlockingQueuedConnection)
self.connect(self, SIGNAL("sigAskModulesWait"), self.sigAskModulesWait,
Qt.BlockingQueuedConnection)
self.connect(self, SIGNAL("sigAsk"), self.sigAsk,
Qt.BlockingQueuedConnection)
def __init__(self, parent):
QObject.__init__(self)
self.parent = parent
self.vfs = vfs.vfs()
self.taskmanager = TaskManager()
self.loader = loader.loader()
def __init__(self): Script.__init__(self, "post_process") self.tm = TaskManager()
def __init__(self):
Script.__init__(self, "carverui")
self.tm = TaskManager()
def ppModules(self):
return TaskManager().ppModules
def fillFromAnalyse(self):
TaskManager().addAnalyseDependencies()
self.tree.update()
def start(self, args):
try:
self.node = args["file"].value()
TaskManager().add("sqlitedb", args, [])
except Exception as e:
self.node = None
