def test_handle_packet_in_raw_in(self):
"""Test handling packet_in raw in message."""
event_name = 'kytos/core.openflow.raw.in'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:02"),
ConnectionState.ESTABLISHED)
data = b'\x04\x0a\x00\x94\x00\x00\x00\x00\x00\x00\x00\x02\x00\x2a\x01'
data += b'\x01\x00\x01\x02\x03\x00\x00\x00\x00\x00\x01\x00\x50\x80\x00'
data += b'\x00\x04\x00\x00\x00\x06\x80\x00\x0a\x02\x08\x06\x80\x00\x06'
data += b'\x06\xff\xff\xff\xff\xff\xff\x80\x00\x08\x06\xf2\x0b\xa4\x7d'
data += b'\xf8\xea\x80\x00\x2a\x02\x00\x01\x80\x00\x2c\x04\x0a\x00\x00'
data += b'\x01\x80\x00\x2e\x04\x0a\x00\x00\x03\x80\x00\x30\x06\xf2\x0b'
data += b'\xa4\x7d\xf8\xea\x80\x00\x32\x06\x00\x00\x00\x00\x00\x00\x00'
data += b'\x00\xff\xff\xff\xff\xff\xff\xf2\x0b\xa4\x7d\xf8\xea\x08\x06'
data += b'\x00\x01\x08\x00\x06\x04\x00\x01\xf2\x0b\xa4\x7d\xf8\xea\x0a'
data += b'\x00\x00\x01\x00\x00\x00\x00\x00\x00\x0a\x00\x00\x03'
event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'new_data': data
})
self.napp.handle_raw_in(event)
of_event = self.napp.controller.buffers.msg_in.get()
self.assertEqual(of_event.name,
'kytos/of_core.v0x04.messages.in.ofpt_packet_in')
def test_handle_01_features_reply(self):
"""Test handling features reply message."""
event_name = 'kytos/of_core.v0x01.messages.in.ofpt_features_reply'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x01, get_switch_mock("00:00:00:00:00:00:00:02"),
ConnectionState.SETUP)
switch.connection.protocol.state = 'waiting_features_reply'
data = b'\x01\x06\x00\x80\x00\x00\x00\x00\x00\x00\x00\xff\x12\x34\x56'
data += b'\x78\x00\x00\x00\x00\xff\x00\x00\x00\x00\x00\x00\xa9\x00\x00'
data += b'\x08\x43\x00\x07\xf2\x0b\xa4\xd0\x3f\x70\x50\x6f\x72\x74\x37'
data += b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
data += b'\x00\x00\x00\x10\x00\x00\x02\x88\x00\x00\x02\x80\x00\x00\x02'
data += b'\x88\x00\x00\x02\x88\x00\x06\xf2\x0b\xa4\x7d\xf8\xea\x50\x6f'
data += b'\x72\x74\x36\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
data += b'\x00\x00\x00\x00\x00\x00\x02\x00\x00\x02\x88\x00\x00\x02\x80'
data += b'\x00\x00\x02\x88\x00\x00\x02\x88'
features_reply = FReply_v0x01()
features_reply.unpack(data[8:])
event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': features_reply
})
self.napp.handle_features_reply(event)
expected = [
'kytos/of_core.v0x01.messages.out.ofpt_stats_request',
'kytos/of_core.v0x01.messages.out.ofpt_set_config'
]
for message in expected:
of_event = self.napp.controller.buffers.msg_out.get()
self.assertEqual(of_event.name, message)
def test_handle_multipart_reply(self):
"""Test handling ofpt_multipart_reply."""
event_name = 'kytos/of_core.v0x04.messages.in.ofpt_multipart_reply'
switch = get_switch_mock("00:00:00:00:00:00:00:02")
switch.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:01"),
ConnectionState.ESTABLISHED)
self.napp.controller.get_switch_or_create(switch.dpid,
switch.connection)
data = b'\x04\x13\x00\x68\xac\xc8\xdf\x58\x00\x01\x00\x00\x00\x00\x00'
data += b'\x00\x00\x58\x00\x00\x00\x00\x00\x38\x25\xd9\x54\xc0\x03\xe8'
data += b'\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00'
data += b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x12\x00\x00\x00\x00'
data += b'\x00\x00\x02\xf4\x00\x01\x00\x10\x80\x00\x0a\x02\x88\xcc\x80'
data += b'\x00\x0c\x02\x1e\xd7\x00\x04\x00\x18\x00\x00\x00\x00\x00\x00'
data += b'\x00\x10\xff\xff\xff\xfd\xff\xff\x00\x00\x00\x00\x00\x00'
# pylint: disable=protected-access
xid = self.napp._multipart_replies_xids[switch.dpid]
# pylint: enable=protected-access
multipart_reply = MultipartReply(xid=xid)
multipart_reply.unpack(data[8:])
stats_event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': multipart_reply
})
self.napp.handle_multipart_reply(stats_event)
# test ofpmp_desc
data = b'\x04\x12\x00\x10\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
data += b'\x00'
multipart_desc = MultipartReply()
multipart_desc.unpack(data[8:])
stats_desc_event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': multipart_desc
})
target_switch = switch.connection.switch
self.napp.handle_multipart_reply(stats_desc_event)
# pylint: disable=protected-access
self.assertNotIn(xid, self.napp._multipart_replies_xids)
# pylint: enable=protected-access
self.assertGreater(len(target_switch.flows), 0)
self.assertEqual(multipart_desc.body.mfr_desc.value,
target_switch.description["manufacturer"])
self.assertEqual(multipart_desc.body.hw_desc.value,
target_switch.description["hardware"])
self.assertEqual(multipart_desc.body.sw_desc.value,
target_switch.description["software"])
self.assertEqual(multipart_desc.body.serial_num.value,
target_switch.description["serial"])
self.assertEqual(multipart_desc.body.dp_desc.value,
target_switch.description["data_path"])
def test_handle_port_desc_multipart_reply(self):
"""Test handling to ofpt_PORT_DESC."""
event_name = 'kytos/of_core.v0x04.messages.in.ofpt_multipart_reply'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:02"))
data = b'\x04\x13\x00\x90\x00\x00\x00\x00\x00\x0d\x00\x00\x00\x00\x00'
data += b'\x00\x00\x00\x00\x07\x00\x00\x00\x00\xf2\x0b\xa4\xd0\x3f\x70'
data += b'\x00\x00\x50\x6f\x72\x74\x37\x00\x00\x00\x00\x00\x00\x00\x00'
data += b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04\x00\x00\x28\x08'
data += b'\x00\x00\x28\x00\x00\x00\x28\x08\x00\x00\x28\x08\x00\x00\x13'
data += b'\x88\x00\x00\x13\x88\x00\x00\x00\x06\x00\x00\x00\x00\xf2\x0b'
data += b'\xa4\x7d\xf8\xea\x00\x00\x50\x6f\x72\x74\x36\x00\x00\x00\x00'
data += b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04'
data += b'\x00\x00\x28\x08\x00\x00\x28\x00\x00\x00\x28\x08\x00\x00\x28'
data += b'\x08\x00\x00\x13\x88\x00\x00\x13\x88'
port_desc = MultipartReply()
port_desc.unpack(data[8:])
interface_1 = get_interface_mock("interface1", 6)
interface_2 = get_interface_mock("interface2", 7)
switch.connection.switch.interfaces = {6: interface_1, 7: interface_2}
stats_event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': port_desc
})
self.napp.handle_multipart_reply(stats_event)
# Send port_desc pack without interface
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:02"))
stats_event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': port_desc
})
self.napp.handle_multipart_reply(stats_event)
expected_event = 'kytos/of_core.switch.port.created'
expected_dpid = '00:00:00:00:00:00:00:02'
for _ in range(0, 2):
of_event_01 = self.napp.controller.buffers.app.get()
of_event_02 = self.napp.controller.buffers.app.get()
self.assertEqual(of_event_01.name, expected_event)
self.assertEqual(of_event_01.content['switch'], expected_dpid)
self.assertEqual(of_event_01.content['port'], 7)
self.assertEqual(of_event_02.name, expected_event)
self.assertEqual(of_event_02.content['switch'], expected_dpid)
self.assertEqual(of_event_02.content['port'], 6)
def setUp(self):
"""Execute steps before each tests.
Set the server_name_url from kytos/of_core
"""
self.switch_v0x01 = get_switch_mock("00:00:00:00:00:00:00:01")
self.switch_v0x04 = get_switch_mock("00:00:00:00:00:00:00:02")
self.switch_v0x01.connection = get_connection_mock(
0x01, get_switch_mock("00:00:00:00:00:00:00:03"))
self.switch_v0x04.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:04"))
patch('kytos.core.helpers.run_on_thread', lambda x: x).start()
# pylint: disable=bad-option-value
from napps.kytos.of_core.main import Main
self.addCleanup(patch.stopall)
self.napp = Main(get_controller_mock())
def test_handle_echo_request(self):
"""Test handling echo request message."""
event_name = 'kytos/of_core.v0x04.messages.in.ofpt_echo_request'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:02"))
data = b'\x04\x02\x00\x0c\x00\x00\x00\x00\x68\x6f\x67\x65'
echo_request = EchoRequest()
echo_request.unpack(data)
event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': echo_request
})
self.napp.handle_echo_request(event)
of_event = self.napp.controller.buffers.msg_out.get()
self.assertEqual(of_event.name,
'kytos/of_core.v0x04.messages.out.ofpt_echo_reply')
def test_handle_features_request_sent(self):
"""Test handling features request sent message."""
event_name = 'kytos/of_core.v0x01.messages.out.ofpt_features_request'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x01, get_switch_mock("00:00:00:00:00:00:00:02"))
switch.connection.protocol.state = 'sending_features'
data = b'\x04\x05\x00\x08\x00\x00\x00\x03'
features_request = FeaturesRequest()
features_request.unpack(data)
event = KytosEvent(name=event_name,
content={
'destination': switch.connection,
'message': features_request
})
self.napp.handle_features_request_sent(event)
self.assertEqual(event.destination.protocol.state,
'waiting_features_reply')
def test_execute(self):
"""Test 'execute' main method."""
dpid_01 = "00:00:00:00:00:00:00:01"
dpid_02 = "00:00:00:00:00:00:00:02"
sw_01 = get_switch_mock()
sw_01.connection = get_connection_mock(0x01, get_switch_mock(dpid_02),
ConnectionState.ESTABLISHED)
sw_04 = get_switch_mock(dpid_02)
sw_04.connection = get_connection_mock(0x04, get_switch_mock(dpid_01),
ConnectionState.ESTABLISHED)
self.napp.controller.get_switch_or_create(dpid_01, sw_01.connection)
self.napp.controller.get_switch_or_create(dpid_02, sw_04.connection)
self.napp.execute()
expected = [
'kytos/of_core.v0x01.messages.out.ofpt_stats_request',
'kytos/of_core.v0x01.messages.out.ofpt_echo_request',
'kytos/of_core.v0x04.messages.out.ofpt_multipart_request',
'kytos/of_core.v0x04.messages.out.ofpt_echo_request'
]
for message in expected:
of_event = self.napp.controller.buffers.msg_out.get()
self.assertEqual(of_event.name, message)
def test_handle_hello_raw_in(self):
"""Test handling hello raw in message."""
event_name = 'kytos/core.openflow.raw.in'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:02"))
data = b'\x04\x00\x00\x10\x00\x00\x00\x3e'
data += b'\x00\x01\x00\x08\x00\x00\x00\x10'
event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'new_data': data
})
self.napp.handle_raw_in(event)
expected = [
'kytos/of_core.v0x04.messages.out.ofpt_hello',
'kytos/of_core.v0x04.messages.out.ofpt_features_request'
]
for message in expected:
of_event = self.napp.controller.buffers.msg_out.get()
self.assertEqual(of_event.name, message)
def test_handle_stats_reply(self):
"""Test handling stats reply message."""
event_name = 'kytos/of_core.v0x01.messages.in.ofpt_stats_reply'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x01, get_switch_mock("00:00:00:00:00:00:00:02"))
stats_data = b'\x01\x11\x00\x0c\x00\x00\x00\x01\x00\x01\x00\x01'
stats_reply = StatsReply()
stats_reply.unpack(stats_data[8:])
stats_event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': stats_reply
})
self.napp.handle_stats_reply(stats_event)
desc_stats_data = b'\x01\x11\x00\x0c\x00\x00\x00\x0e\x00\x00\x00\x00'
desc_stats_reply = StatsReply()
desc_stats_reply.unpack(desc_stats_data[8:])
desc_stats_event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': desc_stats_reply
})
self.napp.handle_stats_reply(desc_stats_event)
self.assertEqual(desc_stats_reply.body.mfr_desc.value,
switch.connection.switch.description["manufacturer"])
self.assertEqual(desc_stats_reply.body.hw_desc.value,
switch.connection.switch.description["hardware"])
self.assertEqual(desc_stats_reply.body.sw_desc.value,
switch.connection.switch.description["software"])
self.assertEqual(desc_stats_reply.body.serial_num.value,
switch.connection.switch.description["serial"])
self.assertEqual(desc_stats_reply.body.dp_desc.value,
switch.connection.switch.description["data_path"])
def test_handle_04_features_reply(self):
"""Test handling features reply message."""
event_name = 'kytos/of_core.v0x04.messages.in.ofpt_features_reply'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:02"),
ConnectionState.SETUP)
switch.connection.protocol.state = 'waiting_features_reply'
data = b'\x04\x06\x00\x20\x00\x00\x00\x00\x00\x00\x08\x60\x6e\x7f\x74'
data += b'\xe7\x00\x00\x00\x00\xff\x63\x00\x00\x00\x00\x00\x4f\x00\x00'
data += b'\x00\x00'
features_reply = FReply_v0x04()
features_reply.unpack(data[8:])
event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'message': features_reply
})
self.napp.handle_features_reply(event)
target_switch = '00:00:08:60:6e:7f:74:e7'
of_event_01 = self.napp.controller.buffers.app.get()
of_event_02 = self.napp.controller.buffers.app.get()
self.assertEqual("kytos/core.switch.new", of_event_01.name)
self.assertEqual(target_switch, of_event_01.content["switch"].dpid)
self.assertEqual("kytos/of_core.handshake.completed", of_event_02.name)
self.assertEqual(target_switch, of_event_02.content["switch"].dpid)
expected = [
'kytos/of_core.v0x04.messages.out.ofpt_multipart_request',
'kytos/of_core.v0x04.messages.out.ofpt_multipart_request',
'kytos/of_core.v0x04.messages.out.ofpt_set_config'
]
for message in expected:
of_event = self.napp.controller.buffers.msg_out.get()
self.assertEqual(of_event.name, message)
def test_handle_port_status_raw_in(self):
"""Test handling port_status raw in message."""
event_name = 'kytos/core.openflow.raw.in'
switch = get_switch_mock()
switch.connection = get_connection_mock(
0x04, get_switch_mock("00:00:00:00:00:00:00:02"),
ConnectionState.ESTABLISHED)
data = b'\x04\x0c\x00\x50\x00\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00'
data += b'\x00\x00\x00\x00\x01\x00\x00\x00\x00\x62\x43\xe5\xdb\x35\x0a'
data += b'\x00\x00\x73\x31\x2d\x65\x74\x68\x31\x00\x00\x00\x00\x00\x00'
data += b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04\x00\x00\x08\x40'
data += b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x98\x96'
data += b'\x80\x00\x00\x00\x00'
event = KytosEvent(name=event_name,
content={
'source': switch.connection,
'new_data': data
})
self.napp.handle_raw_in(event)
of_event = self.napp.controller.buffers.msg_in.get()
self.assertEqual(of_event.name,
'kytos/of_core.v0x04.messages.in.ofpt_port_status')
