def check_infra(self, infra: nx.classes.digraph.DiGraph) -> bool:
"""Checks if the infrastructure graph has the correct format
:infra: nx.classes.digraph.DiGraph: infrastructure graph
:returns: bool: telling if it satisfies it or not
"""
basic_checker = CheckBasicDigraphs()
if not basic_checker.check_infra(infra):
return False
for h1,h2,e_d in infra.edges(data=True):
if 'reliability' not in e_d:
print("'reliability key not present in infrastructure",
"edge (", h1, h2, ")'", file=sys.stderr)
return False
for host in [h1, h2]:
if 'reliability' not in infra.nodes[host]:
print("'reliability key not present in infrastructure"
"host", host, file=sys.stderr)
return False
if 'lifetime' not in infra.nodes[host]:
print("'lifetime key not present in infrastructure host'",
host, file=sys.stderr)
return False
return True
def check_ns(self, ns: nx.classes.digraph.DiGraph) -> bool:
"""Checks if the network service graph has the correct format
:ns: nx.classes.digraph.DiGraph: network service graph
:returns: bool: telling if it satisfies it or not
"""
for node in ns.nodes():
node_dict = ns.nodes[node]
if 'cpu' not in node_dict or type(node_dict['cpu']) != int:
print("'cpu' key not present in node, or not an int",
file=sys.stderr)
return False
if 'mem' not in node_dict or type(node_dict['mem']) != int:
print("'mem' key not present in node, or not an int",
file=sys.stderr)
return False
if 'disk' not in node_dict or type(node_dict['disk']) != int:
print("'disk' key not present in node, or not an int",
file=sys.stderr)
return False
if 'rats' in node_dict and type(node_dict['rats']) != list:
print("'rats' key does not reference a list",
file=sys.stderr)
return False
if 'location' in node_dict:
if type(node_dict['location']) != dict:
print("'location' key does not reference a dictionary",
file=sys.stderr)
return False
if 'radius' not in node_dict['location'] or\
(type(node_dict['location']['radius']) != int and\
type(node_dict['location']['radius']) != float):
print("'location' does not have a 'radius', or it is not"+\
" a number", file=sys.stderr)
return False
if 'center' not in node_dict['location'] or\
type(node_dict['location']['center']) != tuple or\
type(node_dict['location']['center'][0]) != float or\
type(node_dict['location']['center'][1]) != float:
print("'location' does not have a 'center', or it is bot"+\
" a tuple of floats", file=sys.stderr)
return False
for h1,h2 in ns.edges():
edge_dict = ns[h1][h2]
if 'bw' not in edge_dict or type(edge_dict['bw']) != int:
print("'bw' key not in virtual link, or is not int",
file=sys.stderr)
return False
if 'delay' not in edge_dict or type(edge_dict['delay']) != int:
print("'delay' key not in virtual link, or is not int",
file=sys.stderr)
return False
return True
def map(self, infra: nx.classes.digraph.DiGraph,
ns: nx.classes.digraph.DiGraph) -> dict:
"""Maps a network service on top of an infrastructure.
:infra: nx.classes.digraph.DiGraph: infrastructure graph
:ns: nx.classes.digraph.DiGraph: network service graph
:returns: dict: mapping decissions dictionary
"""
mapping = {'worked': True}
# Check that graphs have correct format
if not self.__checker.check_infra(infra) or\
not self.__checker.check_ns(ns):
mapping['worked'] = False
return mapping
infra_tmp = copy.deepcopy(infra)
for vl in ns.edges():
# Map the VNFs
vnf1, vnf2 = vl
for vnf in set(vl).difference(set(mapping.keys())):
host = self.__get_host(vnf=vnf, infra=infra_tmp, ns=ns)
if not host:
mapping['worked'] = False
return mapping
else:
mapping[vnf] = host
self.__consume_vnf(host, vnf, infra_tmp, ns)
# Map the VL
vl_map = self.__get_path(vl=vl,
src_host=mapping[vnf1],
dst_host=mapping[vnf2],
infra=infra_tmp,
ns=ns)
if vl_map == {}:
mapping['worked'] = False
return mapping
mapping[vl] = vl_map['path']
self.__consume_vl(path=vl_map['path'],
vl=vl,
infra=infra_tmp,
ns=ns)
return mapping
def check_infra(self, infra: nx.classes.digraph.DiGraph) -> bool:
"""Checks if the infrastructure graph has the correct format
:infra: nx.classes.digraph.DiGraph: infrastructure graph
:returns: bool: telling if it satisfies it or not
"""
for node in infra.nodes():
node_dict = infra.nodes[node]
if 'cpu' not in node_dict or type(node_dict['cpu']) != int:
print("'cpu' key not present in node", node,
"or not an int", file=sys.stderr)
return False
if 'mem' not in node_dict or type(node_dict['mem']) != int:
print("'mem' key not present in node, or not an int",
file=sys.stderr)
return False
if 'disk' not in node_dict or type(node_dict['disk']) != int:
print("'disk' key not present in node, or not an int",
file=sys.stderr)
return False
if 'rats' in node_dict and type(node_dict['rats']) != list:
print("'rats' key does not reference a list",
file=sys.stderr)
return False
if 'location' in node_dict and (type(node_dict['location']) != tuple\
or type(node_dict['location'][0]) != float\
or type(node_dict['location'][1]) != float):
print("'location' key does not reference a tuple of floats",
file=sys.stderr)
return False
if 'cost' not in node_dict or type(node_dict['cost']) != dict:
print("'cost' key not inside host, or is not dictionary",
file=sys.stderr)
return False
elif 'cpu' not in node_dict['cost'] or (\
type(node_dict['cost']['cpu']) != float and\
type(node_dict['cost']['cpu']) != int):
print("'cpu' key not inside h['cost'], or is not a number",
file=sys.stderr)
return False
elif 'mem' not in node_dict['cost'] or (\
type(node_dict['cost']['mem']) != float and\
type(node_dict['cost']['mem']) != int):
print("'mem' key not inside h['cost'], or is not a number",
file=sys.stderr)
return False
elif 'disk' not in node_dict['cost'] or (\
type(node_dict['cost']['disk']) != float and\
type(node_dict['cost']['disk']) != int):
print("'disk' key not inside h['cost'], or is not a number",
file=sys.stderr)
return False
for h1,h2 in infra.edges():
edge_dict = infra[h1][h2]
if 'bw' not in edge_dict or type(edge_dict['bw']) != int:
print("'bw' key not in physical link, or is not int",
file=sys.stderr)
return False
if 'delay' not in edge_dict or type(edge_dict['delay']) != int:
print("'delay' key not in physical link, or is not int",
file=sys.stderr)
return False
if 'cost' not in edge_dict or (type(edge_dict['cost']) != int\
and type(edge_dict['cost']) != float):
print("'cost' key not inside h[h1][h2]['cost'],"+\
" or is not a number", file=sys.stderr)
return False
return True