mirror of
https://github.com/espressif/openthread.git
synced 2026-07-02 18:30:22 +00:00
Giuseppe Andreello
6d8d42e9bf
Changed the interpreter directives from '#!/usr/bin/python' to '#!/usr/bin/env python' so the OpenThread tests can be run inside a python virtualenv.
236 lines
12 KiB
Python
Executable File
236 lines
12 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# Copyright (c) 2016, The OpenThread Authors.
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
# 1. Redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer.
|
|
# 2. Redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
# 3. Neither the name of the copyright holder nor the
|
|
# names of its contributors may be used to endorse or promote products
|
|
# derived from this software without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
#
|
|
|
|
import io
|
|
import unittest
|
|
|
|
from common import MacAddress, MacAddressType
|
|
import mac802154
|
|
|
|
longaddrs = bytearray([0x61, 0xcc, 0x00, 0xce, 0xfa])
|
|
shortaddrs = bytearray([0x61, 0x88, 0x00, 0xce, 0xfa])
|
|
longshortaddrs = bytearray([0x61, 0xc8, 0x00, 0xce, 0xfa])
|
|
shortlongaddrs = bytearray([0x61, 0x8c, 0x00, 0xce, 0xfa])
|
|
|
|
|
|
class TestMacParser(unittest.TestCase):
|
|
|
|
def test_should_parse_ack_frame(self):
|
|
frame = mac802154.MacFrame()
|
|
frame.parse(io.BytesIO(bytearray([0x12, 0x00, 0x12, 0x34, 0x56])))
|
|
|
|
self.assertEqual(mac802154.MacHeader.FrameType.ACK, frame.header.frame_type)
|
|
self.assertEqual(True, frame.header.frame_pending)
|
|
self.assertEqual(False, frame.header.ack_request)
|
|
self.assertEqual(0, frame.header.frame_version)
|
|
self.assertEqual(0x12, frame.header.seq)
|
|
self.assertEqual(bytearray([0x34, 0x56]), frame.header.fcs)
|
|
self.assertEqual(None, frame.payload)
|
|
|
|
def test_should_parse_data_frame_with_short_addresses(self):
|
|
frame = mac802154.MacFrame()
|
|
frame.parse(io.BytesIO(bytearray([0x61, 0x88, 0x34, 0xce, 0xfa,
|
|
0xad, 0xde, 0xef, 0xbe, 0x12, 0x34, 0xfe, 0xdc])))
|
|
|
|
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
|
|
self.assertEqual(False, frame.header.frame_pending)
|
|
self.assertEqual(True, frame.header.ack_request)
|
|
self.assertEqual(0, frame.header.frame_version)
|
|
self.assertEqual(0x34, frame.header.seq)
|
|
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
|
|
self.assertEqual(0xface, frame.header.dest_pan_id)
|
|
self.assertEqual(0xdead, frame.header.dest_address.rloc)
|
|
self.assertEqual(0xface, frame.header.src_pan_id)
|
|
self.assertEqual(0xbeef, frame.header.src_address.rloc)
|
|
|
|
self.assertEqual(bytearray([0x12, 0x34]), frame.payload.data)
|
|
|
|
def test_should_parse_data_frame_with_extended_addresses(self):
|
|
frame = mac802154.MacFrame()
|
|
frame.parse(io.BytesIO(bytearray([0x61, 0xcc,
|
|
0x56,
|
|
0xce, 0xfa,
|
|
0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe, 0xef,
|
|
0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0, 0x0d,
|
|
0x12, 0x34,
|
|
0xfe, 0xdc])))
|
|
|
|
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
|
|
self.assertEqual(False, frame.header.frame_pending)
|
|
self.assertEqual(True, frame.header.ack_request)
|
|
self.assertEqual(0, frame.header.frame_version)
|
|
self.assertEqual(0x56, frame.header.seq)
|
|
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
|
|
self.assertEqual(0xface, frame.header.dest_pan_id)
|
|
self.assertEqual(bytearray(reversed([0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe, 0xef])),
|
|
frame.header.dest_address.mac_address)
|
|
self.assertEqual(0xface, frame.header.src_pan_id)
|
|
self.assertEqual(bytearray(reversed([0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0, 0x0d])),
|
|
frame.header.src_address.mac_address)
|
|
|
|
self.assertEqual(bytearray([0x12, 0x34]), frame.payload.data)
|
|
|
|
def test_should_parse_data_frame_with_short_and_extended_addresses(self):
|
|
frame = mac802154.MacFrame()
|
|
frame.parse(io.BytesIO(bytearray([0x61, 0xc8,
|
|
0x56,
|
|
0xce, 0xfa,
|
|
0xad, 0xde,
|
|
0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0, 0x0d,
|
|
0x12, 0x34,
|
|
0xfe, 0xdc])))
|
|
|
|
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
|
|
self.assertEqual(False, frame.header.frame_pending)
|
|
self.assertEqual(True, frame.header.ack_request)
|
|
self.assertEqual(0, frame.header.frame_version)
|
|
self.assertEqual(0x56, frame.header.seq)
|
|
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
|
|
self.assertEqual(0xface, frame.header.dest_pan_id)
|
|
self.assertEqual(0xdead, frame.header.dest_address.rloc)
|
|
self.assertEqual(0xface, frame.header.src_pan_id)
|
|
self.assertEqual(bytearray(reversed([0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0, 0x0d])),
|
|
frame.header.src_address.mac_address)
|
|
|
|
self.assertEqual(bytearray([0x12, 0x34]), frame.payload.data)
|
|
|
|
def test_should_parse_data_frame_with_extended_and_short_addresses(self):
|
|
frame = mac802154.MacFrame()
|
|
frame.parse(io.BytesIO(bytearray([0x61, 0x8c,
|
|
0x56,
|
|
0xce, 0xfa,
|
|
0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe, 0xef,
|
|
0x0d, 0xf0,
|
|
0x12, 0x34,
|
|
0xfe, 0xdc])))
|
|
|
|
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
|
|
self.assertEqual(False, frame.header.frame_pending)
|
|
self.assertEqual(True, frame.header.ack_request)
|
|
self.assertEqual(0, frame.header.frame_version)
|
|
self.assertEqual(0x56, frame.header.seq)
|
|
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
|
|
self.assertEqual(0xface, frame.header.dest_pan_id)
|
|
self.assertEqual(bytearray(reversed([0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe, 0xef])),
|
|
frame.header.dest_address.mac_address)
|
|
self.assertEqual(0xface, frame.header.src_pan_id)
|
|
self.assertEqual(0xf00d, frame.header.src_address.rloc)
|
|
|
|
self.assertEqual(bytearray([0x12, 0x34]), frame.payload.data)
|
|
|
|
def test_should_parse_data_request_command(self):
|
|
frame = mac802154.MacFrame()
|
|
frame.parse(io.BytesIO(bytearray([0x63, 0x88,
|
|
0x78,
|
|
0xce, 0xfa,
|
|
0xad, 0xde,
|
|
0x0d, 0xf0,
|
|
0x04,
|
|
0xfe, 0xdc])))
|
|
|
|
self.assertEqual(mac802154.MacHeader.FrameType.COMMAND, frame.header.frame_type)
|
|
self.assertEqual(False, frame.header.frame_pending)
|
|
self.assertEqual(True, frame.header.ack_request)
|
|
self.assertEqual(0, frame.header.frame_version)
|
|
self.assertEqual(0x78, frame.header.seq)
|
|
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
|
|
self.assertEqual(0xface, frame.header.dest_pan_id)
|
|
self.assertEqual(0xdead, frame.header.dest_address.rloc)
|
|
self.assertEqual(0xface, frame.header.src_pan_id)
|
|
self.assertEqual(0xf00d, frame.header.src_address.rloc)
|
|
|
|
self.assertEqual(bytearray([]), frame.payload.data)
|
|
|
|
def test_should_decrypt_data_frame(self):
|
|
|
|
mac802154.DeviceDescriptors.add(0x2001, MacAddress(bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x07]),
|
|
MacAddressType.LONG))
|
|
|
|
frame = mac802154.MacFrame()
|
|
frame.parse(io.BytesIO(bytearray([0x69, 0x98, 0x68, # FC, seq
|
|
0xce, 0xfa, # Pan Id
|
|
0x00, 0x20, # Dst addr
|
|
0x01, 0x20, # Src addr
|
|
0x0d, 0x00, 0x00, 0x00, 0x00, 0x01, # Aux Security Header
|
|
0xb5, 0x5a, 0x0d, 0x8e, 0x18, 0x5c, 0xb1, 0x06, # Payload
|
|
0xc4, 0x6f, 0x7d, 0x6b, 0xb5, 0x4a, 0x87, 0x14,
|
|
0xae, 0xdd, 0x8e, 0xb7, 0x37, 0x62, 0x27, 0x48,
|
|
0xc9, 0x53, 0x0c, 0x44, 0x31, 0x59, 0x8b, 0xa2,
|
|
0x83, 0x59, 0xa1, 0x43,
|
|
0x74, 0xe0, 0x2a, 0xf6, # MIC (valid)
|
|
0x99, 0xfc]))) # FCS (valid)
|
|
|
|
self.assertEqual(mac802154.MacHeader.FrameType.DATA, frame.header.frame_type)
|
|
self.assertEqual(False, frame.header.frame_pending)
|
|
self.assertEqual(True, frame.header.ack_request)
|
|
self.assertEqual(1, frame.header.frame_version)
|
|
self.assertEqual(0x68, frame.header.seq)
|
|
self.assertEqual(bytearray([0x99, 0xfc]), frame.header.fcs)
|
|
self.assertEqual(0xface, frame.header.dest_pan_id)
|
|
self.assertEqual(0x2000, frame.header.dest_address.rloc)
|
|
self.assertEqual(0xface, frame.header.src_pan_id)
|
|
self.assertEqual(0x2001, frame.header.src_address.rloc)
|
|
|
|
self.assertEqual(0, frame.header.aux_sec_header.frame_counter)
|
|
self.assertEqual(5, frame.header.aux_sec_header.security_level)
|
|
|
|
self.assertEqual(bytes(bytearray([0x7c, 0x77, 0x80, 0xf0, 0x4d, 0x4d, 0x4d, 0x4d,
|
|
0xe0, 0x04, 0x44, 0x02, 0x44, 0x66, 0x13, 0x5f,
|
|
0x22, 0x80, 0xb1, 0x61, 0x02, 0x61, 0x73, 0x11,
|
|
0x2a, 0xff, 0x01, 0x08, 0x16, 0x6e, 0x0a, 0x00,
|
|
0x00, 0x00, 0x00, 0x07])), frame.payload.data)
|
|
|
|
def test_should_decrypt_command_frame(self):
|
|
frame = mac802154.MacFrame()
|
|
frame.parse(io.BytesIO(bytearray([0x6b, 0xdc, 0xce, 0xce, 0xfa, 0x02, 0x00, 0x00,
|
|
0x00, 0x00, 0x0a, 0x6e, 0x16, 0x03, 0x00, 0x00,
|
|
0x00, 0x00, 0x0a, 0x6e, 0x16, 0x0d, 0x00, 0x00,
|
|
0x00, 0x00, 0x01, 0x04, 0x2d, 0xbc, 0x12, 0xbe,
|
|
0x0a, 0x4f])))
|
|
|
|
self.assertEqual(mac802154.MacHeader.FrameType.COMMAND, frame.header.frame_type)
|
|
self.assertEqual(False, frame.header.frame_pending)
|
|
self.assertEqual(True, frame.header.ack_request)
|
|
self.assertEqual(1, frame.header.frame_version)
|
|
self.assertEqual(206, frame.header.seq)
|
|
self.assertEqual(bytearray([0x0a, 0x4f]), frame.header.fcs)
|
|
self.assertEqual(0xface, frame.header.dest_pan_id)
|
|
self.assertEqual(bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x02]),
|
|
frame.header.dest_address.mac_address)
|
|
self.assertEqual(0xface, frame.header.src_pan_id)
|
|
self.assertEqual(bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x03]),
|
|
frame.header.src_address.mac_address)
|
|
|
|
self.assertEqual(0, frame.header.aux_sec_header.frame_counter)
|
|
self.assertEqual(5, frame.header.aux_sec_header.security_level)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|