storage.cfg is not necessary, moving to plugins.cfg to save on files and confusion.

This commit is contained in:
Stefan Midjich 2017-11-16 22:17:40 +01:00
parent b23468ed28
commit 4c9f133f9a
3 changed files with 23 additions and 14 deletions

View File

@ -32,4 +32,16 @@ debug = True
# Simply prepend sudo here if you won't run rq worker as root
ipset_add_cmd = ipset -exist add authenticated-clients {client_ip}
ipset_name = authenticated-clients
ipset_name = authenticated-clients
# Dummy plugin to provide the client plugin with storage config while
# simultaneously providing the manage_client tool with its storage config.
[postgres]
hostname=localhost
username=captiveportal
password=secret.
database=captiveportal
port=5432
enabled = False
mandatory = False

View File

@ -2,6 +2,7 @@
# Python helper tool to add IPtables rule using the iptc library. This must
# of course run as root for iptc to work.
from os import getuid
from sys import exit
from argparse import ArgumentParser, FileType, ArgumentTypeError
from pprint import pprint as pp
@ -97,13 +98,18 @@ config.readfp(args.config)
sr = StoragePostgres(config=config)
if args.refresh:
if getuid() == 0:
use_sudo = False
else:
use_sudo = True
# Sync clients and packet counters from ipset into storage.
proc = run_ipset(
'list',
config.get('ipset', 'set_name'),
config.get('ipset', 'ipset_name'),
'-output',
'save',
use_sudo=False,
use_sudo=use_sudo,
timeout=600
)
@ -130,7 +136,7 @@ if args.refresh:
client = Client(
storage=sr,
ip_address=client_ip,
ipset_name=config.get('ipset', 'set_name')
ipset_name=config.get('ipset', 'ipset_name')
)
except Exception as e:
if args.verbose:
@ -173,7 +179,7 @@ for src_ip in args.src_ip:
client = Client(
storage=sr,
ip_address=src_ip,
ipset_name=config.get('ipset', 'set_name')
ipset_name=config.get('ipset', 'ipset_name')
)
if args.delete:

View File

@ -1,9 +0,0 @@
[postgres]
hostname=localhost
username=captiveportal
password=secret.
database=captiveportal
port=5432
[ipset]
set_name=authenticated-clients