mirror of
https://gitlab2.federez.net/re2o/re2o
synced 2024-09-10 08:13:12 +00:00
76 lines
3.1 KiB
Python
76 lines
3.1 KiB
Python
![]() |
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))
|
||
|
|