planetstack/core/models/network.py

   1 import os
   2 import socket
   3 from django.db import models
   4 from core.models import PlCoreBase, Site, Slice, Sliver, Deployment
   5 from django.contrib.contenttypes.models import ContentType
   6 from django.contrib.contenttypes import generic
   7
   8 # If true, then IP addresses will be allocated by the model. If false, then
   9 # we will assume the observer handles it.
  10 NO_OBSERVER=False
  11
  12 class NetworkTemplate(PlCoreBase):
  13     VISIBILITY_CHOICES = (('public', 'public'), ('private', 'private'))
  14     TRANSLATION_CHOICES = (('none', 'none'), ('NAT', 'NAT'))
  15
  16     name = models.CharField(max_length=32)
  17     description = models.CharField(max_length=1024, blank=True, null=True)
  18     guaranteedBandwidth = models.IntegerField(default=0)
  19     visibility = models.CharField(max_length=30, choices=VISIBILITY_CHOICES, default="private")
  20     translation = models.CharField(max_length=30, choices=TRANSLATION_CHOICES, default="none")
  21     sharedNetworkName = models.CharField(max_length=30, blank=True, null=True)
  22     sharedNetworkId = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network")
  23
  24     def __unicode__(self):  return u'%s' % (self.name)
  25
  26     @staticmethod
  27     def select_by_user(user):
  28         return NetworkTemplate.objects.all()
  29
  30 class Network(PlCoreBase):
  31     name = models.CharField(max_length=32)
  32     template = models.ForeignKey(NetworkTemplate)
  33     subnet = models.CharField(max_length=32, blank=True)
  34     ports = models.CharField(max_length=1024, blank=True, null=True)
  35     labels = models.CharField(max_length=1024, blank=True, null=True)
  36     owner = models.ForeignKey(Slice, related_name="ownedNetworks", help_text="Slice that owns control of this Network")
  37
  38     guaranteedBandwidth = models.IntegerField(default=0)
  39     permitAllSlices = models.BooleanField(default=False)
  40     permittedSlices = models.ManyToManyField(Slice, blank=True, related_name="availableNetworks")
  41     slices = models.ManyToManyField(Slice, blank=True, related_name="networks", through="NetworkSlice")
  42     slivers = models.ManyToManyField(Sliver, blank=True, related_name="networks", through="NetworkSliver")
  43
  44     # for observer/manager
  45     network_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network")
  46     router_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum router id")
  47     subnet_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum subnet id")
  48
  49     def __unicode__(self):  return u'%s' % (self.name)
  50
  51     def save(self, *args, **kwds):
  52         if (not self.subnet) and (NO_OBSERVER):
  53             from util.network_subnet_allocator import find_unused_subnet
  54             self.subnet = find_unused_subnet(existing_subnets=[x.subnet for x in Network.objects.all()])
  55         super(Network, self).save(*args, **kwds)
  56
  57     def can_update(self, user):
  58         return self.owner.can_update(user)
  59
  60     @property
  61     def nat_list(self):
  62         """ Support a list of ports in the format "protocol:port, protocol:port, ..."
  63             examples:
  64                 tcp 123
  65                 tcp 123:133
  66                 tcp 123, tcp 124, tcp 125, udp 201, udp 202
  67
  68             User can put either a "/" or a " " between protocol and ports
  69             Port ranges can be specified with "-" or ":"
  70         """
  71         nats = []
  72         if self.ports:
  73             parts = self.ports.split(",")
  74             for part in parts:
  75                 part = part.strip()
  76                 if "/" in part:
  77                     (protocol, ports) = part.split("/",1)
  78                 elif " " in part:
  79                     (protocol, ports) = part.split(None,1)
  80                 else:
  81                     raise TypeError('malformed port specifier %s, format example: "tcp 123, tcp 201:206, udp 333"' % part)
  82
  83                 protocol = protocol.strip()
  84                 ports = ports.strip()
  85
  86                 if not (protocol in ["udp", "tcp"]):
  87                     raise TypeError('unknown protocol %s' % protocol)
  88
  89                 if "-" in ports:
  90                     (first, last) = ports.split("-")
  91                     first = int(first.strip())
  92                     last = int(last.strip())
  93                     portStr = "%d:%d" % (first, last)
  94                 elif ":" in ports:
  95                     (first, last) = ports.split(":")
  96                     first = int(first.strip())
  97                     last = int(last.strip())
  98                     portStr = "%d:%d" % (first, last)
  99                 else:
 100                     portStr = "%d" % int(ports)
 101
 102                 nats.append( {"l4_protocol": protocol, "l4_port": portStr} )
 103
 104         return nats
 105
 106     @staticmethod
 107     def select_by_user(user):
 108         if user.is_admin:
 109             qs = Network.objects.all()
 110         else:
 111             slices = Slice.select_by_user(user)
 112             #slice_ids = [s.id for s in Slice.select_by_user(user)]
 113             qs = Network.objects.filter(owner__in=slices)
 114         return qs
 115
 116 class NetworkDeployments(PlCoreBase):
 117     # Stores the openstack ids at various deployments
 118     network = models.ForeignKey(Network)
 119     deployment = models.ForeignKey(Deployment)
 120     net_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network")
 121     router_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum router id")
 122     subnet_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum subnet id")
 123     subnet = models.CharField(max_length=32, blank=True)
 124
 125     def can_update(self, user):
 126         return user.is_admin
 127
 128     @staticmethod
 129     def select_by_user(user):
 130         if user.is_admin:
 131             qs = NetworkDeployments.objects.all()
 132         else:
 133             slices = Slice.select_by_user(user)
 134             networks = Network.objects.filter(owner__in=slices)
 135             qs = NetworkDeployments.objects.filter(network__in=networks)
 136         return qs
 137
 138 class NetworkSlice(PlCoreBase):
 139     # This object exists solely so we can implement the permission check when
 140     # adding slices to networks. It adds no additional fields to the relation.
 141
 142     network = models.ForeignKey(Network)
 143     slice = models.ForeignKey(Slice)
 144
 145     def save(self, *args, **kwds):
 146         slice = self.slice
 147         if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices):
 148             # to add a sliver to the network, then one of the following must be true:
 149             #   1) sliver's slice is in network's permittedSlices list,
 150             #   2) sliver's slice is network's owner, or
 151             #   3) network's permitAllSlices is true
 152             raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network)))
 153
 154         super(NetworkSlice, self).save(*args, **kwds)
 155
 156     def __unicode__(self):  return u'%s-%s' % (self.network.name, self.slice.name)
 157
 158     def can_update(self, user):
 159         return self.slice.can_update(user)
 160
 161     @staticmethod
 162     def select_by_user(user):
 163         if user.is_admin:
 164             qs = NetworkSlice.objects.all()
 165         else:
 166             slice_ids = [s.id for s in Slice.select_by_user(user)]
 167             qs = NetworkSlice.objects.filter(id__in=slice_ids)
 168         return qs
 169
 170 class NetworkSliver(PlCoreBase):
 171     network = models.ForeignKey(Network)
 172     sliver = models.ForeignKey(Sliver)
 173     ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True)
 174     port_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum port id")
 175
 176     def save(self, *args, **kwds):
 177         slice = self.sliver.slice
 178         if (slice not in self.network.permittedSlices.all()) and (slice != self.network.owner) and (not self.network.permitAllSlices):
 179             # to add a sliver to the network, then one of the following must be true:
 180             #   1) sliver's slice is in network's permittedSlices list,
 181             #   2) sliver's slice is network's owner, or
 182             #   3) network's permitAllSlices is true
 183             raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network)))
 184
 185         if (not self.ip) and (NO_OBSERVER):
 186             from util.network_subnet_allocator import find_unused_address
 187             self.ip = find_unused_address(self.network.subnet,
 188                                           [x.ip for x in self.network.networksliver_set.all()])
 189         super(NetworkSliver, self).save(*args, **kwds)
 190
 191     def __unicode__(self):  return u'%s-%s' % (self.network.name, self.sliver.instance_name)
 192
 193     def can_update(self, user):
 194         return self.sliver.can_update(user)
 195
 196     @staticmethod
 197     def select_by_user(user):
 198         if user.is_admin:
 199             qs = NetworkSliver.objects.all()
 200         else:
 201             sliver_ids = [s.id for s in NetworkSliver.select_by_user(user)]
 202             qs = NetworkSliver.objects.filter(id__in=sliver_ids)
 203         return qs
 204
 205 class Router(PlCoreBase):
 206     name = models.CharField(max_length=32)
 207     owner = models.ForeignKey(Slice, related_name="routers")
 208     permittedNetworks = models.ManyToManyField(Network, blank=True, related_name="availableRouters")
 209     networks = models.ManyToManyField(Network, blank=True, related_name="routers")
 210
 211     def __unicode__(self):  return u'%s' % (self.name)
 212
 213 class NetworkParameterType(PlCoreBase):
 214     name = models.SlugField(help_text="The name of this parameter", max_length=128)
 215     description = models.CharField(max_length=1024)
 216
 217     def __unicode__(self):  return u'%s' % (self.name)
 218
 219 class NetworkParameter(PlCoreBase):
 220     parameter = models.ForeignKey(NetworkParameterType, related_name="parameters", help_text="The type of the parameter")
 221     value = models.CharField(help_text="The value of this parameter", max_length=1024)
 222
 223     # The required fields to do a ObjectType lookup, and object_id assignment
 224     content_type = models.ForeignKey(ContentType)
 225     object_id = models.PositiveIntegerField()
 226     content_object = generic.GenericForeignKey('content_type', 'object_id')
 227
 228     def __unicode__(self):
 229         return self.parameter.name
 230
 231