mirror of https://gitlab.federez.net/re2o/re2o
committed by
root
1 changed files with 75 additions and 0 deletions
@ -0,0 +1,75 @@ |
|||
import os, sys |
|||
|
|||
proj_path = "/var/www/re2o/" |
|||
# This is so Django knows where to find stuff. |
|||
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "re2o.settings") |
|||
sys.path.append(proj_path) |
|||
|
|||
# This is so my local_settings.py gets loaded. |
|||
os.chdir(proj_path) |
|||
|
|||
# This is so models get loaded. |
|||
from django.core.wsgi import get_wsgi_application |
|||
application = get_wsgi_application() |
|||
|
|||
import argparse |
|||
|
|||
from machines.models import Interface |
|||
from topologie.models import Room, Port, Switch |
|||
from users.models import User |
|||
|
|||
from re2o.settings import RADIUS_VLAN_DECISION |
|||
|
|||
VLAN_NOK = RADIUS_VLAN_DECISION['VLAN_NOK'] |
|||
VLAN_OK = RADIUS_VLAN_DECISION['VLAN_OK'] |
|||
|
|||
def decide_vlan(switch_name, port_number, mac_address): |
|||
# Get port from switch and port number |
|||
switch = Switch.objects.filter(switch_interface=Interface.objects.filter(dns=switch_name)) |
|||
if switch: |
|||
port = Port.objects.filter(switch=switch[0], port=port_number) |
|||
if port: |
|||
port = port[0] |
|||
if port.radius == 'NO': |
|||
# Aucune authentification sur ce port |
|||
decision = ("Pas d'authentification sur ce port", VLAN_OK) |
|||
elif port.radius == 'BLOQ': |
|||
# Prise désactivée |
|||
decision = ('Port desactive', VLAN_NOK) |
|||
elif port.radius == 'COMMON' or port.radius == 'STRICT': |
|||
# Authentification par mac |
|||
interface = Interface.objects.filter(mac_address=mac_address) |
|||
if not interface: |
|||
decision = ('Mac not found', VLAN_NOK) |
|||
elif interface[0].is_active(): |
|||
# Verification de la prise |
|||
if port.radius == 'STRICT': |
|||
if port.room: |
|||
user = User.objects.filter(room=Room.objects.filter(name=port.room)) |
|||
if not user: |
|||
decision = ('Chambre non cotisante', VLAN_NOK) |
|||
elif user[0].has_access(): |
|||
decision = ('Machine OK, Proprio OK', VLAN_OK) |
|||
else: |
|||
decision = ('Chambre inconnue', VLAN_NOK) |
|||
else: |
|||
# Mode COMMON |
|||
decision = ('Machine OK', VLAN_OK) |
|||
else: |
|||
decision = ('Machine non active / adherent non cotisant', VLAN_NOK) |
|||
else: |
|||
decision = ('VLAN forced', int(port.radius)) |
|||
else: |
|||
decision = ('port not found!', VLAN_OK) |
|||
else: |
|||
decision = ('switch not found!', VLAN_OK) |
|||
return decision |
|||
|
|||
if __name__ == '__main__': |
|||
parser = argparse.ArgumentParser(description='Decide radius vlan attribution') |
|||
parser.add_argument('switch_name', action="store") |
|||
parser.add_argument('port_number', action="store", type=int) |
|||
parser.add_argument('mac_address', action="store") |
|||
args = parser.parse_args() |
|||
print(decide_vlan(args.switch_name, args.port_number, args.mac_address)) |
|||
|
|||
Loading…
Reference in new issue