mirror of
https://github.com/espressif/openthread.git
synced 2026-08-03 01:17:46 +00:00
[thread-cert] refactor case 8.1.1 using pktverify (#5883)
This commit is contained in:
@@ -31,25 +31,45 @@ import unittest
|
||||
|
||||
import command
|
||||
import dtls
|
||||
import mle
|
||||
import thread_cert
|
||||
|
||||
from command import CheckType
|
||||
from pktverify.consts import MLE_DISCOVERY_REQUEST, MLE_DISCOVERY_RESPONSE, HANDSHAKE_CLIENT_HELLO, HANDSHAKE_SERVER_HELLO, HANDSHAKE_SERVER_KEY_EXCHANGE, HANDSHAKE_SERVER_HELLO_DONE, HANDSHAKE_CLIENT_KEY_EXCHANGE, HANDSHAKE_HELLO_VERIFY_REQUEST, CONTENT_APPLICATION_DATA, NM_EXTENDED_PAN_ID_TLV, NM_NETWORK_NAME_TLV, NM_STEERING_DATA_TLV, NM_COMMISSIONER_UDP_PORT_TLV, NM_JOINER_UDP_PORT_TLV, NM_DISCOVERY_REQUEST_TLV, NM_DISCOVERY_RESPONSE_TLV, THREAD_DISCOVERY_TLV, CONTENT_CHANGE_CIPHER_SPEC, CONTENT_HANDSHAKE, CONTENT_ALERT
|
||||
from pktverify.packet_verifier import PacketVerifier
|
||||
|
||||
COMMISSIONER = 1
|
||||
JOINER = 2
|
||||
|
||||
# Test Purpose and Description:
|
||||
# -----------------------------
|
||||
# The purpose of this test case is to verify the DTLS sessions between the
|
||||
# on-mesh Commissioner and a Joiner when the correct PSKd is used
|
||||
#
|
||||
# Note that many of the messages/records exchanged are encrypted
|
||||
# and cannot be observe
|
||||
#
|
||||
# Test Topology:
|
||||
# -------------
|
||||
# Commissioner
|
||||
# |
|
||||
# Joiner
|
||||
#
|
||||
# DUT Types:
|
||||
# ----------
|
||||
# Commissioner
|
||||
# Joiner
|
||||
|
||||
|
||||
class Cert_8_1_01_Commissioning(thread_cert.TestCase):
|
||||
SUPPORT_NCP = False
|
||||
|
||||
TOPOLOGY = {
|
||||
COMMISSIONER: {
|
||||
'name': 'COMMISSIONER',
|
||||
'masterkey': '00112233445566778899aabbccddeeff',
|
||||
'mode': 'rdn',
|
||||
'panid': 0xface
|
||||
},
|
||||
JOINER: {
|
||||
'name': 'JOINER',
|
||||
'masterkey': 'deadbeefdeadbeefdeadbeefdeadbeef',
|
||||
'mode': 'rdn',
|
||||
'router_selection_jitter': 1
|
||||
@@ -73,71 +93,181 @@ class Cert_8_1_01_Commissioning(thread_cert.TestCase):
|
||||
self.nodes[JOINER].get_masterkey(),
|
||||
self.nodes[COMMISSIONER].get_masterkey(),
|
||||
)
|
||||
|
||||
joiner_messages = self.simulator.get_messages_sent_by(JOINER)
|
||||
commissioner_messages = self.simulator.get_messages_sent_by(COMMISSIONER)
|
||||
|
||||
# 2 - N/A
|
||||
|
||||
# 3 - Joiner_1
|
||||
msg = joiner_messages.next_mle_message(mle.CommandType.DISCOVERY_REQUEST)
|
||||
command.check_discovery_request(msg, thread_version=self.nodes[JOINER].version)
|
||||
request_src_addr = msg.mac_header.src_address
|
||||
|
||||
# 4 - Commissioner
|
||||
msg = commissioner_messages.next_mle_message(mle.CommandType.DISCOVERY_RESPONSE)
|
||||
command.check_discovery_response(msg,
|
||||
request_src_addr,
|
||||
steering_data=CheckType.CONTAIN,
|
||||
thread_version=self.nodes[COMMISSIONER].version)
|
||||
udp_port_set_by_commissioner = command.get_joiner_udp_port_in_discovery_response(msg)
|
||||
|
||||
# 5.2 - Joiner_1
|
||||
msg = joiner_messages.next_dtls_message(dtls.ContentType.HANDSHAKE, dtls.HandshakeType.CLIENT_HELLO)
|
||||
self.assertEqual(msg.get_dst_udp_port(), udp_port_set_by_commissioner)
|
||||
|
||||
# 5.3 - Commissioner
|
||||
msg = commissioner_messages.next_dtls_message(dtls.ContentType.HANDSHAKE,
|
||||
dtls.HandshakeType.HELLO_VERIFY_REQUEST)
|
||||
commissioner_cookie = msg.dtls.body.cookie
|
||||
|
||||
# 5.4 - Joiner_1
|
||||
msg = joiner_messages.next_dtls_message(dtls.ContentType.HANDSHAKE, dtls.HandshakeType.CLIENT_HELLO)
|
||||
self.assertEqual(commissioner_cookie, msg.dtls.body.cookie)
|
||||
self.assertEqual(msg.get_dst_udp_port(), udp_port_set_by_commissioner)
|
||||
|
||||
# 5.5 - Commissioner
|
||||
commissioner_messages.next_dtls_message(dtls.ContentType.HANDSHAKE, dtls.HandshakeType.SERVER_HELLO)
|
||||
commissioner_messages.next_dtls_message(dtls.ContentType.HANDSHAKE, dtls.HandshakeType.SERVER_KEY_EXCHANGE)
|
||||
commissioner_messages.next_dtls_message(dtls.ContentType.HANDSHAKE, dtls.HandshakeType.SERVER_HELLO_DONE)
|
||||
|
||||
# 5.6 - Joiner_1
|
||||
msg = joiner_messages.next_dtls_message(dtls.ContentType.HANDSHAKE, dtls.HandshakeType.CLIENT_KEY_EXCHANGE)
|
||||
self.assertEqual(msg.get_dst_udp_port(), udp_port_set_by_commissioner)
|
||||
msg = joiner_messages.next_dtls_message(dtls.ContentType.CHANGE_CIPHER_SPEC)
|
||||
self.assertEqual(msg.get_dst_udp_port(), udp_port_set_by_commissioner)
|
||||
|
||||
# TODO(wgtdkp): It's required to verify DTLS FINISHED message here.
|
||||
# Currently not handled as it is encrypted.
|
||||
|
||||
# 5.7 - Commissioner
|
||||
commissioner_messages.next_dtls_message(dtls.ContentType.CHANGE_CIPHER_SPEC)
|
||||
|
||||
# TODO(wgtdkp): It's required to verify DTLS FINISHED message here.
|
||||
# Currently not handled as it is encrypted.
|
||||
|
||||
# 5.8,9,10,11
|
||||
# - Joiner_1
|
||||
# - Joiner
|
||||
command.check_joiner_commissioning_messages(joiner_messages.commissioning_messages)
|
||||
# - Commissioner
|
||||
command.check_commissioner_commissioning_messages(commissioner_messages.commissioning_messages)
|
||||
# As commissioner is also joiner router
|
||||
command.check_joiner_router_commissioning_messages(commissioner_messages.commissioning_messages)
|
||||
|
||||
self.nodes[JOINER].thread_start()
|
||||
self.simulator.go(5)
|
||||
self.assertEqual(self.nodes[JOINER].get_state(), 'router')
|
||||
|
||||
def verify(self, pv):
|
||||
pkts = pv.pkts
|
||||
pv.summary.show()
|
||||
|
||||
COMMISSIONER = pv.vars['COMMISSIONER']
|
||||
COMMISSIONER_VERSION = pv.vars['COMMISSIONER_VERSION']
|
||||
JOINER_VERSION = pv.vars['JOINER_VERSION']
|
||||
|
||||
# Step 3: Joiner sends MLE Discovery Request
|
||||
# MLE Discovery Request message MUST have these values:
|
||||
# - MLE Security Suite: 255 (No MLE Security)
|
||||
# - Thread Discovery TLV
|
||||
# Sub-TLVs:
|
||||
# - Discovery Request TLV
|
||||
# - Protocol Version: 2 or 3
|
||||
# (depends on the Thread stack version in testing)
|
||||
pkts.filter_mle_cmd(MLE_DISCOVERY_REQUEST).\
|
||||
filter_LLARMA().\
|
||||
filter(lambda p:
|
||||
[THREAD_DISCOVERY_TLV] == p.mle.tlv.type and\
|
||||
[NM_DISCOVERY_REQUEST_TLV] == p.thread_meshcop.tlv.type and\
|
||||
p.thread_meshcop.tlv.discovery_req_ver == JOINER_VERSION
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# Step 4: Commissioner sends MLE Discovery Response
|
||||
# MLE Discovery Response message MUST have these values:
|
||||
# - MLE Security Suite: 255 (No MLE Security)
|
||||
# - Source Address in IEEE 802.15.4 header MUST be set to
|
||||
# the MAC Extended Address (64-bit)
|
||||
# - Destination Address in IEEE 802.15.4 header MUST be
|
||||
# set to Discovery Request Source Address
|
||||
# - Thread Discovery TLV
|
||||
# Sub-TLVs:
|
||||
# - Discovery Request TLV
|
||||
# - Protocol Version: 2 or 3
|
||||
# (depends on the Thread stack version in testing)
|
||||
# - Extended PAN ID TLV
|
||||
# - Joiner UDP Port TLV
|
||||
# - Network Name TLV
|
||||
# - Steering Data TLV
|
||||
# - Commissioner UDP Port TLV (optional)
|
||||
_rs_pkt = pkts.filter_wpan_src64(COMMISSIONER).\
|
||||
filter_mle_cmd(MLE_DISCOVERY_RESPONSE).\
|
||||
filter(lambda p: {
|
||||
NM_EXTENDED_PAN_ID_TLV,
|
||||
NM_NETWORK_NAME_TLV,
|
||||
NM_STEERING_DATA_TLV,
|
||||
NM_JOINER_UDP_PORT_TLV,
|
||||
NM_DISCOVERY_RESPONSE_TLV
|
||||
} <= set(p.thread_meshcop.tlv.type) and\
|
||||
p.thread_meshcop.tlv.discovery_rsp_ver == COMMISSIONER_VERSION
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# Step 5: Verify the following details occur in the exchange between
|
||||
# Joiner and the Commissioner
|
||||
# 1. UDP port (Specified by the Commissioner: in Discovery Response)
|
||||
# is used as destination port for UDP datagrams from Joiner to
|
||||
# the Commissioner.
|
||||
|
||||
# 2. Joiner sends an initial DTLS-ClientHello handshake record to the
|
||||
# Commissioner
|
||||
pkts.filter_wpan_dst64(COMMISSIONER).\
|
||||
filter(lambda p:
|
||||
p.dtls.handshake.type == [HANDSHAKE_CLIENT_HELLO] and\
|
||||
p.udp.srcport in _rs_pkt.thread_meshcop.tlv.udp_port and\
|
||||
p.udp.dstport in _rs_pkt.thread_meshcop.tlv.udp_port
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# 3. The Commissioner receives the initial DTLS-ClientHello handshake
|
||||
# record and sends a DTLS-HelloVerifyRequest handshake record Joiner
|
||||
_pkt = pkts.filter_wpan_src64(COMMISSIONER).\
|
||||
filter(lambda p: p.dtls.handshake.type == [HANDSHAKE_HELLO_VERIFY_REQUEST]).\
|
||||
must_next()
|
||||
_pkt.must_verify(lambda p: p.dtls.handshake.cookie is not None)
|
||||
|
||||
# 4. Joiner receives the DTLS-HelloVerifyRequest handshake record and sends
|
||||
# a subsequent DTLS-ClientHello handshake record in one UDP datagram to the
|
||||
# Commissioner
|
||||
# Verify that both DTLS-HelloVerifyRequest and subsequent DTLS-ClientHello
|
||||
# contain the same cookie
|
||||
pkts.filter_wpan_dst64(COMMISSIONER).\
|
||||
filter(lambda p:
|
||||
p.dtls.handshake.type == [HANDSHAKE_CLIENT_HELLO] and\
|
||||
p.dtls.handshake.cookie == _pkt.dtls.handshake.cookie
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# 5. Commissioner must correctly receive the subsequent DTLSClientHello
|
||||
# handshake record and then send, in order, DTLSServerHello,
|
||||
# DTLS-ServerKeyExchange and DTLSServerHelloDone handshake records to Joiner
|
||||
pkts.filter_wpan_src64(COMMISSIONER).\
|
||||
filter(lambda p:
|
||||
p.dtls.handshake.type == [HANDSHAKE_SERVER_HELLO,
|
||||
HANDSHAKE_SERVER_KEY_EXCHANGE,
|
||||
HANDSHAKE_SERVER_HELLO_DONE]
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# 6. Joiner receives the DTLS-ServerHello, DTLSServerKeyExchange and
|
||||
# DTLS-ServerHelloDone handshake records and sends, in order,
|
||||
# a DTLS-ClientKeyExchange handshake record,
|
||||
# a DTLS-ChangeCipherSpec record and
|
||||
# an encrypted DTLS-Finished handshake record to the Commissioner.
|
||||
pkts.filter_wpan_dst64(COMMISSIONER).\
|
||||
filter(lambda p:
|
||||
p.dtls.handshake.type == [HANDSHAKE_CLIENT_KEY_EXCHANGE] and\
|
||||
{
|
||||
CONTENT_CHANGE_CIPHER_SPEC,
|
||||
CONTENT_HANDSHAKE
|
||||
} == set(p.dtls.record.content_type)
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# 7. Commissioner receives the DTLS-ClientKeyExchange handshake record, the
|
||||
# DTLS-ChangeCipherSpec record and the encrypted DTLS-Finished handshake record,
|
||||
# and sends a DTLS-ChangeCipherSpec record and an encrypted DTLSFinished handshake
|
||||
# record in that order to Joiner
|
||||
pkts.filter_wpan_src64(COMMISSIONER).\
|
||||
filter(lambda p: {
|
||||
CONTENT_CHANGE_CIPHER_SPEC,
|
||||
CONTENT_HANDSHAKE
|
||||
} == set(p.dtls.record.content_type)
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# 8. Joiner receives the DTLS-ChangeCipherSpec record and the encrypted DTLS-Finished
|
||||
# handshake record and sends a JOIN_FIN.req message in an encrypted DTLS-ApplicationData
|
||||
# record in a single UDP datagram to Commissioner.
|
||||
pkts.filter_wpan_dst64(COMMISSIONER).\
|
||||
filter(lambda p:
|
||||
[CONTENT_APPLICATION_DATA] == p.dtls.record.content_type
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# 9. Commissioner receives the encrypted DTLS-ApplicationData record and sends a
|
||||
# JOIN_FIN.rsp message in an encrypted DTLS-ApplicationData record in a single
|
||||
# UDP datagram to Joiner
|
||||
pkts.filter_wpan_src64(COMMISSIONER).\
|
||||
filter(lambda p:
|
||||
[CONTENT_APPLICATION_DATA] == p.dtls.record.content_type
|
||||
).\
|
||||
must_next()
|
||||
|
||||
# 10. Commissioner sends an encrypted JOIN_ENT.ntf message to Joiner
|
||||
|
||||
# 11. Joiner receives the encrypted JOIN_ENT.ntf message and sends an encrypted
|
||||
# JOIN_ENT.ntf dummy response to Commissioner
|
||||
|
||||
# Check Step 8 ~ 11 in test()
|
||||
|
||||
# 12. Joiner sends an encrypted DTLS-Alert record with a code of 0 (close_notify)
|
||||
# to Commissioner
|
||||
pkts.filter_wpan_dst64(COMMISSIONER).\
|
||||
filter(lambda p:
|
||||
[CONTENT_ALERT] == p.dtls.record.content_type
|
||||
).\
|
||||
must_next()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
||||
@@ -183,6 +183,7 @@ NM_JOINER_UDP_PORT_TLV = 18
|
||||
NM_DELAY_TIMER_TLV = 52
|
||||
NM_CHANNEL_MASK_TLV = 53
|
||||
NM_ENERGY_LIST_TLV = 57
|
||||
NM_DISCOVERY_REQUEST_TLV = 128
|
||||
NM_DISCOVERY_RESPONSE_TLV = 129
|
||||
|
||||
# Diagnostic TLVs
|
||||
@@ -218,6 +219,7 @@ HANDSHAKE_FINISHED = 20
|
||||
CONTENT_CHANGE_CIPHER_SPEC = 20
|
||||
CONTENT_ALERT = 21
|
||||
CONTENT_HANDSHAKE = 22
|
||||
CONTENT_APPLICATION_DATA = 23
|
||||
|
||||
# Network Data TLVs
|
||||
NWD_HAS_ROUTER_TLV = 0
|
||||
|
||||
@@ -278,6 +278,7 @@ _LAYER_FIELDS = {
|
||||
|
||||
# MLE
|
||||
'mle.cmd': _auto,
|
||||
'mle.sec_suite': _hex,
|
||||
'mle.tlv.type': _list(_dec),
|
||||
'mle.tlv.len': _list(_dec),
|
||||
'mle.tlv.mode.receiver_on_idle': _auto,
|
||||
@@ -568,6 +569,8 @@ _LAYER_FIELDS = {
|
||||
'thread_meshcop.tlv.chan_mask_page': _auto,
|
||||
'thread_meshcop.tlv.chan_mask_len': _auto,
|
||||
'thread_meshcop.tlv.chan_mask_mask': _bytes,
|
||||
'thread_meshcop.tlv.discovery_req_ver': _auto,
|
||||
'thread_meshcop.tlv.discovery_rsp_ver': _auto,
|
||||
'thread_meshcop.tlv.energy_list': _list(_auto),
|
||||
'thread_meshcop.tlv.pan_id': _auto,
|
||||
'thread_meshcop.tlv.xpan_id': _bytes,
|
||||
@@ -583,6 +586,7 @@ _LAYER_FIELDS = {
|
||||
'thread_meshcop.tlv.state': _auto,
|
||||
'thread_meshcop.tlv.steering_data': _list(_auto),
|
||||
'thread_meshcop.tlv.unknown': _bytes,
|
||||
'thread_meshcop.tlv.udp_port': _list(_auto),
|
||||
'thread_meshcop.tlv.ba_locator': _auto,
|
||||
'thread_meshcop.tlv.active_tstamp': _auto,
|
||||
'thread_meshcop.tlv.ipv6_addr': _list(_ipv6_addr),
|
||||
|
||||
@@ -165,6 +165,11 @@ class PacketVerifier(object):
|
||||
logging.info("add extra var: %s = %s", k, v)
|
||||
self._vars[k] = v
|
||||
|
||||
for i, topo in self.test_info.topology.items():
|
||||
name = self.test_info.get_node_name(i)
|
||||
if topo['version']:
|
||||
self._vars[name + '_VERSION'] = {'1.1': 2, '1.2': 3}[topo['version']]
|
||||
|
||||
def verify_attached(self, child: str, parent: str = None, child_type: str = 'FTD', pkts=None) -> VerifyResult:
|
||||
"""
|
||||
Verify that the device attaches to the Thread network.
|
||||
|
||||
Reference in New Issue
Block a user