2018-02-19 09:24:28 +00:00
|
|
|
package compiler
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/binary"
|
2020-01-27 07:59:57 +00:00
|
|
|
"errors"
|
2019-11-22 14:16:52 +00:00
|
|
|
"fmt"
|
2018-02-19 09:24:28 +00:00
|
|
|
"go/ast"
|
|
|
|
"go/constant"
|
|
|
|
"go/token"
|
|
|
|
"go/types"
|
2020-02-21 08:50:03 +00:00
|
|
|
"math"
|
2019-02-15 13:38:16 +00:00
|
|
|
"sort"
|
2018-07-02 13:02:00 +00:00
|
|
|
"strings"
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2020-03-03 14:21:42 +00:00
|
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
2020-12-10 11:08:42 +00:00
|
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
2021-12-08 19:33:03 +00:00
|
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/nef"
|
2021-06-24 15:36:40 +00:00
|
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
2021-10-25 10:54:18 +00:00
|
|
|
"github.com/nspcc-dev/neo-go/pkg/util/bitfield"
|
2020-03-03 14:21:42 +00:00
|
|
|
"github.com/nspcc-dev/neo-go/pkg/vm"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
|
2020-06-03 12:55:06 +00:00
|
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
2021-12-02 14:44:53 +00:00
|
|
|
"golang.org/x/tools/go/packages"
|
2018-02-19 09:24:28 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type codegen struct {
|
2018-02-25 12:26:56 +00:00
|
|
|
// Information about the program with all its dependencies.
|
|
|
|
buildInfo *buildInfo
|
|
|
|
|
2019-10-22 14:56:03 +00:00
|
|
|
// prog holds the output buffer.
|
2019-11-22 10:06:32 +00:00
|
|
|
prog *io.BufBinWriter
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2019-10-22 14:56:03 +00:00
|
|
|
// Type information.
|
2018-02-19 09:24:28 +00:00
|
|
|
typeInfo *types.Info
|
2022-04-20 18:30:09 +00:00
|
|
|
// pkgInfoInline is a stack of type information for packages containing inline functions.
|
2021-12-02 14:44:53 +00:00
|
|
|
pkgInfoInline []*packages.Package
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2018-02-25 12:26:56 +00:00
|
|
|
// A mapping of func identifiers with their scope.
|
2018-02-19 09:24:28 +00:00
|
|
|
funcs map[string]*funcScope
|
|
|
|
|
2020-05-06 15:10:20 +00:00
|
|
|
// A mapping of lambda functions into their scope.
|
|
|
|
lambda map[string]*funcScope
|
|
|
|
|
2021-04-28 13:10:44 +00:00
|
|
|
// reverseOffsetMap maps function offsets to a local variable count.
|
|
|
|
reverseOffsetMap map[int]nameWithLocals
|
|
|
|
|
2018-02-25 12:26:56 +00:00
|
|
|
// Current funcScope being converted.
|
|
|
|
scope *funcScope
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2020-05-07 08:54:35 +00:00
|
|
|
globals map[string]int
|
2021-05-12 11:54:40 +00:00
|
|
|
// staticVariables contains global (static in NDX-DN11) variable names and types.
|
|
|
|
staticVariables []string
|
2021-05-12 12:08:22 +00:00
|
|
|
// initVariables contains variables local to `_initialize` method.
|
|
|
|
initVariables []string
|
2021-05-12 13:03:47 +00:00
|
|
|
// deployVariables contains variables local to `_initialize` method.
|
|
|
|
deployVariables []string
|
2020-05-07 08:54:35 +00:00
|
|
|
|
2020-02-17 13:48:41 +00:00
|
|
|
// A mapping from label's names to their ids.
|
2020-02-21 08:50:03 +00:00
|
|
|
labels map[labelWithType]uint16
|
2020-03-06 12:11:14 +00:00
|
|
|
// A list of nested label names together with evaluation stack depth.
|
|
|
|
labelList []labelWithStackSize
|
2022-07-11 08:38:18 +00:00
|
|
|
// inlineContext contains info about inlined function calls.
|
|
|
|
inlineContext []inlineContextSingle
|
2022-04-20 18:30:09 +00:00
|
|
|
// globalInlineCount contains the amount of auxiliary variables introduced by
|
2021-04-28 13:37:31 +00:00
|
|
|
// function inlining during global variables initialization.
|
|
|
|
globalInlineCount int
|
2020-02-17 13:48:41 +00:00
|
|
|
|
|
|
|
// A label for the for-loop being currently visited.
|
|
|
|
currentFor string
|
2020-02-18 14:07:38 +00:00
|
|
|
// A label for the switch statement being visited.
|
|
|
|
currentSwitch string
|
2020-02-17 13:48:41 +00:00
|
|
|
// A label to be used in the next statement.
|
|
|
|
nextLabel string
|
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// sequencePoints is a mapping from the method name to a slice
|
2020-03-31 13:57:35 +00:00
|
|
|
// containing info about mapping from opcode's offset
|
|
|
|
// to a text span in the source file.
|
|
|
|
sequencePoints map[string][]DebugSeqPoint
|
|
|
|
|
2020-07-24 10:40:54 +00:00
|
|
|
// initEndOffset specifies the end of the initialization method.
|
|
|
|
initEndOffset int
|
2020-10-02 10:16:02 +00:00
|
|
|
// deployEndOffset specifies the end of the deployment method.
|
|
|
|
deployEndOffset int
|
2020-07-24 10:40:54 +00:00
|
|
|
|
2020-07-28 16:35:41 +00:00
|
|
|
// importMap contains mapping from package aliases to full package names for the current file.
|
|
|
|
importMap map[string]string
|
|
|
|
|
2020-07-28 16:50:44 +00:00
|
|
|
// constMap contains constants from foreign packages.
|
|
|
|
constMap map[string]types.TypeAndValue
|
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// currPkg is the current package being processed.
|
2021-12-02 14:44:53 +00:00
|
|
|
currPkg *packages.Package
|
2020-07-29 14:20:00 +00:00
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// mainPkg is the main package metadata.
|
2021-12-02 14:44:53 +00:00
|
|
|
mainPkg *packages.Package
|
2020-07-24 10:40:54 +00:00
|
|
|
|
2020-08-05 07:56:36 +00:00
|
|
|
// packages contains packages in the order they were loaded.
|
2021-12-02 14:44:53 +00:00
|
|
|
packages []string
|
|
|
|
packageCache map[string]*packages.Package
|
2020-08-05 07:56:36 +00:00
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// exceptionIndex is the index of the static slot where the exception is stored.
|
2020-08-21 12:37:46 +00:00
|
|
|
exceptionIndex int
|
|
|
|
|
2020-08-10 10:10:35 +00:00
|
|
|
// documents contains paths to all files used by the program.
|
|
|
|
documents []string
|
2022-04-20 18:30:09 +00:00
|
|
|
// docIndex maps the file path to the index in the documents array.
|
2020-08-10 10:10:35 +00:00
|
|
|
docIndex map[string]int
|
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// emittedEvents contains all events emitted by the contract.
|
2023-05-24 08:52:14 +00:00
|
|
|
emittedEvents map[string][]EmittedEventInfo
|
2020-11-24 13:36:35 +00:00
|
|
|
|
2021-06-24 15:36:40 +00:00
|
|
|
// invokedContracts contains invoked methods of other contracts.
|
|
|
|
invokedContracts map[util.Uint160][]string
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
// Label table for recording jump destinations.
|
|
|
|
l []int
|
2021-12-08 19:33:03 +00:00
|
|
|
|
|
|
|
// Tokens for CALLT instruction
|
|
|
|
callTokens []nef.MethodToken
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2020-02-17 13:48:41 +00:00
|
|
|
type labelOffsetType byte
|
|
|
|
|
|
|
|
const (
|
|
|
|
labelStart labelOffsetType = iota // labelStart is a default label type
|
|
|
|
labelEnd // labelEnd is a type for labels that are targets for break
|
2020-02-17 14:19:54 +00:00
|
|
|
labelPost // labelPost is a type for labels that are targets for continue
|
2020-02-17 13:48:41 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type labelWithType struct {
|
|
|
|
name string
|
|
|
|
typ labelOffsetType
|
|
|
|
}
|
|
|
|
|
2020-03-06 12:11:14 +00:00
|
|
|
type labelWithStackSize struct {
|
|
|
|
name string
|
|
|
|
sz int
|
|
|
|
}
|
|
|
|
|
2021-04-28 13:10:44 +00:00
|
|
|
type nameWithLocals struct {
|
|
|
|
name string
|
|
|
|
count int
|
|
|
|
}
|
|
|
|
|
2022-07-11 08:38:18 +00:00
|
|
|
type inlineContextSingle struct {
|
|
|
|
// labelOffset contains size of labelList at the start of inline call processing.
|
|
|
|
// For such calls, we need to drop only the newly created part of stack.
|
|
|
|
labelOffset int
|
|
|
|
// returnLabel contains label ID pointing to the first instruction right after the call.
|
|
|
|
returnLabel uint16
|
|
|
|
}
|
|
|
|
|
2020-05-07 08:54:35 +00:00
|
|
|
type varType int
|
|
|
|
|
|
|
|
const (
|
|
|
|
varGlobal varType = iota
|
|
|
|
varLocal
|
|
|
|
varArgument
|
|
|
|
)
|
|
|
|
|
2022-09-28 06:58:59 +00:00
|
|
|
// ErrUnsupportedTypeAssertion is returned when type assertion statement is not supported by the compiler.
|
|
|
|
var ErrUnsupportedTypeAssertion = errors.New("type assertion with two return values is not supported")
|
|
|
|
|
2021-05-12 20:17:03 +00:00
|
|
|
// newLabel creates a new label to jump to.
|
2020-02-21 08:50:03 +00:00
|
|
|
func (c *codegen) newLabel() (l uint16) {
|
|
|
|
li := len(c.l)
|
|
|
|
if li > math.MaxUint16 {
|
|
|
|
c.prog.Err = errors.New("label number is too big")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
l = uint16(li)
|
2018-02-19 09:24:28 +00:00
|
|
|
c.l = append(c.l, -1)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// newNamedLabel creates a new label with the specified name.
|
2020-02-21 08:50:03 +00:00
|
|
|
func (c *codegen) newNamedLabel(typ labelOffsetType, name string) (l uint16) {
|
2020-02-17 13:48:41 +00:00
|
|
|
l = c.newLabel()
|
|
|
|
lt := labelWithType{name: name, typ: typ}
|
|
|
|
c.labels[lt] = l
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-21 08:50:03 +00:00
|
|
|
func (c *codegen) setLabel(l uint16) {
|
2018-02-19 09:24:28 +00:00
|
|
|
c.l[l] = c.pc() + 1
|
|
|
|
}
|
|
|
|
|
2019-10-22 14:56:03 +00:00
|
|
|
// pc returns the program offset off the last instruction.
|
2018-02-19 09:24:28 +00:00
|
|
|
func (c *codegen) pc() int {
|
|
|
|
return c.prog.Len() - 1
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *codegen) emitLoadConst(t types.TypeAndValue) {
|
2019-11-22 10:06:32 +00:00
|
|
|
if c.prog.Err != nil {
|
|
|
|
return
|
|
|
|
}
|
2020-05-13 15:09:55 +00:00
|
|
|
|
|
|
|
typ, ok := t.Type.Underlying().(*types.Basic)
|
|
|
|
if !ok {
|
2019-11-22 14:16:52 +00:00
|
|
|
c.prog.Err = fmt.Errorf("compiler doesn't know how to convert this constant: %v", t)
|
|
|
|
return
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2020-01-23 14:28:35 +00:00
|
|
|
switch typ.Kind() {
|
2020-04-28 13:42:29 +00:00
|
|
|
case types.Int, types.UntypedInt, types.Uint,
|
2020-05-13 07:53:11 +00:00
|
|
|
types.Int8, types.Uint8,
|
2020-04-28 13:42:29 +00:00
|
|
|
types.Int16, types.Uint16,
|
|
|
|
types.Int32, types.Uint32, types.Int64, types.Uint64:
|
2020-01-23 14:28:35 +00:00
|
|
|
val, _ := constant.Int64Val(t.Value)
|
2020-02-03 14:58:21 +00:00
|
|
|
emit.Int(c.prog.BinWriter, val)
|
2020-01-23 14:28:35 +00:00
|
|
|
case types.String, types.UntypedString:
|
|
|
|
val := constant.StringVal(t.Value)
|
2020-02-03 14:58:21 +00:00
|
|
|
emit.String(c.prog.BinWriter, val)
|
2020-01-23 14:28:35 +00:00
|
|
|
case types.Bool, types.UntypedBool:
|
|
|
|
val := constant.BoolVal(t.Value)
|
2020-02-03 14:58:21 +00:00
|
|
|
emit.Bool(c.prog.BinWriter, val)
|
2020-01-23 14:28:35 +00:00
|
|
|
default:
|
|
|
|
c.prog.Err = fmt.Errorf("compiler doesn't know how to convert this basic type: %v", t)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-05-07 08:54:35 +00:00
|
|
|
func (c *codegen) emitLoadField(i int) {
|
|
|
|
emit.Int(c.prog.BinWriter, int64(i))
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PICKITEM)
|
2018-03-25 16:21:00 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2020-05-07 08:54:35 +00:00
|
|
|
func (c *codegen) emitStoreStructField(i int) {
|
|
|
|
emit.Int(c.prog.BinWriter, int64(i))
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.ROT, opcode.SETITEM)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// getVarIndex returns variable type and position in the corresponding slot,
|
|
|
|
// according to the current scope.
|
2021-02-05 13:15:26 +00:00
|
|
|
func (c *codegen) getVarIndex(pkg string, name string) *varInfo {
|
2020-07-28 16:35:41 +00:00
|
|
|
if pkg == "" {
|
|
|
|
if c.scope != nil {
|
2021-02-05 13:15:26 +00:00
|
|
|
vi := c.scope.vars.getVarInfo(name)
|
|
|
|
if vi != nil {
|
|
|
|
return vi
|
2020-07-28 16:35:41 +00:00
|
|
|
}
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
|
|
|
}
|
2020-07-28 16:35:41 +00:00
|
|
|
if i, ok := c.globals[c.getIdentName(pkg, name)]; ok {
|
2021-02-05 13:15:26 +00:00
|
|
|
return &varInfo{refType: varGlobal, index: i}
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2021-02-05 13:15:26 +00:00
|
|
|
c.scope.newVariable(varLocal, name)
|
|
|
|
return c.scope.vars.getVarInfo(name)
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func getBaseOpcode(t varType) (opcode.Opcode, opcode.Opcode) {
|
|
|
|
switch t {
|
|
|
|
case varGlobal:
|
|
|
|
return opcode.LDSFLD0, opcode.STSFLD0
|
|
|
|
case varLocal:
|
|
|
|
return opcode.LDLOC0, opcode.STLOC0
|
|
|
|
case varArgument:
|
|
|
|
return opcode.LDARG0, opcode.STARG0
|
|
|
|
default:
|
|
|
|
panic("invalid type")
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// emitLoadVar loads the specified variable to the evaluation stack.
|
2020-07-28 16:35:41 +00:00
|
|
|
func (c *codegen) emitLoadVar(pkg string, name string) {
|
2021-02-05 13:15:26 +00:00
|
|
|
vi := c.getVarIndex(pkg, name)
|
2021-05-25 13:02:28 +00:00
|
|
|
if vi.ctx != nil && c.typeAndValueOf(vi.ctx.expr).Value != nil {
|
|
|
|
c.emitLoadConst(c.typeAndValueOf(vi.ctx.expr))
|
|
|
|
return
|
|
|
|
} else if vi.ctx != nil {
|
|
|
|
var oldScope []map[string]varInfo
|
|
|
|
oldMap := c.importMap
|
|
|
|
c.importMap = vi.ctx.importMap
|
|
|
|
if c.scope != nil {
|
|
|
|
oldScope = c.scope.vars.locals
|
|
|
|
c.scope.vars.locals = vi.ctx.scope
|
|
|
|
}
|
|
|
|
|
|
|
|
ast.Walk(c, vi.ctx.expr)
|
|
|
|
|
|
|
|
if c.scope != nil {
|
|
|
|
c.scope.vars.locals = oldScope
|
|
|
|
}
|
|
|
|
c.importMap = oldMap
|
2021-02-05 13:15:26 +00:00
|
|
|
return
|
2021-02-05 14:26:26 +00:00
|
|
|
} else if vi.index == unspecifiedVarIndex {
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSHNULL)
|
|
|
|
return
|
2021-02-05 13:15:26 +00:00
|
|
|
}
|
|
|
|
c.emitLoadByIndex(vi.refType, vi.index)
|
2020-08-21 12:37:46 +00:00
|
|
|
}
|
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// emitLoadByIndex loads the specified variable type with index i.
|
2020-08-21 12:37:46 +00:00
|
|
|
func (c *codegen) emitLoadByIndex(t varType, i int) {
|
2020-05-07 08:54:35 +00:00
|
|
|
base, _ := getBaseOpcode(t)
|
|
|
|
if i < 7 {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, base+opcode.Opcode(i))
|
2020-05-07 08:54:35 +00:00
|
|
|
} else {
|
|
|
|
emit.Instruction(c.prog.BinWriter, base+7, []byte{byte(i)})
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2020-05-07 08:54:35 +00:00
|
|
|
// emitStoreVar stores top value from the evaluation stack in the specified variable.
|
2020-07-28 16:35:41 +00:00
|
|
|
func (c *codegen) emitStoreVar(pkg string, name string) {
|
2020-05-19 15:15:03 +00:00
|
|
|
if name == "_" {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DROP)
|
2020-05-19 15:15:03 +00:00
|
|
|
return
|
|
|
|
}
|
2021-02-05 13:15:26 +00:00
|
|
|
vi := c.getVarIndex(pkg, name)
|
|
|
|
c.emitStoreByIndex(vi.refType, vi.index)
|
2020-08-21 12:37:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// emitLoadByIndex stores top value in the specified variable type with index i.
|
|
|
|
func (c *codegen) emitStoreByIndex(t varType, i int) {
|
2020-05-07 08:54:35 +00:00
|
|
|
_, base := getBaseOpcode(t)
|
|
|
|
if i < 7 {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, base+opcode.Opcode(i))
|
2020-05-07 08:54:35 +00:00
|
|
|
} else {
|
|
|
|
emit.Instruction(c.prog.BinWriter, base+7, []byte{byte(i)})
|
|
|
|
}
|
2018-02-27 09:04:24 +00:00
|
|
|
}
|
|
|
|
|
2020-05-19 13:57:42 +00:00
|
|
|
func (c *codegen) emitDefault(t types.Type) {
|
|
|
|
switch t := t.Underlying().(type) {
|
|
|
|
case *types.Basic:
|
2020-05-07 08:54:35 +00:00
|
|
|
info := t.Info()
|
|
|
|
switch {
|
|
|
|
case info&types.IsInteger != 0:
|
|
|
|
emit.Int(c.prog.BinWriter, 0)
|
|
|
|
case info&types.IsString != 0:
|
|
|
|
emit.Bytes(c.prog.BinWriter, []byte{})
|
|
|
|
case info&types.IsBoolean != 0:
|
|
|
|
emit.Bool(c.prog.BinWriter, false)
|
|
|
|
default:
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSHNULL)
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
2020-05-19 13:57:42 +00:00
|
|
|
case *types.Struct:
|
2020-05-20 08:32:52 +00:00
|
|
|
num := t.NumFields()
|
2021-11-12 13:35:30 +00:00
|
|
|
for i := num - 1; i >= 0; i-- {
|
2020-05-20 08:32:52 +00:00
|
|
|
c.emitDefault(t.Field(i).Type())
|
|
|
|
}
|
2021-11-12 13:35:30 +00:00
|
|
|
emit.Int(c.prog.BinWriter, int64(num))
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PACKSTRUCT)
|
2020-05-19 13:57:42 +00:00
|
|
|
default:
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSHNULL)
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2019-10-22 14:56:03 +00:00
|
|
|
// convertGlobals traverses the AST and only converts global declarations.
|
2022-04-20 18:30:09 +00:00
|
|
|
// If we call this in convertFuncDecl, it will load all global variables
|
2018-02-25 12:26:56 +00:00
|
|
|
// into the scope of the function.
|
2022-08-04 14:47:32 +00:00
|
|
|
func (c *codegen) convertGlobals(f *ast.File) {
|
2018-02-25 12:26:56 +00:00
|
|
|
ast.Inspect(f, func(node ast.Node) bool {
|
|
|
|
switch n := node.(type) {
|
|
|
|
case *ast.FuncDecl:
|
|
|
|
return false
|
|
|
|
case *ast.GenDecl:
|
2020-07-28 16:50:44 +00:00
|
|
|
ast.Walk(c, n)
|
2018-02-25 12:26:56 +00:00
|
|
|
}
|
|
|
|
return true
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2020-08-04 08:06:23 +00:00
|
|
|
func isInitFunc(decl *ast.FuncDecl) bool {
|
|
|
|
return decl.Name.Name == "init" && decl.Recv == nil &&
|
|
|
|
decl.Type.Params.NumFields() == 0 &&
|
|
|
|
decl.Type.Results.NumFields() == 0
|
|
|
|
}
|
|
|
|
|
2021-06-04 08:47:52 +00:00
|
|
|
func (c *codegen) isVerifyFunc(decl *ast.FuncDecl) bool {
|
|
|
|
return decl.Name.Name == "Verify" && decl.Recv == nil &&
|
|
|
|
decl.Type.Results.NumFields() == 1 &&
|
|
|
|
isBool(c.typeOf(decl.Type.Results.List[0].Type))
|
|
|
|
}
|
|
|
|
|
2020-10-06 15:25:40 +00:00
|
|
|
func (c *codegen) clearSlots(n int) {
|
|
|
|
for i := 0; i < n; i++ {
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSHNULL)
|
|
|
|
c.emitStoreByIndex(varLocal, i)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-04-28 13:10:44 +00:00
|
|
|
// convertInitFuncs converts `init()` functions in file f and returns
|
2022-04-20 18:30:09 +00:00
|
|
|
// the number of locals in the last processed definition as well as maximum locals number encountered.
|
2021-04-28 13:10:44 +00:00
|
|
|
func (c *codegen) convertInitFuncs(f *ast.File, pkg *types.Package, lastCount int) (int, int) {
|
|
|
|
maxCount := -1
|
2020-08-04 08:06:23 +00:00
|
|
|
ast.Inspect(f, func(node ast.Node) bool {
|
|
|
|
switch n := node.(type) {
|
|
|
|
case *ast.FuncDecl:
|
|
|
|
if isInitFunc(n) {
|
2021-04-28 13:10:44 +00:00
|
|
|
if lastCount != -1 {
|
|
|
|
c.clearSlots(lastCount)
|
|
|
|
}
|
|
|
|
|
|
|
|
f := c.convertFuncDecl(f, n, pkg)
|
|
|
|
lastCount = f.vars.localsCnt
|
|
|
|
if lastCount > maxCount {
|
|
|
|
maxCount = lastCount
|
2020-10-06 15:25:40 +00:00
|
|
|
}
|
2020-08-04 08:06:23 +00:00
|
|
|
}
|
|
|
|
case *ast.GenDecl:
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
})
|
2021-04-28 13:10:44 +00:00
|
|
|
return lastCount, maxCount
|
2020-08-04 08:06:23 +00:00
|
|
|
}
|
|
|
|
|
2020-10-02 10:16:02 +00:00
|
|
|
func isDeployFunc(decl *ast.FuncDecl) bool {
|
|
|
|
if decl.Name.Name != "_deploy" || decl.Recv != nil ||
|
2021-01-28 13:31:50 +00:00
|
|
|
decl.Type.Params.NumFields() != 2 ||
|
2020-10-02 10:16:02 +00:00
|
|
|
decl.Type.Results.NumFields() != 0 {
|
|
|
|
return false
|
|
|
|
}
|
2021-01-28 13:31:50 +00:00
|
|
|
typ, ok := decl.Type.Params.List[1].Type.(*ast.Ident)
|
2020-10-02 10:16:02 +00:00
|
|
|
return ok && typ.Name == "bool"
|
|
|
|
}
|
|
|
|
|
2021-04-28 13:10:44 +00:00
|
|
|
func (c *codegen) convertDeployFuncs() int {
|
|
|
|
maxCount, lastCount := 0, -1
|
2020-10-02 10:16:02 +00:00
|
|
|
c.ForEachFile(func(f *ast.File, pkg *types.Package) {
|
|
|
|
ast.Inspect(f, func(node ast.Node) bool {
|
|
|
|
switch n := node.(type) {
|
|
|
|
case *ast.FuncDecl:
|
|
|
|
if isDeployFunc(n) {
|
2021-04-28 13:10:44 +00:00
|
|
|
if lastCount != -1 {
|
|
|
|
c.clearSlots(lastCount)
|
|
|
|
}
|
|
|
|
|
|
|
|
f := c.convertFuncDecl(f, n, pkg)
|
|
|
|
lastCount = f.vars.localsCnt
|
|
|
|
if lastCount > maxCount {
|
|
|
|
maxCount = lastCount
|
2020-10-02 10:16:02 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
case *ast.GenDecl:
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
})
|
|
|
|
})
|
2021-04-28 13:10:44 +00:00
|
|
|
return maxCount
|
2020-10-02 10:16:02 +00:00
|
|
|
}
|
|
|
|
|
2021-04-28 13:10:44 +00:00
|
|
|
func (c *codegen) convertFuncDecl(file ast.Node, decl *ast.FuncDecl, pkg *types.Package) *funcScope {
|
2018-02-19 09:24:28 +00:00
|
|
|
var (
|
2020-05-06 15:10:20 +00:00
|
|
|
f *funcScope
|
|
|
|
ok, isLambda bool
|
2018-02-19 09:24:28 +00:00
|
|
|
)
|
2020-08-04 08:06:23 +00:00
|
|
|
isInit := isInitFunc(decl)
|
2020-10-02 10:16:02 +00:00
|
|
|
isDeploy := isDeployFunc(decl)
|
|
|
|
if isInit || isDeploy {
|
2020-08-04 08:06:23 +00:00
|
|
|
f = c.newFuncScope(decl, c.newLabel())
|
2018-02-19 09:24:28 +00:00
|
|
|
} else {
|
2020-08-04 08:06:23 +00:00
|
|
|
f, ok = c.funcs[c.getFuncNameFromDecl("", decl)]
|
|
|
|
if ok {
|
2023-03-18 07:44:12 +00:00
|
|
|
// If this function is a syscall we will not convert it to bytecode.
|
2022-09-16 16:18:47 +00:00
|
|
|
// If it's a potential custom builtin then it needs more specific usages research,
|
|
|
|
// thus let's emit the code for it.
|
2023-03-18 07:44:12 +00:00
|
|
|
if isSyscall(f) {
|
2021-04-28 13:10:44 +00:00
|
|
|
return f
|
2020-08-04 08:06:23 +00:00
|
|
|
}
|
|
|
|
c.setLabel(f.label)
|
|
|
|
} else if f, ok = c.lambda[c.getIdentName("", decl.Name.Name)]; ok {
|
|
|
|
isLambda = ok
|
|
|
|
c.setLabel(f.label)
|
|
|
|
} else {
|
|
|
|
f = c.newFunc(decl)
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2018-02-24 09:06:48 +00:00
|
|
|
|
2020-03-31 12:56:10 +00:00
|
|
|
f.rng.Start = uint16(c.prog.Len())
|
2018-02-25 12:26:56 +00:00
|
|
|
c.scope = f
|
2018-04-10 09:45:31 +00:00
|
|
|
ast.Inspect(decl, c.scope.analyzeVoidCalls) // @OPTIMIZE
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2018-02-25 12:26:56 +00:00
|
|
|
// All globals copied into the scope of the function need to be added
|
|
|
|
// to the stack size of the function.
|
2020-10-02 10:16:02 +00:00
|
|
|
if !isInit && !isDeploy {
|
2020-10-06 15:25:40 +00:00
|
|
|
sizeArg := f.countArgs()
|
|
|
|
if sizeArg > 255 {
|
|
|
|
c.prog.Err = errors.New("maximum of 255 local variables is allowed")
|
|
|
|
}
|
2021-04-28 13:10:44 +00:00
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.INITSLOT, []byte{byte(0), byte(sizeArg)})
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2020-06-29 13:45:27 +00:00
|
|
|
f.vars.newScope()
|
|
|
|
defer f.vars.dropScope()
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
// We need to handle methods, which in Go, is just syntactic sugar.
|
2022-04-20 18:30:09 +00:00
|
|
|
// The method receiver will be passed in as the first argument.
|
|
|
|
// We check if this declaration has a receiver and load it into the scope.
|
2018-02-19 09:24:28 +00:00
|
|
|
//
|
2022-04-20 18:30:09 +00:00
|
|
|
// FIXME: For now, we will hard cast this to a struct. We can later fine tune this
|
2018-02-19 09:24:28 +00:00
|
|
|
// to support other types.
|
|
|
|
if decl.Recv != nil {
|
|
|
|
for _, arg := range decl.Recv.List {
|
compiler: fix nil method receiver handling
An attempt to compile the following code leads to runtime panic:
```
package foo
type CustomInt int
func Main() int {
var i CustomInt
i = 5
return i.Do(2)
}
func (CustomInt) Do(arg int) int {
return arg
}
```
The panic:
```
panic: runtime error: index out of range [0] with length 0 [recovered]
panic: runtime error: index out of range [0] with length 0
goroutine 22 [running]:
testing.tRunner.func1.2({0xa341c0, 0xc0001606d8})
/usr/local/go/src/testing/testing.go:1209 +0x24e
testing.tRunner.func1()
/usr/local/go/src/testing/testing.go:1212 +0x218
panic({0xa341c0, 0xc0001606d8})
/usr/local/go/src/runtime/panic.go:1038 +0x215
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).convertFuncDecl(0xc00015e3c0, {0xc753b8, 0xc000152c80}, 0xc000266300, 0x30)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:497 +0x10b3
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).compile.func2(0xc000152c80, 0xc00023c410)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2153 +0x3f8
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachFile.func1(0xc000229b80)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:102 +0x82
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachPackage(0xc00015e3c0, 0xc000189bb0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:93 +0xc6
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachFile(0x999a20, 0xc000130d80)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:99 +0x45
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).compile(0xc00015e3c0, 0xc0002669f0, 0x1)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2140 +0x445
github.com/nspcc-dev/neo-go/pkg/compiler.codeGen(0xc0002669f0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2191 +0x353
github.com/nspcc-dev/neo-go/pkg/compiler.CompileWithOptions({0xa6f39a, 0x50b6b3}, {0xc6d1a0, 0xc0002421e0}, 0x0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:218 +0x65
github.com/nspcc-dev/neo-go/pkg/compiler_test.vmAndCompileInterop(0x5648df, {0xa9bf23, 0x94})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:75 +0x113
github.com/nspcc-dev/neo-go/pkg/compiler_test.eval(0xc0002421c0, {0xa9bf23, 0x61be8c7}, {0xa68880, 0xc0002421c0})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:36 +0x2d
github.com/nspcc-dev/neo-go/pkg/compiler_test.TestUnnamedMethodReceiver(0x4079f9)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/function_call_test.go:400 +0x4f
testing.tRunner(0xc000204b60, 0xbcebb0)
/usr/local/go/src/testing/testing.go:1259 +0x102
created by testing.(*T).Run
/usr/local/go/src/testing/testing.go:1306 +0x35a
```
The solution is to use the same approach as for unnamed function
parameters handling introduced in #2204. (c *funcScope).newVariable is
able to properly handle "_" receiver.
2022-08-17 14:41:11 +00:00
|
|
|
// Use underscore instead of unnamed receiver name, e.g.:
|
|
|
|
// func (MyCustomStruct) DoSmth(arg1 int) {...}
|
|
|
|
// Unnamed receiver will never be referenced, thus we can use the same approach as for multiple unnamed parameters handling, see #2204.
|
|
|
|
recvName := "_"
|
|
|
|
if len(arg.Names) != 0 {
|
|
|
|
recvName = arg.Names[0].Name
|
|
|
|
}
|
2020-05-07 08:54:35 +00:00
|
|
|
// only create an argument here, it will be stored via INITSLOT
|
compiler: fix nil method receiver handling
An attempt to compile the following code leads to runtime panic:
```
package foo
type CustomInt int
func Main() int {
var i CustomInt
i = 5
return i.Do(2)
}
func (CustomInt) Do(arg int) int {
return arg
}
```
The panic:
```
panic: runtime error: index out of range [0] with length 0 [recovered]
panic: runtime error: index out of range [0] with length 0
goroutine 22 [running]:
testing.tRunner.func1.2({0xa341c0, 0xc0001606d8})
/usr/local/go/src/testing/testing.go:1209 +0x24e
testing.tRunner.func1()
/usr/local/go/src/testing/testing.go:1212 +0x218
panic({0xa341c0, 0xc0001606d8})
/usr/local/go/src/runtime/panic.go:1038 +0x215
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).convertFuncDecl(0xc00015e3c0, {0xc753b8, 0xc000152c80}, 0xc000266300, 0x30)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:497 +0x10b3
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).compile.func2(0xc000152c80, 0xc00023c410)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2153 +0x3f8
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachFile.func1(0xc000229b80)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:102 +0x82
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachPackage(0xc00015e3c0, 0xc000189bb0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:93 +0xc6
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachFile(0x999a20, 0xc000130d80)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:99 +0x45
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).compile(0xc00015e3c0, 0xc0002669f0, 0x1)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2140 +0x445
github.com/nspcc-dev/neo-go/pkg/compiler.codeGen(0xc0002669f0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2191 +0x353
github.com/nspcc-dev/neo-go/pkg/compiler.CompileWithOptions({0xa6f39a, 0x50b6b3}, {0xc6d1a0, 0xc0002421e0}, 0x0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:218 +0x65
github.com/nspcc-dev/neo-go/pkg/compiler_test.vmAndCompileInterop(0x5648df, {0xa9bf23, 0x94})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:75 +0x113
github.com/nspcc-dev/neo-go/pkg/compiler_test.eval(0xc0002421c0, {0xa9bf23, 0x61be8c7}, {0xa68880, 0xc0002421c0})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:36 +0x2d
github.com/nspcc-dev/neo-go/pkg/compiler_test.TestUnnamedMethodReceiver(0x4079f9)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/function_call_test.go:400 +0x4f
testing.tRunner(0xc000204b60, 0xbcebb0)
/usr/local/go/src/testing/testing.go:1259 +0x102
created by testing.(*T).Run
/usr/local/go/src/testing/testing.go:1306 +0x35a
```
The solution is to use the same approach as for unnamed function
parameters handling introduced in #2204. (c *funcScope).newVariable is
able to properly handle "_" receiver.
2022-08-17 14:41:11 +00:00
|
|
|
c.scope.newVariable(varArgument, recvName)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Load the arguments in scope.
|
|
|
|
for _, arg := range decl.Type.Params.List {
|
2020-05-06 15:20:12 +00:00
|
|
|
for _, id := range arg.Names {
|
2020-05-07 08:54:35 +00:00
|
|
|
// only create an argument here, it will be stored via INITSLOT
|
|
|
|
c.scope.newVariable(varArgument, id.Name)
|
2020-05-06 15:20:12 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2018-04-10 09:45:31 +00:00
|
|
|
|
|
|
|
ast.Walk(c, decl.Body)
|
|
|
|
|
2020-05-06 12:39:25 +00:00
|
|
|
// If we have reached the end of the function without encountering `return` statement,
|
|
|
|
// we should clean alt.stack manually.
|
|
|
|
// This can be the case with void and named-return functions.
|
2021-02-05 11:31:45 +00:00
|
|
|
if !isInit && !isDeploy && !lastStmtIsReturn(decl.Body) {
|
2021-11-12 16:30:50 +00:00
|
|
|
c.processDefers()
|
2020-03-31 13:57:35 +00:00
|
|
|
c.saveSequencePoint(decl.Body)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.RET)
|
2018-02-25 12:26:56 +00:00
|
|
|
}
|
2020-03-31 12:56:10 +00:00
|
|
|
|
2021-05-12 12:08:22 +00:00
|
|
|
if isInit {
|
|
|
|
c.initVariables = append(c.initVariables, f.variables...)
|
2021-05-12 13:03:47 +00:00
|
|
|
} else if isDeploy {
|
|
|
|
c.deployVariables = append(c.deployVariables, f.variables...)
|
2021-05-12 12:08:22 +00:00
|
|
|
}
|
|
|
|
|
2020-03-31 12:56:10 +00:00
|
|
|
f.rng.End = uint16(c.prog.Len() - 1)
|
2020-05-06 15:10:20 +00:00
|
|
|
|
|
|
|
if !isLambda {
|
|
|
|
for _, f := range c.lambda {
|
2021-11-30 15:02:38 +00:00
|
|
|
if _, ok := c.lambda[c.getIdentName("", f.decl.Name.Name)]; !ok {
|
|
|
|
panic("ICE: lambda name doesn't match map key")
|
|
|
|
}
|
2020-07-07 10:26:44 +00:00
|
|
|
c.convertFuncDecl(file, f.decl, pkg)
|
2020-05-06 15:10:20 +00:00
|
|
|
}
|
|
|
|
c.lambda = make(map[string]*funcScope)
|
|
|
|
}
|
2021-04-28 13:10:44 +00:00
|
|
|
|
|
|
|
if !isInit && !isDeploy {
|
|
|
|
c.reverseOffsetMap[int(f.rng.Start)] = nameWithLocals{
|
|
|
|
name: f.name,
|
|
|
|
count: f.vars.localsCnt,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return f
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (c *codegen) Visit(node ast.Node) ast.Visitor {
|
2019-11-22 10:06:32 +00:00
|
|
|
if c.prog.Err != nil {
|
|
|
|
return nil
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
switch n := node.(type) {
|
2018-02-24 09:06:48 +00:00
|
|
|
// General declarations.
|
2018-02-25 12:26:56 +00:00
|
|
|
// var (
|
|
|
|
// x = 2
|
|
|
|
// )
|
2018-02-24 09:06:48 +00:00
|
|
|
case *ast.GenDecl:
|
2020-08-10 14:43:26 +00:00
|
|
|
if n.Tok == token.VAR || n.Tok == token.CONST {
|
|
|
|
c.saveSequencePoint(n)
|
|
|
|
}
|
2020-07-28 16:50:44 +00:00
|
|
|
if n.Tok == token.CONST {
|
|
|
|
for _, spec := range n.Specs {
|
|
|
|
vs := spec.(*ast.ValueSpec)
|
|
|
|
for i := range vs.Names {
|
2021-12-02 14:44:53 +00:00
|
|
|
obj := c.currPkg.Types.Scope().Lookup(vs.Names[i].Name)
|
2022-01-18 08:46:48 +00:00
|
|
|
if obj != nil { // can be nil if unused
|
|
|
|
c.constMap[c.getIdentName("", vs.Names[i].Name)] = types.TypeAndValue{
|
|
|
|
Type: obj.Type(),
|
|
|
|
Value: obj.(*types.Const).Val(),
|
|
|
|
}
|
2020-08-25 07:22:58 +00:00
|
|
|
}
|
2020-07-28 16:50:44 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
2018-02-25 12:26:56 +00:00
|
|
|
for _, spec := range n.Specs {
|
|
|
|
switch t := spec.(type) {
|
|
|
|
case *ast.ValueSpec:
|
2022-09-28 06:58:59 +00:00
|
|
|
// Filter out type assertion with two return values: var i, ok = v.(int)
|
|
|
|
if len(t.Names) == 2 && len(t.Values) == 1 && n.Tok == token.VAR {
|
|
|
|
err := checkTypeAssertWithOK(t.Values[0])
|
|
|
|
if err != nil {
|
|
|
|
c.prog.Err = err
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
compiler: allow multi-return variables declaration
Problem: an attempt to compile the following code leads to a runtime
panic:
```
package foo
var a, b = f()
func Main() int {
return a + b
}
func f() (int, int) {
return 1, 2
}
```
```
panic: runtime error: index out of range [1] with length 1 [recovered]
panic: runtime error: index out of range [1] with length 1
goroutine 22 [running]:
testing.tRunner.func1.2({0xa341c0, 0xc0001647f8})
/usr/local/go/src/testing/testing.go:1209 +0x24e
testing.tRunner.func1()
/usr/local/go/src/testing/testing.go:1212 +0x218
panic({0xa341c0, 0xc0001647f8})
/usr/local/go/src/runtime/panic.go:1038 +0x215
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).Visit(0xc0001623c0, {0xc75520, 0xc000155d80})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:591 +0x6559
go/ast.Walk({0xc6c4e0, 0xc0001623c0}, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:50 +0x5f
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).convertGlobals.func1({0xc75520, 0xc000155d80})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:359 +0x70
go/ast.inspector.Visit(0xc000229740, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:375 +0x31
go/ast.Walk({0xc6d920, 0xc000229740}, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:50 +0x5f
go/ast.walkDeclList({0xc6d920, 0xc000229740}, {0xc000155e80, 0x3, 0x120})
/usr/local/go/src/go/ast/walk.go:36 +0x87
go/ast.Walk({0xc6d920, 0xc000229740}, {0xc75458, 0xc000156c80})
/usr/local/go/src/go/ast/walk.go:355 +0x15c5
go/ast.Inspect(...)
/usr/local/go/src/go/ast/walk.go:387
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).convertGlobals(0xc0001623c0, 0xc000156c80, 0xc000254280)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:354 +0x71
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).traverseGlobals.func2(0xc000254280)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/analysis.go:86 +0x16e
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachPackage(0xc0001623c0, 0xc000191b98)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:93 +0xc6
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).traverseGlobals(0xc0001623c0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/analysis.go:82 +0x22c
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).compile(0xc0001623c0, 0xc000274d20, 0x1)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2118 +0x17c
github.com/nspcc-dev/neo-go/pkg/compiler.codeGen(0xc000274d20)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2191 +0x353
github.com/nspcc-dev/neo-go/pkg/compiler.CompileWithOptions({0xa6f39a, 0xc00023cee0}, {0xc6d240, 0xc00024a460}, 0x0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:218 +0x65
github.com/nspcc-dev/neo-go/pkg/compiler_test.vmAndCompileInterop(0x5648df, {0xa9989f, 0x7d})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:75 +0x113
github.com/nspcc-dev/neo-go/pkg/compiler_test.eval(0xc00024a440, {0xa9989f, 0x129f366e}, {0xa68880, 0xc00024a440})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:36 +0x2d
github.com/nspcc-dev/neo-go/pkg/compiler_test.TestGenDeclWithMultiRet.func2(0x4079f9)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/global_test.go:36 +0x4f
testing.tRunner(0xc00022e9c0, 0xbce760)
/usr/local/go/src/testing/testing.go:1259 +0x102
created by testing.(*T).Run
/usr/local/go/src/testing/testing.go:1306 +0x35a
```
Solution:
Allow using multi-return function calls as general variable declaration
value. It was supported for assignment statements, so do the same for
*ast.GenDecl if it's a variable under the hood.
2022-08-16 12:25:53 +00:00
|
|
|
multiRet := n.Tok == token.VAR && len(t.Values) != 0 && len(t.Names) != len(t.Values)
|
2020-04-02 13:36:11 +00:00
|
|
|
for _, id := range t.Names {
|
2020-09-06 12:26:03 +00:00
|
|
|
if id.Name != "_" {
|
|
|
|
if c.scope == nil {
|
|
|
|
// it is a global declaration
|
|
|
|
c.newGlobal("", id.Name)
|
|
|
|
} else {
|
|
|
|
c.scope.newLocal(id.Name)
|
|
|
|
}
|
compiler: allow multi-return variables declaration
Problem: an attempt to compile the following code leads to a runtime
panic:
```
package foo
var a, b = f()
func Main() int {
return a + b
}
func f() (int, int) {
return 1, 2
}
```
```
panic: runtime error: index out of range [1] with length 1 [recovered]
panic: runtime error: index out of range [1] with length 1
goroutine 22 [running]:
testing.tRunner.func1.2({0xa341c0, 0xc0001647f8})
/usr/local/go/src/testing/testing.go:1209 +0x24e
testing.tRunner.func1()
/usr/local/go/src/testing/testing.go:1212 +0x218
panic({0xa341c0, 0xc0001647f8})
/usr/local/go/src/runtime/panic.go:1038 +0x215
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).Visit(0xc0001623c0, {0xc75520, 0xc000155d80})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:591 +0x6559
go/ast.Walk({0xc6c4e0, 0xc0001623c0}, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:50 +0x5f
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).convertGlobals.func1({0xc75520, 0xc000155d80})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:359 +0x70
go/ast.inspector.Visit(0xc000229740, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:375 +0x31
go/ast.Walk({0xc6d920, 0xc000229740}, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:50 +0x5f
go/ast.walkDeclList({0xc6d920, 0xc000229740}, {0xc000155e80, 0x3, 0x120})
/usr/local/go/src/go/ast/walk.go:36 +0x87
go/ast.Walk({0xc6d920, 0xc000229740}, {0xc75458, 0xc000156c80})
/usr/local/go/src/go/ast/walk.go:355 +0x15c5
go/ast.Inspect(...)
/usr/local/go/src/go/ast/walk.go:387
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).convertGlobals(0xc0001623c0, 0xc000156c80, 0xc000254280)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:354 +0x71
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).traverseGlobals.func2(0xc000254280)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/analysis.go:86 +0x16e
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachPackage(0xc0001623c0, 0xc000191b98)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:93 +0xc6
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).traverseGlobals(0xc0001623c0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/analysis.go:82 +0x22c
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).compile(0xc0001623c0, 0xc000274d20, 0x1)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2118 +0x17c
github.com/nspcc-dev/neo-go/pkg/compiler.codeGen(0xc000274d20)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2191 +0x353
github.com/nspcc-dev/neo-go/pkg/compiler.CompileWithOptions({0xa6f39a, 0xc00023cee0}, {0xc6d240, 0xc00024a460}, 0x0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:218 +0x65
github.com/nspcc-dev/neo-go/pkg/compiler_test.vmAndCompileInterop(0x5648df, {0xa9989f, 0x7d})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:75 +0x113
github.com/nspcc-dev/neo-go/pkg/compiler_test.eval(0xc00024a440, {0xa9989f, 0x129f366e}, {0xa68880, 0xc00024a440})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:36 +0x2d
github.com/nspcc-dev/neo-go/pkg/compiler_test.TestGenDeclWithMultiRet.func2(0x4079f9)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/global_test.go:36 +0x4f
testing.tRunner(0xc00022e9c0, 0xbce760)
/usr/local/go/src/testing/testing.go:1259 +0x102
created by testing.(*T).Run
/usr/local/go/src/testing/testing.go:1306 +0x35a
```
Solution:
Allow using multi-return function calls as general variable declaration
value. It was supported for assignment statements, so do the same for
*ast.GenDecl if it's a variable under the hood.
2022-08-16 12:25:53 +00:00
|
|
|
if !multiRet {
|
|
|
|
c.registerDebugVariable(id.Name, t.Type)
|
|
|
|
}
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
2020-04-02 13:36:11 +00:00
|
|
|
}
|
2022-08-19 11:54:42 +00:00
|
|
|
for i, id := range t.Names {
|
|
|
|
if id.Name != "_" {
|
|
|
|
if len(t.Values) != 0 {
|
|
|
|
if i == 0 || !multiRet {
|
|
|
|
ast.Walk(c, t.Values[i])
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
c.emitDefault(c.typeOf(t.Type))
|
compiler: allow multi-return variables declaration
Problem: an attempt to compile the following code leads to a runtime
panic:
```
package foo
var a, b = f()
func Main() int {
return a + b
}
func f() (int, int) {
return 1, 2
}
```
```
panic: runtime error: index out of range [1] with length 1 [recovered]
panic: runtime error: index out of range [1] with length 1
goroutine 22 [running]:
testing.tRunner.func1.2({0xa341c0, 0xc0001647f8})
/usr/local/go/src/testing/testing.go:1209 +0x24e
testing.tRunner.func1()
/usr/local/go/src/testing/testing.go:1212 +0x218
panic({0xa341c0, 0xc0001647f8})
/usr/local/go/src/runtime/panic.go:1038 +0x215
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).Visit(0xc0001623c0, {0xc75520, 0xc000155d80})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:591 +0x6559
go/ast.Walk({0xc6c4e0, 0xc0001623c0}, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:50 +0x5f
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).convertGlobals.func1({0xc75520, 0xc000155d80})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:359 +0x70
go/ast.inspector.Visit(0xc000229740, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:375 +0x31
go/ast.Walk({0xc6d920, 0xc000229740}, {0xc75520, 0xc000155d80})
/usr/local/go/src/go/ast/walk.go:50 +0x5f
go/ast.walkDeclList({0xc6d920, 0xc000229740}, {0xc000155e80, 0x3, 0x120})
/usr/local/go/src/go/ast/walk.go:36 +0x87
go/ast.Walk({0xc6d920, 0xc000229740}, {0xc75458, 0xc000156c80})
/usr/local/go/src/go/ast/walk.go:355 +0x15c5
go/ast.Inspect(...)
/usr/local/go/src/go/ast/walk.go:387
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).convertGlobals(0xc0001623c0, 0xc000156c80, 0xc000254280)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:354 +0x71
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).traverseGlobals.func2(0xc000254280)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/analysis.go:86 +0x16e
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).ForEachPackage(0xc0001623c0, 0xc000191b98)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:93 +0xc6
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).traverseGlobals(0xc0001623c0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/analysis.go:82 +0x22c
github.com/nspcc-dev/neo-go/pkg/compiler.(*codegen).compile(0xc0001623c0, 0xc000274d20, 0x1)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2118 +0x17c
github.com/nspcc-dev/neo-go/pkg/compiler.codeGen(0xc000274d20)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/codegen.go:2191 +0x353
github.com/nspcc-dev/neo-go/pkg/compiler.CompileWithOptions({0xa6f39a, 0xc00023cee0}, {0xc6d240, 0xc00024a460}, 0x0)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/compiler.go:218 +0x65
github.com/nspcc-dev/neo-go/pkg/compiler_test.vmAndCompileInterop(0x5648df, {0xa9989f, 0x7d})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:75 +0x113
github.com/nspcc-dev/neo-go/pkg/compiler_test.eval(0xc00024a440, {0xa9989f, 0x129f366e}, {0xa68880, 0xc00024a440})
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/vm_test.go:36 +0x2d
github.com/nspcc-dev/neo-go/pkg/compiler_test.TestGenDeclWithMultiRet.func2(0x4079f9)
/home/anna/Documents/GitProjects/nspcc-dev/neo-go/pkg/compiler/global_test.go:36 +0x4f
testing.tRunner(0xc00022e9c0, 0xbce760)
/usr/local/go/src/testing/testing.go:1259 +0x102
created by testing.(*T).Run
/usr/local/go/src/testing/testing.go:1306 +0x35a
```
Solution:
Allow using multi-return function calls as general variable declaration
value. It was supported for assignment statements, so do the same for
*ast.GenDecl if it's a variable under the hood.
2022-08-16 12:25:53 +00:00
|
|
|
}
|
2022-08-19 11:54:42 +00:00
|
|
|
c.emitStoreVar("", t.Names[i].Name)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
// If var decl contains call then the code should be emitted for it, otherwise - do not evaluate.
|
|
|
|
if len(t.Values) == 0 {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
var hasCall bool
|
|
|
|
if i == 0 || !multiRet {
|
|
|
|
hasCall = containsCall(t.Values[i])
|
|
|
|
}
|
|
|
|
if hasCall {
|
|
|
|
ast.Walk(c, t.Values[i])
|
|
|
|
}
|
|
|
|
if hasCall || i != 0 && multiRet {
|
|
|
|
c.emitStoreVar("", "_") // drop unused after walk
|
2020-05-07 08:54:35 +00:00
|
|
|
}
|
2018-02-25 12:26:56 +00:00
|
|
|
}
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
case *ast.AssignStmt:
|
2022-09-28 06:58:59 +00:00
|
|
|
// Filter out type assertion with two return values: i, ok = v.(int)
|
|
|
|
if len(n.Lhs) == 2 && len(n.Rhs) == 1 && (n.Tok == token.DEFINE || n.Tok == token.ASSIGN) {
|
|
|
|
err := checkTypeAssertWithOK(n.Rhs[0])
|
|
|
|
if err != nil {
|
|
|
|
c.prog.Err = err
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
2019-12-24 13:46:43 +00:00
|
|
|
multiRet := len(n.Rhs) != len(n.Lhs)
|
2020-03-31 13:57:35 +00:00
|
|
|
c.saveSequencePoint(n)
|
2020-05-18 08:45:20 +00:00
|
|
|
// Assign operations are grouped https://github.com/golang/go/blob/master/src/go/types/stmt.go#L160
|
|
|
|
isAssignOp := token.ADD_ASSIGN <= n.Tok && n.Tok <= token.AND_NOT_ASSIGN
|
|
|
|
if isAssignOp {
|
|
|
|
// RHS can contain exactly one expression, thus there is no need to iterate.
|
|
|
|
ast.Walk(c, n.Lhs[0])
|
|
|
|
ast.Walk(c, n.Rhs[0])
|
2020-08-23 09:39:58 +00:00
|
|
|
c.emitToken(n.Tok, c.typeOf(n.Rhs[0]))
|
2020-05-18 08:45:20 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
for i := 0; i < len(n.Lhs); i++ {
|
|
|
|
switch t := n.Lhs[i].(type) {
|
|
|
|
case *ast.Ident:
|
2020-05-18 08:45:20 +00:00
|
|
|
if n.Tok == token.DEFINE {
|
2020-04-02 13:36:11 +00:00
|
|
|
if !multiRet {
|
|
|
|
c.registerDebugVariable(t.Name, n.Rhs[i])
|
|
|
|
}
|
2020-05-08 13:10:33 +00:00
|
|
|
if t.Name != "_" {
|
|
|
|
c.scope.newLocal(t.Name)
|
|
|
|
}
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
2020-05-18 08:45:20 +00:00
|
|
|
if !isAssignOp && (i == 0 || !multiRet) {
|
|
|
|
ast.Walk(c, n.Rhs[i])
|
|
|
|
}
|
2020-07-28 16:35:41 +00:00
|
|
|
c.emitStoreVar("", t.Name)
|
2018-02-19 09:24:28 +00:00
|
|
|
|
|
|
|
case *ast.SelectorExpr:
|
2020-05-20 08:08:59 +00:00
|
|
|
if !isAssignOp {
|
|
|
|
ast.Walk(c, n.Rhs[i])
|
|
|
|
}
|
2020-07-28 16:35:41 +00:00
|
|
|
typ := c.typeOf(t.X)
|
2021-12-02 14:44:53 +00:00
|
|
|
if c.isInvalidType(typ) {
|
2020-07-28 16:35:41 +00:00
|
|
|
// Store to other package global variable.
|
|
|
|
c.emitStoreVar(t.X.(*ast.Ident).Name, t.Sel.Name)
|
|
|
|
return nil
|
|
|
|
}
|
2020-08-02 11:42:29 +00:00
|
|
|
strct, ok := c.getStruct(typ)
|
2020-05-20 08:08:59 +00:00
|
|
|
if !ok {
|
2019-11-22 14:16:52 +00:00
|
|
|
c.prog.Err = fmt.Errorf("nested selector assigns not supported yet")
|
|
|
|
return nil
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2020-05-20 08:08:59 +00:00
|
|
|
ast.Walk(c, t.X) // load the struct
|
|
|
|
i := indexOfStruct(strct, t.Sel.Name) // get the index of the field
|
|
|
|
c.emitStoreStructField(i) // store the field
|
2018-10-23 08:23:03 +00:00
|
|
|
|
|
|
|
// Assignments to index expressions.
|
|
|
|
// slice[0] = 10
|
|
|
|
case *ast.IndexExpr:
|
2020-05-18 08:45:20 +00:00
|
|
|
if !isAssignOp {
|
|
|
|
ast.Walk(c, n.Rhs[i])
|
|
|
|
}
|
2020-05-20 14:53:01 +00:00
|
|
|
ast.Walk(c, t.X)
|
2020-05-19 12:35:57 +00:00
|
|
|
ast.Walk(c, t.Index)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.ROT, opcode.SETITEM)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
|
2020-02-10 15:14:34 +00:00
|
|
|
case *ast.SliceExpr:
|
2021-11-30 09:55:51 +00:00
|
|
|
if isCompoundSlice(c.typeOf(n.X).Underlying()) {
|
2020-09-15 06:52:56 +00:00
|
|
|
c.prog.Err = errors.New("subslices are supported only for []byte")
|
|
|
|
return nil
|
|
|
|
}
|
2021-11-30 09:55:51 +00:00
|
|
|
|
|
|
|
ast.Walk(c, n.X)
|
2020-02-10 15:14:34 +00:00
|
|
|
|
|
|
|
if n.Low != nil {
|
|
|
|
ast.Walk(c, n.Low)
|
|
|
|
} else {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSH0)
|
2020-02-10 15:14:34 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if n.High != nil {
|
|
|
|
ast.Walk(c, n.High)
|
|
|
|
} else {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.OVER, opcode.SIZE)
|
2020-02-10 15:14:34 +00:00
|
|
|
}
|
|
|
|
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.OVER, opcode.SUB, opcode.SUBSTR)
|
2020-02-10 15:14:34 +00:00
|
|
|
|
|
|
|
return nil
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
case *ast.ReturnStmt:
|
2018-04-02 15:04:42 +00:00
|
|
|
l := c.newLabel()
|
|
|
|
c.setLabel(l)
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2020-03-06 12:11:14 +00:00
|
|
|
cnt := 0
|
2021-02-26 14:58:08 +00:00
|
|
|
start := 0
|
2022-07-11 08:38:18 +00:00
|
|
|
if len(c.inlineContext) > 0 {
|
|
|
|
start = c.inlineContext[len(c.inlineContext)-1].labelOffset
|
2021-02-26 14:58:08 +00:00
|
|
|
}
|
|
|
|
for i := start; i < len(c.labelList); i++ {
|
2020-03-06 12:11:14 +00:00
|
|
|
cnt += c.labelList[i].sz
|
|
|
|
}
|
|
|
|
c.dropItems(cnt)
|
|
|
|
|
2020-05-06 14:24:32 +00:00
|
|
|
if len(n.Results) == 0 {
|
|
|
|
results := c.scope.decl.Type.Results
|
|
|
|
if results.NumFields() != 0 {
|
|
|
|
// function with named returns
|
|
|
|
for i := len(results.List) - 1; i >= 0; i-- {
|
|
|
|
names := results.List[i].Names
|
|
|
|
for j := len(names) - 1; j >= 0; j-- {
|
2020-07-28 16:35:41 +00:00
|
|
|
c.emitLoadVar("", names[j].Name)
|
2020-05-06 14:24:32 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// first result should be on top of the stack
|
|
|
|
for i := len(n.Results) - 1; i >= 0; i-- {
|
|
|
|
ast.Walk(c, n.Results[i])
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2020-08-20 05:34:14 +00:00
|
|
|
c.processDefers()
|
|
|
|
|
2020-03-31 13:57:35 +00:00
|
|
|
c.saveSequencePoint(n)
|
2021-02-04 12:41:00 +00:00
|
|
|
if len(c.pkgInfoInline) == 0 {
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.RET)
|
2022-07-11 08:38:18 +00:00
|
|
|
} else {
|
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, c.inlineContext[len(c.inlineContext)-1].returnLabel)
|
2021-02-04 12:41:00 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
return nil
|
|
|
|
|
|
|
|
case *ast.IfStmt:
|
2020-06-29 13:57:38 +00:00
|
|
|
c.scope.vars.newScope()
|
|
|
|
defer c.scope.vars.dropScope()
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
lIf := c.newLabel()
|
|
|
|
lElse := c.newLabel()
|
2018-10-20 05:11:00 +00:00
|
|
|
lElseEnd := c.newLabel()
|
|
|
|
|
2020-08-21 08:41:10 +00:00
|
|
|
if n.Init != nil {
|
|
|
|
ast.Walk(c, n.Init)
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
if n.Cond != nil {
|
2020-08-23 11:52:37 +00:00
|
|
|
c.emitBoolExpr(n.Cond, true, false, lElse)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
c.setLabel(lIf)
|
|
|
|
ast.Walk(c, n.Body)
|
|
|
|
if n.Else != nil {
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, lElseEnd)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2018-10-20 05:11:00 +00:00
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
c.setLabel(lElse)
|
|
|
|
if n.Else != nil {
|
|
|
|
ast.Walk(c, n.Else)
|
|
|
|
}
|
2018-10-20 05:11:00 +00:00
|
|
|
c.setLabel(lElseEnd)
|
2018-02-19 09:24:28 +00:00
|
|
|
return nil
|
|
|
|
|
2020-01-28 12:47:56 +00:00
|
|
|
case *ast.SwitchStmt:
|
2020-08-24 15:55:30 +00:00
|
|
|
eqOpcode := opcode.EQUAL
|
|
|
|
if n.Tag != nil {
|
|
|
|
ast.Walk(c, n.Tag)
|
|
|
|
eqOpcode, _ = convertToken(token.EQL, c.typeOf(n.Tag))
|
|
|
|
} else {
|
|
|
|
emit.Bool(c.prog.BinWriter, true)
|
|
|
|
}
|
2020-02-18 14:07:38 +00:00
|
|
|
switchEnd, label := c.generateLabel(labelEnd)
|
|
|
|
|
|
|
|
lastSwitch := c.currentSwitch
|
|
|
|
c.currentSwitch = label
|
2020-03-06 12:11:14 +00:00
|
|
|
c.pushStackLabel(label, 1)
|
2020-01-28 12:47:56 +00:00
|
|
|
|
2020-03-10 09:34:05 +00:00
|
|
|
startLabels := make([]uint16, len(n.Body.List))
|
|
|
|
for i := range startLabels {
|
|
|
|
startLabels[i] = c.newLabel()
|
|
|
|
}
|
2020-01-28 12:47:56 +00:00
|
|
|
for i := range n.Body.List {
|
|
|
|
lEnd := c.newLabel()
|
2020-03-10 09:34:05 +00:00
|
|
|
lStart := startLabels[i]
|
2020-01-28 12:47:56 +00:00
|
|
|
cc := n.Body.List[i].(*ast.CaseClause)
|
|
|
|
|
|
|
|
if l := len(cc.List); l != 0 { // if not `default`
|
|
|
|
for j := range cc.List {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
|
2020-01-28 12:47:56 +00:00
|
|
|
ast.Walk(c, cc.List[j])
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, eqOpcode)
|
2020-01-28 12:47:56 +00:00
|
|
|
if j == l-1 {
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPIFNOTL, lEnd)
|
2020-01-28 12:47:56 +00:00
|
|
|
} else {
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPIFL, lStart)
|
2020-01-28 12:47:56 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-06-29 13:57:38 +00:00
|
|
|
c.scope.vars.newScope()
|
|
|
|
|
2020-01-28 12:47:56 +00:00
|
|
|
c.setLabel(lStart)
|
2020-03-10 09:34:05 +00:00
|
|
|
last := len(cc.Body) - 1
|
|
|
|
for j, stmt := range cc.Body {
|
|
|
|
if j == last && isFallthroughStmt(stmt) {
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, startLabels[i+1])
|
2020-03-10 09:34:05 +00:00
|
|
|
break
|
|
|
|
}
|
2020-01-28 12:47:56 +00:00
|
|
|
ast.Walk(c, stmt)
|
|
|
|
}
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, switchEnd)
|
2020-01-28 12:47:56 +00:00
|
|
|
c.setLabel(lEnd)
|
2020-06-29 13:57:38 +00:00
|
|
|
|
|
|
|
c.scope.vars.dropScope()
|
2020-01-28 12:47:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
c.setLabel(switchEnd)
|
2020-03-06 12:11:14 +00:00
|
|
|
c.dropStackLabel()
|
2020-01-28 12:47:56 +00:00
|
|
|
|
2020-02-18 14:07:38 +00:00
|
|
|
c.currentSwitch = lastSwitch
|
|
|
|
|
2020-01-28 12:47:56 +00:00
|
|
|
return nil
|
|
|
|
|
2020-05-06 15:10:20 +00:00
|
|
|
case *ast.FuncLit:
|
2022-02-02 13:51:10 +00:00
|
|
|
var found bool
|
|
|
|
var l uint16
|
|
|
|
for _, fs := range c.lambda {
|
|
|
|
if fs.decl.Body == n.Body {
|
|
|
|
found = true
|
|
|
|
l = fs.label
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if !found {
|
|
|
|
l = c.newLabel()
|
|
|
|
c.newLambda(l, n)
|
|
|
|
}
|
|
|
|
|
2020-05-06 15:10:20 +00:00
|
|
|
buf := make([]byte, 4)
|
|
|
|
binary.LittleEndian.PutUint16(buf, l)
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.PUSHA, buf)
|
|
|
|
return nil
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
case *ast.BasicLit:
|
2020-05-13 15:09:55 +00:00
|
|
|
c.emitLoadConst(c.typeAndValueOf(n))
|
2018-02-19 09:24:28 +00:00
|
|
|
return nil
|
|
|
|
|
2020-08-02 11:56:47 +00:00
|
|
|
case *ast.StarExpr:
|
|
|
|
_, ok := c.getStruct(c.typeOf(n.X))
|
|
|
|
if !ok {
|
|
|
|
c.prog.Err = errors.New("dereferencing is only supported on structs")
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
ast.Walk(c, n.X)
|
|
|
|
c.emitConvert(stackitem.StructT)
|
|
|
|
return nil
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
case *ast.Ident:
|
2020-05-19 12:20:10 +00:00
|
|
|
if tv := c.typeAndValueOf(n); tv.Value != nil {
|
2020-01-27 10:37:33 +00:00
|
|
|
c.emitLoadConst(tv)
|
2020-06-17 08:21:37 +00:00
|
|
|
} else if n.Name == "nil" {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSHNULL)
|
2018-02-19 09:24:28 +00:00
|
|
|
} else {
|
2020-07-28 16:35:41 +00:00
|
|
|
c.emitLoadVar("", n.Name)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
|
|
|
|
case *ast.CompositeLit:
|
2020-12-10 11:08:42 +00:00
|
|
|
t := c.typeOf(n)
|
|
|
|
switch typ := t.Underlying().(type) {
|
2020-06-24 16:25:08 +00:00
|
|
|
case *types.Struct:
|
2020-08-02 11:42:29 +00:00
|
|
|
c.convertStruct(n, false)
|
2020-06-24 16:25:08 +00:00
|
|
|
case *types.Map:
|
|
|
|
c.convertMap(n)
|
2018-02-19 09:24:28 +00:00
|
|
|
default:
|
2020-12-10 11:08:42 +00:00
|
|
|
if tn, ok := t.(*types.Named); ok && isInteropPath(tn.String()) {
|
2022-12-01 18:05:54 +00:00
|
|
|
st, _, _, _ := scAndVMInteropTypeFromExpr(tn, false)
|
2020-12-10 11:08:42 +00:00
|
|
|
expectedLen := -1
|
|
|
|
switch st {
|
|
|
|
case smartcontract.Hash160Type:
|
|
|
|
expectedLen = 20
|
|
|
|
case smartcontract.Hash256Type:
|
|
|
|
expectedLen = 32
|
|
|
|
}
|
|
|
|
if expectedLen != -1 && expectedLen != len(n.Elts) {
|
|
|
|
c.prog.Err = fmt.Errorf("%s type must have size %d", tn.Obj().Name(), expectedLen)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
ln := len(n.Elts)
|
2019-10-22 14:56:03 +00:00
|
|
|
// ByteArrays needs a different approach than normal arrays.
|
2020-05-13 15:09:55 +00:00
|
|
|
if isByteSlice(typ) {
|
2021-02-16 13:20:19 +00:00
|
|
|
c.convertByteArray(n.Elts)
|
2018-02-27 09:04:24 +00:00
|
|
|
return nil
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
for i := ln - 1; i >= 0; i-- {
|
2020-01-23 12:42:50 +00:00
|
|
|
ast.Walk(c, n.Elts[i])
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2020-02-03 14:58:21 +00:00
|
|
|
emit.Int(c.prog.BinWriter, int64(ln))
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PACK)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2018-02-24 09:06:48 +00:00
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
return nil
|
|
|
|
|
|
|
|
case *ast.BinaryExpr:
|
2020-08-23 11:52:37 +00:00
|
|
|
c.emitBinaryExpr(n, false, false, 0)
|
|
|
|
return nil
|
2018-02-19 09:24:28 +00:00
|
|
|
|
|
|
|
case *ast.CallExpr:
|
|
|
|
var (
|
2018-04-04 19:41:19 +00:00
|
|
|
f *funcScope
|
|
|
|
ok bool
|
2020-05-06 15:10:20 +00:00
|
|
|
name string
|
2018-04-04 19:41:19 +00:00
|
|
|
numArgs = len(n.Args)
|
2020-06-08 11:16:41 +00:00
|
|
|
isBuiltin bool
|
2020-06-24 15:36:21 +00:00
|
|
|
isFunc bool
|
2020-08-19 14:31:02 +00:00
|
|
|
isLiteral bool
|
2018-02-19 09:24:28 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
switch fun := n.Fun.(type) {
|
|
|
|
case *ast.Ident:
|
2021-05-25 13:17:40 +00:00
|
|
|
f, ok = c.getFuncFromIdent(fun)
|
2020-06-08 11:16:41 +00:00
|
|
|
isBuiltin = isGoBuiltin(fun.Name)
|
2018-04-04 19:41:19 +00:00
|
|
|
if !ok && !isBuiltin {
|
2020-05-06 15:10:20 +00:00
|
|
|
name = fun.Name
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2020-06-24 15:36:21 +00:00
|
|
|
// distinguish lambda invocations from type conversions
|
|
|
|
if fun.Obj != nil && fun.Obj.Kind == ast.Var {
|
|
|
|
isFunc = true
|
|
|
|
}
|
2022-09-16 16:18:47 +00:00
|
|
|
if ok && canInline(f.pkg.Path(), f.decl.Name.Name, false) {
|
2021-02-04 12:41:00 +00:00
|
|
|
c.inlineCall(f, n)
|
|
|
|
return nil
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
case *ast.SelectorExpr:
|
2020-07-29 14:20:00 +00:00
|
|
|
name, isMethod := c.getFuncNameFromSelector(fun)
|
2018-08-22 07:51:35 +00:00
|
|
|
|
2020-07-29 14:20:00 +00:00
|
|
|
f, ok = c.funcs[name]
|
2021-02-03 13:10:31 +00:00
|
|
|
if ok {
|
2022-07-11 16:28:15 +00:00
|
|
|
f.selector = fun.X
|
2023-03-18 07:44:12 +00:00
|
|
|
isBuiltin = isPotentialCustomBuiltin(f, n)
|
2022-09-16 16:18:47 +00:00
|
|
|
if canInline(f.pkg.Path(), f.decl.Name.Name, isBuiltin) {
|
2021-02-04 12:41:00 +00:00
|
|
|
c.inlineCall(f, n)
|
|
|
|
return nil
|
|
|
|
}
|
2021-02-03 13:10:31 +00:00
|
|
|
} else {
|
|
|
|
typ := c.typeOf(fun)
|
|
|
|
ast.Walk(c, n.Args[0])
|
|
|
|
c.emitExplicitConvert(c.typeOf(n.Args[0]), typ)
|
2019-11-22 14:16:52 +00:00
|
|
|
return nil
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2022-07-06 14:41:39 +00:00
|
|
|
if isMethod {
|
|
|
|
// If this is a method call we need to walk the AST to load the struct locally.
|
|
|
|
// Otherwise, this is a function call from an imported package and we can call it
|
|
|
|
// directly.
|
|
|
|
ast.Walk(c, fun.X)
|
|
|
|
// Don't forget to add 1 extra argument when it's a method.
|
|
|
|
numArgs++
|
|
|
|
}
|
2018-05-06 06:03:26 +00:00
|
|
|
case *ast.ArrayType:
|
2020-01-27 10:37:33 +00:00
|
|
|
// For now we will assume that there are only byte slice conversions.
|
|
|
|
// E.g. []byte("foobar") or []byte(scriptHash).
|
|
|
|
ast.Walk(c, n.Args[0])
|
2020-06-03 12:55:06 +00:00
|
|
|
c.emitConvert(stackitem.BufferT)
|
2018-05-06 06:03:26 +00:00
|
|
|
return nil
|
2022-07-06 15:04:25 +00:00
|
|
|
case *ast.InterfaceType:
|
|
|
|
// It's a type conversion into some interface. Programmer is responsible
|
|
|
|
// for the conversion to be appropriate, just load the arg.
|
|
|
|
ast.Walk(c, n.Args[0])
|
|
|
|
return nil
|
2020-08-19 14:31:02 +00:00
|
|
|
case *ast.FuncLit:
|
|
|
|
isLiteral = true
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2020-03-31 13:57:35 +00:00
|
|
|
c.saveSequencePoint(n)
|
|
|
|
|
2022-09-16 16:18:47 +00:00
|
|
|
args := transformArgs(f, n.Fun, isBuiltin, n.Args)
|
2020-01-27 07:59:57 +00:00
|
|
|
|
2018-02-24 09:06:48 +00:00
|
|
|
// Handle the arguments
|
2020-01-27 07:59:57 +00:00
|
|
|
for _, arg := range args {
|
2018-02-19 09:24:28 +00:00
|
|
|
ast.Walk(c, arg)
|
2020-08-05 06:33:18 +00:00
|
|
|
typ := c.typeOf(arg)
|
|
|
|
_, ok := typ.Underlying().(*types.Struct)
|
|
|
|
if ok && !isInteropPath(typ.String()) {
|
|
|
|
// To clone struct fields we create a new array and append struct to it.
|
|
|
|
// This way even non-pointer struct fields will be copied.
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.NEWARRAY0,
|
|
|
|
opcode.DUP, opcode.ROT, opcode.APPEND,
|
2021-01-18 13:19:09 +00:00
|
|
|
opcode.POPITEM)
|
2020-08-05 06:33:18 +00:00
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2018-04-04 19:41:19 +00:00
|
|
|
// Do not swap for builtin functions.
|
2021-02-05 16:02:09 +00:00
|
|
|
if !isBuiltin && (f != nil && !isSyscall(f)) {
|
2020-06-27 07:39:30 +00:00
|
|
|
typ, ok := c.typeOf(n.Fun).(*types.Signature)
|
|
|
|
if ok && typ.Variadic() && !n.Ellipsis.IsValid() {
|
|
|
|
// pack variadic args into an array only if last argument is not of form `...`
|
2021-02-08 10:51:25 +00:00
|
|
|
varSize := c.packVarArgs(n, typ)
|
2020-06-26 10:30:48 +00:00
|
|
|
numArgs -= varSize - 1
|
|
|
|
}
|
2020-12-29 10:44:07 +00:00
|
|
|
c.emitReverse(numArgs)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2018-03-25 16:21:00 +00:00
|
|
|
// Check builtin first to avoid nil pointer on funcScope!
|
2019-02-09 15:53:58 +00:00
|
|
|
switch {
|
|
|
|
case isBuiltin:
|
2018-03-25 16:21:00 +00:00
|
|
|
// Use the ident to check, builtins are not in func scopes.
|
|
|
|
// We can be sure builtins are of type *ast.Ident.
|
2018-04-22 18:11:37 +00:00
|
|
|
c.convertBuiltin(n)
|
2020-05-06 15:10:20 +00:00
|
|
|
case name != "":
|
2022-04-20 18:30:09 +00:00
|
|
|
// Function was not found, thus it can only be an invocation of a func-typed variable or type conversion.
|
2020-06-24 15:43:29 +00:00
|
|
|
// We care only about string conversions because all others are effectively no-op in NeoVM.
|
|
|
|
// E.g. one cannot write `bool(int(a))`, only `int32(int(a))`.
|
|
|
|
if isString(c.typeOf(n.Fun)) {
|
|
|
|
c.emitConvert(stackitem.ByteArrayT)
|
|
|
|
} else if isFunc {
|
2020-07-28 16:35:41 +00:00
|
|
|
c.emitLoadVar("", name)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.CALLA)
|
2020-06-24 15:36:21 +00:00
|
|
|
}
|
2020-08-19 14:31:02 +00:00
|
|
|
case isLiteral:
|
|
|
|
ast.Walk(c, n.Fun)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.CALLA)
|
2018-08-22 07:51:35 +00:00
|
|
|
case isSyscall(f):
|
2021-03-02 11:15:25 +00:00
|
|
|
c.convertSyscall(f, n)
|
2019-02-09 15:53:58 +00:00
|
|
|
default:
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Call(c.prog.BinWriter, opcode.CALLL, f.label)
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
|
|
|
|
2020-08-24 15:37:39 +00:00
|
|
|
if c.scope != nil && c.scope.voidCalls[n] {
|
|
|
|
var sz int
|
|
|
|
if f != nil {
|
|
|
|
sz = f.decl.Type.Results.NumFields()
|
|
|
|
} else if !isBuiltin {
|
|
|
|
// lambda invocation
|
|
|
|
f := c.typeOf(n.Fun).Underlying().(*types.Signature)
|
|
|
|
sz = f.Results().Len()
|
|
|
|
}
|
|
|
|
for i := 0; i < sz; i++ {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DROP)
|
2020-08-24 15:37:39 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
return nil
|
|
|
|
|
2020-08-20 05:34:14 +00:00
|
|
|
case *ast.DeferStmt:
|
2020-08-21 12:37:46 +00:00
|
|
|
catch := c.newLabel()
|
2020-08-20 05:34:14 +00:00
|
|
|
finally := c.newLabel()
|
|
|
|
param := make([]byte, 8)
|
2020-08-21 12:37:46 +00:00
|
|
|
binary.LittleEndian.PutUint16(param[0:], catch)
|
2020-08-20 05:34:14 +00:00
|
|
|
binary.LittleEndian.PutUint16(param[4:], finally)
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.TRYL, param)
|
2022-02-03 10:20:40 +00:00
|
|
|
index := c.scope.newLocal(fmt.Sprintf("defer@%d", n.Call.Pos()))
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSH1)
|
|
|
|
c.emitStoreByIndex(varLocal, index)
|
2020-08-20 05:34:14 +00:00
|
|
|
c.scope.deferStack = append(c.scope.deferStack, deferInfo{
|
2020-08-21 12:37:46 +00:00
|
|
|
catchLabel: catch,
|
2020-08-20 05:34:14 +00:00
|
|
|
finallyLabel: finally,
|
|
|
|
expr: n.Call,
|
2022-02-03 10:20:40 +00:00
|
|
|
localIndex: index,
|
2020-08-20 05:34:14 +00:00
|
|
|
})
|
|
|
|
return nil
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
case *ast.SelectorExpr:
|
2020-07-28 16:35:41 +00:00
|
|
|
typ := c.typeOf(n.X)
|
2021-12-02 14:44:53 +00:00
|
|
|
if c.isInvalidType(typ) {
|
2020-07-28 16:35:41 +00:00
|
|
|
// This is a global variable from a package.
|
|
|
|
pkgAlias := n.X.(*ast.Ident).Name
|
2020-07-28 16:50:44 +00:00
|
|
|
name := c.getIdentName(pkgAlias, n.Sel.Name)
|
|
|
|
if tv, ok := c.constMap[name]; ok {
|
|
|
|
c.emitLoadConst(tv)
|
|
|
|
} else {
|
|
|
|
c.emitLoadVar(pkgAlias, n.Sel.Name)
|
|
|
|
}
|
2020-07-28 16:35:41 +00:00
|
|
|
return nil
|
|
|
|
}
|
2020-08-02 11:42:29 +00:00
|
|
|
strct, ok := c.getStruct(typ)
|
2020-05-20 08:05:01 +00:00
|
|
|
if !ok {
|
|
|
|
c.prog.Err = fmt.Errorf("selectors are supported only on structs")
|
2019-11-22 14:16:52 +00:00
|
|
|
return nil
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2020-05-20 08:05:01 +00:00
|
|
|
ast.Walk(c, n.X) // load the struct
|
|
|
|
i := indexOfStruct(strct, n.Sel.Name)
|
|
|
|
c.emitLoadField(i) // load the field
|
2018-02-19 09:24:28 +00:00
|
|
|
return nil
|
2018-02-27 09:04:24 +00:00
|
|
|
|
|
|
|
case *ast.UnaryExpr:
|
2020-08-02 11:42:29 +00:00
|
|
|
if n.Op == token.AND {
|
|
|
|
// We support only taking address from struct literals.
|
|
|
|
// For identifiers we can't support "taking address" in a general way
|
|
|
|
// because both struct and array are reference types.
|
|
|
|
lit, ok := n.X.(*ast.CompositeLit)
|
|
|
|
if ok {
|
|
|
|
c.convertStruct(lit, true)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
c.prog.Err = fmt.Errorf("'&' can be used only with struct literals")
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-04-04 19:41:19 +00:00
|
|
|
ast.Walk(c, n.X)
|
2019-02-19 13:50:34 +00:00
|
|
|
// From https://golang.org/ref/spec#Operators
|
|
|
|
// there can be only following unary operators
|
|
|
|
// "+" | "-" | "!" | "^" | "*" | "&" | "<-" .
|
|
|
|
// of which last three are not used in SC
|
|
|
|
switch n.Op {
|
|
|
|
case token.ADD:
|
|
|
|
// +10 == 10, no need to do anything in this case
|
|
|
|
case token.SUB:
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.NEGATE)
|
2019-02-19 13:50:34 +00:00
|
|
|
case token.NOT:
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.NOT)
|
2019-02-19 13:50:34 +00:00
|
|
|
case token.XOR:
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.INVERT)
|
2019-02-19 13:50:34 +00:00
|
|
|
default:
|
2019-11-22 14:16:52 +00:00
|
|
|
c.prog.Err = fmt.Errorf("invalid unary operator: %s", n.Op)
|
|
|
|
return nil
|
2019-02-19 13:50:34 +00:00
|
|
|
}
|
2018-04-04 19:41:19 +00:00
|
|
|
return nil
|
2018-03-25 16:21:00 +00:00
|
|
|
|
2018-04-02 15:04:42 +00:00
|
|
|
case *ast.IncDecStmt:
|
|
|
|
ast.Walk(c, n.X)
|
2020-08-23 09:39:58 +00:00
|
|
|
c.emitToken(n.Tok, c.typeOf(n.X))
|
2018-04-02 15:04:42 +00:00
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// For now, only identifiers are supported for (post) for stmts.
|
2018-04-02 15:04:42 +00:00
|
|
|
// for i := 0; i < 10; i++ {}
|
|
|
|
// Where the post stmt is ( i++ )
|
|
|
|
if ident, ok := n.X.(*ast.Ident); ok {
|
2020-07-28 16:35:41 +00:00
|
|
|
c.emitStoreVar("", ident.Name)
|
2018-04-02 15:04:42 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
|
2018-03-25 16:21:00 +00:00
|
|
|
case *ast.IndexExpr:
|
|
|
|
// Walk the expression, this could be either an Ident or SelectorExpr.
|
|
|
|
// This will load local whatever X is.
|
|
|
|
ast.Walk(c, n.X)
|
2020-05-19 12:35:57 +00:00
|
|
|
ast.Walk(c, n.Index)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PICKITEM) // just pickitem here
|
2020-01-23 14:28:35 +00:00
|
|
|
|
2018-03-25 16:21:00 +00:00
|
|
|
return nil
|
2018-04-02 15:04:42 +00:00
|
|
|
|
2020-02-17 13:48:41 +00:00
|
|
|
case *ast.BranchStmt:
|
2020-02-18 14:07:38 +00:00
|
|
|
var label string
|
2020-02-17 13:48:41 +00:00
|
|
|
if n.Label != nil {
|
|
|
|
label = n.Label.Name
|
2020-02-18 14:07:38 +00:00
|
|
|
} else if n.Tok == token.BREAK {
|
|
|
|
label = c.currentSwitch
|
|
|
|
} else if n.Tok == token.CONTINUE {
|
|
|
|
label = c.currentFor
|
2020-02-17 13:48:41 +00:00
|
|
|
}
|
|
|
|
|
2020-03-06 12:11:14 +00:00
|
|
|
cnt := 0
|
|
|
|
for i := len(c.labelList) - 1; i >= 0 && c.labelList[i].name != label; i-- {
|
|
|
|
cnt += c.labelList[i].sz
|
|
|
|
}
|
|
|
|
c.dropItems(cnt)
|
|
|
|
|
2020-02-17 13:48:41 +00:00
|
|
|
switch n.Tok {
|
|
|
|
case token.BREAK:
|
|
|
|
end := c.getLabelOffset(labelEnd, label)
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, end)
|
2020-02-17 13:48:41 +00:00
|
|
|
case token.CONTINUE:
|
2020-02-17 14:19:54 +00:00
|
|
|
post := c.getLabelOffset(labelPost, label)
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, post)
|
2020-02-17 13:48:41 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
|
|
|
case *ast.LabeledStmt:
|
|
|
|
c.nextLabel = n.Label.Name
|
|
|
|
|
|
|
|
ast.Walk(c, n.Stmt)
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
2020-06-29 13:59:13 +00:00
|
|
|
case *ast.BlockStmt:
|
|
|
|
c.scope.vars.newScope()
|
|
|
|
defer c.scope.vars.dropScope()
|
|
|
|
|
|
|
|
for i := range n.List {
|
|
|
|
ast.Walk(c, n.List[i])
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
2018-04-02 15:04:42 +00:00
|
|
|
case *ast.ForStmt:
|
2020-06-29 13:57:38 +00:00
|
|
|
c.scope.vars.newScope()
|
|
|
|
defer c.scope.vars.dropScope()
|
|
|
|
|
2020-02-17 13:48:41 +00:00
|
|
|
fstart, label := c.generateLabel(labelStart)
|
|
|
|
fend := c.newNamedLabel(labelEnd, label)
|
2020-02-17 14:19:54 +00:00
|
|
|
fpost := c.newNamedLabel(labelPost, label)
|
2020-02-17 13:48:41 +00:00
|
|
|
|
|
|
|
lastLabel := c.currentFor
|
2020-02-18 14:07:38 +00:00
|
|
|
lastSwitch := c.currentSwitch
|
2020-02-17 13:48:41 +00:00
|
|
|
c.currentFor = label
|
2020-02-18 14:07:38 +00:00
|
|
|
c.currentSwitch = label
|
2018-04-02 15:04:42 +00:00
|
|
|
|
|
|
|
// Walk the initializer and condition.
|
2020-01-23 08:48:41 +00:00
|
|
|
if n.Init != nil {
|
|
|
|
ast.Walk(c, n.Init)
|
|
|
|
}
|
2018-04-02 15:04:42 +00:00
|
|
|
|
|
|
|
// Set label and walk the condition.
|
2020-03-06 12:11:14 +00:00
|
|
|
c.pushStackLabel(label, 0)
|
2018-04-02 15:04:42 +00:00
|
|
|
c.setLabel(fstart)
|
2020-03-26 11:57:57 +00:00
|
|
|
if n.Cond != nil {
|
|
|
|
ast.Walk(c, n.Cond)
|
2018-04-02 15:04:42 +00:00
|
|
|
|
2020-03-26 11:57:57 +00:00
|
|
|
// Jump if the condition is false
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPIFNOTL, fend)
|
2020-03-26 11:57:57 +00:00
|
|
|
}
|
2018-04-02 15:04:42 +00:00
|
|
|
|
|
|
|
// Walk body followed by the iterator (post stmt).
|
|
|
|
ast.Walk(c, n.Body)
|
2020-02-17 14:19:54 +00:00
|
|
|
c.setLabel(fpost)
|
2020-01-23 08:48:41 +00:00
|
|
|
if n.Post != nil {
|
|
|
|
ast.Walk(c, n.Post)
|
|
|
|
}
|
2018-04-02 15:04:42 +00:00
|
|
|
|
|
|
|
// Jump back to condition.
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, fstart)
|
2018-04-02 15:04:42 +00:00
|
|
|
c.setLabel(fend)
|
2020-03-06 12:11:14 +00:00
|
|
|
c.dropStackLabel()
|
2018-04-02 15:04:42 +00:00
|
|
|
|
2020-02-17 13:48:41 +00:00
|
|
|
c.currentFor = lastLabel
|
2020-02-18 14:07:38 +00:00
|
|
|
c.currentSwitch = lastSwitch
|
2020-02-17 13:48:41 +00:00
|
|
|
|
2018-04-02 15:04:42 +00:00
|
|
|
return nil
|
2018-04-10 09:45:31 +00:00
|
|
|
|
2020-02-11 12:26:11 +00:00
|
|
|
case *ast.RangeStmt:
|
2020-06-29 13:57:38 +00:00
|
|
|
c.scope.vars.newScope()
|
|
|
|
defer c.scope.vars.dropScope()
|
|
|
|
|
2020-02-19 08:24:31 +00:00
|
|
|
start, label := c.generateLabel(labelStart)
|
|
|
|
end := c.newNamedLabel(labelEnd, label)
|
|
|
|
post := c.newNamedLabel(labelPost, label)
|
|
|
|
|
|
|
|
lastFor := c.currentFor
|
|
|
|
lastSwitch := c.currentSwitch
|
|
|
|
c.currentFor = label
|
|
|
|
c.currentSwitch = label
|
2020-02-11 12:26:11 +00:00
|
|
|
|
|
|
|
ast.Walk(c, n.X)
|
|
|
|
|
2020-06-19 14:14:00 +00:00
|
|
|
// Implementation is a bit different for slices and maps:
|
2022-04-20 18:30:09 +00:00
|
|
|
// For slices, we iterate through indices from 0 to len-1, storing array, len and index on stack.
|
|
|
|
// For maps, we iterate through indices from 0 to len-1, storing map, keyarray, size and index on stack.
|
2020-06-19 14:14:00 +00:00
|
|
|
_, isMap := c.typeOf(n.X).Underlying().(*types.Map)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
|
2020-06-19 14:14:00 +00:00
|
|
|
if isMap {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.KEYS, opcode.DUP)
|
2020-06-19 14:14:00 +00:00
|
|
|
}
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.SIZE, opcode.PUSH0)
|
2020-06-19 14:14:00 +00:00
|
|
|
|
|
|
|
stackSize := 3 // slice, len(slice), index
|
|
|
|
if isMap {
|
|
|
|
stackSize++ // map, keys, len(keys), index in keys
|
|
|
|
}
|
|
|
|
c.pushStackLabel(label, stackSize)
|
2020-02-11 12:26:11 +00:00
|
|
|
c.setLabel(start)
|
|
|
|
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.OVER, opcode.OVER)
|
2020-06-19 14:14:00 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPLEL, end)
|
2020-02-11 12:26:11 +00:00
|
|
|
|
2023-01-11 13:21:40 +00:00
|
|
|
var (
|
|
|
|
haveKey bool
|
|
|
|
haveVal bool
|
|
|
|
keyIdent *ast.Ident
|
|
|
|
keyLoaded bool
|
|
|
|
valIdent *ast.Ident
|
|
|
|
)
|
|
|
|
if n.Key != nil {
|
|
|
|
keyIdent, haveKey = n.Key.(*ast.Ident)
|
|
|
|
if !haveKey {
|
|
|
|
c.prog.Err = errors.New("only simple identifiers can be used for range loop keys (see #2870)")
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
haveKey = (keyIdent.Name != "_")
|
|
|
|
}
|
|
|
|
if n.Value != nil {
|
|
|
|
valIdent, haveVal = n.Value.(*ast.Ident)
|
|
|
|
if !haveVal {
|
|
|
|
c.prog.Err = errors.New("only simple identifiers can be used for range loop values (see #2870)")
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
haveVal = (valIdent.Name != "_")
|
|
|
|
}
|
|
|
|
if haveKey {
|
2020-06-19 14:14:00 +00:00
|
|
|
if isMap {
|
|
|
|
c.rangeLoadKey()
|
2023-01-11 13:21:40 +00:00
|
|
|
if haveVal {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
|
2020-06-19 14:41:06 +00:00
|
|
|
keyLoaded = true
|
|
|
|
}
|
2020-06-19 14:14:00 +00:00
|
|
|
} else {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
|
2020-06-19 14:14:00 +00:00
|
|
|
}
|
2023-01-11 12:49:10 +00:00
|
|
|
if n.Tok == token.DEFINE {
|
|
|
|
c.scope.newLocal(keyIdent.Name)
|
|
|
|
}
|
|
|
|
c.emitStoreVar("", keyIdent.Name)
|
2020-02-11 12:26:11 +00:00
|
|
|
}
|
2023-01-11 13:21:40 +00:00
|
|
|
if haveVal {
|
2020-06-19 14:41:06 +00:00
|
|
|
if !isMap || !keyLoaded {
|
|
|
|
c.rangeLoadKey()
|
|
|
|
}
|
2020-06-19 14:14:00 +00:00
|
|
|
if isMap {
|
|
|
|
// we have loaded only key from key array, now load value
|
|
|
|
emit.Int(c.prog.BinWriter, 4)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter,
|
|
|
|
opcode.PICK, // load map itself (+1 because key was pushed)
|
|
|
|
opcode.SWAP, // key should be on top
|
|
|
|
opcode.PICKITEM)
|
2020-06-19 14:14:00 +00:00
|
|
|
}
|
2023-01-11 12:49:10 +00:00
|
|
|
if n.Tok == token.DEFINE {
|
|
|
|
c.scope.newLocal(valIdent.Name)
|
|
|
|
}
|
|
|
|
c.emitStoreVar("", valIdent.Name)
|
2020-05-19 15:07:29 +00:00
|
|
|
}
|
2020-02-11 12:26:11 +00:00
|
|
|
|
|
|
|
ast.Walk(c, n.Body)
|
|
|
|
|
2020-02-19 08:24:31 +00:00
|
|
|
c.setLabel(post)
|
|
|
|
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.INC)
|
2020-04-23 08:56:36 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, start)
|
2020-02-11 12:26:11 +00:00
|
|
|
|
|
|
|
c.setLabel(end)
|
2020-03-06 12:11:14 +00:00
|
|
|
c.dropStackLabel()
|
2020-02-11 12:26:11 +00:00
|
|
|
|
2020-02-19 08:24:31 +00:00
|
|
|
c.currentFor = lastFor
|
|
|
|
c.currentSwitch = lastSwitch
|
|
|
|
|
2020-02-11 12:26:11 +00:00
|
|
|
return nil
|
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// We don't really care about assertions for the core logic.
|
2018-04-10 09:45:31 +00:00
|
|
|
// The only thing we need is to please the compiler type checking.
|
|
|
|
// For this to work properly, we only need to walk the expression
|
2022-04-20 18:30:09 +00:00
|
|
|
// which is not the assertion type.
|
2018-04-10 09:45:31 +00:00
|
|
|
case *ast.TypeAssertExpr:
|
|
|
|
ast.Walk(c, n.X)
|
2021-03-02 11:34:23 +00:00
|
|
|
if c.isCallExprSyscall(n.X) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2021-02-05 10:06:42 +00:00
|
|
|
goTyp := c.typeOf(n.Type)
|
|
|
|
if canConvert(goTyp.String()) {
|
|
|
|
typ := toNeoType(goTyp)
|
2021-11-12 16:58:21 +00:00
|
|
|
c.emitConvert(typ)
|
2021-02-05 10:06:42 +00:00
|
|
|
}
|
2018-04-10 09:45:31 +00:00
|
|
|
return nil
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
return c
|
|
|
|
}
|
|
|
|
|
2022-09-28 06:58:59 +00:00
|
|
|
func checkTypeAssertWithOK(n ast.Node) error {
|
|
|
|
if t, ok := n.(*ast.TypeAssertExpr); ok &&
|
|
|
|
t.Type != nil { // not a type switch
|
|
|
|
return ErrUnsupportedTypeAssertion
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2021-02-08 10:51:25 +00:00
|
|
|
// packVarArgs packs variadic arguments into an array
|
2022-04-20 18:30:09 +00:00
|
|
|
// and returns the amount of arguments packed.
|
2021-02-08 10:51:25 +00:00
|
|
|
func (c *codegen) packVarArgs(n *ast.CallExpr, typ *types.Signature) int {
|
|
|
|
varSize := len(n.Args) - typ.Params().Len() + 1
|
|
|
|
c.emitReverse(varSize)
|
|
|
|
emit.Int(c.prog.BinWriter, int64(varSize))
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PACK)
|
|
|
|
return varSize
|
|
|
|
}
|
|
|
|
|
2021-03-02 11:34:23 +00:00
|
|
|
func (c *codegen) isCallExprSyscall(e ast.Expr) bool {
|
|
|
|
ce, ok := e.(*ast.CallExpr)
|
|
|
|
if !ok {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
sel, ok := ce.Fun.(*ast.SelectorExpr)
|
|
|
|
if !ok {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
name, _ := c.getFuncNameFromSelector(sel)
|
|
|
|
f, ok := c.funcs[name]
|
|
|
|
return ok && isSyscall(f)
|
|
|
|
}
|
|
|
|
|
2020-08-20 05:34:14 +00:00
|
|
|
// processDefers emits code for `defer` statements.
|
2020-08-21 12:37:46 +00:00
|
|
|
// TRY-related opcodes handle exception as follows:
|
2022-08-08 10:23:21 +00:00
|
|
|
// 1. CATCH block is executed only if exception has occurred.
|
|
|
|
// 2. FINALLY block is always executed, but after catch block.
|
|
|
|
//
|
2020-08-21 12:37:46 +00:00
|
|
|
// Go `defer` statements are a bit different:
|
2022-08-08 10:23:21 +00:00
|
|
|
// 1. `defer` is always executed irregardless of whether an exception has occurred.
|
|
|
|
// 2. `recover` can or can not handle a possible exception.
|
|
|
|
//
|
2022-04-20 18:30:09 +00:00
|
|
|
// Thus, we use the following approach:
|
2022-08-08 10:23:21 +00:00
|
|
|
// 1. Throwed exception is saved in a static field X, static fields Y and it is set to true.
|
|
|
|
// 2. For each defer local there is a dedicated local variable which is set to 1 if `defer` statement
|
|
|
|
// is encountered during an actual execution.
|
|
|
|
// 3. CATCH and FINALLY blocks are the same, and both contain the same CALLs.
|
|
|
|
// 4. Right before the CATCH block, check a variable from (2). If it is null, jump to the end of CATCH+FINALLY block.
|
|
|
|
// 5. In CATCH block we set Y to true and emit default return values if it is the last defer.
|
|
|
|
// 6. Execute FINALLY block only if Y is false.
|
2020-08-20 05:34:14 +00:00
|
|
|
func (c *codegen) processDefers() {
|
|
|
|
for i := len(c.scope.deferStack) - 1; i >= 0; i-- {
|
|
|
|
stmt := c.scope.deferStack[i]
|
|
|
|
after := c.newLabel()
|
2022-02-03 10:20:40 +00:00
|
|
|
|
|
|
|
c.emitLoadByIndex(varLocal, c.scope.deferStack[i].localIndex)
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.ISNULL)
|
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPIFL, after)
|
|
|
|
|
2020-08-20 05:34:14 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.ENDTRYL, after)
|
2020-08-21 12:37:46 +00:00
|
|
|
|
|
|
|
c.setLabel(stmt.catchLabel)
|
|
|
|
c.emitStoreByIndex(varGlobal, c.exceptionIndex)
|
|
|
|
emit.Int(c.prog.BinWriter, 1)
|
2021-04-28 08:20:41 +00:00
|
|
|
|
|
|
|
finalIndex := c.getVarIndex("", finallyVarName).index
|
|
|
|
c.emitStoreByIndex(varLocal, finalIndex)
|
2020-08-21 12:37:46 +00:00
|
|
|
ast.Walk(c, stmt.expr)
|
|
|
|
if i == 0 {
|
2021-11-12 16:30:50 +00:00
|
|
|
results := c.scope.decl.Type.Results
|
|
|
|
if results.NumFields() != 0 {
|
|
|
|
// After panic, default values must be returns, except for named returns,
|
|
|
|
// which we don't support here for now.
|
|
|
|
for i := len(results.List) - 1; i >= 0; i-- {
|
|
|
|
c.emitDefault(c.typeOf(results.List[i].Type))
|
|
|
|
}
|
2020-08-21 12:37:46 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.ENDTRYL, after)
|
|
|
|
|
2020-08-20 05:34:14 +00:00
|
|
|
c.setLabel(stmt.finallyLabel)
|
2020-08-21 12:37:46 +00:00
|
|
|
before := c.newLabel()
|
2021-04-28 08:20:41 +00:00
|
|
|
c.emitLoadByIndex(varLocal, finalIndex)
|
2020-08-21 12:37:46 +00:00
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPIFL, before)
|
2020-08-20 05:34:14 +00:00
|
|
|
ast.Walk(c, stmt.expr)
|
2020-08-21 12:37:46 +00:00
|
|
|
c.setLabel(before)
|
|
|
|
emit.Int(c.prog.BinWriter, 0)
|
2021-04-28 08:20:41 +00:00
|
|
|
c.emitStoreByIndex(varLocal, finalIndex)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.ENDFINALLY)
|
2020-08-20 05:34:14 +00:00
|
|
|
c.setLabel(after)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-02-03 13:10:31 +00:00
|
|
|
// emitExplicitConvert handles `someType(someValue)` conversions between string/[]byte.
|
|
|
|
// Rules for conversion:
|
2022-08-08 10:23:21 +00:00
|
|
|
// 1. interop.* types are converted to ByteArray if not already.
|
|
|
|
// 2. Otherwise, convert between ByteArray/Buffer.
|
|
|
|
// 3. Rules for types which are not string/[]byte should already
|
|
|
|
// be enforced by go parser.
|
2021-02-03 13:10:31 +00:00
|
|
|
func (c *codegen) emitExplicitConvert(from, to types.Type) {
|
|
|
|
if isInteropPath(to.String()) {
|
|
|
|
if isByteSlice(from) && !isString(from) {
|
|
|
|
c.emitConvert(stackitem.ByteArrayT)
|
|
|
|
}
|
|
|
|
} else if isByteSlice(to) && !isByteSlice(from) {
|
|
|
|
c.emitConvert(stackitem.BufferT)
|
|
|
|
} else if isString(to) && !isString(from) {
|
|
|
|
c.emitConvert(stackitem.ByteArrayT)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-02 14:44:53 +00:00
|
|
|
func (c *codegen) isInvalidType(typ types.Type) bool {
|
|
|
|
tb, ok := typ.(*types.Basic)
|
|
|
|
return typ == nil || ok && tb.Kind() == types.Invalid
|
|
|
|
}
|
|
|
|
|
2020-06-19 14:14:00 +00:00
|
|
|
func (c *codegen) rangeLoadKey() {
|
|
|
|
emit.Int(c.prog.BinWriter, 2)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter,
|
|
|
|
opcode.PICK, // load keys
|
|
|
|
opcode.OVER, // load index in key array
|
|
|
|
opcode.PICKITEM)
|
2020-06-19 14:14:00 +00:00
|
|
|
}
|
|
|
|
|
2020-03-10 09:34:05 +00:00
|
|
|
func isFallthroughStmt(c ast.Node) bool {
|
|
|
|
s, ok := c.(*ast.BranchStmt)
|
|
|
|
return ok && s.Tok == token.FALLTHROUGH
|
|
|
|
}
|
|
|
|
|
2020-08-23 09:33:21 +00:00
|
|
|
func (c *codegen) getCompareWithNilArg(n *ast.BinaryExpr) ast.Expr {
|
|
|
|
if isExprNil(n.X) {
|
|
|
|
return n.Y
|
|
|
|
} else if isExprNil(n.Y) {
|
|
|
|
return n.X
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2020-08-23 11:52:37 +00:00
|
|
|
func (c *codegen) emitJumpOnCondition(cond bool, jmpLabel uint16) {
|
|
|
|
if cond {
|
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPIFL, jmpLabel)
|
|
|
|
} else {
|
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPIFNOTL, jmpLabel)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-08-23 15:00:23 +00:00
|
|
|
// emitBoolExpr emits boolean expression. If needJump is true and expression evaluates to `cond`,
|
|
|
|
// jump to jmpLabel is performed and no item is left on stack.
|
2020-08-23 11:52:37 +00:00
|
|
|
func (c *codegen) emitBoolExpr(n ast.Expr, needJump bool, cond bool, jmpLabel uint16) {
|
|
|
|
if be, ok := n.(*ast.BinaryExpr); ok {
|
|
|
|
c.emitBinaryExpr(be, needJump, cond, jmpLabel)
|
|
|
|
} else {
|
|
|
|
ast.Walk(c, n)
|
|
|
|
if needJump {
|
|
|
|
c.emitJumpOnCondition(cond, jmpLabel)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-08-23 15:00:23 +00:00
|
|
|
// emitBinaryExpr emits binary expression. If needJump is true and expression evaluates to `cond`,
|
|
|
|
// jump to jmpLabel is performed and no item is left on stack.
|
2020-08-23 11:52:37 +00:00
|
|
|
func (c *codegen) emitBinaryExpr(n *ast.BinaryExpr, needJump bool, cond bool, jmpLabel uint16) {
|
|
|
|
// The AST package will try to resolve all basic literals for us.
|
|
|
|
// If the typeinfo.Value is not nil we know that the expr is resolved
|
|
|
|
// and needs no further action. e.g. x := 2 + 2 + 2 will be resolved to 6.
|
|
|
|
// NOTE: Constants will also be automatically resolved be the AST parser.
|
|
|
|
// example:
|
|
|
|
// const x = 10
|
|
|
|
// x + 2 will results into 12
|
|
|
|
tinfo := c.typeAndValueOf(n)
|
|
|
|
if tinfo.Value != nil {
|
|
|
|
c.emitLoadConst(tinfo)
|
|
|
|
if needJump && isBool(tinfo.Type) {
|
|
|
|
c.emitJumpOnCondition(cond, jmpLabel)
|
|
|
|
}
|
|
|
|
return
|
|
|
|
} else if arg := c.getCompareWithNilArg(n); arg != nil {
|
|
|
|
ast.Walk(c, arg)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.ISNULL)
|
2020-08-23 11:52:37 +00:00
|
|
|
if needJump {
|
|
|
|
c.emitJumpOnCondition(cond == (n.Op == token.EQL), jmpLabel)
|
|
|
|
} else if n.Op == token.NEQ {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.NOT)
|
2020-08-23 11:52:37 +00:00
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
switch n.Op {
|
|
|
|
case token.LAND, token.LOR:
|
|
|
|
end := c.newLabel()
|
|
|
|
|
|
|
|
// true || .. == true, false && .. == false
|
|
|
|
condShort := n.Op == token.LOR
|
|
|
|
if needJump {
|
|
|
|
l := end
|
|
|
|
if cond == condShort {
|
|
|
|
l = jmpLabel
|
|
|
|
}
|
|
|
|
c.emitBoolExpr(n.X, true, condShort, l)
|
|
|
|
c.emitBoolExpr(n.Y, true, cond, jmpLabel)
|
|
|
|
} else {
|
|
|
|
push := c.newLabel()
|
|
|
|
c.emitBoolExpr(n.X, true, condShort, push)
|
|
|
|
c.emitBoolExpr(n.Y, false, false, 0)
|
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, end)
|
|
|
|
c.setLabel(push)
|
|
|
|
emit.Bool(c.prog.BinWriter, condShort)
|
|
|
|
}
|
|
|
|
c.setLabel(end)
|
|
|
|
|
|
|
|
default:
|
|
|
|
ast.Walk(c, n.X)
|
|
|
|
ast.Walk(c, n.Y)
|
2020-08-23 15:00:23 +00:00
|
|
|
typ := c.typeOf(n.X)
|
|
|
|
if !needJump {
|
|
|
|
c.emitToken(n.Op, typ)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
op, ok := getJumpForToken(n.Op, typ)
|
|
|
|
if !ok {
|
|
|
|
c.emitToken(n.Op, typ)
|
2020-08-23 11:52:37 +00:00
|
|
|
c.emitJumpOnCondition(cond, jmpLabel)
|
2020-08-23 15:00:23 +00:00
|
|
|
return
|
2020-08-23 11:52:37 +00:00
|
|
|
}
|
2020-08-23 15:00:23 +00:00
|
|
|
if !cond {
|
|
|
|
op = negateJmp(op)
|
|
|
|
}
|
|
|
|
emit.Jmp(c.prog.BinWriter, op, jmpLabel)
|
2020-08-23 11:52:37 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-06 12:11:14 +00:00
|
|
|
func (c *codegen) pushStackLabel(name string, size int) {
|
|
|
|
c.labelList = append(c.labelList, labelWithStackSize{
|
|
|
|
name: name,
|
|
|
|
sz: size,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *codegen) dropStackLabel() {
|
|
|
|
last := len(c.labelList) - 1
|
|
|
|
c.dropItems(c.labelList[last].sz)
|
|
|
|
c.labelList = c.labelList[:last]
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *codegen) dropItems(n int) {
|
|
|
|
if n < 4 {
|
|
|
|
for i := 0; i < n; i++ {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DROP)
|
2020-03-06 12:11:14 +00:00
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
emit.Int(c.prog.BinWriter, int64(n))
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PACK, opcode.DROP)
|
2020-03-06 12:11:14 +00:00
|
|
|
}
|
|
|
|
|
2020-02-10 07:43:29 +00:00
|
|
|
// emitReverse reverses top num items of the stack.
|
|
|
|
func (c *codegen) emitReverse(num int) {
|
|
|
|
switch num {
|
2020-06-19 14:28:07 +00:00
|
|
|
case 0, 1:
|
2020-02-10 07:43:29 +00:00
|
|
|
case 2:
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.SWAP)
|
2020-02-10 07:43:29 +00:00
|
|
|
case 3:
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.REVERSE3)
|
2020-05-06 09:12:29 +00:00
|
|
|
case 4:
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.REVERSE4)
|
2020-02-10 07:43:29 +00:00
|
|
|
default:
|
2020-05-06 09:12:29 +00:00
|
|
|
emit.Int(c.prog.BinWriter, int64(num))
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.REVERSEN)
|
2020-02-10 07:43:29 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-17 13:48:41 +00:00
|
|
|
// generateLabel returns a new label.
|
2020-02-21 08:50:03 +00:00
|
|
|
func (c *codegen) generateLabel(typ labelOffsetType) (uint16, string) {
|
2020-02-17 13:48:41 +00:00
|
|
|
name := c.nextLabel
|
|
|
|
if name == "" {
|
|
|
|
name = fmt.Sprintf("@%d", len(c.l))
|
|
|
|
}
|
|
|
|
|
|
|
|
c.nextLabel = ""
|
|
|
|
return c.newNamedLabel(typ, name), name
|
|
|
|
}
|
|
|
|
|
2020-02-21 08:50:03 +00:00
|
|
|
func (c *codegen) getLabelOffset(typ labelOffsetType, name string) uint16 {
|
2020-02-17 13:48:41 +00:00
|
|
|
return c.labels[labelWithType{name: name, typ: typ}]
|
|
|
|
}
|
|
|
|
|
2020-08-23 15:00:23 +00:00
|
|
|
// For `&&` and `||` it return an opcode which jumps only if result is known:
|
2021-05-12 20:17:03 +00:00
|
|
|
// false && .. == false, true || .. = true.
|
2020-08-23 15:00:23 +00:00
|
|
|
func getJumpForToken(tok token.Token, typ types.Type) (opcode.Opcode, bool) {
|
|
|
|
switch tok {
|
|
|
|
case token.GTR:
|
|
|
|
return opcode.JMPGTL, true
|
|
|
|
case token.GEQ:
|
|
|
|
return opcode.JMPGEL, true
|
|
|
|
case token.LSS:
|
|
|
|
return opcode.JMPLTL, true
|
|
|
|
case token.LEQ:
|
|
|
|
return opcode.JMPLEL, true
|
|
|
|
case token.EQL, token.NEQ:
|
|
|
|
if isNumber(typ) {
|
|
|
|
if tok == token.EQL {
|
|
|
|
return opcode.JMPEQL, true
|
|
|
|
}
|
|
|
|
return opcode.JMPNEL, true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0, false
|
|
|
|
}
|
|
|
|
|
2021-12-08 19:33:03 +00:00
|
|
|
func (c *codegen) getCallToken(hash util.Uint160, method string, pcount int, hasReturn bool, flag callflag.CallFlag) (uint16, error) {
|
|
|
|
needed := nef.MethodToken{
|
|
|
|
Hash: hash,
|
|
|
|
Method: method,
|
|
|
|
ParamCount: uint16(pcount),
|
|
|
|
HasReturn: hasReturn,
|
|
|
|
CallFlag: flag,
|
|
|
|
}
|
|
|
|
for i := range c.callTokens {
|
|
|
|
if c.callTokens[i] == needed {
|
|
|
|
return uint16(i), nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if len(c.callTokens) == math.MaxUint16 {
|
|
|
|
return 0, errors.New("call token overflow")
|
|
|
|
}
|
|
|
|
c.callTokens = append(c.callTokens, needed)
|
|
|
|
return uint16(len(c.callTokens) - 1), nil
|
|
|
|
}
|
|
|
|
|
2021-03-02 11:15:25 +00:00
|
|
|
func (c *codegen) convertSyscall(f *funcScope, expr *ast.CallExpr) {
|
2021-12-08 19:33:03 +00:00
|
|
|
var callArgs = expr.Args[1:]
|
|
|
|
|
|
|
|
if strings.HasPrefix(f.name, "CallWithToken") {
|
|
|
|
callArgs = expr.Args[3:]
|
|
|
|
}
|
|
|
|
for _, arg := range callArgs {
|
2021-02-05 16:02:09 +00:00
|
|
|
ast.Walk(c, arg)
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
2021-02-05 16:02:09 +00:00
|
|
|
tv := c.typeAndValueOf(expr.Args[0])
|
2021-12-08 19:33:03 +00:00
|
|
|
if tv.Value == nil || !isString(tv.Type) {
|
|
|
|
c.prog.Err = fmt.Errorf("bad intrinsic argument")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
arg0Str := constant.StringVal(tv.Value)
|
|
|
|
|
2021-03-02 11:15:25 +00:00
|
|
|
if strings.HasPrefix(f.name, "Syscall") {
|
2021-12-08 19:33:03 +00:00
|
|
|
c.emitReverse(len(callArgs))
|
|
|
|
emit.Syscall(c.prog.BinWriter, arg0Str)
|
|
|
|
} else if strings.HasPrefix(f.name, "CallWithToken") {
|
|
|
|
var hasRet = !strings.HasSuffix(f.name, "NoRet")
|
|
|
|
|
|
|
|
c.emitReverse(len(callArgs))
|
|
|
|
|
|
|
|
hash, err := util.Uint160DecodeBytesBE([]byte(arg0Str))
|
|
|
|
if err != nil {
|
|
|
|
c.prog.Err = fmt.Errorf("bad callt hash: %w", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
tv = c.typeAndValueOf(expr.Args[1])
|
|
|
|
if tv.Value == nil || !isString(tv.Type) {
|
|
|
|
c.prog.Err = errors.New("bad callt method")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
method := constant.StringVal(tv.Value)
|
|
|
|
|
|
|
|
tv = c.typeAndValueOf(expr.Args[2])
|
|
|
|
if tv.Value == nil || !isNumber(tv.Type) {
|
|
|
|
c.prog.Err = errors.New("bad callt call flags")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
flag, ok := constant.Uint64Val(tv.Value)
|
|
|
|
if !ok || flag > 255 {
|
|
|
|
c.prog.Err = errors.New("invalid callt flag")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
c.appendInvokedContract(hash, method, flag)
|
|
|
|
|
|
|
|
tokNum, err := c.getCallToken(hash, method, len(callArgs), hasRet, callflag.CallFlag(flag))
|
|
|
|
if err != nil {
|
|
|
|
c.prog.Err = err
|
|
|
|
return
|
|
|
|
}
|
|
|
|
tokBuf := make([]byte, 2)
|
|
|
|
binary.LittleEndian.PutUint16(tokBuf, tokNum)
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.CALLT, tokBuf)
|
2021-03-02 11:15:25 +00:00
|
|
|
} else {
|
2021-12-08 19:33:03 +00:00
|
|
|
op, err := opcode.FromString(arg0Str)
|
2021-03-02 11:15:25 +00:00
|
|
|
if err != nil {
|
|
|
|
c.prog.Err = fmt.Errorf("invalid opcode: %s", op)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
emit.Opcodes(c.prog.BinWriter, op)
|
|
|
|
}
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
|
|
|
|
2020-08-24 09:54:01 +00:00
|
|
|
// emitSliceHelper emits 3 items on stack: slice, its first index, and its size.
|
|
|
|
func (c *codegen) emitSliceHelper(e ast.Expr) {
|
|
|
|
if !isByteSlice(c.typeOf(e)) {
|
|
|
|
c.prog.Err = fmt.Errorf("copy is supported only for byte-slices")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
var hasLowIndex bool
|
|
|
|
switch src := e.(type) {
|
|
|
|
case *ast.SliceExpr:
|
|
|
|
ast.Walk(c, src.X)
|
|
|
|
if src.High != nil {
|
|
|
|
ast.Walk(c, src.High)
|
|
|
|
} else {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP, opcode.SIZE)
|
2020-08-24 09:54:01 +00:00
|
|
|
}
|
|
|
|
if src.Low != nil {
|
|
|
|
ast.Walk(c, src.Low)
|
|
|
|
hasLowIndex = true
|
|
|
|
} else {
|
|
|
|
emit.Int(c.prog.BinWriter, 0)
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
ast.Walk(c, src)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP, opcode.SIZE)
|
2020-08-24 09:54:01 +00:00
|
|
|
emit.Int(c.prog.BinWriter, 0)
|
|
|
|
}
|
|
|
|
if !hasLowIndex {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.SWAP)
|
2020-08-24 09:54:01 +00:00
|
|
|
} else {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP, opcode.ROT, opcode.SWAP, opcode.SUB)
|
2020-08-24 09:54:01 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-04-22 18:11:37 +00:00
|
|
|
func (c *codegen) convertBuiltin(expr *ast.CallExpr) {
|
|
|
|
var name string
|
|
|
|
switch t := expr.Fun.(type) {
|
|
|
|
case *ast.Ident:
|
|
|
|
name = t.Name
|
|
|
|
case *ast.SelectorExpr:
|
|
|
|
name = t.Sel.Name
|
|
|
|
}
|
|
|
|
|
2018-03-25 16:21:00 +00:00
|
|
|
switch name {
|
2020-08-24 09:54:01 +00:00
|
|
|
case "copy":
|
|
|
|
// stack for MEMCPY is: dst, dst_index, src, src_index, count
|
|
|
|
c.emitSliceHelper(expr.Args[0])
|
|
|
|
c.emitSliceHelper(expr.Args[1])
|
|
|
|
emit.Int(c.prog.BinWriter, 3)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.ROLL, opcode.MIN)
|
2020-09-02 12:29:59 +00:00
|
|
|
if !c.scope.voidCalls[expr] {
|
|
|
|
// insert top item to the bottom of MEMCPY args, so that it is left on stack
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
|
2020-09-02 12:29:59 +00:00
|
|
|
emit.Int(c.prog.BinWriter, 6)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.REVERSEN)
|
2020-09-02 12:29:59 +00:00
|
|
|
emit.Int(c.prog.BinWriter, 5)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.REVERSEN)
|
2020-09-02 12:29:59 +00:00
|
|
|
}
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.MEMCPY)
|
2020-08-24 08:58:29 +00:00
|
|
|
case "make":
|
|
|
|
typ := c.typeOf(expr.Args[0])
|
|
|
|
switch {
|
|
|
|
case isMap(typ):
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.NEWMAP)
|
2020-08-24 08:58:29 +00:00
|
|
|
default:
|
|
|
|
if len(expr.Args) == 3 {
|
|
|
|
c.prog.Err = fmt.Errorf("`make()` with a capacity argument is not supported")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
ast.Walk(c, expr.Args[1])
|
|
|
|
if isByteSlice(typ) {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.NEWBUFFER)
|
2020-08-24 08:58:29 +00:00
|
|
|
} else {
|
|
|
|
neoT := toNeoType(typ.(*types.Slice).Elem())
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.NEWARRAYT, []byte{byte(neoT)})
|
|
|
|
}
|
|
|
|
}
|
2018-03-25 16:21:00 +00:00
|
|
|
case "len":
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP, opcode.ISNULL)
|
2020-07-07 18:49:21 +00:00
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.JMPIF, []byte{2 + 1 + 2})
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.SIZE)
|
2020-07-07 18:49:21 +00:00
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.JMP, []byte{2 + 1 + 1})
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DROP, opcode.PUSH0)
|
2018-03-25 16:21:00 +00:00
|
|
|
case "append":
|
2019-09-12 14:39:23 +00:00
|
|
|
arg := expr.Args[0]
|
2022-09-27 11:35:13 +00:00
|
|
|
typ := c.typeOf(arg)
|
2021-02-16 13:20:19 +00:00
|
|
|
ast.Walk(c, arg)
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP, opcode.ISNULL)
|
2020-05-13 15:09:55 +00:00
|
|
|
if isByteSlice(typ) {
|
2021-02-16 13:20:19 +00:00
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.JMPIFNOT, []byte{2 + 3})
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DROP, opcode.PUSHDATA1, 0)
|
|
|
|
if expr.Ellipsis.IsValid() {
|
|
|
|
ast.Walk(c, expr.Args[1])
|
2020-07-09 10:51:49 +00:00
|
|
|
} else {
|
2021-02-16 13:20:19 +00:00
|
|
|
c.convertByteArray(expr.Args[1:])
|
|
|
|
}
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.CAT)
|
|
|
|
} else {
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.JMPIFNOT, []byte{2 + 2})
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DROP, opcode.NEWARRAY0)
|
2021-02-16 13:45:06 +00:00
|
|
|
if expr.Ellipsis.IsValid() {
|
|
|
|
ast.Walk(c, expr.Args[1]) // x y
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSH0) // x y cnt=0
|
|
|
|
start := c.newLabel()
|
|
|
|
c.setLabel(start)
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSH2, opcode.PICK) // x y cnt x
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSH2, opcode.PICK) // x y cnt x y
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP, opcode.SIZE) // x y cnt x y len(y)
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSH3, opcode.PICK) // x y cnt x y len(y) cnt
|
|
|
|
after := c.newLabel()
|
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPEQL, after) // x y cnt x y
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSH2, opcode.PICK, // x y cnt x y cnt
|
|
|
|
opcode.PICKITEM, // x y cnt x y[cnt]
|
|
|
|
opcode.APPEND, // x=append(x, y[cnt]) y cnt
|
|
|
|
opcode.INC) // x y cnt+1
|
|
|
|
emit.Jmp(c.prog.BinWriter, opcode.JMPL, start)
|
|
|
|
c.setLabel(after)
|
|
|
|
for i := 0; i < 4; i++ { // leave x on stack
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DROP)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
for _, e := range expr.Args[1:] {
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
|
|
|
|
ast.Walk(c, e)
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.APPEND)
|
|
|
|
}
|
2020-07-09 10:51:49 +00:00
|
|
|
}
|
2019-09-12 14:39:23 +00:00
|
|
|
}
|
2020-01-28 14:18:38 +00:00
|
|
|
case "panic":
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.THROW)
|
2020-08-21 12:37:46 +00:00
|
|
|
case "recover":
|
2020-09-02 12:20:11 +00:00
|
|
|
if !c.scope.voidCalls[expr] {
|
|
|
|
c.emitLoadByIndex(varGlobal, c.exceptionIndex)
|
|
|
|
}
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PUSHNULL)
|
2020-08-21 12:37:46 +00:00
|
|
|
c.emitStoreByIndex(varGlobal, c.exceptionIndex)
|
2020-09-06 12:49:41 +00:00
|
|
|
case "delete":
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.REMOVE)
|
2023-03-18 07:44:12 +00:00
|
|
|
case "ToHash160":
|
2022-09-16 16:18:47 +00:00
|
|
|
// We can be sure that this is an ast.BasicLit just containing a simple
|
2019-02-13 18:01:10 +00:00
|
|
|
// address string. Note that the string returned from calling Value will
|
|
|
|
// contain double quotes that need to be stripped.
|
2018-07-02 13:02:00 +00:00
|
|
|
addressStr := expr.Args[0].(*ast.BasicLit).Value
|
|
|
|
addressStr = strings.Replace(addressStr, "\"", "", 2)
|
2019-12-25 14:34:18 +00:00
|
|
|
uint160, err := address.StringToUint160(addressStr)
|
2018-07-02 13:02:00 +00:00
|
|
|
if err != nil {
|
2019-11-22 14:16:52 +00:00
|
|
|
c.prog.Err = err
|
|
|
|
return
|
2018-07-02 13:02:00 +00:00
|
|
|
}
|
2019-11-27 09:20:31 +00:00
|
|
|
bytes := uint160.BytesBE()
|
2020-02-03 14:58:21 +00:00
|
|
|
emit.Bytes(c.prog.BinWriter, bytes)
|
2020-06-03 12:55:06 +00:00
|
|
|
c.emitConvert(stackitem.BufferT)
|
2018-03-25 16:21:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-01-29 07:08:24 +00:00
|
|
|
// transformArgs returns a list of function arguments
|
|
|
|
// which should be put on stack.
|
|
|
|
// There are special cases for builtins:
|
2023-03-18 07:44:12 +00:00
|
|
|
// 1. With ToHash160 in case if it behaves like builtin,
|
2022-09-16 16:18:47 +00:00
|
|
|
// parameter conversion is happening at compile-time so there is no need to
|
|
|
|
// push parameters on stack and perform an actual call
|
2022-08-08 10:23:21 +00:00
|
|
|
// 2. With panic, the generated code depends on the fact if an argument was nil or a string;
|
|
|
|
// so, it should be handled accordingly.
|
2022-09-16 16:18:47 +00:00
|
|
|
func transformArgs(fs *funcScope, fun ast.Expr, isBuiltin bool, args []ast.Expr) []ast.Expr {
|
2020-01-29 07:08:24 +00:00
|
|
|
switch f := fun.(type) {
|
|
|
|
case *ast.SelectorExpr:
|
2023-03-18 07:44:12 +00:00
|
|
|
if isBuiltin && f.Sel.Name == "ToHash160" {
|
2020-01-29 07:08:24 +00:00
|
|
|
return args[1:]
|
|
|
|
}
|
2021-02-05 16:02:09 +00:00
|
|
|
if fs != nil && isSyscall(fs) {
|
|
|
|
return nil
|
|
|
|
}
|
2020-01-29 07:08:24 +00:00
|
|
|
case *ast.Ident:
|
2020-08-24 09:54:01 +00:00
|
|
|
switch f.Name {
|
2021-02-16 13:20:19 +00:00
|
|
|
case "make", "copy", "append":
|
2020-08-24 08:58:29 +00:00
|
|
|
return nil
|
2020-01-29 07:08:24 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return args
|
|
|
|
}
|
|
|
|
|
2022-04-20 18:30:09 +00:00
|
|
|
// emitConvert converts the top stack item to the specified type.
|
2020-06-03 12:55:06 +00:00
|
|
|
func (c *codegen) emitConvert(typ stackitem.Type) {
|
2022-10-25 15:20:55 +00:00
|
|
|
if typ == stackitem.BooleanT {
|
|
|
|
// DUP + ISTYPE + JMPIF costs 3 already with CONVERT of a cost 8192.
|
|
|
|
// NOT+NOT at the same time costs 4 and always works (and is shorter).
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.NOT, opcode.NOT)
|
|
|
|
} else {
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.ISTYPE, []byte{byte(typ)})
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.JMPIF, []byte{2 + 2}) // After CONVERT.
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.CONVERT, []byte{byte(typ)})
|
|
|
|
}
|
2020-05-13 08:06:12 +00:00
|
|
|
}
|
|
|
|
|
2021-02-16 13:20:19 +00:00
|
|
|
func (c *codegen) convertByteArray(elems []ast.Expr) {
|
|
|
|
buf := make([]byte, len(elems))
|
2020-09-24 17:20:34 +00:00
|
|
|
varIndices := []int{}
|
2021-02-16 13:20:19 +00:00
|
|
|
for i := 0; i < len(elems); i++ {
|
|
|
|
t := c.typeAndValueOf(elems[i])
|
2020-09-24 17:20:34 +00:00
|
|
|
if t.Value != nil {
|
|
|
|
val, _ := constant.Int64Val(t.Value)
|
|
|
|
buf[i] = byte(val)
|
|
|
|
} else {
|
|
|
|
varIndices = append(varIndices, i)
|
|
|
|
}
|
2018-02-27 09:04:24 +00:00
|
|
|
}
|
2020-02-03 14:58:21 +00:00
|
|
|
emit.Bytes(c.prog.BinWriter, buf)
|
2020-06-03 12:55:06 +00:00
|
|
|
c.emitConvert(stackitem.BufferT)
|
2020-09-24 17:20:34 +00:00
|
|
|
for _, i := range varIndices {
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
|
2020-09-24 17:20:34 +00:00
|
|
|
emit.Int(c.prog.BinWriter, int64(i))
|
2021-02-16 13:20:19 +00:00
|
|
|
ast.Walk(c, elems[i])
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.SETITEM)
|
2020-09-24 17:20:34 +00:00
|
|
|
}
|
2018-02-27 09:04:24 +00:00
|
|
|
}
|
|
|
|
|
2020-01-23 14:06:15 +00:00
|
|
|
func (c *codegen) convertMap(lit *ast.CompositeLit) {
|
2021-11-12 13:43:59 +00:00
|
|
|
l := len(lit.Elts)
|
|
|
|
for i := l - 1; i >= 0; i-- {
|
2020-01-23 14:06:15 +00:00
|
|
|
elem := lit.Elts[i].(*ast.KeyValueExpr)
|
|
|
|
ast.Walk(c, elem.Value)
|
2021-11-12 13:43:59 +00:00
|
|
|
ast.Walk(c, elem.Key)
|
2020-01-23 14:06:15 +00:00
|
|
|
}
|
2021-11-12 13:43:59 +00:00
|
|
|
emit.Int(c.prog.BinWriter, int64(l))
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PACKMAP)
|
2020-01-23 14:06:15 +00:00
|
|
|
}
|
|
|
|
|
2020-08-02 11:42:29 +00:00
|
|
|
func (c *codegen) getStruct(typ types.Type) (*types.Struct, bool) {
|
|
|
|
switch t := typ.Underlying().(type) {
|
|
|
|
case *types.Struct:
|
|
|
|
return t, true
|
|
|
|
case *types.Pointer:
|
|
|
|
strct, ok := t.Elem().Underlying().(*types.Struct)
|
|
|
|
return strct, ok
|
|
|
|
default:
|
|
|
|
return nil, false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *codegen) convertStruct(lit *ast.CompositeLit, ptr bool) {
|
2018-02-24 09:06:48 +00:00
|
|
|
// Create a new structScope to initialize and store
|
|
|
|
// the positions of its variables.
|
2020-05-13 15:09:55 +00:00
|
|
|
strct, ok := c.typeOf(lit).Underlying().(*types.Struct)
|
2018-02-24 09:06:48 +00:00
|
|
|
if !ok {
|
2019-11-22 14:16:52 +00:00
|
|
|
c.prog.Err = fmt.Errorf("the given literal is not of type struct: %v", lit)
|
|
|
|
return
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
|
|
|
|
2020-06-24 16:33:58 +00:00
|
|
|
keyedLit := len(lit.Elts) > 0
|
|
|
|
if keyedLit {
|
|
|
|
_, ok := lit.Elts[0].(*ast.KeyValueExpr)
|
|
|
|
keyedLit = keyedLit && ok
|
|
|
|
}
|
2018-02-24 09:06:48 +00:00
|
|
|
// We need to locally store all the fields, even if they are not initialized.
|
|
|
|
// We will initialize all fields to their "zero" value.
|
2021-11-12 13:13:41 +00:00
|
|
|
for i := strct.NumFields() - 1; i >= 0; i-- {
|
2018-02-27 09:04:24 +00:00
|
|
|
sField := strct.Field(i)
|
2020-07-09 09:26:16 +00:00
|
|
|
var initialized bool
|
2018-02-24 09:06:48 +00:00
|
|
|
|
2020-07-09 09:26:16 +00:00
|
|
|
if !keyedLit {
|
|
|
|
if len(lit.Elts) > i {
|
|
|
|
ast.Walk(c, lit.Elts[i])
|
|
|
|
initialized = true
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Fields initialized by the program.
|
|
|
|
for _, field := range lit.Elts {
|
|
|
|
f := field.(*ast.KeyValueExpr)
|
|
|
|
fieldName := f.Key.(*ast.Ident).Name
|
|
|
|
|
|
|
|
if sField.Name() == fieldName {
|
|
|
|
ast.Walk(c, f.Value)
|
|
|
|
initialized = true
|
|
|
|
break
|
|
|
|
}
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
|
|
|
}
|
2020-07-09 09:26:16 +00:00
|
|
|
if !initialized {
|
|
|
|
c.emitDefault(sField.Type())
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
2021-11-12 13:13:41 +00:00
|
|
|
}
|
|
|
|
emit.Int(c.prog.BinWriter, int64(strct.NumFields()))
|
|
|
|
if ptr {
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PACK)
|
|
|
|
} else {
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.PACKSTRUCT)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-08-23 09:39:58 +00:00
|
|
|
func (c *codegen) emitToken(tok token.Token, typ types.Type) {
|
|
|
|
op, err := convertToken(tok, typ)
|
|
|
|
if err != nil {
|
|
|
|
c.prog.Err = err
|
|
|
|
return
|
|
|
|
}
|
2020-10-02 08:30:15 +00:00
|
|
|
emit.Opcodes(c.prog.BinWriter, op)
|
2020-08-23 09:39:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func convertToken(tok token.Token, typ types.Type) (opcode.Opcode, error) {
|
2018-02-24 09:06:48 +00:00
|
|
|
switch tok {
|
2020-08-23 10:12:10 +00:00
|
|
|
case token.ADD_ASSIGN, token.ADD:
|
|
|
|
// VM has separate opcodes for number and string concatenation
|
|
|
|
if isString(typ) {
|
|
|
|
return opcode.CAT, nil
|
|
|
|
}
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.ADD, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.SUB_ASSIGN:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.SUB, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.MUL_ASSIGN:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.MUL, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.QUO_ASSIGN:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.DIV, nil
|
2019-12-19 10:11:05 +00:00
|
|
|
case token.REM_ASSIGN:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.MOD, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.SUB:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.SUB, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.MUL:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.MUL, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.QUO:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.DIV, nil
|
2019-12-19 10:11:05 +00:00
|
|
|
case token.REM:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.MOD, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.LSS:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.LT, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.LEQ:
|
2021-04-29 11:34:23 +00:00
|
|
|
return opcode.LE, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.GTR:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.GT, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
case token.GEQ:
|
2021-04-29 11:34:23 +00:00
|
|
|
return opcode.GE, nil
|
2018-04-02 15:04:42 +00:00
|
|
|
case token.EQL:
|
2020-08-23 09:39:58 +00:00
|
|
|
// VM has separate opcodes for number and string equality
|
|
|
|
if isNumber(typ) {
|
|
|
|
return opcode.NUMEQUAL, nil
|
|
|
|
}
|
|
|
|
return opcode.EQUAL, nil
|
2018-04-02 15:04:42 +00:00
|
|
|
case token.NEQ:
|
2020-08-23 09:39:58 +00:00
|
|
|
// VM has separate opcodes for number and string equality
|
|
|
|
if isNumber(typ) {
|
|
|
|
return opcode.NUMNOTEQUAL, nil
|
|
|
|
}
|
|
|
|
return opcode.NOTEQUAL, nil
|
2018-04-02 15:04:42 +00:00
|
|
|
case token.DEC:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.DEC, nil
|
2018-04-02 15:04:42 +00:00
|
|
|
case token.INC:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.INC, nil
|
2018-04-04 19:41:19 +00:00
|
|
|
case token.NOT:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.NOT, nil
|
2018-10-26 14:02:32 +00:00
|
|
|
case token.AND:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.AND, nil
|
2018-10-26 14:02:32 +00:00
|
|
|
case token.OR:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.OR, nil
|
2018-10-26 14:02:32 +00:00
|
|
|
case token.SHL:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.SHL, nil
|
2018-10-26 14:02:32 +00:00
|
|
|
case token.SHR:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.SHR, nil
|
2018-10-26 14:02:32 +00:00
|
|
|
case token.XOR:
|
2020-08-23 09:39:58 +00:00
|
|
|
return opcode.XOR, nil
|
2018-02-24 09:06:48 +00:00
|
|
|
default:
|
2020-08-23 09:39:58 +00:00
|
|
|
return 0, fmt.Errorf("compiler could not convert token: %s", tok)
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-02-19 09:24:28 +00:00
|
|
|
func (c *codegen) newFunc(decl *ast.FuncDecl) *funcScope {
|
2020-07-29 14:20:00 +00:00
|
|
|
f := c.newFuncScope(decl, c.newLabel())
|
|
|
|
c.funcs[c.getFuncNameFromDecl("", decl)] = f
|
2018-02-19 09:24:28 +00:00
|
|
|
return f
|
|
|
|
}
|
|
|
|
|
2021-05-25 13:17:40 +00:00
|
|
|
func (c *codegen) getFuncFromIdent(fun *ast.Ident) (*funcScope, bool) {
|
|
|
|
var pkgName string
|
|
|
|
if len(c.pkgInfoInline) != 0 {
|
2021-12-02 14:44:53 +00:00
|
|
|
pkgName = c.pkgInfoInline[len(c.pkgInfoInline)-1].PkgPath
|
2021-05-25 13:17:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
f, ok := c.funcs[c.getIdentName(pkgName, fun.Name)]
|
|
|
|
return f, ok
|
|
|
|
}
|
|
|
|
|
2020-07-29 14:20:00 +00:00
|
|
|
// getFuncNameFromSelector returns fully-qualified function name from the selector expression.
|
|
|
|
// Second return value is true iff this was a method call, not foreign package call.
|
|
|
|
func (c *codegen) getFuncNameFromSelector(e *ast.SelectorExpr) (string, bool) {
|
|
|
|
if c.typeInfo.Selections[e] != nil {
|
2022-07-11 16:28:15 +00:00
|
|
|
typ := c.typeInfo.Types[e.X].Type.String()
|
2022-07-06 14:52:57 +00:00
|
|
|
name := c.getIdentName(typ, e.Sel.Name)
|
|
|
|
if name[0] == '*' {
|
|
|
|
name = name[1:]
|
|
|
|
}
|
|
|
|
return name, true
|
2020-07-29 14:20:00 +00:00
|
|
|
}
|
2022-07-11 16:28:15 +00:00
|
|
|
|
|
|
|
ident := e.X.(*ast.Ident)
|
2020-07-29 14:20:00 +00:00
|
|
|
return c.getIdentName(ident.Name, e.Sel.Name), false
|
|
|
|
}
|
|
|
|
|
2020-05-06 15:10:20 +00:00
|
|
|
func (c *codegen) newLambda(u uint16, lit *ast.FuncLit) {
|
|
|
|
name := fmt.Sprintf("lambda@%d", u)
|
2020-07-29 14:20:00 +00:00
|
|
|
f := c.newFuncScope(&ast.FuncDecl{
|
2020-05-06 15:10:20 +00:00
|
|
|
Name: ast.NewIdent(name),
|
|
|
|
Type: lit.Type,
|
|
|
|
Body: lit.Body,
|
|
|
|
}, u)
|
2020-07-29 14:20:00 +00:00
|
|
|
c.lambda[c.getFuncNameFromDecl("", f.decl)] = f
|
2020-05-06 15:10:20 +00:00
|
|
|
}
|
|
|
|
|
2021-12-02 14:44:53 +00:00
|
|
|
func (c *codegen) compile(info *buildInfo, pkg *packages.Package) error {
|
2020-07-28 15:40:41 +00:00
|
|
|
c.mainPkg = pkg
|
2020-08-05 07:56:36 +00:00
|
|
|
c.analyzePkgOrder()
|
2020-08-10 10:10:35 +00:00
|
|
|
c.fillDocumentInfo()
|
2022-08-04 14:47:32 +00:00
|
|
|
funUsage := c.analyzeFuncAndGlobalVarUsage()
|
2022-07-14 12:34:57 +00:00
|
|
|
if c.prog.Err != nil {
|
|
|
|
return c.prog.Err
|
|
|
|
}
|
2018-02-27 09:04:24 +00:00
|
|
|
|
2019-10-22 14:56:03 +00:00
|
|
|
// Bring all imported functions into scope.
|
2020-07-28 15:40:41 +00:00
|
|
|
c.ForEachFile(c.resolveFuncDecls)
|
2018-02-19 09:24:28 +00:00
|
|
|
|
2021-04-28 13:10:44 +00:00
|
|
|
hasDeploy := c.traverseGlobals()
|
2020-10-02 10:16:02 +00:00
|
|
|
|
|
|
|
if hasDeploy {
|
2021-04-28 13:10:44 +00:00
|
|
|
deployOffset := c.prog.Len()
|
|
|
|
emit.Instruction(c.prog.BinWriter, opcode.INITSLOT, []byte{0, 2})
|
|
|
|
locCount := c.convertDeployFuncs()
|
|
|
|
c.reverseOffsetMap[deployOffset] = nameWithLocals{
|
|
|
|
name: "_deploy",
|
|
|
|
count: locCount,
|
|
|
|
}
|
2020-10-02 10:16:02 +00:00
|
|
|
c.deployEndOffset = c.prog.Len()
|
|
|
|
emit.Opcodes(c.prog.BinWriter, opcode.RET)
|
2020-07-24 10:40:54 +00:00
|
|
|
}
|
2018-02-25 12:26:56 +00:00
|
|
|
|
2019-10-22 14:56:03 +00:00
|
|
|
// sort map keys to generate code deterministically.
|
2021-12-02 14:44:53 +00:00
|
|
|
keys := make([]*types.Package, 0, len(info.program))
|
|
|
|
for _, p := range info.program {
|
|
|
|
keys = append(keys, p.Types)
|
2019-02-15 13:38:16 +00:00
|
|
|
}
|
|
|
|
sort.Slice(keys, func(i, j int) bool { return keys[i].Path() < keys[j].Path() })
|
|
|
|
|
2019-10-22 14:56:03 +00:00
|
|
|
// Generate the code for the program.
|
2020-07-28 15:40:41 +00:00
|
|
|
c.ForEachFile(func(f *ast.File, pkg *types.Package) {
|
|
|
|
for _, decl := range f.Decls {
|
|
|
|
switch n := decl.(type) {
|
|
|
|
case *ast.FuncDecl:
|
|
|
|
// Don't convert the function if it's not used. This will save a lot
|
|
|
|
// of bytecode space.
|
2021-03-04 11:36:15 +00:00
|
|
|
pkgPath := ""
|
2021-12-02 14:44:53 +00:00
|
|
|
if pkg != c.mainPkg.Types { // not a main package
|
2021-03-04 11:36:15 +00:00
|
|
|
pkgPath = pkg.Path()
|
|
|
|
}
|
|
|
|
name := c.getFuncNameFromDecl(pkgPath, n)
|
2021-02-08 08:06:01 +00:00
|
|
|
if !isInitFunc(n) && !isDeployFunc(n) && funUsage.funcUsed(name) &&
|
2022-09-16 16:18:47 +00:00
|
|
|
(!isInteropPath(pkg.Path()) && !canInline(pkg.Path(), n.Name.Name, false)) {
|
2020-07-28 15:40:41 +00:00
|
|
|
c.convertFuncDecl(f, n, pkg)
|
2018-02-24 09:06:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-07-28 15:40:41 +00:00
|
|
|
})
|
2018-02-24 09:06:48 +00:00
|
|
|
|
2020-03-31 13:16:32 +00:00
|
|
|
return c.prog.Err
|
|
|
|
}
|
|
|
|
|
2021-12-02 14:44:53 +00:00
|
|
|
func newCodegen(info *buildInfo, pkg *packages.Package) *codegen {
|
2020-03-31 13:16:32 +00:00
|
|
|
return &codegen{
|
2021-04-28 13:10:44 +00:00
|
|
|
buildInfo: info,
|
|
|
|
prog: io.NewBufBinWriter(),
|
|
|
|
l: []int{},
|
|
|
|
funcs: map[string]*funcScope{},
|
|
|
|
lambda: map[string]*funcScope{},
|
|
|
|
reverseOffsetMap: map[int]nameWithLocals{},
|
|
|
|
globals: map[string]int{},
|
|
|
|
labels: map[labelWithType]uint16{},
|
2021-12-02 14:44:53 +00:00
|
|
|
typeInfo: pkg.TypesInfo,
|
2021-04-28 13:10:44 +00:00
|
|
|
constMap: map[string]types.TypeAndValue{},
|
|
|
|
docIndex: map[string]int{},
|
2021-12-02 14:44:53 +00:00
|
|
|
packageCache: map[string]*packages.Package{},
|
2020-03-31 13:57:35 +00:00
|
|
|
|
2020-10-02 10:16:02 +00:00
|
|
|
initEndOffset: -1,
|
|
|
|
deployEndOffset: -1,
|
|
|
|
|
2023-05-24 08:52:14 +00:00
|
|
|
emittedEvents: make(map[string][]EmittedEventInfo),
|
2021-06-24 15:36:40 +00:00
|
|
|
invokedContracts: make(map[util.Uint160][]string),
|
|
|
|
sequencePoints: make(map[string][]DebugSeqPoint),
|
2020-03-31 13:16:32 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-08 19:33:03 +00:00
|
|
|
// codeGen compiles the program to bytecode.
|
|
|
|
func codeGen(info *buildInfo) (*nef.File, *DebugInfo, error) {
|
compiler: avoid panic on empty package list
An attempt to compile unexisting file via neo-go-vm CLI leads to the following
panic:
```
NEO-GO-VM > loadgo ./1-print.go
panic: runtime error: index out of range [0] with length 0
goroutine 1 [running]:
github.com/nspcc-dev/neo-go/pkg/compiler.codeGen(0xc000047300)
github.com/nspcc-dev/neo-go/pkg/compiler/codegen.go:2188 +0x60c
github.com/nspcc-dev/neo-go/pkg/compiler.CompileWithOptions({0xc000047300, 0xf66c92}, {0x0, 0x0}, 0xc0003a0000)
github.com/nspcc-dev/neo-go/pkg/compiler/compiler.go:218 +0x65
github.com/nspcc-dev/neo-go/pkg/vm/cli.handleLoadGo(0xc0000f5340)
github.com/nspcc-dev/neo-go/pkg/vm/cli/cli.go:480 +0x1a5
github.com/urfave/cli.HandleAction({0xde74a0, 0x10d61d0}, 0x6)
github.com/urfave/cli@v1.22.5/app.go:524 +0xa8
github.com/urfave/cli.Command.Run({{0xf632f0, 0x6}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0xf9915b, 0x38}, {0xf6a57b, ...}, ...}, ...)
github.com/urfave/cli@v1.22.5/command.go:173 +0x652
github.com/urfave/cli.(*App).Run(0xc0002c2000, {0xc00036c330, 0x3, 0x3})
github.com/urfave/cli@v1.22.5/app.go:277 +0x705
github.com/nspcc-dev/neo-go/pkg/vm/cli.(*VMCLI).Run(0xc000290890)
github.com/nspcc-dev/neo-go/pkg/vm/cli/cli.go:694 +0x317
github.com/nspcc-dev/neo-go/cli/vm.startVMPrompt(0xc0001b9e40)
github.com/nspcc-dev/neo-go/cli/vm/vm.go:29 +0x92
github.com/urfave/cli.HandleAction({0xde74a0, 0x10d5d10}, 0x2)
github.com/urfave/cli@v1.22.5/app.go:524 +0xa8
github.com/urfave/cli.Command.Run({{0xf60a2f, 0x2}, {0x0, 0x0}, {0x0, 0x0, 0x0}, {0xf7880b, 0x19}, {0x0, ...}, ...}, ...)
github.com/urfave/cli@v1.22.5/command.go:173 +0x652
github.com/urfave/cli.(*App).Run(0xc0001c8fc0, {0xc000116020, 0x2, 0x2})
github.com/urfave/cli@v1.22.5/app.go:277 +0x705
main.main()
./main.go:21 +0x33
```
2022-08-16 14:08:49 +00:00
|
|
|
if len(info.program) == 0 {
|
|
|
|
return nil, nil, errors.New("empty package")
|
|
|
|
}
|
2021-12-02 14:44:53 +00:00
|
|
|
pkg := info.program[0]
|
2020-03-31 13:16:32 +00:00
|
|
|
c := newCodegen(info, pkg)
|
|
|
|
|
|
|
|
if err := c.compile(info, pkg); err != nil {
|
2020-04-02 12:38:53 +00:00
|
|
|
return nil, nil, err
|
2019-11-22 10:06:32 +00:00
|
|
|
}
|
2020-03-31 13:16:32 +00:00
|
|
|
|
2020-08-19 09:36:17 +00:00
|
|
|
buf, err := c.writeJumps(c.prog.Bytes())
|
|
|
|
if err != nil {
|
2020-04-02 12:38:53 +00:00
|
|
|
return nil, nil, err
|
2020-02-21 08:57:24 +00:00
|
|
|
}
|
2021-10-25 10:54:18 +00:00
|
|
|
|
|
|
|
methods := bitfield.New(len(buf))
|
|
|
|
di := c.emitDebugInfo(buf)
|
|
|
|
for i := range di.Methods {
|
|
|
|
methods.Set(int(di.Methods[i].Range.Start))
|
|
|
|
}
|
2021-12-08 19:33:03 +00:00
|
|
|
f, err := nef.NewFile(buf)
|
|
|
|
if err != nil {
|
|
|
|
return nil, nil, fmt.Errorf("error while trying to create .nef file: %w", err)
|
|
|
|
}
|
|
|
|
if c.callTokens != nil {
|
|
|
|
f.Tokens = c.callTokens
|
|
|
|
}
|
|
|
|
f.Checksum = f.CalculateChecksum()
|
|
|
|
return f, di, vm.IsScriptCorrect(buf, methods)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
|
2020-06-08 11:16:41 +00:00
|
|
|
func (c *codegen) resolveFuncDecls(f *ast.File, pkg *types.Package) {
|
2018-02-19 09:24:28 +00:00
|
|
|
for _, decl := range f.Decls {
|
|
|
|
switch n := decl.(type) {
|
|
|
|
case *ast.FuncDecl:
|
2021-02-04 12:41:00 +00:00
|
|
|
fs := c.newFunc(n)
|
|
|
|
fs.file = f
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-08-19 09:36:17 +00:00
|
|
|
func (c *codegen) writeJumps(b []byte) ([]byte, error) {
|
2020-02-03 09:10:07 +00:00
|
|
|
ctx := vm.NewContext(b)
|
2022-07-12 10:16:32 +00:00
|
|
|
var nopOffsets []int
|
2021-08-13 08:58:21 +00:00
|
|
|
for op, param, err := ctx.Next(); err == nil && ctx.IP() < len(b); op, param, err = ctx.Next() {
|
2020-02-03 09:10:07 +00:00
|
|
|
switch op {
|
2020-04-23 08:56:36 +00:00
|
|
|
case opcode.JMP, opcode.JMPIFNOT, opcode.JMPIF, opcode.CALL,
|
|
|
|
opcode.JMPEQ, opcode.JMPNE,
|
|
|
|
opcode.JMPGT, opcode.JMPGE, opcode.JMPLE, opcode.JMPLT:
|
2020-08-20 05:34:14 +00:00
|
|
|
case opcode.TRYL:
|
2021-08-13 08:58:21 +00:00
|
|
|
_, err := c.replaceLabelWithOffset(ctx.IP(), param)
|
2020-08-21 12:37:46 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2021-08-13 08:58:21 +00:00
|
|
|
_, err = c.replaceLabelWithOffset(ctx.IP(), param[4:])
|
2020-08-20 05:34:14 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-04-23 08:56:36 +00:00
|
|
|
case opcode.JMPL, opcode.JMPIFL, opcode.JMPIFNOTL,
|
|
|
|
opcode.JMPEQL, opcode.JMPNEL,
|
|
|
|
opcode.JMPGTL, opcode.JMPGEL, opcode.JMPLEL, opcode.JMPLTL,
|
2020-08-20 05:34:14 +00:00
|
|
|
opcode.CALLL, opcode.PUSHA, opcode.ENDTRYL:
|
2021-08-13 08:58:21 +00:00
|
|
|
offset, err := c.replaceLabelWithOffset(ctx.IP(), param)
|
2020-08-20 05:34:14 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
2020-08-19 09:36:17 +00:00
|
|
|
if op != opcode.PUSHA && math.MinInt8 <= offset && offset <= math.MaxInt8 {
|
2022-07-12 08:44:32 +00:00
|
|
|
if op == opcode.JMPL && offset == 5 {
|
|
|
|
copy(b[ctx.IP():], []byte{byte(opcode.NOP), byte(opcode.NOP), byte(opcode.NOP), byte(opcode.NOP), byte(opcode.NOP)})
|
|
|
|
nopOffsets = append(nopOffsets, ctx.IP(), ctx.IP()+1, ctx.IP()+2, ctx.IP()+3, ctx.IP()+4)
|
|
|
|
} else {
|
|
|
|
copy(b[ctx.IP():], []byte{byte(toShortForm(op)), byte(offset), byte(opcode.NOP), byte(opcode.NOP), byte(opcode.NOP)})
|
|
|
|
nopOffsets = append(nopOffsets, ctx.IP()+2, ctx.IP()+3, ctx.IP()+4)
|
|
|
|
}
|
2020-08-19 09:36:17 +00:00
|
|
|
}
|
2021-04-28 13:10:44 +00:00
|
|
|
case opcode.INITSLOT:
|
|
|
|
nextIP := ctx.NextIP()
|
|
|
|
info := c.reverseOffsetMap[ctx.IP()]
|
|
|
|
if argCount := b[nextIP-1]; info.count == 0 && argCount == 0 {
|
2022-07-12 10:16:32 +00:00
|
|
|
copy(b[ctx.IP():], []byte{byte(opcode.NOP), byte(opcode.NOP), byte(opcode.NOP)})
|
|
|
|
nopOffsets = append(nopOffsets, ctx.IP(), ctx.IP()+1, ctx.IP()+2)
|
2021-04-28 13:10:44 +00:00
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
if info.count > 255 {
|
|
|
|
return nil, fmt.Errorf("func '%s' has %d local variables (maximum is 255)", info.name, info.count)
|
|
|
|
}
|
|
|
|
b[nextIP-2] = byte(info.count)
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|
|
|
|
}
|
2021-02-11 12:41:49 +00:00
|
|
|
|
|
|
|
if c.deployEndOffset >= 0 {
|
2022-07-12 10:16:32 +00:00
|
|
|
_, end := correctRange(uint16(c.initEndOffset+1), uint16(c.deployEndOffset), nopOffsets)
|
2021-02-11 12:41:49 +00:00
|
|
|
c.deployEndOffset = int(end)
|
|
|
|
}
|
|
|
|
if c.initEndOffset > 0 {
|
2022-07-12 10:16:32 +00:00
|
|
|
_, end := correctRange(0, uint16(c.initEndOffset), nopOffsets)
|
2021-02-11 12:41:49 +00:00
|
|
|
c.initEndOffset = int(end)
|
|
|
|
}
|
|
|
|
|
2020-08-19 09:36:17 +00:00
|
|
|
// Correct function ip range.
|
|
|
|
// Note: indices are sorted in increasing order.
|
|
|
|
for _, f := range c.funcs {
|
2022-07-12 10:16:32 +00:00
|
|
|
f.rng.Start, f.rng.End = correctRange(f.rng.Start, f.rng.End, nopOffsets)
|
2020-08-19 09:36:17 +00:00
|
|
|
}
|
2022-07-12 10:16:32 +00:00
|
|
|
return removeNOPs(b, nopOffsets), nil
|
2020-08-19 09:36:17 +00:00
|
|
|
}
|
|
|
|
|
2021-02-11 12:41:49 +00:00
|
|
|
func correctRange(start, end uint16, offsets []int) (uint16, uint16) {
|
|
|
|
newStart, newEnd := start, end
|
|
|
|
loop:
|
|
|
|
for _, ind := range offsets {
|
|
|
|
switch {
|
|
|
|
case ind > int(end):
|
|
|
|
break loop
|
|
|
|
case ind < int(start):
|
2022-07-12 10:16:32 +00:00
|
|
|
newStart--
|
|
|
|
newEnd--
|
2021-02-11 12:41:49 +00:00
|
|
|
case ind >= int(start):
|
2022-07-12 10:16:32 +00:00
|
|
|
newEnd--
|
2021-02-11 12:41:49 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return newStart, newEnd
|
|
|
|
}
|
|
|
|
|
2020-08-20 05:34:14 +00:00
|
|
|
func (c *codegen) replaceLabelWithOffset(ip int, arg []byte) (int, error) {
|
|
|
|
index := binary.LittleEndian.Uint16(arg)
|
|
|
|
if int(index) > len(c.l) {
|
|
|
|
return 0, fmt.Errorf("unexpected label number: %d (max %d)", index, len(c.l))
|
|
|
|
}
|
2021-10-25 07:51:50 +00:00
|
|
|
if c.l[index] < 0 {
|
|
|
|
return 0, fmt.Errorf("invalid label target: %d at %d", c.l[index], ip)
|
|
|
|
}
|
2020-08-20 05:34:14 +00:00
|
|
|
offset := c.l[index] - ip
|
|
|
|
if offset > math.MaxInt32 || offset < math.MinInt32 {
|
|
|
|
return 0, fmt.Errorf("label offset is too big at the instruction %d: %d (max %d, min %d)",
|
|
|
|
ip, offset, math.MaxInt32, math.MinInt32)
|
|
|
|
}
|
|
|
|
binary.LittleEndian.PutUint32(arg, uint32(offset))
|
|
|
|
return offset, nil
|
|
|
|
}
|
|
|
|
|
2022-07-12 10:16:32 +00:00
|
|
|
// removeNOPs converts b to a program where all long JMP*/CALL* specified by absolute offsets
|
2020-08-19 09:36:17 +00:00
|
|
|
// are replaced with their corresponding short counterparts. It panics if either b or offsets are invalid.
|
|
|
|
// This is done in 2 passes:
|
|
|
|
// 1. Alter jump offsets taking into account parts to be removed.
|
|
|
|
// 2. Perform actual removal of jump targets.
|
|
|
|
// Note: after jump offsets altering, there can appear new candidates for conversion.
|
|
|
|
// These are ignored for now.
|
2022-07-12 10:16:32 +00:00
|
|
|
func removeNOPs(b []byte, nopOffsets []int) []byte {
|
|
|
|
if len(nopOffsets) == 0 {
|
2020-08-19 09:36:17 +00:00
|
|
|
return b
|
|
|
|
}
|
2022-07-12 10:16:32 +00:00
|
|
|
for i := range nopOffsets {
|
|
|
|
if b[nopOffsets[i]] != byte(opcode.NOP) {
|
|
|
|
panic("NOP offset is invalid")
|
|
|
|
}
|
|
|
|
}
|
2020-08-19 09:36:17 +00:00
|
|
|
|
|
|
|
// 1. Alter existing jump offsets.
|
|
|
|
ctx := vm.NewContext(b)
|
|
|
|
for op, _, err := ctx.Next(); err == nil && ctx.IP() < len(b); op, _, err = ctx.Next() {
|
|
|
|
// we can't use arg returned by ctx.Next() because it is copied
|
|
|
|
nextIP := ctx.NextIP()
|
|
|
|
ip := ctx.IP()
|
|
|
|
switch op {
|
|
|
|
case opcode.JMP, opcode.JMPIFNOT, opcode.JMPIF, opcode.CALL,
|
|
|
|
opcode.JMPEQ, opcode.JMPNE,
|
2020-08-20 05:34:14 +00:00
|
|
|
opcode.JMPGT, opcode.JMPGE, opcode.JMPLE, opcode.JMPLT, opcode.ENDTRY:
|
2020-08-19 09:36:17 +00:00
|
|
|
offset := int(int8(b[nextIP-1]))
|
2022-07-12 10:16:32 +00:00
|
|
|
offset += calcOffsetCorrection(ip, ip+offset, nopOffsets)
|
2020-08-19 09:36:17 +00:00
|
|
|
b[nextIP-1] = byte(offset)
|
2020-08-20 05:34:14 +00:00
|
|
|
case opcode.TRY:
|
2020-08-21 12:37:46 +00:00
|
|
|
catchOffset := int(int8(b[nextIP-2]))
|
2022-07-12 10:16:32 +00:00
|
|
|
catchOffset += calcOffsetCorrection(ip, ip+catchOffset, nopOffsets)
|
2020-08-21 12:37:46 +00:00
|
|
|
b[nextIP-1] = byte(catchOffset)
|
2020-08-20 05:34:14 +00:00
|
|
|
finallyOffset := int(int8(b[nextIP-1]))
|
2022-07-12 10:16:32 +00:00
|
|
|
finallyOffset += calcOffsetCorrection(ip, ip+finallyOffset, nopOffsets)
|
2020-08-20 05:34:14 +00:00
|
|
|
b[nextIP-1] = byte(finallyOffset)
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPL, opcode.JMPIFL, opcode.JMPIFNOTL,
|
|
|
|
opcode.JMPEQL, opcode.JMPNEL,
|
|
|
|
opcode.JMPGTL, opcode.JMPGEL, opcode.JMPLEL, opcode.JMPLTL,
|
2020-08-20 05:34:14 +00:00
|
|
|
opcode.CALLL, opcode.PUSHA, opcode.ENDTRYL:
|
2020-08-19 09:36:17 +00:00
|
|
|
arg := b[nextIP-4:]
|
|
|
|
offset := int(int32(binary.LittleEndian.Uint32(arg)))
|
2022-07-12 10:16:32 +00:00
|
|
|
offset += calcOffsetCorrection(ip, ip+offset, nopOffsets)
|
2020-08-19 09:36:17 +00:00
|
|
|
binary.LittleEndian.PutUint32(arg, uint32(offset))
|
2020-08-20 05:34:14 +00:00
|
|
|
case opcode.TRYL:
|
2020-08-21 12:37:46 +00:00
|
|
|
arg := b[nextIP-8:]
|
|
|
|
catchOffset := int(int32(binary.LittleEndian.Uint32(arg)))
|
2022-07-12 10:16:32 +00:00
|
|
|
catchOffset += calcOffsetCorrection(ip, ip+catchOffset, nopOffsets)
|
2020-08-21 12:37:46 +00:00
|
|
|
binary.LittleEndian.PutUint32(arg, uint32(catchOffset))
|
|
|
|
arg = b[nextIP-4:]
|
2020-08-20 05:34:14 +00:00
|
|
|
finallyOffset := int(int32(binary.LittleEndian.Uint32(arg)))
|
2022-07-12 10:16:32 +00:00
|
|
|
finallyOffset += calcOffsetCorrection(ip, ip+finallyOffset, nopOffsets)
|
2020-08-20 05:34:14 +00:00
|
|
|
binary.LittleEndian.PutUint32(arg, uint32(finallyOffset))
|
2020-08-19 09:36:17 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// 2. Convert instructions.
|
|
|
|
copyOffset := 0
|
2022-07-12 10:16:32 +00:00
|
|
|
l := len(nopOffsets)
|
2020-08-19 09:36:17 +00:00
|
|
|
for i := 0; i < l; i++ {
|
2022-07-12 10:16:32 +00:00
|
|
|
start := nopOffsets[i]
|
2020-08-19 09:36:17 +00:00
|
|
|
end := len(b)
|
|
|
|
if i != l-1 {
|
2022-07-12 10:16:32 +00:00
|
|
|
end = nopOffsets[i+1]
|
2020-08-19 09:36:17 +00:00
|
|
|
}
|
2022-07-12 10:16:32 +00:00
|
|
|
copy(b[start-copyOffset:], b[start+1:end])
|
|
|
|
copyOffset++
|
2020-08-19 09:36:17 +00:00
|
|
|
}
|
|
|
|
return b[:len(b)-copyOffset]
|
|
|
|
}
|
|
|
|
|
|
|
|
func calcOffsetCorrection(ip, target int, offsets []int) int {
|
|
|
|
cnt := 0
|
|
|
|
start := sort.Search(len(offsets), func(i int) bool {
|
|
|
|
return offsets[i] >= ip || offsets[i] >= target
|
|
|
|
})
|
|
|
|
for i := start; i < len(offsets) && (offsets[i] < target || offsets[i] <= ip); i++ {
|
|
|
|
ind := offsets[i]
|
2022-07-12 10:16:32 +00:00
|
|
|
if ip <= ind && ind < target || target <= ind && ind < ip {
|
|
|
|
cnt++
|
2020-08-19 09:36:17 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if ip < target {
|
|
|
|
return -cnt
|
|
|
|
}
|
|
|
|
return cnt
|
|
|
|
}
|
|
|
|
|
2020-08-23 15:00:23 +00:00
|
|
|
func negateJmp(op opcode.Opcode) opcode.Opcode {
|
|
|
|
switch op {
|
|
|
|
case opcode.JMPIFL:
|
|
|
|
return opcode.JMPIFNOTL
|
|
|
|
case opcode.JMPIFNOTL:
|
|
|
|
return opcode.JMPIFL
|
|
|
|
case opcode.JMPEQL:
|
|
|
|
return opcode.JMPNEL
|
|
|
|
case opcode.JMPNEL:
|
|
|
|
return opcode.JMPEQL
|
|
|
|
case opcode.JMPGTL:
|
|
|
|
return opcode.JMPLEL
|
|
|
|
case opcode.JMPGEL:
|
|
|
|
return opcode.JMPLTL
|
|
|
|
case opcode.JMPLEL:
|
|
|
|
return opcode.JMPGTL
|
|
|
|
case opcode.JMPLTL:
|
|
|
|
return opcode.JMPGEL
|
|
|
|
default:
|
|
|
|
panic(fmt.Errorf("invalid opcode in negateJmp: %s", op))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-08-23 10:56:55 +00:00
|
|
|
func toShortForm(op opcode.Opcode) opcode.Opcode {
|
|
|
|
switch op {
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMP
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPIFL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMPIF
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPIFNOTL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMPIFNOT
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPEQL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMPEQ
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPNEL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMPNE
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPGTL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMPGT
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPGEL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMPGE
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPLEL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMPLE
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.JMPLTL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.JMPLT
|
2020-08-19 09:36:17 +00:00
|
|
|
case opcode.CALLL:
|
2020-08-23 10:56:55 +00:00
|
|
|
return opcode.CALL
|
2020-08-20 05:34:14 +00:00
|
|
|
case opcode.ENDTRYL:
|
|
|
|
return opcode.ENDTRY
|
2020-08-19 09:36:17 +00:00
|
|
|
default:
|
|
|
|
panic(fmt.Errorf("invalid opcode: %s", op))
|
|
|
|
}
|
2018-02-19 09:24:28 +00:00
|
|
|
}
|