diff --git a/mitmproxy/net/check.py b/mitmproxy/net/check.py index f793d3971..ac2b6830d 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}(? 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 9dbc02e02..0ffd6b2e7 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")