PoC: WebAssembly API

cmd/check/main.go

@@ -237,7 +237,10 @@ func runPath(
 	location := common.NewStringLocation(nil, path)
 
 	// standard library handler is only needed for execution, but we're only checking
-	standardLibraryValues := stdlib.DefaultScriptStandardLibraryValues(nil)
+	standardLibraryValues := stdlib.DefaultScriptStandardLibraryValues(
+		nil,
+		stdlib.DefaultStandardLibraryOptions,
+	)
 
 	func() {
 		defer func() {

cmd/cmd.go

@@ -171,6 +171,7 @@ func PrepareInterpreter(filename string, debugger *interpreter.Debugger) (*inter
 
 	standardLibraryValues := stdlib.DefaultScriptStandardLibraryValues(
 		&StandardLibraryHandler{},
+		stdlib.DefaultStandardLibraryOptions,
 	)
 
 	checker, must := PrepareChecker(
@@ -403,6 +404,10 @@ func (*StandardLibraryHandler) BLSAggregateSignatures(_ [][]byte) ([]byte, error
 	return nil, goerrors.New("crypto functionality is not available in this environment")
 }
 
+func (*StandardLibraryHandler) CompileWebAssembly(_ []byte) (stdlib.WebAssemblyModule, error) {
+	return nil, goerrors.New("WebAssembly functionality is not available in this environment")
+}
+
 func (h *StandardLibraryHandler) NewOnEventEmittedHandler() interpreter.OnEventEmittedFunc {
 	return func(
 		inter *interpreter.Interpreter,

cmd/compile/main.go

@@ -46,7 +46,10 @@ func main() {
 	program, must := cmd.PrepareProgramFromFile(location, codes)
 
 	// standard library handler is only needed for execution, but we're only checking
-	standardLibraryValues := stdlib.DefaultScriptStandardLibraryValues(nil)
+	standardLibraryValues := stdlib.DefaultScriptStandardLibraryValues(
+		nil,
+		stdlib.DefaultStandardLibraryOptions,
+	)
 
 	checker, must := cmd.PrepareChecker(
 		program,

cmd/info/main.go

@@ -240,7 +240,10 @@ func dumpBuiltinValues() {
 	}
 
 	allBaseSemaValueTypes := sema_utils.AllBaseSemaValueTypes()
-	standardLibraryValues := stdlib.DefaultScriptStandardLibraryValues(nil)
+	standardLibraryValues := stdlib.DefaultScriptStandardLibraryValues(
+		nil,
+		stdlib.DefaultStandardLibraryOptions,
+	)
 
 	valueTypes := make([]valueType, 0, len(allBaseSemaValueTypes)+len(standardLibraryValues))
 

common/computationkind.go

@@ -142,4 +142,8 @@ const (
 	// RLP
 	ComputationKindSTDLIBRLPDecodeString
 	ComputationKindSTDLIBRLPDecodeList
+	_
+	_
+	// WebAssembly
+	ComputationKindWebAssemblyFuel
 )

go.mod

@@ -4,7 +4,6 @@ go 1.22
 
 require (
 	github.com/bits-and-blooms/bitset v1.5.0
-	github.com/bytecodealliance/wasmtime-go/v7 v7.0.0
 	github.com/c-bata/go-prompt v0.2.6
 	github.com/dave/dst v0.27.2
 	github.com/fxamacker/cbor/v2 v2.4.1-0.20230228173756-c0c9f774e40c
@@ -31,6 +30,7 @@ require (
 
 require (
 	github.com/SaveTheRbtz/mph v0.1.1-0.20240117162131-4166ec7869bc
+	github.com/bytecodealliance/wasmtime-go/v25 v25.0.0
 	github.com/k0kubun/pp v3.0.1+incompatible
 	github.com/kodova/html-to-markdown v1.0.1
 	github.com/onflow/crypto v0.25.0

go.sum

@@ -5,8 +5,8 @@ github.com/ajstarks/svgo v0.0.0-20180226025133-644b8db467af/go.mod h1:K08gAheRH3
 github.com/andybalholm/cascadia v1.0.0/go.mod h1:GsXiBklL0woXo1j/WYWtSYYC4ouU9PqHO0sqidkEA4Y=
 github.com/bits-and-blooms/bitset v1.5.0 h1:NpE8frKRLGHIcEzkR+gZhiioW1+WbYV6fKwD6ZIpQT8=
 github.com/bits-and-blooms/bitset v1.5.0/go.mod h1:gIdJ4wp64HaoK2YrL1Q5/N7Y16edYb8uY+O0FJTyyDA=
-github.com/bytecodealliance/wasmtime-go/v7 v7.0.0 h1:/rBNjgFju2HCZnkPb1eL+W4GBwP8DMbaQu7i+GR9DH4=
-github.com/bytecodealliance/wasmtime-go/v7 v7.0.0/go.mod h1:bu6fic7trDt20w+LMooX7j3fsOwv4/ln6j8gAdP6vmA=
+github.com/bytecodealliance/wasmtime-go/v25 v25.0.0 h1:ZTn4Ho+srrk0466ugqPfTDCITczsWdT48A0ZMA/TpRU=
+github.com/bytecodealliance/wasmtime-go/v25 v25.0.0/go.mod h1:8mMIYQ92CpVDwXPIb6udnhtFGI3vDZ/937cGeQr5I68=
 github.com/c-bata/go-prompt v0.2.6 h1:POP+nrHE+DfLYx370bedwNhsqmpCUynWPxuHi0C5vZI=
 github.com/c-bata/go-prompt v0.2.6/go.mod h1:/LMAke8wD2FsNu9EXNdHxNLbd9MedkPnCdfpU9wwHfY=
 github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=

interpreter/config.go

@@ -35,6 +35,8 @@ type Config struct {
 	OnResourceOwnerChange OnResourceOwnerChangeFunc
 	// OnMeterComputation is triggered when a computation is about to happen
 	OnMeterComputation OnMeterComputationFunc
+	// OnComputationRemaining is used to determine how much computation is remaining
+	OnComputationRemaining OnComputationRemainingFunc
 	// InjectedCompositeFieldsHandler is used to initialize new composite values' fields
 	InjectedCompositeFieldsHandler InjectedCompositeFieldsHandlerFunc
 	// ContractValueHandler is used to handle imports of values

interpreter/interpreter.go

@@ -106,6 +106,11 @@ type OnMeterComputationFunc func(
 	intensity uint,
 )
 
+// OnComputationRemainingFunc is a function that is used to determine how much computation is remaining.
+type OnComputationRemainingFunc func(
+	kind common.ComputationKind,
+) uint
+
 // CapabilityBorrowHandlerFunc is a function that is used to borrow ID capabilities.
 type CapabilityBorrowHandlerFunc func(
 	inter *Interpreter,
@@ -4996,6 +5001,17 @@ func (interpreter *Interpreter) ReportComputation(compKind common.ComputationKin
 	}
 }
 
+func (interpreter *Interpreter) ComputationRemaining(compKind common.ComputationKind) uint {
+	config := interpreter.SharedState.Config
+
+	onComputationRemaining := config.OnComputationRemaining
+	if onComputationRemaining != nil {
+		return onComputationRemaining(compKind)
+	}
+
+	return math.MaxUint
+}
+
 func (interpreter *Interpreter) getAccessOfMember(self Value, identifier string) sema.Access {
 	typ, err := interpreter.ConvertStaticToSemaType(self.StaticType(interpreter))
 	// some values (like transactions) do not have types that can be looked up this way. These types

interpreter/memory_metering_test.go

@@ -1033,7 +1033,10 @@ func TestInterpretHostFunctionMetering(t *testing.T) {
 
 		baseValueActivation := sema.NewVariableActivation(sema.BaseValueActivation)
 		baseActivation := activations.NewActivation(nil, interpreter.BaseActivation)
-		for _, valueDeclaration := range stdlib.DefaultStandardLibraryValues(nil) {
+		for _, valueDeclaration := range stdlib.DefaultStandardLibraryValues(
+			nil,
+			stdlib.DefaultStandardLibraryOptions,
+		) {
 			baseValueActivation.DeclareValue(valueDeclaration)
 			interpreter.Declare(baseActivation, valueDeclaration)
 		}

runtime/account_test.go

@@ -630,10 +630,10 @@ func TestRuntimeAuthAccountKeysAdd(t *testing.T) {
 	err := rt.ExecuteTransaction(
 		Script{
 			Source: []byte(code),
-			Arguments: encodeArgs([]cadence.Value{
+			Arguments: encodeArgs(
 				pubKey1Value,
 				pubKey2Value,
-			}),
+			),
 		},
 		Context{
 			Location:  nextTransactionLocation(),
@@ -1370,7 +1370,7 @@ func addPublicKeyValidation(runtimeInterface *TestRuntimeInterface, returnError
 	}
 }
 
-func encodeArgs(argValues []cadence.Value) [][]byte {
+func encodeArgs(argValues ...cadence.Value) [][]byte {
 	args := make([][]byte, len(argValues))
 	for i, arg := range argValues {
 		var err error
@@ -1393,7 +1393,7 @@ func (test accountKeyTestCase) executeTransaction(
 	runtimeInterface *TestRuntimeInterface,
 	location Location,
 ) error {
-	args := encodeArgs(test.args)
+	args := encodeArgs(test.args...)
 
 	err := runtime.ExecuteTransaction(
 		Script{
@@ -1413,7 +1413,7 @@ func (test accountKeyTestCase) executeScript(
 	runtimeInterface *TestRuntimeInterface,
 ) (cadence.Value, error) {
 
-	args := encodeArgs(test.args)
+	args := encodeArgs(test.args...)
 
 	value, err := runtime.ExecuteScript(
 		Script{

runtime/config.go

@@ -37,4 +37,6 @@ type Config struct {
 	CoverageReport *CoverageReport
 	// LegacyContractUpgradeEnabled enabled specifies whether to use the old parser when parsing an old contract
 	LegacyContractUpgradeEnabled bool
+	// WebAssemblyEnabled specifies if the WebAssembly API is enabled
+	WebAssemblyEnabled bool
 }

runtime/contract_update_test.go

@@ -385,10 +385,10 @@ func TestRuntimeContractRedeployInSameTransaction(t *testing.T) {
 		err := runtime.ExecuteTransaction(
 			Script{
 				Source: tx,
-				Arguments: encodeArgs([]cadence.Value{
+				Arguments: encodeArgs(
 					cadence.String(foo1),
 					cadence.String(foo2),
-				}),
+				),
 			},
 			Context{
 				Interface: runtimeInterface,

runtime/crypto_test.go

@@ -276,7 +276,7 @@ func TestRuntimeSignatureAlgorithmImport(t *testing.T) {
 		value, err := runtime.ExecuteScript(
 			Script{
 				Source: []byte(script),
-				Arguments: encodeArgs([]cadence.Value{
+				Arguments: encodeArgs(
 					cadence.NewEnum([]cadence.Value{
 						cadence.UInt8(algo.RawValue()),
 					}).WithType(cadence.NewEnumType(
@@ -291,7 +291,7 @@ func TestRuntimeSignatureAlgorithmImport(t *testing.T) {
 						},
 						nil,
 					)),
-				}),
+				),
 			},
 			Context{
 				Interface: runtimeInterface,
@@ -357,7 +357,7 @@ func TestRuntimeHashAlgorithmImport(t *testing.T) {
 		value, err := runtime.ExecuteScript(
 			Script{
 				Source: []byte(script),
-				Arguments: encodeArgs([]cadence.Value{
+				Arguments: encodeArgs(
 					cadence.NewEnum([]cadence.Value{
 						cadence.UInt8(algo.RawValue()),
 					}).WithType(cadence.NewEnumType(
@@ -372,7 +372,7 @@ func TestRuntimeHashAlgorithmImport(t *testing.T) {
 						},
 						nil,
 					)),
-				}),
+				),
 			},
 			Context{
 				Interface: runtimeInterface,
@@ -734,7 +734,7 @@ func TestRuntimeTraversingMerkleProof(t *testing.T) {
 	_, err := runtime.ExecuteScript(
 		Script{
 			Source:    script,
-			Arguments: encodeArgs([]cadence.Value{rootHash, address, accountProof}),
+			Arguments: encodeArgs(rootHash, address, accountProof),
 		},
 		Context{
 			Interface: runtimeInterface,

runtime/empty.go

@@ -29,6 +29,7 @@ import (
 	"github.com/onflow/cadence/common"
 	"github.com/onflow/cadence/interpreter"
 	"github.com/onflow/cadence/sema"
+	"github.com/onflow/cadence/stdlib"
 )
 
 // EmptyRuntimeInterface is an empty implementation of runtime.Interface.
@@ -63,6 +64,10 @@ func (EmptyRuntimeInterface) ComputationUsed() (uint64, error) {
 	panic("unexpected call to ComputationUsed")
 }
 
+func (i EmptyRuntimeInterface) ComputationRemaining(_ common.ComputationKind) uint {
+	panic("unexpected call to ComputationRemaining")
+}
+
 func (EmptyRuntimeInterface) MemoryUsed() (uint64, error) {
 	panic("unexpected call to MemoryUsed")
 }
@@ -264,3 +269,7 @@ func (EmptyRuntimeInterface) ValidateAccountCapabilitiesPublish(
 func (EmptyRuntimeInterface) MinimumRequiredVersion() (string, error) {
 	return "0.0.0", nil
 }
+
+func (EmptyRuntimeInterface) CompileWebAssembly(_ []byte) (stdlib.WebAssemblyModule, error) {
+	panic("unexpected call to CompileWebAssembly")
+}

runtime/environment.go

@@ -163,7 +163,9 @@ func newInterpreterEnvironment(config Config) *interpreterEnvironment {
 	}
 	env.InterpreterConfig = env.newInterpreterConfig()
 	env.CheckerConfig = env.newCheckerConfig()
+
 	env.compositeValueFunctionsHandlers = stdlib.DefaultStandardLibraryCompositeValueFunctionHandlers(env)
+
 	return env
 }
 
@@ -191,6 +193,7 @@ func (e *interpreterEnvironment) newInterpreterConfig() *interpreter.Config {
 		Debugger:                                  e.config.Debugger,
 		OnStatement:                               e.newOnStatementHandler(),
 		OnMeterComputation:                        e.newOnMeterComputation(),
+		OnComputationRemaining:                    e.newOnComputationRemaining(),
 		OnFunctionInvocation:                      e.newOnFunctionInvocationHandler(),
 		OnInvokedFunctionReturn:                   e.newOnInvokedFunctionReturnHandler(),
 		CapabilityBorrowHandler:                   e.newCapabilityBorrowHandler(),
@@ -213,19 +216,33 @@ func (e *interpreterEnvironment) newCheckerConfig() *sema.Config {
 	}
 }
 
+func StandardLibraryOptionsFromConfig(config Config) stdlib.StandardLibraryOptions {
+	return stdlib.StandardLibraryOptions{
+		WebAssemblyEnabled: config.WebAssemblyEnabled,
+	}
+}
+
 func NewBaseInterpreterEnvironment(config Config) *interpreterEnvironment {
 	env := newInterpreterEnvironment(config)
-	for _, valueDeclaration := range stdlib.DefaultStandardLibraryValues(env) {
+	options := StandardLibraryOptionsFromConfig(config)
+	for _, valueDeclaration := range stdlib.DefaultStandardLibraryValues(env, options) {
 		env.DeclareValue(valueDeclaration, nil)
 	}
+	for _, typeDeclaration := range stdlib.DefaultStandardLibraryTypes(options) {
+		env.DeclareType(typeDeclaration, nil)
+	}
 	return env
 }
 
 func NewScriptInterpreterEnvironment(config Config) Environment {
 	env := newInterpreterEnvironment(config)
-	for _, valueDeclaration := range stdlib.DefaultScriptStandardLibraryValues(env) {
+	options := StandardLibraryOptionsFromConfig(config)
+	for _, valueDeclaration := range stdlib.DefaultScriptStandardLibraryValues(env, options) {
 		env.DeclareValue(valueDeclaration, nil)
 	}
+	for _, typeDeclaration := range stdlib.DefaultStandardLibraryTypes(options) {
+		env.DeclareType(typeDeclaration, nil)
+	}
 	return env
 }
 
@@ -914,6 +931,10 @@ func (e *interpreterEnvironment) Hash(data []byte, tag string, algorithm sema.Ha
 	return e.runtimeInterface.Hash(data, tag, algorithm)
 }
 
+func (e *interpreterEnvironment) CompileWebAssembly(bytes []byte) (stdlib.WebAssemblyModule, error) {
+	return e.runtimeInterface.CompileWebAssembly(bytes)
+}
+
 func (e *interpreterEnvironment) DecodeArgument(argument []byte, argumentType cadence.Type) (cadence.Value, error) {
 	return e.runtimeInterface.DecodeArgument(argument, argumentType)
 }
@@ -1150,6 +1171,16 @@ func (e *interpreterEnvironment) newOnMeterComputation() interpreter.OnMeterComp
 	}
 }
 
+func (e *interpreterEnvironment) newOnComputationRemaining() interpreter.OnComputationRemainingFunc {
+	return func(compKind common.ComputationKind) uint {
+		var remaining uint
+		errors.WrapPanic(func() {
+			remaining = e.runtimeInterface.ComputationRemaining(compKind)
+		})
+		return remaining
+	}
+}
+
 func (e *interpreterEnvironment) InterpretContract(
 	location common.AddressLocation,
 	program *interpreter.Program,