text: // Copyright (c) 2017-2019 The Pexa 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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          keystore.SetupLegacyScriptPubKeyMan();
          LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);
  
          CScript witnessscript = GetScriptForDestination(WitnessV0KeyHash(PKHash(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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          keystore.SetupLegacyScriptPubKeyMan();
          LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);
          BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(keys[0]));
  
          scriptPubKey = GetScriptForDestination(WitnessV0KeyHash(PKHash(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(), WalletLocation(), WalletDatabase::CreateDummy());
          keystore.SetupLegacyScriptPubKeyMan();
          LOCK(keystore.GetLegacyScriptPubKeyMan()->cs_KeyStore);
          BOOST_CHECK(keystore.GetLegacyScriptPubKeyMan()->AddKey(uncompressedKey));
  
          scriptPubKey = GetScriptForDestination(WitnessV0KeyHash(PKHash(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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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(), WalletLocation(), WalletDatabase::CreateDummy());
          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()
  