[srp-client] support saving selected server info (by auto-start) (#6672)

This commit adds support for a new feature in SRP client which allows
it to save the selected server info (by the auto-start feature) in
non-volatile settings. On SRP client restart (e.g., due to a device
reset) the client will select the same server when searching to
discover and pick one from the Thread Network Data service entries.
Config `OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE` can
be used to enable/disable this feature.

The server info is saved only after the host info is successfully
registered with the server and if it is selected by auto-start from a
network data SRP *unicast* service entry.

This commit also adds `test_srp_client_save_server_info.py` test-case
which verifies the behavior of the new feature.
This commit is contained in:
Abtin Keshavarzian
2021-05-25 18:34:07 -07:00
committed by GitHub
parent e503c949b9
commit 1796f50264
8 changed files with 377 additions and 2 deletions
+1 -1
View File
@@ -53,7 +53,7 @@ extern "C" {
* @note This number versions both OpenThread platform and user APIs.
*
*/
#define OPENTHREAD_API_VERSION (118)
#define OPENTHREAD_API_VERSION (119)
/**
* @addtogroup api-instance
+1
View File
@@ -71,6 +71,7 @@ enum
OT_SETTINGS_KEY_OMR_PREFIX = 0x0009, ///< Off-mesh routable (OMR) prefix.
OT_SETTINGS_KEY_ON_LINK_PREFIX = 0x000a, ///< On-link prefix for infrastructure link.
OT_SETTINGS_KEY_SRP_ECDSA_KEY = 0x000b, ///< SRP client ECDSA public/private key pair.
OT_SETTINGS_KEY_SRP_CLIENT_INFO = 0x000c, ///< The SRP client info (selected SRP server address).
};
/**
+57
View File
@@ -91,6 +91,14 @@ void SettingsBase::LogPrefix(const char *aAction, const char *aPrefixName, const
}
#endif
#if OPENTHREAD_CONFIG_SRP_CLIENT_ENABLE && OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
void SettingsBase::LogSrpClientInfo(const char *aAction, const SrpClientInfo &aSrpClientInfo) const
{
otLogInfoCore("Non-volatile: %s SrpClientInfo {Server:[%s]:%u}", aAction,
aSrpClientInfo.GetServerAddress().ToString().AsCString(), aSrpClientInfo.GetServerPort());
}
#endif
#endif // #if (OPENTHREAD_CONFIG_LOG_LEVEL >= OT_LOG_LEVEL_INFO)
#if (OPENTHREAD_CONFIG_LOG_LEVEL >= OT_LOG_LEVEL_WARN)
@@ -571,6 +579,55 @@ exit:
return error;
}
#if OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
Error Settings::SaveSrpClientInfo(const SrpClientInfo &aSrpClientInfo)
{
Error error = kErrorNone;
SrpClientInfo prevInfo;
uint16_t length = sizeof(SrpClientInfo);
if ((Read(kKeySrpClientInfo, &prevInfo, length) == kErrorNone) && (length == sizeof(SrpClientInfo)) &&
(prevInfo == aSrpClientInfo))
{
LogSrpClientInfo("Re-saved", aSrpClientInfo);
ExitNow();
}
SuccessOrExit(error = Save(kKeySrpClientInfo, &aSrpClientInfo, sizeof(SrpClientInfo)));
LogSrpClientInfo("Saved", aSrpClientInfo);
exit:
LogFailure(error, "saving SrpClientInfo", /* aIsDelete */ false);
return error;
}
Error Settings::ReadSrpClientInfo(SrpClientInfo &aSrpClientInfo) const
{
Error error;
uint16_t length = sizeof(SrpClientInfo);
aSrpClientInfo.Init();
SuccessOrExit(error = Read(kKeySrpClientInfo, &aSrpClientInfo, length));
LogSrpClientInfo("Read", aSrpClientInfo);
exit:
return error;
}
Error Settings::DeleteSrpClientInfo(void)
{
Error error;
SuccessOrExit(error = Delete(kKeySrpClientInfo));
otLogInfoCore("Non-volatile: Deleted SrpClientInfo");
exit:
LogFailure(error, "deleting SrpClientInfo", true);
return error;
}
#endif // OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
#endif // OPENTHREAD_CONFIG_SRP_CLIENT_ENABLE
Error Settings::Read(Key aKey, void *aBuffer, uint16_t &aSize) const
+89
View File
@@ -589,6 +589,57 @@ public:
uint8_t mDadCounter; ///< Dad Counter used to resolve address conflict in Thread 1.2 DUA feature.
} OT_TOOL_PACKED_END;
/**
* This structure represents the SRP client info (selected server address).
*
*/
OT_TOOL_PACKED_BEGIN
class SrpClientInfo : public Equatable<SrpClientInfo>, private Clearable<SrpClientInfo>
{
public:
/**
* This method initializes the `SrpClientInfo` object.
*
*/
void Init(void) { Clear(); }
/**
* This method returns the server IPv6 address.
*
* @returns The server IPv6 address.
*
*/
const Ip6::Address &GetServerAddress(void) const { return mServerAddress; }
/**
* This method sets the server IPv6 address.
*
* @param[in] aAddress The server IPv6 address.
*
*/
void SetServerAddress(const Ip6::Address &aAddress) { mServerAddress = aAddress; }
/**
* This method returns the server port number.
*
* @returns The server port number.
*
*/
uint16_t GetServerPort(void) const { return Encoding::LittleEndian::HostSwap16(mServerPort); }
/**
* This method sets the server port number.
*
* @param[in] aPort The server port number.
*
*/
void SetServerPort(uint16_t aPort) { mServerPort = Encoding::LittleEndian::HostSwap16(aPort); }
private:
Ip6::Address mServerAddress;
uint16_t mServerPort; // (in little-endian encoding)
} OT_TOOL_PACKED_END;
/**
* This enumeration defines the keys of settings.
*
@@ -606,6 +657,7 @@ public:
kKeyOmrPrefix = OT_SETTINGS_KEY_OMR_PREFIX,
kKeyOnLinkPrefix = OT_SETTINGS_KEY_ON_LINK_PREFIX,
kKeySrpEcdsaKey = OT_SETTINGS_KEY_SRP_ECDSA_KEY,
kKeySrpClientInfo = OT_SETTINGS_KEY_SRP_CLIENT_INFO,
};
protected:
@@ -624,6 +676,7 @@ protected:
#if OPENTHREAD_CONFIG_BORDER_ROUTING_ENABLE
void LogPrefix(const char *aAction, const char *aPrefixName, const Ip6::Prefix &aOmrPrefix) const;
#endif
void LogSrpClientInfo(const char *aAction, const SrpClientInfo &aSrpClientInfo) const;
#else // (OPENTHREAD_CONFIG_LOG_LEVEL >= OT_LOG_LEVEL_INFO) && (OPENTHREAD_CONFIG_LOG_UTIL != 0)
void LogNetworkInfo(const char *, const NetworkInfo &) const {}
void LogParentInfo(const char *, const ParentInfo &) const {}
@@ -634,6 +687,7 @@ protected:
#if OPENTHREAD_CONFIG_BORDER_ROUTING_ENABLE
void LogPrefix(const char *, const char *, const Ip6::Prefix &) const {}
#endif
void LogSrpClientInfo(const char *, const SrpClientInfo &) const {}
#endif // (OPENTHREAD_CONFIG_LOG_LEVEL >= OT_LOG_LEVEL_INFO) && (OPENTHREAD_CONFIG_LOG_UTIL != 0)
#if (OPENTHREAD_CONFIG_LOG_LEVEL >= OT_LOG_LEVEL_WARN) && (OPENTHREAD_CONFIG_LOG_UTIL != 0)
@@ -1092,6 +1146,41 @@ public:
*
*/
Error DeleteSrpKey(void);
#if OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
/**
* This method saves SRP client info.
*
* @param[in] aSrpClientInfo The `SrpClientInfo` to save.
*
* @retval kErrorNone Successfully saved the information in settings.
* @retval kErrorNotImplemented The platform does not implement settings functionality.
*
*/
Error SaveSrpClientInfo(const SrpClientInfo &aSrpClientInfo);
/**
* This method reads SRP client info.
*
* @param[out] aSrpClientInfo A reference to a `SrpClientInfo` to output the read content.
*
* @retval kErrorNone Successfully read the information.
* @retval kErrorNotFound No corresponding value in the setting store.
* @retval kErrorNotImplemented The platform does not implement settings functionality.
*
*/
Error ReadSrpClientInfo(SrpClientInfo &aSrpClientInfo) const;
/**
* This method deletes SRP client info from settings.
*
* @retval kErrorNone Successfully deleted the value.
* @retval kErrorNotImplemented The platform does not implement settings functionality.
*
*/
Error DeleteSrpClientInfo(void);
#endif // OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
#endif // OPENTHREAD_CONFIG_SRP_CLIENT_ENABLE
private:
+17 -1
View File
@@ -51,7 +51,7 @@
* Define to 1 to enable SRP Client auto-start feature and its APIs.
*
* When enabled, the SRP client can be configured to automatically start when it detects the presence of an SRP server
* (by monitoring the Thread Network Data for SRP Server Service entries).
* (by monitoring the Thread Network Data for SRP Server Service entries).
*
*/
#ifndef OPENTHREAD_CONFIG_SRP_CLIENT_AUTO_START_API_ENABLE
@@ -80,6 +80,22 @@
#define OPENTHREAD_CONFIG_SRP_CLIENT_DOMAIN_NAME_API_ENABLE 0
#endif
/**
* @def OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
*
* Define to 1 to enable SRP client feature to save the selected server in non-volatile settings.
*
* When enabled, the SRP client will save the selected server info by auto-start feature in the non-volatile settings
* and on a client restart (e.g., due to a device reset) it will select the same server when searching to discover and
* pick one from the Thread Network Data service entries. The server info is saved only after the host info is
* successfully registered with the server and if it is selected by auto-start from a network data SRP *unicast*
* service entry.
*
*/
#ifndef OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
#define OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE 1
#endif
/**
* @def OPENTHREAD_CONFIG_SRP_CLIENT_DEFAULT_LEASE
*
+48
View File
@@ -542,12 +542,36 @@ exit:
void Client::ChangeHostAndServiceStates(const ItemState *aNewStates)
{
#if OPENTHREAD_CONFIG_SRP_CLIENT_AUTO_START_API_ENABLE && OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
ItemState oldHostState = mHostInfo.GetState();
#endif
mHostInfo.SetState(aNewStates[mHostInfo.GetState()]);
for (Service *service = mServices.GetHead(); service != nullptr; service = service->GetNext())
{
service->SetState(aNewStates[service->GetState()]);
}
#if OPENTHREAD_CONFIG_SRP_CLIENT_AUTO_START_API_ENABLE && OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
if (mAutoStartModeEnabled && mAutoStartDidSelectServer && (oldHostState != kRegistered) &&
(mHostInfo.GetState() == kRegistered))
{
if (mAutoStartIsUsingAnycastAddress)
{
IgnoreError(Get<Settings>().DeleteSrpClientInfo());
}
else
{
Settings::SrpClientInfo info;
info.SetServerAddress(GetServerAddress().GetAddress());
info.SetServerPort(GetServerAddress().GetPort());
IgnoreError(Get<Settings>().SaveSrpClientInfo(info));
}
}
#endif
}
void Client::InvokeCallback(Error aError) const
@@ -1515,6 +1539,10 @@ void Client::ProcessAutoStart(void)
Ip6::SockAddr serverSockAddr;
bool serverIsAnycast = false;
NetworkData::Service::DnsSrpAnycast::Info anycastInfo;
#if OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
Settings::SrpClientInfo savedInfo;
bool hasSavedServerInfo = false;
#endif
VerifyOrExit(mAutoStartModeEnabled);
@@ -1535,6 +1563,13 @@ void Client::ProcessAutoStart(void)
// Now `IsRunning()` implies `mAutoStartDidSelectServer`.
#if OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
if (!IsRunning())
{
hasSavedServerInfo = (Get<Settings>().ReadSrpClientInfo(savedInfo) == kErrorNone);
}
#endif
if (Get<NetworkData::Service::Manager>().FindPreferredDnsSrpAnycastInfo(anycastInfo) == kErrorNone)
{
if (IsRunning() && mAutoStartIsUsingAnycastAddress && (mServerSequenceNumber == anycastInfo.mSequenceNumber) &&
@@ -1564,6 +1599,19 @@ void Client::ProcessAutoStart(void)
ExitNow();
}
#if OPENTHREAD_CONFIG_SRP_CLIENT_SAVE_SELECTED_SERVER_ENABLE
if (hasSavedServerInfo && (unicastInfo.mSockAddr.GetAddress() == savedInfo.GetServerAddress()) &&
(unicastInfo.mSockAddr.GetPort() == savedInfo.GetServerPort()))
{
// Stop the search if we see a match for the previously
// saved server info in the network data entries.
serverSockAddr = unicastInfo.mSockAddr;
serverIsAnycast = false;
break;
}
#endif
numServers++;
// Choose a server randomly (with uniform distribution) from
+2
View File
@@ -176,6 +176,7 @@ EXTRA_DIST = \
test_router_reattach.py \
test_service.py \
test_srp_auto_start_mode.py \
test_srp_client_save_server_info.py \
test_srp_lease.py \
test_srp_name_conflicts.py \
test_srp_register_single_service.py \
@@ -233,6 +234,7 @@ check_SCRIPTS = \
test_router_reattach.py \
test_service.py \
test_srp_auto_start_mode.py \
test_srp_client_save_server_info.py \
test_srp_lease.py \
test_srp_name_conflicts.py \
test_srp_register_single_service.py \
@@ -0,0 +1,162 @@
#!/usr/bin/env python3
#
# Copyright (c) 2021, 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 ipaddress
import unittest
import command
import thread_cert
# Test description:
# This test verifies SRP client behavior to save the selected server address info (unicast) by
# auto-start feature.
#
# Topology:
#
# 4 node (SRP client as leader, 3 SRP servers).
#
CLIENT = 1
SERVER1 = 2
SERVER2 = 3
SERVER3 = 4
WAIT_TIME = 5
MAX_ITER = 5
class SrpAutoStartMode(thread_cert.TestCase):
USE_MESSAGE_FACTORY = False
SUPPORT_NCP = False
TOPOLOGY = {
CLIENT: {
'name': 'SRP_CLIENT',
'masterkey': '00112233445566778899aabbccddeeff',
'mode': 'rdn',
},
SERVER1: {
'name': 'SRP_SERVER1',
'masterkey': '00112233445566778899aabbccddeeff',
'mode': 'rn',
},
SERVER2: {
'name': 'SRP_SERVER2',
'masterkey': '00112233445566778899aabbccddeeff',
'mode': 'rn',
},
SERVER3: {
'name': 'SRP_SERVER3',
'masterkey': '00112233445566778899aabbccddeeff',
'mode': 'rn',
},
}
def test(self):
client = self.nodes[CLIENT]
server1 = self.nodes[SERVER1]
server2 = self.nodes[SERVER2]
server3 = self.nodes[SERVER3]
# Start the server & client devices.
client.start()
self.simulator.go(WAIT_TIME)
self.assertEqual(client.get_state(), 'leader')
for node in [server1, server2, server3]:
node.start()
self.simulator.go(WAIT_TIME)
self.assertEqual(node.get_state(), 'child')
# Enable auto start mode on client and check that server1 is used.
server1.srp_server_set_enabled(True)
client.srp_client_set_host_name('host')
client.srp_client_set_host_address('2001::1')
client.srp_client_add_service('my-service', '_ipps._tcp', 12345)
self.assertEqual(client.srp_client_get_state(), 'Disabled')
client.srp_client_enable_auto_start_mode()
self.assertEqual(client.srp_client_get_auto_start_mode(), 'Enabled')
self.simulator.go(WAIT_TIME)
self.assertEqual(client.srp_client_get_state(), 'Enabled')
self.assertTrue(server1.has_ipaddr(client.srp_client_get_server_address()))
self.assertEqual(client.srp_client_get_host_state(), 'Registered')
# Enable server2 and server3 and check that server1 is still used.
server2.srp_server_set_enabled(True)
server3.srp_server_set_enabled(True)
self.simulator.go(WAIT_TIME)
self.assertTrue(server1.has_ipaddr(client.srp_client_get_server_address()))
# Stop and restart the client (multiple times) and verify that
# server1 is always picked.
for iter in range(0, MAX_ITER):
client.srp_client_stop()
client.srp_client_enable_auto_start_mode()
self.simulator.go(WAIT_TIME)
self.assertTrue(server1.has_ipaddr(client.srp_client_get_server_address()))
self.assertEqual(client.srp_client_get_host_state(), 'Registered')
# Stop the client, then stop server1 and restart client and
# verify that server1 is no longer used.
client.srp_client_stop()
server1.srp_server_set_enabled(False)
self.simulator.go(WAIT_TIME)
client.srp_client_enable_auto_start_mode()
self.simulator.go(WAIT_TIME)
server_address = client.srp_client_get_server_address()
self.assertFalse(server1.has_ipaddr(server_address))
self.assertTrue(server2.has_ipaddr(server_address) or server3.has_ipaddr(server_address))
self.assertEqual(client.srp_client_get_host_state(), 'Registered')
# Start back server1, then restart client and verify that now we remain
# with the new saved server info.
server1.srp_server_set_enabled(True)
self.simulator.go(WAIT_TIME)
for iter in range(0, MAX_ITER):
client.srp_client_stop()
self.simulator.go(WAIT_TIME)
client.srp_client_enable_auto_start_mode()
self.simulator.go(WAIT_TIME)
self.assertEqual(client.srp_client_get_server_address(), server_address)
self.assertEqual(client.srp_client_get_host_state(), 'Registered')
if __name__ == '__main__':
unittest.main()