mirror of
https://gitlab.federez.net/re2o/re2o
synced 2024-06-13 05:58:25 +00:00
102 lines
3.8 KiB
Python
Executable file
102 lines
3.8 KiB
Python
Executable file
# Re2o est un logiciel d'administration développé initiallement au rezometz. Il
|
|
# se veut agnostique au réseau considéré, de manière à être installable en
|
|
# quelques clics.
|
|
#
|
|
# Copyright © 2017 Gabriel Détraz
|
|
# Copyright © 2017 Goulven Kermarec
|
|
# Copyright © 2017 Augustin Lemesle
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program; if not, write to the Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
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 django.db.models import Q
|
|
from machines.models import Interface, IpList, Domain
|
|
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_id, port_number, mac_address):
|
|
# Get port from switch and port number
|
|
switch = Switch.objects.filter(switch_interface=Interface.objects.filter(Q(ipv4=IpList.objects.filter(ipv4=switch_id)) | Q(domain=Domain.objects.filter(name=switch_id))))
|
|
if not switch:
|
|
return ('?', 'Switch inconnu', VLAN_OK)
|
|
|
|
sw_name = str(switch[0].switch_interface)
|
|
|
|
port = Port.objects.filter(switch=switch[0], port=port_number)
|
|
if not port:
|
|
return (sw_name, 'Port inconnu', VLAN_OK)
|
|
|
|
port = port[0]
|
|
|
|
if port.radius == 'NO':
|
|
return (sw_name, "Pas d'authentification sur ce port", VLAN_OK)
|
|
|
|
if port.radius == 'BLOQ':
|
|
return (sw_name, 'Port desactive', VLAN_NOK)
|
|
|
|
if port.radius == 'STRICT':
|
|
if not port.room:
|
|
return (sw_name, 'Chambre inconnue', VLAN_NOK)
|
|
|
|
room_user = User.objects.filter(room=Room.objects.filter(name=port.room))
|
|
if not room_user:
|
|
return (sw_name, 'Chambre non cotisante', VLAN_NOK)
|
|
elif not room_user[0].has_access:
|
|
return (sw_name, 'Chambre resident desactive', VLAN_NOK)
|
|
# else: user OK, on passe à la verif MAC
|
|
|
|
if port.radius == 'COMMON' or port.radius == 'STRICT':
|
|
# Authentification par mac
|
|
interface = Interface.objects.filter(mac_address=mac_address)
|
|
if not interface:
|
|
return (sw_name, 'Machine inconnue', VLAN_NOK)
|
|
elif not interface[0].is_active:
|
|
return (sw_name, 'Machine non active / adherent non cotisant', VLAN_NOK)
|
|
else:
|
|
return (sw_name, 'Machine OK', VLAN_OK)
|
|
|
|
# On gere bien tous les autres états possibles, il ne reste que le VLAN en dur
|
|
return (sw_name, 'VLAN impose', int(port.radius))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Decide radius vlan attribution')
|
|
parser.add_argument('switch_id', 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_id, args.port_number, args.mac_address))
|
|
|