mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2024-12-21 05:53:09 +00:00
316 lines
8.2 KiB
C++
316 lines
8.2 KiB
C++
/*
|
|
* ZeroTier One - Network Virtualization Everywhere
|
|
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
* --
|
|
*
|
|
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
|
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
|
*
|
|
* If you would like to embed ZeroTier into a commercial application or
|
|
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
|
* LLC. Start here: http://www.zerotier.com/
|
|
*/
|
|
|
|
#ifndef ZT_DICTIONARY_HPP
|
|
#define ZT_DICTIONARY_HPP
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <string>
|
|
#include <map>
|
|
#include <stdexcept>
|
|
|
|
#include "Constants.hpp"
|
|
#include "Utils.hpp"
|
|
|
|
// Three fields are added/updated by sign()
|
|
#define ZT_DICTIONARY_SIGNATURE "~!ed25519"
|
|
#define ZT_DICTIONARY_SIGNATURE_IDENTITY "~!sigid"
|
|
#define ZT_DICTIONARY_SIGNATURE_TIMESTAMP "~!sigts"
|
|
|
|
namespace ZeroTier {
|
|
|
|
class Identity;
|
|
|
|
/**
|
|
* Simple key/value dictionary with string serialization
|
|
*
|
|
* The serialization format is a flat key=value with backslash escape.
|
|
* It does not support comments or other syntactic complexities. It is
|
|
* human-readable if the keys and values in the dictionary are also
|
|
* human-readable. Otherwise it might contain unprintable characters.
|
|
*
|
|
* Keys beginning with "~!" are reserved for signature data fields.
|
|
*
|
|
* Note: the signature code depends on std::map<> being sorted, but no
|
|
* other code does. So if the underlying data structure is ever swapped
|
|
* out for an unsorted one, the signature code will have to be updated
|
|
* to sort before composing the string to sign.
|
|
*/
|
|
class Dictionary : public std::map<std::string,std::string>
|
|
{
|
|
public:
|
|
Dictionary() {}
|
|
|
|
/**
|
|
* @param s String-serialized dictionary
|
|
* @param maxlen Maximum length of buffer
|
|
*/
|
|
Dictionary(const char *s,unsigned int maxlen) { fromString(s,maxlen); }
|
|
|
|
/**
|
|
* @param s String-serialized dictionary
|
|
*/
|
|
Dictionary(const std::string &s) { fromString(s.c_str(),(unsigned int)s.length()); }
|
|
|
|
/**
|
|
* Get a key, throwing an exception if it is not present
|
|
*
|
|
* @param key Key to look up
|
|
* @return Reference to value
|
|
* @throws std::invalid_argument Key not found
|
|
*/
|
|
inline const std::string &get(const std::string &key) const
|
|
throw(std::invalid_argument)
|
|
{
|
|
const_iterator e(find(key));
|
|
if (e == end())
|
|
throw std::invalid_argument(std::string("missing required field: ")+key);
|
|
return e->second;
|
|
}
|
|
|
|
/**
|
|
* Get a key, returning a default if not present
|
|
*
|
|
* @param key Key to look up
|
|
* @param dfl Default if not present
|
|
* @return Value or default
|
|
*/
|
|
inline const std::string &get(const std::string &key,const std::string &dfl) const
|
|
{
|
|
const_iterator e(find(key));
|
|
if (e == end())
|
|
return dfl;
|
|
return e->second;
|
|
}
|
|
|
|
/**
|
|
* @param key Key to get
|
|
* @param dfl Default boolean result if key not found or empty (default: false)
|
|
* @return Boolean value of key
|
|
*/
|
|
inline bool getBoolean(const std::string &key,bool dfl = false) const
|
|
{
|
|
const_iterator e(find(key));
|
|
if (e == end())
|
|
return dfl;
|
|
if (e->second.length() < 1)
|
|
return dfl;
|
|
switch(e->second[0]) {
|
|
case '1':
|
|
case 't':
|
|
case 'T':
|
|
case 'y':
|
|
case 'Y':
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* @param key Key to get
|
|
* @param dfl Default value if not present (default: 0)
|
|
* @return Value converted to unsigned 64-bit int or 0 if not found
|
|
*/
|
|
inline uint64_t getUInt(const std::string &key,uint64_t dfl = 0) const
|
|
{
|
|
const_iterator e(find(key));
|
|
if (e == end())
|
|
return dfl;
|
|
return Utils::strToU64(e->second.c_str());
|
|
}
|
|
|
|
/**
|
|
* @param key Key to get
|
|
* @param dfl Default value if not present (default: 0)
|
|
* @return Value converted to unsigned 64-bit int or 0 if not found
|
|
*/
|
|
inline uint64_t getHexUInt(const std::string &key,uint64_t dfl = 0) const
|
|
{
|
|
const_iterator e(find(key));
|
|
if (e == end())
|
|
return dfl;
|
|
return Utils::hexStrToU64(e->second.c_str());
|
|
}
|
|
|
|
/**
|
|
* @param key Key to get
|
|
* @param dfl Default value if not present (default: 0)
|
|
* @return Value converted to signed 64-bit int or 0 if not found
|
|
*/
|
|
inline int64_t getInt(const std::string &key,int64_t dfl = 0) const
|
|
{
|
|
const_iterator e(find(key));
|
|
if (e == end())
|
|
return dfl;
|
|
return Utils::strTo64(e->second.c_str());
|
|
}
|
|
|
|
/**
|
|
* @param key Key to set
|
|
* @param value String value
|
|
*/
|
|
inline void set(const std::string &key,const char *value)
|
|
{
|
|
(*this)[key] = value;
|
|
}
|
|
|
|
/**
|
|
* @param key Key to set
|
|
* @param value String value
|
|
*/
|
|
inline void set(const std::string &key,const std::string &value)
|
|
{
|
|
(*this)[key] = value;
|
|
}
|
|
|
|
/**
|
|
* @param key Key to set
|
|
* @param value Boolean value
|
|
*/
|
|
inline void set(const std::string &key,bool value)
|
|
{
|
|
(*this)[key] = ((value) ? "1" : "0");
|
|
}
|
|
|
|
/**
|
|
* @param key Key to set
|
|
* @param value Integer value
|
|
*/
|
|
inline void set(const std::string &key,uint64_t value)
|
|
{
|
|
char tmp[24];
|
|
Utils::snprintf(tmp,sizeof(tmp),"%llu",(unsigned long long)value);
|
|
(*this)[key] = tmp;
|
|
}
|
|
|
|
/**
|
|
* @param key Key to set
|
|
* @param value Integer value
|
|
*/
|
|
inline void set(const std::string &key,int64_t value)
|
|
{
|
|
char tmp[24];
|
|
Utils::snprintf(tmp,sizeof(tmp),"%lld",(long long)value);
|
|
(*this)[key] = tmp;
|
|
}
|
|
|
|
/**
|
|
* @param key Key to set
|
|
* @param value Integer value
|
|
*/
|
|
inline void setHex(const std::string &key,uint64_t value)
|
|
{
|
|
char tmp[24];
|
|
Utils::snprintf(tmp,sizeof(tmp),"%llx",(unsigned long long)value);
|
|
(*this)[key] = tmp;
|
|
}
|
|
|
|
/**
|
|
* @param key Key to check
|
|
* @return True if dictionary contains key
|
|
*/
|
|
inline bool contains(const std::string &key) const { return (find(key) != end()); }
|
|
|
|
/**
|
|
* @return String-serialized dictionary
|
|
*/
|
|
inline std::string toString() const
|
|
{
|
|
std::string s;
|
|
for(const_iterator kv(begin());kv!=end();++kv) {
|
|
_appendEsc(kv->first.data(),(unsigned int)kv->first.length(),s);
|
|
s.push_back('=');
|
|
_appendEsc(kv->second.data(),(unsigned int)kv->second.length(),s);
|
|
s.append(ZT_EOL_S);
|
|
}
|
|
return s;
|
|
}
|
|
|
|
/**
|
|
* Clear and initialize from a string
|
|
*
|
|
* @param s String-serialized dictionary
|
|
* @param maxlen Maximum length of string buffer
|
|
*/
|
|
void fromString(const char *s,unsigned int maxlen);
|
|
inline void fromString(const std::string &s) { fromString(s.c_str(),(unsigned int)s.length()); }
|
|
void updateFromString(const char *s,unsigned int maxlen);
|
|
inline void update(const char *s,unsigned int maxlen) { updateFromString(s, maxlen); }
|
|
inline void update(const std::string &s) { updateFromString(s.c_str(),(unsigned int)s.length()); }
|
|
|
|
/**
|
|
* @return True if this dictionary is cryptographically signed
|
|
*/
|
|
inline bool hasSignature() const { return (find(ZT_DICTIONARY_SIGNATURE) != end()); }
|
|
|
|
/**
|
|
* @return Signing identity in string-serialized format or empty string if none
|
|
*/
|
|
inline std::string signingIdentity() const { return get(ZT_DICTIONARY_SIGNATURE_IDENTITY,std::string()); }
|
|
|
|
/**
|
|
* @return Signature timestamp in milliseconds since epoch or 0 if none
|
|
*/
|
|
uint64_t signatureTimestamp() const;
|
|
|
|
/**
|
|
* Remove any signature from this dictionary
|
|
*/
|
|
inline void removeSignature()
|
|
{
|
|
erase(ZT_DICTIONARY_SIGNATURE);
|
|
erase(ZT_DICTIONARY_SIGNATURE_IDENTITY);
|
|
erase(ZT_DICTIONARY_SIGNATURE_TIMESTAMP);
|
|
}
|
|
|
|
/**
|
|
* Add or update signature fields with a signature of all other keys and values
|
|
*
|
|
* @param with Identity to sign with (must have secret key)
|
|
* @param now Current time
|
|
* @return True on success
|
|
*/
|
|
bool sign(const Identity &id,uint64_t now);
|
|
|
|
/**
|
|
* Verify signature against an identity
|
|
*
|
|
* @param id Identity to verify against
|
|
* @return True if signature verification OK
|
|
*/
|
|
bool verify(const Identity &id) const;
|
|
|
|
private:
|
|
void _mkSigBuf(std::string &buf) const;
|
|
static void _appendEsc(const char *data,unsigned int len,std::string &to);
|
|
};
|
|
|
|
} // namespace ZeroTier
|
|
|
|
#endif
|