mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-11-11 02:35:18 +00:00
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package ceremony
|
|
|
|
import (
|
|
"crypto"
|
|
|
|
"github.com/iden3/go-iden3-crypto/poseidon"
|
|
"github.com/pkg/errors"
|
|
"go.uber.org/zap"
|
|
"source.quilibrium.com/quilibrium/monorepo/node/config"
|
|
"source.quilibrium.com/quilibrium/monorepo/node/keys"
|
|
)
|
|
|
|
func (e *CeremonyDataClockConsensusEngine) GetProvingKey(
|
|
engineConfig *config.EngineConfig,
|
|
) (crypto.Signer, keys.KeyType, []byte, []byte) {
|
|
provingKey, err := e.keyManager.GetSigningKey(engineConfig.ProvingKeyId)
|
|
if errors.Is(err, keys.KeyNotFoundErr) {
|
|
e.logger.Info("could not get proving key, generating")
|
|
provingKey, err = e.keyManager.CreateSigningKey(
|
|
engineConfig.ProvingKeyId,
|
|
keys.KeyTypeEd448,
|
|
)
|
|
}
|
|
|
|
if err != nil {
|
|
e.logger.Error("could not get proving key", zap.Error(err))
|
|
panic(err)
|
|
}
|
|
|
|
rawKey, err := e.keyManager.GetRawKey(engineConfig.ProvingKeyId)
|
|
if err != nil {
|
|
e.logger.Error("could not get proving key type", zap.Error(err))
|
|
panic(err)
|
|
}
|
|
|
|
provingKeyType := rawKey.Type
|
|
|
|
h, err := poseidon.HashBytes(rawKey.PublicKey)
|
|
if err != nil {
|
|
e.logger.Error("could not hash proving key", zap.Error(err))
|
|
panic(err)
|
|
}
|
|
|
|
provingKeyAddress := h.Bytes()
|
|
provingKeyAddress = append(
|
|
make([]byte, 32-len(provingKeyAddress)),
|
|
provingKeyAddress...,
|
|
)
|
|
|
|
return provingKey, provingKeyType, rawKey.PublicKey, provingKeyAddress
|
|
}
|
|
|
|
func (e *CeremonyDataClockConsensusEngine) IsInProverTrie(key []byte) bool {
|
|
h, err := poseidon.HashBytes(key)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
provingKeyAddress := h.Bytes()
|
|
return e.frameProverTrie.Contains(provingKeyAddress)
|
|
}
|