blob: 5e4883788b846792872c9f66d5f73beb8a82bbe4 [file] [log] [blame]
import hashlib
import os
import socket
import sys
import base64
import time
from django.db.models import F, Q
from xos.config import Config
from synchronizers.base.syncstep import SyncStep
from synchronizers.base.ansible import run_template_ssh
from synchronizers.base.SyncInstanceUsingAnsible import SyncInstanceUsingAnsible
from core.models import Service, Slice, Tag
from services.cord.models import VSGService, VSGTenant, VOLTTenant
from services.hpc.models import HpcService, CDNPrefix
from xos.logger import Logger, logging
# hpclibrary will be in steps/..
parentdir = os.path.join(os.path.dirname(__file__),"..")
sys.path.insert(0,parentdir)
from broadbandshield import BBS
logger = Logger(level=logging.INFO)
PARENTAL_MECHANISM="dnsmasq"
ENABLE_QUICK_UPDATE=False
CORD_USE_VTN = getattr(Config(), "networking_use_vtn", False)
class SyncVSGTenant(SyncInstanceUsingAnsible):
provides=[VSGTenant]
observes=VSGTenant
requested_interval=0
template_name = "sync_vcpetenant.yaml"
service_key_name = "/opt/xos/synchronizers/vcpe/vcpe_private_key"
def __init__(self, *args, **kwargs):
super(SyncVSGTenant, self).__init__(*args, **kwargs)
def fetch_pending(self, deleted):
if (not deleted):
objs = VSGTenant.get_tenant_objects().filter(Q(enacted__lt=F('updated')) | Q(enacted=None),Q(lazy_blocked=False))
else:
objs = VSGTenant.get_deleted_tenant_objects()
return objs
def get_vcpe_service(self, o):
if not o.provider_service:
return None
vcpes = VSGService.get_service_objects().filter(id=o.provider_service.id)
if not vcpes:
return None
return vcpes[0]
def get_extra_attributes(self, o):
# This is a place to include extra attributes that aren't part of the
# object itself. In the case of vCPE, we need to know:
# 1) the addresses of dnsdemux, to setup dnsmasq in the vCPE
# 2) CDN prefixes, so we know what URLs to send to dnsdemux
# 3) BroadBandShield server addresses, for parental filtering
# 4) vlan_ids, for setting up networking in the vCPE VM
vcpe_service = self.get_vcpe_service(o)
dnsdemux_ip = None
if vcpe_service.backend_network_label:
# Connect to dnsdemux using the network specified by
# vcpe_service.backend_network_label
for service in HpcService.objects.all():
for slice in service.slices.all():
if "dnsdemux" in slice.name:
for instance in slice.instances.all():
for ns in instance.ports.all():
if ns.ip and ns.network.labels and (vcpe_service.backend_network_label in ns.network.labels):
dnsdemux_ip = ns.ip
if not dnsdemux_ip:
logger.info("failed to find a dnsdemux on network %s" % vcpe_service.backend_network_label)
else:
# Connect to dnsdemux using the instance's public address
for service in HpcService.objects.all():
for slice in service.slices.all():
if "dnsdemux" in slice.name:
for instance in slice.instances.all():
if dnsdemux_ip=="none":
try:
dnsdemux_ip = socket.gethostbyname(instance.node.name)
except:
pass
if not dnsdemux_ip:
logger.info("failed to find a dnsdemux with a public address")
dnsdemux_ip = dnsdemux_ip or "none"
cdn_prefixes = []
for prefix in CDNPrefix.objects.all():
cdn_prefixes.append(prefix.prefix)
# Broadbandshield can either be set up internally, using vcpe_service.bbs_slice,
# or it can be setup externally using vcpe_service.bbs_server.
bbs_addrs = []
if vcpe_service.bbs_slice:
if vcpe_service.backend_network_label:
for bbs_instance in vcpe_service.bbs_slice.instances.all():
for ns in bbs_instance.ports.all():
if ns.ip and ns.network.labels and (vcpe_service.backend_network_label in ns.network.labels):
bbs_addrs.append(ns.ip)
else:
logger.info("unsupported configuration -- bbs_slice is set, but backend_network_label is not")
if not bbs_addrs:
logger.info("failed to find any usable addresses on bbs_slice")
elif vcpe_service.bbs_server:
bbs_addrs.append(vcpe_service.bbs_server)
else:
logger.info("neither bbs_slice nor bbs_server is configured in the vCPE")
vlan_ids = []
s_tags = []
c_tags = []
if o.volt:
vlan_ids.append(o.volt.vlan_id) # XXX remove this
s_tags.append(o.volt.s_tag)
c_tags.append(o.volt.c_tag)
try:
full_setup = Config().observer_full_setup
except:
full_setup = True
safe_macs=[]
if o.volt and o.volt.subscriber:
for user in o.volt.subscriber.users:
level = user.get("level",None)
mac = user.get("mac",None)
if level in ["G", "PG"]:
if mac:
safe_macs.append(mac)
wan_vm_ip=""
wan_vm_mac=""
tags = Tag.select_by_content_object(o.instance).filter(name="vm_wan_addr")
if tags:
parts=tags[0].value.split(",")
if len(parts)!=3:
raise Exception("vm_wan_addr tag is malformed: %s" % value)
wan_vm_ip = parts[1]
wan_vm_mac = parts[2]
else:
if CORD_USE_VTN:
raise Exception("no vm_wan_addr tag for instance %s" % o.instance)
fields = {"vlan_ids": vlan_ids, # XXX remove this
"s_tags": s_tags,
"c_tags": c_tags,
"dnsdemux_ip": dnsdemux_ip,
"cdn_prefixes": cdn_prefixes,
"bbs_addrs": bbs_addrs,
"full_setup": full_setup,
"isolation": o.instance.isolation,
"wan_container_gateway_mac": vcpe_service.wan_container_gateway_mac,
"wan_container_gateway_ip": vcpe_service.wan_container_gateway_ip,
"wan_container_netbits": vcpe_service.wan_container_netbits,
"wan_vm_mac": wan_vm_mac,
"wan_vm_ip": wan_vm_ip,
"safe_browsing_macs": safe_macs,
"dns_servers": [x.strip() for x in vcpe_service.dns_servers.split(",")] }
# add in the sync_attributes that come from the SubscriberRoot object
if o.volt and o.volt.subscriber and hasattr(o.volt.subscriber, "sync_attributes"):
for attribute_name in o.volt.subscriber.sync_attributes:
fields[attribute_name] = getattr(o.volt.subscriber, attribute_name)
return fields
def sync_fields(self, o, fields):
# the super causes the playbook to be run
super(SyncVSGTenant, self).sync_fields(o, fields)
# now do all of our broadbandshield stuff...
service = self.get_vcpe_service(o)
if not service:
# Ansible uses the service's keypair in order to SSH into the
# instance. It would be bad if the slice had no service.
raise Exception("Slice %s is not associated with a service" % instance.slice.name)
# Make sure the slice is configured properly
if (service != o.instance.slice.service):
raise Exception("Slice %s is associated with some service that is not %s" % (str(instance.slice), str(service)))
# only enable filtering if we have a subscriber object (see below)
url_filter_enable = False
# for attributes that come from CordSubscriberRoot
if o.volt and o.volt.subscriber:
url_filter_enable = o.volt.subscriber.url_filter_enable
url_filter_level = o.volt.subscriber.url_filter_level
url_filter_users = o.volt.subscriber.users
if PARENTAL_MECHANISM=="broadbandshield":
# disable url_filter if there are no bbs_addrs
if url_filter_enable and (not fields.get("bbs_addrs",[])):
logger.info("disabling url_filter because there are no bbs_addrs")
url_filter_enable = False
if url_filter_enable:
bbs_hostname = None
if service.bbs_api_hostname and service.bbs_api_port:
bbs_hostname = service.bbs_api_hostname
else:
# TODO: extract from slice
bbs_hostname = "cordcompute01.onlab.us"
if service.bbs_api_port:
bbs_port = service.bbs_api_port
else:
bbs_port = 8018
if not bbs_hostname:
logger.info("broadbandshield is not configured")
else:
tStart = time.time()
bbs = BBS(o.bbs_account, "123", bbs_hostname, bbs_port)
bbs.sync(url_filter_level, url_filter_users)
if o.hpc_client_ip:
logger.info("associate account %s with ip %s" % (o.bbs_account, o.hpc_client_ip))
bbs.associate(o.hpc_client_ip)
else:
logger.info("no hpc_client_ip to associate")
logger.info("bbs update time %d" % int(time.time()-tStart))
def run_playbook(self, o, fields):
ansible_hash = hashlib.md5(repr(sorted(fields.items()))).hexdigest()
quick_update = (o.last_ansible_hash == ansible_hash)
if ENABLE_QUICK_UPDATE and quick_update:
logger.info("quick_update triggered; skipping ansible recipe")
else:
if o.instance.isolation in ["container", "container_vm"]:
super(SyncVSGTenant, self).run_playbook(o, fields, "sync_vcpetenant_new.yaml")
else:
if CORD_USE_VTN:
super(SyncVSGTenant, self).run_playbook(o, fields, template_name="sync_vcpetenant_vtn.yaml")
else:
super(SyncVSGTenant, self).run_playbook(o, fields)
o.last_ansible_hash = ansible_hash
def delete_record(self, m):
pass