[thread-cert] refactor case 5.5.5 using pktverify (#5332)

This commit is contained in:
Jing Ma
2020-08-19 08:41:36 -07:00
committed by GitHub
parent 9fec90ffe1
commit 2ab2f9af3e
3 changed files with 86 additions and 2 deletions
@@ -30,6 +30,8 @@
import unittest
import thread_cert
from pktverify.consts import MLE_ADVERTISEMENT, MLE_LINK_REQUEST, MLE_PARENT_REQUEST, MLE_PARENT_RESPONSE, MLE_CHILD_ID_REQUEST, MLE_CHILD_ID_RESPONSE, ADDR_SOL_URI, VERSION_TLV, TLV_REQUEST_TLV, SOURCE_ADDRESS_TLV, LEADER_DATA_TLV, CHALLENGE_TLV, LINK_MARGIN_TLV, NL_MAC_EXTENDED_ADDRESS_TLV, NL_STATUS_TLV
from pktverify.packet_verifier import PacketVerifier
LEADER = 1
ROUTER1 = 2
@@ -42,23 +44,27 @@ REED1 = 17
class Cert_5_5_5_SplitMergeREED(thread_cert.TestCase):
TOPOLOGY = {
LEADER: {
'name': 'LEADER',
'mode': 'rsdn',
'panid': 0xface,
'whitelist': [ROUTER2, ROUTER3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, ROUTER15]
},
ROUTER1: {
'name': 'ROUTER_1',
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [ROUTER3]
},
ROUTER2: {
'name': 'ROUTER_2',
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER, REED1]
},
ROUTER3: {
'name': 'ROUTER_3',
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
@@ -137,6 +143,7 @@ class Cert_5_5_5_SplitMergeREED(thread_cert.TestCase):
'whitelist': [LEADER]
},
REED1: {
'name': 'REED',
'mode': 'rsdn',
'panid': 0xface,
'whitelist': [ROUTER2]
@@ -170,11 +177,71 @@ class Cert_5_5_5_SplitMergeREED(thread_cert.TestCase):
self.assertEqual(self.nodes[ROUTER1].get_state(), 'child')
self.assertEqual(self.nodes[REED1].get_state(), 'router')
self.collect_ipaddrs()
addrs = self.nodes[LEADER].get_addrs()
for addr in addrs:
if addr[0:4] != 'fe80':
self.assertTrue(self.nodes[ROUTER1].ping(addr))
def verify(self, pv):
pkts = pv.pkts
pv.summary.show()
LEADER = pv.vars['LEADER']
ROUTER_1 = pv.vars['ROUTER_1']
ROUTER_2 = pv.vars['ROUTER_2']
ROUTER_3 = pv.vars['ROUTER_3']
REED = pv.vars['REED']
router1_pkts = pkts.filter_wpan_src64(ROUTER_1)
router2_pkts = pkts.filter_wpan_src64(ROUTER_2)
reed_pkts = pkts.filter_wpan_src64(REED)
router2_pkts.filter_mle_cmd(MLE_CHILD_ID_RESPONSE).filter_wpan_dst64(REED).must_next()
_start_idx = router2_pkts.index
reed_pkts.index = _start_idx
# Step 3: Remove Router_3 from the network
# Router_1 attempt to re-attach to partition.
# Sends multicast Parent Request to Routers and REEDs
router1_pkts.range(_start_idx).must_next()
router1_pkts.filter_mle_cmd(MLE_PARENT_REQUEST).must_next()
# filter MLE_ADVERTISEMENT with 15 routing table entry:
# 1 byte ID Sequence + 8 bytes ID Mask + 15 bytes Routing Table Entry =
# 24 (Router64 tlv length)
pkts.range(_start_idx).filter_mle_cmd(MLE_ADVERTISEMENT).filter(lambda p: 24 in p.mle.tlv.len).must_next()
_end_idx = pkts.index
# Step 2: The DUT MUST NOT attempt to become an active router by sending an Address Solicit Request
reed_pkts.range(_start_idx, _end_idx).filter_coap_request(ADDR_SOL_URI).must_not_next()
# Step 4: The DUT send a Parent Response to Router_1
reed_pkts.filter_mle_cmd(MLE_PARENT_RESPONSE).must_next().must_verify(lambda p: p.wpan.dst64 == ROUTER_1)
# Step 5: Router_1 Send MLE Child ID Request to the DUT
router1_pkts.range(reed_pkts.index).filter_mle_cmd(MLE_CHILD_ID_REQUEST).must_next()
# Step 6: DUT send an Address Solicit Request to Leader,
# receives short address and becomes a router
pkts.range(_end_idx).filter_wpan_src64(REED).filter_coap_request(ADDR_SOL_URI).must_next().must_verify(
lambda p: {NL_MAC_EXTENDED_ADDRESS_TLV, NL_STATUS_TLV} <= set(p.coap.tlv.type))
# Step 7: DUT send a Multicast Link Request
pkts.filter_wpan_src64(REED).filter_mle_cmd(MLE_LINK_REQUEST).must_next().must_verify(lambda p: {
VERSION_TLV, TLV_REQUEST_TLV, SOURCE_ADDRESS_TLV, LEADER_DATA_TLV, CHALLENGE_TLV, LINK_MARGIN_TLV
} <= set(p.mle.tlv.type))
# Step 8: DUT send Child ID Response to Router_1
reed_pkts.filter_mle_cmd(MLE_CHILD_ID_RESPONSE).must_next().must_verify(lambda p: p.wpan.dst64 == ROUTER_1)
# Step 9: The DUT MUST route the ICMPv6 Echo request to the Leader.
# The DUT MUST route the ICMPv6 Echo reply back to Router_1
leader_mleid = pv.vars['LEADER_MLEID']
router1_mleid = pv.vars['ROUTER_1_MLEID']
reed_pkts.filter_ipv6_src_dst(router1_mleid, leader_mleid).filter_ping_request().must_next()
reed_pkts.filter_ipv6_src_dst(leader_mleid, router1_mleid).filter_ping_reply().must_next()
if __name__ == '__main__':
unittest.main()
+14 -2
View File
@@ -72,7 +72,7 @@ MLE_CHILD_ID_RESPONSE = 12
MLE_CHILD_UPDATE_REQUEST = 13
MLE_CHILD_UPDATE_RESPONSE = 14
# COAP URI
# COAP URIs
ADDR_QRY_URI = '/a/aq'
ADDR_NTF_URI = '/a/an'
ADDR_ERR_URI = '/a/ae'
@@ -81,7 +81,7 @@ ADDR_REL_URI = '/a/ar'
SVR_DATA_URI = '/a/sd'
ND_DATA_URI = '/a/nd'
# TLV
# MLE TLVs
SOURCE_ADDRESS_TLV = 0
MODE_TLV = 1
TIMEOUT_TLV = 2
@@ -110,6 +110,18 @@ ACTIVE_OPERATION_DATASET_TLV = 24
PENDING_OPERATION_DATASET_TLV = 25
THREAD_DISCOVERY_TLV = 26
# Network Layer TLVs
NL_TARGET_EID_TLV = 0
NL_MAC_EXTENDED_ADDRESS_TLV = 1
NL_RLOC16_TLV = 2
NL_ML_EID_TLV = 3
NL_STATUS_TLV = 4
NL_TIME_SINCE_LAST_TRANSACTION_TLV = 6
NL_ROUTER_MASK_TLV = 7
NL_ND_OPTION_TLV = 8
NL_ND_DATA_TLV = 9
NL_THREAD_NETWORK_DATA_TLV = 10
# DUA related constants
ADDRESS_QUERY_INITIAL_RETRY_DELAY = 15
@@ -433,6 +433,11 @@ class PacketFilter(object):
assert isinstance(addr, (str, Ipv6Addr))
return self.filter(lambda p: p.ipv6.dst == addr, **kwargs)
def filter_ipv6_src_dst(self, src_addr, dst_addr, **kwargs):
assert isinstance(src_addr, (str, Ipv6Addr))
assert isinstance(dst_addr, (str, Ipv6Addr))
return self.filter(lambda p: p.ipv6.src == src_addr and p.ipv6.dst == dst_addr, **kwargs)
def filter_LLANMA(self, **kwargs):
return self.filter(lambda p: p.ipv6.dst == consts.LINK_LOCAL_ALL_NODES_MULTICAST_ADDRESS, **kwargs)