Browse Source

update baddies detector

pull/3/head
Jeff Becker 8 years ago
parent
commit
d0e76fa6eb
  1. 9
      baddie-detector/baddiefinder/processor.py
  2. 4
      baddie-detector/baddiefinder/util.py
  3. 2
      baddie-detector/baddies.ini

9
baddie-detector/baddiefinder/processor.py

@ -10,15 +10,14 @@ class BaddieProcessor:
def hook(self, entry): def hook(self, entry):
now = datetime.datetime.now()
for f in self._filters: for f in self._filters:
if f.process(entry) is True: if f.process(entry) is True:
self.add_baddie(entry, 'detected by {}'.format(f.name)) self.add_baddie(entry, 'detected by {} on {}'.format(f.name, now.strftime("%c").replace(":",'-')))
def add_baddie(self, entry, reason): def add_baddie(self, entry, reason):
addr = util.getaddress(entry) addr = util.getaddress(entry).decode('ascii')
if addr not in self._baddies: self._baddies[addr] = reason
self._baddies[addr] = ''
self._baddies[addr] += reason + ' '
def write_blocklist(self, f): def write_blocklist(self, f):
f.write('# baddies blocklist generated on {}\n'.format(datetime.datetime.now())) f.write('# baddies blocklist generated on {}\n'.format(datetime.datetime.now()))

4
baddie-detector/baddiefinder/util.py

@ -5,7 +5,9 @@ def getaddress(info):
for addr in info.addrs: for addr in info.addrs:
opts = addr.options opts = addr.options
if b'host' in opts: if b'host' in opts:
return opts[b'host'] h = opts[b'host']
if b':' not in h:
return h
def getcaps(info): def getcaps(info):
""" """

2
baddie-detector/baddies.ini

@ -1,2 +1,2 @@
[thresholds] [thresholds]
max_floodfills_per_ip = 2 max_floodfills_per_ip = 3
Loading…
Cancel
Save