| """ |
| Flow stats test case. |
| Similar to Flow stats test case in the perl test harness. |
| |
| """ |
| |
| import logging |
| |
| import unittest |
| import random |
| |
| from oftest import config |
| import oftest.controller as controller |
| import oftest.cstruct as ofp |
| import oftest.message as message |
| import oftest.dataplane as dataplane |
| import oftest.action as action |
| import oftest.parse as parse |
| import oftest.base_tests as base_tests |
| |
| from oftest.testutils import * |
| from time import sleep |
| |
| # TODO: ovs has problems with VLAN id? |
| WILDCARD_VALUES = [ofp.OFPFW_IN_PORT, |
| # (ofp.OFPFW_DL_VLAN | ofp.OFPFW_DL_VLAN_PCP), |
| ofp.OFPFW_DL_SRC, |
| ofp.OFPFW_DL_DST, |
| (ofp.OFPFW_DL_TYPE | ofp.OFPFW_NW_SRC_ALL | |
| ofp.OFPFW_NW_DST_ALL | ofp.OFPFW_NW_TOS | ofp.OFPFW_NW_PROTO | |
| ofp.OFPFW_TP_SRC | ofp.OFPFW_TP_DST), |
| (ofp.OFPFW_NW_PROTO | ofp.OFPFW_TP_SRC | ofp.OFPFW_TP_DST), |
| ofp.OFPFW_TP_SRC, |
| ofp.OFPFW_TP_DST, |
| ofp.OFPFW_NW_SRC_MASK, |
| ofp.OFPFW_NW_DST_MASK, |
| ofp.OFPFW_DL_VLAN_PCP, |
| ofp.OFPFW_NW_TOS] |
| |
| def sendPacket(obj, pkt, ingress_port, egress_port, test_timeout): |
| |
| logging.info("Sending packet to dp port " + str(ingress_port) + |
| ", expecting output on " + str(egress_port)) |
| obj.dataplane.send(ingress_port, str(pkt)) |
| |
| exp_pkt_arg = None |
| exp_port = None |
| if config["relax"]: |
| exp_pkt_arg = pkt |
| exp_port = egress_port |
| |
| (rcv_port, rcv_pkt, pkt_time) = obj.dataplane.poll(port_number=exp_port, |
| exp_pkt=exp_pkt_arg) |
| obj.assertTrue(rcv_pkt is not None, |
| "Packet not received on port " + str(egress_port)) |
| logging.debug("Packet len " + str(len(rcv_pkt)) + " in on " + |
| str(rcv_port)) |
| obj.assertEqual(rcv_port, egress_port, |
| "Packet received on port " + str(rcv_port) + |
| ", expected port " + str(egress_port)) |
| obj.assertEqual(str(pkt), str(rcv_pkt), |
| 'Response packet does not match send packet') |
| |
| def getStats(obj, port): |
| stat_req = message.port_stats_request() |
| stat_req.port_no = port |
| |
| logging.info("Sending stats request") |
| response, pkt = obj.controller.transact(stat_req, timeout=2) |
| obj.assertTrue(response is not None, |
| "No response to stats request") |
| obj.assertTrue(len(response.stats) == 1, |
| "Did not receive port stats reply") |
| for item in response.stats: |
| logging.info("Sent " + str(item.tx_packets) + " packets") |
| packet_sent = item.tx_packets |
| packet_recv = item.rx_packets |
| logging.info("Port %d stats count: tx %d rx %d" % (port, packet_sent, packet_recv)) |
| return packet_sent, packet_recv |
| |
| def getAllStats(obj): |
| stat_req = message.port_stats_request() |
| stat_req.port_no = ofp.OFPP_NONE |
| |
| logging.info("Sending all port stats request") |
| response, pkt = obj.controller.transact(stat_req, timeout=2) |
| obj.assertTrue(response is not None, |
| "No response to stats request") |
| obj.assertTrue(len(response.stats) >= 3, |
| "Did not receive all port stats reply") |
| stats = {} |
| for item in response.stats: |
| stats[ item.port_no ] = ( item.tx_packets, item.rx_packets ) |
| return stats |
| |
| def verifyStats(obj, port, test_timeout, packet_sent, packet_recv): |
| stat_req = message.port_stats_request() |
| stat_req.port_no = port |
| |
| all_packets_received = 0 |
| all_packets_sent = 0 |
| sent = recv = 0 |
| for i in range(0,test_timeout): |
| logging.info("Sending stats request") |
| response, pkt = obj.controller.transact(stat_req, |
| timeout=test_timeout) |
| obj.assertTrue(response is not None, |
| "No response to stats request") |
| obj.assertTrue(len(response.stats) == 1, |
| "Did not receive port stats reply") |
| for item in response.stats: |
| sent = item.tx_packets |
| recv = item.rx_packets |
| logging.info("Sent " + str(item.tx_packets) + " packets") |
| if item.tx_packets == packet_sent: |
| all_packets_sent = 1 |
| logging.info("Received " + str(item.rx_packets) + " packets") |
| if item.rx_packets == packet_recv: |
| all_packets_received = 1 |
| |
| if all_packets_received and all_packets_sent: |
| break |
| sleep(1) |
| |
| logging.info("Expected port %d stats count: tx %d rx %d" % (port, packet_sent, packet_recv)) |
| logging.info("Actual port %d stats count: tx %d rx %d" % (port, sent, recv)) |
| obj.assertTrue(all_packets_sent, |
| "Packet sent does not match number sent") |
| obj.assertTrue(all_packets_received, |
| "Packet received does not match number sent") |
| |
| class SingleFlowStats(base_tests.SimpleDataPlane): |
| """ |
| Verify flow stats are properly retrieved. |
| |
| Generate a packet |
| Generate and install a flow from port 1 to 2 |
| Send the packet |
| Send port stats request to port 1 & 2 |
| Verify that the packet counter has incremented |
| """ |
| |
| def runTest(self): |
| # TODO: set from command-line parameter |
| test_timeout = 60 |
| |
| of_ports = config["port_map"].keys() |
| of_ports.sort() |
| self.assertTrue(len(of_ports) > 1, "Not enough ports for test") |
| |
| rc = delete_all_flows(self.controller) |
| self.assertEqual(rc, 0, "Failed to delete all flows") |
| |
| # build packet |
| pkt = simple_tcp_packet() |
| match = packet_to_flow_match(self, pkt) |
| match.wildcards &= ~ofp.OFPFW_IN_PORT |
| self.assertTrue(match is not None, |
| "Could not generate flow match from pkt") |
| act = action.action_output() |
| |
| # build flow |
| ingress_port = of_ports[0]; |
| egress_port = of_ports[1]; |
| logging.info("Ingress " + str(ingress_port) + |
| " to egress " + str(egress_port)) |
| match.in_port = ingress_port |
| flow_mod_msg = message.flow_mod() |
| flow_mod_msg.match = match |
| flow_mod_msg.cookie = random.randint(0,9007199254740992) |
| flow_mod_msg.buffer_id = 0xffffffff |
| flow_mod_msg.idle_timeout = 0 |
| flow_mod_msg.hard_timeout = 0 |
| act.port = egress_port |
| self.assertTrue(flow_mod_msg.actions.add(act), "Could not add action") |
| |
| # send flow |
| logging.info("Inserting flow") |
| rv = self.controller.message_send(flow_mod_msg) |
| self.assertTrue(rv != -1, "Error installing flow mod") |
| self.assertEqual(do_barrier(self.controller), 0, "Barrier failed") |
| |
| # get initial port stats count |
| initTxInPort, initRxInPort = getStats(self, ingress_port) |
| initTxOutPort, initRxOutPort = getStats(self, egress_port) |
| |
| # send packet N times |
| num_sends = random.randint(10,20) |
| logging.info("Sending " + str(num_sends) + " test packets") |
| for i in range(0,num_sends): |
| sendPacket(self, pkt, ingress_port, egress_port, |
| test_timeout) |
| |
| verifyStats(self, ingress_port, test_timeout, initTxInPort, initRxInPort + num_sends) |
| verifyStats(self, egress_port, test_timeout, initTxOutPort + num_sends, initRxOutPort) |
| |
| class MultiFlowStats(base_tests.SimpleDataPlane): |
| """ |
| Verify flow stats are properly retrieved. |
| |
| Generate two packets and install two matching flows |
| Send some number of packets |
| Send a port stats request to get packet count |
| Verify that the packet counter has incremented |
| """ |
| |
| def buildFlowModMsg(self, pkt, ingress_port, egress_port): |
| match = packet_to_flow_match(self, pkt) |
| match.wildcards &= ~ofp.OFPFW_IN_PORT |
| self.assertTrue(match is not None, |
| "Could not generate flow match from pkt") |
| match.in_port = ingress_port |
| |
| flow_mod_msg = message.flow_mod() |
| flow_mod_msg.match = match |
| flow_mod_msg.cookie = random.randint(0,9007199254740992) |
| flow_mod_msg.buffer_id = 0xffffffff |
| flow_mod_msg.idle_timeout = 0 |
| flow_mod_msg.hard_timeout = 0 |
| act = action.action_output() |
| act.port = egress_port |
| self.assertTrue(flow_mod_msg.actions.add(act), "Could not add action") |
| |
| logging.info("Ingress " + str(ingress_port) + |
| " to egress " + str(egress_port)) |
| |
| return flow_mod_msg |
| |
| def runTest(self): |
| # TODO: set from command-line parameter |
| test_timeout = 60 |
| |
| of_ports = config["port_map"].keys() |
| of_ports.sort() |
| self.assertTrue(len(of_ports) >= 3, "Not enough ports for test") |
| ingress_port = of_ports[0]; |
| egress_port1 = of_ports[1]; |
| egress_port2 = of_ports[2]; |
| |
| rc = delete_all_flows(self.controller) |
| self.assertEqual(rc, 0, "Failed to delete all flows") |
| |
| pkt1 = simple_tcp_packet() |
| flow_mod_msg1 = self.buildFlowModMsg(pkt1, ingress_port, egress_port1) |
| |
| pkt2 = simple_tcp_packet(dl_src='0:7:7:7:7:7') |
| flow_mod_msg2 = self.buildFlowModMsg(pkt2, ingress_port, egress_port2) |
| |
| logging.info("Inserting flow1") |
| rv = self.controller.message_send(flow_mod_msg1) |
| self.assertTrue(rv != -1, "Error installing flow mod") |
| logging.info("Inserting flow2") |
| rv = self.controller.message_send(flow_mod_msg2) |
| self.assertTrue(rv != -1, "Error installing flow mod") |
| self.assertEqual(do_barrier(self.controller), 0, "Barrier failed") |
| |
| # get initial port stats count |
| initTxInPort, initRxInPort = getStats(self, ingress_port) |
| initTxOutPort1, initRxOutPort1 = getStats(self, egress_port1) |
| initTxOutPort2, initRxOutPort2 = getStats(self, egress_port2) |
| |
| num_pkt1s = random.randint(10,30) |
| logging.info("Sending " + str(num_pkt1s) + " pkt1s") |
| num_pkt2s = random.randint(10,30) |
| logging.info("Sending " + str(num_pkt2s) + " pkt2s") |
| for i in range(0,num_pkt1s): |
| sendPacket(self, pkt1, ingress_port, egress_port1, test_timeout) |
| for i in range(0,num_pkt2s): |
| sendPacket(self, pkt2, ingress_port, egress_port2, test_timeout) |
| |
| verifyStats(self, ingress_port, test_timeout, |
| initTxInPort, initRxInPort + num_pkt1s + num_pkt2s) |
| verifyStats(self, egress_port1, test_timeout, |
| initTxOutPort1 + num_pkt1s, initRxOutPort1) |
| verifyStats(self, egress_port2, test_timeout, |
| initTxOutPort2 + num_pkt2s, initRxOutPort2) |
| |
| class AllPortStats(base_tests.SimpleDataPlane): |
| """ |
| Verify all port stats are properly retrieved. |
| |
| First, get stats from each port. Then get all port stats, verify |
| consistency with single port stats. |
| """ |
| |
| # TODO: This is copied from MultiFlowStats. Need to combine. |
| def buildFlowModMsg(self, pkt, ingress_port, egress_port): |
| match = packet_to_flow_match(self, pkt) |
| match.wildcards &= ~ofp.OFPFW_IN_PORT |
| self.assertTrue(match is not None, |
| "Could not generate flow match from pkt") |
| match.in_port = ingress_port |
| |
| flow_mod_msg = message.flow_mod() |
| flow_mod_msg.match = match |
| flow_mod_msg.cookie = random.randint(0,9007199254740992) |
| flow_mod_msg.buffer_id = 0xffffffff |
| flow_mod_msg.idle_timeout = 0 |
| flow_mod_msg.hard_timeout = 0 |
| act = action.action_output() |
| act.port = egress_port |
| self.assertTrue(flow_mod_msg.actions.add(act), "Could not add action") |
| |
| logging.info("Ingress " + str(ingress_port) + |
| " to egress " + str(egress_port)) |
| |
| return flow_mod_msg |
| |
| def runTest(self): |
| # TODO: set from command-line parameter |
| test_timeout = 60 |
| |
| of_ports = config["port_map"].keys() |
| of_ports.sort() |
| self.assertTrue(len(of_ports) >= 3, "Not enough ports for test") |
| port0 = of_ports[0]; |
| port1 = of_ports[1]; |
| port2 = of_ports[2]; |
| |
| # construct some packets and flows, send to switch |
| pkt1 = simple_tcp_packet() |
| flow_mod_msg1 = self.buildFlowModMsg(pkt1, port0, port1) |
| |
| pkt2 = simple_tcp_packet(dl_src='0:7:7:7:7:7') |
| flow_mod_msg2 = self.buildFlowModMsg(pkt2, port0, port2) |
| |
| logging.info("Inserting flow1") |
| rv = self.controller.message_send(flow_mod_msg1) |
| self.assertTrue(rv != -1, "Error installing flow mod") |
| logging.info("Inserting flow2") |
| rv = self.controller.message_send(flow_mod_msg2) |
| self.assertTrue(rv != -1, "Error installing flow mod") |
| self.assertEqual(do_barrier(self.controller), 0, "Barrier failed") |
| |
| num_pkt1s = random.randint(5,10) |
| logging.info("Sending " + str(num_pkt1s) + " pkt1s") |
| num_pkt2s = random.randint(10,15) |
| logging.info("Sending " + str(num_pkt2s) + " pkt2s") |
| for i in range(0,num_pkt1s): |
| sendPacket(self, pkt1, port0, port1, test_timeout) |
| for i in range(0,num_pkt2s): |
| sendPacket(self, pkt2, port0, port2, test_timeout) |
| |
| # get individual port stats count |
| port_stats = {} |
| port_stats[ port0 ] = getStats(self, port0) |
| port_stats[ port1 ] = getStats(self, port1) |
| port_stats[ port2 ] = getStats(self, port2) |
| |
| all_stats = getAllStats(self) |
| assert(port_stats[ port0 ] == all_stats[ port0 ]) |
| assert(port_stats[ port1 ] == all_stats[ port1 ]) |
| assert(port_stats[ port2 ] == all_stats[ port2 ]) |