ruben/jail2ban
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Try the pfctl_table_op and set http code accordingly

Browse Source
This commit is contained in:
Ruben van Staveren 2023-01-15 18:45:28 +01:00
parent d43a574579
commit dd66bf3dc9
Signed by: ruben
GPG Key ID: 886F6BECD477A93F
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
jail2ban/__init__.py
View File

@ -88,10 +88,17 @@ def create_app():
name = untaint(PAT_NAME, name) name = untaint(PAT_NAME, name)
app.logger.info(f'Flushing table f2b-{name}' app.logger.info(f'Flushing table f2b-{name}'
f' in anchor f2b-jail/{remote_user}') f' in anchor f2b-jail/{remote_user}')
try:
res = pfctl_table_op('f2b-jail/{remote_user}', res = pfctl_table_op('f2b-jail/{remote_user}',
table='f2b-{name}', table='f2b-{name}',
operation='show', operation='show',
verbose=True) verbose=True)
except CalledProcessError as err:
if err.stderr.find(b'pfctl: Table does not exist.') > 0:
res = []
else:
raise err
result = [entry.groupdict() for entry in result = [entry.groupdict() for entry in
re.finditer(_PFCTL_TABLE_PAT, re.finditer(_PFCTL_TABLE_PAT,
'\n'.join([x.decode('ascii') for x in res]), '\n'.join([x.decode('ascii') for x in res]),
@ -100,7 +107,7 @@ def create_app():
return jsonify({'anchor': f'f2b-jail/{remote_user}', return jsonify({'anchor': f'f2b-jail/{remote_user}',
'table': f'f2b-{name}', 'table': f'f2b-{name}',
'operation': 'list', 'operation': 'list',
'result': result }) 'result': result }), 200 if len(res) else 404
@app.route("/register", methods=['PUT', 'DELETE']) @app.route("/register", methods=['PUT', 'DELETE'])
@auth.login_required @auth.login_required