File: set-cookies-samesite_wsh.py

package info (click to toggle)
firefox 144.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 4,637,504 kB
  • sloc: cpp: 7,576,692; javascript: 6,430,831; ansic: 3,748,119; python: 1,398,978; xml: 628,810; asm: 438,679; java: 186,194; sh: 63,212; makefile: 19,159; objc: 13,086; perl: 12,986; yacc: 4,583; cs: 3,846; pascal: 3,448; lex: 1,720; ruby: 1,003; exp: 762; php: 436; lisp: 258; awk: 247; sql: 66; sed: 53; csh: 10
file content (25 lines) | stat: -rw-r--r-- 891 bytes parent folder | download | duplicates (26) 
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()