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')