forked from TrueCloudLab/frostfs-contract
154 lines
3.3 KiB
Go
154 lines
3.3 KiB
Go
package reputationcontract
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neo-go/pkg/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/iterator"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/management"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/std"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/runtime"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
|
|
"github.com/nspcc-dev/neofs-contract/common"
|
|
)
|
|
|
|
const (
|
|
notaryDisabledKey = "notary"
|
|
|
|
version = 1
|
|
)
|
|
|
|
func _deploy(data interface{}, isUpdate bool) {
|
|
args := data.([]interface{})
|
|
notaryDisabled := args[0].(bool)
|
|
owner := args[1].(interop.Hash160)
|
|
|
|
ctx := storage.GetContext()
|
|
|
|
if !common.HasUpdateAccess(ctx) {
|
|
panic("only owner can reinitialize contract")
|
|
}
|
|
|
|
storage.Put(ctx, common.OwnerKey, owner)
|
|
|
|
// initialize the way to collect signatures
|
|
storage.Put(ctx, notaryDisabledKey, notaryDisabled)
|
|
if notaryDisabled {
|
|
common.InitVote(ctx)
|
|
runtime.Log("reputation contract notary disabled")
|
|
}
|
|
|
|
runtime.Log("reputation contract initialized")
|
|
}
|
|
|
|
func Migrate(script []byte, manifest []byte) bool {
|
|
ctx := storage.GetReadOnlyContext()
|
|
|
|
if !common.HasUpdateAccess(ctx) {
|
|
runtime.Log("only owner can update contract")
|
|
return false
|
|
}
|
|
|
|
management.Update(script, manifest)
|
|
runtime.Log("reputation contract updated")
|
|
|
|
return true
|
|
}
|
|
|
|
func Put(epoch int, peerID []byte, value []byte) {
|
|
ctx := storage.GetContext()
|
|
notaryDisabled := storage.Get(ctx, notaryDisabledKey).(bool)
|
|
|
|
var ( // for invocation collection without notary
|
|
alphabet []common.IRNode
|
|
nodeKey []byte
|
|
alphabetCall bool
|
|
)
|
|
|
|
if notaryDisabled {
|
|
alphabet = common.AlphabetNodes()
|
|
nodeKey = common.InnerRingInvoker(alphabet)
|
|
alphabetCall = len(nodeKey) != 0
|
|
} else {
|
|
multiaddr := common.AlphabetAddress()
|
|
alphabetCall = runtime.CheckWitness(multiaddr)
|
|
}
|
|
|
|
if !alphabetCall {
|
|
runtime.Notify("reputationPut", epoch, peerID, value)
|
|
return
|
|
}
|
|
|
|
id := storageID(epoch, peerID)
|
|
|
|
reputationValues := GetByID(id)
|
|
reputationValues = append(reputationValues, value)
|
|
|
|
rawValues := std.Serialize(reputationValues)
|
|
|
|
if notaryDisabled {
|
|
threshold := len(alphabet)*2/3 + 1
|
|
|
|
n := common.Vote(ctx, id, nodeKey)
|
|
if n < threshold {
|
|
return
|
|
}
|
|
|
|
common.RemoveVotes(ctx, id)
|
|
}
|
|
|
|
storage.Put(ctx, id, rawValues)
|
|
}
|
|
|
|
func Get(epoch int, peerID []byte) [][]byte {
|
|
id := storageID(epoch, peerID)
|
|
return GetByID(id)
|
|
}
|
|
|
|
func GetByID(id []byte) [][]byte {
|
|
ctx := storage.GetReadOnlyContext()
|
|
|
|
data := storage.Get(ctx, id)
|
|
if data == nil {
|
|
return [][]byte{}
|
|
}
|
|
|
|
return std.Deserialize(data.([]byte)).([][]byte)
|
|
}
|
|
|
|
// ListByEpoch returns list of IDs that may be used to get reputation data
|
|
// via GetByID method.
|
|
func ListByEpoch(epoch int) [][]byte {
|
|
ctx := storage.GetReadOnlyContext()
|
|
var buf interface{} = epoch
|
|
it := storage.Find(ctx, buf.([]byte), storage.KeysOnly)
|
|
|
|
var result [][]byte
|
|
|
|
ignore := [][]byte{
|
|
[]byte(common.OwnerKey),
|
|
[]byte(notaryDisabledKey),
|
|
}
|
|
|
|
loop:
|
|
for iterator.Next(it) {
|
|
key := iterator.Value(it).([]byte) // iterator MUST BE `storage.KeysOnly`
|
|
for _, ignoreKey := range ignore {
|
|
if common.BytesEqual(key, ignoreKey) {
|
|
continue loop
|
|
}
|
|
}
|
|
|
|
result = append(result, key)
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func Version() int {
|
|
return version
|
|
}
|
|
|
|
func storageID(epoch int, peerID []byte) []byte {
|
|
var buf interface{} = epoch
|
|
|
|
return append(buf.([]byte), peerID...)
|
|
}
|