Only check the first 4 characters.

This commit is contained in:
Zackery Spytz 2020-01-15 19:20:53 -07:00
parent 8ba363238e
commit fcb8ae9b86
1 changed files with 2 additions and 2 deletions

View File

@ -88,7 +88,7 @@ def ToASCII(label):
raise UnicodeError("label empty or too long") raise UnicodeError("label empty or too long")
# Step 5: Check ACE prefix # Step 5: Check ACE prefix
if label.lower().startswith(sace_prefix): if label[:4].lower() == sace_prefix:
raise UnicodeError("Label starts with ACE prefix") raise UnicodeError("Label starts with ACE prefix")
# Step 6: Encode with PUNYCODE # Step 6: Encode with PUNYCODE
@ -121,7 +121,7 @@ def ToUnicode(label):
except UnicodeError: except UnicodeError:
raise UnicodeError("Invalid character in IDN label") raise UnicodeError("Invalid character in IDN label")
# Step 3: Check for ACE prefix # Step 3: Check for ACE prefix
if not label.lower().startswith(ace_prefix): if not label[:4].lower() == ace_prefix:
return str(label, "ascii") return str(label, "ascii")
# Step 4: Remove ACE prefix # Step 4: Remove ACE prefix