From 9870844b38c84e7446b15909758497cecb26301e Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Thu, 20 Oct 2016 10:46:47 +1300
Subject: netlib.utils -> netlib.check

Now only contains is_valid_[host,port]

Intermediate step - this will be in mitproxy.net soon.
---
 netlib/check.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 netlib/check.py

(limited to 'netlib/check.py')

diff --git a/netlib/check.py b/netlib/check.py
new file mode 100644
index 00000000..7b007cb5
--- /dev/null
+++ b/netlib/check.py
@@ -0,0 +1,22 @@
+import re
+
+_label_valid = re.compile(b"(?!-)[A-Z\d-]{1,63}(?<!-)$", re.IGNORECASE)
+
+
+def is_valid_host(host: bytes) -> bool:
+    """
+        Checks if a hostname is valid.
+    """
+    try:
+        host.decode("idna")
+    except ValueError:
+        return False
+    if len(host) > 255:
+        return False
+    if host and host[-1:] == b".":
+        host = host[:-1]
+    return all(_label_valid.match(x) for x in host.split(b"."))
+
+
+def is_valid_port(port):
+    return 0 <= port <= 65535
-- 
cgit v1.2.3