models.py 70.6 KB
Newer Older
1
# -*- mode: python; coding: utf-8 -*-
2 3 4 5
# 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.
#
6
# Copyright © 2016-2018  Gabriel Détraz
7 8
# Copyright © 2017  Goulven Kermarec
# Copyright © 2017  Augustin Lemesle
9
# Copyright © 2018  Charlie Jacomme
10 11 12 13 14 15 16 17 18 19 20 21 22 23
#
# 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.
24 25 26
"""machines.models
The models definitions for the Machines app
"""
27

28 29
from __future__ import unicode_literals

Gabriel Detraz's avatar
Gabriel Detraz committed
30 31
from datetime import timedelta
import re
32
from ipaddress import IPv6Address
33
from itertools import chain
34
from netaddr import mac_bare, EUI, IPSet, IPRange, IPNetwork, IPAddress
35 36
import hashlib
import base64
Gabriel Detraz's avatar
Gabriel Detraz committed
37

38
from django.db import models
39
from django.db.models import Q
Gabriel Detraz's avatar
Gabriel Detraz committed
40
from django.db.models.signals import post_save, post_delete
41
from django.dispatch import receiver
42
from django.forms import ValidationError
43
from django.utils.functional import cached_property
44
from django.utils import timezone
45
from django.utils.translation import ugettext_lazy as _l
46
from django.core.validators import MaxValueValidator, MinValueValidator
Gabriel Detraz's avatar
Gabriel Detraz committed
47

48
from macaddress.fields import MACAddressField
49

50
from re2o.field_permissions import FieldPermissionModelMixin
51
from re2o.mixins import AclMixin, RevMixin
52

Maël Kervella's avatar
Maël Kervella committed
53 54 55
import users.models
import preferences.models

56

57
class Machine(RevMixin, FieldPermissionModelMixin, models.Model):
Gabriel Detraz's avatar
Gabriel Detraz committed
58 59
    """ Class définissant une machine, object parent user, objets fils
    interfaces"""
60
    PRETTY_NAME = "Machine"
Gabriel Detraz's avatar
Gabriel Detraz committed
61

62
    user = models.ForeignKey('users.User', on_delete=models.PROTECT)
Gabriel Detraz's avatar
Gabriel Detraz committed
63 64 65 66 67 68
    name = models.CharField(
        max_length=255,
        help_text="Optionnel",
        blank=True,
        null=True
    )
69
    active = models.BooleanField(default=True)
70

71 72 73
    class Meta:
        permissions = (
            ("view_machine", "Peut voir un objet machine quelquonque"),
74 75
            ("change_machine_user",
             "Peut changer le propriétaire d'une machine"),
76 77
        )

78 79
    @classmethod
    def get_instance(cls, machineid, *_args, **_kwargs):
80 81 82 83
        """Get the Machine instance with machineid.
        :param userid: The id
        :return: The user
        """
84
        return cls.objects.get(pk=machineid)
85

86 87 88
    def linked_objects(self):
        """Return linked objects : machine and domain.
        Usefull in history display"""
89 90 91 92 93 94
        return chain(
            self.interface_set.all(),
            Domain.objects.filter(
                interface_parent__in=self.interface_set.all()
            )
        )
95

96
    @staticmethod
97
    def can_change_user(user_request, *_args, **_kwargs):
98 99 100 101 102 103 104 105 106 107
        """Checks if an user is allowed to change the user who owns a
        Machine.

        Args:
            user_request: The user requesting to change owner.

        Returns:
            A tuple with a boolean stating if edition is allowed and an
            explanation message.
        """
108 109
        return (user_request.has_perm('machines.change_machine_user'),
                "Vous ne pouvez pas modifier l'utilisateur de la machine.")
110

111 112
    @staticmethod
    def can_view_all(user_request, *_args, **_kwargs):
113 114 115 116 117
        """Vérifie qu'on peut bien afficher l'ensemble des machines,
        droit particulier correspondant
        :param user_request: instance user qui fait l'edition
        :return: True ou False avec la raison de l'échec le cas échéant"""
        if not user_request.has_perm('machines.view_machine'):
118 119
            return False, (u"Vous ne pouvez pas afficher l'ensemble des "
                           "machines sans permission")
120 121
        return True, None

122 123
    @staticmethod
    def can_create(user_request, userid, *_args, **_kwargs):
124 125 126 127 128
        """Vérifie qu'un user qui fait la requète peut bien créer la machine
        et n'a pas atteint son quota, et crée bien une machine à lui
        :param user_request: Utilisateur qui fait la requête
        :param userid: id de l'user dont on va créer une machine
        :return: soit True, soit False avec la raison de l'échec"""
Maël Kervella's avatar
Maël Kervella committed
129
        try:
130
            user = users.models.User.objects.get(pk=userid)
Maël Kervella's avatar
Maël Kervella committed
131 132
        except users.models.User.DoesNotExist:
            return False, u"Utilisateur inexistant"
133 134 135 136
        max_lambdauser_interfaces = (preferences.models.OptionalMachine
                                     .get_cached_value(
                                         'max_lambdauser_interfaces'
                                     ))
137
        if not user_request.has_perm('machines.add_machine'):
138 139
            if not (preferences.models.OptionalMachine
                    .get_cached_value('create_machine')):
140
                return False, u"Vous ne pouvez pas ajouter une machine"
Maël Kervella's avatar
Maël Kervella committed
141
            if user != user_request:
142 143
                return False, (u"Vous ne pouvez pas ajouter une machine à un "
                               "autre user que vous sans droit")
Maël Kervella's avatar
Maël Kervella committed
144
            if user.user_interfaces().count() >= max_lambdauser_interfaces:
145 146 147
                return False, (u"Vous avez atteint le maximum d'interfaces "
                               "autorisées que vous pouvez créer vous même "
                               "(%s) " % max_lambdauser_interfaces)
Maël Kervella's avatar
Maël Kervella committed
148 149
        return True, None

150
    def can_edit(self, user_request, *args, **kwargs):
151 152 153 154 155
        """Vérifie qu'on peut bien éditer cette instance particulière (soit
        machine de soi, soit droit particulier
        :param self: instance machine à éditer
        :param user_request: instance user qui fait l'edition
        :return: True ou False avec la raison le cas échéant"""
156
        if self.user != user_request:
157 158 159 160 161 162
            if (not user_request.has_perm('machines.change_interface') or
                    not self.user.can_edit(
                        self.user,
                        user_request,
                        *args,
                        **kwargs
163
            )[0]):
164 165
                return False, (u"Vous ne pouvez pas éditer une machine "
                               "d'un autre user que vous sans droit")
166 167
        return True, None

168
    def can_delete(self, user_request, *args, **kwargs):
169 170 171 172 173
        """Vérifie qu'on peut bien supprimer cette instance particulière (soit
        machine de soi, soit droit particulier
        :param self: instance machine à supprimer
        :param user_request: instance user qui fait l'edition
        :return: True ou False avec la raison de l'échec le cas échéant"""
