forked from TrueCloudLab/neoneo-go
33 lines
732 B
Go
33 lines
732 B
Go
|
package vm
|
||
|
|
|
||
|
|
import "fmt"
|
||
|
|
|
||
|
|
// InteropFunc allows to hook into the VM.
|
||
|
|
type InteropFunc func(vm *VM) error
|
||
|
|
|
||
|
|
// InteropService
|
||
|
|
type InteropService struct {
|
||
|
|
mapping map[string]InteropFunc
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewInteropService returns a new InteropService object.
|
||
|
|
func NewInteropService() *InteropService {
|
||
|
|
return &InteropService{
|
||
|
|
mapping: map[string]InteropFunc{},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Register any API to the interop service.
|
||
|
|
func (i *InteropService) Register(api string, fun InteropFunc) {
|
||
|
|
i.mapping[api] = fun
|
||
|
|
}
|
||
|
|
|
||
|
|
// Call will invoke the service mapped to the given api.
|
||
|
|
func (i *InteropService) Call(api []byte, vm *VM) error {
|
||
|
|
fun, ok := i.mapping[string(api)]
|
||
|
|
if !ok {
|
||
|
|
return fmt.Errorf("api (%s) not in interop mapping", api)
|
||
|
|
}
|
||
|
|
return fun(vm)
|
||
|
|
}
|