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 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: pass # TODO
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 registerPoc(self): pocString = multipleReplace(self.pocString, POC_IMPORTDICT) _, fileName = filepathParser(self.pocName) self.moduleName = "%s%s" % (fileName, hashlib.md5( self.target).hexdigest()[:8]) try: importer = StringImporter(self.moduleName, pocString) importer.load_module(self.moduleName) except ImportError, ex: logger.log(CUSTOM_LOGGING.ERROR, ex)
def _init(self): super(self.__class__, self).__init__() conf.isPycFile = False conf.httpHeaders = HTTP_DEFAULT_HEADER try: kb.registeredPocs except Exception: kb.registeredPocs = {} # _setHTTPtimeout timeout = float(TASK_TIMEOUT) socket.setdefaulttimeout(timeout) # registerPoc 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.error(ex)