diff options
-rw-r--r-- | mitmproxy/net/check.py | 9 | ||||
-rw-r--r-- | test/mitmproxy/net/test_check.py | 1 |
2 files changed, 9 insertions, 1 deletions
diff --git a/mitmproxy/net/check.py b/mitmproxy/net/check.py index f793d397..ac2b6830 100644 --- a/mitmproxy/net/check.py +++ b/mitmproxy/net/check.py @@ -1,4 +1,5 @@ import re +from ipaddress import ip_address # Allow underscore in host name _label_valid = re.compile(b"(?!-)[A-Z\d\-_]{1,63}(?<!-)$", re.IGNORECASE) @@ -16,7 +17,13 @@ def is_valid_host(host: bytes) -> bool: return False if host and host[-1:] == b".": host = host[:-1] - return all(_label_valid.match(x) for x in host.split(b".")) + if all(_label_valid.match(x) for x in host.split(b".")): + return True + try: + ip_address(host.decode('idna')) + return True + except ValueError: + return False def is_valid_port(port): diff --git a/test/mitmproxy/net/test_check.py b/test/mitmproxy/net/test_check.py index 9dbc02e0..0ffd6b2e 100644 --- a/test/mitmproxy/net/test_check.py +++ b/test/mitmproxy/net/test_check.py @@ -11,3 +11,4 @@ def test_is_valid_host(): assert check.is_valid_host(b"one.two.") # Allow underscore assert check.is_valid_host(b"one_two") + assert check.is_valid_host(b"::1") |