Cleanup, indentation, pyflakes

This commit is contained in:
Scott Kitterman
2018-03-10 00:52:45 -05:00
parent fd39384e78
commit 6348bdcdc7
3 changed files with 18 additions and 27 deletions
+18 -25
View File
@@ -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
-1
View File
@@ -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
-1
View File
@@ -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):