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