core/xgo
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

use atomic.Load/Store to avoid data race

Browse source
This commit is contained in:
Daniel Kang 2019-01-20 06:33:22 -08:00
parent 04bb0f5e62
commit 88dd0224d5
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
runtime/vm.go
View file

@ -3,6 +3,7 @@ package runtime
import (
"errors"
"fmt"
"sync/atomic"
"github.com/d5/tengo/compiler"
"github.com/d5/tengo/compiler/token"
@ -38,7 +39,7 @@ type VM struct {
curFrame *Frame
curInsts []byte
curIPLimit int
aborting bool
aborting int64
}
// NewVM creates a VM.
@ -72,14 +73,14 @@ func NewVM(bytecode *compiler.Bytecode, globals []*objects.Object) *VM {
// Abort aborts the execution.
func (v *VM) Abort() {
v.aborting = true
atomic.StoreInt64(&v.aborting, 1)
}
// Run starts the execution.
func (v *VM) Run() error {
var ip int
for v.curFrame.ip < v.curIPLimit && !v.aborting {
for v.curFrame.ip < v.curIPLimit && (atomic.LoadInt64(&v.aborting) == 0) {
v.curFrame.ip++
ip = v.curFrame.ip
@ -1120,7 +1121,7 @@ func (v *VM) Run() error {
}
// check if stack still has some objects left
if v.sp > 0 && !v.aborting {
if v.sp > 0 && atomic.LoadInt64(&v.aborting) == 0 {
return fmt.Errorf("non empty stack after execution")
}