diff options
| author | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2020-04-10 14:08:40 +0200 |
|---|---|---|
| committer | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2020-04-11 11:12:18 +0200 |
| commit | 5ea75a1a8176f764429fb45875626b1b7ef6644c (patch) | |
| tree | 5e94610f249d4bc4633328ec005412ec74fd8868 /examples/complex/sslstrip.py | |
| parent | 12b37c39e19df082e48825cfcf59fc067bffb4d0 (diff) | |
| download | mitmproxy-5ea75a1a8176f764429fb45875626b1b7ef6644c.tar.gz mitmproxy-5ea75a1a8176f764429fb45875626b1b7ef6644c.tar.bz2 mitmproxy-5ea75a1a8176f764429fb45875626b1b7ef6644c.zip | |
examples: reformat to prevent too long lines
Diffstat (limited to 'examples/complex/sslstrip.py')
| -rw-r--r-- | examples/complex/sslstrip.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/examples/complex/sslstrip.py b/examples/complex/sslstrip.py index 8b904216..16d9b59a 100644 --- a/examples/complex/sslstrip.py +++ b/examples/complex/sslstrip.py @@ -51,9 +51,11 @@ def response(flow: http.HTTPFlow) -> None: flow.response.headers['Location'] = location.replace('https://', 'http://', 1) # strip upgrade-insecure-requests in Content-Security-Policy header - if re.search('upgrade-insecure-requests', flow.response.headers.get('Content-Security-Policy', ''), flags=re.IGNORECASE): + csp_header = flow.response.headers.get('Content-Security-Policy', '') + if re.search('upgrade-insecure-requests', csp_header, flags=re.IGNORECASE): csp = flow.response.headers['Content-Security-Policy'] - flow.response.headers['Content-Security-Policy'] = re.sub(r'upgrade-insecure-requests[;\s]*', '', csp, flags=re.IGNORECASE) + new_header = re.sub(r'upgrade-insecure-requests[;\s]*', '', csp, flags=re.IGNORECASE) + flow.response.headers['Content-Security-Policy'] = new_header # strip secure flag from 'Set-Cookie' headers cookies = flow.response.headers.get_all('Set-Cookie') |