mirror of
https://github.com/jlu5/PyLink.git
synced 2024-12-25 20:22:45 +01:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
# servprotect.py: Protects against KILL and nick collision floods
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
|
|
||
|
|
from expiringdict import ExpiringDict
|
||
|
|
|
||
|
|
import utils
|
||
|
|
from log import log
|
||
|
|
|
||
|
|
# TODO: make length and time configurable
|
||
|
|
savecache = ExpiringDict(max_len=5, max_age_seconds=10)
|
||
|
|
killcache = ExpiringDict(max_len=5, max_age_seconds=10)
|
||
|
|
|
||
|
|
def handle_kill(irc, numeric, command, args):
|
||
|
|
"""
|
||
|
|
Tracks kills against PyLink clients. If too many are received,
|
||
|
|
automatically disconnects from the network.
|
||
|
|
"""
|
||
|
|
if killcache.setdefault(irc.name, 1) >= 5:
|
||
|
|
log.error('(%s) servprotect: Too many kills received, aborting!', irc.name)
|
||
|
|
irc.disconnect()
|
||
|
|
|
||
|
|
log.debug('(%s) servprotect: Incrementing killcache by 1', irc.name)
|
||
|
|
killcache[irc.name] += 1
|
||
|
|
|
||
|
|
utils.add_hook(handle_kill, 'KILL')
|
||
|
|
|
||
|
|
def handle_save(irc, numeric, command, args):
|
||
|
|
"""
|
||
|
|
Tracks SAVEs (nick collision) against PyLink clients. If too many are received,
|
||
|
|
automatically disconnects from the network.
|
||
|
|
"""
|
||
|
|
if savecache.setdefault(irc.name, 0) >= 5:
|
||
|
|
log.error('(%s) servprotect: Too many nick collisions, aborting!', irc.name)
|
||
|
|
irc.disconnect()
|
||
|
|
|
||
|
|
log.debug('(%s) servprotect: Incrementing savecache by 1', irc.name)
|
||
|
|
savecache[irc.name] += 1
|
||
|
|
|
||
|
|
utils.add_hook(handle_save, 'SAVE')
|