diff options
Diffstat (limited to 'src/keystore.h')
-rw-r--r-- | src/keystore.h | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/src/keystore.h b/src/keystore.h index 2f37ec5078..3b6869b42d 100644 --- a/src/keystore.h +++ b/src/keystore.h @@ -4,7 +4,23 @@ #ifndef BITCOIN_KEYSTORE_H #define BITCOIN_KEYSTORE_H -bool AddKey(const CKey& key); -std::vector<unsigned char> GenerateNewKey(); +class CKeyStore +{ +public: + std::map<std::vector<unsigned char>, CPrivKey> mapKeys; + mutable CCriticalSection cs_mapKeys; + virtual bool AddKey(const CKey& key); + bool HaveKey(const std::vector<unsigned char> &vchPubKey) const + { + return (mapKeys.count(vchPubKey) > 0); + } + CPrivKey GetPrivKey(const std::vector<unsigned char> &vchPubKey) const + { + std::map<std::vector<unsigned char>, CPrivKey>::const_iterator mi = mapKeys.find(vchPubKey); + if (mi != mapKeys.end()) + return (*mi).second; + } + std::vector<unsigned char> GenerateNewKey(); +}; #endif |