1
|
|
|
from spike.model.naxsi_rules import NaxsiRules |
2
|
|
|
|
3
|
|
|
from tests import TestsThatNeedsRules |
4
|
|
|
|
5
|
|
|
|
6
|
|
|
try: |
7
|
|
|
from urlparse import urlparse |
8
|
|
|
except ImportError: # python3 |
9
|
|
|
from urllib.parse import urlparse |
10
|
|
|
|
11
|
|
|
|
12
|
|
|
class FlaskrTestCase(TestsThatNeedsRules): |
13
|
|
|
def test_sandbox_rule(self): |
14
|
|
|
rv = self.app.get('/sandbox/rule') |
15
|
|
|
self.assertEqual(rv.status_code, 405) |
16
|
|
|
|
17
|
|
|
rv = self.app.post('/sandbox/rule') |
18
|
|
|
self.assertEqual(rv.status_code, 200) |
19
|
|
|
|
20
|
|
|
def test_sandbox_visualize(self): |
21
|
|
|
data = {'rule': 'MainRule "rx:^POUET$" "msg: sqli" "mz:BODY|URL|ARGS|$HEADERS_VAR:Cookie" "s:$SQL:8" id:1005;', |
22
|
|
|
'visualise_rule': '1'} |
23
|
|
|
rv = self.app.post('/sandbox/rule', data=data) |
24
|
|
|
self.assertEqual(rv.status_code, 302) |
25
|
|
|
self.assertIn('https://regexper.com/#^POUET$', str(rv.data)) |
26
|
|
|
|
27
|
|
|
del data['visualise_rule'] |
28
|
|
|
data['explain_rule'] = 1 |
29
|
|
|
rv = self.app.post('/sandbox/rule', data=data) |
30
|
|
|
_rule = NaxsiRules('sqli', 'rx:^POUET$', 'BODY|URL|ARGS|$HEADERS_VAR:Cookie', '$SQL:8', '1005', "", "sqli") |
31
|
|
|
self.assertIn(str(_rule.explain()), str(rv.data).replace('\\', '')) |
32
|
|
|
|
33
|
|
|
def test_explain_rule(self): |
34
|
|
|
rv = self.app.get('/sandbox/explain_rule/') |
35
|
|
|
self.assertEqual(rv.status_code, 302) |
36
|
|
|
self.assertEqual(urlparse(rv.location).path, '/sandbox/') |
37
|
|
|
|
38
|
|
|
_rule = NaxsiRules.query.order_by(NaxsiRules.sid.desc()).first() |
|
|
|
|
39
|
|
|
rv = self.app.get('/sandbox/explain_rule/?rule={0}'.format(_rule.sid + 1), follow_redirects=True) |
40
|
|
|
self.assertIn('Not rule with id {0}'.format(_rule.sid + 1), str(rv.data)) |
41
|
|
|
|
42
|
|
|
rv = self.app.get('/sandbox/explain_rule/?rule={0}'.format(_rule.sid)) |
43
|
|
|
self.assertEqual(rv.status_code, 200) |
44
|
|
|
self.assertIn(_rule.explain(), str(rv.data)) |
45
|
|
|
|
46
|
|
|
rv = self.app.get('/sandbox/explain_rule/?rule=lol') |
47
|
|
|
self.assertEqual(rv.status_code, 302) |
48
|
|
|
self.assertEqual(urlparse(rv.location).path, '/sandbox/') |
49
|
|
|
|
50
|
|
|
data = 'MainRule "rx:^POUET$" "msg: sqli" "mz:BODY|URL|ARGS|$HEADERS_VAR:Cookie" "s:$SQL:8" id:1005;' |
51
|
|
|
rv = self.app.post('/sandbox/explain_rule/', data={'rule': data}) |
52
|
|
|
self.assertEqual(rv.status_code, 200) |
53
|
|
|
_rule = NaxsiRules() |
54
|
|
|
_rule.parse_rule(data) |
55
|
|
|
self.assertIn(_rule.explain(), str(rv.data)) |
56
|
|
|
|
57
|
|
|
def test_explain_nxlog(self): |
58
|
|
|
rv = self.app.get('/sandbox/explain_nxlog/') |
59
|
|
|
self.assertEqual(rv.status_code, 405) # we only accept POST there. |
60
|
|
|
|
61
|
|
|
rv = self.app.post('/sandbox/explain_nxlog/') |
62
|
|
|
self.assertEqual(rv.status_code, 302) |
63
|
|
|
|
64
|
|
|
rv = self.app.post('/sandbox/explain_nxlog/', data={'nxlog': '1234, lol'}) |
65
|
|
|
self.assertEqual(rv.status_code, 302) |
66
|
|
|
|
67
|
|
|
rv = self.app.post('/sandbox/explain_nxlog/', data={'nxlog': 'ip=1234'}) |
68
|
|
|
self.assertEqual(rv.status_code, 302) |
69
|
|
|
|
70
|
|
|
nxlog = '2013/11/10 07:36:19 [error] 8278#0: *5932 NAXSI_FMT: ip=X.X.X.X&server=Y.Y.Y.Y&' |
71
|
|
|
nxlog += 'uri=/phpMyAdmin-2.8.2/scripts/setup.php&learning=0&vers=0.52&total_processed=472&total_blocked=204&' |
72
|
|
|
nxlog += 'block=0&cscore0=$UWA&score0=8&zone0=HEADERS&id0=42000227&var_name0=user-agent, client: X.X.X.X,' |
73
|
|
|
nxlog += 'server: blog.memze.ro, request: "GET /phpMyAdmin-2.8.2/scripts/setup.php HTTP/1.1", host: "X.X.X.X"' |
74
|
|
|
|
75
|
|
|
rv = self.app.post('/sandbox/explain_nxlog/', data={'nxlog': nxlog}) |
76
|
|
|
|
77
|
|
|
self.assertIn('performed a request to', str(rv.data)) |
78
|
|
|
|
79
|
|
|
def test_explain_whitelist(self): |
80
|
|
|
rv = self.app.get('/sandbox/explain_whitelist/', |
81
|
|
|
data={'whitelist': 'BasicRule wl:1000 "mz:$ARGS_VAR:foo|$URL:/bar";'}) |
82
|
|
|
self.assertEqual(rv.status_code, 200) |
83
|
|
|
self.assertIn('Whitelist all rules if matching in $ARGS_VAR:foo|$URL:/bar.', str(rv.data)) |
84
|
|
|
|
85
|
|
|
rv = self.app.get('/sandbox/explain_whitelist/', |
86
|
|
|
data={'whitelist': 'BasicRule wl:1000 "lol:pouet";'}) |
87
|
|
|
self.assertEqual(rv.status_code, 200) |
88
|
|
|
self.assertIn('Unknown fragment:', str(rv.data)) |
89
|
|
|
|
This check looks for calls to members that are non-existent. These calls will fail.
The member could have been renamed or removed.