| """This module holds the network. |
| |
| Copyright(C) 2009, Stanford University |
| Date October 2009 |
| Created by ykk |
| """ |
| import random |
| import openflow |
| |
| class network: |
| """Class holding information about OpenFlow network |
| """ |
| def __init__(self): |
| """Initialize |
| """ |
| ##List of switches |
| self.switches = [] |
| ##Dictionary of links |
| self.links = {} |
| ##Reference to connections |
| self.connections = openflow.connections() |
| |
| def add_switch(self, sw): |
| """Add switch to network |
| """ |
| self.switches.append(sw) |
| self.connections.add_connection(sw, sw.connection) |
| |
| def add_link(self, link): |
| """Add link to network |
| """ |
| try: |
| self.links[link.switch1,link.switch2].append(link) |
| except KeyError: |
| self.links[link.switch1,link.switch2] = [] |
| self.links[link.switch1,link.switch2].append(link) |
| |
| class link: |
| """Class to hold information about link |
| |
| Copyright(C) 2009, Stanford University |
| Date November 2009 |
| Created by ykk |
| """ |
| def __init__(self, switch1, switch2): |
| """Initialize link between specified switches |
| """ |
| ##Reference to first switch |
| self.switch1 = switch1 |
| ##Reference to second switch |
| self.switch2 = switch2 |
| |
| class switch: |
| """Class holding information about OpenFlow switch |
| |
| Copyright(C) 2009, Stanford University |
| Date October 2009 |
| Created by ykk |
| """ |
| def __init__(self, miss_send_len=128, |
| sock=None, dpid=None, n_buffers=100, n_tables=1, |
| capability=None): |
| """Initialize switch |
| """ |
| ##Socket to controller |
| self.sock = sock |
| ##Datapath id of switch |
| if (dpid != None): |
| self.datapath_id = dpid |
| else: |
| self.datapath_id = random.randrange(1, pow(2,64)) |
| ##Number of buffers |
| self.n_buffers = n_buffers |
| ##Number of tables |
| self.n_tables= n_tables |
| ##Capabilities |
| if (isinstance(capability, switch_capabilities)): |
| self.capability = capability |
| else: |
| self.capability = switch_capabilities(miss_send_len) |
| ##Valid Actions |
| self.valid_actions = 0 |
| ##List of port |
| self.port = [] |
| |
| class switch_capabilities: |
| """Class to hold switch capabilities |
| """ |
| def __init__(self, miss_send_len=128): |
| """Initialize |
| |
| Copyright(C) 2009, Stanford University |
| Date October 2009 |
| Created by ykk |
| """ |
| ##Capabilities support by datapath |
| self.flow_stats = True |
| self.table_stats = True |
| self.port_stats = True |
| self.stp = True |
| self.multi_phy_tx = True |
| self.ip_resam = False |
| ##Switch config |
| self.send_exp = None |
| self.ip_frag = 0 |
| self.miss_send_len = miss_send_len |
| ##Valid actions |
| self.act_output = True |
| self.act_set_vlan_vid = True |
| self.act_set_vlan_pcp = True |
| self.act_strip_vlan = True |
| self.act_set_dl_src = True |
| self.act_set_dl_dst = True |
| self.act_set_nw_src = True |
| self.act_set_nw_dst = True |
| self.act_set_tp_src = True |
| self.act_set_tp_dst = True |
| self.act_vendor = False |
| |
| def get_capability(self, ofmsg): |
| """Return value for uint32_t capability field |
| """ |
| value = 0 |
| if (self.flow_stats): |
| value += ofmsg.get_value("OFPC_FLOW_STATS") |
| if (self.table_stats): |
| value += ofmsg.get_value("OFPC_TABLE_STATS") |
| if (self.port_stats): |
| value += ofmsg.get_value("OFPC_PORT_STATS") |
| if (self.stp): |
| value += ofmsg.get_value("OFPC_STP") |
| if (self.multi_phy_tx): |
| value += ofmsg.get_value("OFPC_MULTI_PHY_TX") |
| if (self.ip_resam): |
| value += ofmsg.get_value("OFPC_IP_REASM") |
| return value |
| |
| def get_actions(self, ofmsg): |
| """Return value for uint32_t action field |
| """ |
| value = 0 |
| if (self.act_output): |
| value += (1 << (ofmsg.get_value("OFPAT_OUTPUT")+1)) |
| if (self.act_set_vlan_vid): |
| value += (1 << (ofmsg.get_value("OFPAT_SET_VLAN_VID")+1)) |
| if (self.act_set_vlan_pcp): |
| value += (1 << (ofmsg.get_value("OFPAT_SET_VLAN_PCP")+1)) |
| if (self.act_strip_vlan): |
| value += (1 << (ofmsg.get_value("OFPAT_STRIP_VLAN")+1)) |
| if (self.act_set_dl_src): |
| value += (1 << (ofmsg.get_value("OFPAT_SET_DL_SRC")+1)) |
| if (self.act_set_dl_dst): |
| value += (1 << (ofmsg.get_value("OFPAT_SET_DL_DST")+1)) |
| if (self.act_set_nw_src): |
| value += (1 << (ofmsg.get_value("OFPAT_SET_NW_SRC")+1)) |
| if (self.act_set_nw_dst): |
| value += (1 << (ofmsg.get_value("OFPAT_SET_NW_DST")+1)) |
| if (self.act_set_tp_src): |
| value += (1 << (ofmsg.get_value("OFPAT_SET_TP_SRC")+1)) |
| if (self.act_set_tp_dst): |
| value += (1 << (ofmsg.get_value("OFPAT_SET_TP_DST")+1)) |
| return value |
| |
| class port: |
| """Class to hold information about port |
| |
| Copyright(C) 2009, Stanford University |
| Date October 2009 |
| Created by ykk |
| """ |
| def __init__(self, port_no, stp=(2 << 8), hw_addr=None, name=""): |
| """Initialize |
| """ |
| ##Port properties |
| self.port_no = port_no |
| if (hw_addr != None): |
| self.hw_addr = hw_addr |
| else: |
| self.hw_addr = random.randrange(1, pow(2,48)) |
| self.name = name |
| ##Port config |
| self.port_down = False |
| self.no_stp = False |
| self.no_recv = False |
| self.no_recv_stp = False |
| self.no_flood = False |
| self.no_fwd = False |
| self.no_packet_in = False |
| #Port state |
| self.link_down = False |
| self.stp = stp |