/usr/share/w3af/plugins/tests/helper.py is in w3af-console 1.1svn5547-1.1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | '''
helper.py
Copyright 2012 Andres Riancho
This file is part of w3af, w3af.sourceforge.net .
w3af is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation version 2 of the License.
w3af is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with w3af; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
'''
import os
import unittest
from itertools import chain
import core.data.kb.knowledgeBase as kb
from core.controllers.w3afCore import w3afCore
from core.controllers.w3afException import w3afException
from core.controllers.coreHelpers.exception_handler import exception_handler
from core.controllers.misc.homeDir import W3AF_LOCAL_PATH
from core.data.options.option import option as Option
from core.data.options.comboOption import comboOption as ComboOption
from core.data.options.optionList import optionList as OptionList
os.chdir(W3AF_LOCAL_PATH)
class PluginTest(unittest.TestCase):
runconfig = {}
kb = kb.kb
def setUp(self):
self.kb.cleanup()
self.w3afcore = w3afCore()
def _scan(self, target, plugins):
'''
Setup env and start scan. Typically called from children's
test methods.
@param target: The target to scan.
@param plugins: PluginConfig objects to activate and setup before
the test runs.
'''
def _targetoptions(*target):
opts = OptionList()
opt = Option('target', '', '', Option.LIST)
opt.setValue(','.join(target))
opts.add(opt)
opt = ComboOption(
'targetOS', ('unknown','unix','windows'), '', 'combo')
opts.add(opt)
opt = ComboOption(
'targetFramework',
('unknown', 'php','asp', 'asp.net',
'java','jsp','cfm','ruby','perl'),
'', 'combo'
)
opts.add(opt)
return opts
# Set target(s)
if isinstance(target, basestring):
target = (target,)
self.w3afcore.target.setOptions(_targetoptions(*target))
# Enable plugins to be tested
for ptype, plugincfgs in plugins.items():
self.w3afcore.plugins.setPlugins([p.name for p in plugincfgs], ptype)
for pcfg in plugincfgs:
plugin_instance = self.w3afcore.plugins.getPluginInstance(pcfg.name, ptype)
default_option_list = plugin_instance.getOptions()
unit_test_options = pcfg.options
for option in default_option_list:
if option.getName() not in unit_test_options:
unit_test_options.add(option)
self.w3afcore.plugins.setPluginOptions(ptype, pcfg.name, unit_test_options)
# Verify env and start the scan
self.w3afcore.plugins.init_plugins()
self.w3afcore.verifyEnvironment()
self.w3afcore.start()
def tearDown(self):
# I want to make sure that we don't have *any hidden* exceptions in our tests.
self.assertEquals( len(exception_handler.get_all_exceptions() ), 0)
self.w3afcore.quit()
self.kb.cleanup()
class PluginConfig(object):
BOOL = 'boolean'
STR = 'string'
LIST = 'list'
INT = 'integer'
URL = 'url'
def __init__(self, name, *opts):
self._name = name
self._options = OptionList()
for optname, optval, optty in opts:
self._options.append( Option(optname, optval, '', optty) )
@property
def name(self):
return self._name
@property
def options(self):
return self._options
|