/usr/share/denyhosts/DenyHosts/counter.py is in denyhosts 2.10-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 | import logging
import time
debug = logging.getLogger("counter").debug
# TODO define __eq__ for this class
class CounterRecord(object):
def __init__(self, count=0, date=None):
self.__count = count
if not date:
self.__date = time.asctime()
else:
self.__date = date
def __str__(self):
return "%d:%s" % (self.__count, self.__date)
def __repr__(self):
return "CountRecord <%d - %s>" % (self.__count, self.__date)
def __add__(self, increment):
# AAAAH!
# Expressions like `a + 4` are usually assumed to not have any side effects,
# but this is not the case with CounterRecord objects. With `c = CounterRecord()`,
# simply evaluating `c + 1` will increment c.__count by 1. This is horrifying.
self.__count += increment
self.__date = time.asctime()
return self
def get_count(self):
return self.__count
def get_date(self):
return self.__date
def reset_count(self):
self.__count = 0
def age_count(self, age):
cutoff = long(time.time()) - age
epoch = time.mktime(time.strptime(self.__date))
#debug("cutoff : %d", cutoff)
#debug("epoch : %d", epoch)
if cutoff > epoch:
self.__count = 0
# TODO replace this with collections.defaultdict
class Counter(dict):
"""
Behaves like a dictionary, except that if the key isn't found, 0 is returned
rather than an exception. This is suitable for situations like:
c = Counter()
c['x'] += 1
"""
def __init__(self):
dict.__init__(self)
def __getitem__(self, k):
try:
return dict.__getitem__(self, k)
except KeyError:
count_rec = CounterRecord(0)
#debug("%s - %s", k, count_rec)
self.__setitem__(k, count_rec)
#debug("dict: %s", dict.values(self))
#debug("count_rec: %s", count_rec)
return count_rec
|