1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
from cookies.resources.helpers import makeCookieHeader, readCookies, setNoCacheAndCORSHeaders
# Step 4/6 (workers/same-site-cookies/{})
def main(request, response):
headers = setNoCacheAndCORSHeaders(request, response)
headers[0] = (b"Content-Type", b"text/javascript")
headers.append(makeCookieHeader(b"samesite_strict_set_on_load", b"test", {b"SameSite":b"Strict", b"path":b"/", b"Secure":b""}))
headers.append(makeCookieHeader(b"samesite_lax_set_on_load", b"test", {b"SameSite":b"Lax", b"path":b"/", b"Secure":b""}))
headers.append(makeCookieHeader(b"samesite_none_set_on_load", b"test", {b"SameSite":b"None", b"path":b"/", b"Secure":b""}))
cookies = readCookies(request)
message = b"ReadOnLoad:"
if b"samesite_strict_set_before_load" in cookies:
message += b"Strict"
if b"samesite_lax_set_before_load" in cookies:
message += b"Lax"
if b"samesite_none_set_before_load" in cookies:
message += b"None"
document = b"""
self.onconnect = (e) => {
fetch("/workers/same-site-cookies/resources/get_cookies_redirect.py", {credentials: 'include'}).then((resp) => {
resp.json().then((cookies) => {
let message = \"""" + message + b""",ReadOnFetch:";
if (cookies.hasOwnProperty("samesite_strict_set_before_load")) {
message += "Strict";
}
if (cookies.hasOwnProperty("samesite_lax_set_before_load")) {
message += "Lax";
}
if (cookies.hasOwnProperty("samesite_none_set_before_load")) {
message += "None";
}
e.ports[0].postMessage(message);
self.close();
});
});
}
"""
return headers, document
|
