HTTPTransport: callers should be able to obtain the HTTP response body

util/net/http_transport_test_server.py

 #!/usr/bin/env python
 # coding: utf-8
 
 # Copyright 2014 The Crashpad Authors. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
 """A one-shot testing webserver.
 
-When invoked, this server will write an integer to stdout, indiciating on which
-port the server is listening. It will then read one integer from stdin,
-indiciating the response code to set for a request. The server will process
-one HTTP request, writing it out entirely to stdout, and will then terminate.
+When invoked, this server will write a short integer to stdout, indiciating on
+which port the server is listening. It will then read one integer from stdin,
+indiciating the response code to be sent in response to a request. It also reads
+8 characters from stdin, which, after having "\r\n" appended, will form the
+response body in a successful response (one with code 200). The server will
+process one HTTP request, deliver the prearranged response to the client, and
+write the entire request to stdout. It will then terminate.
 
 This server is written in Python since it provides a simple HTTP stack, and
 because parsing Chunked encoding is safer and easier in a memory-safe language.
 This could easily have been written in C++ instead.
 """
 
 import BaseHTTPServer
 import struct
 import sys
 
@@ skipping 16 matching lines @@
 
   def flush(self):
     self.file.flush()
 
   def close(self):
     self.file.close()
 
 
 class RequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
   response_code = 500
+  response_body = ''
 
   def handle_one_request(self):
     # Wrap the rfile in the buffering file object so that the raw header block
     # can be written to stdout after it is parsed.
     self.rfile = BufferedReadFile(self.rfile)
     BaseHTTPServer.BaseHTTPRequestHandler.handle_one_request(self)
 
   def do_POST(self):
     writer = sys.stdout
 
     writer.write(self.rfile.buffer)
     self.rfile.buffer = ''
 
     if self.headers.get('Transfer-Encoding', '') == 'Chunked':
       body = self.handle_chunked_encoding()
     else:
       length = int(self.headers.get('Content-Length', -1))
       body = self.rfile.read(length)
 
     self.send_response(self.response_code)
     self.end_headers()
+    if self.response_code == 200:
+      self.wfile.write(self.response_body)
+      self.wfile.write('\r\n')
 
     writer.write(body)
     writer.flush()
 
   def handle_chunked_encoding(self):
     """This parses a "Transfer-Encoding: Chunked" body in accordance with
     RFC 7230 §4.1. This returns the result as a string.
     """
     body = ''
     chunk_size = self.read_chunk_size()
@@ skipping 33 matching lines @@
     import os, msvcrt
     msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
 
   # Start the server.
   server = BaseHTTPServer.HTTPServer(('127.0.0.1', 0), RequestHandler)
 
   # Write the port as an unsigned short to the parent process.
   sys.stdout.write(struct.pack('=H', server.server_address[1]))
   sys.stdout.flush()
 
-  # Read the desired test response code as an unsigned short from the parent
-  # process.
-  RequestHandler.response_code = \
-      struct.unpack('=H', sys.stdin.read(struct.calcsize('=H')))[0]
+  # Read the desired test response code as an unsigned short and the desired
+  # response body as an 8-byte string from the parent process.
+  RequestHandler.response_code, RequestHandler.response_body = \
+      struct.unpack('=H8s', sys.stdin.read(struct.calcsize('=H8s')))
 
   # Handle the request.
   server.handle_request()
 
 if __name__ == '__main__':
   Main()