| /* |
| * 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. |
| */ |
| |
| /** |
| * @file |
| * This file implements common methods for manipulating MeshCoP Datasets. |
| * |
| */ |
| |
| #include <stdio.h> |
| |
| #include <common/code_utils.hpp> |
| #include <common/settings.hpp> |
| #include <platform/settings.h> |
| #include <thread/meshcop_tlvs.hpp> |
| #include <thread/meshcop_dataset.hpp> |
| |
| namespace Thread { |
| namespace MeshCoP { |
| |
| Dataset::Dataset(otInstance *aInstance, const Tlv::Type aType) : |
| mType(aType), |
| mLength(0), |
| mInstance(aInstance) |
| { |
| } |
| |
| void Dataset::Clear(bool isLocal) |
| { |
| mLength = 0; |
| |
| if (isLocal) |
| { |
| otPlatSettingsDelete(mInstance, mType == Tlv::kActiveTimestamp ? kKeyActiveDataset : kKeyPendingDataset, -1); |
| } |
| } |
| |
| Tlv *Dataset::Get(Tlv::Type aType) |
| { |
| Tlv *cur = reinterpret_cast<Tlv *>(mTlvs); |
| Tlv *end = reinterpret_cast<Tlv *>(mTlvs + mLength); |
| Tlv *rval = NULL; |
| |
| while (cur < end) |
| { |
| if (cur->GetType() == aType) |
| { |
| ExitNow(rval = cur); |
| } |
| |
| cur = cur->GetNext(); |
| } |
| |
| exit: |
| return rval; |
| } |
| |
| const Tlv *Dataset::Get(Tlv::Type aType) const |
| { |
| const Tlv *cur = reinterpret_cast<const Tlv *>(mTlvs); |
| const Tlv *end = reinterpret_cast<const Tlv *>(mTlvs + mLength); |
| const Tlv *rval = NULL; |
| |
| while (cur < end) |
| { |
| if (cur->GetType() == aType) |
| { |
| ExitNow(rval = cur); |
| } |
| |
| cur = cur->GetNext(); |
| } |
| |
| exit: |
| return rval; |
| } |
| |
| void Dataset::Get(otOperationalDataset &aDataset) const |
| { |
| const Tlv *cur = reinterpret_cast<const Tlv *>(mTlvs); |
| const Tlv *end = reinterpret_cast<const Tlv *>(mTlvs + mLength); |
| |
| memset(&aDataset, 0, sizeof(aDataset)); |
| |
| while (cur < end) |
| { |
| switch (cur->GetType()) |
| { |
| |
| case Tlv::kActiveTimestamp: |
| { |
| const ActiveTimestampTlv *tlv = static_cast<const ActiveTimestampTlv *>(cur); |
| aDataset.mActiveTimestamp = tlv->GetSeconds(); |
| aDataset.mIsActiveTimestampSet = true; |
| break; |
| } |
| |
| case Tlv::kChannel: |
| { |
| const ChannelTlv *tlv = static_cast<const ChannelTlv *>(cur); |
| aDataset.mChannel = tlv->GetChannel(); |
| aDataset.mIsChannelSet = true; |
| break; |
| } |
| |
| case Tlv::kChannelMask: |
| { |
| uint8_t length = cur->GetLength(); |
| const uint8_t *entry = reinterpret_cast<const uint8_t *>(cur) + sizeof(Tlv); |
| const uint8_t *entryEnd = entry + length; |
| |
| while (entry < entryEnd) |
| { |
| if (reinterpret_cast<const ChannelMaskEntry *>(entry)->GetChannelPage() == 0) |
| { |
| uint8_t i = sizeof(ChannelMaskEntry); |
| aDataset.mChannelMaskPage0 = static_cast<uint32_t>(entry[i] | (entry[i + 1] << 8) | |
| (entry[i + 2] << 16) | (entry[i + 3] << 24)); |
| aDataset.mIsChannelMaskPage0Set = true; |
| break; |
| } |
| |
| entry += (reinterpret_cast<const ChannelMaskEntry *>(entry)->GetMaskLength() + |
| sizeof(ChannelMaskEntry)); |
| } |
| |
| break; |
| } |
| |
| case Tlv::kDelayTimer: |
| { |
| const DelayTimerTlv *tlv = static_cast<const DelayTimerTlv *>(cur); |
| aDataset.mDelay = tlv->GetDelayTimer(); |
| aDataset.mIsDelaySet = true; |
| break; |
| } |
| |
| case Tlv::kExtendedPanId: |
| { |
| const ExtendedPanIdTlv *tlv = static_cast<const ExtendedPanIdTlv *>(cur); |
| memcpy(aDataset.mExtendedPanId.m8, tlv->GetExtendedPanId(), sizeof(aDataset.mExtendedPanId)); |
| aDataset.mIsExtendedPanIdSet = true; |
| break; |
| } |
| |
| case Tlv::kMeshLocalPrefix: |
| { |
| const MeshLocalPrefixTlv *tlv = static_cast<const MeshLocalPrefixTlv *>(cur); |
| memcpy(aDataset.mMeshLocalPrefix.m8, tlv->GetMeshLocalPrefix(), sizeof(aDataset.mMeshLocalPrefix)); |
| aDataset.mIsMeshLocalPrefixSet = true; |
| break; |
| } |
| |
| case Tlv::kNetworkMasterKey: |
| { |
| const NetworkMasterKeyTlv *tlv = static_cast<const NetworkMasterKeyTlv *>(cur); |
| memcpy(aDataset.mMasterKey.m8, tlv->GetNetworkMasterKey(), sizeof(aDataset.mMasterKey)); |
| aDataset.mIsMasterKeySet = true; |
| break; |
| } |
| |
| case Tlv::kNetworkName: |
| { |
| const NetworkNameTlv *tlv = static_cast<const NetworkNameTlv *>(cur); |
| memcpy(aDataset.mNetworkName.m8, tlv->GetNetworkName(), tlv->GetLength()); |
| aDataset.mIsNetworkNameSet = true; |
| break; |
| } |
| |
| case Tlv::kPanId: |
| { |
| const PanIdTlv *panid = static_cast<const PanIdTlv *>(cur); |
| aDataset.mPanId = panid->GetPanId(); |
| aDataset.mIsPanIdSet = true; |
| break; |
| } |
| |
| case Tlv::kPendingTimestamp: |
| { |
| const PendingTimestampTlv *tlv = static_cast<const PendingTimestampTlv *>(cur); |
| aDataset.mPendingTimestamp = tlv->GetSeconds(); |
| aDataset.mIsPendingTimestampSet = true; |
| break; |
| } |
| |
| case Tlv::kPSKc: |
| { |
| const PSKcTlv *tlv = static_cast<const PSKcTlv *>(cur); |
| memcpy(aDataset.mPSKc.m8, tlv->GetPSKc(), tlv->GetLength()); |
| aDataset.mIsPSKcSet = true; |
| break; |
| } |
| |
| case Tlv::kSecurityPolicy: |
| { |
| const SecurityPolicyTlv *tlv = static_cast<const SecurityPolicyTlv *>(cur); |
| aDataset.mSecurityPolicy.mRotationTime = tlv->GetRotationTime(); |
| aDataset.mSecurityPolicy.mFlags = tlv->GetFlags(); |
| aDataset.mIsSecurityPolicySet = true; |
| break; |
| } |
| |
| default: |
| { |
| break; |
| } |
| } |
| |
| cur = cur->GetNext(); |
| } |
| } |
| |
| ThreadError Dataset::Set(const Dataset &aDataset) |
| { |
| memcpy(mTlvs, aDataset.mTlvs, aDataset.mLength); |
| mLength = aDataset.mLength; |
| |
| if (mType == Tlv::kActiveTimestamp) |
| { |
| Remove(Tlv::kPendingTimestamp); |
| Remove(Tlv::kDelayTimer); |
| } |
| |
| return kThreadError_None; |
| } |
| |
| ThreadError Dataset::Set(const otOperationalDataset &aDataset) |
| { |
| ThreadError error = kThreadError_None; |
| MeshCoP::ActiveTimestampTlv activeTimestampTlv; |
| |
| VerifyOrExit(aDataset.mIsActiveTimestampSet, error = kThreadError_InvalidArgs); |
| |
| activeTimestampTlv.Init(); |
| activeTimestampTlv.SetSeconds(aDataset.mActiveTimestamp); |
| activeTimestampTlv.SetTicks(0); |
| Set(activeTimestampTlv); |
| |
| if (mType == Tlv::kPendingTimestamp) |
| { |
| MeshCoP::PendingTimestampTlv pendingTimestampTlv; |
| |
| VerifyOrExit(aDataset.mIsPendingTimestampSet, error = kThreadError_InvalidArgs); |
| |
| pendingTimestampTlv.Init(); |
| pendingTimestampTlv.SetSeconds(aDataset.mPendingTimestamp); |
| pendingTimestampTlv.SetTicks(0); |
| Set(pendingTimestampTlv); |
| |
| if (aDataset.mIsDelaySet) |
| { |
| MeshCoP::DelayTimerTlv tlv; |
| tlv.Init(); |
| tlv.SetDelayTimer(aDataset.mDelay); |
| Set(tlv); |
| } |
| } |
| |
| if (aDataset.mIsChannelSet) |
| { |
| MeshCoP::ChannelTlv tlv; |
| tlv.Init(); |
| tlv.SetChannelPage(0); |
| tlv.SetChannel(aDataset.mChannel); |
| Set(tlv); |
| } |
| |
| if (aDataset.mIsChannelMaskPage0Set) |
| { |
| OT_TOOL_PACKED_BEGIN |
| struct |
| { |
| MeshCoP::ChannelMaskTlv tlv; |
| MeshCoP::ChannelMaskEntry entry; |
| uint8_t mask[sizeof(aDataset.mChannelMaskPage0)]; |
| } OT_TOOL_PACKED_END channelMask; |
| |
| channelMask.tlv.Init(); |
| channelMask.tlv.SetLength(sizeof(MeshCoP::ChannelMaskEntry) + sizeof(aDataset.mChannelMaskPage0)); |
| |
| channelMask.entry.SetChannelPage(0); |
| channelMask.entry.SetMaskLength(sizeof(aDataset.mChannelMaskPage0)); |
| |
| for (uint8_t index = 0; index < sizeof(aDataset.mChannelMaskPage0); index++) |
| { |
| channelMask.mask[index] = (aDataset.mChannelMaskPage0 >> (8 * index)) & 0xff; |
| } |
| |
| Set(channelMask.tlv); |
| } |
| |
| if (aDataset.mIsExtendedPanIdSet) |
| { |
| MeshCoP::ExtendedPanIdTlv tlv; |
| tlv.Init(); |
| tlv.SetExtendedPanId(aDataset.mExtendedPanId.m8); |
| Set(tlv); |
| } |
| |
| if (aDataset.mIsMeshLocalPrefixSet) |
| { |
| MeshCoP::MeshLocalPrefixTlv tlv; |
| tlv.Init(); |
| tlv.SetMeshLocalPrefix(aDataset.mMeshLocalPrefix.m8); |
| Set(tlv); |
| } |
| |
| if (aDataset.mIsMasterKeySet) |
| { |
| MeshCoP::NetworkMasterKeyTlv tlv; |
| tlv.Init(); |
| tlv.SetNetworkMasterKey(aDataset.mMasterKey.m8); |
| Set(tlv); |
| } |
| |
| if (aDataset.mIsNetworkNameSet) |
| { |
| MeshCoP::NetworkNameTlv tlv; |
| tlv.Init(); |
| tlv.SetNetworkName(aDataset.mNetworkName.m8); |
| Set(tlv); |
| } |
| |
| if (aDataset.mIsPanIdSet) |
| { |
| MeshCoP::PanIdTlv tlv; |
| tlv.Init(); |
| tlv.SetPanId(aDataset.mPanId); |
| Set(tlv); |
| } |
| |
| if (aDataset.mIsPSKcSet) |
| { |
| MeshCoP::PSKcTlv tlv; |
| tlv.Init(); |
| tlv.SetPSKc(aDataset.mPSKc.m8); |
| Set(tlv); |
| } |
| |
| if (aDataset.mIsSecurityPolicySet) |
| { |
| MeshCoP::SecurityPolicyTlv tlv; |
| tlv.Init(); |
| tlv.SetRotationTime(aDataset.mSecurityPolicy.mRotationTime); |
| tlv.SetFlags(aDataset.mSecurityPolicy.mFlags); |
| Set(tlv); |
| } |
| |
| exit: |
| return error; |
| } |
| |
| const Timestamp *Dataset::GetTimestamp(void) const |
| { |
| const Timestamp *timestamp = NULL; |
| |
| if (mType == Tlv::kActiveTimestamp) |
| { |
| const ActiveTimestampTlv *tlv = static_cast<const ActiveTimestampTlv *>(Get(mType)); |
| VerifyOrExit(tlv != NULL, ;); |
| timestamp = static_cast<const Timestamp *>(tlv); |
| } |
| else |
| { |
| const PendingTimestampTlv *tlv = static_cast<const PendingTimestampTlv *>(Get(mType)); |
| VerifyOrExit(tlv != NULL, ;); |
| timestamp = static_cast<const Timestamp *>(tlv); |
| } |
| |
| exit: |
| return timestamp; |
| } |
| |
| void Dataset::SetTimestamp(const Timestamp &aTimestamp) |
| { |
| OT_TOOL_PACKED_BEGIN |
| struct |
| { |
| Tlv tlv; |
| Timestamp timestamp; |
| } OT_TOOL_PACKED_END timestampTlv; |
| |
| timestampTlv.tlv.SetType(mType); |
| timestampTlv.tlv.SetLength(sizeof(Timestamp)); |
| timestampTlv.timestamp = aTimestamp; |
| |
| Set(timestampTlv.tlv); |
| } |
| |
| int Dataset::Compare(const Dataset &aCompare) const |
| { |
| const Timestamp *thisTimestamp = GetTimestamp(); |
| const Timestamp *compareTimestamp = aCompare.GetTimestamp(); |
| int rval; |
| |
| if (compareTimestamp == NULL && thisTimestamp == NULL) |
| { |
| rval = 0; |
| } |
| else if (compareTimestamp == NULL && thisTimestamp != NULL) |
| { |
| rval = -1; |
| } |
| else if (compareTimestamp != NULL && thisTimestamp == NULL) |
| { |
| rval = 1; |
| } |
| else |
| { |
| rval = thisTimestamp->Compare(*compareTimestamp); |
| } |
| |
| return rval; |
| } |
| |
| ThreadError Dataset::Restore(void) |
| { |
| return otPlatSettingsGet(mInstance, mType == Tlv::kActiveTimestamp ? kKeyActiveDataset : kKeyPendingDataset, 0, mTlvs, |
| &mLength); |
| } |
| |
| ThreadError Dataset::Store(void) |
| { |
| return otPlatSettingsSet(mInstance, mType == Tlv::kActiveTimestamp ? kKeyActiveDataset : kKeyPendingDataset, mTlvs, |
| mLength); |
| } |
| |
| ThreadError Dataset::Set(const Tlv &aTlv) |
| { |
| ThreadError error = kThreadError_None; |
| uint16_t bytesAvailable = sizeof(mTlvs) - mLength; |
| Tlv *old = Get(aTlv.GetType()); |
| |
| if (old != NULL) |
| { |
| bytesAvailable += sizeof(Tlv) + old->GetLength(); |
| } |
| |
| VerifyOrExit(sizeof(Tlv) + aTlv.GetLength() <= bytesAvailable, error = kThreadError_NoBufs); |
| |
| // remove old TLV |
| if (old != NULL) |
| { |
| Remove(reinterpret_cast<uint8_t *>(old), sizeof(Tlv) + old->GetLength()); |
| } |
| |
| // add new TLV |
| memcpy(mTlvs + mLength, &aTlv, sizeof(Tlv) + aTlv.GetLength()); |
| mLength += sizeof(Tlv) + aTlv.GetLength(); |
| |
| exit: |
| return error; |
| } |
| |
| ThreadError Dataset::Set(const Message &aMessage, uint16_t aOffset, uint8_t aLength) |
| { |
| aMessage.Read(aOffset, aLength, mTlvs); |
| mLength = aLength; |
| return kThreadError_None; |
| } |
| |
| void Dataset::Remove(Tlv::Type aType) |
| { |
| Tlv *tlv; |
| |
| VerifyOrExit((tlv = Get(aType)) != NULL, ;); |
| Remove(reinterpret_cast<uint8_t *>(tlv), sizeof(Tlv) + tlv->GetLength()); |
| |
| exit: |
| return; |
| } |
| |
| void Dataset::Remove(uint8_t *aStart, uint8_t aLength) |
| { |
| memmove(aStart, aStart + aLength, mLength - (static_cast<uint8_t>(aStart - mTlvs) + aLength)); |
| mLength -= aLength; |
| } |
| |
| } // namespace MeshCoP |
| } // namespace Thread |