Scott Kitterman
267 lines
9.7 KiB
Python
267 lines
9.7 KiB
Python
#! /usr/bin/python2
|
|
# Original dkim-milter.py code:
|
|
# Author: Stuart D. Gathman <stuart@bmsi.com>
|
|
# Copyright 2007 Business Management Systems, Inc.
|
|
# This code is under GPL. See COPYING for details.
|
|
|
|
# dkimpy-milter: A DKIM signing/verification Milter application
|
|
# Author: Scott Kitterman <scott@kitterman.com>
|
|
# Copyright 2018 Scott Kitterman
|
|
""" This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with this program; if not, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA."""
|
|
|
|
import sys
|
|
import syslog
|
|
import Milter
|
|
import dkim
|
|
from dkim.dnsplug import get_txt
|
|
from dkim.util import parse_tag_value
|
|
import authres
|
|
import os
|
|
import tempfile
|
|
import StringIO
|
|
import re
|
|
from Milter.config import MilterConfigParser
|
|
from Milter.utils import iniplist,parse_addr,parseaddr
|
|
import dkimpy_milter.config as config
|
|
from dkimpy_milter.util import drop_privileges
|
|
from dkimpy_milter.util import setExceptHook
|
|
from dkimpy_milter.util import write_pid
|
|
from dkimpy_milter.util import read_keyfile
|
|
|
|
FWS = re.compile(r'\r?\n[ \t]+')
|
|
|
|
class dkimMilter(Milter.Base):
|
|
"Milter to check and sign DKIM. Each connection gets its own instance."
|
|
|
|
def __init__(self, milterconfig, privatersa=False, privateed25519=False):
|
|
self.mailfrom = None
|
|
self.id = Milter.uniqueID()
|
|
# we don't want config used to change during a connection
|
|
self.conf = milterconfig
|
|
self.privatersa = privatersa
|
|
self.privateed25519 = privateed25519
|
|
self.fp = None
|
|
|
|
@Milter.noreply
|
|
def connect(self,hostname,unused,hostaddr):
|
|
self.internal_connection = False
|
|
self.hello_name = None
|
|
# sometimes people put extra space in sendmail config, so we strip
|
|
self.receiver = self.getsymval('j').strip()
|
|
if hostaddr and len(hostaddr) > 0:
|
|
ipaddr = hostaddr[0]
|
|
"""if iniplist(ipaddr,self.conf.internal_connect): FIXME
|
|
self.internal_connection = True"""
|
|
else: ipaddr = ''
|
|
self.connectip = ipaddr
|
|
if self.internal_connection:
|
|
connecttype = 'INTERNAL'
|
|
else:
|
|
connecttype = 'EXTERNAL'
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog("connect from %s at %s %s" % (hostname,hostaddr,connecttype))
|
|
return Milter.CONTINUE
|
|
|
|
# multiple messages can be received on a single connection
|
|
# envfrom (MAIL FROM in the SMTP protocol) seems to mark the start
|
|
# of each message.
|
|
@Milter.noreply
|
|
def envfrom(self,f,*str):
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog("mail from",f,str)
|
|
self.fp = StringIO.StringIO()
|
|
self.mailfrom = f
|
|
t = parse_addr(f)
|
|
if len(t) == 2: t[1] = t[1].lower()
|
|
self.canon_from = '@'.join(t)
|
|
self.user = self.getsymval('{auth_authen}')
|
|
self.has_dkim = 0
|
|
self.author = None
|
|
self.arheaders = []
|
|
self.arresults = []
|
|
'''if self.user:
|
|
# Very simple SMTP AUTH policy by default:
|
|
# any successful authentication is considered INTERNAL
|
|
self.internal_connection = True
|
|
auth_type = self.getsymval('{auth_type}')
|
|
ssl_bits = self.getsymval('{cipher_bits}')
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog(
|
|
"SMTP AUTH:",self.user,"sslbits =",ssl_bits, auth_type,
|
|
"ssf =",self.getsymval('{auth_ssf}'), "INTERNAL"
|
|
)
|
|
# Detailed authorization policy is configured in the access file below.
|
|
self.arresults.append(
|
|
authres.SMTPAUTHAuthenticationResult(result = 'pass',
|
|
result_comment = auth_type+' sslbits='+ssl_bits, smtp_auth = self.user)
|
|
)'''
|
|
return Milter.CONTINUE
|
|
|
|
@Milter.noreply
|
|
def header(self,name,val):
|
|
lname = name.lower()
|
|
if lname == 'dkim-signature':
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog("%s: %s" % (name,val))
|
|
self.has_dkim += 1
|
|
if lname == 'from':
|
|
fname,self.author = parseaddr(val)
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog("%s: %s" % (name,val))
|
|
elif lname == 'authentication-results':
|
|
self.arheaders.append(val)
|
|
if self.fp:
|
|
self.fp.write("%s: %s\n" % (name,val))
|
|
return Milter.CONTINUE
|
|
|
|
@Milter.noreply
|
|
def eoh(self):
|
|
if self.fp:
|
|
self.fp.write("\n") # terminate headers
|
|
self.bodysize = 0
|
|
return Milter.CONTINUE
|
|
|
|
@Milter.noreply
|
|
def body(self,chunk): # copy body to temp file
|
|
if self.fp:
|
|
self.fp.write(chunk) # IOError causes TEMPFAIL in milter
|
|
self.bodysize += len(chunk)
|
|
return Milter.CONTINUE
|
|
|
|
def eom(self):
|
|
if not self.fp:
|
|
return Milter.ACCEPT # no message collected - so no eom processing
|
|
# Remove existing Authentication-Results headers for our authserv_id
|
|
for i,val in enumerate(self.arheaders,1):
|
|
# FIXME: don't delete A-R headers from trusted MTAs
|
|
ar = authres.AuthenticationResultsHeader.parse_value(FWS.sub('',val))
|
|
if ar.authserv_id == self.receiver:
|
|
self.chgheader('authentication-results',i,'')
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog('REMOVE: ',val)
|
|
# Check or sign DKIM
|
|
self.fp.seek(0)
|
|
if self.internal_connection or conf.get('Mode') == 's' or conf.get('Mode') == 'sv':
|
|
txt = self.fp.read()
|
|
self.sign_dkim(txt)
|
|
result = None
|
|
if self.has_dkim and (conf.get('Mode') == 'v' or conf.get('Mode') == 'sv'):
|
|
txt = self.fp.read()
|
|
self.check_dkim(txt)
|
|
else:
|
|
result = 'none'
|
|
if self.arresults:
|
|
h = authres.AuthenticationResultsHeader(authserv_id = self.receiver,
|
|
results=self.arresults)
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog(h)
|
|
name,val = str(h).split(': ',1)
|
|
self.addheader(name,val,0)
|
|
return Milter.CONTINUE
|
|
|
|
def sign_dkim(self,txt):
|
|
conf = self.conf
|
|
try:
|
|
d = dkim.DKIM(txt)
|
|
h = d.sign(conf.get('Selector'),conf.get('Domain'), self.privatersa,
|
|
canonicalize=('relaxed','simple'))
|
|
name,val = h.split(': ',1)
|
|
self.addheader(name,val.strip().replace('\r\n','\n'),0)
|
|
if self.privateed25519:
|
|
d = dkim.DKIM(txt)
|
|
h = d.sign(conf.get('SelectorEd25519'),conf.get('Domain'), self.privateed25519,
|
|
canonicalize=('relaxed','simple'))
|
|
name,val = h.split(': ',1)
|
|
self.addheader(name,val.strip().replace('\r\n','\n'),0)
|
|
except dkim.DKIMException as x:
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog('DKIM: %s'%x)
|
|
except Exception as x:
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog("sign_dkim: %s",x,exc_info=True)
|
|
|
|
def check_dkim(self,txt):
|
|
res = False
|
|
conf = self.conf
|
|
d = dkim.DKIM(txt)
|
|
for y in range(self.has_dkim): # Verify _ALL_ the signatures
|
|
try:
|
|
res = d.verify(idx=y)
|
|
if res:
|
|
self.dkim_comment = 'Good %d bit signature.' % d.keysize
|
|
else:
|
|
self.dkim_comment = 'Bad %d bit signature.' % d.keysize
|
|
except dkim.DKIMException as x:
|
|
self.dkim_comment = str(x)
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog('DKIM: %s'%x)
|
|
except Exception as x:
|
|
self.dkim_comment = str(x)
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog("check_dkim: %s",x,exc_info=True)
|
|
self.header_i = d.signature_fields.get(b'i')
|
|
self.header_d = d.signature_fields.get(b'd')
|
|
if res:
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog('DKIM: Pass (%s)'%d.domain)
|
|
self.dkim_domain = d.domain
|
|
else:
|
|
fd,fname = tempfile.mkstemp(".dkim")
|
|
with os.fdopen(fd,"w+b") as fp:
|
|
fp.write(txt)
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog('DKIM: Fail (saved as %s)'%fname)
|
|
if res:
|
|
result = 'pass'
|
|
else:
|
|
result = 'fail'
|
|
self.arresults.append(
|
|
authres.DKIMAuthenticationResult(result=result,
|
|
header_i = self.header_i, header_d = self.header_d,
|
|
result_comment = self.dkim_comment)
|
|
)
|
|
return
|
|
|
|
def main():
|
|
privateRSA = False
|
|
privateEd25519 = False
|
|
configFile = '/etc/dkimpy-milter.conf'
|
|
if len(sys.argv) > 1:
|
|
if sys.argv[1] in ( '-?', '--help', '-h' ):
|
|
print('usage: dkimpy-milter [<configfilename>]')
|
|
sys.exit(1)
|
|
configFile = sys.argv[1]
|
|
milterconfig = config._processConfigFile(filename = configFile)
|
|
if milterconfig.get('Syslog'):
|
|
syslog.openlog(os.path.basename(sys.argv[0]), syslog.LOG_PID, syslog.LOG_MAIL)
|
|
setExceptHook()
|
|
write_pid(milterconfig)
|
|
if milterconfig.get('KeyFile'):
|
|
privateRSA = read_keyfile(milterconfig, 'RSA')
|
|
if milterconfig.get('KeyFileEd25519'):
|
|
privateEd25519 = read_keyfile(milterconfig, 'Ed25519')
|
|
drop_privileges(milterconfig)
|
|
if milterconfig.get('Syslog'):
|
|
syslog.syslog('dkimpy-milter started. user: {0}'.format(milterconfig.get('UserID')))
|
|
Milter.factory = dkimMilter(milterconfig, privatersa=privateRSA, privateed25519=privateEd25519)
|
|
Milter.set_flags(Milter.CHGHDRS + Milter.ADDHDRS)
|
|
miltername = 'dkimpy-filter'
|
|
socketname = milterconfig.get('Socket')
|
|
sys.stdout.flush()
|
|
Milter.runmilter(miltername,socketname,240)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|