Pieter Wuille
10 years ago
8 changed files with 234 additions and 164 deletions
@ -1,113 +0,0 @@ |
|||||||
// Copyright (c) 2012-2013 The Bitcoin Core developers
|
|
||||||
// Distributed under the MIT/X11 software license, see the accompanying
|
|
||||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
||||||
|
|
||||||
//
|
|
||||||
// Unit tests for canonical signatures
|
|
||||||
//
|
|
||||||
|
|
||||||
#include "data/sig_noncanonical.json.h" |
|
||||||
#include "data/sig_canonical.json.h" |
|
||||||
#include "key.h" |
|
||||||
#include "random.h" |
|
||||||
#include "script/interpreter.h" |
|
||||||
#include "util.h" |
|
||||||
#include "utilstrencodings.h" |
|
||||||
|
|
||||||
#include <boost/foreach.hpp> |
|
||||||
#include <boost/test/unit_test.hpp> |
|
||||||
#include "json/json_spirit_writer_template.h" |
|
||||||
#include <openssl/ecdsa.h> |
|
||||||
|
|
||||||
using namespace std; |
|
||||||
using namespace json_spirit; |
|
||||||
|
|
||||||
// In script_tests.cpp
|
|
||||||
extern Array read_json(const std::string& jsondata); |
|
||||||
|
|
||||||
BOOST_AUTO_TEST_SUITE(canonical_tests) |
|
||||||
|
|
||||||
// OpenSSL-based test for canonical signature (without test for hashtype byte)
|
|
||||||
bool static IsCanonicalSignature_OpenSSL_inner(const std::vector<unsigned char>& vchSig) |
|
||||||
{ |
|
||||||
if (vchSig.size() == 0) |
|
||||||
return false; |
|
||||||
const unsigned char *input = &vchSig[0]; |
|
||||||
ECDSA_SIG *psig = NULL; |
|
||||||
d2i_ECDSA_SIG(&psig, &input, vchSig.size()); |
|
||||||
if (psig == NULL) |
|
||||||
return false; |
|
||||||
unsigned char buf[256]; |
|
||||||
unsigned char *pbuf = buf; |
|
||||||
unsigned int nLen = i2d_ECDSA_SIG(psig, NULL); |
|
||||||
if (nLen != vchSig.size()) { |
|
||||||
ECDSA_SIG_free(psig); |
|
||||||
return false; |
|
||||||
} |
|
||||||
nLen = i2d_ECDSA_SIG(psig, &pbuf); |
|
||||||
ECDSA_SIG_free(psig); |
|
||||||
return (memcmp(&vchSig[0], &buf[0], nLen) == 0); |
|
||||||
} |
|
||||||
|
|
||||||
// OpenSSL-based test for canonical signature
|
|
||||||
bool static IsCanonicalSignature_OpenSSL(const std::vector<unsigned char> &vchSignature) { |
|
||||||
if (vchSignature.size() < 1) |
|
||||||
return false; |
|
||||||
if (vchSignature.size() > 127) |
|
||||||
return false; |
|
||||||
if (vchSignature[vchSignature.size() - 1] & 0x7C) |
|
||||||
return false; |
|
||||||
|
|
||||||
std::vector<unsigned char> vchSig(vchSignature); |
|
||||||
vchSig.pop_back(); |
|
||||||
if (!IsCanonicalSignature_OpenSSL_inner(vchSig)) |
|
||||||
return false; |
|
||||||
return true; |
|
||||||
} |
|
||||||
|
|
||||||
BOOST_AUTO_TEST_CASE(script_canon) |
|
||||||
{ |
|
||||||
Array tests = read_json(std::string(json_tests::sig_canonical, json_tests::sig_canonical + sizeof(json_tests::sig_canonical))); |
|
||||||
|
|
||||||
BOOST_FOREACH(Value &tv, tests) { |
|
||||||
string test = tv.get_str(); |
|
||||||
if (IsHex(test)) { |
|
||||||
std::vector<unsigned char> sig = ParseHex(test); |
|
||||||
BOOST_CHECK_MESSAGE(IsCanonicalSignature(sig, SCRIPT_VERIFY_STRICTENC), test); |
|
||||||
BOOST_CHECK_MESSAGE(IsCanonicalSignature_OpenSSL(sig), test); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
BOOST_AUTO_TEST_CASE(script_noncanon) |
|
||||||
{ |
|
||||||
Array tests = read_json(std::string(json_tests::sig_noncanonical, json_tests::sig_noncanonical + sizeof(json_tests::sig_noncanonical))); |
|
||||||
|
|
||||||
BOOST_FOREACH(Value &tv, tests) { |
|
||||||
string test = tv.get_str(); |
|
||||||
if (IsHex(test)) { |
|
||||||
std::vector<unsigned char> sig = ParseHex(test); |
|
||||||
BOOST_CHECK_MESSAGE(!IsCanonicalSignature(sig, SCRIPT_VERIFY_STRICTENC), test); |
|
||||||
BOOST_CHECK_MESSAGE(!IsCanonicalSignature_OpenSSL(sig), test); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
BOOST_AUTO_TEST_CASE(script_signstrict) |
|
||||||
{ |
|
||||||
for (int i=0; i<100; i++) { |
|
||||||
CKey key; |
|
||||||
key.MakeNewKey(i & 1); |
|
||||||
std::vector<unsigned char> sig; |
|
||||||
uint256 hash = GetRandHash(); |
|
||||||
|
|
||||||
BOOST_CHECK(key.Sign(hash, sig)); // Generate a random signature.
|
|
||||||
BOOST_CHECK(key.GetPubKey().Verify(hash, sig)); // Check it.
|
|
||||||
sig.push_back(0x01); // Append a sighash type.
|
|
||||||
|
|
||||||
BOOST_CHECK(IsCanonicalSignature(sig, SCRIPT_VERIFY_STRICTENC | SCRIPT_VERIFY_LOW_S)); |
|
||||||
BOOST_CHECK(IsCanonicalSignature_OpenSSL(sig)); |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
BOOST_AUTO_TEST_SUITE_END() |
|
||||||
Loading…
Reference in new issue