mirror of
https://github.com/espressif/openthread.git
synced 2026-08-12 13:47:47 +00:00
[windows] remove unsupported platform (#3878)
This commit is contained in:
Kamil Sroka
committed by
Jonathan Hui
Jonathan Hui
parent
e482edb0f1
commit
c14cc60962
@@ -138,7 +138,6 @@ EXTRA_DIST = \
|
||||
network_data.py \
|
||||
network_layer.py \
|
||||
node.py \
|
||||
node_cli.py \
|
||||
pcap.py \
|
||||
simulator.py \
|
||||
sniffer.py \
|
||||
|
||||
@@ -1,77 +0,0 @@
|
||||
#!/usr/bin/env 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 unittest
|
||||
import ctypes
|
||||
|
||||
class GUID(ctypes.Structure):
|
||||
_fields_ = [("Data1", ctypes.c_uint),
|
||||
("Data2", ctypes.c_ushort),
|
||||
("Data3", ctypes.c_ushort),
|
||||
("Data4", ctypes.c_ubyte * 8)]
|
||||
|
||||
class otDeviceList(ctypes.Structure):
|
||||
_fields_ = [("aDevicesLength", ctypes.c_ushort),
|
||||
("aDevices", GUID * 64)]
|
||||
|
||||
class Cert_otLwf(unittest.TestCase):
|
||||
def setUp(self):
|
||||
|
||||
# Load the DLL
|
||||
self.Api = ctypes.WinDLL("otApi.dll")
|
||||
if self.Api == None:
|
||||
raise OSError("Failed to load otApi.dll!")
|
||||
|
||||
# Define the functions
|
||||
self.Api.otApiInit.restype = ctypes.c_void_p
|
||||
self.Api.otApiFinalize.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otApiFinalize.restype = None
|
||||
self.Api.otFreeMemory.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otFreeMemory.restype = None
|
||||
self.Api.otEnumerateDevices.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otEnumerateDevices.restype = ctypes.POINTER(otDeviceList)
|
||||
|
||||
def tearDown(self):
|
||||
if self.ApiInstance:
|
||||
self.Api.otApiFinalize(self.ApiInstance)
|
||||
|
||||
def test(self):
|
||||
# Instantiate the API
|
||||
self.ApiInstance = self.Api.otApiInit()
|
||||
# Assert that it didn't return NULL
|
||||
self.assertNotEqual(self.ApiInstance, None)
|
||||
# Query the device list
|
||||
devices = self.Api.otEnumerateDevices(self.ApiInstance)
|
||||
# Assert that it didn't return NULL
|
||||
self.assertNotEqual(devices, None)
|
||||
# Print the number of devices
|
||||
print("devices found: %d" % devices.contents.aDevicesLength)
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
+749
-114
File diff suppressed because it is too large
Load Diff
@@ -1,602 +0,0 @@
|
||||
#!/usr/bin/env 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 os
|
||||
import sys
|
||||
import time
|
||||
import ctypes
|
||||
|
||||
class otApi:
|
||||
def __init__(self, nodeid):
|
||||
self.verbose = int(float(os.getenv('VERBOSE', 0)))
|
||||
self.__init_dll(nodeid)
|
||||
|
||||
def __del__(self):
|
||||
self.Api.otNodeFinalize(self.otNode)
|
||||
|
||||
def set_mode(self, mode):
|
||||
if self.Api.otNodeSetMode(self.otNode, mode.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeSetMode failed!")
|
||||
|
||||
def interface_up(self):
|
||||
if self.Api.otNodeInterfaceUp(self.otNode) != 0:
|
||||
raise OSError("otNodeInterfaceUp failed!")
|
||||
|
||||
def interface_down(self):
|
||||
if self.Api.otNodeInterfaceDown(self.otNode) != 0:
|
||||
raise OSError("otNodeInterfaceDown failed!")
|
||||
|
||||
def thread_start(self):
|
||||
if self.Api.otNodeThreadStart(self.otNode) != 0:
|
||||
raise OSError("otNodeThreadStart failed!")
|
||||
|
||||
def thread_stop(self):
|
||||
if self.Api.otNodeThreadStop(self.otNode) != 0:
|
||||
raise OSError("otNodeThreadStop failed!")
|
||||
|
||||
def commissioner_start(self):
|
||||
if self.Api.otNodeCommissionerStart(self.otNode) != 0:
|
||||
raise OSError("otNodeCommissionerStart failed!")
|
||||
|
||||
def commissioner_add_joiner(self, addr, psk):
|
||||
if self.Api.otNodeCommissionerJoinerAdd(self.otNode, addr.encode('utf-8'), psk.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeCommissionerJoinerAdd failed!")
|
||||
|
||||
def joiner_start(self, pskd='', provisioning_url=''):
|
||||
if self.Api.otNodeJoinerStart(self.otNode, pskd.encode('utf-8'), provisioning_url.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeJoinerStart failed!")
|
||||
|
||||
def clear_whitelist(self):
|
||||
if self.Api.otNodeClearWhitelist(self.otNode) != 0:
|
||||
raise OSError("otNodeClearWhitelist failed!")
|
||||
|
||||
def enable_whitelist(self):
|
||||
if self.Api.otNodeEnableWhitelist(self.otNode) != 0:
|
||||
raise OSError("otNodeEnableWhitelist failed!")
|
||||
|
||||
def disable_whitelist(self):
|
||||
if self.Api.otNodeDisableWhitelist(self.otNode) != 0:
|
||||
raise OSError("otNodeDisableWhitelist failed!")
|
||||
|
||||
def add_whitelist(self, addr, rssi=None):
|
||||
if rssi == None:
|
||||
rssi = 0
|
||||
if self.Api.otNodeAddWhitelist(self.otNode, addr.encode('utf-8'), ctypes.c_byte(rssi)) != 0:
|
||||
raise OSError("otNodeAddWhitelist failed!")
|
||||
|
||||
def remove_whitelist(self, addr):
|
||||
if self.Api.otNodeRemoveWhitelist(self.otNode, addr.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeRemoveWhitelist failed!")
|
||||
|
||||
def get_addr16(self):
|
||||
return self.Api.otNodeGetAddr16(self.otNode)
|
||||
|
||||
def get_addr64(self):
|
||||
return self.Api.otNodeGetAddr64(self.otNode).decode('utf-8')
|
||||
|
||||
def get_eui64(self):
|
||||
return self.Api.otNodeGetEui64(self.otNode).decode('utf-8')
|
||||
|
||||
def get_joiner_id(self):
|
||||
return self.Api.otNodeGetJoinerId(self.otNode).decode('utf-8')
|
||||
|
||||
def get_channel(self):
|
||||
return self.Api.otNodeGetChannel(self.otNode)
|
||||
|
||||
def set_channel(self, channel):
|
||||
if self.Api.otNodeSetChannel(self.otNode, ctypes.c_ubyte(channel)) != 0:
|
||||
raise OSError("otNodeSetChannel failed!")
|
||||
|
||||
def get_masterkey(self):
|
||||
return self.Api.otNodeGetMasterkey(self.otNode).decode("utf-8")
|
||||
|
||||
def set_masterkey(self, masterkey):
|
||||
if self.Api.otNodeSetMasterkey(self.otNode, masterkey.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeSetMasterkey failed!")
|
||||
|
||||
def get_key_sequence_counter(self):
|
||||
return self.Api.otNodeGetKeySequenceCounter(self.otNode)
|
||||
|
||||
def set_key_sequence_counter(self, key_sequence_counter):
|
||||
if self.Api.otNodeSetKeySequenceCounter(self.otNode, ctypes.c_uint(key_sequence_counter)) != 0:
|
||||
raise OSError("otNodeSetKeySequenceCounter failed!")
|
||||
|
||||
def set_key_switch_guardtime(self, key_switch_guardtime):
|
||||
if self.Api.otNodeSetKeySwitchGuardTime(self.otNode, ctypes.c_uint(key_switch_guardtime)) != 0:
|
||||
raise OSError("otNodeSetKeySwitchGuardTime failed!")
|
||||
|
||||
def set_network_id_timeout(self, network_id_timeout):
|
||||
if self.Api.otNodeSetNetworkIdTimeout(self.otNode, ctypes.c_ubyte(network_id_timeout)) != 0:
|
||||
raise OSError("otNodeSetNetworkIdTimeout failed!")
|
||||
|
||||
def get_network_name(self):
|
||||
return self.Api.otNodeGetNetworkName(self.otNode).decode("utf-8")
|
||||
|
||||
def set_network_name(self, network_name):
|
||||
if self.Api.otNodeSetNetworkName(self.otNode, network_name.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeSetNetworkName failed!")
|
||||
|
||||
def get_panid(self):
|
||||
return int(self.Api.otNodeGetPanId(self.otNode))
|
||||
|
||||
def set_panid(self, panid):
|
||||
if self.Api.otNodeSetPanId(self.otNode, ctypes.c_ushort(panid)) != 0:
|
||||
raise OSError("otNodeSetPanId failed!")
|
||||
|
||||
def get_partition_id(self):
|
||||
return int(self.Api.otNodeGetPartitionId(self.otNode))
|
||||
|
||||
def set_partition_id(self, partition_id):
|
||||
if self.Api.otNodeSetPartitionId(self.otNode, ctypes.c_uint(partition_id)) != 0:
|
||||
raise OSError("otNodeSetPartitionId failed!")
|
||||
|
||||
def set_router_upgrade_threshold(self, threshold):
|
||||
if self.Api.otNodeSetRouterUpgradeThreshold(self.otNode, ctypes.c_ubyte(threshold)) != 0:
|
||||
raise OSError("otNodeSetRouterUpgradeThreshold failed!")
|
||||
|
||||
def set_router_downgrade_threshold(self, threshold):
|
||||
if self.Api.otNodeSetRouterDowngradeThreshold(self.otNode, ctypes.c_ubyte(threshold)) != 0:
|
||||
raise OSError("otNodeSetRouterDowngradeThreshold failed!")
|
||||
|
||||
def release_router_id(self, router_id):
|
||||
if self.Api.otNodeReleaseRouterId(self.otNode, ctypes.c_ubyte(router_id)) != 0:
|
||||
raise OSError("otNodeReleaseRouterId failed!")
|
||||
|
||||
def get_state(self):
|
||||
return self.Api.otNodeGetState(self.otNode).decode('utf-8')
|
||||
|
||||
def set_state(self, state):
|
||||
if self.Api.otNodeSetState(self.otNode, state.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeSetState failed!")
|
||||
|
||||
def get_timeout(self):
|
||||
return int(self.Api.otNodeGetTimeout(self.otNode))
|
||||
|
||||
def set_timeout(self, timeout):
|
||||
if self.Api.otNodeSetTimeout(self.otNode, ctypes.c_uint(timeout)) != 0:
|
||||
raise OSError("otNodeSetTimeout failed!")
|
||||
|
||||
def set_max_children(self, number):
|
||||
if self.Api.otNodeSetMaxChildren(self.otNode, ctypes.c_ubyte(number)) != 0:
|
||||
raise OSError("otNodeSetMaxChildren failed!")
|
||||
|
||||
def get_weight(self):
|
||||
return int(self.Api.otNodeGetWeight(self.otNode))
|
||||
|
||||
def set_weight(self, weight):
|
||||
if self.Api.otNodeSetWeight(self.otNode, ctypes.c_ubyte(weight)) != 0:
|
||||
raise OSError("otNodeSetWeight failed!")
|
||||
|
||||
def add_ipaddr(self, ipaddr):
|
||||
if self.Api.otNodeAddIpAddr(self.otNode, ipaddr.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeAddIpAddr failed!")
|
||||
|
||||
def get_addrs(self):
|
||||
return self.Api.otNodeGetAddrs(self.otNode).decode("utf-8").split("\n")
|
||||
|
||||
def add_service(self, enterpriseNumber, serviceData, serverData):
|
||||
raise OSError("otServerAddService wrapper not implemented!")
|
||||
|
||||
def remove_service(self, enterpriseNumber, serviceData):
|
||||
raise OSError("otServerRemoveService wrapper not implemented!")
|
||||
|
||||
def get_context_reuse_delay(self):
|
||||
return int(self.Api.otNodeGetContextReuseDelay(self.otNode))
|
||||
|
||||
def set_context_reuse_delay(self, delay):
|
||||
if self.Api.otNodeSetContextReuseDelay(self.otNode, ctypes.c_uint(delay)) != 0:
|
||||
raise OSError("otNodeSetContextReuseDelay failed!")
|
||||
|
||||
def add_prefix(self, prefix, flags, prf = 'med'):
|
||||
if self.Api.otNodeAddPrefix(self.otNode, prefix.encode('utf-8'), flags.encode('utf-8'), prf.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeAddPrefix failed!")
|
||||
|
||||
def remove_prefix(self, prefix):
|
||||
if self.Api.otNodeRemovePrefix(self.otNode, prefix.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeRemovePrefix failed!")
|
||||
|
||||
def add_route(self, prefix, prf = 'med'):
|
||||
if self.Api.otNodeAddRoute(self.otNode, prefix.encode('utf-8'), prf.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeAddRoute failed!")
|
||||
|
||||
def remove_route(self, prefix):
|
||||
if self.Api.otNodeRemoveRoute(self.otNode, prefix.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeRemovePrefix failed!")
|
||||
|
||||
def register_netdata(self):
|
||||
if self.Api.otNodeRegisterNetdata(self.otNode) != 0:
|
||||
raise OSError("otNodeRegisterNetdata failed!")
|
||||
|
||||
def energy_scan(self, mask, count, period, scan_duration, ipaddr):
|
||||
if self.Api.otNodeEnergyScan(self.otNode, ctypes.c_uint(mask), ctypes.c_ubyte(count), ctypes.c_ushort(period), ctypes.c_ushort(scan_duration), ipaddr.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeEnergyScan failed!")
|
||||
|
||||
def panid_query(self, panid, mask, ipaddr):
|
||||
if self.Api.otNodePanIdQuery(self.otNode, ctypes.c_ushort(panid), ctypes.c_uint(mask), ipaddr.encode('utf-8')) != 0:
|
||||
raise OSError("otNodePanIdQuery failed!")
|
||||
|
||||
def scan(self):
|
||||
return self.Api.otNodeScan(self.otNode).decode("utf-8").split("\n")
|
||||
|
||||
def ping(self, ipaddr, num_responses=1, size=None, timeout=5000):
|
||||
if size == None:
|
||||
size = 100
|
||||
numberOfResponders = self.Api.otNodePing(self.otNode, ipaddr.encode('utf-8'), ctypes.c_ushort(size),
|
||||
ctypes.c_uint(num_responses), ctypes.c_uint16(timeout))
|
||||
return numberOfResponders >= num_responses
|
||||
|
||||
def set_router_selection_jitter(self, jitter):
|
||||
if self.Api.otNodeSetRouterSelectionJitter(self.otNode, ctypes.c_ubyte(jitter)) != 0:
|
||||
raise OSError("otNodeSetRouterSelectionJitter failed!")
|
||||
|
||||
def set_active_dataset(self, timestamp, panid=None, channel=None, channel_mask=None, master_key=None):
|
||||
if panid == None:
|
||||
panid = 0
|
||||
if channel == None:
|
||||
channel = 0
|
||||
if channel_mask == None:
|
||||
channel_mask = 0
|
||||
if master_key == None:
|
||||
master_key = ""
|
||||
if self.Api.otNodeSetActiveDataset(
|
||||
self.otNode,
|
||||
ctypes.c_ulonglong(timestamp),
|
||||
ctypes.c_ushort(panid),
|
||||
ctypes.c_ushort(channel),
|
||||
ctypes.c_uint(channel_mask),
|
||||
master_key.encode('utf-8')
|
||||
) != 0:
|
||||
raise OSError("otNodeSetActiveDataset failed!")
|
||||
|
||||
def set_pending_dataset(self, pendingtimestamp, activetimestamp, panid=None, channel=None):
|
||||
if pendingtimestamp == None:
|
||||
pendingtimestamp = 0
|
||||
if activetimestamp == None:
|
||||
activetimestamp = 0
|
||||
if panid == None:
|
||||
panid = 0
|
||||
if channel == None:
|
||||
channel = 0
|
||||
if self.Api.otNodeSetPendingDataset(
|
||||
self.otNode,
|
||||
ctypes.c_ulonglong(activetimestamp),
|
||||
ctypes.c_ulonglong(pendingtimestamp),
|
||||
ctypes.c_ushort(panid),
|
||||
ctypes.c_ushort(channel)
|
||||
) != 0:
|
||||
raise OSError("otNodeSetPendingDataset failed!")
|
||||
|
||||
def announce_begin(self, mask, count, period, ipaddr):
|
||||
if self.Api.otNodeCommissionerAnnounceBegin(self.otNode, ctypes.c_uint(mask), ctypes.c_ubyte(count), ctypes.c_ushort(period), ipaddr.encode('utf-8')) != 0:
|
||||
raise OSError("otNodeCommissionerAnnounceBegin failed!")
|
||||
|
||||
def send_mgmt_active_set(self, active_timestamp=None, channel=None, channel_mask=None, extended_panid=None,
|
||||
panid=None, master_key=None, mesh_local=None, network_name=None, binary=None):
|
||||
if active_timestamp == None:
|
||||
active_timestamp = 0
|
||||
if panid == None:
|
||||
panid = 0
|
||||
if channel == None:
|
||||
channel = 0
|
||||
if channel_mask == None:
|
||||
channel_mask = 0
|
||||
if extended_panid == None:
|
||||
extended_panid = ""
|
||||
if master_key == None:
|
||||
master_key = ""
|
||||
if mesh_local == None:
|
||||
mesh_local = ""
|
||||
if network_name == None:
|
||||
network_name = ""
|
||||
if binary == None:
|
||||
binary = ""
|
||||
if self.Api.otNodeSendActiveSet(
|
||||
self.otNode,
|
||||
ctypes.c_ulonglong(active_timestamp),
|
||||
ctypes.c_ushort(panid),
|
||||
ctypes.c_ushort(channel),
|
||||
ctypes.c_uint(channel_mask),
|
||||
extended_panid.encode('utf-8'),
|
||||
master_key.encode('utf-8'),
|
||||
mesh_local.encode('utf-8'),
|
||||
network_name.encode('utf-8'),
|
||||
binary.encode('utf-8')
|
||||
) != 0:
|
||||
raise OSError("otNodeSendActiveSet failed!")
|
||||
|
||||
def send_mgmt_pending_set(self, pending_timestamp=None, active_timestamp=None, delay_timer=None, channel=None,
|
||||
panid=None, master_key=None, mesh_local=None, network_name=None):
|
||||
if pending_timestamp == None:
|
||||
pending_timestamp = 0
|
||||
if active_timestamp == None:
|
||||
active_timestamp = 0
|
||||
if delay_timer == None:
|
||||
delay_timer = 0
|
||||
if panid == None:
|
||||
panid = 0
|
||||
if channel == None:
|
||||
channel = 0
|
||||
if master_key == None:
|
||||
master_key = ""
|
||||
if mesh_local == None:
|
||||
mesh_local = ""
|
||||
if network_name == None:
|
||||
network_name = ""
|
||||
if self.Api.otNodeSendPendingSet(
|
||||
self.otNode,
|
||||
ctypes.c_ulonglong(active_timestamp),
|
||||
ctypes.c_ulonglong(pending_timestamp),
|
||||
ctypes.c_uint(delay_timer),
|
||||
ctypes.c_ushort(panid),
|
||||
ctypes.c_ushort(channel),
|
||||
master_key.encode('utf-8'),
|
||||
mesh_local.encode('utf-8'),
|
||||
network_name.encode('utf-8')
|
||||
) != 0:
|
||||
raise OSError("otNodeSendPendingSet failed!")
|
||||
|
||||
def log(self, message):
|
||||
self.Api.otNodeLog(message)
|
||||
|
||||
def __init_dll(self, nodeid):
|
||||
""" Initialize the API from a Windows DLL. """
|
||||
|
||||
# Load the DLL
|
||||
self.Api = ctypes.WinDLL("otnodeapi.dll")
|
||||
if self.Api == None:
|
||||
raise OSError("Failed to load otnodeapi.dll!")
|
||||
|
||||
# Define the functions
|
||||
self.Api.otNodeLog.argtypes = [ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeInit.argtypes = [ctypes.c_uint]
|
||||
self.Api.otNodeInit.restype = ctypes.c_void_p
|
||||
|
||||
self.Api.otNodeFinalize.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeSetMode.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeInterfaceUp.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeInterfaceDown.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeThreadStart.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeThreadStop.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeCommissionerStart.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeCommissionerJoinerAdd.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeCommissionerStop.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeJoinerStart.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeJoinerStop.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeClearWhitelist.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeEnableWhitelist.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeDisableWhitelist.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeAddWhitelist.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_byte]
|
||||
|
||||
self.Api.otNodeRemoveWhitelist.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeGetAddr16.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetAddr16.restype = ctypes.c_ushort
|
||||
|
||||
self.Api.otNodeGetAddr64.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetAddr64.restype = ctypes.c_char_p
|
||||
|
||||
self.Api.otNodeGetEui64.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetEui64.restype = ctypes.c_char_p
|
||||
|
||||
self.Api.otNodeGetJoinerId.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetJoinerId.restype = ctypes.c_char_p
|
||||
|
||||
self.Api.otNodeSetChannel.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ubyte]
|
||||
|
||||
self.Api.otNodeGetChannel.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetChannel.restype = ctypes.c_ubyte
|
||||
|
||||
self.Api.otNodeSetMasterkey.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeGetMasterkey.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetMasterkey.restype = ctypes.c_char_p
|
||||
|
||||
self.Api.otNodeGetKeySequenceCounter.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetKeySequenceCounter.restype = ctypes.c_uint
|
||||
|
||||
self.Api.otNodeSetKeySequenceCounter.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_uint]
|
||||
|
||||
self.Api.otNodeSetKeySwitchGuardTime.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_uint]
|
||||
|
||||
self.Api.otNodeSetNetworkIdTimeout.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ubyte]
|
||||
|
||||
self.Api.otNodeGetNetworkName.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetNetworkName.restype = ctypes.c_char_p
|
||||
|
||||
self.Api.otNodeSetNetworkName.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeGetPanId.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetPanId.restype = ctypes.c_ushort
|
||||
|
||||
self.Api.otNodeSetPanId.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ushort]
|
||||
|
||||
self.Api.otNodeGetPartitionId.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetPartitionId.restype = ctypes.c_uint
|
||||
|
||||
self.Api.otNodeSetPartitionId.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_uint]
|
||||
|
||||
self.Api.otNodeSetRouterUpgradeThreshold.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ubyte]
|
||||
|
||||
self.Api.otNodeSetRouterDowngradeThreshold.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ubyte]
|
||||
|
||||
self.Api.otNodeReleaseRouterId.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ubyte]
|
||||
|
||||
self.Api.otNodeGetState.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetState.restype = ctypes.c_char_p
|
||||
|
||||
self.Api.otNodeSetState.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeGetTimeout.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetTimeout.restype = ctypes.c_uint
|
||||
|
||||
self.Api.otNodeSetTimeout.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_uint]
|
||||
|
||||
self.Api.otNodeGetWeight.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetWeight.restype = ctypes.c_ubyte
|
||||
|
||||
self.Api.otNodeSetWeight.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ubyte]
|
||||
|
||||
self.Api.otNodeAddIpAddr.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeGetAddrs.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetAddrs.restype = ctypes.c_char_p
|
||||
|
||||
self.Api.otNodeGetContextReuseDelay.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeGetContextReuseDelay.restype = ctypes.c_uint
|
||||
|
||||
self.Api.otNodeSetContextReuseDelay.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_uint]
|
||||
|
||||
self.Api.otNodeAddPrefix.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeRemovePrefix.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeAddRoute.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeRemoveRoute.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeRegisterNetdata.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otNodeEnergyScan.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_uint,
|
||||
ctypes.c_ubyte,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodePanIdQuery.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_uint,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeScan.argtypes = [ctypes.c_void_p]
|
||||
self.Api.otNodeScan.restype = ctypes.c_char_p
|
||||
|
||||
self.Api.otNodePing.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_uint,
|
||||
ctypes.c_uint16]
|
||||
self.Api.otNodePing.restype = ctypes.c_uint
|
||||
|
||||
self.Api.otNodeSetRouterSelectionJitter.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ubyte]
|
||||
|
||||
self.Api.otNodeCommissionerAnnounceBegin.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_uint,
|
||||
ctypes.c_ubyte,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeSetActiveDataset.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ulonglong,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_uint,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeSetPendingDataset.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ulonglong,
|
||||
ctypes.c_ulonglong,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_ushort]
|
||||
|
||||
self.Api.otNodeSendPendingSet.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ulonglong,
|
||||
ctypes.c_ulonglong,
|
||||
ctypes.c_uint,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeSendActiveSet.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ulonglong,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_ushort,
|
||||
ctypes.c_uint,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p,
|
||||
ctypes.c_char_p]
|
||||
|
||||
self.Api.otNodeSetMaxChildren.argtypes = [ctypes.c_void_p,
|
||||
ctypes.c_ubyte]
|
||||
|
||||
# Initialize a new node
|
||||
self.otNode = self.Api.otNodeInit(ctypes.c_uint(nodeid))
|
||||
if self.otNode == None:
|
||||
raise OSError("otNodeInit failed!")
|
||||
@@ -1,937 +0,0 @@
|
||||
#!/usr/bin/env 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 os
|
||||
import sys
|
||||
import time
|
||||
import pexpect
|
||||
import pexpect.popen_spawn
|
||||
import re
|
||||
import socket
|
||||
import ipaddress
|
||||
|
||||
import config
|
||||
import simulator
|
||||
|
||||
class otCli:
|
||||
def __init__(self, nodeid, is_mtd, simulator=None):
|
||||
self.nodeid = nodeid
|
||||
self.verbose = int(float(os.getenv('VERBOSE', 0)))
|
||||
self.node_type = os.getenv('NODE_TYPE', 'sim')
|
||||
self.simulator = simulator
|
||||
if self.simulator:
|
||||
self.simulator.add_node(self)
|
||||
|
||||
mode = os.environ.get('USE_MTD') is '1' and is_mtd and 'mtd' or 'ftd'
|
||||
|
||||
if self.node_type == 'soc':
|
||||
self.__init_soc(nodeid)
|
||||
elif self.node_type == 'ncp-sim':
|
||||
# TODO use mode after ncp-mtd is available.
|
||||
self.__init_ncp_sim(nodeid, 'ftd')
|
||||
else:
|
||||
self.__init_sim(nodeid, mode)
|
||||
|
||||
if self.verbose:
|
||||
if sys.version_info[0] == 2:
|
||||
self.pexpect.logfile_read = sys.stdout
|
||||
else:
|
||||
self.pexpect.logfile_read = sys.stdout.buffer
|
||||
|
||||
self._initialized = True
|
||||
|
||||
def __init_sim(self, nodeid, mode):
|
||||
""" Initialize a simulation node. """
|
||||
if 'OT_CLI_PATH' in os.environ.keys():
|
||||
cmd = os.environ['OT_CLI_PATH']
|
||||
elif 'top_builddir' in os.environ.keys():
|
||||
srcdir = os.environ['top_builddir']
|
||||
cmd = '%s/examples/apps/cli/ot-cli-%s' % (srcdir, mode)
|
||||
else:
|
||||
cmd = './ot-cli-%s' % mode
|
||||
|
||||
if 'RADIO_DEVICE' in os.environ:
|
||||
cmd += ' %s' % os.environ['RADIO_DEVICE']
|
||||
os.environ['NODE_ID'] = str(nodeid)
|
||||
|
||||
cmd += ' %d' % nodeid
|
||||
print ("%s" % cmd)
|
||||
|
||||
self.pexpect = pexpect.popen_spawn.PopenSpawn(cmd, timeout=4)
|
||||
|
||||
# Add delay to ensure that the process is ready to receive commands.
|
||||
timeout = 0.4
|
||||
while timeout > 0:
|
||||
self.pexpect.send('\r\n')
|
||||
try:
|
||||
self.pexpect.expect('> ', timeout=0.1)
|
||||
break
|
||||
except pexpect.TIMEOUT:
|
||||
timeout -= 0.1
|
||||
|
||||
def __init_ncp_sim(self, nodeid, mode):
|
||||
""" Initialize an NCP simulation node. """
|
||||
if 'RADIO_DEVICE' in os.environ:
|
||||
args = ' %s' % os.environ['RADIO_DEVICE']
|
||||
os.environ['NODE_ID'] = str(nodeid)
|
||||
else:
|
||||
args = ''
|
||||
|
||||
if 'OT_NCP_PATH' in os.environ.keys():
|
||||
cmd = 'spinel-cli.py -p "%s%s" -n' % (os.environ['OT_NCP_PATH'], args)
|
||||
elif "top_builddir" in os.environ.keys():
|
||||
builddir = os.environ['top_builddir']
|
||||
cmd = 'spinel-cli.py -p "%s/examples/apps/ncp/ot-ncp-%s%s" -n' % (builddir, mode, args)
|
||||
else:
|
||||
cmd = 'spinel-cli.py -p "./ot-ncp-%s%s" -n' % (mode, args)
|
||||
cmd += ' %d' % nodeid
|
||||
print ("%s" % cmd)
|
||||
|
||||
self.pexpect = pexpect.spawn(cmd, timeout=4)
|
||||
|
||||
# Add delay to ensure that the process is ready to receive commands.
|
||||
time.sleep(0.2)
|
||||
self._expect('spinel-cli >')
|
||||
self.debug(int(os.getenv('DEBUG', '0')))
|
||||
|
||||
def _expect(self, pattern, timeout=-1, *args, **kwargs):
|
||||
""" Process simulator events until expected the pattern. """
|
||||
if timeout == -1:
|
||||
timeout = self.pexpect.timeout
|
||||
|
||||
assert timeout > 0
|
||||
|
||||
while timeout > 0:
|
||||
try:
|
||||
return self.pexpect.expect(pattern, 0.1, *args, **kwargs)
|
||||
except pexpect.TIMEOUT:
|
||||
timeout -= 0.1
|
||||
self.simulator.go(0)
|
||||
if timeout <= 0:
|
||||
raise
|
||||
|
||||
def __init_soc(self, nodeid):
|
||||
""" Initialize a System-on-a-chip node connected via UART. """
|
||||
import fdpexpect
|
||||
serialPort = '/dev/ttyUSB%d' % ((nodeid-1)*2)
|
||||
self.pexpect = fdpexpect.fdspawn(os.open(serialPort, os.O_RDWR|os.O_NONBLOCK|os.O_NOCTTY))
|
||||
|
||||
def __del__(self):
|
||||
self.destroy()
|
||||
|
||||
def destroy(self):
|
||||
if not self._initialized:
|
||||
return
|
||||
|
||||
if hasattr(self.pexpect, 'proc') and self.pexpect.proc.poll() is None or \
|
||||
not hasattr(self.pexpect, 'proc') and self.pexpect.isalive():
|
||||
print("%d: exit" % self.nodeid)
|
||||
self.pexpect.send('exit\n')
|
||||
self.pexpect.expect(pexpect.EOF)
|
||||
self.pexpect.wait()
|
||||
self._initialized = False
|
||||
|
||||
def read_cert_messages_in_commissioning_log(self, timeout=-1):
|
||||
"""Get the log of the traffic after DTLS handshake.
|
||||
"""
|
||||
format_str = br"=+?\[\[THCI\].*?type=%s.*?\].*?=+?[\s\S]+?-{40,}"
|
||||
join_fin_req = format_str % br"JOIN_FIN\.req"
|
||||
join_fin_rsp = format_str % br"JOIN_FIN\.rsp"
|
||||
dummy_format_str = br"\[THCI\].*?type=%s.*?"
|
||||
join_ent_ntf = dummy_format_str % br"JOIN_ENT\.ntf"
|
||||
join_ent_rsp = dummy_format_str % br"JOIN_ENT\.rsp"
|
||||
pattern = (b"(" + join_fin_req + b")|(" + join_fin_rsp + b")|("+ join_ent_ntf + b")|(" + join_ent_rsp + b")")
|
||||
|
||||
messages = []
|
||||
# There are at most 4 cert messages both for joiner and commissioner
|
||||
for _ in range(0, 4):
|
||||
try:
|
||||
self._expect(pattern, timeout=timeout)
|
||||
log = self.pexpect.match.group(0)
|
||||
messages.append(self._extract_cert_message(log))
|
||||
except:
|
||||
break
|
||||
return messages
|
||||
|
||||
def _extract_cert_message(self, log):
|
||||
res = re.search(br"direction=\w+", log)
|
||||
assert res
|
||||
direction = res.group(0).split(b'=')[1].strip()
|
||||
|
||||
res = re.search(br"type=\S+", log)
|
||||
assert res
|
||||
type = res.group(0).split(b'=')[1].strip()
|
||||
|
||||
payload = bytearray([])
|
||||
payload_len = 0
|
||||
if type in [b"JOIN_FIN.req", b"JOIN_FIN.rsp"]:
|
||||
res = re.search(br"len=\d+", log)
|
||||
assert res
|
||||
payload_len = int(res.group(0).split(b'=')[1].strip())
|
||||
|
||||
hex_pattern = br"\|(\s([0-9a-fA-F]{2}|\.\.))+?\s+?\|"
|
||||
while True:
|
||||
res = re.search(hex_pattern, log)
|
||||
if not res:
|
||||
break
|
||||
data = [int(hex, 16) for hex in res.group(0)[1:-1].split(b' ') if hex and hex != b'..']
|
||||
payload += bytearray(data)
|
||||
log = log[res.end()-1:]
|
||||
|
||||
assert len(payload) == payload_len
|
||||
return (direction, type, payload)
|
||||
|
||||
def send_command(self, cmd, go=True):
|
||||
print("%d: %s" % (self.nodeid, cmd))
|
||||
self.pexpect.send(cmd + '\n')
|
||||
if go:
|
||||
self.simulator.go(0, nodeid=self.nodeid)
|
||||
sys.stdout.flush()
|
||||
|
||||
def get_commands(self):
|
||||
self.send_command('?')
|
||||
self._expect('Commands:')
|
||||
commands = []
|
||||
while True:
|
||||
i = self._expect(['Done', r'(\S+)'])
|
||||
if i != 0:
|
||||
commands.append(self.pexpect.match.groups()[0])
|
||||
else:
|
||||
break
|
||||
return commands
|
||||
|
||||
def set_mode(self, mode):
|
||||
cmd = 'mode ' + mode
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def debug(self, level):
|
||||
# `debug` command will not trigger interaction with simulator
|
||||
self.send_command('debug '+ str(level), go=False)
|
||||
|
||||
def interface_up(self):
|
||||
self.send_command('ifconfig up')
|
||||
self._expect('Done')
|
||||
|
||||
def interface_down(self):
|
||||
self.send_command('ifconfig down')
|
||||
self._expect('Done')
|
||||
|
||||
def thread_start(self):
|
||||
self.send_command('thread start')
|
||||
self._expect('Done')
|
||||
|
||||
def thread_stop(self):
|
||||
self.send_command('thread stop')
|
||||
self._expect('Done')
|
||||
|
||||
def commissioner_start(self):
|
||||
cmd = 'commissioner start'
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def commissioner_add_joiner(self, addr, psk):
|
||||
cmd = 'commissioner joiner add ' + addr + ' ' + psk
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def joiner_start(self, pskd='', provisioning_url=''):
|
||||
cmd = 'joiner start ' + pskd + ' ' + provisioning_url
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def clear_whitelist(self):
|
||||
cmd = 'macfilter addr clear'
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def enable_whitelist(self):
|
||||
cmd = 'macfilter addr whitelist'
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def disable_whitelist(self):
|
||||
cmd = 'macfilter addr disable'
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def add_whitelist(self, addr, rssi=None):
|
||||
cmd = 'macfilter addr add ' + addr
|
||||
|
||||
if rssi != None:
|
||||
cmd += ' ' + str(rssi)
|
||||
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def remove_whitelist(self, addr):
|
||||
cmd = 'macfilter addr remove ' + addr
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_addr16(self):
|
||||
self.send_command('rloc16')
|
||||
i = self._expect('([0-9a-fA-F]{4})')
|
||||
if i == 0:
|
||||
addr16 = int(self.pexpect.match.groups()[0], 16)
|
||||
self._expect('Done')
|
||||
return addr16
|
||||
|
||||
def get_router_id(self):
|
||||
rloc16 = self.get_addr16()
|
||||
return (rloc16 >> 10)
|
||||
|
||||
def get_addr64(self):
|
||||
self.send_command('extaddr')
|
||||
i = self._expect('([0-9a-fA-F]{16})')
|
||||
if i == 0:
|
||||
addr64 = self.pexpect.match.groups()[0].decode("utf-8")
|
||||
|
||||
self._expect('Done')
|
||||
return addr64
|
||||
|
||||
def get_eui64(self):
|
||||
self.send_command('eui64')
|
||||
i = self._expect('([0-9a-fA-F]{16})')
|
||||
if i == 0:
|
||||
addr64 = self.pexpect.match.groups()[0].decode("utf-8")
|
||||
|
||||
self._expect('Done')
|
||||
return addr64
|
||||
|
||||
def get_joiner_id(self):
|
||||
self.send_command('joiner id')
|
||||
i = self._expect('([0-9a-fA-F]{16})')
|
||||
if i == 0:
|
||||
addr = self.pexpect.match.groups()[0].decode("utf-8")
|
||||
self._expect('Done')
|
||||
return addr
|
||||
|
||||
def get_channel(self):
|
||||
self.send_command('channel')
|
||||
i = self._expect(r'(\d+)\r?\n')
|
||||
if i == 0:
|
||||
channel = int(self.pexpect.match.groups()[0])
|
||||
self._expect('Done')
|
||||
return channel
|
||||
|
||||
def set_channel(self, channel):
|
||||
cmd = 'channel %d' % channel
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_masterkey(self):
|
||||
self.send_command('masterkey')
|
||||
i = self._expect('([0-9a-fA-F]{32})')
|
||||
if i == 0:
|
||||
masterkey = self.pexpect.match.groups()[0].decode("utf-8")
|
||||
self._expect('Done')
|
||||
return masterkey
|
||||
|
||||
def set_masterkey(self, masterkey):
|
||||
cmd = 'masterkey ' + masterkey
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_key_sequence_counter(self):
|
||||
self.send_command('keysequence counter')
|
||||
i = self._expect(r'(\d+)\r?\n')
|
||||
if i == 0:
|
||||
key_sequence_counter = int(self.pexpect.match.groups()[0])
|
||||
self._expect('Done')
|
||||
return key_sequence_counter
|
||||
|
||||
def set_key_sequence_counter(self, key_sequence_counter):
|
||||
cmd = 'keysequence counter %d' % key_sequence_counter
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def set_key_switch_guardtime(self, key_switch_guardtime):
|
||||
cmd = 'keysequence guardtime %d' % key_switch_guardtime
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def set_network_id_timeout(self, network_id_timeout):
|
||||
cmd = 'networkidtimeout %d' % network_id_timeout
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_network_name(self):
|
||||
self.send_command('networkname')
|
||||
while True:
|
||||
i = self._expect(['Done', r'(\S+)'])
|
||||
if i != 0:
|
||||
network_name = self.pexpect.match.groups()[0].decode('utf-8')
|
||||
else:
|
||||
break
|
||||
return network_name
|
||||
|
||||
def set_network_name(self, network_name):
|
||||
cmd = 'networkname ' + network_name
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_panid(self):
|
||||
self.send_command('panid')
|
||||
i = self._expect('([0-9a-fA-F]{4})')
|
||||
if i == 0:
|
||||
panid = int(self.pexpect.match.groups()[0], 16)
|
||||
self._expect('Done')
|
||||
return panid
|
||||
|
||||
def set_panid(self, panid):
|
||||
cmd = 'panid %d' % panid
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_partition_id(self):
|
||||
self.send_command('leaderpartitionid')
|
||||
i = self._expect(r'(\d+)\r?\n')
|
||||
if i == 0:
|
||||
weight = self.pexpect.match.groups()[0]
|
||||
self._expect('Done')
|
||||
return weight
|
||||
|
||||
def set_partition_id(self, partition_id):
|
||||
cmd = 'leaderpartitionid %d' % partition_id
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def set_router_upgrade_threshold(self, threshold):
|
||||
cmd = 'routerupgradethreshold %d' % threshold
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def set_router_downgrade_threshold(self, threshold):
|
||||
cmd = 'routerdowngradethreshold %d' % threshold
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def release_router_id(self, router_id):
|
||||
cmd = 'releaserouterid %d' % router_id
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_state(self):
|
||||
states = ['detached', 'child', 'router', 'leader']
|
||||
self.send_command('state')
|
||||
match = self._expect(states)
|
||||
self._expect('Done')
|
||||
return states[match]
|
||||
|
||||
def set_state(self, state):
|
||||
cmd = 'state ' + state
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_timeout(self):
|
||||
self.send_command('childtimeout')
|
||||
i = self._expect(r'(\d+)\r?\n')
|
||||
if i == 0:
|
||||
timeout = self.pexpect.match.groups()[0]
|
||||
self._expect('Done')
|
||||
return timeout
|
||||
|
||||
def set_timeout(self, timeout):
|
||||
cmd = 'childtimeout %d' % timeout
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def set_max_children(self, number):
|
||||
cmd = 'childmax %d' % number
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_weight(self):
|
||||
self.send_command('leaderweight')
|
||||
i = self._expect(r'(\d+)\r?\n')
|
||||
if i == 0:
|
||||
weight = self.pexpect.match.groups()[0]
|
||||
self._expect('Done')
|
||||
return weight
|
||||
|
||||
def set_weight(self, weight):
|
||||
cmd = 'leaderweight %d' % weight
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def add_ipaddr(self, ipaddr):
|
||||
cmd = 'ipaddr add ' + ipaddr
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def get_addrs(self):
|
||||
addrs = []
|
||||
self.send_command('ipaddr')
|
||||
|
||||
while True:
|
||||
i = self._expect([r'(\S+(:\S*)+)\r?\n', 'Done'])
|
||||
if i == 0:
|
||||
addrs.append(self.pexpect.match.groups()[0].decode("utf-8"))
|
||||
elif i == 1:
|
||||
break
|
||||
|
||||
return addrs
|
||||
|
||||
def get_addr(self, prefix):
|
||||
network = ipaddress.ip_network(u'%s' % str(prefix))
|
||||
addrs = self.get_addrs()
|
||||
|
||||
for addr in addrs:
|
||||
if isinstance(addr, bytearray):
|
||||
addr = bytes(addr)
|
||||
elif isinstance(addr, str) and sys.version_info[0] == 2:
|
||||
addr = addr.decode("utf-8")
|
||||
ipv6_address = ipaddress.ip_address(addr)
|
||||
if ipv6_address in network:
|
||||
return ipv6_address.exploded
|
||||
|
||||
return None
|
||||
|
||||
def get_addr_rloc(self):
|
||||
addrs = self.get_addrs()
|
||||
for addr in addrs:
|
||||
segs = addr.split(':')
|
||||
if segs[4] == '0' and segs[5] == 'ff' and segs[6] == 'fe00' and segs[7] != 'fc00':
|
||||
return addr
|
||||
return None
|
||||
|
||||
def get_addr_leader_aloc(self):
|
||||
addrs = self.get_addrs()
|
||||
for addr in addrs:
|
||||
segs = addr.split(':')
|
||||
if segs[4] == '0' and segs[5] == 'ff' and segs[6] == 'fe00' and segs[7] == 'fc00':
|
||||
return addr
|
||||
return None
|
||||
|
||||
def get_eidcaches(self):
|
||||
eidcaches = []
|
||||
self.send_command('eidcache')
|
||||
|
||||
while True:
|
||||
i = self._expect([r'([a-fA-F0-9\:]+) ([a-fA-F0-9]+)\r?\n', 'Done'])
|
||||
if i == 0:
|
||||
eid = self.pexpect.match.groups()[0].decode("utf-8")
|
||||
rloc = self.pexpect.match.groups()[1].decode("utf-8")
|
||||
eidcaches.append((eid, rloc))
|
||||
elif i == 1:
|
||||
break
|
||||
|
||||
return eidcaches
|
||||
|
||||
def add_service(self, enterpriseNumber, serviceData, serverData):
|
||||
cmd = 'service add ' + enterpriseNumber + ' ' + serviceData+ ' ' + serverData
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def remove_service(self, enterpriseNumber, serviceData):
|
||||
cmd = 'service remove ' + enterpriseNumber + ' ' + serviceData
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def __getLinkLocalAddress(self):
|
||||
for ip6Addr in self.get_addrs():
|
||||
if re.match(config.LINK_LOCAL_REGEX_PATTERN, ip6Addr, re.I):
|
||||
return ip6Addr
|
||||
|
||||
return None
|
||||
|
||||
def __getGlobalAddress(self):
|
||||
global_address = []
|
||||
for ip6Addr in self.get_addrs():
|
||||
if (not re.match(config.LINK_LOCAL_REGEX_PATTERN, ip6Addr, re.I)) and \
|
||||
(not re.match(config.MESH_LOCAL_PREFIX_REGEX_PATTERN, ip6Addr, re.I)) and \
|
||||
(not re.match(config.ROUTING_LOCATOR_REGEX_PATTERN, ip6Addr, re.I)):
|
||||
global_address.append(ip6Addr)
|
||||
|
||||
return global_address
|
||||
|
||||
def __getRloc(self):
|
||||
for ip6Addr in self.get_addrs():
|
||||
if re.match(config.MESH_LOCAL_PREFIX_REGEX_PATTERN, ip6Addr, re.I) and \
|
||||
re.match(config.ROUTING_LOCATOR_REGEX_PATTERN, ip6Addr, re.I) and \
|
||||
not(re.match(config.ALOC_FLAG_REGEX_PATTERN, ip6Addr, re.I)):
|
||||
return ip6Addr
|
||||
return None
|
||||
|
||||
def __getAloc(self):
|
||||
aloc = []
|
||||
for ip6Addr in self.get_addrs():
|
||||
if re.match(config.MESH_LOCAL_PREFIX_REGEX_PATTERN, ip6Addr, re.I) and \
|
||||
re.match(config.ROUTING_LOCATOR_REGEX_PATTERN, ip6Addr, re.I) and \
|
||||
re.match(config.ALOC_FLAG_REGEX_PATTERN, ip6Addr, re.I):
|
||||
aloc.append(ip6Addr)
|
||||
|
||||
return aloc
|
||||
|
||||
def __getMleid(self):
|
||||
for ip6Addr in self.get_addrs():
|
||||
if re.match(config.MESH_LOCAL_PREFIX_REGEX_PATTERN, ip6Addr, re.I) and \
|
||||
not(re.match(config.ROUTING_LOCATOR_REGEX_PATTERN, ip6Addr, re.I)):
|
||||
return ip6Addr
|
||||
|
||||
return None
|
||||
|
||||
def get_ip6_address(self, address_type):
|
||||
"""Get specific type of IPv6 address configured on thread device.
|
||||
|
||||
Args:
|
||||
address_type: the config.ADDRESS_TYPE type of IPv6 address.
|
||||
|
||||
Returns:
|
||||
IPv6 address string.
|
||||
"""
|
||||
if address_type == config.ADDRESS_TYPE.LINK_LOCAL:
|
||||
return self.__getLinkLocalAddress()
|
||||
elif address_type == config.ADDRESS_TYPE.GLOBAL:
|
||||
return self.__getGlobalAddress()
|
||||
elif address_type == config.ADDRESS_TYPE.RLOC:
|
||||
return self.__getRloc()
|
||||
elif address_type == config.ADDRESS_TYPE.ALOC:
|
||||
return self.__getAloc()
|
||||
elif address_type == config.ADDRESS_TYPE.ML_EID:
|
||||
return self.__getMleid()
|
||||
else:
|
||||
return None
|
||||
|
||||
return None
|
||||
|
||||
def get_context_reuse_delay(self):
|
||||
self.send_command('contextreusedelay')
|
||||
i = self._expect(r'(\d+)\r?\n')
|
||||
if i == 0:
|
||||
timeout = self.pexpect.match.groups()[0]
|
||||
self._expect('Done')
|
||||
return timeout
|
||||
|
||||
def set_context_reuse_delay(self, delay):
|
||||
cmd = 'contextreusedelay %d' % delay
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def add_prefix(self, prefix, flags, prf = 'med'):
|
||||
cmd = 'prefix add ' + prefix + ' ' + flags + ' ' + prf
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def remove_prefix(self, prefix):
|
||||
cmd = 'prefix remove ' + prefix
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def add_route(self, prefix, prf = 'med'):
|
||||
cmd = 'route add ' + prefix + ' ' + prf
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def remove_route(self, prefix):
|
||||
cmd = 'route remove ' + prefix
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def register_netdata(self):
|
||||
self.send_command('netdataregister')
|
||||
self._expect('Done')
|
||||
|
||||
def energy_scan(self, mask, count, period, scan_duration, ipaddr):
|
||||
cmd = 'commissioner energy ' + str(mask) + ' ' + str(count) + ' ' + str(period) + ' ' + str(scan_duration) + ' ' + ipaddr
|
||||
self.send_command(cmd)
|
||||
|
||||
if isinstance(self.simulator, simulator.VirtualTime):
|
||||
self.simulator.go(8)
|
||||
timeout = 1
|
||||
else:
|
||||
timeout = 8
|
||||
|
||||
self._expect('Energy:', timeout=timeout)
|
||||
|
||||
def panid_query(self, panid, mask, ipaddr):
|
||||
cmd = 'commissioner panid ' + str(panid) + ' ' + str(mask) + ' ' + ipaddr
|
||||
self.send_command(cmd)
|
||||
|
||||
if isinstance(self.simulator, simulator.VirtualTime):
|
||||
self.simulator.go(8)
|
||||
timeout = 1
|
||||
else:
|
||||
timeout = 8
|
||||
|
||||
self._expect('Conflict:', timeout=timeout)
|
||||
|
||||
def scan(self):
|
||||
self.send_command('scan')
|
||||
|
||||
results = []
|
||||
while True:
|
||||
i = self._expect([r'\|\s(\S+)\s+\|\s(\S+)\s+\|\s([0-9a-fA-F]{4})\s\|\s([0-9a-fA-F]{16})\s\|\s(\d+)\r?\n',
|
||||
'Done'])
|
||||
if i == 0:
|
||||
results.append(self.pexpect.match.groups())
|
||||
else:
|
||||
break
|
||||
|
||||
return results
|
||||
|
||||
def ping(self, ipaddr, num_responses=1, size=None, timeout=5):
|
||||
cmd = 'ping ' + ipaddr
|
||||
if size != None:
|
||||
cmd += ' ' + str(size)
|
||||
|
||||
self.send_command(cmd)
|
||||
|
||||
if isinstance(self.simulator, simulator.VirtualTime):
|
||||
self.simulator.go(timeout)
|
||||
|
||||
result = True
|
||||
try:
|
||||
responders = {}
|
||||
while len(responders) < num_responses:
|
||||
i = self._expect([r'from (\S+):'])
|
||||
if i == 0:
|
||||
responders[self.pexpect.match.groups()[0]] = 1
|
||||
self._expect('\n')
|
||||
except (pexpect.TIMEOUT, socket.timeout):
|
||||
result = False
|
||||
if isinstance(self.simulator, simulator.VirtualTime):
|
||||
self.simulator.sync_devices()
|
||||
|
||||
return result
|
||||
|
||||
def reset(self):
|
||||
self.send_command('reset')
|
||||
time.sleep(0.1)
|
||||
|
||||
def set_router_selection_jitter(self, jitter):
|
||||
cmd = 'routerselectionjitter %d' % jitter
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def set_active_dataset(self, timestamp, panid=None, channel=None, channel_mask=None, master_key=None):
|
||||
self.send_command('dataset clear')
|
||||
self._expect('Done')
|
||||
|
||||
cmd = 'dataset activetimestamp %d' % timestamp
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
if panid != None:
|
||||
cmd = 'dataset panid %d' % panid
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
if channel != None:
|
||||
cmd = 'dataset channel %d' % channel
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
if channel_mask != None:
|
||||
cmd = 'dataset channelmask %d' % channel_mask
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
if master_key != None:
|
||||
cmd = 'dataset masterkey ' + master_key
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
self.send_command('dataset commit active')
|
||||
self._expect('Done')
|
||||
|
||||
def set_pending_dataset(self, pendingtimestamp, activetimestamp, panid=None, channel=None):
|
||||
self.send_command('dataset clear')
|
||||
self._expect('Done')
|
||||
|
||||
cmd = 'dataset pendingtimestamp %d' % pendingtimestamp
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
cmd = 'dataset activetimestamp %d' % activetimestamp
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
if panid != None:
|
||||
cmd = 'dataset panid %d' % panid
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
if channel != None:
|
||||
cmd = 'dataset channel %d' % channel
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
self.send_command('dataset commit pending')
|
||||
self._expect('Done')
|
||||
|
||||
def announce_begin(self, mask, count, period, ipaddr):
|
||||
cmd = 'commissioner announce ' + str(mask) + ' ' + str(count) + ' ' + str(period) + ' ' + ipaddr
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def send_mgmt_active_set(self, active_timestamp=None, channel=None, channel_mask=None, extended_panid=None,
|
||||
panid=None, master_key=None, mesh_local=None, network_name=None, binary=None):
|
||||
cmd = 'dataset mgmtsetcommand active '
|
||||
|
||||
if active_timestamp != None:
|
||||
cmd += 'activetimestamp %d ' % active_timestamp
|
||||
|
||||
if channel != None:
|
||||
cmd += 'channel %d ' % channel
|
||||
|
||||
if channel_mask != None:
|
||||
cmd += 'channelmask %d ' % channel_mask
|
||||
|
||||
if extended_panid != None:
|
||||
cmd += 'extpanid ' + extended_panid + ' '
|
||||
|
||||
if panid != None:
|
||||
cmd += 'panid %d ' % panid
|
||||
|
||||
if master_key != None:
|
||||
cmd += 'masterkey ' + master_key + ' '
|
||||
|
||||
if mesh_local != None:
|
||||
cmd += 'localprefix ' + mesh_local + ' '
|
||||
|
||||
if network_name != None:
|
||||
cmd += 'networkname ' + network_name + ' '
|
||||
|
||||
if binary != None:
|
||||
cmd += 'binary ' + binary + ' '
|
||||
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def send_mgmt_pending_set(self, pending_timestamp=None, active_timestamp=None, delay_timer=None, channel=None,
|
||||
panid=None, master_key=None, mesh_local=None, network_name=None):
|
||||
cmd = 'dataset mgmtsetcommand pending '
|
||||
if pending_timestamp != None:
|
||||
cmd += 'pendingtimestamp %d ' % pending_timestamp
|
||||
|
||||
if active_timestamp != None:
|
||||
cmd += 'activetimestamp %d ' % active_timestamp
|
||||
|
||||
if delay_timer != None:
|
||||
cmd += 'delaytimer %d ' % delay_timer
|
||||
|
||||
if channel != None:
|
||||
cmd += 'channel %d ' % channel
|
||||
|
||||
if panid != None:
|
||||
cmd += 'panid %d ' % panid
|
||||
|
||||
if master_key != None:
|
||||
cmd += 'masterkey ' + master_key + ' '
|
||||
|
||||
if mesh_local != None:
|
||||
cmd += 'localprefix ' + mesh_local + ' '
|
||||
|
||||
if network_name != None:
|
||||
cmd += 'networkname ' + network_name + ' '
|
||||
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def coaps_start_psk(self, psk, pskIdentity):
|
||||
cmd = 'coaps psk ' + psk + ' ' + pskIdentity
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
cmd = 'coaps start'
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def coaps_start_x509(self):
|
||||
cmd = 'coaps x509'
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
cmd = 'coaps start'
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def coaps_set_resource_path(self, path):
|
||||
cmd = 'coaps resource ' + path
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def coaps_stop(self):
|
||||
cmd = 'coaps stop'
|
||||
self.send_command(cmd)
|
||||
|
||||
if isinstance(self.simulator, simulator.VirtualTime):
|
||||
self.simulator.go(5)
|
||||
timeout = 1
|
||||
else:
|
||||
timeout = 5
|
||||
|
||||
self._expect('Done', timeout=timeout)
|
||||
|
||||
def coaps_connect(self, ipaddr):
|
||||
cmd = 'coaps connect ' + ipaddr
|
||||
self.send_command(cmd)
|
||||
|
||||
if isinstance(self.simulator, simulator.VirtualTime):
|
||||
self.simulator.go(5)
|
||||
timeout = 1
|
||||
else:
|
||||
timeout = 5
|
||||
|
||||
self._expect('coaps connected', timeout=timeout)
|
||||
|
||||
def coaps_disconnect(self):
|
||||
cmd = 'coaps disconnect'
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
self.simulator.go(5)
|
||||
|
||||
def coaps_get(self):
|
||||
cmd = 'coaps get test'
|
||||
self.send_command(cmd)
|
||||
|
||||
if isinstance(self.simulator, simulator.VirtualTime):
|
||||
self.simulator.go(5)
|
||||
timeout = 1
|
||||
else:
|
||||
timeout = 5
|
||||
|
||||
self._expect('coaps response', timeout=timeout)
|
||||
|
||||
def commissioner_mgmtset(self, tlvs_binary):
|
||||
cmd = 'commissioner mgmtset binary ' + tlvs_binary
|
||||
self.send_command(cmd)
|
||||
self._expect('Done')
|
||||
|
||||
def bytes_to_hex_str(self, src):
|
||||
return ''.join(format(x, '02x') for x in src)
|
||||
|
||||
def commissioner_mgmtset_with_tlvs(self, tlvs):
|
||||
payload = bytearray()
|
||||
for tlv in tlvs:
|
||||
payload += tlv.to_hex()
|
||||
self.commissioner_mgmtset(self.bytes_to_hex_str(payload))
|
||||
|
||||
@@ -153,66 +153,7 @@ class MacFrame(ctypes.Structure):
|
||||
("length", ctypes.c_ubyte),
|
||||
("nodeid", ctypes.c_uint)]
|
||||
|
||||
class SnifferVirtualTransport(SnifferTransport):
|
||||
""" Virtual interface based implementation of sniffer transport. """
|
||||
|
||||
def __init__(self, nodeid):
|
||||
self.Handle = None
|
||||
|
||||
# Load the DLL
|
||||
self.Api = ctypes.WinDLL("otnodeapi.dll")
|
||||
if self.Api == None:
|
||||
raise OSError("Failed to load otnodeapi.dll!")
|
||||
|
||||
# Define the functions
|
||||
self.Api.otListenerInit.argtypes = [ctypes.c_uint]
|
||||
self.Api.otListenerInit.restype = ctypes.c_void_p
|
||||
|
||||
self.Api.otListenerFinalize.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self.Api.otListenerRead.argtypes = [ctypes.c_void_p, ctypes.POINTER(MacFrame)]
|
||||
|
||||
def __del__(self):
|
||||
if not self.is_opened:
|
||||
return
|
||||
|
||||
self.close()
|
||||
|
||||
def open(self):
|
||||
if self.is_opened:
|
||||
raise RuntimeError("Transport is already opened.")
|
||||
|
||||
# Initialize a listener
|
||||
self.Handle = self.Api.otListenerInit(0)
|
||||
|
||||
if not self.is_opened:
|
||||
raise RuntimeError("Transport opening failed.")
|
||||
|
||||
def close(self):
|
||||
if not self.is_opened:
|
||||
raise RuntimeError("Transport is closed.")
|
||||
|
||||
self.Api.otListenerFinalize(self.Handle);
|
||||
self.Handle = None
|
||||
|
||||
@property
|
||||
def is_opened(self):
|
||||
return bool(self.Handle is not None)
|
||||
|
||||
def recv(self, bufsize):
|
||||
frame = MacFrame()
|
||||
pFrame = ctypes.pointer(frame);
|
||||
|
||||
self.Api.otListenerRead(self.Handle, pFrame)
|
||||
|
||||
return bytearray(frame.buffer)[:frame.length], frame.nodeid
|
||||
|
||||
|
||||
class SnifferTransportFactory(object):
|
||||
|
||||
def create_transport(self, nodeid):
|
||||
if sys.platform != "win32":
|
||||
return SnifferSocketTransport(nodeid)
|
||||
|
||||
else:
|
||||
return SnifferVirtualTransport(nodeid)
|
||||
return SnifferSocketTransport(nodeid)
|
||||
|
||||
@@ -29,16 +29,16 @@
|
||||
|
||||
import unittest
|
||||
|
||||
import node_cli
|
||||
import node
|
||||
import config
|
||||
|
||||
class TestDiag(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.simulator = config.create_default_simulator()
|
||||
self.node_cli = node_cli.otCli(1, False, simulator=self.simulator)
|
||||
self.node = node.Node(1, False, simulator=self.simulator)
|
||||
|
||||
def tearDown(self):
|
||||
self.node_cli.destroy()
|
||||
self.node.destroy()
|
||||
self.simulator.stop()
|
||||
|
||||
def test(self):
|
||||
@@ -80,9 +80,9 @@ class TestDiag(unittest.TestCase):
|
||||
]
|
||||
|
||||
for case in cases:
|
||||
self.node_cli.send_command(case[0])
|
||||
self.node.send_command(case[0])
|
||||
self.simulator.go(1)
|
||||
self.node_cli._expect(case[1])
|
||||
self.node._expect(case[1])
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user