Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2025-01-30 14:58:38 +00:00
Code Issues Projects Releases Wiki Activity

Absolute IPv6 addresses supported

Browse Source
This commit is contained in:
Ujjwal Verma 2017-02-20 21:02:43 +05:30
parent 18401dda8f
commit 11134b669e
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mitmproxy/net/check.py
View File

@ -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):

1
test/mitmproxy/net/test_check.py
View File

@ -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")