You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.5 KiB
95 lines
2.5 KiB
package blsgen
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
bls_core "github.com/harmony-one/bls/ffi/go/bls"
|
|
"github.com/harmony-one/harmony/multibls"
|
|
)
|
|
|
|
// loadHelper defines the helper interface to load bls keys. Implemented by
|
|
// multiKeyLoader - load key files with a slice of target key files
|
|
// blsDirLoader - load key files from a directory
|
|
type loadHelper interface {
|
|
loadKeys() (multibls.PrivateKeys, error)
|
|
}
|
|
|
|
// multiKeyLoader load keys from multiple bls key files
|
|
type multiKeyLoader struct {
|
|
keyFiles []string
|
|
decrypters map[string]keyDecrypter
|
|
|
|
loadedSecrets []*bls_core.SecretKey
|
|
}
|
|
|
|
func newMultiKeyLoader(keyFiles []string, decrypters []keyDecrypter) (*multiKeyLoader, error) {
|
|
dm := make(map[string]keyDecrypter)
|
|
for _, decrypter := range decrypters {
|
|
dm[decrypter.extension()] = decrypter
|
|
}
|
|
for _, keyFile := range keyFiles {
|
|
ext := filepath.Ext(keyFile)
|
|
if _, supported := dm[ext]; !supported {
|
|
return nil, fmt.Errorf("unsupported key extension: %v", ext)
|
|
}
|
|
}
|
|
return &multiKeyLoader{
|
|
keyFiles: keyFiles,
|
|
decrypters: dm,
|
|
loadedSecrets: make([]*bls_core.SecretKey, 0, len(keyFiles)),
|
|
}, nil
|
|
}
|
|
|
|
func (loader *multiKeyLoader) loadKeys() (multibls.PrivateKeys, error) {
|
|
for _, keyFile := range loader.keyFiles {
|
|
decrypter := loader.decrypters[filepath.Ext(keyFile)]
|
|
secret, err := decrypter.decryptFile(keyFile)
|
|
if err != nil {
|
|
return multibls.PrivateKeys{}, err
|
|
}
|
|
loader.loadedSecrets = append(loader.loadedSecrets, secret)
|
|
}
|
|
return multibls.GetPrivateKeys(loader.loadedSecrets...), nil
|
|
}
|
|
|
|
type blsDirLoader struct {
|
|
keyDir string
|
|
decrypters map[string]keyDecrypter
|
|
|
|
loadedSecrets []*bls_core.SecretKey
|
|
}
|
|
|
|
func newBlsDirLoader(keyDir string, decrypters []keyDecrypter) (*blsDirLoader, error) {
|
|
dm := make(map[string]keyDecrypter)
|
|
for _, decrypter := range decrypters {
|
|
dm[decrypter.extension()] = decrypter
|
|
}
|
|
if err := checkIsDir(keyDir); err != nil {
|
|
return nil, err
|
|
}
|
|
return &blsDirLoader{
|
|
keyDir: keyDir,
|
|
decrypters: dm,
|
|
}, nil
|
|
}
|
|
|
|
func (loader *blsDirLoader) loadKeys() (multibls.PrivateKeys, error) {
|
|
filepath.Walk(loader.keyDir, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
decrypter, exist := loader.decrypters[filepath.Ext(path)]
|
|
if !exist {
|
|
return nil
|
|
}
|
|
secret, err := decrypter.decryptFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
loader.loadedSecrets = append(loader.loadedSecrets, secret)
|
|
return nil
|
|
})
|
|
return multibls.GetPrivateKeys(loader.loadedSecrets...), nil
|
|
}
|
|
|