def load_ttf(self):
'''
loads topology transfer function
'''
f = TF(1)
f.load_from_json("%s/topology.tf.json" % self.settings["input_path"])
return f
def make_TTF(num_mbox):
TTF = TF(format["length"]*2)
rule = {}
rule["in_ports"] = [0]
rule["out_ports"] = [1]
TTF.add_link_rule(rule)
rule["in_ports"] = [1]
rule["out_ports"] = [0]
TTF.add_link_rule(rule)
rule["in_ports"] = [2]
rule["out_ports"] = [3]
TTF.add_link_rule(rule)
rule["in_ports"] = [3]
rule["out_ports"] = [2]
TTF.add_link_rule(rule)
rule["in_ports"] = [5]
rule["out_ports"] = [4]
TTF.add_link_rule(rule)
rule["in_ports"] = [4]
rule["out_ports"] = [5]
TTF.add_link_rule(rule)
rule["in_ports"] = [6]
rule["out_ports"] = [7]
TTF.add_link_rule(rule)
rule["in_ports"] = [7]
rule["out_ports"] = [6]
TTF.add_link_rule(rule)
rule["in_ports"] = [8]
rule["out_ports"] = [9]
TTF.add_link_rule(rule)
rule["in_ports"] = [9]
rule["out_ports"] = [8]
TTF.add_link_rule(rule)
rule["in_ports"] = [10]
rule["out_ports"] = [11]
TTF.add_link_rule(rule)
rule["in_ports"] = [11]
rule["out_ports"] = [10]
TTF.add_link_rule(rule)
rule["in_ports"] = [12]
rule["out_ports"] = [13]
TTF.add_link_rule(rule)
rule["in_ports"] = [13]
rule["out_ports"] = [12]
TTF.add_link_rule(rule)
for i in range(num_mbox):
rule["in_ports"] = [14 + i*4]
rule["out_ports"] = [15 + i*4]
TTF.add_link_rule(rule)
rule["in_ports"] = [15 + i*4]
rule["out_ports"] = [14 + i*4]
TTF.add_link_rule(rule)
rule["in_ports"] = [16 + i*4]
rule["out_ports"] = [17 + i*4]
TTF.add_link_rule(rule)
rule["in_ports"] = [17 + i*4]
rule["out_ports"] = [16 + i*4]
TTF.add_link_rule(rule)
return TTF
def load_ntf(self):
'''
load transfer functions into a emulated transfer function with @layer layers.
'''
if "remove_duplicates" in self.settings.keys() and \
self.settings["remove_duplicates"]:
emul_tf = emulated_tf(self.settings["num_layers"], True)
else:
emul_tf = emulated_tf(self.settings["num_layers"], False)
emul_tf.set_fwd_engine_stage(self.settings["fwd_engine_layer"])
emul_tf.set_multipliers(self.settings["switch_id_multiplier"], \
self.settings["port_type_multiplier"], \
self.settings["out_port_type_const"])
for rtr_name in self.settings["rtr_names"]:
f = TF(1)
f.load_from_json("%s/%s.tf.json" %
(self.settings["input_path"], rtr_name))
if "hash_table" in self.settings.keys():
f.activate_hash_table(self.settings["hash_table"])
emul_tf.append_tf(f)
emul_tf.length = f.length
return emul_tf
def testRW1(self):
tf = TF(1)
tf.add_rewrite_rule(TF.create_standard_rule([1], "10xxxxxx", [2], \
"10011111", "01100000"))
hs = headerspace(1)
hs.add_hs(wildcard_create_from_string("1001xxxx"))
result = tf.T(hs, 1)
self.assertEqual(len(result), 1)
self.assert_(wildcard_is_equal(result[0][0].hs_list[0],\
wildcard_create_from_string("1111xxxx")))
def testDependency(self):
tf = TF(1)
tf.add_fwd_rule(TF.create_standard_rule([1], "10xxxxxx", [2], \
None, None))
tf.add_rewrite_rule(TF.create_standard_rule([1], "1xxxxxxx", [3], "00111111", "10000000","",[]))
hs = headerspace(1)
hs.add_hs(wildcard_create_from_string("xxxxxxxx"))
result = tf.T(hs, 1)
self.assertEqual(len(result), 2, "Expecting both rules to be matched")
self.assertTrue(wildcard_is_equal(
result[1][0].hs_list[0],\
wildcard_create_from_string("10xxxxxx"),\
), \
"unexpected second byte array")
def testInverse(self):
tf = TF(1)
tf.add_rewrite_rule(TF.create_standard_rule([1], "10xxxxxx", [2], \
"10011111", "01100000"))
hs = headerspace(1)
hs.add_hs(wildcard_create_from_string("111xxxxx"))
hs.diff_hs(wildcard_create_from_string("1110xxxx"))
result = tf.T_inv(hs, 2)
self.assertEqual(len(result), 1)
self.assertEqual(result[0][0].count(),1)
self.assertEqual(result[0][0].count_diff(),1)
self.assert_(wildcard_is_equal(result[0][0].hs_list[0],\
wildcard_create_from_string("10xxxxxx"),\
))
self.assert_(wildcard_is_equal(result[0][0].hs_diff[0][0],\
wildcard_create_from_string("10x0xxxx"),\
))
format["ip_dst_pos"] = 0
format["ip_dst_len"] = 4
format["length"] = 4
rtr_names = [
"bbra_rtr",
"bbrb_rtr",
"boza_rtr",
"bozb_rtr",
"coza_rtr",
"cozb_rtr",
"goza_rtr",
"gozb_rtr",
"poza_rtr",
"pozb_rtr",
"roza_rtr",
"rozb_rtr",
"soza_rtr",
"sozb_rtr",
"yoza_rtr",
"yozb_rtr",
]
for rtr_name in rtr_names:
f = TF(1)
f.load_object_from_file("../work/tf_simple_stanford_backbone/%s.tf" %
rtr_name)
#OFG = OpenFlow_Rule_Generator(f,ciscoRouter(1).HS_FORMAT())
OFG = OpenFlow_Rule_Generator(f, format)
OFG.generate_of_rules("%s.of" % rtr_name)
def generate_transfer_functions(settings):
st = time()
if ("replace_vlans" in settings.keys()):
has_replaced_vlan = True
else:
has_replaced_vlan = False
if "arp_table_file_sfx" in settings.keys():
arp_sfx = settings["arp_table_file_sfx"]
else:
arp_sfx = "_arp_table.txt"
if "mac_table_file_sfx" in settings.keys():
mac_sfx = settings["mac_table_file_sfx"]
else:
mac_sfx = "_mac_table.txt"
if "config_file_sfx" in settings.keys():
config_sfx = settings["config_file_sfx"]
else:
config_sfx = "_config.txt"
if "spanning_tree_file_sfx" in settings.keys():
span_sfx = settings["spanning_tree_file_sfx"]
else:
span_sfx = "_spanning_tree.txt"
if "route_table_file_sfx" in settings.keys():
route_sfx = settings["route_table_file_sfx"]
else:
route_sfx = "_route.txt"
# generate transfer functions
L = 0
id = 1
cs_list = {}
for i in range(len(settings["rtr_names"])):
rtr_name = settings["rtr_names"][i]
cs = cisco_router(id)
if has_replaced_vlan:
cs.set_replaced_vlan(settings["replace_vlans"][i])
if "hs_format" in settings.keys():
cs.set_hs_format(settings["hs_format"])
L = cs.hs_format["length"]
tf = TF(L)
tf.set_prefix_id(rtr_name)
cs.read_arp_table_file("%s/%s%s" %
(settings["input_path"], rtr_name, arp_sfx))
cs.read_mac_table_file("%s/%s%s" %
(settings["input_path"], rtr_name, mac_sfx))
cs.read_spanning_tree_file("%s/%s%s"%\
(settings["input_path"],rtr_name,span_sfx))
cs.read_config_file("%s/%s%s" %
(settings["input_path"], rtr_name, config_sfx))
cs.read_route_file("%s/%s%s" %
(settings["input_path"], rtr_name, route_sfx))
if ("optimize_fwd_table" not in settings.keys() or \
settings["optimize_fwd_table"]):
cs.optimize_forwarding_table()
if ("fwd_table_only" in settings.keys()
and settings["fwd_table_only"]):
cs.generate_port_ids_only_for_output_ports()
cs.generate_fwd_table_tf(tf)
else:
cs.generate_port_ids([])
cs.generate_transfer_function(tf)
if (not os.path.isdir(settings["output_path"])):
os.makedirs(settings["output_path"])
tf.save_as_json("%s/%s.tf.json" % (settings["output_path"], rtr_name))
tf.save_object_to_file("%s/%s.tf" %
(settings["output_path"], rtr_name))
id += 1
cs_list[rtr_name] = cs
#generate port maps
f = open("%s/port_map.json" % settings["output_path"], 'w')
port_map = {}
for rtr in cs_list.keys():
cs = cs_list[rtr]
port_map[rtr] = cs.port_to_id
f.write(json.dumps(port_map))
f.close()
#write topology:
if "topology" in settings.keys():
print "===Generating Topology==="
out_port_addition = cisco_router.PORT_TYPE_MULTIPLIER * \
cisco_router.OUTPUT_PORT_TYPE_CONST
topology = settings["topology"]
tf = TF(L)
for (from_router, from_port, to_router, to_port) in topology:
from_cs = cs_list[from_router]
to_cs = cs_list[to_router]
rule = TF.create_standard_rule(\
[from_cs.get_port_id(from_port) + out_port_addition],\
None,[to_cs.get_port_id(to_port)],\
None, None, "", [])
tf.add_link_rule(rule)
rule = TF.create_standard_rule(\
[to_cs.get_port_id(to_port) + out_port_addition], \
None,[from_cs.get_port_id(from_port)], \
None, None, "", [])
tf.add_link_rule(rule)
tf.save_as_json("%s/topology.tf.json" % settings["output_path"])
tf.save_object_to_file("%s/topology.tf" % settings["output_path"])
en = time()
print "completed in ", en - st, "seconds"
from headerspace.tf import TF
import json
import sys
rtr_names = [
"atla", "chic", "hous", "kans", "losa", "newy32aoa", "salt", "seat", "wash"
]
PORT_TYPE_MULTIPLIER = 10000
SWITCH_ID_MULTIPLIER = 100000
path = sys.argv[1]
out_path = sys.argv[2]
#path = "i2_tfs"
#out_path = "i2_json"
table_id = 0
topo = TF(1)
topo.load_object_from_file("%s/%s.tf" % (path, "backbone_topology"))
topology = {"topology": []}
for rule in topo.rules:
in_ports = rule["in_ports"]
out_ports = rule["out_ports"]
for in_port in in_ports:
for out_port in out_ports:
topology["topology"].append({"src": in_port, "dst": out_port})
topo.save_as_json("%s/%s.json" % (path, "topology"))
for rtr in rtr_names:
tf = TF(1)
tf.load_object_from_file("%s/%s.tf" % (path, rtr))
#tf.save_as_json("%s/%s.tf.json"%(path,rtr))
table_id += 1
],
"input_path":
"../examples/stanford/tf_stanford_backbone",
"switch_id_multiplier":
cisco_router.SWITCH_ID_MULTIPLIER,
"port_type_multiplier":
cisco_router.PORT_TYPE_MULTIPLIER,
"mid_port_type_const":
cisco_router.INTERMEDIATE_PORT_TYPE_CONST,
}
loader = net_loader(settings)
(map, inv_map) = loader.load_port_map()
#topology
f = TF(1)
f.load_object_from_file("%s/topology.tf" % settings["input_path"])
#net plumber instance
N = NetPlumber(f.length)
#adding links
for rule in f.rules:
input_ports = rule["in_ports"]
output_ports = rule["out_ports"]
for input_port in input_ports:
for output_port in output_ports:
N.add_link(input_port, output_port)
# add links for intermediate port
f = open("%s/port_map.txt" % settings["input_path"], 'r')
def make_NTF(num_mbox):
line_counter = 1
num_stack = num_mbox + 1
NTF = TF(format["length"])
m_addr = [dotted_ip_to_int("10.1.1.1"),
dotted_ip_to_int("10.1.2.2"),
dotted_ip_to_int("10.1.3.3"),
dotted_ip_to_int("10.1.4.4"),
dotted_ip_to_int("10.1.5.5"),]
m_subnet = [dotted_ip_to_int("10.1.1.0"),
dotted_ip_to_int("10.1.2.0"),
dotted_ip_to_int("10.1.3.0"),
dotted_ip_to_int("10.1.4.0"),
dotted_ip_to_int("10.1.5.0"),]
sender_addr = dotted_ip_to_int("10.1.10.1")
receiver_addr = dotted_ip_to_int("10.1.20.1")
network_subnet = dotted_ip_to_int("10.1.0.0")
# rules for R1
for i in range(num_mbox):
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,m_addr[i],32)
rule = TF.create_standard_rule([1,7], ip_match, [2], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,receiver_addr,32)
rule = TF.create_standard_rule([1,2], ip_match, [7], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,sender_addr,32)
rule = TF.create_standard_rule([2,7], ip_match, [1], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
# Add rule for R2:
all_ip_star_ingress_ports = []
for i in range(num_mbox):
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,m_addr[i],32)
rule = TF.create_standard_rule([3,4], ip_match, [14+4*i], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
all_ip_star_ingress_ports.append(14+4*i)
for i in range(num_mbox):
rule = TF.create_custom_rule(make_IP_str_match(all_ip_star_ingress_ports),
make_IP_str_transform(m_addr[i], 32, [all_ip_star_ingress_ports[i]]),
make_IP_str_inv_match([all_ip_star_ingress_ports[i]], m_addr[i], 32),
make_IP_str_inv_transform(all_ip_star_ingress_ports),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
rule = TF.create_custom_rule(make_IP_str_match(all_ip_star_ingress_ports),
make_IP_str_transform(sender_addr, 32, [4]),
make_IP_str_inv_match([4], sender_addr, 32),
make_IP_str_inv_transform(all_ip_star_ingress_ports),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
rule = TF.create_custom_rule(make_IP_str_match(all_ip_star_ingress_ports),
make_IP_str_transform(receiver_addr, 32, [3]),
make_IP_str_inv_match([3], receiver_addr, 32),
make_IP_str_inv_transform(all_ip_star_ingress_ports),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
# add rule for R3:
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,receiver_addr,32)
rule = TF.create_standard_rule([5,6], ip_match, [8], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,sender_addr,32)
rule = TF.create_standard_rule([5], ip_match, [6], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
rule = TF.create_custom_rule(make_IP_str_match([8]),
make_IP_str_transform(receiver_addr, 32, [8]),
make_IP_str_inv_match([8], receiver_addr, 32),
make_IP_str_inv_transform([8]),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
rule = TF.create_custom_rule(make_IP_str_match([8]),
make_IP_str_transform(sender_addr, 32, [6]),
make_IP_str_inv_match([6], sender_addr, 32),
make_IP_str_inv_transform([8]),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
for i in range(num_mbox):
rule = TF.create_custom_rule(make_IP_str_match([8]),
make_IP_str_transform(m_addr[i], 32, [5]),
make_IP_str_inv_match([5], m_addr[i], 32),
make_IP_str_inv_transform([8]),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,m_addr[i],32)
rule = TF.create_standard_rule([6], ip_match, [5], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
# add rule for R4,R5,R8,R9,...
for i in range(num_mbox):
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,m_subnet[i],24)
rule = TF.create_standard_rule([15 + 4*i], ip_match, [16 + 4*i], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,network_subnet,18)
rule = TF.create_standard_rule([16 + 4*i], ip_match, [15 + 4*i], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
# add rule for R6
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,receiver_addr,32)
rule = TF.create_standard_rule([9], ip_match, [10], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
ip_match = make_byte_array_ip_star_hdr(None,[],[],0,0,network_subnet,16)
rule = TF.create_standard_rule([10], ip_match, [9], None, None, "sample.txt", [line_counter])
NTF.add_fwd_rule(rule)
line_counter += 1
# add rule for R7
rule = TF.create_custom_rule(make_IP_str_match([11]),
make_IP_str_transform(receiver_addr, 32, [12]),
make_IP_str_inv_match([12], receiver_addr, 32),
make_IP_str_inv_transform([11]),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
rule = TF.create_custom_rule(make_IP_str_match([11]),
make_IP_str_transform(network_subnet, 21, [11]),
make_IP_str_inv_match([11], network_subnet, 21),
make_IP_str_inv_transform([11]),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
# rule for middle boxes:
for i in range(num_mbox):
rule = TF.create_custom_rule(make_mbox_match(m_addr[i], 32, 17+4*i),
make_mbox_tf(),
make_mbox_inv_match(m_addr[i], 32, 17+4*i),
make_mbox_tf(),
"sample.txt", [line_counter])
NTF.add_custom_rule(rule)
line_counter += 1
return NTF
