src/key.cpp

   1 // Copyright (c) 2009-2014 The Bitcoin Core developers
   2 // Copyright (c) 2017 The Zcash developers
   3 // Distributed under the MIT software license, see the accompanying
   4 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   5
   6 #include "key.h"
   7
   8 #include "arith_uint256.h"
   9 #include "crypto/common.h"
  10 #include "crypto/hmac_sha512.h"
  11 #include "pubkey.h"
  12 #include "random.h"
  13
  14 #include <secp256k1.h>
  15 #include <secp256k1_recovery.h>
  16
  17 static secp256k1_context* secp256k1_context_sign = NULL;
  18
  19 /** These functions are taken from the libsecp256k1 distribution and are very ugly. */
  20
  21 /**
  22  * This parses a format loosely based on a DER encoding of the ECPrivateKey type from
  23  * section C.4 of SEC 1 <http://www.secg.org/sec1-v2.pdf>, with the following caveats:
  24  *
  25  * * The octet-length of the SEQUENCE must be encoded as 1 or 2 octets. It is not
  26  *   required to be encoded as one octet if it is less than 256, as DER would require.
  27  * * The octet-length of the SEQUENCE must not be greater than the remaining
  28  *   length of the key encoding, but need not match it (i.e. the encoding may contain
  29  *   junk after the encoded SEQUENCE).
  30  * * The privateKey OCTET STRING is zero-filled on the left to 32 octets.
  31  * * Anything after the encoding of the privateKey OCTET STRING is ignored, whether
  32  *   or not it is validly encoded DER.
  33  *
  34  * out32 must point to an output buffer of length at least 32 bytes.
  35  */
  36 static int ec_privkey_import_der(const secp256k1_context* ctx, unsigned char *out32, const unsigned char *privkey, size_t privkeylen) {
  37     const unsigned char *end = privkey + privkeylen;
  38     memset(out32, 0, 32);
  39     /* sequence header */
  40     if (end - privkey < 1 || *privkey != 0x30u) {
  41         return 0;
  42     }
  43     privkey++;
  44     /* sequence length constructor */
  45     if (end - privkey < 1 || !(*privkey & 0x80u)) {
  46         return 0;
  47     }
  48     size_t lenb = *privkey & ~0x80u; privkey++;
  49     if (lenb < 1 || lenb > 2) {
  50         return 0;
  51     }
  52     if (end - privkey < lenb) {
  53         return 0;
  54     }
  55     /* sequence length */
  56     size_t len = privkey[lenb-1] | (lenb > 1 ? privkey[lenb-2] << 8 : 0u);
  57     privkey += lenb;
  58     if (end - privkey < len) {
  59         return 0;
  60     }
  61     /* sequence element 0: version number (=1) */
  62     if (end - privkey < 3 || privkey[0] != 0x02u || privkey[1] != 0x01u || privkey[2] != 0x01u) {
  63         return 0;
  64     }
  65     privkey += 3;
  66     /* sequence element 1: octet string, up to 32 bytes */
  67     if (end - privkey < 2 || privkey[0] != 0x04u) {
  68         return 0;
  69     }
  70     size_t oslen = privkey[1];
  71     privkey += 2;
  72     if (oslen > 32 || end - privkey < oslen) {
  73         return 0;
  74     }
  75     memcpy(out32 + (32 - oslen), privkey, oslen);
  76     if (!secp256k1_ec_seckey_verify(ctx, out32)) {
  77         memset(out32, 0, 32);
  78         return 0;
  79     }
  80     return 1;
  81 }
  82
  83 /**
  84  * This serializes to a DER encoding of the ECPrivateKey type from section C.4 of SEC 1
  85  * <http://www.secg.org/sec1-v2.pdf>. The optional parameters and publicKey fields are
  86  * included.
  87  *
  88  * privkey must point to an output buffer of length at least PRIVATE_KEY_SIZE bytes.
  89  * privkeylen must initially be set to the size of the privkey buffer. Upon return it
  90  * will be set to the number of bytes used in the buffer.
  91  * key32 must point to a 32-byte raw private key.
  92  */
  93 static int ec_privkey_export_der(const secp256k1_context *ctx, unsigned char *privkey, size_t *privkeylen, const unsigned char *key32, int compressed) {
  94     assert(*privkeylen >= PRIVATE_KEY_SIZE);
  95     static_assert(
  96         PRIVATE_KEY_SIZE >= COMPRESSED_PRIVATE_KEY_SIZE,
  97         "COMPRESSED_PRIVATE_KEY_SIZE is larger than PRIVATE_KEY_SIZE");
  98     secp256k1_pubkey pubkey;
  99     size_t pubkeylen = 0;
 100     if (!secp256k1_ec_pubkey_create(ctx, &pubkey, key32)) {
 101         *privkeylen = 0;
 102         return 0;
 103     }
 104     if (compressed) {
 105         static const unsigned char begin[] = {
 106             0x30,0x81,0xD3,0x02,0x01,0x01,0x04,0x20
 107         };
 108         static const unsigned char middle[] = {
 109             0xA0,0x81,0x85,0x30,0x81,0x82,0x02,0x01,0x01,0x30,0x2C,0x06,0x07,0x2A,0x86,0x48,
 110             0xCE,0x3D,0x01,0x01,0x02,0x21,0x00,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
 111             0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
 112             0xFF,0xFF,0xFE,0xFF,0xFF,0xFC,0x2F,0x30,0x06,0x04,0x01,0x00,0x04,0x01,0x07,0x04,
 113             0x21,0x02,0x79,0xBE,0x66,0x7E,0xF9,0xDC,0xBB,0xAC,0x55,0xA0,0x62,0x95,0xCE,0x87,
 114             0x0B,0x07,0x02,0x9B,0xFC,0xDB,0x2D,0xCE,0x28,0xD9,0x59,0xF2,0x81,0x5B,0x16,0xF8,
 115             0x17,0x98,0x02,0x21,0x00,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
 116             0xFF,0xFF,0xFF,0xFF,0xFE,0xBA,0xAE,0xDC,0xE6,0xAF,0x48,0xA0,0x3B,0xBF,0xD2,0x5E,
 117             0x8C,0xD0,0x36,0x41,0x41,0x02,0x01,0x01,0xA1,0x24,0x03,0x22,0x00
 118         };
 119         unsigned char *ptr = privkey;
 120         memcpy(ptr, begin, sizeof(begin)); ptr += sizeof(begin);
 121         memcpy(ptr, key32, 32); ptr += 32;
 122         memcpy(ptr, middle, sizeof(middle)); ptr += sizeof(middle);
 123         pubkeylen = COMPRESSED_PUBLIC_KEY_SIZE;
 124         secp256k1_ec_pubkey_serialize(ctx, ptr, &pubkeylen, &pubkey, SECP256K1_EC_COMPRESSED);
 125         ptr += pubkeylen;
 126         *privkeylen = ptr - privkey;
 127         assert(*privkeylen == COMPRESSED_PRIVATE_KEY_SIZE);
 128     } else {
 129         static const unsigned char begin[] = {
 130             0x30,0x82,0x01,0x13,0x02,0x01,0x01,0x04,0x20
 131         };
 132         static const unsigned char middle[] = {
 133             0xA0,0x81,0xA5,0x30,0x81,0xA2,0x02,0x01,0x01,0x30,0x2C,0x06,0x07,0x2A,0x86,0x48,
 134             0xCE,0x3D,0x01,0x01,0x02,0x21,0x00,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
 135             0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
 136             0xFF,0xFF,0xFE,0xFF,0xFF,0xFC,0x2F,0x30,0x06,0x04,0x01,0x00,0x04,0x01,0x07,0x04,
 137             0x41,0x04,0x79,0xBE,0x66,0x7E,0xF9,0xDC,0xBB,0xAC,0x55,0xA0,0x62,0x95,0xCE,0x87,
 138             0x0B,0x07,0x02,0x9B,0xFC,0xDB,0x2D,0xCE,0x28,0xD9,0x59,0xF2,0x81,0x5B,0x16,0xF8,
 139             0x17,0x98,0x48,0x3A,0xDA,0x77,0x26,0xA3,0xC4,0x65,0x5D,0xA4,0xFB,0xFC,0x0E,0x11,
 140             0x08,0xA8,0xFD,0x17,0xB4,0x48,0xA6,0x85,0x54,0x19,0x9C,0x47,0xD0,0x8F,0xFB,0x10,
 141             0xD4,0xB8,0x02,0x21,0x00,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
 142             0xFF,0xFF,0xFF,0xFF,0xFE,0xBA,0xAE,0xDC,0xE6,0xAF,0x48,0xA0,0x3B,0xBF,0xD2,0x5E,
 143             0x8C,0xD0,0x36,0x41,0x41,0x02,0x01,0x01,0xA1,0x44,0x03,0x42,0x00
 144         };
 145         unsigned char *ptr = privkey;
 146         memcpy(ptr, begin, sizeof(begin)); ptr += sizeof(begin);
 147         memcpy(ptr, key32, 32); ptr += 32;
 148         memcpy(ptr, middle, sizeof(middle)); ptr += sizeof(middle);
 149         pubkeylen = PUBLIC_KEY_SIZE;
 150         secp256k1_ec_pubkey_serialize(ctx, ptr, &pubkeylen, &pubkey, SECP256K1_EC_UNCOMPRESSED);
 151         ptr += pubkeylen;
 152         *privkeylen = ptr - privkey;
 153         assert(*privkeylen == PRIVATE_KEY_SIZE);
 154     }
 155     return 1;
 156 }
 157
 158 bool CKey::Check(const unsigned char *vch) {
 159     return secp256k1_ec_seckey_verify(secp256k1_context_sign, vch);
 160 }
 161
 162 void CKey::MakeNewKey(bool fCompressedIn) {
 163     do {
 164         GetRandBytes(vch, sizeof(vch));
 165     } while (!Check(vch));
 166     fValid = true;
 167     fCompressed = fCompressedIn;
 168 }
 169
 170 bool CKey::SetPrivKey(const CPrivKey &privkey, bool fCompressedIn) {
 171     if (!ec_privkey_import_der(secp256k1_context_sign, (unsigned char*)begin(), &privkey[0], privkey.size()))
 172         return false;
 173     fCompressed = fCompressedIn;
 174     fValid = true;
 175     return true;
 176 }
 177
 178 CPrivKey CKey::GetPrivKey() const {
 179     assert(fValid);
 180     CPrivKey privkey;
 181     int ret;
 182     size_t privkeylen;
 183     privkey.resize(PRIVATE_KEY_SIZE);
 184     privkeylen = PRIVATE_KEY_SIZE;
 185     ret = ec_privkey_export_der(secp256k1_context_sign, (unsigned char*)&privkey[0], &privkeylen, begin(), fCompressed ? SECP256K1_EC_COMPRESSED : SECP256K1_EC_UNCOMPRESSED);
 186     assert(ret);
 187     privkey.resize(privkeylen);
 188     return privkey;
 189 }
 190
 191 CPubKey CKey::GetPubKey() const {
 192     assert(fValid);
 193     secp256k1_pubkey pubkey;
 194     size_t clen = PUBLIC_KEY_SIZE;
 195     CPubKey result;
 196     int ret = secp256k1_ec_pubkey_create(secp256k1_context_sign, &pubkey, begin());
 197     assert(ret);
 198     secp256k1_ec_pubkey_serialize(secp256k1_context_sign, (unsigned char*)result.begin(), &clen, &pubkey, fCompressed ? SECP256K1_EC_COMPRESSED : SECP256K1_EC_UNCOMPRESSED);
 199     assert(result.size() == clen);
 200     assert(result.IsValid());
 201     return result;
 202 }
 203
 204 bool CKey::Sign(const uint256 &hash, std::vector<unsigned char>& vchSig, uint32_t test_case) const {
 205     if (!fValid)
 206         return false;
 207     vchSig.resize(SIGNATURE_SIZE);
 208     size_t nSigLen = SIGNATURE_SIZE;
 209     unsigned char extra_entropy[32] = {0};
 210     WriteLE32(extra_entropy, test_case);
 211     secp256k1_ecdsa_signature sig;
 212     int ret = secp256k1_ecdsa_sign(secp256k1_context_sign, &sig, hash.begin(), begin(), secp256k1_nonce_function_rfc6979, test_case ? extra_entropy : NULL);
 213     assert(ret);
 214     secp256k1_ecdsa_signature_serialize_der(secp256k1_context_sign, (unsigned char*)&vchSig[0], &nSigLen, &sig);
 215     vchSig.resize(nSigLen);
 216     return true;
 217 }
 218
 219 bool CKey::VerifyPubKey(const CPubKey& pubkey) const {
 220     if (pubkey.IsCompressed() != fCompressed) {
 221         return false;
 222     }
 223     unsigned char rnd[8];
 224     std::string str = "Zcash key verification\n";
 225     GetRandBytes(rnd, sizeof(rnd));
 226     uint256 hash;
 227     CHash256().Write((unsigned char*)str.data(), str.size()).Write(rnd, sizeof(rnd)).Finalize(hash.begin());
 228     std::vector<unsigned char> vchSig;
 229     Sign(hash, vchSig);
 230     return pubkey.Verify(hash, vchSig);
 231 }
 232
 233 bool CKey::SignCompact(const uint256 &hash, std::vector<unsigned char>& vchSig) const {
 234     if (!fValid)
 235         return false;
 236     vchSig.resize(COMPACT_SIGNATURE_SIZE);
 237     int rec = -1;
 238     secp256k1_ecdsa_recoverable_signature sig;
 239     int ret = secp256k1_ecdsa_sign_recoverable(secp256k1_context_sign, &sig, hash.begin(), begin(), secp256k1_nonce_function_rfc6979, NULL);
 240     assert(ret);
 241     secp256k1_ecdsa_recoverable_signature_serialize_compact(secp256k1_context_sign, (unsigned char*)&vchSig[1], &rec, &sig);
 242     assert(ret);
 243     assert(rec != -1);
 244     vchSig[0] = 27 + rec + (fCompressed ? 4 : 0);
 245     return true;
 246 }
 247
 248 bool CKey::Load(CPrivKey &privkey, CPubKey &vchPubKey, bool fSkipCheck=false) {
 249     if (!ec_privkey_import_der(secp256k1_context_sign, (unsigned char*)begin(), &privkey[0], privkey.size()))
 250         return false;
 251     fCompressed = vchPubKey.IsCompressed();
 252     fValid = true;
 253
 254     if (fSkipCheck)
 255         return true;
 256
 257     return VerifyPubKey(vchPubKey);
 258 }
 259
 260 bool CKey::Derive(CKey& keyChild, ChainCode &ccChild, unsigned int nChild, const ChainCode& cc) const {
 261     assert(IsValid());
 262     assert(IsCompressed());
 263     unsigned char out[64];
 264     LockObject(out);
 265     if ((nChild >> 31) == 0) {
 266         CPubKey pubkey = GetPubKey();
 267         assert(pubkey.size() == COMPRESSED_PUBLIC_KEY_SIZE);
 268         BIP32Hash(cc, nChild, *pubkey.begin(), pubkey.begin()+1, out);
 269     } else {
 270         assert(size() == 32);
 271         BIP32Hash(cc, nChild, 0, begin(), out);
 272     }
 273     memcpy(ccChild.begin(), out+32, 32);
 274     memcpy((unsigned char*)keyChild.begin(), begin(), 32);
 275     bool ret = secp256k1_ec_privkey_tweak_add(secp256k1_context_sign, (unsigned char*)keyChild.begin(), out);
 276     UnlockObject(out);
 277     keyChild.fCompressed = true;
 278     keyChild.fValid = ret;
 279     return ret;
 280 }
 281
 282 bool CExtKey::Derive(CExtKey &out, unsigned int nChild) const {
 283     out.nDepth = nDepth + 1;
 284     CKeyID id = key.GetPubKey().GetID();
 285     memcpy(&out.vchFingerprint[0], &id, 4);
 286     out.nChild = nChild;
 287     return key.Derive(out.key, out.chaincode, nChild, chaincode);
 288 }
 289
 290 void CExtKey::SetMaster(const unsigned char *seed, unsigned int nSeedLen) {
 291     static const unsigned char hashkey[] = {'B','i','t','c','o','i','n',' ','s','e','e','d'};
 292     unsigned char out[64];
 293     LockObject(out);
 294     CHMAC_SHA512(hashkey, sizeof(hashkey)).Write(seed, nSeedLen).Finalize(out);
 295     key.Set(&out[0], &out[32], true);
 296     memcpy(chaincode.begin(), &out[32], 32);
 297     UnlockObject(out);
 298     nDepth = 0;
 299     nChild = 0;
 300     memset(vchFingerprint, 0, sizeof(vchFingerprint));
 301 }
 302
 303 CExtPubKey CExtKey::Neuter() const {
 304     CExtPubKey ret;
 305     ret.nDepth = nDepth;
 306     memcpy(&ret.vchFingerprint[0], &vchFingerprint[0], 4);
 307     ret.nChild = nChild;
 308     ret.pubkey = key.GetPubKey();
 309     ret.chaincode = chaincode;
 310     return ret;
 311 }
 312
 313 void CExtKey::Encode(unsigned char code[74]) const {
 314     code[0] = nDepth;
 315     memcpy(code+1, vchFingerprint, 4);
 316     code[5] = (nChild >> 24) & 0xFF; code[6] = (nChild >> 16) & 0xFF;
 317     code[7] = (nChild >>  8) & 0xFF; code[8] = (nChild >>  0) & 0xFF;
 318     memcpy(code+9, chaincode.begin(), 32);
 319     code[41] = 0;
 320     assert(key.size() == 32);
 321     memcpy(code+42, key.begin(), 32);
 322 }
 323
 324 void CExtKey::Decode(const unsigned char code[74]) {
 325     nDepth = code[0];
 326     memcpy(vchFingerprint, code+1, 4);
 327     nChild = (code[5] << 24) | (code[6] << 16) | (code[7] << 8) | code[8];
 328     memcpy(chaincode.begin(), code+9, 32);
 329     key.Set(code+42, code+74, true);
 330 }
 331
 332 bool ECC_InitSanityCheck() {
 333     CKey key;
 334     key.MakeNewKey(true);
 335     CPubKey pubkey = key.GetPubKey();
 336     return key.VerifyPubKey(pubkey);
 337 }
 338
 339 void ECC_Start() {
 340     assert(secp256k1_context_sign == NULL);
 341
 342     secp256k1_context *ctx = secp256k1_context_create(SECP256K1_CONTEXT_SIGN);
 343     assert(ctx != NULL);
 344
 345     {
 346         // Pass in a random blinding seed to the secp256k1 context.
 347         unsigned char seed[32];
 348         LockObject(seed);
 349         GetRandBytes(seed, 32);
 350         bool ret = secp256k1_context_randomize(ctx, seed);
 351         assert(ret);
 352         UnlockObject(seed);
 353     }
 354
 355     secp256k1_context_sign = ctx;
 356 }
 357
 358 void ECC_Stop() {
 359     secp256k1_context *ctx = secp256k1_context_sign;
 360     secp256k1_context_sign = NULL;
 361
 362     if (ctx) {
 363         secp256k1_context_destroy(ctx);
 364     }
 365 }