Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Require content-length header

Browse Source 
fixes #908
This commit is contained in:
Jack Danger Canty 2017-05-01 12:13:12 -07:00
parent df84675ae6
commit a2fd72d2f8
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
moto/s3/responses.py
View File

@ -294,6 +294,8 @@ class ResponseObject(_TemplateEnvironmentMixin):
) )
def _bucket_response_put(self, request, body, region_name, bucket_name, querystring, headers): def _bucket_response_put(self, request, body, region_name, bucket_name, querystring, headers):
if not request.headers.get('Content-Length'):
return 411, {}, "Content-Length required"
if 'versioning' in querystring: if 'versioning' in querystring:
ver = re.search('<Status>([A-Za-z]+)</Status>', body) ver = re.search('<Status>([A-Za-z]+)</Status>', body)
if ver: if ver:
@ -355,6 +357,8 @@ class ResponseObject(_TemplateEnvironmentMixin):
return 409, {}, template.render(bucket=removed_bucket) return 409, {}, template.render(bucket=removed_bucket)
def _bucket_response_post(self, request, body, bucket_name, headers): def _bucket_response_post(self, request, body, bucket_name, headers):
if not request.headers.get('Content-Length'):
return 411, {}, "Content-Length required"
path = request.path if hasattr(request, 'path') else request.path_url path = request.path if hasattr(request, 'path') else request.path_url
if self.is_delete_keys(request, path, bucket_name): if self.is_delete_keys(request, path, bucket_name):
return self._bucket_response_delete_keys(request, body, bucket_name, headers) return self._bucket_response_delete_keys(request, body, bucket_name, headers)

11
tests/test_s3/test_server.py
View File

@ -66,3 +66,14 @@ def test_s3_server_post_to_bucket():
res = test_client.get('/the-key', 'http://tester.localhost:5000/') res = test_client.get('/the-key', 'http://tester.localhost:5000/')
res.status_code.should.equal(200) res.status_code.should.equal(200)
res.data.should.equal(b"nothing") res.data.should.equal(b"nothing")
def test_s3_server_post_without_content_length():
backend = server.create_backend_app("s3")
test_client = backend.test_client()
res = test_client.put('/', 'http://tester.localhost:5000/', environ_overrides={'CONTENT_LENGTH': ''})
res.status_code.should.equal(411)
res = test_client.post('/', "https://tester.localhost:5000/", environ_overrides={'CONTENT_LENGTH': ''})
res.status_code.should.equal(411)