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.INFO)
  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 if json data is a dict
  84             data = request.args
  85             # try json payload
  86             if data is None:
  87                 data = request.json
  88             # then no data
  89             if data is None:
  90                 data = {}
  91             elif type(data) is not dict:
  92                 data = json.loads(request.json)
  93
  94             vnf_src_name = data.get("vnf_src_name")
  95             vnf_dst_name = data.get("vnf_dst_name")
  96             vnf_src_interface = data.get("vnf_src_interface")
  97             vnf_dst_interface = data.get("vnf_dst_interface")
  98             weight = data.get("weight")
  99             match = data.get("match")
 100             bidirectional = data.get("bidirectional")
 101             cookie = data.get("cookie")
 102             priority = data.get("priority")
 103             skip_vlan_tag = data.get("skip_vlan_tag")
 104             monitor = data.get("monitor")
 105             monitor_placement = data.get("monitor_placement")
 106
 107             c = net.setChain(
 108                 vnf_src_name, vnf_dst_name,
 109                 vnf_src_interface=vnf_src_interface,
 110                 vnf_dst_interface=vnf_dst_interface,
 111                 cmd=command,
 112                 weight=weight,
 113                 match=match,
 114                 bidirectional=bidirectional,
 115                 cookie=cookie,
 116                 priority=priority,
 117                 skip_vlan_tag=skip_vlan_tag,
 118                 monitor=monitor,
 119                 monitor_placement=monitor_placement)
 120             # return setChain response
 121             return str(c), 200, CORS_HEADER
 122         except Exception as ex:
 123             logging.exception("API error.")
 124             return ex.message, 500, CORS_HEADER
 125
 126
 127 class DrawD3jsgraph(Resource):
 128
 129     global net
 130
 131     def get(self):
 132         nodes = list()
 133         nodes2 = list()
 134         links = list()
 135         # add all DCs
 136         node_attr = networkx.get_node_attributes(net.DCNetwork_graph, 'type')
 137         for node_name in net.DCNetwork_graph.nodes():
 138             nodes2.append(node_name)
 139             node_index = nodes2.index(node_name)
 140             type = node_attr[node_name]
 141             node_dict = {"name":node_name,"group":type}
 142             nodes.append(node_dict)
 143
 144         # add links between other DCs
 145         for node1_name in net.DCNetwork_graph.nodes():
 146             node1_index = nodes2.index(node1_name)
 147             for node2_name in net.DCNetwork_graph.neighbors(node1_name):
 148                 node2_index = nodes2.index(node2_name)
 149                 edge_dict = {"source": node1_index, "target": node2_index, "value": 10}
 150                 links.append(edge_dict)
 151
 152         json = {"nodes":nodes, "links":links}
 153         return json, 200, CORS_HEADER