blob: 59f0a4a598ebaf67c6583e196fba2c40b89dc01b (
plain)
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
|
import urllib
def web_socket_do_extra_handshake(request):
url_parts = urllib.parse.urlsplit(request.uri)
max_age = ""
if "clear" in url_parts.query:
max_age = "; Max-Age=0"
value = "1"
if "value" in url_parts.query:
value = urllib.parse.parse_qs(url_parts.query)["value"][0]
cookies = [
"samesite-unspecified={}; Path=/".format(value) + max_age,
"samesite-lax={}; Path=/; SameSite=Lax".format(value) + max_age,
"samesite-strict={}; Path=/; SameSite=Strict".format(value) + max_age,
# SameSite=None cookies must be Secure.
"samesite-none={}; Path=/; SameSite=None; Secure".format(value) + max_age
]
for cookie in cookies:
request.extra_headers.append(("Set-Cookie", cookie))
def web_socket_transfer_data(request):
# Expect close() from user agent.
request.ws_stream.receive_message()
|
