[thread-cert] refactor case 5.1.13 using pktverify (#5766)

This commit is contained in:
Jing Ma
2020-11-04 10:42:56 -08:00
committed by GitHub
parent a60cc488df
commit a698677dd4
@@ -31,19 +31,43 @@ import unittest
import mle
import thread_cert
from pktverify.consts import MLE_ADVERTISEMENT, MLE_LINK_REQUEST, MLE_LINK_ACCEPT, MLE_LINK_ACCEPT_AND_REQUEST, SOURCE_ADDRESS_TLV, CHALLENGE_TLV, RESPONSE_TLV, LINK_LAYER_FRAME_COUNTER_TLV, ROUTE64_TLV, ADDRESS16_TLV, LEADER_DATA_TLV, TLV_REQUEST_TLV, VERSION_TLV
from pktverify.packet_verifier import PacketVerifier
from pktverify.null_field import nullField
LEADER = 1
ROUTER = 2
# Test Purpose and Description:
# -----------------------------
# The purpose of this test case is to validate that when a router resets,
# it will synchronize upon returning by using the Router Synchronization
# after Reset procedure.
#
# Test Topology:
# -------------
# Leader
# |
# Router
#
# DUT Types:
# ----------
# Leader
# Router
class Cert_5_1_13_RouterReset(thread_cert.TestCase):
USE_MESSAGE_FACTORY = False
TOPOLOGY = {
LEADER: {
'name': 'LEADER',
'mode': 'rdn',
'panid': 0xface,
'allowlist': [ROUTER]
},
ROUTER: {
'name': 'ROUTER',
'mode': 'rdn',
'panid': 0xface,
'router_selection_jitter': 1,
@@ -64,8 +88,7 @@ class Cert_5_1_13_RouterReset(thread_cert.TestCase):
self.nodes[ROUTER].start()
self.simulator.go(5)
self.assertEqual(self.nodes[ROUTER].get_state(), 'router')
rloc16 = self.nodes[ROUTER].get_addr16()
self.collect_rloc16s()
self.nodes[ROUTER].reset()
self._setUpRouter()
@@ -74,70 +97,110 @@ class Cert_5_1_13_RouterReset(thread_cert.TestCase):
self.nodes[ROUTER].start()
self.simulator.go(5)
self.assertEqual(self.nodes[ROUTER].get_state(), 'router')
self.assertEqual(self.nodes[ROUTER].get_addr16(), rloc16)
leader_messages = self.simulator.get_messages_sent_by(LEADER)
router1_messages = self.simulator.get_messages_sent_by(ROUTER)
def verify(self, pv):
pkts = pv.pkts
pv.summary.show()
# 1 - All
leader_messages.next_mle_message(mle.CommandType.ADVERTISEMENT)
LEADER = pv.vars['LEADER']
ROUTER = pv.vars['ROUTER']
ROUTER_RLOC16 = pv.vars['ROUTER_RLOC16']
router1_messages.next_mle_message(mle.CommandType.PARENT_REQUEST)
leader_messages.next_mle_message(mle.CommandType.PARENT_RESPONSE)
# Step 1: Verify topology is formed correctly.
router1_messages.next_mle_message(mle.CommandType.CHILD_ID_REQUEST)
leader_messages.next_mle_message(mle.CommandType.CHILD_ID_RESPONSE)
# Step 2: Devices MUST send properly formatted MLE Advertisements with
# an IP Hop Limit of 255 to the Link-Local All Nodes multicast
# address (FF02::1).
# The following TLVs MUST be present in the MLE Advertisements:
# - Leader Data TLV
# - Route64 TLV
# - Source Address TLV
msg = router1_messages.next_coap_message("0.02")
msg.assertCoapMessageRequestUriPath("/a/as")
pv.verify_attached('ROUTER')
pkts.filter_wpan_src64(ROUTER).\
filter_LLANMA().\
filter_mle_cmd(MLE_ADVERTISEMENT).\
filter(lambda p: {
LEADER_DATA_TLV,
ROUTE64_TLV,
SOURCE_ADDRESS_TLV
} == set(p.mle.tlv.type) and\
p.ipv6.hlim == 255
).\
must_next()
pkts.filter_wpan_src64(LEADER).\
filter_LLANMA().\
filter_mle_cmd(MLE_ADVERTISEMENT).\
filter(lambda p: {
LEADER_DATA_TLV,
ROUTE64_TLV,
SOURCE_ADDRESS_TLV
} == set(p.mle.tlv.type) and\
p.ipv6.hlim == 255
).\
must_next()
msg = leader_messages.next_coap_message("2.04")
# Step 4: Router sends multicast Link Request message
#
# The Link Request Message MUST be sent to the Link-Local All Routers
# Routers multicast address (FF02::2) and contain the following TLVs:
# - Challenge TLV
# - TLV Request TLV
# - Address16 TLV
# - Route64 TLV
# - Version TLV
#
router1_messages.next_mle_message(mle.CommandType.LINK_REQUEST)
msg = leader_messages.next_mle_message_of_one_of_command_types(
mle.CommandType.LINK_ACCEPT_AND_REQUEST,
mle.CommandType.LINK_ACCEPT,
)
self.assertIsNotNone(msg)
_pkt_lq = pkts.filter_mle_cmd(MLE_LINK_REQUEST).\
filter_LLARMA().\
filter(lambda p: {
CHALLENGE_TLV,
VERSION_TLV,
TLV_REQUEST_TLV,
ADDRESS16_TLV,
ROUTE64_TLV
} <= set(p.mle.tlv.type) and\
p.mle.tlv.addr16 is nullField and \
p.mle.tlv.route64.id_mask is nullField
).\
must_next()
# 2 - Router1 / Leader
msg = router1_messages.next_mle_message(mle.CommandType.ADVERTISEMENT)
msg.assertSentWithHopLimit(255)
msg.assertSentToDestinationAddress("ff02::1")
msg.assertMleMessageContainsTlv(mle.SourceAddress)
msg.assertMleMessageContainsTlv(mle.LeaderData)
msg.assertMleMessageContainsTlv(mle.Route64)
# Step 5: Leader replies to Router with Link Accept message
# The following TLVs MUST be present in the Link Accept message:
# - Leader Data TLV
# - Link-layer Frame Counter TLV
# - Link Margin TLV
# - Response TLV
# - Source Address TLV
# - Version TLV
# - TLV Request TLV: Link Margin
# - Challenge TLV (optional)
# - MLE Frame Counter TLV (optional)
# The Challenge TLV MUST be included if the response is an
# Accept and Request message.
# Responses to multicast Link Requests MUST be delayed by a
# random time of up to MLE_MAX_RESPONSE_DELAY (1 second).
msg = leader_messages.next_mle_message(mle.CommandType.ADVERTISEMENT)
msg.assertSentWithHopLimit(255)
msg.assertSentToDestinationAddress("ff02::1")
msg.assertMleMessageContainsTlv(mle.SourceAddress)
msg.assertMleMessageContainsTlv(mle.LeaderData)
msg.assertMleMessageContainsTlv(mle.Route64)
# 4 - Router1
msg = router1_messages.next_mle_message(mle.CommandType.LINK_REQUEST)
msg.assertSentToDestinationAddress("ff02::2")
msg.assertMleMessageContainsTlv(mle.Challenge)
msg.assertMleMessageContainsTlv(mle.Version)
msg.assertMleMessageContainsTlv(mle.TlvRequest)
tlv_request = msg.get_mle_message_tlv(mle.TlvRequest)
self.assertIn(mle.TlvType.ROUTE64, tlv_request.tlvs)
self.assertIn(mle.TlvType.ADDRESS16, tlv_request.tlvs)
# 5 - Leader
msg = leader_messages.next_mle_message(mle.CommandType.LINK_ACCEPT)
msg.assertSentToNode(self.nodes[ROUTER])
msg.assertMleMessageContainsTlv(mle.SourceAddress)
msg.assertMleMessageContainsTlv(mle.LeaderData)
msg.assertMleMessageContainsTlv(mle.Response)
msg.assertMleMessageContainsTlv(mle.LinkLayerFrameCounter)
msg.assertMleMessageContainsOptionalTlv(mle.MleFrameCounter)
msg.assertMleMessageContainsTlv(mle.Address16)
msg.assertMleMessageContainsTlv(mle.Version)
msg.assertMleMessageContainsTlv(mle.Route64)
msg.assertMleMessageContainsOptionalTlv(mle.Challenge)
_pkt = pkts.filter_wpan_src64(LEADER).\
filter_wpan_dst64(ROUTER).\
filter_mle_cmd2(MLE_LINK_ACCEPT, MLE_LINK_ACCEPT_AND_REQUEST).\
filter(lambda p: {
ADDRESS16_TLV,
LEADER_DATA_TLV,
LINK_LAYER_FRAME_COUNTER_TLV,
RESPONSE_TLV,
ROUTE64_TLV,
SOURCE_ADDRESS_TLV,
VERSION_TLV
} <= set(p.mle.tlv.type) and\
p.mle.tlv.addr16 is not nullField and \
p.mle.tlv.route64.id_mask is not nullField and\
p.mle.tlv.addr16 == ROUTER_RLOC16 and\
p.sniff_timestamp - _pkt_lq.sniff_timestamp <= 1
).\
must_next()
if _pkt.mle.cmd == MLE_LINK_ACCEPT_AND_REQUEST:
_pkt.must_verify(lambda p: [CHALLENGE_TLV] in p.mle.tlv.type)
if __name__ == '__main__':