r.loginov/frostfs-contract
1
0
Fork 0
forked from TrueCloudLab/frostfs-contract
Code Pull requests Activity

[#23] alphabet: Add OnPayment method

Browse source 
Without OnPayment method alphabet contracts won't be able
to receive NEO and produce GAS.

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
Alex Vanin 2020-12-10 13:41:16 +03:00 committed by Alex Vanin
parent 37d8905d55
commit bd45e50480
8 changed files with 86 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
alphabet/alphabet.tpl
View file

@ -1,6 +1,7 @@
package alphabetcontract package alphabetcontract
import ( import (
"github.com/nspcc-dev/neo-go/pkg/interop"
"github.com/nspcc-dev/neo-go/pkg/interop/binary" "github.com/nspcc-dev/neo-go/pkg/interop/binary"
"github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/contract"
"github.com/nspcc-dev/neo-go/pkg/interop/crypto" "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
@ -51,6 +52,14 @@ func init() {
ctx = storage.GetContext() ctx = storage.GetContext()
} }
// OnPayment is a callback for NEP-17 compatible native GAS and NEO contracts.
func OnPayment(from interop.Hash160, amount int, data interface{}) {
caller := runtime.GetCallingScriptHash()
if !bytesEqual(caller, []byte(gasHash)) && !bytesEqual(caller, []byte(neoHash)) {
panic("onPayment: alphabet contract accepts GAS and NEO only")
}
}
func Init(addrNetmap []byte) { func Init(addrNetmap []byte) {
if storage.Get(ctx, netmapContractKey) != nil { if storage.Get(ctx, netmapContractKey) != nil {
panic("contract already deployed") panic("contract already deployed")

9
alphabet/az/az_contract.go
View file

@ -1,6 +1,7 @@
package alphabetcontract package alphabetcontract
import ( import (
"github.com/nspcc-dev/neo-go/pkg/interop"
"github.com/nspcc-dev/neo-go/pkg/interop/binary" "github.com/nspcc-dev/neo-go/pkg/interop/binary"
"github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/contract"
"github.com/nspcc-dev/neo-go/pkg/interop/crypto" "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
@ -51,6 +52,14 @@ func init() {
ctx = storage.GetContext() ctx = storage.GetContext()
} }
// OnPayment is a callback for NEP-17 compatible native GAS and NEO contracts.
func OnPayment(from interop.Hash160, amount int, data interface{}) {
caller := runtime.GetCallingScriptHash()
if !bytesEqual(caller, []byte(gasHash)) && !bytesEqual(caller, []byte(neoHash)) {
panic("onPayment: alphabet contract accepts GAS and NEO only")
}
}
func Init(addrNetmap []byte) { func Init(addrNetmap []byte) {
if storage.Get(ctx, netmapContractKey) != nil { if storage.Get(ctx, netmapContractKey) != nil {
panic("contract already deployed") panic("contract already deployed")

9
alphabet/buky/buky_contract.go
View file

@ -1,6 +1,7 @@
package alphabetcontract package alphabetcontract
import ( import (
"github.com/nspcc-dev/neo-go/pkg/interop"
"github.com/nspcc-dev/neo-go/pkg/interop/binary" "github.com/nspcc-dev/neo-go/pkg/interop/binary"
"github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/contract"
"github.com/nspcc-dev/neo-go/pkg/interop/crypto" "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
@ -51,6 +52,14 @@ func init() {
ctx = storage.GetContext() ctx = storage.GetContext()
} }
// OnPayment is a callback for NEP-17 compatible native GAS and NEO contracts.
func OnPayment(from interop.Hash160, amount int, data interface{}) {
caller := runtime.GetCallingScriptHash()
if !bytesEqual(caller, []byte(gasHash)) && !bytesEqual(caller, []byte(neoHash)) {
panic("onPayment: alphabet contract accepts GAS and NEO only")
}
}
func Init(addrNetmap []byte) { func Init(addrNetmap []byte) {
if storage.Get(ctx, netmapContractKey) != nil { if storage.Get(ctx, netmapContractKey) != nil {
panic("contract already deployed") panic("contract already deployed")

9
alphabet/dobro/dobro_contract.go
View file

@ -1,6 +1,7 @@
package alphabetcontract package alphabetcontract
import ( import (
"github.com/nspcc-dev/neo-go/pkg/interop"
"github.com/nspcc-dev/neo-go/pkg/interop/binary" "github.com/nspcc-dev/neo-go/pkg/interop/binary"
"github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/contract"
"github.com/nspcc-dev/neo-go/pkg/interop/crypto" "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
@ -51,6 +52,14 @@ func init() {
ctx = storage.GetContext() ctx = storage.GetContext()
} }
// OnPayment is a callback for NEP-17 compatible native GAS and NEO contracts.
func OnPayment(from interop.Hash160, amount int, data interface{}) {
caller := runtime.GetCallingScriptHash()
if !bytesEqual(caller, []byte(gasHash)) && !bytesEqual(caller, []byte(neoHash)) {
panic("onPayment: alphabet contract accepts GAS and NEO only")
}
}
func Init(addrNetmap []byte) { func Init(addrNetmap []byte) {
if storage.Get(ctx, netmapContractKey) != nil { if storage.Get(ctx, netmapContractKey) != nil {
panic("contract already deployed") panic("contract already deployed")

9
alphabet/glagoli/glagoli_contract.go
View file

@ -1,6 +1,7 @@
package alphabetcontract package alphabetcontract
import ( import (
"github.com/nspcc-dev/neo-go/pkg/interop"
"github.com/nspcc-dev/neo-go/pkg/interop/binary" "github.com/nspcc-dev/neo-go/pkg/interop/binary"
"github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/contract"
"github.com/nspcc-dev/neo-go/pkg/interop/crypto" "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
@ -51,6 +52,14 @@ func init() {
ctx = storage.GetContext() ctx = storage.GetContext()
} }
// OnPayment is a callback for NEP-17 compatible native GAS and NEO contracts.
func OnPayment(from interop.Hash160, amount int, data interface{}) {
caller := runtime.GetCallingScriptHash()
if !bytesEqual(caller, []byte(gasHash)) && !bytesEqual(caller, []byte(neoHash)) {
panic("onPayment: alphabet contract accepts GAS and NEO only")
}
}
func Init(addrNetmap []byte) { func Init(addrNetmap []byte) {
if storage.Get(ctx, netmapContractKey) != nil { if storage.Get(ctx, netmapContractKey) != nil {
panic("contract already deployed") panic("contract already deployed")

9
alphabet/jest/jest_contract.go
View file

@ -1,6 +1,7 @@
package alphabetcontract package alphabetcontract
import ( import (
"github.com/nspcc-dev/neo-go/pkg/interop"
"github.com/nspcc-dev/neo-go/pkg/interop/binary" "github.com/nspcc-dev/neo-go/pkg/interop/binary"
"github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/contract"
"github.com/nspcc-dev/neo-go/pkg/interop/crypto" "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
@ -51,6 +52,14 @@ func init() {
ctx = storage.GetContext() ctx = storage.GetContext()
} }
// OnPayment is a callback for NEP-17 compatible native GAS and NEO contracts.
func OnPayment(from interop.Hash160, amount int, data interface{}) {
caller := runtime.GetCallingScriptHash()
if !bytesEqual(caller, []byte(gasHash)) && !bytesEqual(caller, []byte(neoHash)) {
panic("onPayment: alphabet contract accepts GAS and NEO only")
}
}
func Init(addrNetmap []byte) { func Init(addrNetmap []byte) {
if storage.Get(ctx, netmapContractKey) != nil { if storage.Get(ctx, netmapContractKey) != nil {
panic("contract already deployed") panic("contract already deployed")

9
alphabet/vedi/vedi_contract.go
View file

@ -1,6 +1,7 @@
package alphabetcontract package alphabetcontract
import ( import (
"github.com/nspcc-dev/neo-go/pkg/interop"
"github.com/nspcc-dev/neo-go/pkg/interop/binary" "github.com/nspcc-dev/neo-go/pkg/interop/binary"
"github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/contract"
"github.com/nspcc-dev/neo-go/pkg/interop/crypto" "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
@ -51,6 +52,14 @@ func init() {
ctx = storage.GetContext() ctx = storage.GetContext()
} }
// OnPayment is a callback for NEP-17 compatible native GAS and NEO contracts.
func OnPayment(from interop.Hash160, amount int, data interface{}) {
caller := runtime.GetCallingScriptHash()
if !bytesEqual(caller, []byte(gasHash)) && !bytesEqual(caller, []byte(neoHash)) {
panic("onPayment: alphabet contract accepts GAS and NEO only")
}
}
func Init(addrNetmap []byte) { func Init(addrNetmap []byte) {
if storage.Get(ctx, netmapContractKey) != nil { if storage.Get(ctx, netmapContractKey) != nil {
panic("contract already deployed") panic("contract already deployed")

9
alphabet/zhivete/zhivete_contract.go
View file

@ -1,6 +1,7 @@
package alphabetcontract package alphabetcontract
import ( import (
"github.com/nspcc-dev/neo-go/pkg/interop"
"github.com/nspcc-dev/neo-go/pkg/interop/binary" "github.com/nspcc-dev/neo-go/pkg/interop/binary"
"github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/contract"
"github.com/nspcc-dev/neo-go/pkg/interop/crypto" "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
@ -51,6 +52,14 @@ func init() {
ctx = storage.GetContext() ctx = storage.GetContext()
} }
// OnPayment is a callback for NEP-17 compatible native GAS and NEO contracts.
func OnPayment(from interop.Hash160, amount int, data interface{}) {
caller := runtime.GetCallingScriptHash()
if !bytesEqual(caller, []byte(gasHash)) && !bytesEqual(caller, []byte(neoHash)) {
panic("onPayment: alphabet contract accepts GAS and NEO only")
}
}
func Init(addrNetmap []byte) { func Init(addrNetmap []byte) {
if storage.Get(ctx, netmapContractKey) != nil { if storage.Get(ctx, netmapContractKey) != nil {
panic("contract already deployed") panic("contract already deployed")