Index: net/tools/testserver/testserver.py |
diff --git a/net/tools/testserver/testserver.py b/net/tools/testserver/testserver.py |
index f9177897cc6e17b64a3fa075498365b96f0a7b81..40ac17d48b1540b94c8cefdfd87b74d95948946e 100755 |
--- a/net/tools/testserver/testserver.py |
+++ b/net/tools/testserver/testserver.py |
@@ -909,7 +909,7 @@ class TestPageHandler(BasePageHandler): |
if not self.path.startswith(prefix): |
return False |
# Consume a request body if present. |
- if self.command == 'POST' or self.command == 'PUT' : |
+ if self.command == 'POST' or self.command == 'PUT': |
self.ReadRequestBody() |
return self._FileHandlerHelper(prefix) |
@@ -918,7 +918,15 @@ class TestPageHandler(BasePageHandler): |
prefix = urlparse.urljoin(self.server.file_root_url, 'post/') |
if not self.path.startswith(prefix): |
return False |
- self.ReadRequestBody() |
+ body = self.ReadRequestBody() |
+ _, _, _, _, query_str, _ = urlparse.urlparse(self.path) |
+ query_dict = cgi.parse_qs(query_str) |
+ expected_body = query_dict.get('expected_body', []) |
+ if body not in expected_body: |
+ self.send_response(404) |
+ self.end_headers() |
+ self.wfile.write('') |
+ return True |
return self._FileHandlerHelper(prefix) |
def _FileHandlerHelper(self, prefix): |