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_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)
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_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_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 _parse_mac_frame(self, data):
mac_frame = mac802154.MacFrame()
mac_frame.parse(data)
return mac_frame
