diff options
Diffstat (limited to 'node/EllipticCurveKeyPair.hpp')
-rw-r--r-- | node/EllipticCurveKeyPair.hpp | 128 |
1 files changed, 128 insertions, 0 deletions
diff --git a/node/EllipticCurveKeyPair.hpp b/node/EllipticCurveKeyPair.hpp new file mode 100644 index 00000000..2649f4c4 --- /dev/null +++ b/node/EllipticCurveKeyPair.hpp @@ -0,0 +1,128 @@ +/* + * ZeroTier One - Global Peer to Peer Ethernet + * Copyright (C) 2012-2013 ZeroTier Networks LLC + * + * 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_ELLIPTICCURVEKEYPAIR_HPP +#define _ZT_ELLIPTICCURVEKEYPAIR_HPP + +#include <string> +#include "EllipticCurveKey.hpp" + +namespace ZeroTier { + +/** + * An elliptic curve key pair supporting generation and key agreement + */ +class EllipticCurveKeyPair +{ +public: + EllipticCurveKeyPair(); + EllipticCurveKeyPair(const EllipticCurveKeyPair &pair); + EllipticCurveKeyPair(const EllipticCurveKey &pubk,const EllipticCurveKey &privk); + ~EllipticCurveKeyPair(); + + const EllipticCurveKeyPair &operator=(const EllipticCurveKeyPair &pair); + + /** + * Fill this structure with a newly generated public/private key pair + * + * @return True if key generation is successful + */ + bool generate(); + + /** + * Perform elliptic curve key agreement + * + * @param theirPublicKey Remote side's public key + * @param agreedUponKey Buffer to fill with agreed-upon symmetric key + * @param agreedUponKeyLength Number of bytes to generate + * @return True if key agreement is successful + */ + bool agree(const EllipticCurveKey &theirPublicKey,unsigned char *agreedUponKey,unsigned int agreedUponKeyLength) const; + + /** + * Sign a SHA256 hash + * + * @param sha256 Pointer to 256-bit / 32-byte SHA hash to sign + * @return ECDSA signature (r and s in binary format, each prefixed by an 8-bit size) + */ + std::string sign(const void *sha256) const; + + /** + * Sign something with this pair's private key, computing its hash first + * + * @param data Data to hash and sign + * @param len Length of data + * @return Signature bytes + */ + std::string sign(const void *data,unsigned int len) const; + + /** + * Verify a signature + * + * @param sha256 Pointer to 256-bit / 32-byte SHA hash to verify + * @param pk Public key to verify against + * @param sigbytes Signature bytes + * @param siglen Length of signature + */ + static bool verify(const void *sha256,const EllipticCurveKey &pk,const void *sigbytes,unsigned int siglen); + + /** + * Verify a signature + * + * @param data Data to verify + * @param len Length of data + * @param pk Public key to verify against + * @param sigbytes Signature bytes + * @param siglen Length of signature + */ + static bool verify(const void *data,unsigned int len,const EllipticCurveKey &pk,const void *sigbytes,unsigned int siglen); + + inline bool operator==(const EllipticCurveKeyPair &kp) const + throw() + { + return ((_pub == kp._pub)&&(_priv == kp._priv)); + } + inline bool operator!=(const EllipticCurveKeyPair &kp) const + throw() + { + return ((_pub != kp._pub)||(_priv != kp._priv)); + } + + inline const EllipticCurveKey &pub() const throw() { return _pub; } + inline const EllipticCurveKey &priv() const throw() { return _priv; } + +private: + bool initInternalKey(); + + EllipticCurveKey _pub; + EllipticCurveKey _priv; + void *_internal_key; +}; + +} // namespace ZeroTier + +#endif |