forked from xmendez/wfuzz
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_req_parse.py
160 lines (125 loc) · 5.01 KB
/
test_req_parse.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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
import unittest
from wfuzz.fuzzobjects import FuzzRequest
http_post_request = '''POST /slipstream/view HTTP/1.1
Host: www
User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10.14; rv:70.0) Gecko/20100101 Firefox/70.0
Accept: */*
Accept-Language: en-GB,en;q=0.5
Accept-Encoding: gzip, deflate
Referer: https://www
Content-Type: text/plain;charset=UTF-8
Origin: https://www
Content-Length: 3387
Connection: close
a=1'''
http_get_request = '''GET /sttc/bpk-fonts/55b577a1.woff2 HTTP/1.1
Host: js.skyscnr.com
User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10.14; rv:70.0) Gecko/20100101 Firefox/70.0
Accept: application/font-woff2;q=1.0,application/font-woff;q=0.9,*/*;q=0.8
Accept-Language: en-GB,en;q=0.5
Accept-Encoding: gzip, deflate
Origin: https://www.skyscanner.es
Connection: close
Referer: https://js.skyscnr.com/sttc/oc-registry/components/base-stylesheet/0.1.33/build//static/css/main.e09b44e2.css
'''
http_response = '''HTTP/1.1 201 Created
Content-Type: application/json
Content-Length: 51
Cache-Control: no-store, no-cache, must-revalidate, post-check=0, pre-check=0, max-age=0
Expires: -1
Last-Modified: Mon, 30 Dec 2019 13:16:57 GMT
Pragma: no-cache
Server: Unspecified
Date: Mon, 30 Dec 2019 13:16:57 GMT
Connection: close
LINE_1'''
http_response_no_content = '''HTTP/1.1 201 Created
Content-Type: application/json
Content-Length: 51
Cache-Control: no-store, no-cache, must-revalidate, post-check=0, pre-check=0, max-age=0
Expires: -1
Last-Modified: Mon, 30 Dec 2019 13:16:57 GMT
Pragma: no-cache
Server: Unspecified
Date: Mon, 30 Dec 2019 13:16:57 GMT
Connection: close
'''
http_multi_request = '''POST /tr/ HTTP/1.1
Host: www.facebook.com
User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10.14; rv:70.0) Gecko/20100101 Firefox/70.0
Accept: */*
Accept-Language: en-GB,en;q=0.5
Accept-Encoding: gzip, deflate
Content-Type: multipart/form-data; boundary=---------------------------18698393981150719881279620016
Content-Length: 3320
Origin: https://www.skyscanner.es
Connection: close
Referer: https://www.skyscanner.es/
-----------------------------18698393981150719881279620016
Content-Disposition: form-data; name="id"
561358470665569
-----------------------------18698393981150719881279620016
Content-Disposition: form-data; name="rqm"
SB
-----------------------------18698393981150719881279620016--
'''
http_follow_response = '''HTTP/1.1 301 Moved Permanently
Location: http://www.google.com/
Content-Type: text/html; charset=UTF-8
Date: Mon, 30 Dec 2019 20:26:23 GMT
Expires: Wed, 29 Jan 2020 20:26:23 GMT
Cache-Control: public, max-age=2592000
Server: gws
Content-Length: 219
X-XSS-Protection: 0
X-Frame-Options: SAMEORIGIN
HTTP/1.1 200 OK
Date: Mon, 30 Dec 2019 20:26:23 GMT
Expires: -1
Cache-Control: private, max-age=0
Content-Type: text/html; charset=ISO-8859-1
P3P: CP="This is not a P3P policy! See g.co/p3phelp for more info."
Server: gws
X-XSS-Protection: 0
X-Frame-Options: SAMEORIGIN
Set-Cookie: 1P_JAR=2019-12-30-20; expires=Wed, 29-Jan-2020 20:26:23 GMT; path=/; domain=.google.com
Set-Cookie: NID=194=Tygb5SRWSRvznMKZn4Dnl0SIkI9zcjk_U9OnBb9RlhyXWKlvEJSCorghYsp5IPR-bAm31XZlKGiL0RjLxjGigGqkGguTVmJ1C4Ae6JUKLoAYLbR-C8xAvuwoXm6Nw61Wep9U1zkq6evNZ-WbKyfYvOS6WrUi_3TXU7BqUaWZsJY; expires=Tue, 30-Jun-2020 20:26:23 GMT; path=/; domain=.google.com; HttpOnly
Accept-Ranges: none
Vary: Accept-Encoding
Transfer-Encoding: chunked
LINE_1'''
class ParseRequestTest(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(ParseRequestTest, self).__init__(*args, **kwargs)
self.maxDiff = 1000
def test_2_ways_of_parsing_content(self):
fr = FuzzRequest()
fr.update_from_raw_http(http_multi_request, "https", http_response)
fr2 = FuzzRequest()
fr2.update_from_raw_http(http_multi_request, "https", http_response_no_content, b"LINE_1")
# raw content takes precedence
fr3 = FuzzRequest()
fr3.update_from_raw_http(http_multi_request, "https", http_response, b"LINE_0")
self.assertEqual(fr.content, fr2.content)
self.assertEqual(fr3.content, "LINE_0")
def test_parse_multi_raw_request(self):
fr = FuzzRequest()
fr.update_from_raw_http(http_multi_request, "https", http_response)
self.assertEqual(fr.params.post.id, "561358470665569")
self.assertEqual(fr.params.post.rqm, "SB")
self.assertEqual(fr.content, "LINE_1")
def test_parse_raw_multi_response(self):
fr = FuzzRequest()
fr.update_from_raw_http(http_multi_request, "https", http_follow_response)
self.assertEqual(fr.content, "LINE_1")
self.assertEqual(fr.code, 200)
def test_parse_get_crlf_request(self):
fr = FuzzRequest()
fr.update_from_raw_http(http_get_request, "https", "\n\n\n")
self.assertEqual(fr.method, "GET")
self.assertEqual(fr.params.raw_post, None)
def test_parse_crlf_post_request(self):
fr = FuzzRequest()
fr.update_from_raw_http(http_post_request, "https", "\n\n\n")
self.assertEqual(fr.method, "POST")
self.assertEqual(fr.params.post, {'a': '1'})