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