mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-04-26 17:03:43 +02:00
146 lines
3.4 KiB
C++
146 lines
3.4 KiB
C++
/*
|
|
* Copyright (c)2013-2020 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: 2024-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_MAP_HPP
|
|
#define ZT_MAP_HPP
|
|
|
|
/* This defines a Map, SortedMap, Vector, etc. based on STL templates. */
|
|
|
|
#include "Constants.hpp"
|
|
#include "Utils.hpp"
|
|
|
|
#ifdef __CPP11__
|
|
#include <unordered_map>
|
|
#endif
|
|
#include <map>
|
|
#include <vector>
|
|
#include <list>
|
|
#include <set>
|
|
#include <string>
|
|
|
|
namespace ZeroTier {
|
|
|
|
#ifdef __CPP11__
|
|
struct _MapHasher
|
|
{
|
|
template<typename O>
|
|
std::size_t operator()(const O &obj) const noexcept { return (std::size_t)obj.hashCode(); }
|
|
std::size_t operator()(const uint64_t i) const noexcept { return (std::size_t)Utils::hash64(i ^ Utils::s_mapNonce); }
|
|
std::size_t operator()(const int64_t i) const noexcept { return (std::size_t)Utils::hash64((uint64_t)i ^ Utils::s_mapNonce); }
|
|
std::size_t operator()(const uint32_t i) const noexcept { return (std::size_t)Utils::hash32(i ^ (uint32_t)Utils::s_mapNonce); }
|
|
std::size_t operator()(const int32_t i) const noexcept { return (std::size_t)Utils::hash32((uint32_t)i ^ (uint32_t)Utils::s_mapNonce); }
|
|
};
|
|
|
|
template<typename K,typename V>
|
|
class Map : public std::unordered_map< K,V,_MapHasher,std::equal_to<K>,Utils::Mallocator< std::pair<const K,V> > >
|
|
{
|
|
public:
|
|
ZT_INLINE V *get(const K &key) noexcept
|
|
{
|
|
typename Map::iterator i(this->find(key));
|
|
if (i == this->end())
|
|
return nullptr;
|
|
return &(i->second);
|
|
}
|
|
|
|
ZT_INLINE const V *get(const K &key) const noexcept
|
|
{
|
|
typename Map::const_iterator i(this->find(key));
|
|
if (i == this->end())
|
|
return nullptr;
|
|
return &(i->second);
|
|
}
|
|
|
|
ZT_INLINE void set(const K &key,const V &value)
|
|
{
|
|
this->emplace(key,value);
|
|
}
|
|
};
|
|
#else
|
|
template<typename K,typename V>
|
|
class Map : public std::map< K,V,std::less<K>,Utils::Mallocator< std::pair<const K,V> > >
|
|
{
|
|
public:
|
|
ZT_INLINE V *get(const K &key) noexcept
|
|
{
|
|
typename Map::iterator i(this->find(key));
|
|
if (i == this->end())
|
|
return nullptr;
|
|
return &(i->second);
|
|
}
|
|
|
|
ZT_INLINE const V *get(const K &key) const noexcept
|
|
{
|
|
typename Map::const_iterator i(this->find(key));
|
|
if (i == this->end())
|
|
return nullptr;
|
|
return &(i->second);
|
|
}
|
|
|
|
ZT_INLINE void set(const K &key,const V &value)
|
|
{
|
|
(*this)[key] = value;
|
|
}
|
|
};
|
|
#endif
|
|
|
|
template<typename K,typename V>
|
|
class SortedMap : public std::map< K,V,std::less<K>,Utils::Mallocator< std::pair<const K,V> > >
|
|
{
|
|
public:
|
|
ZT_INLINE V *get(const K &key) noexcept
|
|
{
|
|
typename SortedMap::iterator i(this->find(key));
|
|
if (i == this->end())
|
|
return nullptr;
|
|
return &(i->second);
|
|
}
|
|
|
|
ZT_INLINE const V *get(const K &key) const noexcept
|
|
{
|
|
typename SortedMap::const_iterator i(this->find(key));
|
|
if (i == this->end())
|
|
return nullptr;
|
|
return &(i->second);
|
|
}
|
|
|
|
ZT_INLINE void set(const K &key,const V &value)
|
|
{
|
|
(*this)[key] = value;
|
|
}
|
|
};
|
|
|
|
template<typename V>
|
|
class Vector : public std::vector< V,Utils::Mallocator<V> >
|
|
{
|
|
};
|
|
|
|
template<typename V>
|
|
class List : public std::list< V,Utils::Mallocator<V> >
|
|
{
|
|
};
|
|
|
|
template<typename V>
|
|
class Set : public std::set< V,std::less<V>,Utils::Mallocator<V> >
|
|
{
|
|
};
|
|
|
|
class String : public std::basic_string< char,std::char_traits<char>,Utils::Mallocator<char> >
|
|
{
|
|
public:
|
|
explicit ZT_INLINE String(const char *const s) { assign(s); }
|
|
};
|
|
|
|
} // ZeroTier
|
|
|
|
#endif
|