Use Referrer-Policy headers for CSS stylesheets

third_party/WebKit/LayoutTests/external/wpt/referrer-policy/generic/subresource/image.py

 import os, sys, array, json, math, StringIO
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 import subresource
 
 class Image:
     """This class partially implements the interface of the PIL.Image.Image.
        One day in the future WPT might support the PIL module or another imaging
        library, so this hacky BMP implementation will no longer be required.
     """
     def __init__(self, width, height):
@@ skipping 70 matching lines @@
     img = Image.new("RGB", (sqrt, sqrt), "black")
     img.putdata(color_data)
 
     # Flush image to string.
     f = StringIO.StringIO()
     img.save(f, "BMP")
     f.seek(0)
 
     return f.read()
 
-def generate_payload(server_data):
+def generate_payload(request, server_data):
     data = ('{"headers": %(headers)s}') % server_data
-    return encode_string_as_bmp_image(data)
+    if "id" in request.GET:
+        request.server.stash.put(request.GET["id"], data)
+    data = encode_string_as_bmp_image(data)
+    return data
+
+def generate_report_headers_payload(request, server_data):
+    stashed_data = request.server.stash.take(request.GET["id"])
+    return stashed_data
 
 def main(request, response):
+    handler = lambda data: generate_payload(request, data)
+    content_type = 'image/bmp'
+
+    if "report-headers" in request.GET:
+        handler = lambda data: generate_report_headers_payload(request, data)
+        content_type = 'application/json'
+
     subresource.respond(request,
                         response,
-                        payload_generator = generate_payload,
-                        content_type = "image/bmp",
+                        payload_generator = handler,
+                        content_type = content_type,
                         access_control_allow_origin = "*")