src/script/ismine.cpp

   1 // Copyright (c) 2009-2010 Satoshi Nakamoto
   2 // Copyright (c) 2009-2016 The Bitcoin Core developers
   3 // Distributed under the MIT software license, see the accompanying
   4 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   5
   6 #include "ismine.h"
   7
   8 #include "key.h"
   9 #include "keystore.h"
  10 #include "script/script.h"
  11 #include "script/standard.h"
  12 #include "script/sign.h"
  13
  14 #include <boost/foreach.hpp>
  15
  16 typedef std::vector<unsigned char> valtype;
  17
  18 unsigned int HaveKeys(const std::vector<valtype>& pubkeys, const CKeyStore& keystore)
  19 {
  20     unsigned int nResult = 0;
  21     for (const valtype& pubkey : pubkeys)
  22     {
  23         CKeyID keyID = CPubKey(pubkey).GetID();
  24         if (keystore.HaveKey(keyID))
  25             ++nResult;
  26     }
  27     return nResult;
  28 }
  29
  30 isminetype IsMine(const CKeyStore& keystore, const CScript& scriptPubKey, SigVersion sigversion)
  31 {
  32     bool isInvalid = false;
  33     return IsMine(keystore, scriptPubKey, isInvalid, sigversion);
  34 }
  35
  36 isminetype IsMine(const CKeyStore& keystore, const CTxDestination& dest, SigVersion sigversion)
  37 {
  38     bool isInvalid = false;
  39     return IsMine(keystore, dest, isInvalid, sigversion);
  40 }
  41
  42 isminetype IsMine(const CKeyStore &keystore, const CTxDestination& dest, bool& isInvalid, SigVersion sigversion)
  43 {
  44     CScript script = GetScriptForDestination(dest);
  45     return IsMine(keystore, script, isInvalid, sigversion);
  46 }
  47
  48 isminetype IsMine(const CKeyStore &keystore, const CScript& scriptPubKey, bool& isInvalid, SigVersion sigversion)
  49 {
  50     std::vector<valtype> vSolutions;
  51     txnouttype whichType;
  52     if (!Solver(scriptPubKey, whichType, vSolutions)) {
  53         if (keystore.HaveWatchOnly(scriptPubKey))
  54             return ISMINE_WATCH_UNSOLVABLE;
  55         return ISMINE_NO;
  56     }
  57
  58     CKeyID keyID;
  59     switch (whichType)
  60     {
  61     case TX_NONSTANDARD:
  62     case TX_NULL_DATA:
  63         break;
  64     case TX_PUBKEY:
  65         keyID = CPubKey(vSolutions[0]).GetID();
  66         if (sigversion != SIGVERSION_BASE && vSolutions[0].size() != 33) {
  67             isInvalid = true;
  68             return ISMINE_NO;
  69         }
  70         if (keystore.HaveKey(keyID))
  71             return ISMINE_SPENDABLE;
  72         break;
  73     case TX_WITNESS_V0_KEYHASH:
  74     {
  75         if (!keystore.HaveCScript(CScriptID(CScript() << OP_0 << vSolutions[0]))) {
  76             // We do not support bare witness outputs unless the P2SH version of it would be
  77             // acceptable as well. This protects against matching before segwit activates.
  78             // This also applies to the P2WSH case.
  79             break;
  80         }
  81         isminetype ret = ::IsMine(keystore, GetScriptForDestination(CKeyID(uint160(vSolutions[0]))), isInvalid, SIGVERSION_WITNESS_V0);
  82         if (ret == ISMINE_SPENDABLE || ret == ISMINE_WATCH_SOLVABLE || (ret == ISMINE_NO && isInvalid))
  83             return ret;
  84         break;
  85     }
  86     case TX_PUBKEYHASH:
  87         keyID = CKeyID(uint160(vSolutions[0]));
  88         if (sigversion != SIGVERSION_BASE) {
  89             CPubKey pubkey;
  90             if (keystore.GetPubKey(keyID, pubkey) && !pubkey.IsCompressed()) {
  91                 isInvalid = true;
  92                 return ISMINE_NO;
  93             }
  94         }
  95         if (keystore.HaveKey(keyID))
  96             return ISMINE_SPENDABLE;
  97         break;
  98     case TX_SCRIPTHASH:
  99     {
 100         CScriptID scriptID = CScriptID(uint160(vSolutions[0]));
 101         CScript subscript;
 102         if (keystore.GetCScript(scriptID, subscript)) {
 103             isminetype ret = IsMine(keystore, subscript, isInvalid);
 104             if (ret == ISMINE_SPENDABLE || ret == ISMINE_WATCH_SOLVABLE || (ret == ISMINE_NO && isInvalid))
 105                 return ret;
 106         }
 107         break;
 108     }
 109     case TX_WITNESS_V0_SCRIPTHASH:
 110     {
 111         if (!keystore.HaveCScript(CScriptID(CScript() << OP_0 << vSolutions[0]))) {
 112             break;
 113         }
 114         uint160 hash;
 115         CRIPEMD160().Write(&vSolutions[0][0], vSolutions[0].size()).Finalize(hash.begin());
 116         CScriptID scriptID = CScriptID(hash);
 117         CScript subscript;
 118         if (keystore.GetCScript(scriptID, subscript)) {
 119             isminetype ret = IsMine(keystore, subscript, isInvalid, SIGVERSION_WITNESS_V0);
 120             if (ret == ISMINE_SPENDABLE || ret == ISMINE_WATCH_SOLVABLE || (ret == ISMINE_NO && isInvalid))
 121                 return ret;
 122         }
 123         break;
 124     }
 125
 126     case TX_MULTISIG:
 127     {
 128         // Only consider transactions "mine" if we own ALL the
 129         // keys involved. Multi-signature transactions that are
 130         // partially owned (somebody else has a key that can spend
 131         // them) enable spend-out-from-under-you attacks, especially
 132         // in shared-wallet situations.
 133         std::vector<valtype> keys(vSolutions.begin()+1, vSolutions.begin()+vSolutions.size()-1);
 134         if (sigversion != SIGVERSION_BASE) {
 135             for (size_t i = 0; i < keys.size(); i++) {
 136                 if (keys[i].size() != 33) {
 137                     isInvalid = true;
 138                     return ISMINE_NO;
 139                 }
 140             }
 141         }
 142         if (HaveKeys(keys, keystore) == keys.size())
 143             return ISMINE_SPENDABLE;
 144         break;
 145     }
 146     }
 147
 148     if (keystore.HaveWatchOnly(scriptPubKey)) {
 149         // TODO: This could be optimized some by doing some work after the above solver
 150         SignatureData sigs;
 151         return ProduceSignature(DummySignatureCreator(&keystore), scriptPubKey, sigs) ? ISMINE_WATCH_SOLVABLE : ISMINE_WATCH_UNSOLVABLE;
 152     }
 153     return ISMINE_NO;
 154 }