From 7b37e2cb8dde7b787823fcecd5608ace797e2793 Mon Sep 17 00:00:00 2001 From: Scott Kitterman Date: Sun, 19 Apr 2020 02:28:22 -0400 Subject: [PATCH] Fix dnsfunc definition for test suite for compatibility with dkimpy >= 1.0 --- dkimpy_milter/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/dkimpy_milter/__init__.py b/dkimpy_milter/__init__.py index b912660..5118ae8 100644 --- a/dkimpy_milter/__init__.py +++ b/dkimpy_milter/__init__.py @@ -363,8 +363,12 @@ class dkimMilter(Milter.Base): try: dnsoverride = self.conf.get('DNSOverride') if isinstance(dnsoverride, str): + timeout = 5 + domain = self.fdomain + def dnsfunc(domain, timeout=timeout, dnsoverride=dnsoverride): + return dnsoverride syslog.syslog("DNSOverride: {0}".format(dnsoverride)) - res = d.verify(idx=y, dnsfunc=lambda _x: dnsoverride) + res = d.verify(idx=y, dnsfunc=dnsfunc) else: res = d.verify(idx=y) algo = codecs.decode(d.signature_fields.get(b'a'), 'ascii')