From 176325b5a62794fcf6aeb0320360d95df495a6b4 Mon Sep 17 00:00:00 2001 From: Jonathan Hui Date: Fri, 7 Oct 2016 15:50:09 -0700 Subject: [PATCH] Add Thread Certification 9.2.11 to automated test. (#764) --- tests/scripts/Makefile.am | 3 + .../thread-cert/Cert_9_2_11_MasterKey.py | 160 ++++++++++++++++++ tests/scripts/thread-cert/node.py | 12 +- 3 files changed, 173 insertions(+), 2 deletions(-) create mode 100755 tests/scripts/thread-cert/Cert_9_2_11_MasterKey.py diff --git a/tests/scripts/Makefile.am b/tests/scripts/Makefile.am index 736640931..18fba85e7 100644 --- a/tests/scripts/Makefile.am +++ b/tests/scripts/Makefile.am @@ -107,6 +107,7 @@ EXTRA_DIST = \ thread-cert/Cert_9_2_07_DelayTimer.py \ thread-cert/Cert_9_2_09_PendingPartition.py \ thread-cert/Cert_9_2_10_PendingPartition.py \ + thread-cert/Cert_9_2_11_MasterKey.py \ thread-cert/Cert_9_2_12_Announce.py \ thread-cert/Cert_9_2_13_EnergyScan.py \ thread-cert/Cert_9_2_14_PanIdQuery.py \ @@ -229,6 +230,7 @@ check_SCRIPTS += \ thread-cert/Cert_9_2_07_DelayTimer.py \ thread-cert/Cert_9_2_09_PendingPartition.py \ thread-cert/Cert_9_2_10_PendingPartition.py \ + thread-cert/Cert_9_2_11_MasterKey.py \ thread-cert/Cert_9_2_12_Announce.py \ thread-cert/Cert_9_2_13_EnergyScan.py \ thread-cert/Cert_9_2_14_PanIdQuery.py \ @@ -253,6 +255,7 @@ XFAIL_NCP_TESTS = \ thread-cert/Cert_9_2_07_DelayTimer.py \ thread-cert/Cert_9_2_09_PendingPartition.py \ thread-cert/Cert_9_2_10_PendingPartition.py \ + thread-cert/Cert_9_2_11_MasterKey.py \ thread-cert/Cert_9_2_12_Announce.py \ thread-cert/Cert_9_2_13_EnergyScan.py \ thread-cert/Cert_9_2_14_PanIdQuery.py \ diff --git a/tests/scripts/thread-cert/Cert_9_2_11_MasterKey.py b/tests/scripts/thread-cert/Cert_9_2_11_MasterKey.py new file mode 100755 index 000000000..77cbda0e8 --- /dev/null +++ b/tests/scripts/thread-cert/Cert_9_2_11_MasterKey.py @@ -0,0 +1,160 @@ +#!/usr/bin/python +# +# 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 time +import unittest + +import node + +KEY1 = '000102030405060708090a0b0c0d0e0f' +KEY2 = '0f0e0d0c0b0a09080706050403020100' + +CHANNEL_INIT = 19 +PANID_INIT = 0xface + +COMMISSIONER = 1 +LEADER = 2 +ROUTER1 = 3 +ED1 = 4 +SED1 = 5 + +class Cert_9_2_11_MasterKey(unittest.TestCase): + def setUp(self): + self.nodes = {} + for i in range(1,6): + self.nodes[i] = node.Node(i) + + self.nodes[COMMISSIONER].set_active_dataset(10, channel=CHANNEL_INIT, panid=PANID_INIT, master_key=KEY1) + self.nodes[COMMISSIONER].set_mode('rsdn') + self.nodes[COMMISSIONER].add_whitelist(self.nodes[LEADER].get_addr64()) + self.nodes[COMMISSIONER].enable_whitelist() + self.nodes[COMMISSIONER].set_router_selection_jitter(1) + + self.nodes[LEADER].set_active_dataset(10, channel=CHANNEL_INIT, panid=PANID_INIT, master_key=KEY1) + self.nodes[LEADER].set_mode('rsdn') + self.nodes[LEADER].add_whitelist(self.nodes[COMMISSIONER].get_addr64()) + self.nodes[LEADER].add_whitelist(self.nodes[ROUTER1].get_addr64()) + self.nodes[LEADER].enable_whitelist() + self.nodes[LEADER].set_router_selection_jitter(1) + + self.nodes[ROUTER1].set_active_dataset(10, channel=CHANNEL_INIT, panid=PANID_INIT, master_key=KEY1) + self.nodes[ROUTER1].set_mode('rsdn') + self.nodes[ROUTER1].add_whitelist(self.nodes[LEADER].get_addr64()) + self.nodes[ROUTER1].add_whitelist(self.nodes[ED1].get_addr64()) + self.nodes[ROUTER1].add_whitelist(self.nodes[SED1].get_addr64()) + self.nodes[ROUTER1].enable_whitelist() + self.nodes[ROUTER1].set_router_selection_jitter(1) + + self.nodes[ED1].set_active_dataset(10, channel=CHANNEL_INIT, panid=PANID_INIT, master_key=KEY1) + self.nodes[ED1].set_mode('rsn') + self.nodes[ED1].add_whitelist(self.nodes[ROUTER1].get_addr64()) + self.nodes[ED1].enable_whitelist() + self.nodes[ED1].set_router_selection_jitter(1) + + self.nodes[SED1].set_active_dataset(10, channel=CHANNEL_INIT, panid=PANID_INIT, master_key=KEY1) + self.nodes[SED1].set_mode('s') + self.nodes[SED1].add_whitelist(self.nodes[ROUTER1].get_addr64()) + self.nodes[SED1].enable_whitelist() + self.nodes[SED1].set_timeout(3) + self.nodes[SED1].set_router_selection_jitter(1) + + def tearDown(self): + for node in list(self.nodes.values()): + node.stop() + del self.nodes + + def test(self): + self.nodes[LEADER].start() + self.nodes[LEADER].set_state('leader') + self.assertEqual(self.nodes[LEADER].get_state(), 'leader') + + self.nodes[COMMISSIONER].start() + time.sleep(5) + self.assertEqual(self.nodes[COMMISSIONER].get_state(), 'router') + + self.nodes[ROUTER1].start() + time.sleep(5) + self.assertEqual(self.nodes[ROUTER1].get_state(), 'router') + + self.nodes[ED1].start() + time.sleep(5) + self.assertEqual(self.nodes[ED1].get_state(), 'child') + + self.nodes[SED1].start() + time.sleep(5) + self.assertEqual(self.nodes[SED1].get_state(), 'child') + + self.nodes[COMMISSIONER].send_mgmt_pending_set(pending_timestamp=10, + active_timestamp=70, + delay_timer=10000, + master_key=KEY2) + time.sleep(15) + + print(self.nodes[COMMISSIONER].get_masterkey()) + print(self.nodes[LEADER].get_masterkey()) + print(self.nodes[ROUTER1].get_masterkey()) + print(self.nodes[ED1].get_masterkey()) + print(self.nodes[SED1].get_masterkey()) + + self.assertEqual(self.nodes[COMMISSIONER].get_masterkey(), KEY2) + self.assertEqual(self.nodes[LEADER].get_masterkey(), KEY2) + self.assertEqual(self.nodes[ROUTER1].get_masterkey(), KEY2) + self.assertEqual(self.nodes[ED1].get_masterkey(), KEY2) + self.assertEqual(self.nodes[SED1].get_masterkey(), KEY2) + + ipaddrs = self.nodes[ED1].get_addrs() + for ipaddr in ipaddrs: + if ipaddr[0:4] != 'fe80': + self.assertTrue(self.nodes[LEADER].ping(ipaddr)) + + self.nodes[COMMISSIONER].send_mgmt_pending_set(pending_timestamp=20, + active_timestamp=30, + delay_timer=10000, + master_key=KEY1) + time.sleep(15) + + print(self.nodes[COMMISSIONER].get_masterkey()) + print(self.nodes[LEADER].get_masterkey()) + print(self.nodes[ROUTER1].get_masterkey()) + print(self.nodes[ED1].get_masterkey()) + print(self.nodes[SED1].get_masterkey()) + + self.assertEqual(self.nodes[COMMISSIONER].get_masterkey(), KEY1) + self.assertEqual(self.nodes[LEADER].get_masterkey(), KEY1) + self.assertEqual(self.nodes[ROUTER1].get_masterkey(), KEY1) + self.assertEqual(self.nodes[ED1].get_masterkey(), KEY1) + self.assertEqual(self.nodes[SED1].get_masterkey(), KEY1) + + ipaddrs = self.nodes[ED1].get_addrs() + for ipaddr in ipaddrs: + if ipaddr[0:4] != 'fe80': + self.assertTrue(self.nodes[LEADER].ping(ipaddr)) + +if __name__ == '__main__': + unittest.main() diff --git a/tests/scripts/thread-cert/node.py b/tests/scripts/thread-cert/node.py index 39384ec33..0f5e7ae0b 100755 --- a/tests/scripts/thread-cert/node.py +++ b/tests/scripts/thread-cert/node.py @@ -447,7 +447,7 @@ class Node: self.send_command(cmd) self.pexpect.expect('Done') - def set_active_dataset(self, timestamp, panid=None, channel=None): + def set_active_dataset(self, timestamp, panid=None, channel=None, master_key=None): self.send_command('dataset clear') self.pexpect.expect('Done') @@ -465,6 +465,11 @@ class Node: self.send_command(cmd) self.pexpect.expect('Done') + if master_key != None: + cmd = 'dataset masterkey ' + master_key + self.send_command(cmd) + self.pexpect.expect('Done') + self.send_command('dataset commit active') self.pexpect.expect('Done') @@ -499,7 +504,7 @@ class Node: self.pexpect.expect('Done') def send_mgmt_pending_set(self, pending_timestamp=None, active_timestamp=None, delay_timer=None, channel=None, - panid=None): + panid=None, master_key=None): cmd = 'dataset mgmtsetcommand pending ' if pending_timestamp != None: @@ -517,6 +522,9 @@ class Node: if panid != None: cmd += 'panid %d ' % panid + if master_key != None: + cmd += 'masterkey ' + master_key + ' ' + self.send_command(cmd) self.pexpect.expect('Done')