mirror of https://github.com/fail2ban/fail2ban
introduces new command "fail2ban-python", as automatically created symlink to python executable, where fail2ban currently installed (resp. its modules are located);
fixed pythonic filters and test scripts (running via "fail2ban-python" now); fixed test case "testSetupInstallRoot" not for default python (also using direct call, out of virtualenv); # Conflicts: # config/filter.d/ignorecommands/apache-fakegooglebot # fail2ban/tests/files/config/apache-auth/digest.py # fail2ban/tests/files/ignorecommand.py # fail2ban/tests/misctestcase.pypull/1511/head
parent
9d70c49ea8
commit
38d53a72fd
|
@ -38,11 +38,14 @@ if os.path.exists("fail2ban/__init__.py"):
|
||||||
from fail2ban.version import version
|
from fail2ban.version import version
|
||||||
|
|
||||||
from fail2ban.tests.utils import gatherTests
|
from fail2ban.tests.utils import gatherTests
|
||||||
from fail2ban.helpers import FormatterWithTraceBack, getLogger
|
from fail2ban.helpers import updatePyExec, FormatterWithTraceBack, getLogger
|
||||||
from fail2ban.server.mytime import MyTime
|
from fail2ban.server.mytime import MyTime
|
||||||
|
|
||||||
from optparse import OptionParser, Option
|
from optparse import OptionParser, Option
|
||||||
|
|
||||||
|
# Update fail2ban-python env to current python version (where f2b-modules located/installed)
|
||||||
|
updatePyExec(os.path.dirname(__file__))
|
||||||
|
|
||||||
def get_opt_parser():
|
def get_opt_parser():
|
||||||
# use module docstring for help output
|
# use module docstring for help output
|
||||||
p = OptionParser(
|
p = OptionParser(
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
#!/usr/bin/python
|
#!/usr/bin/env fail2ban-python
|
||||||
# Inspired by https://isc.sans.edu/forums/diary/When+Google+isnt+Google/15968/
|
# Inspired by https://isc.sans.edu/forums/diary/When+Google+isnt+Google/15968/
|
||||||
#
|
#
|
||||||
# Written in Python to reuse built-in Python batteries and not depend on
|
# Written in Python to reuse built-in Python batteries and not depend on
|
||||||
|
|
|
@ -113,6 +113,24 @@ class FormatterWithTraceBack(logging.Formatter):
|
||||||
return logging.Formatter.format(self, record)
|
return logging.Formatter.format(self, record)
|
||||||
|
|
||||||
|
|
||||||
|
def updatePyExec(bindir, executable=None):
|
||||||
|
"""Update fail2ban-python link to current python version (where f2b-modules located/installed)
|
||||||
|
"""
|
||||||
|
bindir = os.path.realpath(bindir)
|
||||||
|
if executable is None:
|
||||||
|
executable = sys.executable
|
||||||
|
pypath = os.path.join(bindir, 'fail2ban-python')
|
||||||
|
# if not exists or point to another version - update link:
|
||||||
|
isfile = os.path.isfile(pypath)
|
||||||
|
if not isfile or os.path.realpath(pypath) != os.path.realpath(executable):
|
||||||
|
if isfile:
|
||||||
|
os.unlink(pypath)
|
||||||
|
os.symlink(executable, pypath)
|
||||||
|
# extend current environment path (e.g. if fail2ban not yet installed):
|
||||||
|
if bindir not in os.environ["PATH"].split(os.pathsep):
|
||||||
|
os.environ["PATH"] = os.environ["PATH"] + os.pathsep + bindir;
|
||||||
|
|
||||||
|
|
||||||
def getLogger(name):
|
def getLogger(name):
|
||||||
"""Get logging.Logger instance with Fail2Ban logger name convention
|
"""Get logging.Logger instance with Fail2Ban logger name convention
|
||||||
"""
|
"""
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
#!/usr/bin/python
|
#!/usr/bin/env fail2ban-python
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
#!/usr/bin/python
|
#!/usr/bin/env fail2ban-python
|
||||||
import sys
|
import sys
|
||||||
if sys.argv[1] == "10.0.0.1":
|
if sys.argv[1] == "10.0.0.1":
|
||||||
exit(0)
|
exit(0)
|
||||||
|
|
|
@ -70,6 +70,17 @@ class HelpersTest(unittest.TestCase):
|
||||||
self.assertEqual(splitwords(' 1\n 2, 3'), ['1', '2', '3'])
|
self.assertEqual(splitwords(' 1\n 2, 3'), ['1', '2', '3'])
|
||||||
|
|
||||||
|
|
||||||
|
def _getSysPythonVersion():
|
||||||
|
import subprocess, locale
|
||||||
|
sysVerCmd = "fail2ban-python -c 'import sys; print(tuple(sys.version_info))'"
|
||||||
|
if sys.version_info >= (2,7):
|
||||||
|
sysVer = subprocess.check_output(sysVerCmd, shell=True)
|
||||||
|
else:
|
||||||
|
sysVer = subprocess.Popen(sysVerCmd, shell=True, stdout=subprocess.PIPE).stdout.read()
|
||||||
|
if sys.version_info >= (3,):
|
||||||
|
sysVer = sysVer.decode(locale.getpreferredencoding(), 'replace')
|
||||||
|
return str(sysVer).rstrip()
|
||||||
|
|
||||||
class SetupTest(unittest.TestCase):
|
class SetupTest(unittest.TestCase):
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
|
@ -79,6 +90,12 @@ class SetupTest(unittest.TestCase):
|
||||||
raise unittest.SkipTest(
|
raise unittest.SkipTest(
|
||||||
"Seems to be running not out of source distribution"
|
"Seems to be running not out of source distribution"
|
||||||
" -- cannot locate setup.py")
|
" -- cannot locate setup.py")
|
||||||
|
# compare current version of python installed resp. active one:
|
||||||
|
sysVer = _getSysPythonVersion()
|
||||||
|
if sysVer != str(tuple(sys.version_info)):
|
||||||
|
raise unittest.SkipTest(
|
||||||
|
"Seems to be running with python distribution %s"
|
||||||
|
" -- install can be tested only with system distribution %s" % (str(tuple(sys.version_info)), sysVer))
|
||||||
|
|
||||||
def testSetupInstallRoot(self):
|
def testSetupInstallRoot(self):
|
||||||
if not self.setup:
|
if not self.setup:
|
||||||
|
@ -125,6 +142,10 @@ class SetupTest(unittest.TestCase):
|
||||||
'etc/fail2ban/jail.conf'):
|
'etc/fail2ban/jail.conf'):
|
||||||
self.assertTrue(os.path.exists(os.path.join(tmp, f)),
|
self.assertTrue(os.path.exists(os.path.join(tmp, f)),
|
||||||
msg="Can't find %s" % f)
|
msg="Can't find %s" % f)
|
||||||
|
self.assertEqual(
|
||||||
|
os.path.realpath(os.path.join(tmp, 'usr/local/bin/fail2ban-python')),
|
||||||
|
os.path.realpath(sys.executable))
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
# clean up
|
# clean up
|
||||||
shutil.rmtree(tmp)
|
shutil.rmtree(tmp)
|
||||||
|
|
40
setup.py
40
setup.py
|
@ -40,12 +40,46 @@ except ImportError:
|
||||||
# python 2.x
|
# python 2.x
|
||||||
from distutils.command.build_py import build_py
|
from distutils.command.build_py import build_py
|
||||||
from distutils.command.build_scripts import build_scripts
|
from distutils.command.build_scripts import build_scripts
|
||||||
|
# all versions
|
||||||
|
from distutils.command.install_scripts import install_scripts
|
||||||
|
|
||||||
import os
|
import os
|
||||||
from os.path import isfile, join, isdir, realpath
|
from os.path import isfile, join, isdir, realpath
|
||||||
import sys
|
import sys
|
||||||
import warnings
|
import warnings
|
||||||
from glob import glob
|
from glob import glob
|
||||||
|
|
||||||
|
|
||||||
|
def updatePyExec(bindir, executable=None):
|
||||||
|
"""Update fail2ban-python link to current python version (where f2b-modules located/installed)
|
||||||
|
"""
|
||||||
|
bindir = os.path.realpath(bindir)
|
||||||
|
if executable is None:
|
||||||
|
executable = sys.executable
|
||||||
|
pypath = os.path.join(bindir, 'fail2ban-python')
|
||||||
|
# if not exists or point to another version - update link:
|
||||||
|
isfile = os.path.isfile(pypath)
|
||||||
|
if not isfile or os.path.realpath(pypath) != os.path.realpath(executable):
|
||||||
|
if isfile:
|
||||||
|
os.unlink(pypath)
|
||||||
|
os.symlink(executable, pypath)
|
||||||
|
|
||||||
|
|
||||||
|
# Wrapper to install python binding (to current python version):
|
||||||
|
class install_scripts_f2b(install_scripts):
|
||||||
|
|
||||||
|
def get_outputs(self):
|
||||||
|
outputs = install_scripts.get_outputs(self)
|
||||||
|
fn = None
|
||||||
|
for fn in outputs:
|
||||||
|
if os.path.basename(fn) == 'fail2ban-server':
|
||||||
|
break
|
||||||
|
bindir = os.path.dirname(fn)
|
||||||
|
print('creating fail2ban-python binding -> %s' % (bindir,))
|
||||||
|
updatePyExec(bindir)
|
||||||
|
return outputs
|
||||||
|
|
||||||
|
|
||||||
if setuptools and "test" in sys.argv:
|
if setuptools and "test" in sys.argv:
|
||||||
import logging
|
import logging
|
||||||
logSys = logging.getLogger("fail2ban")
|
logSys = logging.getLogger("fail2ban")
|
||||||
|
@ -113,12 +147,16 @@ setup(
|
||||||
url = "http://www.fail2ban.org",
|
url = "http://www.fail2ban.org",
|
||||||
license = "GPL",
|
license = "GPL",
|
||||||
platforms = "Posix",
|
platforms = "Posix",
|
||||||
cmdclass = {'build_py': build_py, 'build_scripts': build_scripts},
|
cmdclass = {
|
||||||
|
'build_py': build_py, 'build_scripts': build_scripts,
|
||||||
|
'install_scripts': install_scripts_f2b
|
||||||
|
},
|
||||||
scripts = [
|
scripts = [
|
||||||
'bin/fail2ban-client',
|
'bin/fail2ban-client',
|
||||||
'bin/fail2ban-server',
|
'bin/fail2ban-server',
|
||||||
'bin/fail2ban-regex',
|
'bin/fail2ban-regex',
|
||||||
'bin/fail2ban-testcases',
|
'bin/fail2ban-testcases',
|
||||||
|
# 'bin/fail2ban-python', -- link (binary), will be installed via install_scripts_f2b wrapper
|
||||||
],
|
],
|
||||||
packages = [
|
packages = [
|
||||||
'fail2ban',
|
'fail2ban',
|
||||||
|
|
Loading…
Reference in New Issue