def loadPoc(pocFile): if pocFile.endswith(".pyc"): conf.isPycFile = True if conf.isPocString: poc = conf.pocFile if not conf.pocname: if conf.pocFile: conf.pocname = os.path.split(conf.pocFile)[1] else: errMsg = "Use pocString must provide pocname" logger.log(CUSTOM_LOGGING.ERROR, errMsg) pocname = conf.pocname else: pocname = os.path.split(pocFile)[1] poc = readFile(pocFile) if not conf.isPycFile: if not re.search(POC_REGISTER_REGEX, poc): warnMsg = "poc: %s register is missing" % pocname logger.log(CUSTOM_LOGGING.WARNING, warnMsg) className = getPocClassName(poc) poc += POC_REGISTER_STRING.format(className) retVal = multipleReplace(poc, POC_IMPORTDICT) else: retVal = poc return {pocname: retVal}
def setTemporaryPoc(pocFile): pocFilename = "_" + os.path.split(pocFile)[1] if not os.path.isdir(paths.POCSUITE_TMP_PATH): os.makedirs(paths.POCSUITE_TMP_PATH) pocname = os.path.join(paths.POCSUITE_TMP_PATH, pocFilename) poc = readFile(pocFile) retVal = multipleReplace(poc, POC_IMPORTDICT) # TODO 直接写入tmp文件夹 没有考虑是否存在文件 或者使用后删除文件 writeFile(pocname, retVal) return pocname
def setTemporaryPoc(pocFile): pocFilename = os.path.split(pocFile)[1] if not pocFilename.startswith("_"): pocFilename = "_" + pocFilename pocname = os.path.join(paths.POCSUITE_TMP_PATH, pocFilename) poc = readFile(pocFile) retVal = multipleReplace(poc, POC_IMPORTDICT) # TODO 直接写入tmp文件夹 没有考虑是否存在文件 或者使用后删除文件 writeFile(pocname, retVal) return pocname
def isOldVersionPoc(filename): poc = readFile(filename) for _ in OLD_VERSION_CHARACTER: if _ not in poc: return False return True