blob: 0419fbee2ea046bfb1e20c91bf88edc71f82d361 [file] [log] [blame]
from header import *
#from core.models.service import Service
from core.models import Service
from core.models import AddressPool#from core.models.tenant import Tenant
from core.models import Tenant
class VRouterApp(XOSBase):
KIND = "vROUTER"
class Meta:
app_label = "vrouter"
name = "vrouter"
verbose_name = "vRouter Service"
# Primitive Fields (Not Relations)
name = CharField( help_text = "application name", max_length = 50, null = False, db_index = False, blank = False )
control_plane_connect_point = CharField( help_text = "port identifier in ONOS", max_length = 21, null = False, db_index = False, blank = False )
ospf_enabled = BooleanField( default = True, help_text = "ospf enabled", null = False, db_index = False, blank = True )
# Relations
vrouter_service = ForeignKey(VRouterService, db_index = True, related_name = 'apps', null = False, blank = False )
def _get_interfaces(self):
app_interfaces = []
devices = VRouterDevice.objects.filter(vrouter_service=self.vrouter_service)
for device in devices:
ports = VRouterPort.objects.filter(vrouter_device=device.id)
for port in ports:
interfaces = VRouterInterface.objects.filter(vrouter_port=port.id)
for iface in interfaces:
app_interfaces.append(iface.name)
return app_interfaces
pass
class VRouterDevice(XOSBase):
KIND = "vROUTER"
class Meta:
app_label = "vrouter"
name = "vrouter"
verbose_name = "vRouter Service"
# Primitive Fields (Not Relations)
name = CharField( help_text = "device friendly name", max_length = 20, null = True, db_index = False, blank = True )
openflow_id = CharField( help_text = "device identifier in ONOS", max_length = 20, null = False, db_index = False, blank = False )
config_key = CharField( default = "basic", max_length = 32, blank = False, help_text = "configuration key", null = False, db_index = False )
driver = CharField( help_text = "driver type", max_length = 32, null = False, db_index = False, blank = False )
# Relations
vrouter_service = ForeignKey(VRouterService, db_index = True, related_name = 'devices', null = False, blank = False )
pass
class VRouterInterface(XOSBase):
KIND = "vROUTER"
class Meta:
app_label = "vrouter"
name = "vrouter"
verbose_name = "vRouter Service"
# Primitive Fields (Not Relations)
name = CharField( help_text = "interface name", max_length = 10, null = False, db_index = False, blank = False )
mac = CharField( help_text = "interface mac", max_length = 17, null = False, db_index = False, blank = False )
vlan = CharField( help_text = "interface vlan id", max_length = 10, null = True, db_index = False, blank = True )
# Relations
vrouter_port = ForeignKey(VRouterPort, db_index = True, related_name = 'interfaces', null = False, blank = False )
pass
class VRouterIp(XOSBase):
KIND = "vROUTER"
class Meta:
app_label = "vrouter"
name = "vrouter"
verbose_name = "vRouter Service"
# Primitive Fields (Not Relations)
name = CharField( help_text = "ip friendly name", max_length = 20, null = True, db_index = False, blank = True )
ip = CharField( help_text = "interface ips", max_length = 19, null = False, db_index = False, blank = False )
# Relations
vrouter_interface = ForeignKey(VRouterInterface, db_index = True, related_name = 'ips', null = False, blank = False )
pass
class VRouterPort(XOSBase):
KIND = "vROUTER"
class Meta:
app_label = "vrouter"
name = "vrouter"
verbose_name = "vRouter Service"
# Primitive Fields (Not Relations)
name = CharField( help_text = "port friendly name", max_length = 20, null = True, db_index = False, blank = True )
openflow_id = CharField( help_text = "port identifier in ONOS", max_length = 21, null = False, db_index = False, blank = False )
# Relations
vrouter_device = ForeignKey(VRouterDevice, db_index = True, related_name = 'ports', null = False, blank = False )
vrouter_service = ForeignKey(VRouterService, db_index = True, related_name = 'device_ports', null = False, blank = False )
pass
class VRouterService(Service):
KIND = "vROUTER"
class Meta:
app_label = "vrouter"
name = "vrouter"
verbose_name = "vRouter Service"
# Primitive Fields (Not Relations)
rest_hostname = StrippedCharField( db_index = False, max_length = 255, null = True, blank = True )
rest_port = IntegerField( default = 8181, null = False, blank = False, db_index = False )
rest_user = StrippedCharField( default = "onos", max_length = 255, null = False, db_index = False, blank = False )
rest_pass = StrippedCharField( default = "rocks", max_length = 255, null = False, db_index = False, blank = False )
# Relations
def ip_to_mac(self, ip):
(a, b, c, d) = ip.split('.')
return "02:42:%02x:%02x:%02x:%02x" % (int(a), int(b), int(c), int(d))
def get_gateways(self):
gateways = []
aps = self.addresspools.all()
for ap in aps:
gateways.append({"gateway_ip": ap.gateway_ip, "gateway_mac": ap.gateway_mac})
return gateways
def get_address_pool(self, name):
ap = AddressPool.objects.filter(name=name, service=self)
if not ap:
raise Exception("vRouter unable to find addresspool %s" % name)
return ap[0]
def get_tenant(self, **kwargs):
address_pool_name = kwargs.pop("address_pool_name")
ap = self.get_address_pool(address_pool_name)
ip = ap.get_address()
if not ip:
raise Exception("AddressPool '%s' has run out of addresses." % ap.name)
t = VRouterTenant(provider_service=self, **kwargs)
t.public_ip = ip
t.public_mac = self.ip_to_mac(ip)
t.address_pool_id = ap.id
t.save()
return t
pass
class VRouterTenant(Tenant):
KIND = "vROUTER"
class Meta:
app_label = "vrouter"
name = "vrouter"
verbose_name = "vRouter Service"
# Primitive Fields (Not Relations)
public_ip = StrippedCharField( db_index = False, max_length = 30, null = True, blank = True )
public_mac = StrippedCharField( db_index = False, max_length = 30, null = True, blank = True )
# Relations
address_pool = ForeignKey(AddressPool, db_index = True, related_name = 'vrouter_tenants', null = True, blank = True )
@property
def gateway_ip(self):
if not self.address_pool:
return None
return self.address_pool.gateway_ip
@property
def gateway_mac(self):
if not self.address_pool:
return None
return self.address_pool.gateway_mac
@property
def cidr(self):
if not self.address_pool:
return None
return self.address_pool.cidr
@property
def netbits(self):
# return number of bits in the network portion of the cidr
if self.cidr:
parts = self.cidr.split("/")
if len(parts) == 2:
return int(parts[1].strip())
return None
def cleanup_addresspool(self):
if self.address_pool:
ap = self.address_pool
if ap:
ap[0].put_address(self.public_ip)
self.public_ip = None
def delete(self, *args, **kwargs):
self.cleanup_addresspool()
super(VRouterTenant, self).delete(*args, **kwargs)
pass