def __init__(self):
GrepPlugin.__init__(self)
self._total_count = 0
self._vuln_count = 0
self._vulns = DiskList()
self._ids = DiskList()
def __init__(self):
'''
Class init
'''
GrepPlugin.__init__(self)
self._total_count = 0
self._vulns = DiskList()
self._urls = DiskList()
def test_thread_safe(self):
dl = DiskList()
def worker(range_inst):
for i in range_inst:
dl.append(i)
threads = []
_min = 0
for _max in xrange(0, 1100, 100):
th = threading.Thread(target=worker, args=(xrange(_min, _max), ))
threads.append(th)
_min = _max
for th in threads:
th.start()
for th in threads:
th.join()
for i in xrange(0, 1000):
self.assertTrue(i in dl, i)
dl_as_list = list(dl)
self.assertEqual(len(dl_as_list), len(set(dl_as_list)))
dl_as_list.sort()
self.assertEqual(dl_as_list, range(1000))
def test_fuzzable_request(self):
dl = DiskList()
uri = URL('http://w3af.org/?id=2')
qsr1 = HTTPQSRequest(uri,
method='GET',
headers=Headers([('Referer', 'http://w3af.org/')
]))
uri = URL('http://w3af.org/?id=3')
qsr2 = HTTPQSRequest(uri,
method='OPTIONS',
headers=Headers([('Referer', 'http://w3af.org/')
]))
uri = URL('http://w3af.org/?id=7')
qsr3 = HTTPQSRequest(uri,
method='FOO',
headers=Headers([('Referer', 'http://w3af.org/')
]))
dl.append(qsr1)
dl.append(qsr2)
self.assertEqual(dl[0], qsr1)
self.assertEqual(dl[1], qsr2)
self.assertFalse(qsr3 in dl)
self.assertTrue(qsr2 in dl)
def test_len(self):
dl = DiskList()
for i in xrange(0, 100):
_ = dl.append(i)
self.assertEqual(len(dl) == 100, True)
def __init__(self, scroll_bar, active_filter, possible):
'''
:param scroll_bar: Gtk Vertical Scrollbar object
:param active_filter: the filter active at startup.
:param possible: all filter keys
'''
gtk.TextView.__init__(self)
MessageConsumer.__init__(self)
self.set_editable(False)
self.set_cursor_visible(False)
self.set_wrap_mode(gtk.WRAP_WORD)
self.textbuffer = self.get_buffer()
self.show()
self.possible = set(possible)
self.active_filter = active_filter
self.text_position = 0
self.all_messages = DiskList()
# scroll bar
self.freeze_scrollbar = False
scroll_bar.connect("value-changed", self.scroll_changed)
# colors
self.textbuffer.create_tag("red-fg", foreground="red")
self.textbuffer.create_tag("blue-fg", foreground="blue")
self.textbuffer.create_tag("brown-fg", foreground="brown")
self.bg_colors = {
"vulnerability": "red-fg",
"information": "blue-fg",
"error": "brown-fg",
}
def __init__(self):
AuditPlugin.__init__(self)
# Internal variables
self._potential_vulns = DiskList()
# User configured variables
self._diff_ratio = 0.30
def test_slice_all(self):
disk_list = DiskList()
disk_list.append('1')
disk_list.append('2')
dl_copy = disk_list[:]
self.assertIn('1', dl_copy)
self.assertIn('2', dl_copy)
def test_many_instances(self):
all_instances = []
amount = 200
for _ in xrange(amount):
disk_list = DiskList()
all_instances.append(disk_list)
self.assertEqual(len(all_instances), amount)
def __init__(self):
AuditPlugin.__init__(self)
# Internal variables
self._expected_res_mutant = DiskDict()
self._freq_list = DiskList()
re_str = '<!--#exec cmd="echo -n (.*?);echo -n (.*?)" -->'
self._extract_results_re = re.compile(re_str)
def test_slice_first_N(self):
disk_list = DiskList()
disk_list.append('1')
disk_list.append('2')
disk_list.append('3')
dl_copy = disk_list[:1]
self.assertIn('1', dl_copy)
self.assertNotIn('2', dl_copy)
self.assertNotIn('3', dl_copy)
def test_urlobject(self):
dl = DiskList()
dl.append(URL('http://w3af.org/?id=2'))
dl.append(URL('http://w3af.org/?id=3'))
self.assertEqual(dl[0], URL('http://w3af.org/?id=2'))
self.assertEqual(dl[1], URL('http://w3af.org/?id=3'))
self.assertFalse(URL('http://w3af.org/?id=4') in dl)
self.assertTrue(URL('http://w3af.org/?id=2') in dl)
def test_remove_table(self):
disk_list = DiskList()
table_name = disk_list.table_name
db = get_default_temp_db_instance()
self.assertTrue(db.table_exists(table_name))
disk_list.cleanup()
self.assertFalse(db.table_exists(table_name))
def test_sorted(self):
dl = DiskList()
dl.append('abc')
dl.append('def')
dl.append('aaa')
sorted_dl = sorted(dl)
self.assertEqual(['aaa', 'abc', 'def'], sorted_dl)
def __init__(self):
GrepPlugin.__init__(self)
# Internal variables
self._already_added = DiskList()
# Compile all regular expressions and store information to avoid
# multiple queries to the same function
self._common_directories = get_common_directories()
self._compiled_regexes = {}
self._compile_regex()
def test_getitem(self):
dl = DiskList()
dl.append('a')
dl.append(1)
dl.append([3, 2, 1])
self.assertEqual(dl[0] == 'a', True)
self.assertEqual(dl[1] == 1, True)
self.assertEqual(dl[2] == [3, 2, 1], True)
self.assertRaises(IndexError, dl.__getitem__, 3)
def test_islice(self):
disk_list = DiskList()
disk_list.extend('ABCDEFG')
EXPECTED = 'CDEFG'
result = ''
for c in itertools.islice(disk_list, 2, None, None):
result += c
self.assertEqual(EXPECTED, result)
def test_getitem_negative(self):
dl = DiskList()
dl.append('a')
dl.append('b')
dl.append('c')
self.assertEqual(dl[-1], 'c')
self.assertEqual(dl[-2], 'b')
self.assertEqual(dl[-3], 'a')
self.assertRaises(IndexError, dl.__getitem__, -4)
def test_reverse_iteration(self):
dl = DiskList()
dl.append(1)
dl.append(2)
dl.append(3)
reverse_iter_res = []
for i in reversed(dl):
reverse_iter_res.append(i)
self.assertEqual(reverse_iter_res, [3, 2, 1])
def test_extend(self):
dl = DiskList()
dl.append('a')
dl.extend([1, 2, 3])
self.assertEqual(len(dl), 4)
self.assertEqual(dl[0], 'a')
self.assertEqual(dl[1], 1)
self.assertEqual(dl[2], 2)
self.assertEqual(dl[3], 3)
def test_clear(self):
dl = DiskList()
dl.append('a')
dl.append('b')
self.assertEqual(len(dl), 2)
dl.clear()
self.assertEqual(len(dl), 0)
def test_ordered_iter(self):
dl = DiskList()
dl.append('abc')
dl.append('def')
dl.append('aaa')
sorted_dl = []
for i in dl.ordered_iter():
sorted_dl.append(i)
self.assertEqual(['aaa', 'abc', 'def'], sorted_dl)
def test_string(self):
dl = DiskList()
for i in xrange(0, 1000):
rnd = ''.join(random.choice(string.letters) for i in xrange(40))
_ = dl.append(rnd)
self.assertEqual(rnd in dl, True)
for i in string.letters:
self.assertNotIn(i, dl)
self.assertIn(rnd, dl)
def test_int(self):
dl = DiskList()
for i in xrange(0, 1000):
_ = dl.append(i)
for i in xrange(0, 1000 / 2):
r = random.randint(0, 1000 - 1)
self.assertEqual(r in dl, True)
for i in xrange(0, 1000 / 2):
r = random.randint(1000, 1000 * 2)
self.assertEqual(r in dl, False)
def test_pickle(self):
dl = DiskList()
dl.append('a')
dl.append(1)
dl.append([3, 2, 1])
values = []
for i in dl:
values.append(i)
self.assertEqual(values[0] == 'a', True)
self.assertEqual(values[1] == 1, True)
self.assertEqual(values[2] == [3, 2, 1], True)
def test_unicode(self):
dl = DiskList()
dl.append(u'à')
dl.append(u'המלצת השבוע')
dl.append([
u'à',
])
self.assertEqual(dl[0], u'à')
self.assertEqual(dl[1], u'המלצת השבוע')
self.assertEqual(dl[2], [
u'à',
])
def __init__(self):
OutputPlugin.__init__(self)
# Internal variables
self._initialized = False
self._style_output_file = os.path.join('plugins', 'output',
'html_file', 'style.css')
# These attributes hold the file pointers
self._file = None
self._aditional_info = DiskList()
# User configured parameters
self._verbose = False
self._output_file_name = '~/report.html'
def __init__(self, w3af):
gtk.DrawingArea.__init__(self)
MessageConsumer.__init__(self)
self.w3af = w3af
self.pangolayout = self.create_pango_layout("")
# store all messages to be able to redraw
self.all_messages = DiskList()
self._need_redraw = 0
# control variables
self.alreadyStopped = False
self.timeGrouping = 2
self.timeBase = int(time.time() * 1000)
self.realLeftMargin = MIZQ
self.gc = None
self._redraw_gen = None
# Go live!
self.connect("expose-event", self.area_expose_cb)
gobject.timeout_add(500, self.draw_handler)
self.show()
def test_not(self):
dl = DiskList()
self.assertFalse(dl)
