mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-11-26 19:42:23 +00:00
3b11f98cd0
This method returns persisted block height and doesn't take into account
persisting block height. Some of the callers of this method relay on
the wrong assumption that BlockHeight() returns persisting block index.
Fix improper usages of this method and adjust tests. Ref.
61a066583e/src/Neo/SmartContract/ApplicationEngine.cs (L634).
Signed-off-by: Anna Shaleva <shaleva.ann@nspcc.ru>
105 lines
2.9 KiB
Go
105 lines
2.9 KiB
Go
package native
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/config"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/trigger"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
)
|
|
|
|
// Call calls the specified native contract method.
|
|
func Call(ic *interop.Context) error {
|
|
version := ic.VM.Estack().Pop().BigInt().Int64()
|
|
if version != 0 {
|
|
return fmt.Errorf("native contract of version %d is not active", version)
|
|
}
|
|
var meta *interop.ContractMD
|
|
curr := ic.VM.GetCurrentScriptHash()
|
|
for _, ctr := range ic.Natives {
|
|
m := ctr.Metadata()
|
|
if m.Hash == curr {
|
|
meta = m
|
|
break
|
|
}
|
|
}
|
|
if meta == nil {
|
|
return fmt.Errorf("native contract %s (version %d) not found", curr.StringLE(), version)
|
|
}
|
|
history := meta.UpdateHistory
|
|
if len(history) == 0 {
|
|
return fmt.Errorf("native contract %s is disabled", meta.Name)
|
|
}
|
|
if history[0] > ic.BlockHeight() { // persisting block must not be taken into account.
|
|
return fmt.Errorf("native contract %s is active after height = %d", meta.Name, history[0])
|
|
}
|
|
m, ok := meta.GetMethodByOffset(ic.VM.Context().IP())
|
|
if !ok {
|
|
return fmt.Errorf("method not found")
|
|
}
|
|
reqFlags := m.RequiredFlags
|
|
if !ic.IsHardforkEnabled(config.HFAspidochelone) && meta.ID == ManagementContractID &&
|
|
(m.MD.Name == "deploy" || m.MD.Name == "update") {
|
|
reqFlags &= callflag.States | callflag.AllowNotify
|
|
}
|
|
if !ic.VM.Context().GetCallFlags().Has(reqFlags) {
|
|
return fmt.Errorf("missing call flags for native %d `%s` operation call: %05b vs %05b",
|
|
version, m.MD.Name, ic.VM.Context().GetCallFlags(), reqFlags)
|
|
}
|
|
invokeFee := m.CPUFee*ic.BaseExecFee() +
|
|
m.StorageFee*ic.BaseStorageFee()
|
|
if !ic.VM.AddGas(invokeFee) {
|
|
return errors.New("gas limit exceeded")
|
|
}
|
|
ctx := ic.VM.Context()
|
|
args := make([]stackitem.Item, len(m.MD.Parameters))
|
|
for i := range args {
|
|
args[i] = ic.VM.Estack().Peek(i).Item()
|
|
}
|
|
result := m.Func(ic, args)
|
|
for range m.MD.Parameters {
|
|
ic.VM.Estack().Pop()
|
|
}
|
|
if m.MD.ReturnType != smartcontract.VoidType {
|
|
ctx.Estack().PushItem(result)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// OnPersist calls OnPersist methods for all native contracts.
|
|
func OnPersist(ic *interop.Context) error {
|
|
if ic.Trigger != trigger.OnPersist {
|
|
return errors.New("onPersist must be trigered by system")
|
|
}
|
|
for _, c := range ic.Natives {
|
|
if !c.Metadata().IsActive(ic.Block.Index) {
|
|
continue
|
|
}
|
|
err := c.OnPersist(ic)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// PostPersist calls PostPersist methods for all native contracts.
|
|
func PostPersist(ic *interop.Context) error {
|
|
if ic.Trigger != trigger.PostPersist {
|
|
return errors.New("postPersist must be trigered by system")
|
|
}
|
|
for _, c := range ic.Natives {
|
|
if !c.Metadata().IsActive(ic.Block.Index) {
|
|
continue
|
|
}
|
|
err := c.PostPersist(ic)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|