/
filter_manager.py
105 lines (90 loc) · 3.69 KB
/
filter_manager.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import sys
import logging
from spam_filter import FilterMessage
import utils
logz=logging
logging.basicConfig(level=logging.DEBUG, stream=sys.stdout)
class FilterManager():
def __init__(self, msg, data):
self.msg=msg
self.data=data
self.score=0.0
self.str_scan=""
self.passed_tests=""
def run(self):
score=0
filter=FilterMessage(self.msg, self.data)
ret=self.msg.hasSubject()
if ret==True:
self.passed_tests=self.passed_tests+"Subject exists"+"\n"
else:
score=utils.getScore("SUBJECT_MISSING", self.data)
self.str_scan=self.str_scan+"\tSUBJECT_MISSING"+"\trule score:"+str(score)+"\n"
self.score=self.score+score
ret=self.msg.hasFrom()
if ret==True:
self.passed_tests=self.passed_tests+"Subject exists"+"\n"
else:
score=utils.getScore("FROM_MISSING", self.data)
self.str_scan=self.str_scan+"\tFROM_MISSING"+"\trule score:"+str(score)+"\n"
self.score=self.score+score
ret=filter.check_header('Message-Id')
if ret==0:
self.passed_tests=self.passed_tests+"Header Message-Id is valid"+"\n"
else:
score=utils.getScore("MESSAGEID_INVALID", self.data)
self.str_scan=self.str_scan+"\tMESSAGEID_INVALID"+"\trule score:"+str(score)+"\n"
self.score=self.score+score
ret=filter.check_header('Date')
if ret==0:
self.passed_tests=self.passed_tests+"Header Date is valid"+"\n"
else:
score=utils.getScore("DATE_INVALID", self.data)
self.str_scan=self.str_scan+"\tDATE_INVALID"+"\trule score:"+str(score)+"\n"
if score is not None:
self.score=self.score+float(score)
ret=filter.check_header('From')
if ret==0:
self.passed_tests=self.passed_tests+"Header From is valid"+"\n"
else:
score=utils.getScore("FROM_INVALID", self.data)
self.str_scan=self.str_scan+"\tFROM_INVALID"+"\trule score:"+str(score)+"\n"
self.score=self.score+score
ret=filter.check_forged_received_ip()
if ret==0:
self.passed_tests=self.passed_tests+"Not forged header"+"\n"
else:
score=utils.getScore("RECEIVED_FORGED", self.data)
self.str_scan=self.str_scan+"\tRECEIVED_FORGED"+"\trule score:"+str(score)+"\n"
self.score=self.score+score
ret=filter.filter_source_ip()
if ret is None:
self.passed_tests=self.passed_tests+"Source IP is not Spam"+"\n"
else:
score=utils.getScore("DNSBL_ANSWER", self.data)
self.str_scan=self.str_scan+"\tDNSBL_ANSWER"+"\trule score:"+str(score)+"\n"
self.score=self.score+score
self.printResult()
ret=self.msg.mark_subject(self.getLevelSpam())
if ret!=0:
logz.info("Couldn't mark subject")
def getLevelSpam(self):
if self.score<5.0:
level="No Spam"
elif self.score>=5.0 and self.score<10.0:
level="Spam:Low"
elif self.score>=10.0 and self.score<15.0:
level="Spam:Medium"
elif self.score>=15.0 and self.score<20.0:
level="Spam:High"
elif self.score>=20.0:
level="Spam:Very High"
else:
pass
return level
def printResult(self):
print "----------------------------------------------------------"
print "Failed Tests:"
print self.str_scan
print "Final Score:[%2.2f]" %self.score
print "Result:%s" %self.getLevelSpam()