Page Menu
Home
Phabricator
Search
Configure Global Search
Log In
Files
F10907642
keystore.cpp
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Flag For Later
Size
5 KB
Subscribers
None
keystore.cpp
View Options
// Copyright (c) 2009-2010 Satoshi Nakamoto
// Copyright (c) 2009-2012 The Bitcoin developers
// Distributed under the MIT/X11 software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
#include
"keystore.h"
#include
"script.h"
bool
CKeyStore
::
GetPubKey
(
const
CKeyID
&
address
,
CPubKey
&
vchPubKeyOut
)
const
{
CKey
key
;
if
(
!
GetKey
(
address
,
key
))
return
false
;
vchPubKeyOut
=
key
.
GetPubKey
();
return
true
;
}
bool
CBasicKeyStore
::
AddKey
(
const
CKey
&
key
)
{
bool
fCompressed
=
false
;
CSecret
secret
=
key
.
GetSecret
(
fCompressed
);
{
LOCK
(
cs_KeyStore
);
mapKeys
[
key
.
GetPubKey
().
GetID
()]
=
make_pair
(
secret
,
fCompressed
);
}
return
true
;
}
bool
CBasicKeyStore
::
AddCScript
(
const
CScript
&
redeemScript
)
{
{
LOCK
(
cs_KeyStore
);
mapScripts
[
redeemScript
.
GetID
()]
=
redeemScript
;
}
return
true
;
}
bool
CBasicKeyStore
::
HaveCScript
(
const
CScriptID
&
hash
)
const
{
bool
result
;
{
LOCK
(
cs_KeyStore
);
result
=
(
mapScripts
.
count
(
hash
)
>
0
);
}
return
result
;
}
bool
CBasicKeyStore
::
GetCScript
(
const
CScriptID
&
hash
,
CScript
&
redeemScriptOut
)
const
{
{
LOCK
(
cs_KeyStore
);
ScriptMap
::
const_iterator
mi
=
mapScripts
.
find
(
hash
);
if
(
mi
!=
mapScripts
.
end
())
{
redeemScriptOut
=
(
*
mi
).
second
;
return
true
;
}
}
return
false
;
}
bool
CCryptoKeyStore
::
SetCrypted
()
{
{
LOCK
(
cs_KeyStore
);
if
(
fUseCrypto
)
return
true
;
if
(
!
mapKeys
.
empty
())
return
false
;
fUseCrypto
=
true
;
}
return
true
;
}
bool
CCryptoKeyStore
::
Lock
()
{
if
(
!
SetCrypted
())
return
false
;
{
LOCK
(
cs_KeyStore
);
vMasterKey
.
clear
();
}
NotifyStatusChanged
(
this
);
return
true
;
}
bool
CCryptoKeyStore
::
Unlock
(
const
CKeyingMaterial
&
vMasterKeyIn
)
{
{
LOCK
(
cs_KeyStore
);
if
(
!
SetCrypted
())
return
false
;
CryptedKeyMap
::
const_iterator
mi
=
mapCryptedKeys
.
begin
();
for
(;
mi
!=
mapCryptedKeys
.
end
();
++
mi
)
{
const
CPubKey
&
vchPubKey
=
(
*
mi
).
second
.
first
;
const
std
::
vector
<
unsigned
char
>
&
vchCryptedSecret
=
(
*
mi
).
second
.
second
;
CSecret
vchSecret
;
if
(
!
DecryptSecret
(
vMasterKeyIn
,
vchCryptedSecret
,
vchPubKey
.
GetHash
(),
vchSecret
))
return
false
;
if
(
vchSecret
.
size
()
!=
32
)
return
false
;
CKey
key
;
key
.
SetPubKey
(
vchPubKey
);
key
.
SetSecret
(
vchSecret
);
if
(
key
.
GetPubKey
()
==
vchPubKey
)
break
;
return
false
;
}
vMasterKey
=
vMasterKeyIn
;
}
NotifyStatusChanged
(
this
);
return
true
;
}
bool
CCryptoKeyStore
::
AddKey
(
const
CKey
&
key
)
{
{
LOCK
(
cs_KeyStore
);
if
(
!
IsCrypted
())
return
CBasicKeyStore
::
AddKey
(
key
);
if
(
IsLocked
())
return
false
;
std
::
vector
<
unsigned
char
>
vchCryptedSecret
;
CPubKey
vchPubKey
=
key
.
GetPubKey
();
bool
fCompressed
;
if
(
!
EncryptSecret
(
vMasterKey
,
key
.
GetSecret
(
fCompressed
),
vchPubKey
.
GetHash
(),
vchCryptedSecret
))
return
false
;
if
(
!
AddCryptedKey
(
key
.
GetPubKey
(),
vchCryptedSecret
))
return
false
;
}
return
true
;
}
bool
CCryptoKeyStore
::
AddCryptedKey
(
const
CPubKey
&
vchPubKey
,
const
std
::
vector
<
unsigned
char
>
&
vchCryptedSecret
)
{
{
LOCK
(
cs_KeyStore
);
if
(
!
SetCrypted
())
return
false
;
mapCryptedKeys
[
vchPubKey
.
GetID
()]
=
make_pair
(
vchPubKey
,
vchCryptedSecret
);
}
return
true
;
}
bool
CCryptoKeyStore
::
GetKey
(
const
CKeyID
&
address
,
CKey
&
keyOut
)
const
{
{
LOCK
(
cs_KeyStore
);
if
(
!
IsCrypted
())
return
CBasicKeyStore
::
GetKey
(
address
,
keyOut
);
CryptedKeyMap
::
const_iterator
mi
=
mapCryptedKeys
.
find
(
address
);
if
(
mi
!=
mapCryptedKeys
.
end
())
{
const
CPubKey
&
vchPubKey
=
(
*
mi
).
second
.
first
;
const
std
::
vector
<
unsigned
char
>
&
vchCryptedSecret
=
(
*
mi
).
second
.
second
;
CSecret
vchSecret
;
if
(
!
DecryptSecret
(
vMasterKey
,
vchCryptedSecret
,
vchPubKey
.
GetHash
(),
vchSecret
))
return
false
;
if
(
vchSecret
.
size
()
!=
32
)
return
false
;
keyOut
.
SetPubKey
(
vchPubKey
);
keyOut
.
SetSecret
(
vchSecret
);
return
true
;
}
}
return
false
;
}
bool
CCryptoKeyStore
::
GetPubKey
(
const
CKeyID
&
address
,
CPubKey
&
vchPubKeyOut
)
const
{
{
LOCK
(
cs_KeyStore
);
if
(
!
IsCrypted
())
return
CKeyStore
::
GetPubKey
(
address
,
vchPubKeyOut
);
CryptedKeyMap
::
const_iterator
mi
=
mapCryptedKeys
.
find
(
address
);
if
(
mi
!=
mapCryptedKeys
.
end
())
{
vchPubKeyOut
=
(
*
mi
).
second
.
first
;
return
true
;
}
}
return
false
;
}
bool
CCryptoKeyStore
::
EncryptKeys
(
CKeyingMaterial
&
vMasterKeyIn
)
{
{
LOCK
(
cs_KeyStore
);
if
(
!
mapCryptedKeys
.
empty
()
||
IsCrypted
())
return
false
;
fUseCrypto
=
true
;
BOOST_FOREACH
(
KeyMap
::
value_type
&
mKey
,
mapKeys
)
{
CKey
key
;
if
(
!
key
.
SetSecret
(
mKey
.
second
.
first
,
mKey
.
second
.
second
))
return
false
;
const
CPubKey
vchPubKey
=
key
.
GetPubKey
();
std
::
vector
<
unsigned
char
>
vchCryptedSecret
;
bool
fCompressed
;
if
(
!
EncryptSecret
(
vMasterKeyIn
,
key
.
GetSecret
(
fCompressed
),
vchPubKey
.
GetHash
(),
vchCryptedSecret
))
return
false
;
if
(
!
AddCryptedKey
(
vchPubKey
,
vchCryptedSecret
))
return
false
;
}
mapKeys
.
clear
();
}
return
true
;
}
File Metadata
Details
Attached
Mime Type
text/x-c
Expires
Mon, Nov 25, 08:07 (1 d, 19 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
4511496
Default Alt Text
keystore.cpp (5 KB)
Attached To
rABC Bitcoin ABC
Event Timeline
Log In to Comment