src/emuvim/api/rest/network.py

   1 """
   2 Copyright (c) 2015 SONATA-NFV and Paderborn University
   3 ALL RIGHTS RESERVED.
   4
   5 Licensed under the Apache License, Version 2.0 (the "License");
   6 you may not use this file except in compliance with the License.
   7 You may obtain a copy of the License at
   8
   9     http://www.apache.org/licenses/LICENSE-2.0
  10
  11 Unless required by applicable law or agreed to in writing, software
  12 distributed under the License is distributed on an "AS IS" BASIS,
  13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 See the License for the specific language governing permissions and
  15 limitations under the License.
  16
  17 Neither the name of the SONATA-NFV [, ANY ADDITIONAL AFFILIATION]
  18 nor the names of its contributors may be used to endorse or promote
  19 products derived from this software without specific prior written
  20 permission.
  21
  22 This work has been performed in the framework of the SONATA project,
  23 funded by the European Commission under Grant number 671517 through
  24 the Horizon 2020 and 5G-PPP programmes. The authors would like to
  25 acknowledge the contributions of their colleagues of the SONATA
  26 partner consortium (www.sonata-nfv.eu).
  27 """
  28
  29 """
  30 Distributed Cloud Emulator (dcemulator)
  31 Networking and monitoring functions
  32 (c) 2015 by Steven Van Rossem <steven.vanrossem@intec.ugent.be>
  33 """
  34
  35 import logging
  36 from flask_restful import Resource
  37 from flask import request
  38 import json
  39 import networkx
  40
  41 logging.basicConfig(level=logging.DEBUG)
  42
  43 CORS_HEADER = {'Access-Control-Allow-Origin': '*'}
  44
  45 # the global net is set from the topology file, and connected via connectDCNetwork function in rest_api_endpoint.py
  46 net = None
  47
  48
  49 class NetworkAction(Resource):
  50     """
  51     Add or remove chains between VNFs. These chain links are implemented as flow entries in the networks' SDN switches.
  52     :param vnf_src_name: VNF name of the source of the link
  53     :param vnf_dst_name: VNF name of the destination of the link
  54     :param vnf_src_interface: VNF interface name of the source of the link
  55     :param vnf_dst_interface: VNF interface name of the destination of the link
  56     :param weight: weight of the link (can be useful for routing calculations)
  57     :param match: OpenFlow match format of the flow entry
  58     :param bidirectional: boolean value if the link needs to be implemented from src to dst and back
  59     :param cookie: cookie value, identifier of the flow entry to be installed.
  60     :param priority: integer indicating the priority of the flow entry
  61     :param skip_vlan_tag: boolean to indicate whether a new vlan tag should be created for this chain
  62     :param monitor: boolean to indicate whether a new vlan tag should be created for this chain
  63     :param monitor_placement: 'tx'=place the monitoring flowrule at the beginning of the chain, 'rx'=place at the end of the chain
  64     :return: message string indicating if the chain action is succesful or not
  65     """
  66
  67     global net
  68
  69     def put(self):
  70         logging.debug("REST CALL: network chain add")
  71         command = 'add-flow'
  72         return self._NetworkAction(command=command)
  73
  74     def delete(self):
  75         logging.debug("REST CALL: network chain remove")
  76         command = 'del-flows'
  77         return self._NetworkAction(command=command)
  78
  79     def _NetworkAction(self, command=None):
  80         # call DCNetwork method, not really datacenter specific API for now...
  81         # no check if vnfs are really connected to this datacenter...
  82         try:
  83             # check json payload
  84             logging.debug("json: {}".format(request.json))
  85             logging.debug("args: {}".format(request.args))
  86
  87             data = request.json
  88             if data is None:
  89                 data = request.args
  90             if data is None:
  91                 data = {}
  92
  93             vnf_src_name = data.get("vnf_src_name")
  94             vnf_dst_name = data.get("vnf_dst_name")
  95             vnf_src_interface = data.get("vnf_src_interface")
  96             vnf_dst_interface = data.get("vnf_dst_interface")
  97             weight = data.get("weight")
  98             match = data.get("match")
  99             bidirectional = data.get("bidirectional")
 100             cookie = data.get("cookie")
 101             priority = data.get("priority")
 102             skip_vlan_tag = data.get("skip_vlan_tag")
 103             monitor = data.get("monitor")
 104             monitor_placement = data.get("monitor_placement")
 105
 106             c = net.setChain(
 107                 vnf_src_name, vnf_dst_name,
 108                 vnf_src_interface=vnf_src_interface,
 109                 vnf_dst_interface=vnf_dst_interface,
 110                 cmd=command,
 111                 weight=weight,
 112                 match=match,
 113                 bidirectional=bidirectional,
 114                 cookie=cookie,
 115                 priority=priority,
 116                 skip_vlan_tag=skip_vlan_tag,
 117                 monitor=monitor,
 118                 monitor_placement=monitor_placement)
 119             # return setChain response
 120             return str(c), 200, CORS_HEADER
 121         except Exception as ex:
 122             logging.exception("API error.")
 123             return ex.message, 500, CORS_HEADER
 124
 125
 126 class DrawD3jsgraph(Resource):
 127
 128     global net
 129
 130     def get(self):
 131         nodes = list()
 132         nodes2 = list()
 133         links = list()
 134         # add all DCs
 135         node_attr = networkx.get_node_attributes(net.DCNetwork_graph, 'type')
 136         for node_name in net.DCNetwork_graph.nodes():
 137             nodes2.append(node_name)
 138             node_index = nodes2.index(node_name)
 139             type = node_attr[node_name]
 140             node_dict = {"name":node_name,"group":type}
 141             nodes.append(node_dict)
 142
 143         # add links between other DCs
 144         for node1_name in net.DCNetwork_graph.nodes():
 145             node1_index = nodes2.index(node1_name)
 146             for node2_name in net.DCNetwork_graph.neighbors(node1_name):
 147                 node2_index = nodes2.index(node2_name)
 148                 edge_dict = {"source": node1_index, "target": node2_index, "value": 10}
 149                 links.append(edge_dict)
 150
 151         json = {"nodes":nodes, "links":links}
 152         return json, 200, CORS_HEADER