def do_load(self, args): if args.isdigit(): conf.pocFile = kb.unloadedList[int(args)] del kb.unloadedList[int(args)] pass else: conf.pocFile = args conf.pocname = os.path.split(conf.pocFile)[1] setPoc() print '[*] load poc file(s) success!' print pass
def init(): _setHTTPUserAgent() _setHTTPReferer() _setHTTPCookies() _setHTTPTimeout() _setHTTPExtraHeaders() if conf.pocFile: setPoc() if conf.vulKeyword: conf.pocFile = '%s/%s' % (paths.PENEWORK_MODULES_PATH, conf.vulKeyword) setPoc() requiresCheck() registerPocFromDict() pocViolation() setMultipleTarget() _setHTTPProxy()