src/test/key_tests.cpp

   1 #include <boost/test/unit_test.hpp>
   2
   3 #include <string>
   4 #include <vector>
   5
   6 #include "key.h"
   7 #include "base58.h"
   8 #include "uint256.h"
   9 #include "util.h"
  10
  11 using namespace std;
  12
  13 static const string strSecret1     ("5HxWvvfubhXpYYpS3tJkw6fq9jE9j18THftkZjHHfmFiWtmAbrj");
  14 static const string strSecret2     ("5KC4ejrDjv152FGwP386VD1i2NYc5KkfSMyv1nGy1VGDxGHqVY3");
  15 static const string strSecret1C    ("Kwr371tjA9u2rFSMZjTNun2PXXP3WPZu2afRHTcta6KxEUdm1vEw");
  16 static const string strSecret2C    ("L3Hq7a8FEQwJkW1M2GNKDW28546Vp5miewcCzSqUD9kCAXrJdS3g");
  17 static const CBitcoinAddress addr1 ("1QFqqMUD55ZV3PJEJZtaKCsQmjLT6JkjvJ");
  18 static const CBitcoinAddress addr2 ("1F5y5E5FMc5YzdJtB9hLaUe43GDxEKXENJ");
  19 static const CBitcoinAddress addr1C("1NoJrossxPBKfCHuJXT4HadJrXRE9Fxiqs");
  20 static const CBitcoinAddress addr2C("1CRj2HyM1CXWzHAXLQtiGLyggNT9WQqsDs");
  21
  22
  23 static const string strAddressBad("1HV9Lc3sNHZxwj4Zk6fB38tEmBryq2cBiF");
  24
  25
  26 #ifdef KEY_TESTS_DUMPINFO
  27 void dumpKeyInfo(uint256 privkey)
  28 {
  29     CSecret secret;
  30     secret.resize(32);
  31     memcpy(&secret[0], &privkey, 32);
  32     vector<unsigned char> sec;
  33     sec.resize(32);
  34     memcpy(&sec[0], &secret[0], 32);
  35     printf("  * secret (hex): %s\n", HexStr(sec).c_str());
  36
  37     for (int nCompressed=0; nCompressed<2; nCompressed++)
  38     {
  39         bool fCompressed = nCompressed == 1;
  40         printf("  * %s:\n", fCompressed ? "compressed" : "uncompressed");
  41         CBitcoinSecret bsecret;
  42         bsecret.SetSecret(secret, fCompressed);
  43         printf("    * secret (base58): %s\n", bsecret.ToString().c_str());
  44         CKey key;
  45         key.SetSecret(secret, fCompressed);
  46         vector<unsigned char> vchPubKey = key.GetPubKey();
  47         printf("    * pubkey (hex): %s\n", HexStr(vchPubKey).c_str());
  48         printf("    * address (base58): %s\n", CBitcoinAddress(vchPubKey).ToString().c_str());
  49     }
  50 }
  51 #endif
  52
  53
  54 BOOST_AUTO_TEST_SUITE(key_tests)
  55
  56 BOOST_AUTO_TEST_CASE(key_test1)
  57 {
  58     CBitcoinSecret bsecret1, bsecret2, bsecret1C, bsecret2C, baddress1;
  59     BOOST_CHECK( bsecret1.SetString (strSecret1));
  60     BOOST_CHECK( bsecret2.SetString (strSecret2));
  61     BOOST_CHECK( bsecret1C.SetString(strSecret1C));
  62     BOOST_CHECK( bsecret2C.SetString(strSecret2C));
  63     BOOST_CHECK(!baddress1.SetString(strAddressBad));
  64
  65     bool fCompressed;
  66     CSecret secret1  = bsecret1.GetSecret (fCompressed);
  67     BOOST_CHECK(fCompressed == false);
  68     CSecret secret2  = bsecret2.GetSecret (fCompressed);
  69     BOOST_CHECK(fCompressed == false);
  70     CSecret secret1C = bsecret1C.GetSecret(fCompressed);
  71     BOOST_CHECK(fCompressed == true);
  72     CSecret secret2C = bsecret2C.GetSecret(fCompressed);
  73     BOOST_CHECK(fCompressed == true);
  74
  75     BOOST_CHECK(secret1 == secret1C);
  76     BOOST_CHECK(secret2 == secret2C);
  77
  78     CKey key1, key2, key1C, key2C;
  79     key1.SetSecret(secret1, false);
  80     key2.SetSecret(secret2, false);
  81     key1C.SetSecret(secret1, true);
  82     key2C.SetSecret(secret2, true);
  83
  84     BOOST_CHECK(addr1.Get()  == CTxDestination(key1.GetPubKey().GetID()));
  85     BOOST_CHECK(addr2.Get()  == CTxDestination(key2.GetPubKey().GetID()));
  86     BOOST_CHECK(addr1C.Get() == CTxDestination(key1C.GetPubKey().GetID()));
  87     BOOST_CHECK(addr2C.Get() == CTxDestination(key2C.GetPubKey().GetID()));
  88
  89     for (int n=0; n<16; n++)
  90     {
  91         string strMsg = strprintf("Very secret message %i: 11", n);
  92         uint256 hashMsg = Hash(strMsg.begin(), strMsg.end());
  93
  94         // normal signatures
  95
  96         vector<unsigned char> sign1, sign2, sign1C, sign2C;
  97
  98         BOOST_CHECK(key1.Sign (hashMsg, sign1));
  99         BOOST_CHECK(key2.Sign (hashMsg, sign2));
 100         BOOST_CHECK(key1C.Sign(hashMsg, sign1C));
 101         BOOST_CHECK(key2C.Sign(hashMsg, sign2C));
 102
 103         BOOST_CHECK( key1.Verify(hashMsg, sign1));
 104         BOOST_CHECK(!key1.Verify(hashMsg, sign2));
 105         BOOST_CHECK( key1.Verify(hashMsg, sign1C));
 106         BOOST_CHECK(!key1.Verify(hashMsg, sign2C));
 107
 108         BOOST_CHECK(!key2.Verify(hashMsg, sign1));
 109         BOOST_CHECK( key2.Verify(hashMsg, sign2));
 110         BOOST_CHECK(!key2.Verify(hashMsg, sign1C));
 111         BOOST_CHECK( key2.Verify(hashMsg, sign2C));
 112
 113         BOOST_CHECK( key1C.Verify(hashMsg, sign1));
 114         BOOST_CHECK(!key1C.Verify(hashMsg, sign2));
 115         BOOST_CHECK( key1C.Verify(hashMsg, sign1C));
 116         BOOST_CHECK(!key1C.Verify(hashMsg, sign2C));
 117
 118         BOOST_CHECK(!key2C.Verify(hashMsg, sign1));
 119         BOOST_CHECK( key2C.Verify(hashMsg, sign2));
 120         BOOST_CHECK(!key2C.Verify(hashMsg, sign1C));
 121         BOOST_CHECK( key2C.Verify(hashMsg, sign2C));
 122
 123         // compact signatures (with key recovery)
 124
 125         vector<unsigned char> csign1, csign2, csign1C, csign2C;
 126
 127         BOOST_CHECK(key1.SignCompact (hashMsg, csign1));
 128         BOOST_CHECK(key2.SignCompact (hashMsg, csign2));
 129         BOOST_CHECK(key1C.SignCompact(hashMsg, csign1C));
 130         BOOST_CHECK(key2C.SignCompact(hashMsg, csign2C));
 131
 132         CKey rkey1, rkey2, rkey1C, rkey2C;
 133
 134         BOOST_CHECK(rkey1.SetCompactSignature (hashMsg, csign1));
 135         BOOST_CHECK(rkey2.SetCompactSignature (hashMsg, csign2));
 136         BOOST_CHECK(rkey1C.SetCompactSignature(hashMsg, csign1C));
 137         BOOST_CHECK(rkey2C.SetCompactSignature(hashMsg, csign2C));
 138
 139
 140         BOOST_CHECK(rkey1.GetPubKey()  == key1.GetPubKey());
 141         BOOST_CHECK(rkey2.GetPubKey()  == key2.GetPubKey());
 142         BOOST_CHECK(rkey1C.GetPubKey() == key1C.GetPubKey());
 143         BOOST_CHECK(rkey2C.GetPubKey() == key2C.GetPubKey());
 144     }
 145 }
 146
 147 BOOST_AUTO_TEST_SUITE_END()