def registerPoc(self): pocString = multipleReplace(self.pocString, POC_IMPORTDICT) _, self.moduleName = filepathParser(self.pocName) try: importer = StringImporter(self.moduleName, pocString) importer.load_module(self.moduleName) except ImportError, ex: logger.log(CUSTOM_LOGGING.ERROR, ex)
def registerJsonPoc(path): _, pocname = filepathParser(path) if pocname in kb.registeredPocs: return with open(path) as f: jsonPoc = json.load(f) kb.registeredPocs[pocname] = jsonPoc
def registerPyPoc(path): _, moduleName = filepathParser(path) module = changeToPyImportType(path) try: __import__(module, fromlist=["*"]) except ImportError: # TODO 需要再搞一下 addSysPath(_) __import__(moduleName, fromlist=["*"])
def registerPyPoc(pocDict): pocname = pocDict.keys()[0] _, moduleName = filepathParser(pocname) try: importer = StringImporter(moduleName, pocDict[pocname]) importer.load_module(moduleName) except ImportError, ex: errMsg = "%s register failed \"%s\"" % (moduleName, str(ex)) logger.log(CUSTOM_LOGGING.ERROR, errMsg)
def do_search(self, args): graph = PrettyTable(["pocId", "PocName"]) graph.align["PocName"] = "m" graph.padding_width = 1 for k, v in kb.unloadedList.iteritems(): if str(args) in v: graph.add_row([k, filepathParser(v)[1]]) print graph pass
def avaliable(): graph = PrettyTable(["pocId", "avaliablePocName", "Folder"]) graph.align["PocsName"] = "m" graph.padding_width = 1 for k, v in kb.unloadedList.iteritems(): path, name = filepathParser(v) graph.add_row([k, name, os.path.relpath(path, paths.POCSUITE_ROOT_PATH)]) print graph print
def registerPyPoc(path): _, moduleName = filepathParser(path) module = changeToPyImportType(path) try: __import__(module, fromlist=["*"]) except ImportError: # TODO 需要再搞一下 addSysPath(_) try: __import__(moduleName, fromlist=["*"]) except ImportError, ex: errMsg = "%s register failed \"%s\"" % (path, str(ex)) logger.log(CUSTOM_LOGGING.ERROR, errMsg)