mirror of
https://github.com/jlu5/PyLink.git
synced 2024-11-01 01:09:22 +01:00
Rewrite is_server_name() to fail on hostnames with - and _
This commit is contained in:
parent
462fa91622
commit
083dc6a58f
14
classes.py
14
classes.py
@ -774,16 +774,16 @@ class PyLinkNetworkCoreWithUtils(PyLinkNetworkCore):
|
|||||||
"""
|
"""
|
||||||
return str(obj).startswith('#')
|
return str(obj).startswith('#')
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _isASCII(s):
|
|
||||||
"""Returns whether the given string only contains non-whitespace ASCII characters."""
|
|
||||||
chars = string.ascii_letters + string.digits + string.punctuation
|
|
||||||
return all(char in chars for char in str(s))
|
|
||||||
|
|
||||||
|
# Modified from https://stackoverflow.com/a/106223 (RFC 1123):
|
||||||
|
# - Allow hostnames that end in '.'
|
||||||
|
# - Require at least one '.' in the hostname
|
||||||
|
_HOSTNAME_RE = re.compile(r'^(([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-]*[a-zA-Z0-9])\.)+'
|
||||||
|
r'([A-Za-z0-9]|[A-Za-z0-9][A-Za-z0-9\-]*[A-Za-z0-9])*$')
|
||||||
@classmethod
|
@classmethod
|
||||||
def is_server_name(cls, s):
|
def is_server_name(cls, text):
|
||||||
"""Returns whether the string given is a valid server name."""
|
"""Returns whether the string given is a valid server name."""
|
||||||
return cls._isASCII(s) and '.' in s and not s.startswith('.')
|
return bool(cls._HOSTNAME_RE.match(text))
|
||||||
|
|
||||||
_HOSTMASK_RE = re.compile(r'^\S+!\S+@\S+$')
|
_HOSTMASK_RE = re.compile(r'^\S+!\S+@\S+$')
|
||||||
@classmethod
|
@classmethod
|
||||||
|
@ -178,15 +178,6 @@ class BaseProtocolTest(unittest.TestCase):
|
|||||||
assertF('&channel') # we don't support these yet
|
assertF('&channel') # we don't support these yet
|
||||||
assertF('lorem ipsum')
|
assertF('lorem ipsum')
|
||||||
|
|
||||||
def test_is_ascii(self):
|
|
||||||
assertT = lambda inp: self.assertTrue(self.p._isASCII(inp))
|
|
||||||
assertF = lambda inp: self.assertFalse(self.p._isASCII(inp))
|
|
||||||
|
|
||||||
assertT('iotgjw@sy9!4py645ujg990rEYghiwaK0r4{SEFIre')
|
|
||||||
assertT('touche')
|
|
||||||
assertF('touché')
|
|
||||||
assertF('测试1')
|
|
||||||
|
|
||||||
def test_is_server_name(self):
|
def test_is_server_name(self):
|
||||||
self.assertTrue(self.p.is_server_name('test.local'))
|
self.assertTrue(self.p.is_server_name('test.local'))
|
||||||
self.assertTrue(self.p.is_server_name('IRC.example.com'))
|
self.assertTrue(self.p.is_server_name('IRC.example.com'))
|
||||||
|
Loading…
Reference in New Issue
Block a user