mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-04-26 17:03:43 +02:00
84 lines
2.4 KiB
C++
84 lines
2.4 KiB
C++
/*
|
|
* Copyright (c)2013-2021 ZeroTier, Inc.
|
|
*
|
|
* Use of this software is governed by the Business Source License included
|
|
* in the LICENSE.TXT file in the project's root directory.
|
|
*
|
|
* Change Date: 2026-01-01
|
|
*
|
|
* On the date above, in accordance with the Business Source License, use
|
|
* of this software will be governed by version 2.0 of the Apache License.
|
|
*/
|
|
/****/
|
|
|
|
#ifndef ZT_STORE_HPP
|
|
#define ZT_STORE_HPP
|
|
|
|
#include "Constants.hpp"
|
|
#include "Containers.hpp"
|
|
#include "RuntimeEnvironment.hpp"
|
|
|
|
namespace ZeroTier {
|
|
|
|
/**
|
|
* Wrapper around API callbacks for data store
|
|
*/
|
|
class Store
|
|
{
|
|
public:
|
|
ZT_INLINE Store(const RuntimeEnvironment *const renv): RR(renv)
|
|
{}
|
|
|
|
/**
|
|
* Get a state object
|
|
*
|
|
* @param tPtr Thread pointer to pass through
|
|
* @param type Object type
|
|
* @param id Object ID
|
|
* @param idSize Size of object ID in qwords
|
|
* @return Data or empty vector if not found
|
|
*/
|
|
ZT_INLINE Vector< uint8_t > get(void *tPtr, ZT_StateObjectType type, const uint64_t *id, unsigned int idSize) const
|
|
{
|
|
Vector< uint8_t > dv;
|
|
void *data = nullptr;
|
|
void (*freeFunc)(void *) = nullptr;
|
|
const int r = RR->cb.stateGetFunction(reinterpret_cast<ZT_Node *>(RR->node), RR->uPtr, tPtr, type, id, idSize, &data, &freeFunc);
|
|
if (r > 0)
|
|
dv.assign(reinterpret_cast<const uint8_t *>(data), reinterpret_cast<const uint8_t *>(data) + r);
|
|
if ((data) && (freeFunc))
|
|
freeFunc(data);
|
|
return dv;
|
|
}
|
|
|
|
/**
|
|
* Store a state object
|
|
*
|
|
* @param tPtr Thread pointer to pass through
|
|
* @param type Object type
|
|
* @param id Object ID
|
|
* @param idSize Size of object ID in qwords
|
|
* @param data Data to store
|
|
* @param len Length of data
|
|
*/
|
|
ZT_INLINE void put(void *const tPtr, ZT_StateObjectType type, const uint64_t *const id, const unsigned int idSize, const void *const data, const unsigned int len) noexcept
|
|
{ RR->cb.statePutFunction(reinterpret_cast<ZT_Node *>(this), RR->uPtr, tPtr, type, id, idSize, data, (int)len); }
|
|
|
|
/**
|
|
* Erase a state object from the object store
|
|
*
|
|
* @param tPtr Thread pointer to pass through
|
|
* @param type Object type
|
|
* @param id Object ID
|
|
* @param idSize Size of object ID in qwords
|
|
*/
|
|
ZT_INLINE void erase(void *const tPtr, ZT_StateObjectType type, const uint64_t *const id, const unsigned int idSize) noexcept
|
|
{ RR->cb.statePutFunction(reinterpret_cast<ZT_Node *>(this), RR->uPtr, tPtr, type, id, idSize, nullptr, -1); }
|
|
|
|
private:
|
|
const RuntimeEnvironment *RR;
|
|
};
|
|
|
|
} // namespace ZeroTier
|
|
|
|
#endif
|