You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
155 lines
5.3 KiB
155 lines
5.3 KiB
// Copyright (c) 2009-2010 Satoshi Nakamoto |
|
// Copyright (c) 2009-2017 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 <script/ismine.h> |
|
|
|
#include <key.h> |
|
#include <keystore.h> |
|
#include <script/script.h> |
|
#include <script/sign.h> |
|
|
|
|
|
typedef std::vector<unsigned char> valtype; |
|
|
|
unsigned int HaveKeys(const std::vector<valtype>& pubkeys, const CKeyStore& keystore) |
|
{ |
|
unsigned int nResult = 0; |
|
for (const valtype& pubkey : pubkeys) |
|
{ |
|
CKeyID keyID = CPubKey(pubkey).GetID(); |
|
if (keystore.HaveKey(keyID)) |
|
++nResult; |
|
} |
|
return nResult; |
|
} |
|
|
|
isminetype IsMine(const CKeyStore& keystore, const CScript& scriptPubKey, SigVersion sigversion) |
|
{ |
|
bool isInvalid = false; |
|
return IsMine(keystore, scriptPubKey, isInvalid, sigversion); |
|
} |
|
|
|
isminetype IsMine(const CKeyStore& keystore, const CTxDestination& dest, SigVersion sigversion) |
|
{ |
|
bool isInvalid = false; |
|
return IsMine(keystore, dest, isInvalid, sigversion); |
|
} |
|
|
|
isminetype IsMine(const CKeyStore &keystore, const CTxDestination& dest, bool& isInvalid, SigVersion sigversion) |
|
{ |
|
CScript script = GetScriptForDestination(dest); |
|
return IsMine(keystore, script, isInvalid, sigversion); |
|
} |
|
|
|
isminetype IsMine(const CKeyStore &keystore, const CScript& scriptPubKey, bool& isInvalid, SigVersion sigversion) |
|
{ |
|
isInvalid = false; |
|
|
|
std::vector<valtype> vSolutions; |
|
txnouttype whichType; |
|
if (!Solver(scriptPubKey, whichType, vSolutions)) { |
|
if (keystore.HaveWatchOnly(scriptPubKey)) |
|
return ISMINE_WATCH_UNSOLVABLE; |
|
return ISMINE_NO; |
|
} |
|
|
|
CKeyID keyID; |
|
switch (whichType) |
|
{ |
|
case TX_NONSTANDARD: |
|
case TX_NULL_DATA: |
|
case TX_WITNESS_UNKNOWN: |
|
break; |
|
case TX_PUBKEY: |
|
keyID = CPubKey(vSolutions[0]).GetID(); |
|
if (sigversion != SIGVERSION_BASE && vSolutions[0].size() != 33) { |
|
isInvalid = true; |
|
return ISMINE_NO; |
|
} |
|
if (keystore.HaveKey(keyID)) |
|
return ISMINE_SPENDABLE; |
|
break; |
|
case TX_WITNESS_V0_KEYHASH: |
|
{ |
|
if (!keystore.HaveCScript(CScriptID(CScript() << OP_0 << vSolutions[0]))) { |
|
// We do not support bare witness outputs unless the P2SH version of it would be |
|
// acceptable as well. This protects against matching before segwit activates. |
|
// This also applies to the P2WSH case. |
|
break; |
|
} |
|
isminetype ret = ::IsMine(keystore, GetScriptForDestination(CKeyID(uint160(vSolutions[0]))), isInvalid, SIGVERSION_WITNESS_V0); |
|
if (ret == ISMINE_SPENDABLE || ret == ISMINE_WATCH_SOLVABLE || (ret == ISMINE_NO && isInvalid)) |
|
return ret; |
|
break; |
|
} |
|
case TX_PUBKEYHASH: |
|
keyID = CKeyID(uint160(vSolutions[0])); |
|
if (sigversion != SIGVERSION_BASE) { |
|
CPubKey pubkey; |
|
if (keystore.GetPubKey(keyID, pubkey) && !pubkey.IsCompressed()) { |
|
isInvalid = true; |
|
return ISMINE_NO; |
|
} |
|
} |
|
if (keystore.HaveKey(keyID)) |
|
return ISMINE_SPENDABLE; |
|
break; |
|
case TX_SCRIPTHASH: |
|
{ |
|
CScriptID scriptID = CScriptID(uint160(vSolutions[0])); |
|
CScript subscript; |
|
if (keystore.GetCScript(scriptID, subscript)) { |
|
isminetype ret = IsMine(keystore, subscript, isInvalid); |
|
if (ret == ISMINE_SPENDABLE || ret == ISMINE_WATCH_SOLVABLE || (ret == ISMINE_NO && isInvalid)) |
|
return ret; |
|
} |
|
break; |
|
} |
|
case TX_WITNESS_V0_SCRIPTHASH: |
|
{ |
|
if (!keystore.HaveCScript(CScriptID(CScript() << OP_0 << vSolutions[0]))) { |
|
break; |
|
} |
|
uint160 hash; |
|
CRIPEMD160().Write(&vSolutions[0][0], vSolutions[0].size()).Finalize(hash.begin()); |
|
CScriptID scriptID = CScriptID(hash); |
|
CScript subscript; |
|
if (keystore.GetCScript(scriptID, subscript)) { |
|
isminetype ret = IsMine(keystore, subscript, isInvalid, SIGVERSION_WITNESS_V0); |
|
if (ret == ISMINE_SPENDABLE || ret == ISMINE_WATCH_SOLVABLE || (ret == ISMINE_NO && isInvalid)) |
|
return ret; |
|
} |
|
break; |
|
} |
|
|
|
case TX_MULTISIG: |
|
{ |
|
// Only consider transactions "mine" if we own ALL the |
|
// keys involved. Multi-signature transactions that are |
|
// partially owned (somebody else has a key that can spend |
|
// them) enable spend-out-from-under-you attacks, especially |
|
// in shared-wallet situations. |
|
std::vector<valtype> keys(vSolutions.begin()+1, vSolutions.begin()+vSolutions.size()-1); |
|
if (sigversion != SIGVERSION_BASE) { |
|
for (size_t i = 0; i < keys.size(); i++) { |
|
if (keys[i].size() != 33) { |
|
isInvalid = true; |
|
return ISMINE_NO; |
|
} |
|
} |
|
} |
|
if (HaveKeys(keys, keystore) == keys.size()) |
|
return ISMINE_SPENDABLE; |
|
break; |
|
} |
|
} |
|
|
|
if (keystore.HaveWatchOnly(scriptPubKey)) { |
|
// TODO: This could be optimized some by doing some work after the above solver |
|
SignatureData sigs; |
|
return ProduceSignature(DummySignatureCreator(&keystore), scriptPubKey, sigs) ? ISMINE_WATCH_SOLVABLE : ISMINE_WATCH_UNSOLVABLE; |
|
} |
|
return ISMINE_NO; |
|
}
|
|
|