Cleanup, indentation, pyflakes
This commit is contained in:
+18
-25
@@ -25,15 +25,12 @@ import sys
|
|||||||
import syslog
|
import syslog
|
||||||
import Milter
|
import Milter
|
||||||
import dkim
|
import dkim
|
||||||
from dkim.dnsplug import get_txt
|
|
||||||
from dkim.util import parse_tag_value
|
|
||||||
import authres
|
import authres
|
||||||
import os
|
import os
|
||||||
import tempfile
|
import tempfile
|
||||||
import StringIO
|
import StringIO
|
||||||
import re
|
import re
|
||||||
from Milter.config import MilterConfigParser
|
from Milter.utils import parse_addr,parseaddr
|
||||||
from Milter.utils import iniplist,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
|
||||||
from dkimpy_milter.util import setExceptHook
|
from dkimpy_milter.util import setExceptHook
|
||||||
@@ -173,15 +170,12 @@ class dkimMilter(Milter.Base):
|
|||||||
domain = milterconfig.get('Domain')
|
domain = milterconfig.get('Domain')
|
||||||
else:
|
else:
|
||||||
domain = ''
|
domain = ''
|
||||||
if (self.fdomain in domain) and ((not milterconfig.get('Mode') == 'v') or not self.external_connection):
|
if (self.fdomain in domain) and not milterconfig.get('Mode') == 'v' and not self.external_connection:
|
||||||
txt = self.fp.read()
|
txt = self.fp.read()
|
||||||
self.sign_dkim(txt)
|
self.sign_dkim(txt)
|
||||||
result = None
|
|
||||||
if (self.has_dkim) and (not self.internal_connection) and (milterconfig.get('Mode') == 'v' or milterconfig.get('Mode') == 'sv'):
|
if (self.has_dkim) and (not self.internal_connection) and (milterconfig.get('Mode') == 'v' or milterconfig.get('Mode') == 'sv'):
|
||||||
txt = self.fp.read()
|
txt = self.fp.read()
|
||||||
self.check_dkim(txt)
|
self.check_dkim(txt)
|
||||||
else:
|
|
||||||
result = 'none'
|
|
||||||
if self.arresults:
|
if self.arresults:
|
||||||
h = authres.AuthenticationResultsHeader(authserv_id = self.AuthservID,
|
h = authres.AuthenticationResultsHeader(authserv_id = self.AuthservID,
|
||||||
results=self.arresults)
|
results=self.arresults)
|
||||||
@@ -225,23 +219,22 @@ class dkimMilter(Milter.Base):
|
|||||||
|
|
||||||
def check_dkim(self,txt):
|
def check_dkim(self,txt):
|
||||||
res = False
|
res = False
|
||||||
conf = self.conf
|
|
||||||
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)
|
res = d.verify(idx=y)
|
||||||
if res:
|
if res:
|
||||||
self.dkim_comment = 'Good {0} bit {1} signature.'.format(d.keysize, d.signature_fields.get(b'a'))
|
self.dkim_comment = 'Good {0} bit {1} signature.'.format(d.keysize, d.signature_fields.get(b'a'))
|
||||||
else:
|
else:
|
||||||
self.dkim_comment = 'Bad {0} bit {1} signature.'.format(d.keysize, d.signature_fields.get(b'a'))
|
self.dkim_comment = 'Bad {0} bit {1} signature.'.format(d.keysize, 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'):
|
||||||
syslog.syslog('DKIM: {0}'.format(x))
|
syslog.syslog('DKIM: {0}'.format(x))
|
||||||
except Exception as x:
|
except Exception as x:
|
||||||
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 = d.signature_fields.get(b'i')
|
||||||
self.header_d = d.signature_fields.get(b'd')
|
self.header_d = d.signature_fields.get(b'd')
|
||||||
self.header_a = d.signature_fields.get(b'a')
|
self.header_a = d.signature_fields.get(b'a')
|
||||||
@@ -259,14 +252,14 @@ class dkimMilter(Milter.Base):
|
|||||||
else:
|
else:
|
||||||
syslog.syslog('DKIM: Fail ({0})'.format(d.domain))
|
syslog.syslog('DKIM: Fail ({0})'.format(d.domain))
|
||||||
if res:
|
if res:
|
||||||
result = 'pass'
|
result = 'pass'
|
||||||
else:
|
else:
|
||||||
result = 'fail'
|
result = 'fail'
|
||||||
res = False
|
res = False
|
||||||
self.arresults.append(
|
self.arresults.append(
|
||||||
authres.DKIMAuthenticationResult(result=result,
|
authres.DKIMAuthenticationResult(result=result,
|
||||||
header_i = self.header_i, header_d = self.header_d, header_a = self.header_a,
|
header_i = self.header_i, header_d = self.header_d, header_a = self.header_a,
|
||||||
result_comment = self.dkim_comment)
|
result_comment = self.dkim_comment)
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
|
|||||||
@@ -28,7 +28,6 @@ import syslog
|
|||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import re
|
import re
|
||||||
import urllib
|
|
||||||
import stat
|
import stat
|
||||||
import dkim
|
import dkim
|
||||||
import socket
|
import socket
|
||||||
|
|||||||
@@ -46,7 +46,6 @@ def fold(header):
|
|||||||
j = i + 1
|
j = i + 1
|
||||||
pre += header[:j] + b"\n "
|
pre += header[:j] + b"\n "
|
||||||
header = header[j:]
|
header = header[j:]
|
||||||
namelen = 0
|
|
||||||
return pre + header
|
return pre + header
|
||||||
|
|
||||||
def user_group(userid):
|
def user_group(userid):
|
||||||
|
|||||||
Reference in New Issue
Block a user