Make encodings all ascii/utf-8 as appropriate.
This commit is contained in:
+2
-2
@@ -430,8 +430,8 @@ def verify(message, logger=None, dnsfunc=dnstxt):
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
# dnstxt wants Unicode
|
# dnstxt wants Unicode
|
||||||
selector = sig[b's'].decode('latin-1')
|
selector = sig[b's'].decode('ascii')
|
||||||
domain = sig[b'd'].decode('latin-1')
|
domain = sig[b'd'].decode('ascii')
|
||||||
name = "%s._domainkey.%s." % (selector, domain)
|
name = "%s._domainkey.%s." % (selector, domain)
|
||||||
s = dnsfunc(name).encode('utf-8')
|
s = dnsfunc(name).encode('utf-8')
|
||||||
if not s:
|
if not s:
|
||||||
|
|||||||
@@ -53,7 +53,7 @@ class TestSignAndVerify(unittest.TestCase):
|
|||||||
|
|
||||||
def dnsfunc(self, domain):
|
def dnsfunc(self, domain):
|
||||||
self.assertEqual('test._domainkey.example.com.', domain)
|
self.assertEqual('test._domainkey.example.com.', domain)
|
||||||
return read_test_data("test.txt").decode('latin1')
|
return read_test_data("test.txt").decode('utf-8')
|
||||||
|
|
||||||
def test_verifies(self):
|
def test_verifies(self):
|
||||||
# A message verifies after being signed.
|
# A message verifies after being signed.
|
||||||
|
|||||||
+3
-3
@@ -36,11 +36,11 @@ if sys.version_info[0] >= 3:
|
|||||||
sys.stdin = sys.stdin.detach()
|
sys.stdin = sys.stdin.detach()
|
||||||
sys.stdout = sys.stdout.detach()
|
sys.stdout = sys.stdout.detach()
|
||||||
|
|
||||||
selector = sys.argv[1].encode('utf-8')
|
selector = sys.argv[1].encode('ascii')
|
||||||
domain = sys.argv[2].encode('utf-8')
|
domain = sys.argv[2].encode('ascii')
|
||||||
privatekeyfile = sys.argv[3]
|
privatekeyfile = sys.argv[3]
|
||||||
if len(sys.argv) > 5:
|
if len(sys.argv) > 5:
|
||||||
identity = sys.argv[4].encode('utf-8')
|
identity = sys.argv[4].encode('ascii')
|
||||||
else:
|
else:
|
||||||
identity = None
|
identity = None
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user