Source code
Revision control
Copy as Markdown
Other Tools
# -*- coding: utf-8 -*-
# Active wptserve handler for cookie operations.
#
# This must support the following requests:
#
# - GET with the following query parameters:
# - charset: (optional) character set for response (default: utf-8)
# A cookie: request header (if present) is echoed in the body with a
# cookie= prefix followed by the urlencoded bytes from the header.
# Used to inspect the cookie jar from an HTTP request header context.
# - POST with form-data in the body and the following query-or-form parameters:
# - set-cookie: (optional; repeated) echoed in the set-cookie: response
# header and also echoed in the body with a set-cookie= prefix
# followed by the urlencoded bytes from the parameter; multiple occurrences
# are CRLF-delimited.
# Used to set cookies from an HTTP response header context.
#
# The response has 200 status and content-type: text/plain; charset=<charset>
import encodings, re
from urllib.parse import parse_qs, quote
from wptserve.utils import isomorphic_decode, isomorphic_encode
# NOTE: These are intentionally very lax to permit testing
DISALLOWED_IN_COOKIE_NAME_RE = re.compile(br'[;\0-\x1f\x7f]')
DISALLOWED_IN_HEADER_RE = re.compile(br'[\0-\x1f\x7f]')
# Ensure common charset names do not end up with different
# capitalization or punctuation
CHARSET_OVERRIDES = {
encodings.codecs.lookup(charset).name: charset
for charset in (u'utf-8', u'iso-8859-1',)
}
def quote_str(cookie_str):
return isomorphic_encode(quote(isomorphic_decode(cookie_str), u'', encoding=u'iso-8859-1'))
def parse_qs_str(query_str):
args = parse_qs(isomorphic_decode(query_str), keep_blank_values=True, encoding=u'iso-8859-1')
binary_args = {}
for key, val in args.items():
binary_args[isomorphic_encode(key)] = [isomorphic_encode(x) for x in val]
return binary_args
def main(request, response):
assert request.method in (
u'GET',
u'POST',
), u'request method was neither GET nor POST: %r' % request.method
qd = (isomorphic_encode(request.url).split(b'#')[0].split(b'?', 1) + [b''])[1]
if request.method == u'POST':
qd += b'&' + request.body
args = parse_qs_str(qd)
charset = encodings.codecs.lookup([isomorphic_decode(x) for x in args.get(b'charset', [u'utf-8'])][-1]).name
charset = CHARSET_OVERRIDES.get(charset, charset)
headers = [(b'content-type', b'text/plain; charset=' + isomorphic_encode(charset))]
body = []
if request.method == u'POST':
for set_cookie in args.get(b'set-cookie', []):
if b'=' in set_cookie.split(b';', 1)[0]:
name, rest = set_cookie.split(b'=', 1)
assert re.search(
DISALLOWED_IN_COOKIE_NAME_RE,
name
) is None, b'name had disallowed characters: %r' % name
else:
rest = set_cookie
assert re.search(
DISALLOWED_IN_HEADER_RE,
rest
) is None, b'rest had disallowed characters: %r' % rest
headers.append((b'set-cookie', set_cookie))
body.append(b'set-cookie=' + quote_str(set_cookie))
else:
cookie = request.headers.get(b'cookie')
if cookie is not None:
body.append(b'cookie=' + quote_str(cookie))
body = b'\r\n'.join(body)
headers.append((b'content-length', len(body)))
return 200, headers, body