Make encodings all ascii/utf-8 as appropriate.

This commit is contained in:
William Grant
2011-03-19 22:44:46 +11:00
parent b6d9e35a1d
commit d9e3b49a8e
3 changed files with 6 additions and 6 deletions
+2 -2
View File
@@ -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:
+1 -1
View File
@@ -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
View File
@@ -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