|
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
# emacs: -*- mode: python; py-indent-offset: 4; indent-tabs-mode: t -*-
|
|
|
|
|
# vi: set ft=python sts=4 ts=4 sw=4 noet :
|
|
|
|
|
"""Script to run Fail2Ban tests battery
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
# This file is part of Fail2Ban.
|
|
|
|
|
#
|
|
|
|
|
# Fail2Ban is free software; you can redistribute it and/or modify
|
|
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
|
|
|
# (at your option) any later version.
|
|
|
|
|
#
|
|
|
|
|
# Fail2Ban is distributed in the hope that it will be useful,
|
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
|
#
|
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
|
# along with Fail2Ban; if not, write to the Free Software
|
|
|
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
|
|
|
|
|
|
__author__ = "Cyril Jaquier"
|
|
|
|
|
__copyright__ = "Copyright (c) 2004 Cyril Jaquier, 2012- Yaroslav Halchenko"
|
|
|
|
|
__license__ = "GPL"
|
|
|
|
|
|
|
|
|
|
import logging
|
|
|
|
|
import os
|
|
|
|
|
import sys
|
|
|
|
|
import time
|
|
|
|
|
import unittest
|
|
|
|
|
|
|
|
|
|
# Check if local fail2ban module exists, and use if it exists by
|
|
|
|
|
# modifying the path. This is done so that tests can be used in dev
|
|
|
|
|
# environment.
|
|
|
|
|
if os.path.exists("fail2ban/__init__.py"):
|
|
|
|
|
sys.path.insert(0, ".")
|
|
|
|
|
from fail2ban.version import version
|
|
|
|
|
|
|
|
|
|
from fail2ban.tests.utils import getOptParser, initProcess, gatherTests
|
|
|
|
|
from fail2ban.setup import updatePyExec
|
|
|
|
|
|
|
|
|
|
# Update fail2ban-python env to current python version (where f2b-modules located/installed)
|
|
|
|
|
bindir = os.path.dirname(
|
|
|
|
|
# __file__ seems to be overwritten sometimes on some python versions (e.g. bug of 2.6 by running under cProfile, etc.):
|
|
|
|
|
sys.argv[0] if os.path.basename(sys.argv[0]) == 'fail2ban-testcases' else __file__
|
|
|
|
|
)
|
|
|
|
|
updatePyExec(bindir)
|
|
|
|
|
|
|
|
|
|
(opts, regexps) = getOptParser(__doc__).parse_args()
|
|
|
|
|
|
|
|
|
|
#
|
|
|
|
|
# Process initialization corresponding options (logging, default options, etc.)
|
|
|
|
|
#
|
|
|
|
|
opts = initProcess(opts)
|
|
|
|
|
verbosity = opts.verbosity
|
|
|
|
|
|
|
|
|
|
#
|
|
|
|
|
# Gather tests (and filter corresponding options)
|
|
|
|
|
#
|
- performance of fail2ban optimized
-- cache dnsToIp, ipToName to prevent long wait during retrieving of ip/name for wrong dns or lazy dns-system;
-- instead of simple "sleep" used conditional wait "wait_for", that internal increases sleep interval up to sleeptime;
-- ticket / banmanager / failmanager modules are performance optimized;
-- api of filter (log files), jail, etc. rewritten and extended for performance purposes;
- performance of test cases optimized:
-- added option "--fast" to decrease wait intervals, avoid passive waiting, and skip few very slow test cases;
- code review after partially cherry pick of branch 'ban-time-incr' (see gh-716)
-- ticket module prepared to easy merge with newest version of 'ban-time-incr', now additionally holds banTime, banCount and json-data;
-- executeCmd partially moved from action to new module utils, etc.
-- python 2.6 compatibility;
- testExecuteTimeoutWithNastyChildren: test case repaired - wait for pid file inside bash, kill tree in any case (gh-1155);
- testSocket: test case repaired - wait for server thread starts a socket (listener)
10 years ago
|
|
|
tests = gatherTests(regexps, opts)
|
|
|
|
|
|
|
|
|
|
#
|
|
|
|
|
# Run the tests
|
|
|
|
|
#
|
|
|
|
|
testRunner = unittest.TextTestRunner(verbosity=verbosity)
|
|
|
|
|
|
|
|
|
|
tests_results = testRunner.run(tests)
|
|
|
|
|
|
|
|
|
|
if not tests_results.wasSuccessful(): # pragma: no cover
|
|
|
|
|
sys.exit(1)
|
