src/keystore.cpp

   1 // Copyright (c) 2009-2010 Satoshi Nakamoto
   2 // Copyright (c) 2009-2014 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 "keystore.h"
   7
   8 #include "key.h"
   9 #include "util.h"
  10
  11 #include <boost/foreach.hpp>
  12
  13 bool CKeyStore::GetPubKey(const CKeyID &address, CPubKey &vchPubKeyOut) const
  14 {
  15     CKey key;
  16     if (!GetKey(address, key))
  17         return false;
  18     vchPubKeyOut = key.GetPubKey();
  19     return true;
  20 }
  21
  22 bool CKeyStore::AddKey(const CKey &key) {
  23     return AddKeyPubKey(key, key.GetPubKey());
  24 }
  25
  26 bool CBasicKeyStore::AddKeyPubKey(const CKey& key, const CPubKey &pubkey)
  27 {
  28     LOCK(cs_KeyStore);
  29     mapKeys[pubkey.GetID()] = key;
  30     return true;
  31 }
  32
  33 bool CBasicKeyStore::AddCScript(const CScript& redeemScript)
  34 {
  35     if (redeemScript.size() > MAX_SCRIPT_ELEMENT_SIZE)
  36         return error("CBasicKeyStore::AddCScript(): redeemScripts > %i bytes are invalid", MAX_SCRIPT_ELEMENT_SIZE);
  37
  38     LOCK(cs_KeyStore);
  39     mapScripts[CScriptID(redeemScript)] = redeemScript;
  40     return true;
  41 }
  42
  43 bool CBasicKeyStore::HaveCScript(const CScriptID& hash) const
  44 {
  45     LOCK(cs_KeyStore);
  46     return mapScripts.count(hash) > 0;
  47 }
  48
  49 bool CBasicKeyStore::GetCScript(const CScriptID &hash, CScript& redeemScriptOut) const
  50 {
  51     LOCK(cs_KeyStore);
  52     ScriptMap::const_iterator mi = mapScripts.find(hash);
  53     if (mi != mapScripts.end())
  54     {
  55         redeemScriptOut = (*mi).second;
  56         return true;
  57     }
  58     return false;
  59 }
  60
  61 bool CBasicKeyStore::AddWatchOnly(const CScript &dest)
  62 {
  63     LOCK(cs_KeyStore);
  64     setWatchOnly.insert(dest);
  65     return true;
  66 }
  67
  68 bool CBasicKeyStore::RemoveWatchOnly(const CScript &dest)
  69 {
  70     LOCK(cs_KeyStore);
  71     setWatchOnly.erase(dest);
  72     return true;
  73 }
  74
  75 bool CBasicKeyStore::HaveWatchOnly(const CScript &dest) const
  76 {
  77     LOCK(cs_KeyStore);
  78     return setWatchOnly.count(dest) > 0;
  79 }
  80
  81 bool CBasicKeyStore::HaveWatchOnly() const
  82 {
  83     LOCK(cs_KeyStore);
  84     return (!setWatchOnly.empty());
  85 }
  86
  87 bool CBasicKeyStore::AddSpendingKey(const libzcash::SpendingKey &sk)
  88 {
  89     LOCK(cs_SpendingKeyStore);
  90     auto address = sk.address();
  91     mapSpendingKeys[address] = sk;
  92     mapNoteDecryptors.insert(std::make_pair(address, ZCNoteDecryption(sk.viewing_key())));
  93     return true;
  94 }