Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 1 | import os |
| 2 | import socket |
Scott Baker | 723fd25 | 2015-01-06 15:11:29 -0800 | [diff] [blame] | 3 | import sys |
Scott Baker | a6a5b29 | 2016-02-12 18:05:34 -0800 | [diff] [blame] | 4 | from django.db import models, transaction |
Scott Baker | 09a3f64 | 2016-04-12 16:29:40 -0700 | [diff] [blame] | 5 | from core.models import PlCoreBase, Site, Slice, Instance, Controller, Service |
Tony Mack | 51c4a7d | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 6 | from core.models import ControllerLinkManager,ControllerLinkDeletionManager |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 7 | from django.contrib.contenttypes.models import ContentType |
| 8 | from django.contrib.contenttypes import generic |
Scott Baker | a289ed7 | 2014-10-17 16:22:20 -0700 | [diff] [blame] | 9 | from django.core.exceptions import ValidationError |
Scott Baker | 34d16d4 | 2015-10-07 22:50:21 -0700 | [diff] [blame] | 10 | from django.db.models import Q |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 11 | |
| 12 | # If true, then IP addresses will be allocated by the model. If false, then |
| 13 | # we will assume the observer handles it. |
Scott Baker | 026bfe7 | 2013-07-29 16:03:50 -0700 | [diff] [blame] | 14 | NO_OBSERVER=False |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 15 | |
Scott Baker | a289ed7 | 2014-10-17 16:22:20 -0700 | [diff] [blame] | 16 | def ParseNatList(ports): |
| 17 | """ Support a list of ports in the format "protocol:port, protocol:port, ..." |
| 18 | examples: |
| 19 | tcp 123 |
| 20 | tcp 123:133 |
| 21 | tcp 123, tcp 124, tcp 125, udp 201, udp 202 |
| 22 | |
| 23 | User can put either a "/" or a " " between protocol and ports |
| 24 | Port ranges can be specified with "-" or ":" |
| 25 | """ |
| 26 | nats = [] |
| 27 | if ports: |
| 28 | parts = ports.split(",") |
| 29 | for part in parts: |
| 30 | part = part.strip() |
| 31 | if "/" in part: |
| 32 | (protocol, ports) = part.split("/",1) |
| 33 | elif " " in part: |
| 34 | (protocol, ports) = part.split(None,1) |
| 35 | else: |
| 36 | raise TypeError('malformed port specifier %s, format example: "tcp 123, tcp 201:206, udp 333"' % part) |
| 37 | |
| 38 | protocol = protocol.strip() |
| 39 | ports = ports.strip() |
| 40 | |
| 41 | if not (protocol in ["udp", "tcp"]): |
| 42 | raise ValueError('unknown protocol %s' % protocol) |
| 43 | |
| 44 | if "-" in ports: |
| 45 | (first, last) = ports.split("-") |
| 46 | first = int(first.strip()) |
| 47 | last = int(last.strip()) |
| 48 | portStr = "%d:%d" % (first, last) |
| 49 | elif ":" in ports: |
| 50 | (first, last) = ports.split(":") |
| 51 | first = int(first.strip()) |
| 52 | last = int(last.strip()) |
| 53 | portStr = "%d:%d" % (first, last) |
| 54 | else: |
| 55 | portStr = "%d" % int(ports) |
| 56 | |
| 57 | nats.append( {"l4_protocol": protocol, "l4_port": portStr} ) |
| 58 | |
| 59 | return nats |
| 60 | |
| 61 | def ValidateNatList(ports): |
| 62 | try: |
| 63 | ParseNatList(ports) |
| 64 | except Exception,e: |
| 65 | raise ValidationError(str(e)) |
| 66 | |
Scott Baker | d0f74ec | 2015-11-16 15:07:43 -0800 | [diff] [blame] | 67 | class ParameterMixin(object): |
| 68 | # helper classes for dealing with NetworkParameter |
| 69 | |
| 70 | def get_parameters(self): |
| 71 | parameter_dict = {} |
| 72 | |
| 73 | instance_type = ContentType.objects.get_for_model(self) |
| 74 | for param in NetworkParameter.objects.filter(content_type__pk=instance_type.id, object_id=self.id): |
| 75 | parameter_dict[param.parameter.name] = param.value |
| 76 | |
| 77 | return parameter_dict |
| 78 | |
| 79 | def set_parameter(self, name, value): |
| 80 | instance_type = ContentType.objects.get_for_model(self) |
| 81 | existing_params = NetworkParameter.objects.filter(parameter__name=name, content_type__pk=instance_type.id, object_id=self.id) |
| 82 | if existing_params: |
| 83 | p=existing_params[0] |
| 84 | p.value = value |
| 85 | p.save() |
| 86 | else: |
| 87 | pt = NetworkParameterType.objects.get(name=name) |
| 88 | p = NetworkParameter(parameter=pt, content_type=instance_type, object_id=self.id, value=value) |
| 89 | p.save() |
| 90 | |
| 91 | def unset_parameter(self, name): |
| 92 | instance_type = ContentType.objects.get_for_model(self) |
| 93 | existing_params = NetworkParameter.objects.filter(parameter__name=name, content_type__pk=instance_type.id, object_id=self.id) |
| 94 | for p in existing_params: |
| 95 | p.delete() |
| 96 | |
| 97 | |
| 98 | class NetworkTemplate(PlCoreBase, ParameterMixin): |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 99 | VISIBILITY_CHOICES = (('public', 'public'), ('private', 'private')) |
Scott Baker | 87e5e09 | 2013-08-07 18:58:10 -0700 | [diff] [blame] | 100 | TRANSLATION_CHOICES = (('none', 'none'), ('NAT', 'NAT')) |
Scott Baker | 59078f8 | 2014-11-17 16:03:49 -0800 | [diff] [blame] | 101 | TOPOLOGY_CHOICES = (('bigswitch', 'BigSwitch'), ('physical', 'Physical'), ('custom', 'Custom')) |
| 102 | CONTROLLER_CHOICES = ((None, 'None'), ('onos', 'ONOS'), ('custom', 'Custom')) |
Scott Baker | a36a990 | 2015-12-09 15:44:55 -0800 | [diff] [blame] | 103 | ACCESS_CHOICES = ((None, 'None'), ('indirect', 'Indirect'), ('direct', 'Direct')) |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 104 | |
| 105 | name = models.CharField(max_length=32) |
| 106 | description = models.CharField(max_length=1024, blank=True, null=True) |
Scott Baker | 369f9b9 | 2015-01-03 12:03:38 -0800 | [diff] [blame] | 107 | guaranteed_bandwidth = models.IntegerField(default=0) |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 108 | visibility = models.CharField(max_length=30, choices=VISIBILITY_CHOICES, default="private") |
Scott Baker | 87e5e09 | 2013-08-07 18:58:10 -0700 | [diff] [blame] | 109 | translation = models.CharField(max_length=30, choices=TRANSLATION_CHOICES, default="none") |
Scott Baker | a36a990 | 2015-12-09 15:44:55 -0800 | [diff] [blame] | 110 | access = models.CharField(max_length=30, null=True, blank=True, choices=ACCESS_CHOICES, help_text="Advertise this network as a means for other slices to contact this slice") |
Scott Baker | 369f9b9 | 2015-01-03 12:03:38 -0800 | [diff] [blame] | 111 | shared_network_name = models.CharField(max_length=30, blank=True, null=True) |
| 112 | shared_network_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network") |
Scott Baker | e60c73b | 2015-12-09 16:31:26 -0800 | [diff] [blame] | 113 | topology_kind = models.CharField(null=False, blank=False, max_length=30, choices=TOPOLOGY_CHOICES, default="bigswitch") |
Scott Baker | 369f9b9 | 2015-01-03 12:03:38 -0800 | [diff] [blame] | 114 | controller_kind = models.CharField(null=True, blank=True, max_length=30, choices=CONTROLLER_CHOICES, default=None) |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 115 | |
Scott Baker | 201f7da | 2014-12-23 10:56:06 -0800 | [diff] [blame] | 116 | def __init__(self, *args, **kwargs): |
| 117 | super(NetworkTemplate, self).__init__(*args, **kwargs) |
| 118 | |
| 119 | # somehow these got set wrong inside of the live database. Remove this |
| 120 | # code after all is well... |
Scott Baker | 369f9b9 | 2015-01-03 12:03:38 -0800 | [diff] [blame] | 121 | if (self.topology_kind=="BigSwitch"): |
Scott Baker | 723fd25 | 2015-01-06 15:11:29 -0800 | [diff] [blame] | 122 | print >> sys.stderr, "XXX warning: topology_kind invalid case" |
Scott Baker | 369f9b9 | 2015-01-03 12:03:38 -0800 | [diff] [blame] | 123 | self.topology_kind="bigswitch" |
| 124 | elif (self.topology_kind=="Physical"): |
Scott Baker | 723fd25 | 2015-01-06 15:11:29 -0800 | [diff] [blame] | 125 | print >> sys.stderr, "XXX warning: topology_kind invalid case" |
Scott Baker | 369f9b9 | 2015-01-03 12:03:38 -0800 | [diff] [blame] | 126 | self.topology_kind="physical" |
| 127 | elif (self.topology_kind=="Custom"): |
Scott Baker | 723fd25 | 2015-01-06 15:11:29 -0800 | [diff] [blame] | 128 | print >> sys.stderr, "XXX warning: topology_kind invalid case" |
Scott Baker | 369f9b9 | 2015-01-03 12:03:38 -0800 | [diff] [blame] | 129 | self.toplogy_kind="custom" |
Scott Baker | 201f7da | 2014-12-23 10:56:06 -0800 | [diff] [blame] | 130 | |
Scott Baker | e60c73b | 2015-12-09 16:31:26 -0800 | [diff] [blame] | 131 | def save(self, *args, **kwargs): |
| 132 | self.enforce_choices(self.access, self.ACCESS_CHOICES) |
| 133 | super(NetworkTemplate, self).save(*args, **kwargs) |
| 134 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 135 | def __unicode__(self): return u'%s' % (self.name) |
| 136 | |
Scott Baker | d0f74ec | 2015-11-16 15:07:43 -0800 | [diff] [blame] | 137 | class Network(PlCoreBase, ParameterMixin): |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 138 | name = models.CharField(max_length=32) |
| 139 | template = models.ForeignKey(NetworkTemplate) |
| 140 | subnet = models.CharField(max_length=32, blank=True) |
Scott Baker | a289ed7 | 2014-10-17 16:22:20 -0700 | [diff] [blame] | 141 | ports = models.CharField(max_length=1024, blank=True, null=True, validators=[ValidateNatList]) |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 142 | labels = models.CharField(max_length=1024, blank=True, null=True) |
Siobhan Tully | ce652d0 | 2013-10-08 21:52:35 -0400 | [diff] [blame] | 143 | owner = models.ForeignKey(Slice, related_name="ownedNetworks", help_text="Slice that owns control of this Network") |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 144 | |
Scott Baker | 549aa25 | 2015-01-03 12:29:29 -0800 | [diff] [blame] | 145 | guaranteed_bandwidth = models.IntegerField(default=0) |
| 146 | permit_all_slices = models.BooleanField(default=False) |
| 147 | permitted_slices = models.ManyToManyField(Slice, blank=True, related_name="availableNetworks") |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 148 | slices = models.ManyToManyField(Slice, blank=True, related_name="networks", through="NetworkSlice") |
Scott Baker | c682292 | 2015-09-14 11:41:05 -0700 | [diff] [blame] | 149 | instances = models.ManyToManyField(Instance, blank=True, related_name="networks", through="Port") |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 150 | |
Scott Baker | 549aa25 | 2015-01-03 12:29:29 -0800 | [diff] [blame] | 151 | topology_parameters = models.TextField(null=True, blank=True) |
| 152 | controller_url = models.CharField(null=True, blank=True, max_length=1024) |
| 153 | controller_parameters = models.TextField(null=True, blank=True) |
Scott Baker | 59078f8 | 2014-11-17 16:03:49 -0800 | [diff] [blame] | 154 | |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 155 | # for observer/manager |
| 156 | network_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network") |
| 157 | router_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum router id") |
| 158 | subnet_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum subnet id") |
| 159 | |
Scott Baker | 3789cb2 | 2015-08-21 16:40:53 -0700 | [diff] [blame] | 160 | autoconnect = models.BooleanField(default=True, help_text="This network can be autoconnected to the slice that owns it") |
| 161 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 162 | def __unicode__(self): return u'%s' % (self.name) |
| 163 | |
| 164 | def save(self, *args, **kwds): |
| 165 | if (not self.subnet) and (NO_OBSERVER): |
| 166 | from util.network_subnet_allocator import find_unused_subnet |
| 167 | self.subnet = find_unused_subnet(existing_subnets=[x.subnet for x in Network.objects.all()]) |
| 168 | super(Network, self).save(*args, **kwds) |
| 169 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 170 | def can_update(self, user): |
Tony Mack | 3428e6e | 2015-02-08 21:38:41 -0500 | [diff] [blame] | 171 | return user.can_update_slice(self.owner) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 172 | |
Scott Baker | 5bbaa23 | 2014-08-14 17:23:15 -0700 | [diff] [blame] | 173 | @property |
| 174 | def nat_list(self): |
Scott Baker | a289ed7 | 2014-10-17 16:22:20 -0700 | [diff] [blame] | 175 | return ParseNatList(self.ports) |
Scott Baker | 5bbaa23 | 2014-08-14 17:23:15 -0700 | [diff] [blame] | 176 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 177 | @staticmethod |
| 178 | def select_by_user(user): |
| 179 | if user.is_admin: |
| 180 | qs = Network.objects.all() |
| 181 | else: |
Tony Mack | 5efa133 | 2014-04-02 15:45:48 -0400 | [diff] [blame] | 182 | slices = Slice.select_by_user(user) |
| 183 | #slice_ids = [s.id for s in Slice.select_by_user(user)] |
| 184 | qs = Network.objects.filter(owner__in=slices) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 185 | return qs |
| 186 | |
Scott Baker | d0f74ec | 2015-11-16 15:07:43 -0800 | [diff] [blame] | 187 | def get_parameters(self): |
| 188 | # returns parameters from the template, updated by self. |
| 189 | p={} |
| 190 | if self.template: |
| 191 | p = self.template.get_parameters() |
| 192 | p.update(ParameterMixin.get_parameters(self)) |
| 193 | return p |
| 194 | |
Tony Mack | 3066a95 | 2015-01-05 22:48:11 -0500 | [diff] [blame] | 195 | class ControllerNetwork(PlCoreBase): |
Tony Mack | 51c4a7d | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 196 | objects = ControllerLinkManager() |
| 197 | deleted_objects = ControllerLinkDeletionManager() |
Sapan Bhatia | ed7b83b | 2014-09-22 14:52:59 -0400 | [diff] [blame] | 198 | |
Tony Mack | 51c4a7d | 2014-11-30 15:33:35 -0500 | [diff] [blame] | 199 | # Stores the openstack ids at various controllers |
| 200 | network = models.ForeignKey(Network, related_name='controllernetworks') |
| 201 | controller = models.ForeignKey(Controller, related_name='controllernetworks') |
Tony Mack | 457c84c | 2014-04-08 16:37:56 -0400 | [diff] [blame] | 202 | net_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum network") |
Tony Mack | e9b0869 | 2014-04-07 19:38:28 -0400 | [diff] [blame] | 203 | router_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum router id") |
Scott Baker | 95d81c7 | 2014-08-12 18:29:27 -0700 | [diff] [blame] | 204 | subnet_id = models.CharField(null=True, blank=True, max_length=256, help_text="Quantum subnet id") |
| 205 | subnet = models.CharField(max_length=32, blank=True) |
Scott Baker | 3927886 | 2016-05-18 08:56:50 -0700 | [diff] [blame] | 206 | segmentation_id = models.IntegerField(null=True, blank=True) |
Scott Baker | e4ea895 | 2015-10-26 15:12:13 -0700 | [diff] [blame] | 207 | |
Tony Mack | 5d93a9e | 2015-04-11 12:17:59 -0400 | [diff] [blame] | 208 | class Meta: |
| 209 | unique_together = ('network', 'controller') |
Scott Baker | d0f74ec | 2015-11-16 15:07:43 -0800 | [diff] [blame] | 210 | |
Sapan Bhatia | d0e8ed8 | 2016-04-06 19:30:24 +0200 | [diff] [blame] | 211 | def tologdict(self): |
| 212 | d=super(ControllerNetwork,self).tologdict() |
| 213 | try: |
| 214 | d['network_name']=self.network.name |
| 215 | d['controller_name']=self.controller.name |
| 216 | except: |
| 217 | pass |
| 218 | return d |
| 219 | |
Tony Mack | e9b0869 | 2014-04-07 19:38:28 -0400 | [diff] [blame] | 220 | @staticmethod |
| 221 | def select_by_user(user): |
| 222 | if user.is_admin: |
Scott Baker | 7d85a03 | 2015-02-23 17:22:33 -0800 | [diff] [blame] | 223 | qs = ControllerNetwork.objects.all() |
Tony Mack | e9b0869 | 2014-04-07 19:38:28 -0400 | [diff] [blame] | 224 | else: |
| 225 | slices = Slice.select_by_user(user) |
| 226 | networks = Network.objects.filter(owner__in=slices) |
Scott Baker | 7d85a03 | 2015-02-23 17:22:33 -0800 | [diff] [blame] | 227 | qs = ControllerNetwork.objects.filter(network__in=networks) |
Scott Baker | 95d81c7 | 2014-08-12 18:29:27 -0700 | [diff] [blame] | 228 | return qs |
Tony Mack | e9b0869 | 2014-04-07 19:38:28 -0400 | [diff] [blame] | 229 | |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 230 | class NetworkSlice(PlCoreBase): |
| 231 | # This object exists solely so we can implement the permission check when |
| 232 | # adding slices to networks. It adds no additional fields to the relation. |
| 233 | |
Sapan Bhatia | 6bfa2ca | 2014-11-11 21:47:45 -0500 | [diff] [blame] | 234 | network = models.ForeignKey(Network,related_name='networkslices') |
| 235 | slice = models.ForeignKey(Slice,related_name='networkslices') |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 236 | |
Tony Mack | 5d93a9e | 2015-04-11 12:17:59 -0400 | [diff] [blame] | 237 | class Meta: |
| 238 | unique_together = ('network', 'slice') |
Tony Mack | c8836df | 2015-03-09 17:13:14 -0400 | [diff] [blame] | 239 | |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 240 | def save(self, *args, **kwds): |
| 241 | slice = self.slice |
Scott Baker | b3c363e | 2015-04-13 17:23:28 -0700 | [diff] [blame] | 242 | if (slice not in self.network.permitted_slices.all()) and (slice != self.network.owner) and (not self.network.permit_all_slices): |
Tony Mack | d851547 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 243 | # to add a instance to the network, then one of the following must be true: |
| 244 | # 1) instance's slice is in network's permittedSlices list, |
| 245 | # 2) instance's slice is network's owner, or |
Scott Baker | 87191e7 | 2013-08-06 08:55:07 -0700 | [diff] [blame] | 246 | # 3) network's permitAllSlices is true |
| 247 | raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network))) |
| 248 | |
| 249 | super(NetworkSlice, self).save(*args, **kwds) |
| 250 | |
| 251 | def __unicode__(self): return u'%s-%s' % (self.network.name, self.slice.name) |
| 252 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 253 | def can_update(self, user): |
Tony Mack | 3428e6e | 2015-02-08 21:38:41 -0500 | [diff] [blame] | 254 | return user.can_update_slice(self.slice) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 255 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 256 | @staticmethod |
| 257 | def select_by_user(user): |
| 258 | if user.is_admin: |
| 259 | qs = NetworkSlice.objects.all() |
| 260 | else: |
| 261 | slice_ids = [s.id for s in Slice.select_by_user(user)] |
Scott Baker | 34d16d4 | 2015-10-07 22:50:21 -0700 | [diff] [blame] | 262 | network_ids = [network.id for network in Network.select_by_user(user)] |
| 263 | qs = NetworkSlice.objects.filter(Q(slice__in=slice_ids) | Q(network__in=network_ids)) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 264 | return qs |
| 265 | |
Scott Baker | d0f74ec | 2015-11-16 15:07:43 -0800 | [diff] [blame] | 266 | class Port(PlCoreBase, ParameterMixin): |
Scott Baker | c682292 | 2015-09-14 11:41:05 -0700 | [diff] [blame] | 267 | network = models.ForeignKey(Network,related_name='links') |
| 268 | instance = models.ForeignKey(Instance, null=True, blank=True, related_name='ports') |
Tony Mack | d851547 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 269 | ip = models.GenericIPAddressField(help_text="Instance ip address", blank=True, null=True) |
Scott Baker | e50115b | 2015-11-16 20:15:41 -0800 | [diff] [blame] | 270 | port_id = models.CharField(null=True, blank=True, max_length=256, help_text="Neutron port id") |
Scott Baker | e553b47 | 2015-09-08 18:22:15 -0700 | [diff] [blame] | 271 | mac = models.CharField(null=True, blank=True, max_length=256, help_text="MAC address associated with this port") |
Scott Baker | c5dfb5f | 2016-02-16 11:59:07 -0800 | [diff] [blame] | 272 | xos_created = models.BooleanField(default=False) # True if XOS created this port in Neutron, False if port created by Neutron and observed by XOS |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 273 | |
Tony Mack | 5d93a9e | 2015-04-11 12:17:59 -0400 | [diff] [blame] | 274 | class Meta: |
Tony Mack | d851547 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 275 | unique_together = ('network', 'instance') |
Tony Mack | c8836df | 2015-03-09 17:13:14 -0400 | [diff] [blame] | 276 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 277 | def save(self, *args, **kwds): |
Tony Mack | d851547 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 278 | if self.instance: |
| 279 | slice = self.instance.slice |
Scott Baker | a92af1f | 2015-08-18 17:04:01 -0700 | [diff] [blame] | 280 | if (slice not in self.network.permitted_slices.all()) and (slice != self.network.owner) and (not self.network.permit_all_slices): |
Tony Mack | d851547 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 281 | # to add a instance to the network, then one of the following must be true: |
| 282 | # 1) instance's slice is in network's permittedSlices list, |
| 283 | # 2) instance's slice is network's owner, or |
Scott Baker | a92af1f | 2015-08-18 17:04:01 -0700 | [diff] [blame] | 284 | # 3) network's permitAllSlices is true |
| 285 | raise ValueError("Slice %s is not allowed to connect to network %s" % (str(slice), str(self.network))) |
| 286 | |
Scott Baker | c682292 | 2015-09-14 11:41:05 -0700 | [diff] [blame] | 287 | super(Port, self).save(*args, **kwds) |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 288 | |
Scott Baker | a92af1f | 2015-08-18 17:04:01 -0700 | [diff] [blame] | 289 | def __unicode__(self): |
Tony Mack | d851547 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 290 | if self.instance: |
| 291 | return u'%s-%s' % (self.network.name, self.instance.instance_name) |
Scott Baker | a92af1f | 2015-08-18 17:04:01 -0700 | [diff] [blame] | 292 | else: |
Scott Baker | d0ca597 | 2015-08-25 23:24:36 -0700 | [diff] [blame] | 293 | return u'%s-unboundport-%s' % (self.network.name, self.id) |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 294 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 295 | def can_update(self, user): |
Tony Mack | d851547 | 2015-08-19 11:58:18 -0400 | [diff] [blame] | 296 | if self.instance: |
| 297 | return user.can_update_slice(self.instance.slice) |
Scott Baker | a92af1f | 2015-08-18 17:04:01 -0700 | [diff] [blame] | 298 | if self.network: |
| 299 | return user.can_update_slice(self.network.owner) |
| 300 | return False |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 301 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 302 | @staticmethod |
| 303 | def select_by_user(user): |
| 304 | if user.is_admin: |
Scott Baker | c682292 | 2015-09-14 11:41:05 -0700 | [diff] [blame] | 305 | qs = Port.objects.all() |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 306 | else: |
Scott Baker | 34d16d4 | 2015-10-07 22:50:21 -0700 | [diff] [blame] | 307 | instances = Instance.select_by_user(user) |
| 308 | instance_ids = [instance.id for instance in instances] |
| 309 | networks = Network.select_by_user(user) |
| 310 | network_ids = [network.id for network in networks] |
| 311 | qs = Port.objects.filter(Q(instance__in=instance_ids) | Q(network__in=network_ids)) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 312 | return qs |
| 313 | |
Scott Baker | d0f74ec | 2015-11-16 15:07:43 -0800 | [diff] [blame] | 314 | def get_parameters(self): |
| 315 | # returns parameters from the network, updated by self. |
| 316 | p={} |
| 317 | if self.network: |
| 318 | p = self.network.get_parameters() |
| 319 | p.update(ParameterMixin.get_parameters(self)) |
| 320 | return p |
| 321 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 322 | class Router(PlCoreBase): |
| 323 | name = models.CharField(max_length=32) |
| 324 | owner = models.ForeignKey(Slice, related_name="routers") |
| 325 | permittedNetworks = models.ManyToManyField(Network, blank=True, related_name="availableRouters") |
| 326 | networks = models.ManyToManyField(Network, blank=True, related_name="routers") |
| 327 | |
| 328 | def __unicode__(self): return u'%s' % (self.name) |
| 329 | |
Tony Mack | 3428e6e | 2015-02-08 21:38:41 -0500 | [diff] [blame] | 330 | def can_update(self, user): |
| 331 | return user.can_update_slice(self.owner) |
| 332 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 333 | class NetworkParameterType(PlCoreBase): |
| 334 | name = models.SlugField(help_text="The name of this parameter", max_length=128) |
| 335 | description = models.CharField(max_length=1024) |
| 336 | |
| 337 | def __unicode__(self): return u'%s' % (self.name) |
| 338 | |
| 339 | class NetworkParameter(PlCoreBase): |
Sapan Bhatia | 6bfa2ca | 2014-11-11 21:47:45 -0500 | [diff] [blame] | 340 | parameter = models.ForeignKey(NetworkParameterType, related_name="networkparameters", help_text="The type of the parameter") |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 341 | value = models.CharField(help_text="The value of this parameter", max_length=1024) |
| 342 | |
| 343 | # The required fields to do a ObjectType lookup, and object_id assignment |
| 344 | content_type = models.ForeignKey(ContentType) |
| 345 | object_id = models.PositiveIntegerField() |
| 346 | content_object = generic.GenericForeignKey('content_type', 'object_id') |
| 347 | |
| 348 | def __unicode__(self): |
| 349 | return self.parameter.name |
| 350 | |
Scott Baker | 6167c92 | 2016-02-12 17:31:04 -0800 | [diff] [blame] | 351 | class AddressPool(PlCoreBase): |
| 352 | name = models.CharField(max_length=32) |
| 353 | addresses = models.TextField(blank=True, null=True) |
Scott Baker | 09a3f64 | 2016-04-12 16:29:40 -0700 | [diff] [blame] | 354 | gateway_ip = models.CharField(max_length=32, null=True) |
| 355 | gateway_mac = models.CharField(max_length=32, null=True) |
Scott Baker | fca52df | 2016-04-13 16:40:50 -0700 | [diff] [blame] | 356 | cidr = models.CharField(max_length=32, null=True) |
Scott Baker | 2e60860 | 2016-02-12 17:41:52 -0800 | [diff] [blame] | 357 | inuse = models.TextField(blank=True, null=True) |
Scott Baker | 09a3f64 | 2016-04-12 16:29:40 -0700 | [diff] [blame] | 358 | service = models.ForeignKey(Service, related_name="addresspools", null=True, blank=True) |
Scott Baker | 6167c92 | 2016-02-12 17:31:04 -0800 | [diff] [blame] | 359 | |
| 360 | def __unicode__(self): return u'%s' % (self.name) |
| 361 | |
| 362 | def get_address(self): |
| 363 | with transaction.atomic(): |
| 364 | ap = AddressPool.objects.get(pk=self.pk) |
| 365 | if ap.addresses: |
Scott Baker | 272949f | 2016-02-18 12:21:09 -0800 | [diff] [blame] | 366 | avail_ips = ap.addresses.split() |
Scott Baker | 6167c92 | 2016-02-12 17:31:04 -0800 | [diff] [blame] | 367 | else: |
Scott Baker | 272949f | 2016-02-18 12:21:09 -0800 | [diff] [blame] | 368 | avail_ips = [] |
| 369 | |
| 370 | if ap.inuse: |
| 371 | inuse_ips = ap.inuse.split() |
| 372 | else: |
| 373 | inuse_ips = [] |
| 374 | |
| 375 | while avail_ips: |
| 376 | addr = avail_ips.pop(0) |
| 377 | |
| 378 | if addr in inuse_ips: |
| 379 | # This may have happened if someone re-ran the tosca |
| 380 | # recipe and 'refilled' the AddressPool while some addresses |
| 381 | # were still in use. |
| 382 | continue |
| 383 | |
| 384 | inuse_ips.insert(0,addr) |
| 385 | |
| 386 | ap.inuse = " ".join(inuse_ips) |
| 387 | ap.addresses = " ".join(avail_ips) |
| 388 | ap.save() |
| 389 | return addr |
| 390 | |
| 391 | addr = None |
Scott Baker | 6167c92 | 2016-02-12 17:31:04 -0800 | [diff] [blame] | 392 | return addr |
| 393 | |
| 394 | def put_address(self, addr): |
| 395 | with transaction.atomic(): |
| 396 | ap = AddressPool.objects.get(pk=self.pk) |
Scott Baker | a6a5b29 | 2016-02-12 18:05:34 -0800 | [diff] [blame] | 397 | addresses = ap.addresses or "" |
| 398 | parts = addresses.split() |
Scott Baker | 6167c92 | 2016-02-12 17:31:04 -0800 | [diff] [blame] | 399 | if addr not in parts: |
Scott Baker | a6a5b29 | 2016-02-12 18:05:34 -0800 | [diff] [blame] | 400 | parts.insert(0,addr) |
Scott Baker | 6167c92 | 2016-02-12 17:31:04 -0800 | [diff] [blame] | 401 | ap.addresses = " ".join(parts) |
Scott Baker | a6a5b29 | 2016-02-12 18:05:34 -0800 | [diff] [blame] | 402 | |
| 403 | inuse = ap.inuse or "" |
| 404 | parts = inuse.split() |
| 405 | if addr in parts: |
| 406 | parts.remove(addr) |
| 407 | ap.inuse = " ".join(parts) |
| 408 | |
| 409 | ap.save() |
| 410 | |
Scott Baker | 58a9c7a | 2013-07-29 15:43:07 -0700 | [diff] [blame] | 411 | |