Compare commits
17 Commits
1.0.1
..
dkg/python3
| Author | SHA1 | Date | |
|---|---|---|---|
| ea09bab1a8 | |||
| 391b5352f3 | |||
| ad8f396db0 | |||
| 479820a07d | |||
| 7bfb87fab7 | |||
| 5c1d5d6e52 | |||
| ae31730593 | |||
| 72ed000ccf | |||
| b3db013754 | |||
| bd1d25d83e | |||
| 1c6030024d | |||
| 71c0c3f20a | |||
| a9a6893c89 | |||
| bb44f36519 | |||
| 9e11b75ec3 | |||
| f60ea12e86 | |||
| e872bd44b0 |
@@ -1,3 +1,4 @@
|
|||||||
dist
|
dist
|
||||||
dkimpy_milter.egg-info
|
dkimpy_milter.egg-info
|
||||||
*.pyc
|
*.pyc
|
||||||
|
*~
|
||||||
|
|||||||
+40
-34
@@ -1,4 +1,4 @@
|
|||||||
#! /usr/bin/python2
|
#! /usr/bin/python3
|
||||||
# Original dkim-milter.py code:
|
# Original dkim-milter.py code:
|
||||||
# Author: Stuart D. Gathman <stuart@bmsi.com>
|
# Author: Stuart D. Gathman <stuart@bmsi.com>
|
||||||
# Copyright 2007 Business Management Systems, Inc.
|
# Copyright 2007 Business Management Systems, Inc.
|
||||||
@@ -28,8 +28,9 @@ import dkim
|
|||||||
import authres
|
import authres
|
||||||
import os
|
import os
|
||||||
import tempfile
|
import tempfile
|
||||||
import StringIO
|
import io
|
||||||
import re
|
import re
|
||||||
|
import codecs
|
||||||
from Milter.utils import parse_addr, parseaddr
|
from Milter.utils import parse_addr, parseaddr
|
||||||
import dkimpy_milter.config as config
|
import dkimpy_milter.config as config
|
||||||
from dkimpy_milter.util import drop_privileges
|
from dkimpy_milter.util import drop_privileges
|
||||||
@@ -61,7 +62,9 @@ class dkimMilter(Milter.Base):
|
|||||||
self.external_connection = False
|
self.external_connection = False
|
||||||
self.hello_name = None
|
self.hello_name = None
|
||||||
# sometimes people put extra space in sendmail config, so we strip
|
# sometimes people put extra space in sendmail config, so we strip
|
||||||
self.receiver = self.getsymval('j').strip()
|
self.receiver = self.getsymval('j')
|
||||||
|
if self.receiver is not None:
|
||||||
|
self.receiver = self.receiver.strip()
|
||||||
try:
|
try:
|
||||||
self.AuthservID = milterconfig['AuthservID']
|
self.AuthservID = milterconfig['AuthservID']
|
||||||
except:
|
except:
|
||||||
@@ -108,7 +111,7 @@ class dkimMilter(Milter.Base):
|
|||||||
def envfrom(self, f, *str):
|
def envfrom(self, f, *str):
|
||||||
if milterconfig.get('Syslog') and milterconfig.get('debugLevel') >= 2:
|
if milterconfig.get('Syslog') and milterconfig.get('debugLevel') >= 2:
|
||||||
syslog.syslog("mail from: {0} {1}".format(f, str))
|
syslog.syslog("mail from: {0} {1}".format(f, str))
|
||||||
self.fp = StringIO.StringIO()
|
self.fp = io.BytesIO()
|
||||||
self.mailfrom = f
|
self.mailfrom = f
|
||||||
t = parse_addr(f)
|
t = parse_addr(f)
|
||||||
if len(t) == 2:
|
if len(t) == 2:
|
||||||
@@ -140,13 +143,13 @@ class dkimMilter(Milter.Base):
|
|||||||
elif lname == 'authentication-results':
|
elif lname == 'authentication-results':
|
||||||
self.arheaders.append(val)
|
self.arheaders.append(val)
|
||||||
if self.fp:
|
if self.fp:
|
||||||
self.fp.write("%s: %s\n" % (name, val))
|
self.fp.write(b"%s: %s\n" % (codecs.encode(name, 'ascii'), codecs.encode(val, 'ascii')))
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
@Milter.noreply
|
@Milter.noreply
|
||||||
def eoh(self):
|
def eoh(self):
|
||||||
if self.fp:
|
if self.fp:
|
||||||
self.fp.write("\n") # terminate headers
|
self.fp.write(b"\n") # terminate headers
|
||||||
self.bodysize = 0
|
self.bodysize = 0
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
@@ -193,20 +196,20 @@ class dkimMilter(Milter.Base):
|
|||||||
h = authres.AuthenticationResultsHeader(authserv_id=
|
h = authres.AuthenticationResultsHeader(authserv_id=
|
||||||
self.AuthservID,
|
self.AuthservID,
|
||||||
results=self.arresults)
|
results=self.arresults)
|
||||||
h = fold(str(h))
|
h = fold(codecs.encode(str(h), 'ascii'))
|
||||||
if (milterconfig.get('Syslog') and
|
if (milterconfig.get('Syslog') and
|
||||||
milterconfig.get('debugLevel') >= 2):
|
milterconfig.get('debugLevel') >= 2):
|
||||||
syslog.syslog(str(h))
|
syslog.syslog(codecs.decode(h, 'ascii'))
|
||||||
name, val = str(h).split(': ', 1)
|
name, val = codecs.decode(h, 'ascii').split(': ', 1)
|
||||||
self.addheader(name, val, 0)
|
self.addheader(name, val, 0)
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
def sign_dkim(self, txt):
|
def sign_dkim(self, txt):
|
||||||
canon = milterconfig.get('Canonicalization')
|
canon = codecs.encode(milterconfig.get('Canonicalization'), 'ascii')
|
||||||
canonicalize = []
|
canonicalize = []
|
||||||
if len(canon.split('/')) == 2:
|
if len(canon.split(b'/')) == 2:
|
||||||
canonicalize.append(canon.split('/')[0])
|
canonicalize.append(canon.split(b'/')[0])
|
||||||
canonicalize.append(canon.split('/')[1])
|
canonicalize.append(canon.split(b'/')[1])
|
||||||
else:
|
else:
|
||||||
canonicalize.append(canon)
|
canonicalize.append(canon)
|
||||||
canonicalize.append(canon)
|
canonicalize.append(canon)
|
||||||
@@ -216,11 +219,12 @@ class dkimMilter(Milter.Base):
|
|||||||
try:
|
try:
|
||||||
if privateRSA:
|
if privateRSA:
|
||||||
d = dkim.DKIM(txt)
|
d = dkim.DKIM(txt)
|
||||||
h = d.sign(milterconfig.get('Selector'), self.fdomain,
|
h = d.sign(codecs.encode(milterconfig.get('Selector'), 'ascii'), codecs.encode(self.fdomain, 'ascii'),
|
||||||
privateRSA, canonicalize=(canonicalize[0],
|
codecs.encode(privateRSA, 'ascii'),
|
||||||
canonicalize[1]))
|
canonicalize=(canonicalize[0],
|
||||||
name, val = h.split(': ', 1)
|
canonicalize[1]))
|
||||||
self.addheader(name, val.strip().replace('\r\n', '\n'), 0)
|
name, val = h.split(b': ', 1)
|
||||||
|
self.addheader(codecs.decode(name, 'ascii'), codecs.decode(val, 'ascii').strip().replace('\r\n', '\n'), 0)
|
||||||
if (milterconfig.get('Syslog') and
|
if (milterconfig.get('Syslog') and
|
||||||
(milterconfig.get('SyslogSuccess')
|
(milterconfig.get('SyslogSuccess')
|
||||||
or milterconfig.get('debugLevel') >= 1)):
|
or milterconfig.get('debugLevel') >= 1)):
|
||||||
@@ -231,12 +235,12 @@ class dkimMilter(Milter.Base):
|
|||||||
d.domain.lower()))
|
d.domain.lower()))
|
||||||
if privateEd25519:
|
if privateEd25519:
|
||||||
d = dkim.DKIM(txt)
|
d = dkim.DKIM(txt)
|
||||||
h = d.sign(milterconfig.get('SelectorEd25519'), self.fdomain,
|
h = d.sign(codecs.encode(milterconfig.get('SelectorEd25519'), 'ascii'), codecs.encode(self.fdomain, 'ascii'),
|
||||||
privateEd25519, canonicalize=(canonicalize[0],
|
privateEd25519, canonicalize=(canonicalize[0],
|
||||||
canonicalize[1]),
|
canonicalize[1]),
|
||||||
signature_algorithm='ed25519-sha256')
|
signature_algorithm=b'ed25519-sha256')
|
||||||
name, val = h.split(': ', 1)
|
name, val = h.split(b': ', 1)
|
||||||
self.addheader(name, val.strip().replace('\r\n', '\n'), 0)
|
self.addheader(codecs.decode(name, 'ascii'), codecs.decode(val, 'ascii').strip().replace('\r\n', '\n'), 0)
|
||||||
if (milterconfig.get('Syslog') and
|
if (milterconfig.get('Syslog') and
|
||||||
(milterconfig.get('SyslogSuccess')
|
(milterconfig.get('SyslogSuccess')
|
||||||
or milterconfig.get('debugLevel') >= 1)):
|
or milterconfig.get('debugLevel') >= 1)):
|
||||||
@@ -258,21 +262,23 @@ class dkimMilter(Milter.Base):
|
|||||||
for y in range(self.has_dkim): # Verify _ALL_ the signatures
|
for y in range(self.has_dkim): # Verify _ALL_ the signatures
|
||||||
d = dkim.DKIM(txt)
|
d = dkim.DKIM(txt)
|
||||||
try:
|
try:
|
||||||
res = d.verify(idx=y)
|
dnsoverride = milterconfig.get('DNSOverride')
|
||||||
|
if isinstance(dnsoverride, str):
|
||||||
|
syslog.syslog("DNSOverride: {0}".format(dnsoverride))
|
||||||
|
res = d.verify(idx=y, dnsfunc=lambda _x: dnsoverride)
|
||||||
|
else:
|
||||||
|
res = d.verify(idx=y)
|
||||||
|
algo = codecs.decode(d.signature_fields.get(b'a'), 'ascii')
|
||||||
if res:
|
if res:
|
||||||
if d.signature_fields.get(b'a') == 'ed25519-sha256':
|
if algo == 'ed25519-sha256':
|
||||||
self.dkim_comment = ('Good {0} signature'
|
self.dkim_comment = ('Good {0} signature'
|
||||||
.format(d.signature_fields
|
.format(algo))
|
||||||
.get(b'a')))
|
|
||||||
else:
|
else:
|
||||||
self.dkim_comment = ('Good {0} bit {1} signature'
|
self.dkim_comment = ('Good {0} bit {1} signature'
|
||||||
.format(d.keysize,
|
.format(d.keysize, algo))
|
||||||
d.signature_fields
|
|
||||||
.get(b'a')))
|
|
||||||
else:
|
else:
|
||||||
self.dkim_comment = ('Bad {0} bit {1} signature.'
|
self.dkim_comment = ('Bad {0} bit {1} signature.'
|
||||||
.format(d.keysize,
|
.format(d.keysize, algo))
|
||||||
d.signature_fields.get(b'a')))
|
|
||||||
except dkim.DKIMException as x:
|
except dkim.DKIMException as x:
|
||||||
self.dkim_comment = str(x)
|
self.dkim_comment = str(x)
|
||||||
if milterconfig.get('Syslog'):
|
if milterconfig.get('Syslog'):
|
||||||
@@ -281,9 +287,9 @@ class dkimMilter(Milter.Base):
|
|||||||
self.dkim_comment = str(x)
|
self.dkim_comment = str(x)
|
||||||
if milterconfig.get('Syslog'):
|
if milterconfig.get('Syslog'):
|
||||||
syslog.syslog("check_dkim: {0}".format(x))
|
syslog.syslog("check_dkim: {0}".format(x))
|
||||||
self.header_i = d.signature_fields.get(b'i')
|
self.header_i = codecs.decode(d.signature_fields.get(b'i'), 'ascii')
|
||||||
self.header_d = d.signature_fields.get(b'd')
|
self.header_d = codecs.decode(d.signature_fields.get(b'd'), 'ascii')
|
||||||
self.header_a = d.signature_fields.get(b'a')
|
self.header_a = codecs.decode(d.signature_fields.get(b'a'), 'ascii')
|
||||||
if res:
|
if res:
|
||||||
if (milterconfig.get('Syslog') and
|
if (milterconfig.get('Syslog') and
|
||||||
(milterconfig.get('SyslogSuccess') or
|
(milterconfig.get('SyslogSuccess') or
|
||||||
|
|||||||
@@ -0,0 +1,6 @@
|
|||||||
|
#!/usr/bin/python3
|
||||||
|
|
||||||
|
from dkimpy_milter import main
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
+17
-12
@@ -31,13 +31,13 @@ import stat
|
|||||||
import dkim
|
import dkim
|
||||||
import socket
|
import socket
|
||||||
import ipaddress
|
import ipaddress
|
||||||
from dnsplug import Session
|
from .dnsplug import Session
|
||||||
|
|
||||||
# default values
|
# default values
|
||||||
defaultConfigData = {
|
defaultConfigData = {
|
||||||
'Syslog': 'yes',
|
'Syslog': 'yes',
|
||||||
'SyslogFacility': 'mail',
|
'SyslogFacility': 'mail',
|
||||||
'UMask': 007,
|
'UMask': 0o07,
|
||||||
'Mode': 'sv',
|
'Mode': 'sv',
|
||||||
'Socket': 'local:/var/run/dkimpy-milter/dkimpy-milter.sock',
|
'Socket': 'local:/var/run/dkimpy-milter/dkimpy-milter.sock',
|
||||||
'PidFile': '/var/run/dkimpy-milter/dkimpy-milter.pid',
|
'PidFile': '/var/run/dkimpy-milter/dkimpy-milter.pid',
|
||||||
@@ -48,6 +48,7 @@ defaultConfigData = {
|
|||||||
'DiagnosticDirectory': '',
|
'DiagnosticDirectory': '',
|
||||||
'MacroList': '',
|
'MacroList': '',
|
||||||
'MacroListVerify': '',
|
'MacroListVerify': '',
|
||||||
|
'DNSOverride': None,
|
||||||
'debugLevel': 0 # Undocumented config item for developer use
|
'debugLevel': 0 # Undocumented config item for developer use
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -84,14 +85,14 @@ class HostsDataset(object):
|
|||||||
self.item = item[1:]
|
self.item = item[1:]
|
||||||
self.negative = True
|
self.negative = True
|
||||||
try:
|
try:
|
||||||
self.item = ipaddress.ip_address(unicode(self.item, "utf-8"))
|
self.item = ipaddress.ip_address(str(self.item, "utf-8"))
|
||||||
if isinstance(self.item, ipaddress.IPv4Address):
|
if isinstance(self.item, ipaddress.IPv4Address):
|
||||||
self.isipv4 = True
|
self.isipv4 = True
|
||||||
elif isinstance(self.item, ipaddress.IPv6Address):
|
elif isinstance(self.item, ipaddress.IPv6Address):
|
||||||
self.isipv6 = True
|
self.isipv6 = True
|
||||||
except ValueError as e:
|
except ValueError as e:
|
||||||
try:
|
try:
|
||||||
self.item = ipaddress.ip_network(unicode
|
self.item = ipaddress.ip_network(str
|
||||||
(self.item, "utf-8"),
|
(self.item, "utf-8"),
|
||||||
strict=False)
|
strict=False)
|
||||||
if isinstance(self.item, ipaddress.IPv4Network):
|
if isinstance(self.item, ipaddress.IPv4Network):
|
||||||
@@ -109,7 +110,7 @@ class HostsDataset(object):
|
|||||||
|
|
||||||
def match(self, connectip):
|
def match(self, connectip):
|
||||||
'''Check if the connect IP is part of the dataset'''
|
'''Check if the connect IP is part of the dataset'''
|
||||||
source = ipaddress.ip_address(unicode(connectip, "utf-8"))
|
source = ipaddress.ip_address(str(connectip, "utf-8"))
|
||||||
for item in self.dataset:
|
for item in self.dataset:
|
||||||
if item.isdomain or item.ishostname:
|
if item.isdomain or item.ishostname:
|
||||||
result = self.matchname(source) # Match host/domains first
|
result = self.matchname(source) # Match host/domains first
|
||||||
@@ -159,13 +160,13 @@ class HostsDataset(object):
|
|||||||
if isinstance(source, ipaddress.IPv4Address):
|
if isinstance(source, ipaddress.IPv4Address):
|
||||||
ips = s.dns(name, 'A')
|
ips = s.dns(name, 'A')
|
||||||
for ip in ips:
|
for ip in ips:
|
||||||
ip = ipaddress.IPv4Address(unicode(ip, 'UTF-8'))
|
ip = ipaddress.IPv4Address(str(ip, 'UTF-8'))
|
||||||
if ip == source:
|
if ip == source:
|
||||||
results.append(name)
|
results.append(name)
|
||||||
if isinstance(source, ipaddress.IPv6Address):
|
if isinstance(source, ipaddress.IPv6Address):
|
||||||
ips = s.dns(name, 'AAAA')
|
ips = s.dns(name, 'AAAA')
|
||||||
for ip in ips:
|
for ip in ips:
|
||||||
ip = ipaddress.IPv6Address(unicode(ip, 'UTF-8'))
|
ip = ipaddress.IPv6Address(str(ip, 'UTF-8'))
|
||||||
if ip == source:
|
if ip == source:
|
||||||
results.append(name)
|
results.append(name)
|
||||||
return results
|
return results
|
||||||
@@ -224,13 +225,13 @@ def _processConfigFile(filename=None, configdata=None, useSyslog=1,
|
|||||||
'''Load the specified config file, exit and log errors if it fails,
|
'''Load the specified config file, exit and log errors if it fails,
|
||||||
otherwise return a config dictionary.'''
|
otherwise return a config dictionary.'''
|
||||||
|
|
||||||
import config
|
from . import config
|
||||||
if configdata is None:
|
if configdata is None:
|
||||||
configdata = config.defaultConfigData
|
configdata = config.defaultConfigData
|
||||||
if filename is not None:
|
if filename is not None:
|
||||||
try:
|
try:
|
||||||
_readConfigFile(filename, configdata)
|
_readConfigFile(filename, configdata)
|
||||||
except Exception, e:
|
except Exception as e:
|
||||||
raise
|
raise
|
||||||
if useSyslog:
|
if useSyslog:
|
||||||
syslog.syslog(e.args[0])
|
syslog.syslog(e.args[0])
|
||||||
@@ -334,13 +335,14 @@ def _readConfigFile(path, configData=None, configGlobal={}):
|
|||||||
'DiagnosticDirectory': 'str',
|
'DiagnosticDirectory': 'str',
|
||||||
'MacroList': 'dataset',
|
'MacroList': 'dataset',
|
||||||
'MacroListVerify': 'dataset',
|
'MacroListVerify': 'dataset',
|
||||||
|
'DNSOverride': 'str',
|
||||||
'debugLevel': 'int'
|
'debugLevel': 'int'
|
||||||
}
|
}
|
||||||
|
|
||||||
# check to see if it's a file
|
# check to see if it's a file
|
||||||
try:
|
try:
|
||||||
mode = os.stat(path)[0]
|
mode = os.stat(path)[0]
|
||||||
except OSError, e:
|
except OSError as e:
|
||||||
syslog.syslog(syslog.LOG_ERR, 'ERROR stating "%s": %s'
|
syslog.syslog(syslog.LOG_ERR, 'ERROR stating "%s": %s'
|
||||||
% (path, e.strerror))
|
% (path, e.strerror))
|
||||||
return(configData)
|
return(configData)
|
||||||
@@ -388,7 +390,10 @@ def _readConfigFile(path, configData=None, configGlobal={}):
|
|||||||
if conversion == 'bool':
|
if conversion == 'bool':
|
||||||
configData[name] = _find_boolean(value)
|
configData[name] = _find_boolean(value)
|
||||||
elif conversion == 'str':
|
elif conversion == 'str':
|
||||||
configData[name] = str(value)
|
if isinstance(value, list):
|
||||||
|
configData[name] = line.split(None, 1)[1]
|
||||||
|
else:
|
||||||
|
configData[name] = str(value)
|
||||||
elif conversion == 'int':
|
elif conversion == 'int':
|
||||||
configData[name] = int(value)
|
configData[name] = int(value)
|
||||||
elif conversion == 'dataset':
|
elif conversion == 'dataset':
|
||||||
@@ -399,7 +404,7 @@ def _readConfigFile(path, configData=None, configGlobal={}):
|
|||||||
configData[name] = conversion(value)
|
configData[name] = conversion(value)
|
||||||
fp.close()
|
fp.close()
|
||||||
try:
|
try:
|
||||||
configData['AuthservID'] = _make_authserv_id(configData['AuthservID'])
|
configData['AuthservID'] = _make_authserv_id(configData.get('AuthservID', 'HOSTNAME'))
|
||||||
configData['IntHosts'] = HostsDataset(configData['InternalHosts'])
|
configData['IntHosts'] = HostsDataset(configData['InternalHosts'])
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|||||||
@@ -84,7 +84,7 @@ class Session(object):
|
|||||||
raise DNSError('Length of CNAME chain exceeds %d' % MAX_CNAME)
|
raise DNSError('Length of CNAME chain exceeds %d' % MAX_CNAME)
|
||||||
cnames[name] = cname
|
cnames[name] = cname
|
||||||
if cname in cnames:
|
if cname in cnames:
|
||||||
raise DNSError, 'CNAME loop'
|
raise DNSError('CNAME loop')
|
||||||
result = self.dns(cname, qtype, cnames=cnames)
|
result = self.dns(cname, qtype, cnames=cnames)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
@@ -103,16 +103,16 @@ def DNSLookup_pydns(name, qtype, tcpfallback=True, timeout=30):
|
|||||||
#
|
#
|
||||||
if resp.header['tc'] == True:
|
if resp.header['tc'] == True:
|
||||||
if not tcpfallback:
|
if not tcpfallback:
|
||||||
raise DNS.DNSError, 'DNS: Truncated UDP Reply, SPF records should fit in a UDP packet'
|
raise DNS.DNSError('DNS: Truncated UDP Reply, SPF records should fit in a UDP packet')
|
||||||
try:
|
try:
|
||||||
req = DNS.DnsRequest(name, qtype=qtype, protocol='tcp',
|
req = DNS.DnsRequest(name, qtype=qtype, protocol='tcp',
|
||||||
timeout=timeout)
|
timeout=timeout)
|
||||||
resp = req.req()
|
resp = req.req()
|
||||||
except DNS.DNSError, x:
|
except DNS.DNSError as x:
|
||||||
raise DNS.DNSError, 'TCP Fallback error: ' + str(x)
|
raise DNS.DNSError('TCP Fallback error: ' + str(x))
|
||||||
return [((a['name'], a['typename']), a['data']) for a in resp.answers]
|
return [((a['name'], a['typename']), a['data']) for a in resp.answers]
|
||||||
except IOError, x:
|
except IOError as x:
|
||||||
raise DNS.DNSError, 'DNS: ' + str(x)
|
raise DNS.DNSError('DNS: ' + str(x))
|
||||||
|
|
||||||
def DNSLookup_dnspython(name,qtype,tcpfallback=True,timeout=30):
|
def DNSLookup_dnspython(name,qtype,tcpfallback=True,timeout=30):
|
||||||
retVal = []
|
retVal = []
|
||||||
@@ -164,5 +164,5 @@ if __name__ == '__main__':
|
|||||||
import sys
|
import sys
|
||||||
s = Session()
|
s = Session()
|
||||||
for n,t in zip(*[iter(sys.argv[1:])]*2):
|
for n,t in zip(*[iter(sys.argv[1:])]*2):
|
||||||
print n,t
|
print(n,t)
|
||||||
print s.dns(n,t)
|
print(s.dns(n,t))
|
||||||
|
|||||||
+12
-4
@@ -150,10 +150,18 @@ def own_socketfile(milterconfig):
|
|||||||
"""If socket is Unix socket, chown to UserID before dropping privileges"""
|
"""If socket is Unix socket, chown to UserID before dropping privileges"""
|
||||||
import os
|
import os
|
||||||
user, group = user_group(milterconfig.get('UserID'))
|
user, group = user_group(milterconfig.get('UserID'))
|
||||||
if milterconfig.get('Socket')[:1] == '/':
|
offset = None
|
||||||
os.chown(milterconfig.get('Socket')[1:], user, group)
|
sockname = milterconfig.get('Socket')
|
||||||
if milterconfig.get('Socket')[:6] == "local:":
|
if sockname[:1] == '/':
|
||||||
os.chown(milterconfig.get('Socket')[6:], user, group)
|
offset = 0
|
||||||
|
elif sockname[:6] == "local:":
|
||||||
|
offset = 6
|
||||||
|
elif sockname[:5] == "unix:":
|
||||||
|
offset = 5
|
||||||
|
|
||||||
|
if offset is not None:
|
||||||
|
if os.path.exists(sockname[offset:]):
|
||||||
|
os.chown(sockname[offset:], user, group)
|
||||||
|
|
||||||
|
|
||||||
def read_keyfile(milterconfig, keytype):
|
def read_keyfile(milterconfig, keytype):
|
||||||
|
|||||||
@@ -311,6 +311,13 @@ be set:
|
|||||||
(b) KeyTable, SigningTable, no Domain, no KeyFile, no Selector;
|
(b) KeyTable, SigningTable, no Domain, no KeyFile, no Selector;
|
||||||
[fooTable options NOT IMPLEMENTED]
|
[fooTable options NOT IMPLEMENTED]
|
||||||
|
|
||||||
|
.TP
|
||||||
|
.I DNSOverride (string)
|
||||||
|
Provide a text string that a verifying milter should use instead of
|
||||||
|
consulting the DNS on each message. This is useful primarily for
|
||||||
|
testing purposes in environments where it is awkward to modify the
|
||||||
|
system DNS resolution. It should not be used in production.
|
||||||
|
|
||||||
.TP
|
.TP
|
||||||
.I PeerList (dataset)
|
.I PeerList (dataset)
|
||||||
Identifies a set of "peers" that identifies clients whose connections
|
Identifies a set of "peers" that identifies clients whose connections
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
#! /usr/bin/python
|
#! /usr/bin/python3
|
||||||
# dkimpy-milter: A DKIM signing/verification Milter application
|
# dkimpy-milter: A DKIM signing/verification Milter application
|
||||||
# Author: Scott Kitterman <scott@kitterman.com>
|
# Author: Scott Kitterman <scott@kitterman.com>
|
||||||
# Copyright 2018 Scott Kitterman
|
# Copyright 2018 Scott Kitterman
|
||||||
@@ -23,10 +23,10 @@ description = "Domain Keys Identified Mail (DKIM) signing/verifying milter for P
|
|||||||
|
|
||||||
kw = {} # Work-around for lack of 'or' requires in setuptools.
|
kw = {} # Work-around for lack of 'or' requires in setuptools.
|
||||||
try:
|
try:
|
||||||
import DNS
|
import dns
|
||||||
kw['install_requires'] = ['dkimpy>=0.7', 'pymilter', 'authres>=1.1.0', 'PyNaCl', 'ipaddress', 'PyDNS']
|
kw['install_requires'] = ['dkimpy>=0.7', 'pymilter', 'authres>=1.1.0', 'PyNaCl', 'dnspython']
|
||||||
except ImportError: # If PyDNS is not installed, prefer dnspython
|
except ImportError: # If PyDNS is not installed, prefer dnspython
|
||||||
kw['install_requires'] = ['dkimpy>=0.7', 'pymilter', 'authres>=1.1.0', 'PyNaCl', 'ipaddress', 'dnspython']
|
kw['install_requires'] = ['dkimpy>=0.7', 'pymilter', 'authres>=1.1.0', 'PyNaCl', 'PyDNS']
|
||||||
|
|
||||||
setup(
|
setup(
|
||||||
name='dkimpy-milter',
|
name='dkimpy-milter',
|
||||||
@@ -43,7 +43,7 @@ setup(
|
|||||||
'License :: OSI Approved :: GNU General Public License (GPL)',
|
'License :: OSI Approved :: GNU General Public License (GPL)',
|
||||||
'Natural Language :: English',
|
'Natural Language :: English',
|
||||||
'Operating System :: POSIX',
|
'Operating System :: POSIX',
|
||||||
'Programming Language :: Python :: 2 :: Only',
|
'Programming Language :: Python :: 3 :: Only',
|
||||||
'Topic :: Communications :: Email :: Mail Transport Agents',
|
'Topic :: Communications :: Email :: Mail Transport Agents',
|
||||||
'Topic :: Communications :: Email :: Filters',
|
'Topic :: Communications :: Email :: Filters',
|
||||||
'Topic :: Security',
|
'Topic :: Security',
|
||||||
|
|||||||
@@ -0,0 +1,12 @@
|
|||||||
|
-- -*- lua -*-
|
||||||
|
for _, keytype in ipairs({"ed25519", "rsa"}) do
|
||||||
|
for _, func in ipairs({"signing", "verify"}) do
|
||||||
|
mt.echo("testing "..keytype.." "..func)
|
||||||
|
conn = mt.connect("unix:"..keytype.."."..func..".sock")
|
||||||
|
if conn == nil then
|
||||||
|
error("mt.connect() failed "..keytype.." "..func)
|
||||||
|
end
|
||||||
|
mt.disconnect(conn)
|
||||||
|
mt.echo(keytype.." "..func.." complete")
|
||||||
|
end
|
||||||
|
end
|
||||||
Executable
+40
@@ -0,0 +1,40 @@
|
|||||||
|
-- -*- lua -*-
|
||||||
|
for _, keytype in ipairs({"ed25519", "rsa"}) do
|
||||||
|
for _, func in ipairs({"signing", "verify"}) do
|
||||||
|
mt.echo("testing "..keytype.." "..func)
|
||||||
|
conn = mt.connect("unix:"..keytype.."."..func..".sock")
|
||||||
|
if conn == nil then
|
||||||
|
error("mt.connect() failed "..keytype.." "..func)
|
||||||
|
end
|
||||||
|
if mt.conninfo(conn, "localhost", "127.0.0.1") ~= nil then
|
||||||
|
error("mt.conninfo() failed "..keytype.." "..func)
|
||||||
|
end
|
||||||
|
if mt.getreply(conn) ~= SMFIR_CONTINUE then
|
||||||
|
error("mt.conninfo() unexpected reply "..keytype.." "..func)
|
||||||
|
end
|
||||||
|
|
||||||
|
if mt.test_action(conn, SMFIF_ADDHDRS) then
|
||||||
|
print("could add headers "..keytype.." "..func)
|
||||||
|
else
|
||||||
|
error("mt.test_action() says could not add headers "..keytype.." "..func)
|
||||||
|
end
|
||||||
|
|
||||||
|
if mt.test_action(conn, SMFIF_CHGHDRS) then
|
||||||
|
print("could change headers "..keytype.." "..func)
|
||||||
|
else
|
||||||
|
error("mt.test_action() says could not change headers "..keytype.." "..func)
|
||||||
|
end
|
||||||
|
|
||||||
|
-- -- FIXME: this part of the test fails, as apparently the
|
||||||
|
-- -- dkimpy-milter claims the right to change the body of a message,
|
||||||
|
-- -- even though it shouldn't. How can we fix the negotiation?
|
||||||
|
-- if mt.test_action(conn, SMFIF_CHGBODY) then
|
||||||
|
-- error("mt.test_action() says could change body "..keytype.." "..func)
|
||||||
|
-- else
|
||||||
|
-- print("could not change body "..keytype.." "..func)
|
||||||
|
-- end
|
||||||
|
|
||||||
|
mt.disconnect(conn)
|
||||||
|
mt.echo(keytype.." "..func.." test complete")
|
||||||
|
end
|
||||||
|
end
|
||||||
@@ -0,0 +1,100 @@
|
|||||||
|
-- -*- lua -*-
|
||||||
|
|
||||||
|
msg = {
|
||||||
|
['headers'] = {
|
||||||
|
['From'] = 'Alice <alice@example.net>',
|
||||||
|
['Message-Id'] = '<dkimpy-milter-test-02@example.net>',
|
||||||
|
['To'] = 'Bob <bob@example.biz>',
|
||||||
|
['Date'] = 'Mon, 18 Feb 2019 08:32:50 -0500',
|
||||||
|
['Subject'] = 'Signing test',
|
||||||
|
['Content-Type'] = 'text/plain',
|
||||||
|
},
|
||||||
|
['body'] = "This is a test!\r\n",
|
||||||
|
}
|
||||||
|
|
||||||
|
-- returns miltertest connection object
|
||||||
|
function connect_and_send (sockname, headers, body)
|
||||||
|
conn = mt.connect(sockname)
|
||||||
|
if conn == nil then
|
||||||
|
error "mt.connect() failed"
|
||||||
|
end
|
||||||
|
if mt.conninfo(conn, "localhost", "127.0.0.1") ~= nil then
|
||||||
|
error "mt.conninfo() failed"
|
||||||
|
end
|
||||||
|
if mt.getreply(conn) ~= SMFIR_CONTINUE then
|
||||||
|
error "mt.conninfo() unexpected reply"
|
||||||
|
end
|
||||||
|
|
||||||
|
-- mt.macro(conn, SMFIC_MAIL, "i", "simple-message")
|
||||||
|
if mt.mailfrom(conn, "<alice@example.net>") ~= nil then
|
||||||
|
error "mt.mailfrom() failed"
|
||||||
|
end
|
||||||
|
if mt.getreply(conn) ~= SMFIR_CONTINUE then
|
||||||
|
error "mt.mailfrom() unexpected reply"
|
||||||
|
end
|
||||||
|
-- mt.rcptto() is called implicitly
|
||||||
|
|
||||||
|
-- send headers
|
||||||
|
for key,value in pairs(headers) do
|
||||||
|
if mt.header(conn, key, value) ~= nil then
|
||||||
|
error("mt.header(" .. key .. ") failed")
|
||||||
|
end
|
||||||
|
if mt.getreply(conn) ~= SMFIR_CONTINUE then
|
||||||
|
error("mt.header(" .. key .. ") unexpected reply")
|
||||||
|
end
|
||||||
|
end
|
||||||
|
-- send EOH
|
||||||
|
if mt.eoh(conn) ~= nil then
|
||||||
|
error "mt.eoh() failed"
|
||||||
|
end
|
||||||
|
if mt.getreply(conn) ~= SMFIR_CONTINUE then
|
||||||
|
error "mt.eoh() unexpected reply"
|
||||||
|
end
|
||||||
|
|
||||||
|
-- send body
|
||||||
|
if mt.bodystring(conn, body) ~= nil then
|
||||||
|
error "mt.bodystring() failed"
|
||||||
|
end
|
||||||
|
if mt.getreply(conn) ~= SMFIR_CONTINUE then
|
||||||
|
error "mt.bodystring() unexpected reply"
|
||||||
|
end
|
||||||
|
-- end of message; let the filter react
|
||||||
|
if mt.eom(conn) ~= nil then
|
||||||
|
error "mt.eom() failed"
|
||||||
|
end
|
||||||
|
reply = mt.getreply(conn)
|
||||||
|
if reply ~= SMFIR_CONTINUE then
|
||||||
|
error ("mt.eom() unexpected reply: " .. reply)
|
||||||
|
end
|
||||||
|
return conn
|
||||||
|
end
|
||||||
|
|
||||||
|
for _, keytype in ipairs({"ed25519", "rsa"}) do
|
||||||
|
mt.echo("testing "..keytype)
|
||||||
|
signing = connect_and_send("unix:"..keytype..".signing.sock", msg.headers, msg.body)
|
||||||
|
-- verify that a test header field got added
|
||||||
|
if not mt.eom_check(signing, MT_HDRINSERT) then
|
||||||
|
error "no header added by signer"
|
||||||
|
end
|
||||||
|
|
||||||
|
signature = mt.getheader(signing, "DKIM-Signature", 0)
|
||||||
|
|
||||||
|
mt.disconnect(signing)
|
||||||
|
|
||||||
|
mt.echo("DKIM-Signature: " .. signature)
|
||||||
|
|
||||||
|
msg.headers['DKIM-Signature'] = signature
|
||||||
|
|
||||||
|
verify = connect_and_send("unix:"..keytype..".verify.sock", msg.headers, msg.body)
|
||||||
|
|
||||||
|
if not mt.eom_check(verify, MT_HDRINSERT) then
|
||||||
|
error "no header added in verify"
|
||||||
|
end
|
||||||
|
|
||||||
|
authres = mt.getheader(verify, "Authentication-Results", 0)
|
||||||
|
mt.echo("Authentication-Results: "..authres)
|
||||||
|
|
||||||
|
mt.disconnect(verify)
|
||||||
|
|
||||||
|
mt.echo(keytype.." complete")
|
||||||
|
end
|
||||||
Executable
+2
@@ -0,0 +1,2 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
python3 -m dkimpy_milter "$@"
|
||||||
Executable
+84
@@ -0,0 +1,84 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
set -e
|
||||||
|
WORKDIR=$(mktemp -d)
|
||||||
|
TESTDIR=$(realpath "$(dirname "$0")")
|
||||||
|
DKIMPY_MILTER=${DKIMPY_MILTER:-"$TESTDIR/dkimpy-milter"}
|
||||||
|
KEY_TYPES=(ed25519 rsa)
|
||||||
|
|
||||||
|
cd "$WORKDIR"
|
||||||
|
|
||||||
|
printf "Testing %s from directory %s\n" "$DKIMPY_MILTER" "$WORKDIR"
|
||||||
|
|
||||||
|
for keytype in "${KEY_TYPES[@]}"; do
|
||||||
|
dknewkey --ktype "$keytype" "testkey.$keytype"
|
||||||
|
if [ "$keytype" = ed25519 ]; then
|
||||||
|
keyfile=KeyFileEd25519
|
||||||
|
selector=SelectorEd25519
|
||||||
|
else
|
||||||
|
keyfile=KeyFile
|
||||||
|
selector=Selector
|
||||||
|
fi
|
||||||
|
cat > "$keytype.signing.conf" <<EOF
|
||||||
|
Domain example.net
|
||||||
|
$keyfile testkey.$keytype.key
|
||||||
|
$selector testkey
|
||||||
|
Socket unix:$keytype.signing.sock
|
||||||
|
PidFile $keytype.signing.pid
|
||||||
|
Mode s
|
||||||
|
UserID $(id --name --user):$(id --name --group)
|
||||||
|
EOF
|
||||||
|
|
||||||
|
cat > "$keytype.verify.conf" <<EOF
|
||||||
|
Socket unix:$keytype.verify.sock
|
||||||
|
PidFile $keytype.verify.pid
|
||||||
|
Mode v
|
||||||
|
DNSOverride $(cat testkey.$keytype.dns)
|
||||||
|
UserID $(id --name --user):$(id --name --group)
|
||||||
|
EOF
|
||||||
|
done
|
||||||
|
|
||||||
|
cleanup() {
|
||||||
|
echo cleaning up jobs:
|
||||||
|
jobs
|
||||||
|
for keytype in "${KEY_TYPES[@]}"; do
|
||||||
|
for func in signing verify; do
|
||||||
|
if [ -s "$keytype.$func.pid" ] && kill -0 "$(cat "$keytype.$func.pid")"; then
|
||||||
|
kill "$(cat $keytype.$func.pid)"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
done
|
||||||
|
wait
|
||||||
|
for keytype in "${KEY_TYPES[@]}"; do
|
||||||
|
for func in signing verify; do
|
||||||
|
errdata="$keytype.$func.stderr"
|
||||||
|
if [ -s "$errdata" ]; then
|
||||||
|
printf -- "-> %s:\n" "$errdata"
|
||||||
|
cat "$errdata"
|
||||||
|
printf -- "-> end %s\n" "$errdata"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
done
|
||||||
|
rm -rf "$WORKDIR"
|
||||||
|
}
|
||||||
|
|
||||||
|
for keytype in "${KEY_TYPES[@]}"; do
|
||||||
|
for func in signing verify; do
|
||||||
|
PYTHONPATH="$(dirname "$TESTDIR")" "$DKIMPY_MILTER" "$keytype.$func.conf" 2>"$keytype.$func.stderr" &
|
||||||
|
done
|
||||||
|
done
|
||||||
|
trap cleanup EXIT
|
||||||
|
|
||||||
|
# ugly ugly (how are we supposed to know that the milters are all ready?):
|
||||||
|
sleep 2
|
||||||
|
|
||||||
|
# uses miltertest from opendkim:
|
||||||
|
for x in ${TESTS:-"$TESTDIR"/*.miltertest}; do
|
||||||
|
if ! [ -e "$x" ]; then
|
||||||
|
if [ -e "$TESTDIR/$x" ]; then
|
||||||
|
x="$TESTDIR/$x"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
printf -- "-> running %s...\n" "$x"
|
||||||
|
miltertest -s "$x"
|
||||||
|
done
|
||||||
Reference in New Issue
Block a user