def test_verify_url(self):
ctarget = CoreTarget()
self.assertRaises(BaseFrameworkException, ctarget._verify_url,
URL_KLASS('ftp://www.google.com/'))
self.assertTrue(ctarget._verify_url(URL_KLASS('http://www.google.com/')))
self.assertTrue(ctarget._verify_url(URL_KLASS('http://www.google.com:39/')))
def get_target(self):
"""
:return: The profile target with the options (target_os,
target_framework, etc.)
"""
# Get the plugin defaults with their types
target_instance = CoreTarget()
options = target_instance.get_options()
for section in self._config.sections():
# Section is something like audit.xss or crawl.web_spider
# or [profile] or [target]
if section == 'target':
for option in self._config.options(section):
options[option].set_value(self._config.get(
section, option))
return options
def get_target(self):
"""
:return: The profile target with the options (target_os,
target_framework, etc.)
"""
# Get the plugin defaults with their types
target_instance = CoreTarget()
options = target_instance.get_options()
for section in self._config.sections():
# Section is something like audit.xss or crawl.web_spider
# or [profile] or [target]
if section == 'target':
for option in self._config.options(section):
options[option].set_value(
self._config.get(section, option))
return options
def test_verify_file_target(self):
ctarget = CoreTarget()
target_file = '/tmp/moth.target'
target = 'file://%s' % target_file
target_file_handler = file(target_file, 'w')
target_file_handler.write('http://moth/1\n')
target_file_handler.write('http://moth/2\n')
target_file_handler.close()
options = ctarget.get_options()
options['target'].set_value(target)
ctarget.set_options(options)
moth1 = URL_KLASS('http://moth/1')
moth2 = URL_KLASS('http://moth/2')
self.assertIn(moth1, cf.cf.get('targets'))
self.assertIn(moth2, cf.cf.get('targets'))
def test_verify_file_target(self):
ctarget = CoreTarget()
target_file = '/tmp/moth.target'
target = 'file://%s' % target_file
target_file_handler = file(target_file, 'w')
target_file_handler.write('http://moth/1\n')
target_file_handler.write('http://moth/2\n')
target_file_handler.close()
options = ctarget.get_options()
options['target'].set_value(target)
ctarget.set_options(options)
moth1 = URL_KLASS('http://moth/1')
moth2 = URL_KLASS('http://moth/2')
self.assertIn(moth1, cf.cf.get('targets'))
self.assertIn(moth2, cf.cf.get('targets'))
def test_basic(self):
opt_lst = CoreTarget().get_options()
for opt in opt_lst:
self.assertIn(opt.get_type(), OPTION_TYPES)
self.assertTrue(opt.get_name())
self.assertEqual(opt, opt)
# Just verify that this doesn't crash and that the types
# are correct
self.assertIsInstance(opt.get_name(), basestring)
self.assertIsInstance(opt.get_desc(), basestring)
self.assertIsInstance(opt.get_type(), basestring)
self.assertIsInstance(opt.get_help(), basestring)
self.assertIsInstance(opt.get_value_str(), basestring)