def setUp(self):
self.true_strings = ['1', 't', 'true', 'y', 'yes']
self.false_strings = ['', 'false', 'ye', 'tr']
self.prefs = Prefs()
self.prefs._Prefs__data['WORK_DIR'] = ospj(dirname(__file__), 'data/utils/')
self.prefs._Prefs__data['DAEMON_LOG'] = ospj(self.prefs.get('WORK_DIR'), 'daemon.log')
self.prefs._Prefs__data['ADMIN_EMAIL'] = '*****@*****.**'
self.prefs._Prefs__data['SMTP_FROM'] = '*****@*****.**'
self.prefs._Prefs__data['SMTP_SUBJECT'] = 'Denyhosts Mailinator'
self.prefs._Prefs__data['SMTP_DATE_FORMAT'] = '%Y%m%d'
self.prefs._Prefs__data['SMTP_HOST'] = 'localhost'
self.prefs._Prefs__data['SMTP_PORT'] = 2500
self.prefs._Prefs__data['SMTP_SSL'] = 'False'
def setUp(self):
self.work_dir = mkdtemp()
self.warned_hosts_filename = ospj(self.work_dir, ALLOWED_WARNED_HOSTS)
self.prefs = Prefs()
self.prefs._Prefs__data['WORK_DIR'] = self.work_dir
self.prefs._Prefs__data['ALLOWED_HOSTS_HOSTNAME_LOOKUP'] = 'false'
self.allowed_hosts = AllowedHosts(self.prefs)
def setUp(self):
super(SyncTestReceiveHosts, self).setUp()
self.prefs = Prefs()
self.prefs._Prefs__data['SYNC_SERVER'] = LOCAL_SYNC_SERVER_URL
self.work_dir = mkdtemp()
self.prefs._Prefs__data['WORK_DIR'] = self.work_dir
self.test_hosts = ['host1', 'host2', 'host3']
self.remote_sync_server.add_hosts(self.test_hosts)
def setUp(self):
self.prefs = Prefs()
work_dir = ospj(dirname(__file__), 'data/purgecounter/dynamic')
self.filename = ospj(work_dir, PURGE_HISTORY)
self.prefs._Prefs__data['WORK_DIR'] = work_dir
self.counter = Counter()
host = 'host'
count = 1
self.counter[host] += count
self.test_string = '%s:%s\n' % (host, self.counter[host])
def setUp(self):
super(SyncTestSendHosts, self).setUp()
self.prefs = Prefs()
self.prefs._Prefs__data['SYNC_SERVER'] = LOCAL_SYNC_SERVER_URL
work_dir = mkdtemp()
self.prefs._Prefs__data['WORK_DIR'] = work_dir
self.test_hosts = ['host1', 'host2', 'host3']
with open(ospj(work_dir, SYNC_HOSTS), 'w') as f:
for host in self.test_hosts:
print(host, file=f)
def setUp(self):
self.directory = ospj(dirname(__file__), 'data/deny_hosts')
self.work_dir = ospj(self.directory, 'work')
self.logfile = ospj(self.work_dir, 'logfile')
self.prefs = Prefs()
self.lock_file = LockFile(ospj(self.directory, 'lockfile'))
self.lock_file.remove(die_=False)
self.lock_file.create()
self.prefs._Prefs__data['ETC_DIR'] = ospj(self.directory, 'etc')
self.prefs._Prefs__data['WORK_DIR'] = self.work_dir
def setup_denyhosts(self):
self.prefs = Prefs()
self.lock_file = LockFile(ospj(self.DIRECTORY, 'lockfile'))
self.lock_file.remove(die_=False)
self.lock_file.create()
self.prefs._Prefs__data['ETC_DIR'] = ospj(self.DIRECTORY, 'etc')
self.prefs._Prefs__data['WORK_DIR'] = self.WORK_DIR
self.prefs._Prefs__data['HOSTS_DENY'] = ospj(self.prefs._Prefs__data['ETC_DIR'], 'hosts.deny')
self.prefs._Prefs__data['DENY_THRESHOLD_INVALID'] = 5
self.prefs._Prefs__data['DENY_THRESHOLD_VALID'] = 5
self.prefs._Prefs__data['DENY_THRESHOLD_ROOT'] = 0
self.prefs._Prefs__data['DENY_THRESHOLD_RESTRICTED'] = 5
def setUp(self):
self.allowed_hosts = ['host1', 'host2']
self.prefs = Prefs()
self.work_dir = mkdtemp()
self.prefs._Prefs__data['WORK_DIR'] = self.work_dir
keys = [
'DENY_THRESHOLD_INVALID',
'DENY_THRESHOLD_VALID',
'DENY_THRESHOLD_ROOT',
'DENY_THRESHOLD_RESTRICTED',
]
for key in keys:
self.prefs._Prefs__data[key] = 0
config_file = arg
if opt in ('-m', '--migrate'):
migrate = 1
if opt in ('-p', '--purge'):
purge = 1
if opt in ('-s', '--sync'):
sync_mode = 1
if opt == '--daemon':
daemon = 1
if opt == '--upgrade099':
upgrade099 = 1
if opt == '--version':
print "DenyHosts version:", VERSION
sys.exit(0)
prefs = Prefs(config_file)
first_time = 0
try:
os.makedirs(prefs.get('WORK_DIR'))
first_time = 1
except Exception, e:
if e[0] != 17:
print e
sys.exit(1)
setup_logging(prefs, enable_debug, verbose, daemon)
if not logfiles or daemon:
logfiles = [prefs.get('SECURE_LOG')]
elif len(logfiles) > 1:
class UtilsTest(unittest.TestCase):
def setUp(self):
self.true_strings = ['1', 't', 'true', 'y', 'yes']
self.false_strings = ['', 'false', 'ye', 'tr']
self.prefs = Prefs()
self.prefs._Prefs__data['WORK_DIR'] = ospj(dirname(__file__),
'data/utils/')
self.prefs._Prefs__data['DAEMON_LOG'] = ospj(
self.prefs.get('WORK_DIR'), 'daemon.log')
self.prefs._Prefs__data['ADMIN_EMAIL'] = '*****@*****.**'
self.prefs._Prefs__data['SMTP_FROM'] = '*****@*****.**'
self.prefs._Prefs__data['SMTP_SUBJECT'] = 'Denyhosts Mailinator'
self.prefs._Prefs__data['SMTP_DATE_FORMAT'] = '%Y%m%d'
self.prefs._Prefs__data['SMTP_HOST'] = 'localhost'
self.prefs._Prefs__data['SMTP_PORT'] = 2500
self.prefs._Prefs__data['SMTP_SSL'] = 'False'
def test_is_true(self):
for string in self.true_strings:
self.assertTrue(util.is_true(string))
for string in self.false_strings:
self.assertFalse(util.is_true(string))
def test_is_false(self):
for string in self.true_strings:
self.assertFalse(util.is_false(string))
for string in self.false_strings:
self.assertTrue(util.is_false(string))
def test_parse_host(self):
for line, expected in HOST_TEST_DATA:
self.assertEqual(util.parse_host(line), expected)
def test_valid_ip(self):
for ip, expected in TEST_IPS:
self.assertEqual(util.is_valid_ip_address(ip), expected)
def test_setup_logging_with_debug(self):
self.assertIsNone(util.setup_logging(self.prefs, True, False, False))
def test_setup_logging_no_debug(self):
self.assertIsNone(util.setup_logging(self.prefs, False, False, False))
def test_seconds(self):
self.assertEqual(util.calculate_seconds(1), 1)
def test_invalid_seconds_format(self):
with self.assertRaises(Exception) as cm:
util.calculate_seconds(datetime.now())
self.assertEqual(cm.exception, 'Error')
def test_seconds_zero_false(self):
with self.assertRaises(Exception) as cm:
util.calculate_seconds('0', False)
self.assertEqual(cm.exception, 'Error')
def test_send_email_success(self):
self.assertIsNone(util.send_email(self.prefs,
'testing report success'))
def test_send_email_error(self):
self.assertIsNone(util.send_email(self.prefs,
'testing report failure'))
def test_whitespace(self):
self.assertEqual(
util.normalize_whitespace('testing whitespace for denyhosts'),
'testing whitespace for denyhosts')
def test_get_hostname_bug_145(self):
self.assertEqual(util.gethostbyname('68.ip-51-38-131.eu'),
'51.38.131.68')
config_file = arg
if opt in ('-m', '--migrate'):
migrate = 1
if opt in ('-p', '--purge'):
purge = 1
if opt in ('-s', '--sync'):
sync_mode = 1
if opt == '--daemon':
daemon = 1
if opt == '--upgrade099':
upgrade099 = 1
if opt == '--version':
print "DenyHosts version:", VERSION
sys.exit(0)
prefs = Prefs(config_file)
first_time = 0
try:
os.makedirs(prefs.get('WORK_DIR'))
first_time = 1
except Exception, e:
if e[0] != 17:
print e
sys.exit(1)
setup_logging(prefs, enable_debug, verbose, daemon)
if not logfiles or daemon:
logfiles = [prefs.get('SECURE_LOG')]
elif len(logfiles) > 1:
config_file = arg
if opt in ("-m", "--migrate"):
migrate = 1
if opt in ("-p", "--purge"):
purge = 1
if opt in ("-s", "--sync"):
sync_mode = 1
if opt == "--daemon":
daemon = 1
if opt == "--upgrade099":
upgrade099 = 1
if opt == "--version":
print "DenyHosts version:", VERSION
sys.exit(0)
prefs = Prefs(config_file)
first_time = 0
try:
os.makedirs(prefs.get("WORK_DIR"))
first_time = 1
except Exception, e:
if e[0] != 17:
print e
sys.exit(1)
setup_logging(prefs, enable_debug, verbose, daemon)
if not logfiles or daemon:
logfiles = [prefs.get("SECURE_LOG")]
elif len(logfiles) > 1:
class UtilsTest(unittest.TestCase):
def setUp(self):
self.true_strings = ['1', 't', 'true', 'y', 'yes']
self.false_strings = ['', 'false', 'ye', 'tr']
self.prefs = Prefs()
self.prefs._Prefs__data['WORK_DIR'] = ospj(dirname(__file__),
'data/utils/')
self.prefs._Prefs__data['DAEMON_LOG'] = ospj(
self.prefs.get('WORK_DIR'), 'daemon.log')
self.prefs._Prefs__data['ADMIN_EMAIL'] = '*****@*****.**'
self.prefs._Prefs__data['SMTP_FROM'] = '*****@*****.**'
self.prefs._Prefs__data['SMTP_SUBJECT'] = 'Denyhosts Mailinator'
self.prefs._Prefs__data['SMTP_DATE_FORMAT'] = '%Y%m%d'
self.prefs._Prefs__data['SMTP_HOST'] = 'localhost'
self.prefs._Prefs__data['SMTP_PORT'] = 2500
def test_is_true(self):
for string in self.true_strings:
self.assertTrue(util.is_true(string))
for string in self.false_strings:
self.assertFalse(util.is_true(string))
def test_is_false(self):
for string in self.true_strings:
self.assertFalse(util.is_false(string))
for string in self.false_strings:
self.assertTrue(util.is_false(string))
def test_parse_host(self):
for line, expected in HOST_TEST_DATA:
self.assertEqual(util.parse_host(line), expected)
def test_mail_command(self):
executable = str(sys.executable).replace(os.sep,
'/') # To support windows
self.assertEqual(
0, send_mail_by_command(executable, ["--version"], "", print))
self.assertNotEquals(
0,
send_mail_by_command(executable, ["--option-that-does-not-exist"],
"", print))
self.assertEquals(
42,
send_mail_by_command(
executable +
' -c "import sys; c = sys.stdin.read(); exit(int(c))"', [],
'42', print))
self.assertEquals(
3,
send_mail_by_command(
executable + ' -c "import sys; c = sys.argv[3]; exit(int(c))"',
["1", "2", "3"], "spam", print))
def test_valid_ip(self):
for ip, expected in TEST_IPS:
self.assertEqual(util.is_valid_ip_address(ip), expected)
def test_setup_logging_with_debug(self):
self.assertIsNone(util.setup_logging(self.prefs, True, False, False))
def test_setup_logging_no_debug(self):
self.assertIsNone(util.setup_logging(self.prefs, False, False, False))
def test_seconds(self):
self.assertEqual(util.calculate_seconds(1), 1)
def test_invalid_seconds_format(self):
with self.assertRaises(Exception) as cm:
util.calculate_seconds(datetime.now())
self.assertEqual(cm.exception, 'Error')
def test_seconds_zero_false(self):
with self.assertRaises(Exception) as cm:
util.calculate_seconds('0', False)
self.assertEqual(cm.exception, 'Error')
def test_send_email_success(self):
self.assertIsNone(util.send_email(self.prefs, 'testing report'))
def test_send_email_error(self):
self.prefs._Prefs__data['SMTP_PORT'] = 25
self.assertIsNone(util.send_email(self.prefs, 'testing report'))
def test_whitespace(self):
self.assertEqual(
util.normalize_whitespace('testing whitespace for denyhosts'),
'testing whitespace for denyhosts')
def setUp(self):
data_dir = ospj(dirname(abspath(__file__)), 'data')
self.prefs = Prefs()
self.prefs._Prefs__data['WORK_DIR'] = data_dir
self.prefs._Prefs__data['ALLOWED_HOSTS_HOSTNAME_LOOKUP'] = 'false'
self.allowed_hosts = AllowedHosts(self.prefs)
def setUp(self):
self.prefs = Prefs()
self.prefs._Prefs__data['WORK_DIR'] = ospj(dirname(__file__),
'data/purgecounter/static')
self.prefs._Prefs__data['PURGE_THRESHOLD'] = 1
def setUp(self):
self.prefs = Prefs()
self.prefs._Prefs__data['ETC_DIR'] = mkdtemp()
def setUp(self):
super(SyncTestBasic, self).setUp()
self.prefs = Prefs()
self.prefs._Prefs__data['SYNC_SERVER'] = LOCAL_SYNC_SERVER_URL
self.prefs._Prefs__data['WORK_DIR'] = ospj(dirname(__file__),
'data/sync/static')
def setUp(self):
self.prefs = Prefs()
self.prefs._Prefs__data['WORK_DIR'] = ospj(dirname(__file__),
'data/sync/dynamic')
self.sync = Sync(self.prefs)
self.value = randint(0, 1e9)
def setUp(self):
self.prefs = Prefs()
self.prefs._Prefs__data['WORK_DIR'] = ospj(dirname(__file__),
'data/sync/static')
self.sync = Sync(self.prefs)
if opt == '--purge-all':
purge_all = 1
if opt == '--purgeip':
purgeip_list.append(arg)
purgeip = 1
if opt == '--upgrade099':
upgrade099 = 1
if opt == '--version':
print("DenyHosts version:", VERSION)
sys.exit(0)
# This is generally expected to be in the environment, but there's no
# non-hackish way to get systemd to set it, so just hack it in here.
os.environ['HOSTNAME'] = platform.node()
prefs = Prefs(config_file)
first_time = 0
try:
if not os.path.exists(prefs.get('WORK_DIR')):
os.makedirs(prefs.get('WORK_DIR'))
first_time = 1
except Exception as e:
if e[0] != 17:
print(e)
sys.exit(1)
# On some operating systems the file /etc/hosts.deny (or
# whatever HOSTS_DENY is set to, may not exist. We should
# "touch" the file to make sure it is there to avoid errors later.
try:
def setUp(self):
self.prefs = Prefs()
self.prefs._Prefs__data['ETC_DIR'] = ospj(dirname(__file__),
'data/restricted')
self.test_hosts = set(['host1', 'host2'])
def setUp(self):
self.prefs = Prefs()
work_dir = mkdtemp()
self.filename = ospj(work_dir, PURGE_HISTORY)
self.prefs._Prefs__data['WORK_DIR'] = work_dir
self.hosts = set(['host1', 'host2'])
