-
Notifications
You must be signed in to change notification settings - Fork 784
/
Copy pathtest_parse_target.py
112 lines (93 loc) · 4.15 KB
/
test_parse_target.py
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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import unittest
from pocsuite3.lib.core.common import parse_target
from pocsuite3.lib.core.common import OrderedSet
class TestCase(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
def test_domain(self):
result = OrderedSet()
result.add('example.com')
self.assertEqual(parse_target('example.com'), result)
def test_domain_url(self):
result = OrderedSet()
result.add('https://example.com/cgi-bin/test.cgi?a=b&c=d')
self.assertEqual(parse_target('https://example.com/cgi-bin/test.cgi?a=b&c=d'), result)
def test_domain_url_with_additional_ports(self):
result = OrderedSet()
result.add('https://example.com:8080/cgi-bin/test.cgi?a=b&c=d')
result.add('https://example.com:8443/cgi-bin/test.cgi?a=b&c=d')
result.add('http://example.com:10000/cgi-bin/test.cgi?a=b&c=d')
self.assertEqual(parse_target('https://example.com/cgi-bin/test.cgi?a=b&c=d',
[8080, 8443, 'http:10000'], True), result)
def test_ipv4_url(self):
result = OrderedSet()
result.add('172.16.218.1/cgi-bin')
self.assertEqual(parse_target('172.16.218.1/cgi-bin'), result)
def test_ipv6_url(self):
result = OrderedSet()
result.add('https://[fd12:3456:789a:1::f0]:8443/test')
self.assertEqual(parse_target('https://[fd12:3456:789a:1::f0]:8443/test'), result)
def test_ipv4(self):
result = OrderedSet()
result.add('192.168.1.1')
self.assertEqual(parse_target('192.168.1.1'), result)
def test_ipv4_cidr(self):
result = OrderedSet()
result.add('192.168.1.0')
result.add('192.168.1.1')
self.assertEqual(parse_target('192.168.1.1/31'), result)
def test_ipv4_cidr_with_host_32(self):
result = OrderedSet()
result.add('192.168.1.1')
self.assertEqual(parse_target('192.168.1.1/32'), result)
def test_ipv4_with_additional_ports(self):
result = OrderedSet()
result.add('172.16.218.0:8080')
result.add('172.16.218.0:8443')
result.add('https://172.16.218.0:10000')
result.add('172.16.218.1:8080')
result.add('172.16.218.1:8443')
result.add('172.16.218.1:8443')
result.add('https://172.16.218.1:10000')
self.assertEqual(parse_target('172.16.218.1/31', [8080, 8443, 'https:10000'], True), result)
def test_ipv6(self):
result = OrderedSet()
result.add('fd12:3456:789a:1::1')
self.assertEqual(parse_target('fd12:3456:789a:1::1'), result)
def test_ipv6_cidr(self):
result = OrderedSet()
result.add('fd12:3456:789a:1::1')
result.add('fd12:3456:789a:1::2')
result.add('fd12:3456:789a:1::3')
self.assertEqual(parse_target('fd12:3456:789a:1::/126'), result)
def test_ipv6_cidr_with_host_128(self):
result = OrderedSet()
result.add('fd12:3456:789a:1::')
self.assertEqual(parse_target('fd12:3456:789a:1::/128'), result)
def test_ipv6_with_additional_ports(self):
result = OrderedSet()
result.add('fd12:3456:789a:1::1')
result.add('[fd12:3456:789a:1::1]:8080')
result.add('[fd12:3456:789a:1::1]:8443')
result.add('https://[fd12:3456:789a:1::1]:10000')
result.add('fd12:3456:789a:1::2')
result.add('[fd12:3456:789a:1::2]:8080')
result.add('[fd12:3456:789a:1::2]:8443')
result.add('https://[fd12:3456:789a:1::2]:10000')
result.add('fd12:3456:789a:1::3')
result.add('[fd12:3456:789a:1::3]:8080')
result.add('[fd12:3456:789a:1::3]:8443')
result.add('https://[fd12:3456:789a:1::3]:10000')
self.assertEqual(parse_target('fd12:3456:789a:1::/126', [8080, 8443, 'https:10000']), result)
def test_localhost(self):
result = OrderedSet()
result.add('localhost')
self.assertEqual(parse_target('localhost'), result)
def test_random_str(self):
result = OrderedSet()
result.add('!@#$%^&*()_-+=:::::<>""{}[]:::8080')
self.assertEqual(parse_target('!@#$%^&*()_-+=:::::<>""{}[]:::8080'), result)