174
        if self.user != user_request:
175 176 177 178 179 180
            if (not user_request.has_perm('machines.change_interface') or
                    not self.user.can_edit(
                        self.user,
                        user_request,
                        *args,
                        **kwargs
181
            )[0]):
182 183
                return False, (u"Vous ne pouvez pas éditer une machine "
                               "d'un autre user que vous sans droit")
184 185
        return True, None

186
    def can_view(self, user_request, *_args, **_kwargs):
187 188 189 190 191
        """Vérifie qu'on peut bien voir cette instance particulière (soit
        machine de soi, soit droit particulier
        :param self: instance machine à éditer
        :param user_request: instance user qui fait l'edition
        :return: True ou False avec la raison de l'échec le cas échéant"""
192 193 194 195
        if (not user_request.has_perm('machines.view_machine') and
                self.user != user_request):
            return False, (u"Vous n'avez pas droit de voir les machines autre "
                           "que les vôtres")
196
        return True, None
197

198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218
    @cached_property
    def short_name(self):
        """Par defaut, renvoie le nom de la première interface
        de cette machine"""
        return str(self.interface_set.first().domain.name)

    @cached_property
    def all_short_names(self):
        """Renvoie de manière unique, le nom des interfaces de cette
        machine"""
        return Domain.objects.filter(
            interface_parent__machine=self
            ).values_list('name', flat=True).distinct()

    @cached_property
    def all_complete_names(self):
        """Renvoie tous les tls complets de la machine"""
        return [str(domain) for domain in Domain.objects.filter(
            Q(cname__interface_parent__machine=self) | Q(interface_parent__machine=self)
            )]

219 220 221
    def __init__(self, *args, **kwargs):
        super(Machine, self).__init__(*args, **kwargs)
        self.field_permissions = {
222
            'user': self.can_change_user,
223 224
        }

