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 Init(notaryDisabled bool, owner 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 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...) }