def initOptions(inputOptions=AttribDict()): if IS_WIN: coloramainit() # TODO conf.url = inputOptions.url conf.urlFile = inputOptions.urlFile conf.mode = inputOptions.Mode conf.pocFile = inputOptions.pocFile conf.randomAgent = inputOptions.randomAgent conf.agent = inputOptions.agent conf.cookie = inputOptions.cookie conf.headers = inputOptions.headers conf.referer = inputOptions.referer conf.threads = inputOptions.threads conf.report = inputOptions.report conf.proxy = inputOptions.proxy conf.proxyCred = inputOptions.proxyCred conf.timeout = inputOptions.timeout conf.httpHeaders = HTTP_DEFAULT_HEADER conf.params = inputOptions.extra_params if inputOptions.extra_params else None conf.retry = int(inputOptions.retry) if inputOptions.retry else None conf.delay = float(inputOptions.delay) if inputOptions.delay else None if inputOptions.host: conf.httpHeaders.update({'Host': inputOptions.host}) try: conf.isPocString = inputOptions.isPocString conf.pocname = inputOptions.pocname except: conf.isPocString = False conf.isPycFile = False initializeKb()
def init_options(input_options=None): if input_options is None: input_options = AttribDict() # if IS_WIN: # coloramainit() # 好像初始化了没有用 conf.update(input_options) conf.http_headers = HTTP_DEFAULT_HEADER conf.params = input_options.extra_params if input_options.host: conf.http_headers.upadte({'Host': input_options.host}) try: conf.is_poc_string = init_options.is_poc_string conf.poc_name = input_options.poc_name except Exception: conf.is_poc_string = False conf.is_pyc_file = False initialize_kb()
# -*- coding: utf-8 -*- """ Copyright (c) 2014-2015 pocsuite developers (http://sebug.net) See the file 'docs/COPYING' for copying permission """ from pocsuite.lib.core.datatype import AttribDict from pocsuite.lib.core.log import LOGGER from pocsuite.lib.core.defaults import defaults # logger logger = LOGGER # object to share within function and classes command # line options and settings conf = AttribDict() # Dictionary storing # (1)targets, (2)registeredPocs, (3) bruteMode # (4)results, (5)pocFiles # (6)multiThreadMode \ threadContinue \ threadException kb = AttribDict() cmdLineOptions = AttribDict() registeredPocs = {} # pocsuite paths paths = AttribDict() defaults = AttribDict(defaults)
def setup_class(self): self.test1 = AttribDict() self.test2 = AttribDict(indict={'a': 1, 'b': 2}) self.test3 = AttribDict(attribute='test3')