225
    def __str__(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
226 227
        return str(self.user) + ' - ' + str(self.id) + ' - ' + str(self.name)

228

229
class MachineType(RevMixin, AclMixin, models.Model):
230
    """ Type de machine, relié à un type d'ip, affecté aux interfaces"""
231 232
    PRETTY_NAME = "Type de machine"

233
    type = models.CharField(max_length=255)
Gabriel Detraz's avatar
Gabriel Detraz committed
234 235 236 237 238 239
    ip_type = models.ForeignKey(
        'IpType',
        on_delete=models.PROTECT,
        blank=True,
        null=True
    )
240

241 242 243
    class Meta:
        permissions = (
            ("view_machinetype", "Peut voir un objet machinetype"),
244 245
            ("use_all_machinetype",
             "Peut utiliser n'importe quel type de machine"),
246 247
        )

248
    def all_interfaces(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
249 250
        """ Renvoie toutes les interfaces (cartes réseaux) de type
        machinetype"""
251 252
        return Interface.objects.filter(type=self)

253 254
    @staticmethod
    def can_use_all(user_request, *_args, **_kwargs):
255 256 257 258 259 260 261 262
        """Check if an user can use every MachineType.

        Args:
            user_request: The user requesting edition.
        Returns:
            A tuple with a boolean stating if user can acces and an explanation
            message is acces is not allowed.
        """
263
        if not user_request.has_perm('machines.use_all_machinetype'):
264 265
            return False, (u"Vous n'avez pas le droit d'utiliser tout types "
                           "de machines")
266 267
        return True, None

268
    def __str__(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
269 270
        return self.type

271

272
class IpType(RevMixin, AclMixin, models.Model):
273
    """ Type d'ip, définissant un range d'ip, affecté aux machine types"""
274 275
    PRETTY_NAME = "Type d'ip"

276
    type = models.CharField(max_length=255)
277
    extension = models.ForeignKey('Extension', on_delete=models.PROTECT)
278
    need_infra = models.BooleanField(default=False)
279 280
    domaine_ip_start = models.GenericIPAddressField(protocol='IPv4')
    domaine_ip_stop = models.GenericIPAddressField(protocol='IPv4')
281 282 283 284 285 286 287 288 289 290 291 292 293 294
    domaine_ip_network = models.GenericIPAddressField(
        protocol='IPv4',
        null=True,
        blank=True,
        help_text="Network containing the ipv4 range domain ip start/stop. Optional"
    )
    domaine_ip_netmask = models.IntegerField(
        default=24,
        validators=[
            MaxValueValidator(31),
            MinValueValidator(8)
        ],
        help_text="Netmask for the ipv4 range domain"
    )
295
    reverse_v4 = models.BooleanField(
296
            default=False, 
297
            help_text="Enable reverse DNS for IPv4",
298
    )
Gabriel Detraz's avatar
Gabriel Detraz committed
299 300 301 302 303
    prefix_v6 = models.GenericIPAddressField(
        protocol='IPv6',
        null=True,
        blank=True
    )
304 305 306 307 308 309 310
    prefix_v6_length = models.IntegerField(
        default=64,
        validators=[
            MaxValueValidator(128),
            MinValueValidator(0)
        ]
        )
311
    reverse_v6 = models.BooleanField(
312
            default=False,
313
            help_text="Enable reverse DNS for IPv6",
314
    )
Gabriel Detraz's avatar
Gabriel Detraz committed
315 316 317 318 319 320 321 322 323 324 325
    vlan = models.ForeignKey(
        'Vlan',
        on_delete=models.PROTECT,
        blank=True,
        null=True
    )
    ouverture_ports = models.ForeignKey(
        'OuverturePortList',
        blank=True,
        null=True
    )
326

327 328 329 330 331 332
    class Meta:
        permissions = (
            ("view_iptype", "Peut voir un objet iptype"),
            ("use_all_iptype", "Peut utiliser tous les iptype"),
        )

333
    @cached_property
334
    def ip_range(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
335 336
        """ Renvoie un objet IPRange à partir de l'objet IpType"""
        return IPRange(self.domaine_ip_start, end=self.domaine_ip_stop)
337 338 339

    @cached_property
    def ip_set(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
340
        """ Renvoie une IPSet à partir de l'iptype"""
341
        return IPSet(self.ip_range)
342 343 344

    @cached_property
    def ip_set_as_str(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
345
        """ Renvoie une liste des ip en string"""
346 347
        return [str(x) for x in self.ip_set]

348 349 350 351 352
    @cached_property
    def ip_set_cidrs_as_str(self):
        """Renvoie la liste des cidrs du range en str"""
        return [str(ip_range) for ip_range in self.ip_set.iter_cidrs()]

353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379
    @cached_property
    def ip_set_full_info(self):
        """Iter sur les range cidr, et renvoie network, broacast , etc"""
        return [
            {
                'network': str(ip_set.network),
                'netmask': str(ip_set.netmask),
                'netmask_cidr': str(ip_set.prefixlen),
                'broadcast': str(ip_set.broadcast),
                'vlan': str(self.vlan),
                'vlan_id': self.vlan.vlan_id
            } for ip_set in self.ip_set.iter_cidrs()
        ]

    @cached_property
    def ip6_set_full_info(self):
        if self.prefix_v6:
            return {
                'network' : str(self.prefix_v6),
                'netmask' : 'ffff:ffff:ffff:ffff::',
                'netmask_cidr' : str(self.prefix_v6_length),
                'vlan': str(self.vlan),
                'vlan_id': self.vlan.vlan_id
            }
        else:
            return None

380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397
    @cached_property
    def ip_network(self):
        """Renvoie le network parent du range start-stop, si spécifié
        Différent de ip_set_cidrs ou iP_set, car lui est supérieur ou égal"""
        if self.domaine_ip_network:
            return IPNetwork(str(self.domaine_ip_network) + '/' + str(self.domaine_ip_netmask))
        return None

    @cached_property
    def ip_net_full_info(self):
        """Renvoie les infos du network contenant du range"""
        return {
            'network' : str(self.ip_network.network),
            'netmask' : str(self.ip_network.netmask),
            'broadcast' : str(self.ip_network.broadcast),
            'netmask_cidr' : str(self.ip_network.prefixlen),
        }

398 399 400 401 402
    @cached_property
    def complete_prefixv6(self):
        """Return the complete prefix v6 as cidr"""
        return str(self.prefix_v6) + "/" + str(self.prefix_v6_length)

403
    def ip_objects(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
404
        """ Renvoie tous les objets ipv4 relié à ce type"""
405 406 407
        return IpList.objects.filter(ip_type=self)

    def free_ip(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
408
        """ Renvoie toutes les ip libres associées au type donné (self)"""
Gabriel Detraz's avatar
Gabriel Detraz committed
409 410 411
        return IpList.objects.filter(
            interface__isnull=True
        ).filter(ip_type=self)
412 413

    def gen_ip_range(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
414 415 416
        """ Cree les IpList associées au type self. Parcours pédestrement et
        crée les ip une par une. Si elles existent déjà, met à jour le type
        associé à l'ip"""
417
        # Creation du range d'ip dans les objets iplist
418
        ip_obj = [IpList(ip_type=self, ipv4=str(ip)) for ip in self.ip_range]
Gabriel Detraz's avatar
Gabriel Detraz committed
419
        listes_ip = IpList.objects.filter(
420
            ipv4__in=[str(ip) for ip in self.ip_range]
Gabriel Detraz's avatar
Gabriel Detraz committed
421
        )
422 423 424 425 426 427
        # Si il n'y a pas d'ip, on les crée
        if not listes_ip:
            IpList.objects.bulk_create(ip_obj)
        # Sinon on update l'ip_type
        else:
            listes_ip.update(ip_type=self)
428
        return
429 430

    def del_ip_range(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
431 432
        """ Methode dépréciée, IpList est en mode cascade et supprimé
        automatiquement"""
433
        if Interface.objects.filter(ipv4__in=self.ip_objects()):
Gabriel Detraz's avatar
Gabriel Detraz committed
434 435
            raise ValidationError("Une ou plusieurs ip du range sont\
            affectées, impossible de supprimer le range")
436 437 438
        for ip in self.ip_objects():
            ip.delete()

439 440 441 442 443
    def check_replace_prefixv6(self):
        """Remplace les prefixv6 des interfaces liées à ce type d'ip"""
        if not self.prefix_v6:
            return
        else:
444
            for ipv6 in Ipv6List.objects.filter(
445 446 447 448
                interface__in=Interface.objects.filter(
                    type__in=MachineType.objects.filter(ip_type=self)
                )
            ):
449 450
                ipv6.check_and_replace_prefix(prefix=self.prefix_v6)

451 452
    def get_associated_ptr_records(self):
        from re2o.utils import all_active_assigned_interfaces
453 454 455 456 457 458
        if self.reverse_v4:
            return (all_active_assigned_interfaces()
                    .filter(type__ip_type=self)
                    .filter(ipv4__isnull=False))
        else:
            return None
459 460 461

    def get_associated_ptr_v6_records(self):
        from re2o.utils import all_active_interfaces
462 463 464 465 466
        if self.reverse_v6:
            return (all_active_interfaces(full=True)
                    .filter(type__ip_type=self))
        else:
            return None
467

468
    def clean(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
469 470 471 472 473
        """ Nettoyage. Vérifie :
        - Que ip_stop est après ip_start
        - Qu'on ne crée pas plus gros qu'un /16
        - Que le range crée ne recoupe pas un range existant
        - Formate l'ipv6 donnée en /64"""
474 475 476 477
        if IPAddress(self.domaine_ip_start) > IPAddress(self.domaine_ip_stop):
            raise ValidationError("Domaine end doit être après start...")
        # On ne crée pas plus grand qu'un /16
        if self.ip_range.size > 65536:
Gabriel Detraz's avatar
Gabriel Detraz committed
478 479
            raise ValidationError("Le range est trop gros, vous ne devez\
            pas créer plus grand qu'un /16")
480
        # On check que les / ne se recoupent pas
481
        for element in IpType.objects.all().exclude(pk=self.pk):
482
            if not self.ip_set.isdisjoint(element.ip_set):
Gabriel Detraz's avatar
Gabriel Detraz committed
483 484
                raise ValidationError("Le range indiqué n'est pas disjoint\
                des ranges existants")
485 486 487
        # On formate le prefix v6
        if self.prefix_v6:
            self.prefix_v6 = str(IPNetwork(self.prefix_v6 + '/64').network)
488 489 490 491 492
        # On vérifie qu'un domaine network/netmask contiens bien le domaine ip start-stop
        if self.domaine_ip_network:
            if not self.domaine_ip_start in self.ip_network or not self.domaine_ip_stop in self.ip_network:
                raise ValidationError("If you specify a domaine ip network/netmask, it\
                    must contain domaine ipstart-stop range")
493 494 495 496 497 498
        return

    def save(self, *args, **kwargs):
        self.clean()
        super(IpType, self).save(*args, **kwargs)

499 500
    @staticmethod
    def can_use_all(user_request, *_args, **_kwargs):
501 502
        """Superdroit qui permet d'utiliser toutes les extensions sans
        restrictions
503 504
        :param user_request: instance user qui fait l'edition
        :return: True ou False avec la raison de l'échec le cas échéant"""
505
        return user_request.has_perm('machines.use_all_iptype'), None
506

507 508
    def __str__(self):
        return self.type
chirac's avatar
chirac committed
509

Gabriel Detraz's avatar
Gabriel Detraz committed
510

511
class Vlan(RevMixin, AclMixin, models.Model):
512 513
    """ Un vlan : vlan_id et nom
    On limite le vlan id entre 0 et 4096, comme défini par la norme"""
514 515
    PRETTY_NAME = "Vlans"

516
    vlan_id = models.PositiveIntegerField(validators=[MaxValueValidator(4095)])
517 518 519
    name = models.CharField(max_length=256)
    comment = models.CharField(max_length=256, blank=True)

520 521 522 523 524
    class Meta:
        permissions = (
            ("view_vlan", "Peut voir un objet vlan"),
        )

525 526 527
    def __str__(self):
        return self.name

Gabriel Detraz's avatar
Gabriel Detraz committed
528

529
class Nas(RevMixin, AclMixin, models.Model):
Gabriel Detraz's avatar
Gabriel Detraz committed
530
    """ Les nas. Associé à un machine_type.
Gabriel Detraz's avatar
Gabriel Detraz committed
531 532
    Permet aussi de régler le port_access_mode (802.1X ou mac-address) pour
    le radius. Champ autocapture de la mac à true ou false"""
533 534
    PRETTY_NAME = "Correspondance entre les nas et les machines connectées"

535 536 537 538 539 540
    default_mode = '802.1X'
    AUTH = (
        ('802.1X', '802.1X'),
        ('Mac-address', 'Mac-address'),
    )

541
    name = models.CharField(max_length=255, unique=True)
Gabriel Detraz's avatar
Gabriel Detraz committed
542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
    nas_type = models.ForeignKey(
        'MachineType',
        on_delete=models.PROTECT,
        related_name='nas_type'
    )
    machine_type = models.ForeignKey(
        'MachineType',
        on_delete=models.PROTECT,
        related_name='machinetype_on_nas'
    )
    port_access_mode = models.CharField(
        choices=AUTH,
        default=default_mode,
        max_length=32
    )
557
    autocapture_mac = models.BooleanField(default=False)
558

559 560 561 562 563
    class Meta:
        permissions = (
            ("view_nas", "Peut voir un objet Nas"),
        )

564 565 566
    def __str__(self):
        return self.name

Gabriel Detraz's avatar
Gabriel Detraz committed
567

568
class SOA(RevMixin, AclMixin, models.Model):
569 570 571 572 573 574 575
    """
    Un enregistrement SOA associé à une extension
    Les valeurs par défault viennent des recommandations RIPE :
    https://www.ripe.net/publications/docs/ripe-203
    """
    PRETTY_NAME = "Enregistrement SOA"

576
    name = models.CharField(max_length=255)
577 578 579 580
    mail = models.EmailField(
        help_text='Email du contact pour la zone'
    )
    refresh = models.PositiveIntegerField(
581
        default=86400,  # 24 hours
582 583 584 585
        help_text='Secondes avant que les DNS secondaires doivent demander le\
                   serial du DNS primaire pour détecter une modification'
    )
    retry = models.PositiveIntegerField(
586
        default=7200,  # 2 hours
587 588 589 590
        help_text='Secondes avant que les DNS secondaires fassent une nouvelle\
                   demande de serial en cas de timeout du DNS primaire'
    )
    expire = models.PositiveIntegerField(
591
        default=3600000,  # 1000 hours
592 593 594 595 596 597 598 599
        help_text='Secondes après lesquelles les DNS secondaires arrêtent de\
                   de répondre aux requêtes en cas de timeout du DNS primaire'
    )
    ttl = models.PositiveIntegerField(
        default=172800,  # 2 days
        help_text='Time To Live'
    )

600 601 602 603 604
    class Meta:
        permissions = (
            ("view_soa", "Peut voir un objet soa"),
        )

605 606 607 608 609 610 611 612 613 614 615 616 617
    def __str__(self):
        return str(self.name)

    @cached_property
    def dns_soa_param(self):
        """
        Renvoie la partie de l'enregistrement SOA correspondant aux champs :
            <refresh>   ; refresh
            <retry>     ; retry
            <expire>    ; expire
            <ttl>       ; TTL
        """
        return (
618 619 620 621
            '    {refresh}; refresh\n'
            '    {retry}; retry\n'
            '    {expire}; expire\n'
            '    {ttl}; TTL'
622
        ).format(
623 624 625 626
            refresh=str(self.refresh).ljust(12),
            retry=str(self.retry).ljust(12),
            expire=str(self.expire).ljust(12),
            ttl=str(self.ttl).ljust(12)
627 628 629 630 631 632
        )

    @cached_property
    def dns_soa_mail(self):
        """ Renvoie le mail dans l'enregistrement SOA """
        mail_fields = str(self.mail).split('@')
633
        return mail_fields[0].replace('.', '\\.') + '.' + mail_fields[1] + '.'
634 635 636 637 638 639 640

    @classmethod
    def new_default_soa(cls):
        """ Fonction pour créer un SOA par défaut, utile pour les nouvelles
        extensions .
        /!\ Ne jamais supprimer ou renommer cette fonction car elle est
        utilisée dans les migrations de la BDD. """
641 642 643 644
        return cls.objects.get_or_create(
            name="SOA to edit",
            mail="postmaser@example.com"
        )[0].pk
645 646


647
class Extension(RevMixin, AclMixin, models.Model):
Gabriel Detraz's avatar
Gabriel Detraz committed
648 649
    """ Extension dns type example.org. Précise si tout le monde peut
    l'utiliser, associé à un origin (ip d'origine)"""
650 651
    PRETTY_NAME = "Extensions dns"

Gabriel Detraz's avatar
Gabriel Detraz committed
652 653 654 655 656
    name = models.CharField(
        max_length=255,
        unique=True,
        help_text="Nom de la zone, doit commencer par un point (.example.org)"
    )
657
    need_infra = models.BooleanField(default=False)
658
    origin = models.ForeignKey(
Gabriel Detraz's avatar
Gabriel Detraz committed
659 660 661
        'IpList',
        on_delete=models.PROTECT,
        blank=True,
Gabriel Detraz's avatar
Gabriel Detraz committed
662 663
        null=True,
        help_text="Enregistrement A associé à la zone"
Gabriel Detraz's avatar
Gabriel Detraz committed
664
    )
665 666 667
    origin_v6 = models.GenericIPAddressField(
        protocol='IPv6',
        null=True,
Gabriel Detraz's avatar
Gabriel Detraz committed
668
        blank=True,
lhark's avatar
lhark committed
669
        help_text="Enregistrement AAAA associé à la zone"
670
    )
671 672
    soa = models.ForeignKey(
        'SOA',
673
        on_delete=models.CASCADE
674
    )
675

676 677 678 679 680 681
    class Meta:
        permissions = (
            ("view_extension", "Peut voir un objet extension"),
            ("use_all_extension", "Peut utiliser toutes les extension"),
        )

682 683
    @cached_property
    def dns_entry(self):
684 685 686
        """ Une entrée DNS A et AAAA sur origin (zone self)"""
        entry = ""
        if self.origin:
687
            entry += "@               IN  A       " + str(self.origin)
688 689 690
        if self.origin_v6:
            if entry:
                entry += "\n"
691
            entry += "@               IN  AAAA    " + str(self.origin_v6)
692
        return entry
693

694
    def get_associated_sshfp_records(self):
695 696 697
        from re2o.utils import all_active_assigned_interfaces
        return (all_active_assigned_interfaces()
                .filter(type__ip_type__extension=self)
698
                .filter(machine__id__in=SshFp.objects.values('machine')))
699

700
    def get_associated_a_records(self):
701 702
        from re2o.utils import all_active_assigned_interfaces
        return (all_active_assigned_interfaces()
703
                .filter(type__ip_type__extension=self)
704
                .filter(ipv4__isnull=False))
705 706

    def get_associated_aaaa_records(self):
707 708 709
        from re2o.utils import all_active_interfaces
        return (all_active_interfaces(full=True)
                .filter(type__ip_type__extension=self))
710 711

    def get_associated_cname_records(self):
712
        from re2o.utils import all_active_assigned_interfaces
713 714
        return (Domain.objects
                .filter(extension=self)
Charlie Jacomme's avatar
Charlie Jacomme committed
715
                .filter(cname__interface_parent__in=all_active_assigned_interfaces())
716
                .prefetch_related('cname'))
717

718 719
    @staticmethod
    def can_use_all(user_request, *_args, **_kwargs):
720 721
        """Superdroit qui permet d'utiliser toutes les extensions sans
        restrictions
722 723
        :param user_request: instance user qui fait l'edition
        :return: True ou False avec la raison de l'échec le cas échéant"""
724
        return user_request.has_perm('machines.use_all_extension'), None
725

726 727
    def __str__(self):
        return self.name
chirac's avatar
chirac committed
728

729
    def clean(self, *args, **kwargs):
Gabriel Detraz's avatar
Gabriel Detraz committed
730 731 732 733
        if self.name and self.name[0] != '.':
            raise ValidationError("Une extension doit commencer par un point")
        super(Extension, self).clean(*args, **kwargs)

Gabriel Detraz's avatar
Gabriel Detraz committed
734

735
class Mx(RevMixin, AclMixin, models.Model):
Gabriel Detraz's avatar
Gabriel Detraz committed
736 737
    """ Entrées des MX. Enregistre la zone (extension) associée et la
    priorité
Gabriel Detraz's avatar
Gabriel Detraz committed
738
    Todo : pouvoir associer un MX à une interface """
739 740 741
    PRETTY_NAME = "Enregistrements MX"

    zone = models.ForeignKey('Extension', on_delete=models.PROTECT)
742 743
    priority = models.PositiveIntegerField()
    name = models.ForeignKey('Domain', on_delete=models.PROTECT)
744

745 746 747 748 749
    class Meta:
        permissions = (
            ("view_mx", "Peut voir un objet mx"),
        )

750 751
    @cached_property
    def dns_entry(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
752 753
        """Renvoie l'entrée DNS complète pour un MX à mettre dans les
        fichiers de zones"""
754 755
        return "@               IN  MX  {prior} {name}".format(
            prior=str(self.priority).ljust(3),
Maël Kervella's avatar
Maël Kervella committed
756
            name=str(self.name)
757
        )
758

759 760 761
    def __str__(self):
        return str(self.zone) + ' ' + str(self.priority) + ' ' + str(self.name)

Gabriel Detraz's avatar
Gabriel Detraz committed
762

763
class Ns(RevMixin, AclMixin, models.Model):
Gabriel Detraz's avatar
Gabriel Detraz committed
764
    """Liste des enregistrements name servers par zone considéérée"""
765 766 767
    PRETTY_NAME = "Enregistrements NS"

    zone = models.ForeignKey('Extension', on_delete=models.PROTECT)
768
    ns = models.ForeignKey('Domain', on_delete=models.PROTECT)
769

770 771
    class Meta:
        permissions = (
Gabriel Detraz's avatar
Gabriel Detraz committed
772
            ("view_ns", "Peut voir un objet ns"),
773 774
        )

775 776
    @cached_property
    def dns_entry(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
777
        """Renvoie un enregistrement NS complet pour les filezones"""
778
        return "@               IN  NS      " + str(self.ns)
779

780
    def __str__(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
781
        return str(self.zone) + ' ' + str(self.ns)
782

Gabriel Detraz's avatar
Gabriel Detraz committed
783

784
class Txt(RevMixin, AclMixin, models.Model):
Gabriel Detraz's avatar
Gabriel Detraz committed
785
    """ Un enregistrement TXT associé à une extension"""
786
    PRETTY_NAME = "Enregistrement TXT"
Gabriel Detraz's avatar
Gabriel Detraz committed
787 788 789

    zone = models.ForeignKey('Extension', on_delete=models.PROTECT)
    field1 = models.CharField(max_length=255)
790
    field2 = models.TextField(max_length=2047)
Gabriel Detraz's avatar
Gabriel Detraz committed
791

792 793 794 795 796
    class Meta:
        permissions = (
            ("view_txt", "Peut voir un objet txt"),
        )

Gabriel Detraz's avatar
Gabriel Detraz committed
797
    def __str__(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
798 799
        return str(self.zone) + " : " + str(self.field1) + " " +\
            str(self.field2)
Gabriel Detraz's avatar
Gabriel Detraz committed
800 801 802

    @cached_property
    def dns_entry(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
803
        """Renvoie l'enregistrement TXT complet pour le fichier de zone"""
804
        return str(self.field1).ljust(15) + " IN  TXT     " + str(self.field2)
Gabriel Detraz's avatar
Gabriel Detraz committed
805

Gabriel Detraz's avatar
Gabriel Detraz committed
806

Charlie Jacomme's avatar
Charlie Jacomme committed
807
class DName(RevMixin, AclMixin, models.Model):
Maël Kervella's avatar
Maël Kervella committed
808
    """A DNAME entry for the DNS."""
Charlie Jacomme's avatar
Charlie Jacomme committed
809 810 811 812 813
    zone = models.ForeignKey('Extension', on_delete=models.PROTECT)
    alias = models.CharField(max_length=255)

    class Meta:
        permissions = (
Maël Kervella's avatar
Maël Kervella committed
814
            ("view_dname", "Can see a dname object"),
Charlie Jacomme's avatar
Charlie Jacomme committed
815
        )
Maël Kervella's avatar
Maël Kervella committed
816 817
        verbose_name = "DNAME entry"
        verbose_name_plural = "DNAME entries"
Charlie Jacomme's avatar
Charlie Jacomme committed
818 819 820 821 822 823

    def __str__(self):
        return str(self.zone) + " : " + str(self.alias)

    @cached_property
    def dns_entry(self):
Maël Kervella's avatar
Maël Kervella committed
824
        """Returns the DNAME record for the DNS zone file."""
chirac's avatar
chirac committed
825
        return str(self.alias).ljust(15) + " IN  DNAME   " + str(self.zone)
Charlie Jacomme's avatar
Charlie Jacomme committed
826 827


828
class Srv(RevMixin, AclMixin, models.Model):
829
    """ A SRV record """
Gabriel Detraz's avatar
Gabriel Detraz committed
830 831 832 833 834
    PRETTY_NAME = "Enregistrement Srv"

    TCP = 'TCP'
    UDP = 'UDP'

835
    service = models.CharField(max_length=31)
Gabriel Detraz's avatar
Gabriel Detraz committed
836 837 838 839 840
    protocole = models.CharField(
        max_length=3,
        choices=(
            (TCP, 'TCP'),
            (UDP, 'UDP'),
841
        ),
Gabriel Detraz's avatar
Gabriel Detraz committed
842 843 844 845 846 847 848 849
        default=TCP,
    )
    extension = models.ForeignKey('Extension', on_delete=models.PROTECT)
    ttl = models.PositiveIntegerField(
        default=172800,  # 2 days
        help_text='Time To Live'
    )
    priority = models.PositiveIntegerField(
850
        default=0,
Gabriel Detraz's avatar
Gabriel Detraz committed
851
        validators=[MaxValueValidator(65535)],
852 853 854
        help_text=("La priorité du serveur cible (valeur entière non "
                   "négative, plus elle est faible, plus ce serveur sera "
                   "utilisé s'il est disponible)")
Gabriel Detraz's avatar
Gabriel Detraz committed
855 856
    )
    weight = models.PositiveIntegerField(
857
        default=0,
Gabriel Detraz's avatar
Gabriel Detraz committed
858 859 860 861 862 863 864 865 866 867 868 869 870 871
        validators=[MaxValueValidator(65535)],
        help_text="Poids relatif pour les enregistrements de même priorité\
            (valeur entière de 0 à 65535)"
    )
    port = models.PositiveIntegerField(
        validators=[MaxValueValidator(65535)],
        help_text="Port (tcp/udp)"
    )
    target = models.ForeignKey(
        'Domain',
        on_delete=models.PROTECT,
        help_text="Serveur cible"
    )

872 873
    class Meta:
        permissions = (
874
            ("view_srv", "Peut voir un objet srv"),
875 876
        )

Gabriel Detraz's avatar
Gabriel Detraz committed
877 878 879 880 881 882 883 884 885 886 887 888 889 890
    def __str__(self):
        return str(self.service) + ' ' + str(self.protocole) + ' ' +\
            str(self.extension) + ' ' + str(self.priority) +\
            ' ' + str(self.weight) + str(self.port) + str(self.target)

    @cached_property
    def dns_entry(self):
        """Renvoie l'enregistrement SRV complet pour le fichier de zone"""
        return str(self.service) + '._' + str(self.protocole).lower() +\
            str(self.extension) + '. ' + str(self.ttl) + ' IN SRV ' +\
            str(self.priority) + ' ' + str(self.weight) + ' ' +\
            str(self.port) + ' ' + str(self.target) + '.'


891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957
class SshFp(RevMixin, AclMixin, models.Model):
    """A fingerprint of an SSH public key"""

    ALGO = (
        ("ssh-rsa", "ssh-rsa"),
        ("ssh-ed25519", "ssh-ed25519"),
        ("ecdsa-sha2-nistp256", "ecdsa-sha2-nistp256"),
        ("ecdsa-sha2-nistp384", "ecdsa-sha2-nistp384"),
        ("ecdsa-sha2-nistp521", "ecdsa-sha2-nistp521"),
    )

    machine = models.ForeignKey('Machine', on_delete=models.CASCADE)
    pub_key_entry = models.TextField(
        help_text="SSH public key",
        max_length=2048
    )
    algo = models.CharField(
        choices=ALGO,
        max_length=32
    )
    comment = models.CharField(
        help_text="Comment",
        max_length=255,
        null=True,
        blank=True
    )

    @cached_property
    def algo_id(self):
        """Return the id of the algorithm for this key"""
        if "ecdsa" in self.algo:
            return 3
        elif "rsa" in self.algo:
            return 1
        else:
            return 2

    @cached_property
    def hash(self):
        """Return the hashess for the pub key with correct id
        cf RFC, 1 is sha1 , 2 sha256"""
        return {
            "1" : hashlib.sha1(base64.b64decode(self.pub_key_entry)).hexdigest(),
            "2" : hashlib.sha256(base64.b64decode(self.pub_key_entry)).hexdigest(),
        }

    class Meta:
        permissions = (
            ("view_sshfp", "Can see an SSHFP record"),
        )
        verbose_name = "SSHFP record"
        verbose_name_plural = "SSHFP records"

    def can_view(self, user_request, *_args, **_kwargs):
        return self.machine.can_view(user_request, *_args, **_kwargs)

    def can_edit(self, user_request, *args, **kwargs):
        return self.machine.can_edit(user_request, *args, **kwargs)

    def can_delete(self, user_request, *args, **kwargs):
        return self.machine.can_delete(user_request, *args, **kwargs)

    def __str__(self):
        return str(self.algo) + ' ' + str(self.comment)



958
class Interface(RevMixin, AclMixin, FieldPermissionModelMixin, models.Model):
Gabriel Detraz's avatar
Gabriel Detraz committed
959 960 961
    """ Une interface. Objet clef de l'application machine :
    - une address mac unique. Possibilité de la rendre unique avec le
    typemachine
Gabriel Detraz's avatar
Gabriel Detraz committed
962 963 964 965
    - une onetoone vers IpList pour attribution ipv4
    - le type parent associé au range ip et à l'extension
    - un objet domain associé contenant son nom
    - la liste des ports oiuvert"""
966 967
    PRETTY_NAME = "Interface"

Gabriel Detraz's avatar
Gabriel Detraz committed
968 969 970 971 972 973
    ipv4 = models.OneToOneField(
        'IpList',
        on_delete=models.PROTECT,
        blank=True,
        null=True
    )
974
    mac_address = MACAddressField(integer=False)
975
    machine = models.ForeignKey('Machine', on_delete=models.CASCADE)
976
    type = models.ForeignKey('MachineType', on_delete=models.PROTECT)
977
    details = models.CharField(max_length=255, blank=True)
978
    port_lists = models.ManyToManyField('OuverturePortList', blank=True)
chirac's avatar
chirac committed
979

980 981 982
    class Meta:
        permissions = (
            ("view_interface", "Peut voir un objet interface"),
983 984
            ("change_interface_machine",
             "Peut changer le propriétaire d'une interface"),
985 986
        )

987
    @cached_property
Dalahro's avatar
Dalahro committed
988 989 990 991
    def is_active(self):
        """ Renvoie si une interface doit avoir accès ou non """
        machine = self.machine
        user = self.machine.user
992
        return machine.active and user.has_access()
Dalahro's avatar
Dalahro committed
993

994
    @cached_property
995
    def ipv6_slaac(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
996 997
        """ Renvoie un objet type ipv6 à partir du prefix associé à
        l'iptype parent"""
998
        if self.type.ip_type.prefix_v6:
Gabriel Detraz's avatar
Gabriel Detraz committed
999 1000 1001
            return EUI(self.mac_address).ipv6(
                IPNetwork(self.type.ip_type.prefix_v6).network
            )
1002 1003 1004
        else:
            return None

1005 1006 1007
    @cached_property
    def gen_ipv6_dhcpv6(self):
        """Cree une ip, à assigner avec dhcpv6 sur une machine"""
1008
        prefix_v6 = self.type.ip_type.prefix_v6.encode().decode('utf-8')
1009 1010
        if not prefix_v6:
            return None
1011 1012 1013 1014
        return IPv6Address(
            IPv6Address(prefix_v6).exploded[:20] +
            IPv6Address(self.id).exploded[20:]
        )
1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027

    def sync_ipv6_dhcpv6(self):
        """Affecte une ipv6 dhcpv6 calculée à partir de l'id de la machine"""
        ipv6_dhcpv6 = self.gen_ipv6_dhcpv6
        if not ipv6_dhcpv6:
            return
        ipv6 = Ipv6List.objects.filter(ipv6=str(ipv6_dhcpv6)).first()
        if not ipv6:
            ipv6 = Ipv6List(ipv6=str(ipv6_dhcpv6))
        ipv6.interface = self
        ipv6.save()
        return

1028 1029 1030 1031 1032 1033 1034 1035
    def sync_ipv6_slaac(self):
        """Cree, mets à jour et supprime si il y a lieu l'ipv6 slaac associée
        à la machine
        Sans prefixe ipv6, on return
        Si l'ip slaac n'est pas celle qu'elle devrait être, on maj"""
        ipv6_slaac = self.ipv6_slaac
        if not ipv6_slaac:
            return
1036 1037 1038
        ipv6_object = (Ipv6List.objects
                       .filter(interface=self, slaac_ip=True)
                       .first())
1039 1040 1041 1042 1043 1044
        if not ipv6_object:
            ipv6_object = Ipv6List(interface=self, slaac_ip=True)
        if ipv6_object.ipv6 != str(ipv6_slaac):
            ipv6_object.ipv6 = str(ipv6_slaac)
            ipv6_object.save()

1045 1046
    def sync_ipv6(self):
        """Cree et met à jour l'ensemble des ipv6 en fonction du mode choisi"""
1047 1048
        if (preferences.models.OptionalMachine
                .get_cached_value('ipv6_mode') == 'SLAAC'):
1049
            self.sync_ipv6_slaac()
1050 1051
        elif (preferences.models.OptionalMachine
              .get_cached_value('ipv6_mode') == 'DHCPV6'):
1052 1053 1054 1055
            self.sync_ipv6_dhcpv6()
        else:
            return

1056
    def ipv6(self):
1057
        """ Renvoie le queryset de la liste des ipv6
1058 1059 1060 1061
        On renvoie l'ipv6 slaac que si le mode slaac est activé
        (et non dhcpv6)"""
        if (preferences.models.OptionalMachine
                .get_cached_value('ipv6_mode') == 'SLAAC'):
1062
            return self.ipv6list.all()
1063 1064
        elif (preferences.models.OptionalMachine
              .get_cached_value('ipv6_mode') == 'DHCPV6'):
1065
            return self.ipv6list.filter(slaac_ip=False)
1066 1067
        else:
            return None
1068

1069
    def mac_bare(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
1070
        """ Formatage de la mac type mac_bare"""
1071 1072
        return str(EUI(self.mac_address, dialect=mac_bare)).lower()

1073
    def filter_macaddress(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
1074 1075
        """ Tente un formatage mac_bare, si échoue, lève une erreur de
        validation"""
1076 1077
        try:
            self.mac_address = str(EUI(self.mac_address))
Gabriel Detraz's avatar
Gabriel Detraz committed
1078
        except:
1079 1080
            raise ValidationError("La mac donnée est invalide")

1081
    def clean(self, *args, **kwargs):
Gabriel Detraz's avatar
Gabriel Detraz committed
1082 1083
        """ Formate l'addresse mac en mac_bare (fonction filter_mac)
        et assigne une ipv4 dans le bon range si inexistante ou incohérente"""
1084 1085 1086 1087 1088 1089 1090
        # If type was an invalid value, django won't create an attribute type
        # but try clean() as we may be able to create it from another value
        # so even if the error as yet been detected at this point, django
        # continues because the error might not prevent us from creating the
        # instance.
        # But in our case, it's impossible to create a type value so we raise
        # the error.
1091
        if not hasattr(self, 'type'):
1092
            raise ValidationError("Le type d'ip choisi n'est pas valide")
1093
        self.filter_macaddress()
1094
        self.mac_address = str(EUI(self.mac_address)) or None
1095
        if not self.ipv4 or self.type.ip_type != self.ipv4.ip_type:
1096
            self.assign_ipv4()
1097
        super(Interface, self).clean(*args, **kwargs)
1098 1099 1100 1101 1102 1103 1104

    def assign_ipv4(self):
        """ Assigne une ip à l'interface """
        free_ips = self.type.ip_type.free_ip()
        if free_ips:
            self.ipv4 = free_ips[0]
        else:
Gabriel Detraz's avatar
Gabriel Detraz committed
1105 1106
            raise ValidationError("Il n'y a plus d'ip disponibles\
            dans le slash")
1107 1108 1109
        return

    def unassign_ipv4(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
1110
        """ Sans commentaire, désassigne une ipv4"""
1111
        self.ipv4 = None
1112

1113 1114 1115 1116 1117
    def update_type(self):
        """ Lorsque le machinetype est changé de type d'ip, on réassigne"""
        self.clean()
        self.save()

1118
    def save(self, *args, **kwargs):
1119
        self.filter_macaddress()
1120
        # On verifie la cohérence en forçant l'extension par la méthode
1121 1122 1123 1124
        if self.ipv4:
            if self.type.ip_type != self.ipv4.ip_type:
                raise ValidationError("L'ipv4 et le type de la machine ne\
                correspondent pas")
1125 1126
        super(Interface, self).save(*args, **kwargs)

1127 1128
    @staticmethod
    def can_create(user_request, machineid, *_args, **_kwargs):
1129 1130 1131 1132 1133
        """Verifie que l'user a les bons droits infra pour créer
        une interface, ou bien que la machine appartient bien à l'user
        :param macineid: Id de la machine parente de l'interface
        :param user_request: instance utilisateur qui fait la requête
        :return: soit True, soit False avec la raison de l'échec"""
1134
        try:
1135
            machine = Machine.objects.get(pk=machineid)
1136 1137
        except Machine.DoesNotExist:
            return False, u"Machine inexistante"
1138
        if not user_request.has_perm('machines.add_interface'):
1139 1140
            if not (preferences.models.OptionalMachine
                    .get_cached_value('create_machine')):
1141
                return False, u"Vous ne pouvez pas ajouter une machine"
1142 1143 1144 1145
            max_lambdauser_interfaces = (preferences.models.OptionalMachine
                                         .get_cached_value(
                                             'max_lambdauser_interfaces'
                                         ))
1146 1147 1148
            if machine.user != user_request:
                return False, u"Vous ne pouvez pas ajouter une interface à une\
                        machine d'un autre user que vous sans droit"
1149 1150
            if (machine.user.user_interfaces().count() >=
                    max_lambdauser_interfaces):
1151 1152 1153 1154 1155
                return False, u"Vous avez atteint le maximum d'interfaces\
                        autorisées que vous pouvez créer vous même (%s) "\
                        % max_lambdauser_interfaces
        return True, None

1156
    @staticmethod
1157 1158 1159
    def can_change_machine(user_request, *_args, **_kwargs):
        """Check if a user can change the machine associated with an
        Interface object """
1160 1161
        return (user_request.has_perm('machines.change_interface_machine'),
                "Droit requis pour changer la machine")
1162

1163
    def can_edit(self, user_request, *args, **kwargs):
1164 1165 1166 1167 1168
        """Verifie que l'user a les bons droits infra pour editer
        cette instance interface, ou qu'elle lui appartient
        :param self: Instance interface à editer
        :param user_request: Utilisateur qui fait la requête
        :return: soit True, soit False avec la raison de l'échec"""
1169
        if self.machine.user != user_request:
1170 1171 1172 1173 1174
            if (not user_request.has_perm('machines.change_interface') or
                    not self.machine.user.can_edit(
                        user_request,
                        *args,
                        **kwargs
1175
            )[0]):
1176 1177
                return False, (u"Vous ne pouvez pas éditer une machine "
                               "d'un autre user que vous sans droit")
1178 1179
        return True, None

1180
    def can_delete(self, user_request, *args, **kwargs):
1181
        """Verifie que l'user a les bons droits delete object pour del
1182 1183 1184 1185
        cette instance interface, ou qu'elle lui appartient
        :param self: Instance interface à del
        :param user_request: Utilisateur qui fait la requête
        :return: soit True, soit False avec la raison de l'échec"""
1186
        if self.machine.user != user_request:
1187 1188 1189 1190 1191
            if (not user_request.has_perm('machines.change_interface') or
                    not self.machine.user.can_edit(
                        user_request,
                        *args,
                        **kwargs
1192
            )[0]):
1193 1194
                return False, (u"Vous ne pouvez pas éditer une machine "
                               "d'un autre user que vous sans droit")
1195 1196
        return True, None

1197
    def can_view(self, user_request, *_args, **_kwargs):
1198
        """Vérifie qu'on peut bien voir cette instance particulière avec
1199
        droit view objet ou qu'elle appartient à l'user
1200 1201 1202
        :param self: instance interface à voir
        :param user_request: instance user qui fait l'edition
        :return: True ou False avec la raison de l'échec le cas échéant"""
1203 1204 1205 1206
        if (not user_request.has_perm('machines.view_interface') and
                self.machine.user != user_request):
            return False, (u"Vous n'avez pas le droit de voir des machines "
                           "autre que les vôtres")
1207 1208
        return True, None

1209 1210 1211
    def __init__(self, *args, **kwargs):
        super(Interface, self).__init__(*args, **kwargs)
        self.field_permissions = {
1212
            'machine': self.can_change_machine,
1213 1214
        }

1215
    def __str__(self):
chirac's avatar
chirac committed
1216 1217 1218 1219 1220
        try:
            domain = self.domain
        except:
            domain = None
        return str(domain)
1221

1222
    def has_private_ip(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
1223
        """ True si l'ip associée est privée"""
1224
        if self.ipv4:
1225 1226 1227
            return IPAddress(str(self.ipv4)).is_private()
        else:
            return False
1228

1229
    def may_have_port_open(self):
Gabriel Detraz's avatar
Gabriel Detraz committed
1230
        """ True si l'interface a une ip et une ip publique.
Gabriel Detraz's avatar
Gabriel Detraz committed
1231 1232
        Permet de ne pas exporter des ouvertures sur des ip privées
        (useless)"""
1233
        return self.ipv4 and not self.has_private_ip()
1234

Gabriel Detraz's avatar
Gabriel Detraz committed
1235

1236
class Ipv6List(RevMixin, AclMixin, FieldPermissionModelMixin, models.Model):
1237
    """ A list of IPv6 """
1238 1239 1240 1241 1242
    PRETTY_NAME = 'Enregistrements Ipv6 des machines'

    ipv6 = models.GenericIPAddressField(
        protocol='IPv6',
    )
1243 1244 1245 1246 1247
    interface = models.ForeignKey(
        'Interface',
        on_delete=models.CASCADE,
        related_name='ipv6list'
    )
1248 1249 1250 1251 1252
    slaac_ip = models.BooleanField(default=False)

    class Meta:
        permissions = (
            ("view_ipv6list", "Peut voir un objet ipv6"),
1253 1254
            ("change_ipv6list_slaac_ip",
             "Peut changer la valeur slaac sur une ipv6"),
1255 1256
        )

1257 1258
    @staticmethod
    def can_create(user_request, interfaceid, *_args, **_kwargs):
1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274
        """Verifie que l'user a les bons droits infra pour créer
        une ipv6, ou possède l'interface associée
        :param interfaceid: Id de l'interface associée à cet objet domain
        :param user_request: instance utilisateur qui fait la requête
        :return: soit True, soit False avec la raison de l'échec"""
        try:
            interface = Interface.objects.get(pk=interfaceid)
        except Interface.DoesNotExist:
            return False, u"Interface inexistante"
        if not user_request.has_perm('machines.add_ipv6list'):
            if interface.machine.user != user_request:
                return False, u"Vous ne pouvez pas ajouter un alias à une\
                        machine d'un autre user que vous sans droit"
        return True, None

    @staticmethod
1275 1276
    def can_change_slaac_ip(user_request, *_args, **_kwargs):
        """ Check if a user can change the slaac value """
1277 1278
        return (user_request.has_perm('machines.change_ipv6list_slaac_ip'),
                "Droit requis pour changer la valeur slaac ip")
1279 1280 1281 1282 1283 1284 1285 1286

    def can_edit(self, user_request, *args, **kwargs):
        """Verifie que l'user a les bons droits infra pour editer
        cette instance interface, ou qu'elle lui appartient
        :param self: Instance interface à editer
        :param user_request: Utilisateur qui fait la requête
        :return: soit True, soit False avec la raison de l'échec"""
        if self.interface.machine.user != user_request:
1287 1288 1289 1290 1291
            if (not user_request.has_perm('machines.change_ipv6list') or
                    not self.interface.machine.user.can_edit(
                        user_request,
                        *args,
                        **kwargs
1292
            )[0]):
1293 1294
                return False, (u"Vous ne pouvez pas éditer une machine "
                               "d'un autre user que vous sans droit")
1295 1296 1297 1298 1299 1300 1301 1302 1303
        return True, None

    def can_delete(self, user_request, *args, **kwargs):
        """Verifie que l'user a les bons droits delete object pour del
        cette instance interface, ou qu'elle lui appartient
        :param self: Instance interface à del
        :param user_request: Utilisateur qui fait la requête
        :return: soit True, soit False avec la raison de l'échec"""
        if self.interface.machine.user != user_request:
1304 1305 1306 1307 1308
            if (not user_request.has_perm('machines.change_ipv6list') or
                    not self.interface.machine.user.can_edit(
                        user_request,
                        *args,