Copy domain and selector immediately after validating sig fields.

This commit is contained in:
Stuart D. Gathman
2012-08-21 13:07:24 -04:00
parent 02551e35b2
commit fcaee3084a
+2 -2
View File
@@ -493,6 +493,8 @@ class DKIM(object):
logger.debug("sig: %r" % sig) logger.debug("sig: %r" % sig)
validate_signature_fields(sig) validate_signature_fields(sig)
self.domain = sig[b'd']
self.selector = sig[b's']
try: try:
canon_policy = CanonicalizationPolicy.from_c_value(sig.get(b'c')) canon_policy = CanonicalizationPolicy.from_c_value(sig.get(b'c'))
@@ -506,8 +508,6 @@ class DKIM(object):
except KeyError as e: except KeyError as e:
raise MessageFormatError("unknown signature algorithm: %s" % e.args[0]) raise MessageFormatError("unknown signature algorithm: %s" % e.args[0])
self.domain = sig[b'd']
self.selector = sig[b's']
if b'l' in sig: if b'l' in sig:
body = body[:int(sig[b'l'])] body = body[:int(sig[b'l'])]