mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 20:07:17 +00:00
10b438e2c8
* WIP changes for keymanager * WIP fix * WIP needs unit tests * fixing order * adding unit test * fixing linter * updating unit tests and creating more reusable functions * making accountStore copy method part of struct * Update validator/keymanager/local/delete_test.go * Update validator/keymanager/local/delete.go * Update validator/keymanager/local/delete.go Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * Update validator/keymanager/local/import.go Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * Update validator/keymanager/local/delete.go Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * Update validator/keymanager/local/delete.go Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * Update validator/keymanager/local/import.go Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * Update validator/keymanager/local/import.go Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * Update validator/keymanager/local/delete.go Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * addressing suggestion of not reinitializing from reading the file but instead update the information based on memory on hand * Update validator/accounts/wallet_create.go Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * adding changes based on suggestions * making logs more consistent * fixing linting --------- Co-authored-by: Raul Jordan <raul@prysmaticlabs.com> Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com>
77 lines
2.7 KiB
Go
77 lines
2.7 KiB
Go
package local
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
|
|
fieldparams "github.com/prysmaticlabs/prysm/v4/config/fieldparams"
|
|
"github.com/prysmaticlabs/prysm/v4/encoding/bytesutil"
|
|
ethpbservice "github.com/prysmaticlabs/prysm/v4/proto/eth/service"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// DeleteKeystores takes in public keys and removes the accounts from the wallet.
|
|
// This includes their disk keystore and cached keystore, but maintains the slashing
|
|
// protection history in the database.
|
|
// 1) Copy the in memory keystore
|
|
// 2) Delete the keys from copied in memory keystore
|
|
// 3) Save the copy to disk
|
|
// 4) Reinitialize account store and updating the keymanager
|
|
// 5) Return API response
|
|
func (km *Keymanager) DeleteKeystores(
|
|
ctx context.Context, publicKeys [][]byte,
|
|
) ([]*ethpbservice.DeletedKeystoreStatus, error) {
|
|
// Check for duplicate keys and filter them out.
|
|
trackedPublicKeys := make(map[[fieldparams.BLSPubkeyLength]byte]bool)
|
|
statuses := make([]*ethpbservice.DeletedKeystoreStatus, 0, len(publicKeys))
|
|
deletedKeys := make([][]byte, 0, len(publicKeys))
|
|
// 1) Copy the in memory keystore
|
|
storeCopy := km.accountsStore.Copy()
|
|
for _, publicKey := range publicKeys {
|
|
// Check if the key in the request is a duplicate or not found
|
|
if _, ok := trackedPublicKeys[bytesutil.ToBytes48(publicKey)]; ok {
|
|
statuses = append(statuses, ðpbservice.DeletedKeystoreStatus{
|
|
Status: ethpbservice.DeletedKeystoreStatus_NOT_ACTIVE,
|
|
})
|
|
continue
|
|
}
|
|
var index int
|
|
var found bool
|
|
for j, pubKey := range storeCopy.PublicKeys {
|
|
if bytes.Equal(pubKey, publicKey) {
|
|
index = j
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
statuses = append(statuses, ðpbservice.DeletedKeystoreStatus{
|
|
Status: ethpbservice.DeletedKeystoreStatus_NOT_FOUND,
|
|
})
|
|
continue
|
|
}
|
|
// 2) Delete the keys from copied in memory keystore
|
|
deletedPublicKey := storeCopy.PublicKeys[index]
|
|
deletedKeys = append(deletedKeys, deletedPublicKey)
|
|
storeCopy.PrivateKeys = append(storeCopy.PrivateKeys[:index], storeCopy.PrivateKeys[index+1:]...)
|
|
storeCopy.PublicKeys = append(storeCopy.PublicKeys[:index], storeCopy.PublicKeys[index+1:]...)
|
|
statuses = append(statuses, ðpbservice.DeletedKeystoreStatus{
|
|
Status: ethpbservice.DeletedKeystoreStatus_DELETED,
|
|
})
|
|
trackedPublicKeys[bytesutil.ToBytes48(publicKey)] = true
|
|
}
|
|
if len(deletedKeys) == 0 {
|
|
return statuses, nil
|
|
}
|
|
// 3 & 4) save to disk and re-initializes keystore
|
|
if err := km.SaveStoreAndReInitialize(ctx, storeCopy); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
log.WithFields(logrus.Fields{
|
|
"publicKeys": CreatePrintoutOfKeys(deletedKeys),
|
|
}).Info("Successfully deleted validator key(s)")
|
|
// 5) Return API response
|
|
return statuses, nil
|
|
}
|