Evgenii Stratonikov
074a241272
Because `Verify` is flexible enough, any funds transferred to the proxy contract can be moved out with the help of the committee. Thus, implementing onNEP11Payment() is enough. Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
95 lines
2.7 KiB
Go
95 lines
2.7 KiB
Go
package proxy
|
|
|
|
import (
|
|
"git.frostfs.info/TrueCloudLab/frostfs-contract/common"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/gas"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/management"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/runtime"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
|
|
)
|
|
|
|
const accountKeyPrefix = 'a'
|
|
|
|
// OnNEP17Payment is a callback for NEP-17 compatible native GAS contract.
|
|
func OnNEP17Payment(from interop.Hash160, amount int, data any) {
|
|
caller := runtime.GetCallingScriptHash()
|
|
if !common.BytesEqual(caller, []byte(gas.Hash)) {
|
|
common.AbortWithMessage("proxy contract accepts GAS only")
|
|
}
|
|
}
|
|
|
|
// OnNEP11Payment is a callback for NEP-11 compatible NNS contract.
|
|
func OnNEP11Payment(from interop.Hash160, amount int, token []byte, data any) {
|
|
caller := runtime.GetCallingScriptHash()
|
|
nnsHash := management.GetContractByID(1).Hash
|
|
if !common.BytesEqual(caller, []byte(nnsHash)) {
|
|
common.AbortWithMessage("proxy contract accepts NNS tokens only")
|
|
}
|
|
}
|
|
|
|
func _deploy(data any, isUpdate bool) {
|
|
if isUpdate {
|
|
args := data.([]any)
|
|
common.CheckVersion(args[len(args)-1].(int))
|
|
return
|
|
}
|
|
|
|
runtime.Log("proxy contract initialized")
|
|
}
|
|
|
|
// Update method updates contract source code and manifest. It can be invoked
|
|
// only by committee.
|
|
func Update(script []byte, manifest []byte, data any) {
|
|
if !common.HasUpdateAccess() {
|
|
panic("only committee can update contract")
|
|
}
|
|
|
|
management.UpdateWithData(script, manifest, common.AppendVersion(data))
|
|
runtime.Log("proxy contract updated")
|
|
}
|
|
|
|
// Verify method returns true if transaction contains valid multisignature of
|
|
// Alphabet nodes of the Inner Ring or any of the trusted accounts added via AddAccount.
|
|
func Verify() bool {
|
|
if !runtime.GetScriptContainer().Sender.Equals(runtime.GetExecutingScriptHash()) {
|
|
return false
|
|
}
|
|
|
|
signers := runtime.CurrentSigners()
|
|
ctx := storage.GetReadOnlyContext()
|
|
for i := 1; /* skip sender */ i < len(signers); i++ {
|
|
if storage.Get(ctx, append([]byte{accountKeyPrefix}, signers[i].Account...)) != nil {
|
|
return true
|
|
}
|
|
}
|
|
|
|
if runtime.CheckWitness(common.CommitteeAddress()) {
|
|
return true
|
|
}
|
|
|
|
if runtime.CheckWitness(common.AlphabetAddress()) {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
// Version returns the version of the contract.
|
|
func Version() int {
|
|
return common.Version
|
|
}
|
|
|
|
func AddAccount(addr interop.Hash160) {
|
|
common.CheckWitness(common.CommitteeAddress())
|
|
|
|
ctx := storage.GetContext()
|
|
storage.Put(ctx, append([]byte{accountKeyPrefix}, addr...), []byte{1})
|
|
}
|
|
|
|
func RemoveAccount(addr interop.Hash160) {
|
|
common.CheckWitness(common.CommitteeAddress())
|
|
|
|
ctx := storage.GetContext()
|
|
storage.Delete(ctx, append([]byte{accountKeyPrefix}, addr...))
|
|
}
|