| #!/usr/bin/env python3 |
| # |
| # 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([0x4]), 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, # MIC (valid) |
| 0x74, |
| 0xe0, |
| 0x2a, |
| 0xf6, |
| 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() |
