git / moonbite-core / src/wallet/test/ismine_tests.cpp · raw

// Copyright (c) 2017-2019 The Bitcoin Core developers// Distributed under the MIT software license, see the accompanying// file COPYING or http://www.opensource.org/licenses/mit-license.php.#include <key.h>#include <node/context.h>#include <script/script.h>#include <script/standard.h>#include <test/util/setup_common.h>#include <wallet/ismine.h>#include <wallet/wallet.h>#include <boost/test/unit_test.hpp>BOOST_FIXTURE_TEST_SUITE(ismine_tests, BasicTestingSetup)BOOST_AUTO_TEST_CASE(ismine_standard){    CKey keys[2];    CPubKey pubkeys[2];    for (int i = 0; i < 2; i++) {        keys[i].MakeNewKey(true);        pubkeys[i] = keys[i].GetPubKey();    }    CKey uncompressedKey;    uncompressedKey.MakeNewKey(false);    CPubKey uncompressedPubkey = uncompressedKey.GetPubKey();    NodeContext node;    std::unique_ptr<interfaces::Chain> chain = interfaces::MakeChain(node);    CScript scriptPubKey;    isminetype result;    // P2PK compressed    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        scriptPubKey = GetScriptForRawPubKey(pubkeys[0]);        // Keystore does not have key        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has key        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // P2PK uncompressed    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        scriptPubKey = GetScriptForRawPubKey(uncompressedPubkey);        // Keystore does not have key        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has key        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(uncompressedKey));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // P2PKH compressed    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        scriptPubKey = GetScriptForDestination(PKHash(pubkeys[0]));        // Keystore does not have key        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has key        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // P2PKH uncompressed    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        scriptPubKey = GetScriptForDestination(PKHash(uncompressedPubkey));        // Keystore does not have key        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has key        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(uncompressedKey));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // P2SH    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        CScript redeemScript = GetScriptForDestination(PKHash(pubkeys[0]));        scriptPubKey = GetScriptForDestination(ScriptHash(redeemScript));        // Keystore does not have redeemScript or key        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has redeemScript but no key        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(redeemScript));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has redeemScript and key        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // (P2PKH inside) P2SH inside P2SH (invalid)    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        CScript redeemscript_inner = GetScriptForDestination(PKHash(pubkeys[0]));        CScript redeemscript = GetScriptForDestination(ScriptHash(redeemscript_inner));        scriptPubKey = GetScriptForDestination(ScriptHash(redeemscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(redeemscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(redeemscript_inner));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // (P2PKH inside) P2SH inside P2WSH (invalid)    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        CScript redeemscript = GetScriptForDestination(PKHash(pubkeys[0]));        CScript witnessscript = GetScriptForDestination(ScriptHash(redeemscript));        scriptPubKey = GetScriptForDestination(WitnessV0ScriptHash(witnessscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(witnessscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(redeemscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // P2WPKH inside P2WSH (invalid)    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        CScript witnessscript = GetScriptForDestination(WitnessV0KeyHash(pubkeys[0]));        scriptPubKey = GetScriptForDestination(WitnessV0ScriptHash(witnessscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(witnessscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // (P2PKH inside) P2WSH inside P2WSH (invalid)    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        CScript witnessscript_inner = GetScriptForDestination(PKHash(pubkeys[0]));        CScript witnessscript = GetScriptForDestination(WitnessV0ScriptHash(witnessscript_inner));        scriptPubKey = GetScriptForDestination(WitnessV0ScriptHash(witnessscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(witnessscript_inner));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(witnessscript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // P2WPKH compressed    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        scriptPubKey = GetScriptForDestination(WitnessV0KeyHash(pubkeys[0]));        // Keystore implicitly has key and P2SH redeemScript        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // P2WPKH uncompressed    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(uncompressedKey));        scriptPubKey = GetScriptForDestination(WitnessV0KeyHash(uncompressedPubkey));        // Keystore has key, but no P2SH redeemScript        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has key and P2SH redeemScript        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // scriptPubKey multisig    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        scriptPubKey = GetScriptForMultisig(2, {uncompressedPubkey, pubkeys[1]});        // Keystore does not have any keys        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has 1/2 keys        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(uncompressedKey));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has 2/2 keys        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[1]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has 2/2 keys and the script        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // P2SH multisig    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(uncompressedKey));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[1]));        CScript redeemScript = GetScriptForMultisig(2, {uncompressedPubkey, pubkeys[1]});        scriptPubKey = GetScriptForDestination(ScriptHash(redeemScript));        // Keystore has no redeemScript        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has redeemScript        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(redeemScript));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // P2WSH multisig with compressed keys    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[1]));        CScript witnessScript = GetScriptForMultisig(2, {pubkeys[0], pubkeys[1]});        scriptPubKey = GetScriptForDestination(WitnessV0ScriptHash(witnessScript));        // Keystore has keys, but no witnessScript or P2SH redeemScript        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has keys and witnessScript, but no P2SH redeemScript        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(witnessScript));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has keys, witnessScript, P2SH redeemScript        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // P2WSH multisig with uncompressed key    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(uncompressedKey));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[1]));        CScript witnessScript = GetScriptForMultisig(2, {uncompressedPubkey, pubkeys[1]});        scriptPubKey = GetScriptForDestination(WitnessV0ScriptHash(witnessScript));        // Keystore has keys, but no witnessScript or P2SH redeemScript        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has keys and witnessScript, but no P2SH redeemScript        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(witnessScript));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has keys, witnessScript, P2SH redeemScript        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(scriptPubKey));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // P2WSH multisig wrapped in P2SH    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        CScript witnessScript = GetScriptForMultisig(2, {pubkeys[0], pubkeys[1]});        CScript redeemScript = GetScriptForDestination(WitnessV0ScriptHash(witnessScript));        scriptPubKey = GetScriptForDestination(ScriptHash(redeemScript));        // Keystore has no witnessScript, P2SH redeemScript, or keys        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has witnessScript and P2SH redeemScript, but no keys        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(redeemScript));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddCScript(witnessScript));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);        // Keystore has keys, witnessScript, P2SH redeemScript        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[1]));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_SPENDABLE);    }    // OP_RETURN    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        scriptPubKey.clear();        scriptPubKey << OP_RETURN << ToByteVector(pubkeys[0]);        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // witness unspendable    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        scriptPubKey.clear();        scriptPubKey << OP_0 << ToByteVector(ParseHex("aabb"));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // witness unknown    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        scriptPubKey.clear();        scriptPubKey << OP_16 << ToByteVector(ParseHex("aabb"));        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }    // Nonstandard    {        CWallet keystore(chain.get(), "", CreateDummyWalletDatabase());        keystore.SetupLegacyScriptPubKeyMan();        LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);        BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));        scriptPubKey.clear();        scriptPubKey << OP_9 << OP_ADD << OP_11 << OP_EQUAL;        result = keystore.GetLegacyScriptPubKeyMan()->IsMine(scriptPubKey);        BOOST_CHECK_EQUAL(result, ISMINE_NO);    }}BOOST_AUTO_TEST_SUITE_END()