profxadke/captive.whump.shanti-portal
1
0
Fork 0
mirror of https://codeberg.org/prof_x_pvt_ltd/captive.whump.shanti-portal synced 2024-08-14 22:46:42 +00:00
Code Issues Projects Releases Packages Wiki Activity

implementing iptc part.

Browse source
This commit is contained in:
Stefan Midjich 2017-03-07 09:36:56 +01:00
parent 622827b4e8
commit 4cbbc54674
2 changed files with 27 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
tools/client.py
View file

@ -5,7 +5,7 @@ Handles "clients" in IPtables for captive portal.
from uuid import uuid4 from uuid import uuid4
from datetime import datetime from datetime import datetime
#import iptc import iptc
import errors import errors
@ -15,7 +15,7 @@ class Client(object):
def __init__(self, **kw): def __init__(self, **kw):
# Required parameters # Required parameters
self.storage = kw.pop('storage') self.storage = kw.pop('storage')
self.chain = kw.pop('chain') self._chain = kw.pop('chain')
# First try to get an existing client by ID # First try to get an existing client by ID
self.client_id = kw.pop('client_id', None) self.client_id = kw.pop('client_id', None)
@ -46,6 +46,10 @@ class Client(object):
self.last_packets = 0 self.last_packets = 0
self.last_activity = None self.last_activity = None
# Init iptables
self.table = iptc.Table(iptc.Table.MANGLE)
self.chain = iptc.Chain(table, self._chain)
def load_client(self, data): def load_client(self, data):
self.client_id = data.get('client_id') self.client_id = data.get('client_id')
@ -59,7 +63,7 @@ class Client(object):
def commit(self): def commit(self):
self.commit_client() self.commit_client()
#self.commit_rule() self.commit_rule()
def commit_client(self): def commit_client(self):
@ -69,28 +73,31 @@ class Client(object):
def delete(self): def delete(self):
#self.remove_rule() self.remove_rule()
self.storage.remove_client(self) self.storage.remove_client(self)
def find_rule(self): def remove_rule(self):
raise NotImplemented rule = self.find_rule(self.ip_address, self.protocol)
if rule:
self.chain.delete_rule(rule)
def find_rule(self, ip_address, protocol):
for rule in self.chain.rules:
src_ip = rule.src
if src_ip.startswith(ip_address) and rule.protocol == protocol:
return rule
else:
return None
def commit_rule(self): def commit_rule(self):
table = iptc.Table(iptc.Table.MANGLE) rule = find_rule(self.ip_address, self.protocol)
chain = iptc.Chain(table, self.chain) if not rule:
# Check if rule exists
for rule in chain.rules:
src_ip = rule.src
if src_ip.startswith(self.ip_address) and rule.protocol == self.protocol:
print('Rule exists')
break
else:
rule = iptc.Rule() rule = iptc.Rule()
rule.src = self.ip_address rule.src = self.ip_address
rule.protocol = self.protocol rule.protocol = self.protocol
rule.target = iptc.Target(rule, 'RETURN') rule.target = iptc.Target(rule, 'RETURN')
chain.insert_rule(rule) self.chain.insert_rule(rule)

3
tools/errors.py
View file

@ -1,2 +1,5 @@
class StorageNotFound(Exception): class StorageNotFound(Exception):
pass pass
class IPTCRuleExists(Exception):
pass