forked from TrueCloudLab/frostfs-node
108 lines
2 KiB
Go
108 lines
2 KiB
Go
package frostfs
|
|
|
|
import (
|
|
"crypto/elliptic"
|
|
"fmt"
|
|
|
|
"github.com/TrueCloudLab/frostfs-node/pkg/morph/client/neofsid"
|
|
frostfs "github.com/TrueCloudLab/frostfs-node/pkg/morph/event/neofs"
|
|
"github.com/TrueCloudLab/frostfs-sdk-go/user"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type bindCommon interface {
|
|
User() []byte
|
|
Keys() [][]byte
|
|
TxHash() util.Uint256
|
|
}
|
|
|
|
func (np *Processor) processBind(e bindCommon) {
|
|
if !np.alphabetState.IsAlphabet() {
|
|
np.log.Info("non alphabet mode, ignore bind")
|
|
return
|
|
}
|
|
|
|
c := &bindCommonContext{
|
|
bindCommon: e,
|
|
}
|
|
|
|
_, c.bind = e.(frostfs.Bind)
|
|
|
|
err := np.checkBindCommon(c)
|
|
if err != nil {
|
|
np.log.Error("invalid manage key event",
|
|
zap.Bool("bind", c.bind),
|
|
zap.String("error", err.Error()),
|
|
)
|
|
|
|
return
|
|
}
|
|
|
|
np.approveBindCommon(c)
|
|
}
|
|
|
|
type bindCommonContext struct {
|
|
bindCommon
|
|
|
|
bind bool
|
|
|
|
scriptHash util.Uint160
|
|
}
|
|
|
|
func (np *Processor) checkBindCommon(e *bindCommonContext) error {
|
|
var err error
|
|
|
|
e.scriptHash, err = util.Uint160DecodeBytesBE(e.User())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
curve := elliptic.P256()
|
|
|
|
for _, key := range e.Keys() {
|
|
_, err = keys.NewPublicKeyFromBytes(key, curve)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (np *Processor) approveBindCommon(e *bindCommonContext) {
|
|
// calculate wallet address
|
|
scriptHash := e.User()
|
|
|
|
u160, err := util.Uint160DecodeBytesBE(scriptHash)
|
|
if err != nil {
|
|
np.log.Error("could not decode script hash from bytes",
|
|
zap.String("error", err.Error()),
|
|
)
|
|
|
|
return
|
|
}
|
|
|
|
var id user.ID
|
|
id.SetScriptHash(u160)
|
|
|
|
prm := neofsid.CommonBindPrm{}
|
|
prm.SetOwnerID(id.WalletBytes())
|
|
prm.SetKeys(e.Keys())
|
|
prm.SetHash(e.bindCommon.TxHash())
|
|
|
|
var typ string
|
|
if e.bind {
|
|
typ = "bind"
|
|
err = np.frostfsIDClient.AddKeys(prm)
|
|
} else {
|
|
typ = "unbind"
|
|
err = np.frostfsIDClient.RemoveKeys(prm)
|
|
}
|
|
|
|
if err != nil {
|
|
np.log.Error(fmt.Sprintf("could not approve %s", typ),
|
|
zap.String("error", err.Error()))
|
|
}
|
|
}
|