Skip to content

Commit c1d27a1

Browse files
committed
Add buffer overflow test for C version
This follows #6.
1 parent 15a37a0 commit c1d27a1

File tree

1 file changed

+18
-3
lines changed

1 file changed

+18
-3
lines changed

autotest/client.py

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -107,14 +107,29 @@ def test_Scenario5(self):
107107

108108
def test_Scenario6(self):
109109
cli = Client()
110+
uri = "/" \
111+
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" \
112+
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" \
113+
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" \
114+
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
110115
for i in range(10):
111116
self.assertEqual(cli.test_Connect(server), 1)
112-
res = cli.test_PostRequest("/index.html", None)
117+
res = cli.test_GetRequest(uri)
113118
self.assertIsNotNone(res)
114119
self.assertEqual(res.status, 404)
120+
self.assertEqual(res.read(22), b"")
115121
self.assertEqual(cli.test_Close(), 1)
116122

117123
def test_Scenario7(self):
124+
cli = Client()
125+
for i in range(10):
126+
self.assertEqual(cli.test_Connect(server), 1)
127+
res = cli.test_PostRequest("/index.html", None)
128+
self.assertIsNotNone(res)
129+
self.assertEqual(res.status, 404)
130+
self.assertEqual(cli.test_Close(), 1)
131+
132+
def test_Scenario8(self):
118133
cli = Client()
119134
for i in range(40):
120135
self.assertEqual(cli.test_Connect(server), 1)
@@ -125,7 +140,7 @@ def test_Scenario7(self):
125140
self.assertEqual(res.read(len(str.encode(pattern))), str.encode(pattern))
126141
self.assertEqual(cli.test_Close(), 1)
127142

128-
def test_Scenario8(self):
143+
def test_Scenario9(self):
129144
cli = Client()
130145
for i in range(40):
131146
self.assertEqual(cli.test_Connect(server), 1)
@@ -136,7 +151,7 @@ def test_Scenario8(self):
136151
self.assertEqual(res.read(len(str.encode(pattern))), str.encode(pattern))
137152
self.assertEqual(cli.test_Close(), 1)
138153

139-
def test_Scenario9(self):
154+
def test_Scenario10(self):
140155
cli = Client()
141156
for i in range(10):
142157
self.assertEqual(cli.test_Connect(server), 1)

0 commit comments

Comments
 (0)