blob: d5daf42c784588e1e42767992f9c27f3b22e82fb [file] [log] [blame]
peusterme26487b2016-03-08 14:00:21 +01001"""
peusterm79ef6ae2016-07-08 13:53:57 +02002Copyright (c) 2015 SONATA-NFV and Paderborn University
3ALL RIGHTS RESERVED.
4
5Licensed under the Apache License, Version 2.0 (the "License");
6you may not use this file except in compliance with the License.
7You may obtain a copy of the License at
8
9 http://www.apache.org/licenses/LICENSE-2.0
10
11Unless required by applicable law or agreed to in writing, software
12distributed under the License is distributed on an "AS IS" BASIS,
13WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14See the License for the specific language governing permissions and
15limitations under the License.
16
17Neither the name of the SONATA-NFV [, ANY ADDITIONAL AFFILIATION]
18nor the names of its contributors may be used to endorse or promote
19products derived from this software without specific prior written
20permission.
21
22This work has been performed in the framework of the SONATA project,
23funded by the European Commission under Grant number 671517 through
24the Horizon 2020 and 5G-PPP programmes. The authors would like to
25acknowledge the contributions of their colleagues of the SONATA
26partner consortium (www.sonata-nfv.eu).
27"""
28"""
peusterme26487b2016-03-08 14:00:21 +010029This module implements a simple REST API that behaves like SONATA's gatekeeper.
30
31It is only used to support the development of SONATA's SDK tools and to demonstrate
32the year 1 version of the emulator until the integration with WP4's orchestrator is done.
33"""
34
35import logging
36import os
37import uuid
38import hashlib
peusterm786cd542016-03-14 14:12:17 +010039import zipfile
peusterm7ec665d2016-03-14 15:20:44 +010040import yaml
peusterme66edf72016-08-23 11:11:12 +020041import threading
stevenvanrossem8a9df3f2017-01-27 22:35:04 +010042from docker import DockerClient
peusterme26487b2016-03-08 14:00:21 +010043from flask import Flask, request
44import flask_restful as fr
wtaverni5b23b662016-06-20 12:26:21 +020045from collections import defaultdict
stevenvanrossemdb2f9432016-08-20 00:01:11 +020046import pkg_resources
peusterme26487b2016-03-08 14:00:21 +010047
peusterm398cd3b2016-03-21 15:04:54 +010048logging.basicConfig()
peusterm786cd542016-03-14 14:12:17 +010049LOG = logging.getLogger("sonata-dummy-gatekeeper")
50LOG.setLevel(logging.DEBUG)
peusterme26487b2016-03-08 14:00:21 +010051logging.getLogger("werkzeug").setLevel(logging.WARNING)
52
peusterm92237dc2016-03-21 15:45:58 +010053GK_STORAGE = "/tmp/son-dummy-gk/"
54UPLOAD_FOLDER = os.path.join(GK_STORAGE, "uploads/")
55CATALOG_FOLDER = os.path.join(GK_STORAGE, "catalog/")
peusterme26487b2016-03-08 14:00:21 +010056
peusterm82d406e2016-05-02 20:52:06 +020057# Enable Dockerfile build functionality
58BUILD_DOCKERFILE = False
59
peusterm398cd3b2016-03-21 15:04:54 +010060# flag to indicate that we run without the emulator (only the bare API for integration testing)
61GK_STANDALONE_MODE = False
62
peusterm56356cb2016-05-03 10:43:43 +020063# should a new version of an image be pulled even if its available
wtaverni5b23b662016-06-20 12:26:21 +020064FORCE_PULL = False
peusterme26487b2016-03-08 14:00:21 +010065
stevenvanrossemdb2f9432016-08-20 00:01:11 +020066# Automatically deploy SAPs (endpoints) of the service as new containers
peustermb1cf5372016-08-23 14:02:09 +020067# Attention: This is not a configuration switch but a global variable! Don't change its default value.
stevenvanrossemdb2f9432016-08-20 00:01:11 +020068DEPLOY_SAP = False
69
peusterm76eb8652016-09-06 11:07:16 +020070# flag to indicate if we use bidirectional forwarding rules in the automatic chaining process
71BIDIRECTIONAL_CHAIN = False
72
peusterme26487b2016-03-08 14:00:21 +010073class Gatekeeper(object):
74
75 def __init__(self):
peusterm786cd542016-03-14 14:12:17 +010076 self.services = dict()
peusterm082378b2016-03-16 20:14:22 +010077 self.dcs = dict()
stevenvanrossembecc7c52016-11-07 05:52:01 +010078 self.net = None
peusterm3444ae42016-03-16 20:46:41 +010079 self.vnf_counter = 0 # used to generate short names for VNFs (Mininet limitation)
peusterm786cd542016-03-14 14:12:17 +010080 LOG.info("Create SONATA dummy gatekeeper.")
peusterme26487b2016-03-08 14:00:21 +010081
peusterm786cd542016-03-14 14:12:17 +010082 def register_service_package(self, service_uuid, service):
83 """
84 register new service package
85 :param service_uuid
86 :param service object
87 """
88 self.services[service_uuid] = service
89 # lets perform all steps needed to onboard the service
90 service.onboard()
91
peusterm3444ae42016-03-16 20:46:41 +010092 def get_next_vnf_name(self):
93 self.vnf_counter += 1
peusterm398cd3b2016-03-21 15:04:54 +010094 return "vnf%d" % self.vnf_counter
peusterm3444ae42016-03-16 20:46:41 +010095
peusterm786cd542016-03-14 14:12:17 +010096
97class Service(object):
98 """
99 This class represents a NS uploaded as a *.son package to the
100 dummy gatekeeper.
101 Can have multiple running instances of this service.
102 """
103
104 def __init__(self,
105 service_uuid,
106 package_file_hash,
107 package_file_path):
108 self.uuid = service_uuid
109 self.package_file_hash = package_file_hash
110 self.package_file_path = package_file_path
111 self.package_content_path = os.path.join(CATALOG_FOLDER, "services/%s" % self.uuid)
peusterm7ec665d2016-03-14 15:20:44 +0100112 self.manifest = None
113 self.nsd = None
114 self.vnfds = dict()
peustermbdfab7e2016-03-14 16:03:30 +0100115 self.local_docker_files = dict()
peusterm82d406e2016-05-02 20:52:06 +0200116 self.remote_docker_image_urls = dict()
peusterm786cd542016-03-14 14:12:17 +0100117 self.instances = dict()
peusterm6b5224d2016-07-20 13:20:31 +0200118 self.vnf_name2docker_name = dict()
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200119 self.sap_identifiers = set()
peusterm6b5224d2016-07-20 13:20:31 +0200120 # lets generate a set of subnet configurations used for e-line chaining setup
121 self.eline_subnets_src = generate_subnet_strings(50, start=200, subnet_size=24, ip=1)
122 self.eline_subnets_dst = generate_subnet_strings(50, start=200, subnet_size=24, ip=2)
peusterme26487b2016-03-08 14:00:21 +0100123
peusterm786cd542016-03-14 14:12:17 +0100124 def onboard(self):
125 """
126 Do all steps to prepare this service to be instantiated
127 :return:
128 """
129 # 1. extract the contents of the package and store them in our catalog
130 self._unpack_service_package()
131 # 2. read in all descriptor files
peusterm7ec665d2016-03-14 15:20:44 +0100132 self._load_package_descriptor()
133 self._load_nsd()
134 self._load_vnfd()
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200135 if DEPLOY_SAP:
136 self._load_saps()
peusterm786cd542016-03-14 14:12:17 +0100137 # 3. prepare container images (e.g. download or build Dockerfile)
peusterm82d406e2016-05-02 20:52:06 +0200138 if BUILD_DOCKERFILE:
139 self._load_docker_files()
140 self._build_images_from_dockerfiles()
141 else:
142 self._load_docker_urls()
143 self._pull_predefined_dockerimages()
peusterm3bb86bf2016-08-15 09:47:57 +0200144 LOG.info("On-boarded service: %r" % self.manifest.get("name"))
peusterm7ec665d2016-03-14 15:20:44 +0100145
peusterm082378b2016-03-16 20:14:22 +0100146 def start_service(self):
peusterm3444ae42016-03-16 20:46:41 +0100147 """
148 This methods creates and starts a new service instance.
149 It computes placements, iterates over all VNFDs, and starts
150 each VNFD as a Docker container in the data center selected
151 by the placement algorithm.
152 :return:
153 """
154 LOG.info("Starting service %r" % self.uuid)
stevenvanrossemd87fe472016-05-11 11:34:34 +0200155
peusterm3444ae42016-03-16 20:46:41 +0100156 # 1. each service instance gets a new uuid to identify it
peusterm082378b2016-03-16 20:14:22 +0100157 instance_uuid = str(uuid.uuid4())
peusterm3444ae42016-03-16 20:46:41 +0100158 # build a instances dict (a bit like a NSR :))
159 self.instances[instance_uuid] = dict()
160 self.instances[instance_uuid]["vnf_instances"] = list()
stevenvanrossemd87fe472016-05-11 11:34:34 +0200161
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200162 # 2. Configure the chaining of the network functions (currently only E-Line and E-LAN links supported)
163 vnf_id2vnf_name = defaultdict(lambda: "NotExistingNode",
164 reduce(lambda x, y: dict(x, **y),
165 map(lambda d: {d["vnf_id"]: d["vnf_name"]},
166 self.nsd["network_functions"])))
167
168 # 3. compute placement of this service instance (adds DC names to VNFDs)
peusterm398cd3b2016-03-21 15:04:54 +0100169 if not GK_STANDALONE_MODE:
peustermf6459542016-08-31 19:00:17 +0200170 #self._calculate_placement(FirstDcPlacement)
171 self._calculate_placement(RoundRobinDcPlacement)
peusterm3444ae42016-03-16 20:46:41 +0100172 # iterate over all vnfds that we have to start
peusterm082378b2016-03-16 20:14:22 +0100173 for vnfd in self.vnfds.itervalues():
peusterm398cd3b2016-03-21 15:04:54 +0100174 vnfi = None
175 if not GK_STANDALONE_MODE:
176 vnfi = self._start_vnfd(vnfd)
177 self.instances[instance_uuid]["vnf_instances"].append(vnfi)
stevenvanrossemd87fe472016-05-11 11:34:34 +0200178
edmaasf5d0cbe2016-12-11 15:12:26 +0100179 if "virtual_links" in self.nsd:
180 vlinks = self.nsd["virtual_links"]
181 fwd_links = self.nsd["forwarding_graphs"][0]["constituent_virtual_links"]
182 eline_fwd_links = [l for l in vlinks if (l["id"] in fwd_links) and (l["connectivity_type"] == "E-Line")]
183 elan_fwd_links = [l for l in vlinks if (l["id"] in fwd_links) and (l["connectivity_type"] == "E-LAN")]
stevenvanrossemd87fe472016-05-11 11:34:34 +0200184
stevenvanrossem9cc73602017-01-27 23:37:29 +0100185 GK.net.deployed_elines.extend(eline_fwd_links)
186 GK.net.deployed_elans.extend(elan_fwd_links)
stevenvanrossembecc7c52016-11-07 05:52:01 +0100187
edmaasf5d0cbe2016-12-11 15:12:26 +0100188 # 4a. deploy E-Line links
189 # cookie is used as identifier for the flowrules installed by the dummygatekeeper
190 # eg. different services get a unique cookie for their flowrules
191 cookie = 1
192 for link in eline_fwd_links:
193 src_id, src_if_name = link["connection_points_reference"][0].split(":")
194 dst_id, dst_if_name = link["connection_points_reference"][1].split(":")
stevenvanrossemd87fe472016-05-11 11:34:34 +0200195
edmaasf5d0cbe2016-12-11 15:12:26 +0100196 # check if there is a SAP in the link
197 if src_id in self.sap_identifiers:
198 src_docker_name = "{0}_{1}".format(src_id, src_if_name)
199 src_id = src_docker_name
stevenvanrossemdc3bfd02016-11-04 15:33:28 +0100200 else:
edmaasf5d0cbe2016-12-11 15:12:26 +0100201 src_docker_name = src_id
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200202
edmaasf5d0cbe2016-12-11 15:12:26 +0100203 if dst_id in self.sap_identifiers:
204 dst_docker_name = "{0}_{1}".format(dst_id, dst_if_name)
205 dst_id = dst_docker_name
206 else:
207 dst_docker_name = dst_id
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200208
edmaasf5d0cbe2016-12-11 15:12:26 +0100209 src_name = vnf_id2vnf_name[src_id]
210 dst_name = vnf_id2vnf_name[dst_id]
peusterm9fb74ec2016-06-16 11:30:55 +0200211
stevenvanrossem6d5019a2016-08-12 23:00:22 +0200212 LOG.debug(
edmaasf5d0cbe2016-12-11 15:12:26 +0100213 "Setting up E-Line link. %s(%s:%s) -> %s(%s:%s)" % (
214 src_name, src_id, src_if_name, dst_name, dst_id, dst_if_name))
stevenvanrossem6d5019a2016-08-12 23:00:22 +0200215
edmaasf5d0cbe2016-12-11 15:12:26 +0100216 if (src_name in self.vnfds) and (dst_name in self.vnfds):
217 network = self.vnfds[src_name].get("dc").net # there should be a cleaner way to find the DCNetwork
218 LOG.debug(src_docker_name)
219 ret = network.setChain(
220 src_docker_name, dst_docker_name,
221 vnf_src_interface=src_if_name, vnf_dst_interface=dst_if_name,
222 bidirectional=BIDIRECTIONAL_CHAIN, cmd="add-flow", cookie=cookie, priority=10)
stevenvanrossemdc3bfd02016-11-04 15:33:28 +0100223
edmaasf5d0cbe2016-12-11 15:12:26 +0100224 # re-configure the VNFs IP assignment and ensure that a new subnet is used for each E-Link
225 src_vnfi = self._get_vnf_instance(instance_uuid, src_name)
226 if src_vnfi is not None:
227 self._vnf_reconfigure_network(src_vnfi, src_if_name, self.eline_subnets_src.pop(0))
228 dst_vnfi = self._get_vnf_instance(instance_uuid, dst_name)
229 if dst_vnfi is not None:
230 self._vnf_reconfigure_network(dst_vnfi, dst_if_name, self.eline_subnets_dst.pop(0))
231
232 # 4b. deploy E-LAN links
233 base = 10
234 for link in elan_fwd_links:
stevenvanrossem9cc73602017-01-27 23:37:29 +0100235
236 elan_vnf_list=[]
237
edmaasf5d0cbe2016-12-11 15:12:26 +0100238 # generate lan ip address
239 ip = 1
240 for intf in link["connection_points_reference"]:
241 ip_address = generate_lan_string("10.0", base, subnet_size=24, ip=ip)
242 vnf_id, intf_name = intf.split(":")
243 if vnf_id in self.sap_identifiers:
244 src_docker_name = "{0}_{1}".format(vnf_id, intf_name)
245 vnf_id = src_docker_name
stevenvanrossem9cc73602017-01-27 23:37:29 +0100246 else:
247 src_docker_name = vnf_id
edmaasf5d0cbe2016-12-11 15:12:26 +0100248 vnf_name = vnf_id2vnf_name[vnf_id]
249 LOG.debug(
250 "Setting up E-LAN link. %s(%s:%s) -> %s" % (
251 vnf_name, vnf_id, intf_name, ip_address))
252
253 if vnf_name in self.vnfds:
254 # re-configure the VNFs IP assignment and ensure that a new subnet is used for each E-LAN
255 # E-LAN relies on the learning switch capability of Ryu which has to be turned on in the topology
256 # (DCNetwork(controller=RemoteController, enable_learning=True)), so no explicit chaining is necessary.
257 vnfi = self._get_vnf_instance(instance_uuid, vnf_name)
258 if vnfi is not None:
259 self._vnf_reconfigure_network(vnfi, intf_name, ip_address)
260 # increase for the next ip address on this E-LAN
261 ip += 1
stevenvanrossem9cc73602017-01-27 23:37:29 +0100262
263 # add this vnf and interface to the E-LAN for tagging
264 network = self.vnfds[vnf_name].get("dc").net # there should be a cleaner way to find the DCNetwork
265 elan_vnf_list.append({'name':src_docker_name,'interface':intf_name})
stevenvanrossemdc3bfd02016-11-04 15:33:28 +0100266
267
stevenvanrossem9cc73602017-01-27 23:37:29 +0100268 # install the VLAN tags for this E-LAN
269 network.setLAN(elan_vnf_list)
edmaasf5d0cbe2016-12-11 15:12:26 +0100270 # increase the base ip address for the next E-LAN
271 base += 1
stevenvanrossem6d5019a2016-08-12 23:00:22 +0200272
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200273 # 5. run the emulator specific entrypoint scripts in the VNFIs of this service instance
peusterm8484b902016-06-21 09:03:35 +0200274 self._trigger_emulator_start_scripts_in_vnfis(self.instances[instance_uuid]["vnf_instances"])
275
peusterm3444ae42016-03-16 20:46:41 +0100276 LOG.info("Service started. Instance id: %r" % instance_uuid)
peusterm082378b2016-03-16 20:14:22 +0100277 return instance_uuid
278
edmaas9c4fd112016-10-05 19:45:57 +0200279 def stop_service(self, instance_uuid):
edmaasd454d542016-09-29 13:19:22 +0200280 """
281 This method stops a running service instance.
edmaas74d72492016-10-05 19:59:22 +0200282 It iterates over all VNF instances, stopping them each
edmaasd454d542016-09-29 13:19:22 +0200283 and removing them from their data center.
284
edmaas74d72492016-10-05 19:59:22 +0200285 :param instance_uuid: the uuid of the service instance to be stopped
edmaasd454d542016-09-29 13:19:22 +0200286 """
edmaas9c4fd112016-10-05 19:45:57 +0200287 LOG.info("Stopping service %r" % self.uuid)
288 # get relevant information
289 # instance_uuid = str(self.uuid.uuid4())
290 vnf_instances = self.instances[instance_uuid]["vnf_instances"]
291
292 for v in vnf_instances:
edmaas74d72492016-10-05 19:59:22 +0200293 self._stop_vnfi(v)
edmaas9c4fd112016-10-05 19:45:57 +0200294
295 if not GK_STANDALONE_MODE:
296 # remove placement?
297 # self._remove_placement(RoundRobinPlacement)
298 None
299
300 # last step: remove the instance from the list of all instances
301 del self.instances[instance_uuid]
edmaasd454d542016-09-29 13:19:22 +0200302
peusterm398cd3b2016-03-21 15:04:54 +0100303 def _start_vnfd(self, vnfd):
304 """
305 Start a single VNFD of this service
306 :param vnfd: vnfd descriptor dict
307 :return:
308 """
309 # iterate over all deployment units within each VNFDs
310 for u in vnfd.get("virtual_deployment_units"):
311 # 1. get the name of the docker image to start and the assigned DC
peusterm56356cb2016-05-03 10:43:43 +0200312 vnf_name = vnfd.get("name")
313 if vnf_name not in self.remote_docker_image_urls:
314 raise Exception("No image name for %r found. Abort." % vnf_name)
315 docker_name = self.remote_docker_image_urls.get(vnf_name)
peusterm398cd3b2016-03-21 15:04:54 +0100316 target_dc = vnfd.get("dc")
317 # 2. perform some checks to ensure we can start the container
318 assert(docker_name is not None)
319 assert(target_dc is not None)
320 if not self._check_docker_image_exists(docker_name):
321 raise Exception("Docker image %r not found. Abort." % docker_name)
edmaas7e084ea2016-11-28 13:50:23 +0100322
323 # 3. get the resource limits
324 res_req = u.get("resource_requirements")
325 cpu_list = res_req.get("cpu").get("cores")
326 if not cpu_list or len(cpu_list)==0:
327 cpu_list="1"
328 cpu_bw = res_req.get("cpu").get("cpu_bw")
329 if not cpu_bw:
330 cpu_bw=1
331 mem_num = str(res_req.get("memory").get("size"))
332 if len(mem_num)==0:
333 mem_num="2"
334 mem_unit = str(res_req.get("memory").get("size_unit"))
335 if str(mem_unit)==0:
336 mem_unit="GB"
337 mem_limit = float(mem_num)
338 if mem_unit=="GB":
339 mem_limit=mem_limit*1024*1024*1024
340 elif mem_unit=="MB":
341 mem_limit=mem_limit*1024*1024
342 elif mem_unit=="KB":
343 mem_limit=mem_limit*1024
344 mem_lim = int(mem_limit)
345 cpu_period, cpu_quota = self._calculate_cpu_cfs_values(float(cpu_bw))
346
347 # 4. do the dc.startCompute(name="foobar") call to run the container
peusterm398cd3b2016-03-21 15:04:54 +0100348 # TODO consider flavors, and other annotations
stevenvanrossemd87fe472016-05-11 11:34:34 +0200349 intfs = vnfd.get("connection_points")
stevenvanrossemeae73082016-08-05 16:22:12 +0200350
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200351 # TODO: get all vnf id's from the nsd for this vnfd and use those as dockername
stevenvanrossem11a021f2016-08-05 13:43:00 +0200352 # use the vnf_id in the nsd as docker name
353 # so deployed containers can be easily mapped back to the nsd
354 vnf_name2id = defaultdict(lambda: "NotExistingNode",
355 reduce(lambda x, y: dict(x, **y),
356 map(lambda d: {d["vnf_name"]: d["vnf_id"]},
357 self.nsd["network_functions"])))
358 self.vnf_name2docker_name[vnf_name] = vnf_name2id[vnf_name]
359 # self.vnf_name2docker_name[vnf_name] = GK.get_next_vnf_name()
360
peusterm6b5224d2016-07-20 13:20:31 +0200361 LOG.info("Starting %r as %r in DC %r" % (vnf_name, self.vnf_name2docker_name[vnf_name], vnfd.get("dc")))
peusterm761c14d2016-07-19 09:31:19 +0200362 LOG.debug("Interfaces for %r: %r" % (vnf_name, intfs))
edmaasf5d0cbe2016-12-11 15:12:26 +0100363 vnfi = target_dc.startCompute(self.vnf_name2docker_name[vnf_name], network=intfs, image=docker_name, flavor_name="small",
edmaas7e084ea2016-11-28 13:50:23 +0100364 cpu_quota=cpu_quota, cpu_period=cpu_period, cpuset=cpu_list, mem_limit=mem_lim)
peusterm398cd3b2016-03-21 15:04:54 +0100365 return vnfi
366
edmaas74d72492016-10-05 19:59:22 +0200367 def _stop_vnfi(self, vnfi):
edmaasd454d542016-09-29 13:19:22 +0200368 """
edmaas74d72492016-10-05 19:59:22 +0200369 Stop a VNF instance.
edmaasd454d542016-09-29 13:19:22 +0200370
edmaas74d72492016-10-05 19:59:22 +0200371 :param vnfi: vnf instance to be stopped
edmaasd454d542016-09-29 13:19:22 +0200372 """
edmaas9c4fd112016-10-05 19:45:57 +0200373 # Find the correct datacenter
374 status = vnfi.getStatus()
375 dc = vnfi.datacenter
376 # stop the vnfi
edmaas74d72492016-10-05 19:59:22 +0200377 LOG.info("Stopping the vnf instance contained in %r in DC %r" % (status["name"], dc))
edmaas9c4fd112016-10-05 19:45:57 +0200378 dc.stopCompute(status["name"])
edmaasd454d542016-09-29 13:19:22 +0200379
peusterm6b5224d2016-07-20 13:20:31 +0200380 def _get_vnf_instance(self, instance_uuid, name):
381 """
382 Returns the Docker object for the given VNF name (or Docker name).
383 :param instance_uuid: UUID of the service instance to search in.
384 :param name: VNF name or Docker name. We are fuzzy here.
385 :return:
386 """
387 dn = name
388 if name in self.vnf_name2docker_name:
389 dn = self.vnf_name2docker_name[name]
390 for vnfi in self.instances[instance_uuid]["vnf_instances"]:
391 if vnfi.name == dn:
392 return vnfi
393 LOG.warning("No container with name: %r found.")
394 return None
395
396 @staticmethod
397 def _vnf_reconfigure_network(vnfi, if_name, net_str):
398 """
399 Reconfigure the network configuration of a specific interface
400 of a running container.
401 :param vnfi: container instacne
402 :param if_name: interface name
403 :param net_str: network configuration string, e.g., 1.2.3.4/24
404 :return:
405 """
406 intf = vnfi.intf(intf=if_name)
407 if intf is not None:
408 intf.setIP(net_str)
409 LOG.debug("Reconfigured network of %s:%s to %r" % (vnfi.name, if_name, net_str))
410 else:
411 LOG.warning("Interface not found: %s:%s. Network reconfiguration skipped." % (vnfi.name, if_name))
412
413
peusterm8484b902016-06-21 09:03:35 +0200414 def _trigger_emulator_start_scripts_in_vnfis(self, vnfi_list):
415 for vnfi in vnfi_list:
416 config = vnfi.dcinfo.get("Config", dict())
417 env = config.get("Env", list())
418 for env_var in env:
edmaas7e084ea2016-11-28 13:50:23 +0100419 var, cmd = map(str.strip, map(str, env_var.split('=', 1)))
420 LOG.debug("%r = %r" % (var , cmd))
421 if var=="SON_EMU_CMD":
peusterme66edf72016-08-23 11:11:12 +0200422 LOG.info("Executing entry point script in %r: %r" % (vnfi.name, cmd))
423 # execute command in new thread to ensure that GK is not blocked by VNF
424 t = threading.Thread(target=vnfi.cmdPrint, args=(cmd,))
425 t.daemon = True
426 t.start()
peusterm8484b902016-06-21 09:03:35 +0200427
peusterm786cd542016-03-14 14:12:17 +0100428 def _unpack_service_package(self):
429 """
430 unzip *.son file and store contents in CATALOG_FOLDER/services/<service_uuid>/
431 """
peusterm82d406e2016-05-02 20:52:06 +0200432 LOG.info("Unzipping: %r" % self.package_file_path)
peusterm786cd542016-03-14 14:12:17 +0100433 with zipfile.ZipFile(self.package_file_path, "r") as z:
434 z.extractall(self.package_content_path)
435
peusterm82d406e2016-05-02 20:52:06 +0200436
peusterm7ec665d2016-03-14 15:20:44 +0100437 def _load_package_descriptor(self):
438 """
439 Load the main package descriptor YAML and keep it as dict.
440 :return:
441 """
442 self.manifest = load_yaml(
443 os.path.join(
444 self.package_content_path, "META-INF/MANIFEST.MF"))
445
446 def _load_nsd(self):
447 """
448 Load the entry NSD YAML and keep it as dict.
449 :return:
450 """
451 if "entry_service_template" in self.manifest:
452 nsd_path = os.path.join(
453 self.package_content_path,
454 make_relative_path(self.manifest.get("entry_service_template")))
455 self.nsd = load_yaml(nsd_path)
stevenvanrossembecc7c52016-11-07 05:52:01 +0100456 GK.net.deployed_nsds.append(self.nsd)
peusterm757fe9a2016-04-04 14:11:58 +0200457 LOG.debug("Loaded NSD: %r" % self.nsd.get("name"))
peusterm7ec665d2016-03-14 15:20:44 +0100458
459 def _load_vnfd(self):
460 """
461 Load all VNFD YAML files referenced in MANIFEST.MF and keep them in dict.
462 :return:
463 """
464 if "package_content" in self.manifest:
465 for pc in self.manifest.get("package_content"):
466 if pc.get("content-type") == "application/sonata.function_descriptor":
467 vnfd_path = os.path.join(
468 self.package_content_path,
469 make_relative_path(pc.get("name")))
470 vnfd = load_yaml(vnfd_path)
peusterm757fe9a2016-04-04 14:11:58 +0200471 self.vnfds[vnfd.get("name")] = vnfd
472 LOG.debug("Loaded VNFD: %r" % vnfd.get("name"))
peusterm7ec665d2016-03-14 15:20:44 +0100473
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200474 def _load_saps(self):
475 # Each Service Access Point (connection_point) in the nsd is getting its own container
476 SAPs = [p["id"] for p in self.nsd["connection_points"] if p["type"] == "interface"]
477 for sap in SAPs:
478 # endpoints needed in this service
479 sap_vnf_id, sap_vnf_interface = sap.split(':')
480 # set of the connection_point ids found in the nsd (in the examples this is 'ns')
481 self.sap_identifiers.add(sap_vnf_id)
482
peusterm76eb8652016-09-06 11:07:16 +0200483 sap_docker_name = "%s_%s" % (sap_vnf_id, sap_vnf_interface)
stevenvanrossemdb2f9432016-08-20 00:01:11 +0200484
485 # add SAP to self.vnfds
486 sapfile = pkg_resources.resource_filename(__name__, "sap_vnfd.yml")
487 sap_vnfd = load_yaml(sapfile)
488 sap_vnfd["connection_points"][0]["id"] = sap_vnf_interface
489 sap_vnfd["name"] = sap_docker_name
490 self.vnfds[sap_docker_name] = sap_vnfd
491 # add SAP vnf to list in the NSD so it is deployed later on
492 # each SAP get a unique VNFD and vnf_id in the NSD
493 self.nsd["network_functions"].append({"vnf_id": sap_docker_name, "vnf_name": sap_docker_name})
494 LOG.debug("Loaded SAP: %r" % sap_vnfd.get("name"))
495
peusterm7ec665d2016-03-14 15:20:44 +0100496 def _load_docker_files(self):
497 """
peusterm9d7d4b02016-03-23 19:56:44 +0100498 Get all paths to Dockerfiles from VNFDs and store them in dict.
peusterm7ec665d2016-03-14 15:20:44 +0100499 :return:
500 """
peusterm9d7d4b02016-03-23 19:56:44 +0100501 for k, v in self.vnfds.iteritems():
502 for vu in v.get("virtual_deployment_units"):
503 if vu.get("vm_image_format") == "docker":
504 vm_image = vu.get("vm_image")
peusterm7ec665d2016-03-14 15:20:44 +0100505 docker_path = os.path.join(
506 self.package_content_path,
peusterm9d7d4b02016-03-23 19:56:44 +0100507 make_relative_path(vm_image))
508 self.local_docker_files[k] = docker_path
peusterm56356cb2016-05-03 10:43:43 +0200509 LOG.debug("Found Dockerfile (%r): %r" % (k, docker_path))
peusterm7ec665d2016-03-14 15:20:44 +0100510
peusterm82d406e2016-05-02 20:52:06 +0200511 def _load_docker_urls(self):
512 """
513 Get all URLs to pre-build docker images in some repo.
514 :return:
515 """
516 for k, v in self.vnfds.iteritems():
517 for vu in v.get("virtual_deployment_units"):
518 if vu.get("vm_image_format") == "docker":
peusterm35ba4052016-05-02 21:21:14 +0200519 url = vu.get("vm_image")
520 if url is not None:
521 url = url.replace("http://", "")
522 self.remote_docker_image_urls[k] = url
peusterm56356cb2016-05-03 10:43:43 +0200523 LOG.debug("Found Docker image URL (%r): %r" % (k, self.remote_docker_image_urls[k]))
peusterm82d406e2016-05-02 20:52:06 +0200524
peustermbdfab7e2016-03-14 16:03:30 +0100525 def _build_images_from_dockerfiles(self):
526 """
527 Build Docker images for each local Dockerfile found in the package: self.local_docker_files
528 """
peusterm398cd3b2016-03-21 15:04:54 +0100529 if GK_STANDALONE_MODE:
530 return # do not build anything in standalone mode
peustermbdfab7e2016-03-14 16:03:30 +0100531 dc = DockerClient()
532 LOG.info("Building %d Docker images (this may take several minutes) ..." % len(self.local_docker_files))
533 for k, v in self.local_docker_files.iteritems():
534 for line in dc.build(path=v.replace("Dockerfile", ""), tag=k, rm=False, nocache=False):
535 LOG.debug("DOCKER BUILD: %s" % line)
536 LOG.info("Docker image created: %s" % k)
537
peusterm82d406e2016-05-02 20:52:06 +0200538 def _pull_predefined_dockerimages(self):
peustermbdfab7e2016-03-14 16:03:30 +0100539 """
540 If the package contains URLs to pre-build Docker images, we download them with this method.
541 """
peusterm35ba4052016-05-02 21:21:14 +0200542 dc = DockerClient()
543 for url in self.remote_docker_image_urls.itervalues():
peusterm56356cb2016-05-03 10:43:43 +0200544 if not FORCE_PULL: # only pull if not present (speedup for development)
stevenvanrossem8a9df3f2017-01-27 22:35:04 +0100545 if len(dc.images.list(name=url)) > 0:
peusterm56356cb2016-05-03 10:43:43 +0200546 LOG.debug("Image %r present. Skipping pull." % url)
547 continue
peusterm35ba4052016-05-02 21:21:14 +0200548 LOG.info("Pulling image: %r" % url)
549 dc.pull(url,
550 insecure_registry=True)
peusterm786cd542016-03-14 14:12:17 +0100551
peusterm3444ae42016-03-16 20:46:41 +0100552 def _check_docker_image_exists(self, image_name):
peusterm3f307142016-03-16 21:02:53 +0100553 """
554 Query the docker service and check if the given image exists
555 :param image_name: name of the docker image
556 :return:
557 """
stevenvanrossem8a9df3f2017-01-27 22:35:04 +0100558 return len(DockerClient().images.list(name=image_name)) > 0
peusterm3444ae42016-03-16 20:46:41 +0100559
peusterm082378b2016-03-16 20:14:22 +0100560 def _calculate_placement(self, algorithm):
561 """
562 Do placement by adding the a field "dc" to
563 each VNFD that points to one of our
564 data center objects known to the gatekeeper.
565 """
566 assert(len(self.vnfds) > 0)
567 assert(len(GK.dcs) > 0)
568 # instantiate algorithm an place
569 p = algorithm()
570 p.place(self.nsd, self.vnfds, GK.dcs)
571 LOG.info("Using placement algorithm: %r" % p.__class__.__name__)
572 # lets print the placement result
573 for name, vnfd in self.vnfds.iteritems():
574 LOG.info("Placed VNF %r on DC %r" % (name, str(vnfd.get("dc"))))
575
edmaas7e084ea2016-11-28 13:50:23 +0100576 def _calculate_cpu_cfs_values(self, cpu_time_percentage):
577 """
578 Calculate cpu period and quota for CFS
579 :param cpu_time_percentage: percentage of overall CPU to be used
580 :return: cpu_period, cpu_quota
581 """
582 if cpu_time_percentage is None:
583 return -1, -1
584 if cpu_time_percentage < 0:
585 return -1, -1
586 # (see: https://www.kernel.org/doc/Documentation/scheduler/sched-bwc.txt)
587 # Attention minimum cpu_quota is 1ms (micro)
588 cpu_period = 1000000 # lets consider a fixed period of 1000000 microseconds for now
589 LOG.debug("cpu_period is %r, cpu_percentage is %r" % (cpu_period, cpu_time_percentage))
590 cpu_quota = cpu_period * cpu_time_percentage # calculate the fraction of cpu time for this container
591 # ATTENTION >= 1000 to avoid a invalid argument system error ... no idea why
592 if cpu_quota < 1000:
593 LOG.debug("cpu_quota before correcting: %r" % cpu_quota)
594 cpu_quota = 1000
595 LOG.warning("Increased CPU quota to avoid system error.")
596 LOG.debug("Calculated: cpu_period=%f / cpu_quota=%f" % (cpu_period, cpu_quota))
597 return int(cpu_period), int(cpu_quota)
598
peusterm082378b2016-03-16 20:14:22 +0100599
600"""
601Some (simple) placement algorithms
602"""
603
604
605class FirstDcPlacement(object):
606 """
607 Placement: Always use one and the same data center from the GK.dcs dict.
608 """
609 def place(self, nsd, vnfds, dcs):
610 for name, vnfd in vnfds.iteritems():
611 vnfd["dc"] = list(dcs.itervalues())[0]
612
peusterme26487b2016-03-08 14:00:21 +0100613
peustermf6459542016-08-31 19:00:17 +0200614class RoundRobinDcPlacement(object):
615 """
616 Placement: Distribute VNFs across all available DCs in a round robin fashion.
617 """
peustermf6459542016-08-31 19:00:17 +0200618 def place(self, nsd, vnfds, dcs):
619 c = 0
edmaasd454d542016-09-29 13:19:22 +0200620 dcs_list = list(dcs.itervalues())
peustermf6459542016-08-31 19:00:17 +0200621 for name, vnfd in vnfds.iteritems():
622 vnfd["dc"] = dcs_list[c % len(dcs_list)]
623 c += 1 # inc. c to use next DC
624
625
626
627
peusterme26487b2016-03-08 14:00:21 +0100628"""
629Resource definitions and API endpoints
630"""
631
632
633class Packages(fr.Resource):
634
635 def post(self):
636 """
peusterm26455852016-03-08 14:23:53 +0100637 Upload a *.son service package to the dummy gatekeeper.
638
peusterme26487b2016-03-08 14:00:21 +0100639 We expect request with a *.son file and store it in UPLOAD_FOLDER
peusterm26455852016-03-08 14:23:53 +0100640 :return: UUID
peusterme26487b2016-03-08 14:00:21 +0100641 """
642 try:
643 # get file contents
wtavernib8d9ecb2016-03-25 15:18:31 +0100644 print(request.files)
peusterm593ca582016-03-30 19:55:01 +0200645 # lets search for the package in the request
646 if "package" in request.files:
647 son_file = request.files["package"]
648 # elif "file" in request.files:
649 # son_file = request.files["file"]
650 else:
651 return {"service_uuid": None, "size": 0, "sha1": None, "error": "upload failed. file not found."}, 500
peusterme26487b2016-03-08 14:00:21 +0100652 # generate a uuid to reference this package
653 service_uuid = str(uuid.uuid4())
peusterm786cd542016-03-14 14:12:17 +0100654 file_hash = hashlib.sha1(str(son_file)).hexdigest()
peusterme26487b2016-03-08 14:00:21 +0100655 # ensure that upload folder exists
656 ensure_dir(UPLOAD_FOLDER)
657 upload_path = os.path.join(UPLOAD_FOLDER, "%s.son" % service_uuid)
658 # store *.son file to disk
peusterm786cd542016-03-14 14:12:17 +0100659 son_file.save(upload_path)
peusterme26487b2016-03-08 14:00:21 +0100660 size = os.path.getsize(upload_path)
peusterm786cd542016-03-14 14:12:17 +0100661 # create a service object and register it
662 s = Service(service_uuid, file_hash, upload_path)
663 GK.register_service_package(service_uuid, s)
peusterme26487b2016-03-08 14:00:21 +0100664 # generate the JSON result
peusterm938143e2016-09-15 15:39:36 +0200665 return {"service_uuid": service_uuid, "size": size, "sha1": file_hash, "error": None}, 201
peusterme26487b2016-03-08 14:00:21 +0100666 except Exception as ex:
peusterm786cd542016-03-14 14:12:17 +0100667 LOG.exception("Service package upload failed:")
peusterm593ca582016-03-30 19:55:01 +0200668 return {"service_uuid": None, "size": 0, "sha1": None, "error": "upload failed"}, 500
peusterme26487b2016-03-08 14:00:21 +0100669
670 def get(self):
peusterm26455852016-03-08 14:23:53 +0100671 """
672 Return a list of UUID's of uploaded service packages.
673 :return: dict/list
674 """
peusterm075b46a2016-07-20 17:08:00 +0200675 LOG.info("GET /packages")
peusterm786cd542016-03-14 14:12:17 +0100676 return {"service_uuid_list": list(GK.services.iterkeys())}
peusterme26487b2016-03-08 14:00:21 +0100677
678
679class Instantiations(fr.Resource):
680
681 def post(self):
peusterm26455852016-03-08 14:23:53 +0100682 """
683 Instantiate a service specified by its UUID.
684 Will return a new UUID to identify the running service instance.
685 :return: UUID
686 """
peusterm64b45502016-03-16 21:15:14 +0100687 # try to extract the service uuid from the request
peusterm26455852016-03-08 14:23:53 +0100688 json_data = request.get_json(force=True)
peusterm64b45502016-03-16 21:15:14 +0100689 service_uuid = json_data.get("service_uuid")
690
691 # lets be a bit fuzzy here to make testing easier
692 if service_uuid is None and len(GK.services) > 0:
693 # if we don't get a service uuid, we simple start the first service in the list
694 service_uuid = list(GK.services.iterkeys())[0]
695
peustermbea87372016-03-16 19:37:35 +0100696 if service_uuid in GK.services:
peusterm64b45502016-03-16 21:15:14 +0100697 # ok, we have a service uuid, lets start the service
peustermbea87372016-03-16 19:37:35 +0100698 service_instance_uuid = GK.services.get(service_uuid).start_service()
edmaas59b28fc2016-11-01 17:11:47 +0100699 return {"service_instance_uuid": service_instance_uuid}, 201
peustermbea87372016-03-16 19:37:35 +0100700 return "Service not found", 404
peusterme26487b2016-03-08 14:00:21 +0100701
702 def get(self):
peusterm26455852016-03-08 14:23:53 +0100703 """
704 Returns a list of UUIDs containing all running services.
705 :return: dict / list
706 """
peusterm075b46a2016-07-20 17:08:00 +0200707 LOG.info("GET /instantiations")
708 return {"service_instantiations_list": [
peusterm64b45502016-03-16 21:15:14 +0100709 list(s.instances.iterkeys()) for s in GK.services.itervalues()]}
peusterm786cd542016-03-14 14:12:17 +0100710
edmaasd454d542016-09-29 13:19:22 +0200711 def delete(self):
712 """
edmaas74d72492016-10-05 19:59:22 +0200713 Stops a running service specified by its service and instance UUID.
edmaasd454d542016-09-29 13:19:22 +0200714 """
edmaas74d72492016-10-05 19:59:22 +0200715 # try to extract the service and instance UUID from the request
edmaasd454d542016-09-29 13:19:22 +0200716 json_data = request.get_json(force=True)
717 service_uuid = json_data.get("service_uuid")
edmaas59b28fc2016-11-01 17:11:47 +0100718 instance_uuid = json_data.get("service_instance_uuid")
edmaas9c4fd112016-10-05 19:45:57 +0200719
720 # try to be fuzzy
721 if service_uuid is None and len(GK.services) > 0:
722 #if we don't get a service uuid, we simply stop the last service in the list
723 service_uuid = list(GK.services.iterkeys())[0]
724 if instance_uuid is None and len(GK.services[service_uuid].instances) > 0:
725 instance_uuid = list(GK.services[service_uuid].instances.iterkeys())[0]
edmaasd454d542016-09-29 13:19:22 +0200726
edmaas74d72492016-10-05 19:59:22 +0200727 if service_uuid in GK.services and instance_uuid in GK.services[service_uuid].instances:
728 # valid service and instance UUID, stop service
edmaas9c4fd112016-10-05 19:45:57 +0200729 GK.services.get(service_uuid).stop_service(instance_uuid)
edmaasf5d0cbe2016-12-11 15:12:26 +0100730 return "service instance with uuid %r stopped." % instance_uuid,200
edmaasd454d542016-09-29 13:19:22 +0200731 return "Service not found", 404
732
edmaas74d72492016-10-05 19:59:22 +0200733class Exit(fr.Resource):
edmaas9c4fd112016-10-05 19:45:57 +0200734
735 def put(self):
736 """
737 Stop the running Containernet instance regardless of data transmitted
738 """
edmaasf5d0cbe2016-12-11 15:12:26 +0100739 list(GK.dcs.values())[0].net.stop()
edmaas59b28fc2016-11-01 17:11:47 +0100740
741
742def initialize_GK():
743 global GK
744 GK = Gatekeeper()
745
edmaas9c4fd112016-10-05 19:45:57 +0200746
peusterme26487b2016-03-08 14:00:21 +0100747
748# create a single, global GK object
edmaas59b28fc2016-11-01 17:11:47 +0100749GK = None
750initialize_GK()
peusterme26487b2016-03-08 14:00:21 +0100751# setup Flask
752app = Flask(__name__)
753app.config['MAX_CONTENT_LENGTH'] = 512 * 1024 * 1024 # 512 MB max upload
754api = fr.Api(app)
755# define endpoints
peusterm593ca582016-03-30 19:55:01 +0200756api.add_resource(Packages, '/packages')
757api.add_resource(Instantiations, '/instantiations')
edmaas74d72492016-10-05 19:59:22 +0200758api.add_resource(Exit, '/emulator/exit')
peusterme26487b2016-03-08 14:00:21 +0100759
760
edmaas59b28fc2016-11-01 17:11:47 +0100761#def initialize_GK():
762# global GK
763# GK = Gatekeeper()
peusterme26487b2016-03-08 14:00:21 +0100764
765
peusterm082378b2016-03-16 20:14:22 +0100766def start_rest_api(host, port, datacenters=dict()):
peustermbea87372016-03-16 19:37:35 +0100767 GK.dcs = datacenters
stevenvanrossembecc7c52016-11-07 05:52:01 +0100768 GK.net = get_dc_network()
peusterme26487b2016-03-08 14:00:21 +0100769 # start the Flask server (not the best performance but ok for our use case)
770 app.run(host=host,
771 port=port,
772 debug=True,
773 use_reloader=False # this is needed to run Flask in a non-main thread
774 )
775
776
777def ensure_dir(name):
778 if not os.path.exists(name):
peusterm7ec665d2016-03-14 15:20:44 +0100779 os.makedirs(name)
780
781
782def load_yaml(path):
783 with open(path, "r") as f:
784 try:
785 r = yaml.load(f)
786 except yaml.YAMLError as exc:
787 LOG.exception("YAML parse error")
788 r = dict()
789 return r
790
791
792def make_relative_path(path):
peusterm9d7d4b02016-03-23 19:56:44 +0100793 if path.startswith("file://"):
794 path = path.replace("file://", "", 1)
peusterm7ec665d2016-03-14 15:20:44 +0100795 if path.startswith("/"):
peusterm9d7d4b02016-03-23 19:56:44 +0100796 path = path.replace("/", "", 1)
peusterm7ec665d2016-03-14 15:20:44 +0100797 return path
798
799
stevenvanrossem6d5019a2016-08-12 23:00:22 +0200800def generate_lan_string(prefix, base, subnet_size=24, ip=0):
801 """
802 Helper to generate different network configuration strings.
803 """
804 r = "%s.%d.%d/%d" % (prefix, base, ip, subnet_size)
805 return r
806
807
peusterm6b5224d2016-07-20 13:20:31 +0200808def generate_subnet_strings(n, start=1, subnet_size=24, ip=0):
809 """
810 Helper to generate different network configuration strings.
811 """
812 r = list()
813 for i in range(start, start + n):
814 r.append("%d.0.0.%d/%d" % (i, ip, subnet_size))
815 return r
816
stevenvanrossembecc7c52016-11-07 05:52:01 +0100817def get_dc_network():
818 """
819 retrieve the DCnetwork where this dummygatekeeper (GK) connects to.
820 Assume at least 1 datacenter is connected to this GK, and that all datacenters belong to the same DCNetwork
821 :return:
822 """
823 assert (len(GK.dcs) > 0)
824 return GK.dcs.values()[0].net
peusterm6b5224d2016-07-20 13:20:31 +0200825
peusterme26487b2016-03-08 14:00:21 +0100826if __name__ == '__main__':
827 """
828 Lets allow to run the API in standalone mode.
829 """
peusterm398cd3b2016-03-21 15:04:54 +0100830 GK_STANDALONE_MODE = True
peusterme26487b2016-03-08 14:00:21 +0100831 logging.getLogger("werkzeug").setLevel(logging.INFO)
832 start_rest_api("0.0.0.0", 8000)
833