Changeset View
Changeset View
Standalone View
Standalone View
src/test/script_tests.cpp
| Show First 20 Lines • Show All 276 Lines • ▼ Show 20 Lines | std::vector<uint8_t> DoSign(const CKey &key, const uint256 &hash, | ||||
| s = std::vector<uint8_t>(vchSig.begin() + 6 + vchSig[3], | s = std::vector<uint8_t>(vchSig.begin() + 6 + vchSig[3], | ||||
| vchSig.begin() + 6 + vchSig[3] + | vchSig.begin() + 6 + vchSig[3] + | ||||
| vchSig[5 + vchSig[3]]); | vchSig[5 + vchSig[3]]); | ||||
| } while (lenR != r.size() || lenS != s.size()); | } while (lenR != r.size() || lenS != s.size()); | ||||
| return vchSig; | return vchSig; | ||||
| } | } | ||||
| std::vector<uint8_t> DoSignSchnorr(const CKey &key, | |||||
| const uint256 &hash) const { | |||||
| std::vector<uint8_t> vchSig; | |||||
| // no need to iterate for size; schnorrs are always same size. | |||||
| key.SignSchnorr(hash, vchSig); | |||||
| return vchSig; | |||||
| } | |||||
| public: | public: | ||||
| TestBuilder(const CScript &script_, const std::string &comment_, | TestBuilder(const CScript &script_, const std::string &comment_, | ||||
| uint32_t flags_, bool P2SH = false, | uint32_t flags_, bool P2SH = false, | ||||
| Amount nValue_ = Amount::zero()) | Amount nValue_ = Amount::zero()) | ||||
| : script(script_), havePush(false), comment(comment_), flags(flags_), | : script(script_), havePush(false), comment(comment_), flags(flags_), | ||||
| scriptError(SCRIPT_ERR_OK), nValue(nValue_) { | scriptError(SCRIPT_ERR_OK), nValue(nValue_) { | ||||
| CScript scriptPubKey = script; | CScript scriptPubKey = script; | ||||
| if (P2SH) { | if (P2SH) { | ||||
| ▲ Show 20 Lines • Show All 47 Lines • ▼ Show 20 Lines | TestBuilder &PushSig(const CKey &key, | ||||
| uint256 hash = SignatureHash(script, CTransaction(spendTx), 0, | uint256 hash = SignatureHash(script, CTransaction(spendTx), 0, | ||||
| sigHashType, amount, nullptr, sigFlags); | sigHashType, amount, nullptr, sigFlags); | ||||
| std::vector<uint8_t> vchSig = DoSign(key, hash, lenR, lenS); | std::vector<uint8_t> vchSig = DoSign(key, hash, lenR, lenS); | ||||
| vchSig.push_back(static_cast<uint8_t>(sigHashType.getRawSigHashType())); | vchSig.push_back(static_cast<uint8_t>(sigHashType.getRawSigHashType())); | ||||
| DoPush(vchSig); | DoPush(vchSig); | ||||
| return *this; | return *this; | ||||
| } | } | ||||
| TestBuilder & | |||||
| PushSigSchnorr(const CKey &key, SigHashType sigHashType = SigHashType(), | |||||
| Amount amount = Amount::zero(), | |||||
| uint32_t sigFlags = SCRIPT_ENABLE_SIGHASH_FORKID) { | |||||
| uint256 hash = SignatureHash(script, CTransaction(spendTx), 0, | |||||
| sigHashType, amount, nullptr, sigFlags); | |||||
| std::vector<uint8_t> vchSig = DoSignSchnorr(key, hash); | |||||
| vchSig.push_back(static_cast<uint8_t>(sigHashType.getRawSigHashType())); | |||||
| DoPush(vchSig); | |||||
| return *this; | |||||
| } | |||||
| TestBuilder &PushDataSig(const CKey &key, const std::vector<uint8_t> &data, | TestBuilder &PushDataSig(const CKey &key, const std::vector<uint8_t> &data, | ||||
| unsigned int lenR = 32, unsigned int lenS = 32) { | unsigned int lenR = 32, unsigned int lenS = 32) { | ||||
| std::vector<uint8_t> vchHash(32); | std::vector<uint8_t> vchHash(32); | ||||
| CSHA256().Write(data.data(), data.size()).Finalize(vchHash.data()); | CSHA256().Write(data.data(), data.size()).Finalize(vchHash.data()); | ||||
| DoPush(DoSign(key, uint256(vchHash), lenR, lenS)); | DoPush(DoSign(key, uint256(vchHash), lenR, lenS)); | ||||
| return *this; | return *this; | ||||
| } | } | ||||
| TestBuilder &PushDataSigSchnorr(const CKey &key, | |||||
| const std::vector<uint8_t> &data) { | |||||
| std::vector<uint8_t> vchHash(32); | |||||
| CSHA256().Write(data.data(), data.size()).Finalize(vchHash.data()); | |||||
| DoPush(DoSignSchnorr(key, uint256(vchHash))); | |||||
| return *this; | |||||
| } | |||||
| TestBuilder &Push(const CPubKey &pubkey) { | TestBuilder &Push(const CPubKey &pubkey) { | ||||
| DoPush(std::vector<uint8_t>(pubkey.begin(), pubkey.end())); | DoPush(std::vector<uint8_t>(pubkey.begin(), pubkey.end())); | ||||
| return *this; | return *this; | ||||
| } | } | ||||
| TestBuilder &PushRedeem() { | TestBuilder &PushRedeem() { | ||||
| DoPush(std::vector<uint8_t>(redeemscript.begin(), redeemscript.end())); | DoPush(std::vector<uint8_t>(redeemscript.begin(), redeemscript.end())); | ||||
| return *this; | return *this; | ||||
| ▲ Show 20 Lines • Show All 1,161 Lines • ▼ Show 20 Lines | tests.push_back(TestBuilder(CScript() << ToByteVector(keys.pubkey0H) | ||||
| << OP_CHECKDATASIGVERIFY << OP_TRUE, | << OP_CHECKDATASIGVERIFY << OP_TRUE, | ||||
| "CHECKDATASIGVERIFY with invalid hybrid pubkey", | "CHECKDATASIGVERIFY with invalid hybrid pubkey", | ||||
| checkdatasigschnorrflags) | checkdatasigschnorrflags) | ||||
| .PushDataSig(keys.key0, {}) | .PushDataSig(keys.key0, {}) | ||||
| .DamagePush(10) | .DamagePush(10) | ||||
| .Num(0) | .Num(0) | ||||
| .ScriptError(SCRIPT_ERR_PUBKEYTYPE)); | .ScriptError(SCRIPT_ERR_PUBKEYTYPE)); | ||||
| /* | |||||
| * Tests that have Schnorr signatures in them: | |||||
| */ | |||||
| tests.push_back( | |||||
| TestBuilder(CScript() << ToByteVector(keys.pubkey0) << OP_CHECKSIG, | |||||
| "Schnorr P2PK without flag", 0) | |||||
| .PushSigSchnorr(keys.key0) | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back( | |||||
| TestBuilder(CScript() << ToByteVector(keys.pubkey0) << OP_CHECKSIG, | |||||
| "Schnorr P2PK without flag with STRICTENC", | |||||
| SCRIPT_VERIFY_STRICTENC) | |||||
| .PushSigSchnorr(keys.key0) | |||||
| .ScriptError(SCRIPT_ERR_SIG_DER)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_DUP << OP_HASH160 | |||||
| << ToByteVector(keys.pubkey1C.GetID()) | |||||
| << OP_EQUALVERIFY << OP_CHECKSIG, | |||||
| "Schnorr P2PKH without flag", 0) | |||||
| .PushSigSchnorr(keys.key1) | |||||
| .Push(keys.pubkey1C) | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back( | |||||
| TestBuilder(CScript() << ToByteVector(keys.pubkey0C) << OP_CHECKSIG, | |||||
| "Schnorr P2SH(P2PK) without flag", SCRIPT_VERIFY_P2SH, true) | |||||
| .PushSigSchnorr(keys.key0) | |||||
| .PushRedeem() | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_DUP << OP_HASH160 | |||||
| << ToByteVector(keys.pubkey0.GetID()) | |||||
| << OP_EQUALVERIFY << OP_CHECKSIG, | |||||
| "Schnorr P2SH(P2PKH) without flag", | |||||
| SCRIPT_VERIFY_P2SH, true) | |||||
| .PushSigSchnorr(keys.key0) | |||||
| .Push(keys.pubkey0) | |||||
| .PushRedeem() | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_3 << ToByteVector(keys.pubkey0C) | |||||
| << ToByteVector(keys.pubkey1C) | |||||
| << ToByteVector(keys.pubkey2C) << OP_3 | |||||
| << OP_CHECKMULTISIG, | |||||
| "Schnorr 3-of-3 illegal", SCRIPT_ENABLE_SCHNORR) | |||||
| .Num(0) | |||||
| .PushSigSchnorr(keys.key0) | |||||
| .PushSigSchnorr(keys.key1) | |||||
| .PushSigSchnorr(keys.key2) | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_3 << ToByteVector(keys.pubkey0C) | |||||
| << ToByteVector(keys.pubkey1C) | |||||
| << ToByteVector(keys.pubkey2C) << OP_3 | |||||
| << OP_CHECKMULTISIG, | |||||
| "Schnorr-mixed 3-of-3 illegal", | |||||
| SCRIPT_ENABLE_SCHNORR) | |||||
| .Num(0) | |||||
| .PushSig(keys.key0) | |||||
| .PushSig(keys.key1) | |||||
| .PushSigSchnorr(keys.key2) | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_2 << ToByteVector(keys.pubkey0C) | |||||
| << ToByteVector(keys.pubkey1C) | |||||
| << ToByteVector(keys.pubkey2C) << OP_3 | |||||
| << OP_CHECKMULTISIG, | |||||
| "Schnorr P2SH(2-of-3) illegal", | |||||
| SCRIPT_VERIFY_P2SH | SCRIPT_ENABLE_SCHNORR, | |||||
| true) | |||||
| .Num(0) | |||||
| .PushSigSchnorr(keys.key1) | |||||
| .PushSigSchnorr(keys.key2) | |||||
| .PushRedeem() | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_1 << ToByteVector(keys.pubkey1C) | |||||
| << OP_1 << OP_CHECKMULTISIG, | |||||
| "ECDSA 1-of-1 without schnorr flag", 0) | |||||
| .Num(0) | |||||
| .PushSig(keys.key1)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_1 << ToByteVector(keys.pubkey1C) | |||||
| << OP_1 << OP_CHECKMULTISIG, | |||||
| "ECDSA 1-of-1 with schnorr flag", | |||||
| SCRIPT_ENABLE_SCHNORR) | |||||
| .Num(0) | |||||
| .PushSig(keys.key1)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_1 << ToByteVector(keys.pubkey1C) | |||||
| << OP_1 << OP_CHECKMULTISIG, | |||||
| "Schnorr 1-of-1 illegal", SCRIPT_ENABLE_SCHNORR) | |||||
| .Num(0) | |||||
| .PushSigSchnorr(keys.key1) | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_1 << ToByteVector(keys.pubkey1C) | |||||
| << OP_1 << OP_CHECKMULTISIG, | |||||
| "Schnorr 1-of-1 illegal without flag", 0) | |||||
| .Num(0) | |||||
| .PushSigSchnorr(keys.key1) | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back(TestBuilder(CScript() << OP_1 << ToByteVector(keys.pubkey1C) | |||||
| << OP_1 << OP_CHECKMULTISIG, | |||||
| "Schnorr 1-of-1 illegal with STRICTENC", | |||||
| SCRIPT_ENABLE_SCHNORR | SCRIPT_VERIFY_STRICTENC) | |||||
| .Num(0) | |||||
| .PushSigSchnorr(keys.key1) | |||||
| .ScriptError(SCRIPT_ERR_SIG_DER)); | |||||
| tests.push_back( | |||||
| TestBuilder(CScript() << OP_1 << ToByteVector(keys.pubkey1C) << OP_1 | |||||
| << OP_CHECKMULTISIG, | |||||
| "Schnorr 1-of-1 illegal without flag with STRICTENC", | |||||
| SCRIPT_VERIFY_STRICTENC) | |||||
| .Num(0) | |||||
| .PushSigSchnorr(keys.key1) | |||||
| .ScriptError(SCRIPT_ERR_SIG_DER)); | |||||
| tests.push_back( | |||||
| TestBuilder(CScript() << ToByteVector(keys.pubkey1C) << OP_CHECKDATASIG, | |||||
| "Schnorr Standard CHECKDATASIG without flag", | |||||
| checkdatasigflags) | |||||
| .PushDataSigSchnorr(keys.key1, {}) | |||||
| .Num(0) | |||||
| .ScriptError(SCRIPT_ERR_SIG_DER)); | |||||
| tests.push_back( | |||||
| TestBuilder(CScript() << ToByteVector(keys.pubkey1C) << OP_CHECKDATASIG, | |||||
| "Schnorr Standard CHECKDATASIG without flag but no " | |||||
| "NULLFAIL/STRICTENC", | |||||
| SCRIPT_ENABLE_CHECKDATASIG) | |||||
| .PushDataSigSchnorr(keys.key1, {}) | |||||
| .Num(0) | |||||
| .ScriptError(SCRIPT_ERR_EVAL_FALSE)); | |||||
| tests.push_back(TestBuilder(CScript() << ToByteVector(keys.pubkey1C) | |||||
| << OP_CHECKDATASIGVERIFY, | |||||
| "Schnorr CHECKDATASIGVERIFY without flag", | |||||
| checkdatasigflags) | |||||
| .PushDataSigSchnorr(keys.key1, {}) | |||||
| .Num(0) | |||||
| .ScriptError(SCRIPT_ERR_SIG_DER)); | |||||
| tests.push_back( | |||||
| TestBuilder( | |||||
| CScript() << ToByteVector(keys.pubkey1C) << OP_CHECKDATASIGVERIFY, | |||||
| "Schnorr CHECKDATASIGVERIFY without flag but no NULLFAIL/STRICTENC", | |||||
| SCRIPT_ENABLE_CHECKDATASIG) | |||||
| .PushDataSigSchnorr(keys.key1, {}) | |||||
| .Num(0) | |||||
| .ScriptError(SCRIPT_ERR_CHECKDATASIGVERIFY)); | |||||
| std::set<std::string> tests_set; | std::set<std::string> tests_set; | ||||
| { | { | ||||
| UniValue json_tests = read_json(std::string( | UniValue json_tests = read_json(std::string( | ||||
| json_tests::script_tests, | json_tests::script_tests, | ||||
| json_tests::script_tests + sizeof(json_tests::script_tests))); | json_tests::script_tests + sizeof(json_tests::script_tests))); | ||||
| for (unsigned int idx = 0; idx < json_tests.size(); idx++) { | for (unsigned int idx = 0; idx < json_tests.size(); idx++) { | ||||
| ▲ Show 20 Lines • Show All 755 Lines • Show Last 20 Lines | |||||