def validate_start_function(context: Context, start: StartFunction) -> None: context.validate_function_idx(start.function_idx) function_type = context.get_function(start.function_idx) if function_type != FunctionType((), ()): raise ValidationError( "Start function may not have arguments or a result type. Got " f"{function_type}")
def parse_function_type(stream: IO[bytes]) -> FunctionType: flag = parse_single_byte(stream) if flag != 0x60: raise MalformedModule( f"Invalid function type leading byte: {hex(flag)}") params = parse_vector(parse_valtype, stream) results = parse_vector(parse_valtype, stream) return FunctionType(params, results)
def instantiate_spectest_module(store: Store) -> ModuleInstance: logger = logging.getLogger("wasm.tools.fixtures.modules.spectest") def spectest__print_i32(store, arg): logger.debug('print_i32: %s', arg) return store, [] def spectest__print_i64(store, arg): logger.debug('print_i64: %s', arg) return store, [] def spectest__print_f32(store, arg): logger.debug('print_f32: %s', arg) return store, [] def spectest__print_f64(store, arg): logger.debug('print_f64: %s', arg) return store, [] def spectest__print_i32_f32(store, arg): logger.debug('print_i32_f32: %s', arg) return store, [] def spectest__print_f64_f64(store, arg): logger.debug('print_f64_f64: %s', arg) return store, [] def spectest__print(store, arg): logger.debug('print: %s', arg) return store, [] store.allocate_host_function(FunctionType((ValType.i32, ), ()), spectest__print_i32) store.allocate_host_function(FunctionType((ValType.i64, ), ()), spectest__print_i64) store.allocate_host_function(FunctionType((ValType.f32, ), ()), spectest__print_f32) store.allocate_host_function(FunctionType((ValType.f64, ), ()), spectest__print_f64) store.allocate_host_function(FunctionType((ValType.i32, ValType.f32), ()), spectest__print_i32_f32) store.allocate_host_function(FunctionType((ValType.f64, ValType.f64), ()), spectest__print_f64_f64) store.allocate_host_function(FunctionType((), ()), spectest__print) # min:1,max:2 required by import.wast: store.allocate_memory(MemoryType(UInt32(1), UInt32(2))) # 666 required by import.wast store.allocate_global(GlobalType(Mutability.const, ValType.i32), UInt32(666)) store.allocate_global(GlobalType(Mutability.const, ValType.f32), Float32(0.0)) store.allocate_global(GlobalType(Mutability.const, ValType.f64), Float64(0.0)) store.allocate_table( TableType( Limits(UInt32(10), UInt32(20)), FunctionAddress)) # max was 30, changed to 20 for import.wast moduleinst = ModuleInstance( types=( FunctionType((ValType.i32, ), ()), FunctionType((ValType.i64, ), ()), FunctionType((ValType.f32, ), ()), FunctionType((ValType.f64, ), ()), FunctionType((ValType.i32, ValType.f32), ()), FunctionType((ValType.f64, ValType.f64), ()), FunctionType((), ()), ), func_addrs=tuple(FunctionAddress(idx) for idx in range(7)), table_addrs=(TableAddress(0), ), memory_addrs=(MemoryAddress(0), ), global_addrs=(GlobalAddress(0), GlobalAddress(1)), exports=( ExportInstance("print_i32", FunctionAddress(0)), ExportInstance("print_i64", FunctionAddress(1)), ExportInstance("print_f32", FunctionAddress(2)), ExportInstance("print_f64", FunctionAddress(3)), ExportInstance("print_i32_f32", FunctionAddress(4)), ExportInstance("print_f64_f64", FunctionAddress(5)), ExportInstance("print", FunctionAddress(6)), ExportInstance("memory", MemoryAddress(0)), ExportInstance("global_i32", GlobalAddress(0)), ExportInstance("global_f32", GlobalAddress(1)), ExportInstance("global_f64", GlobalAddress(2)), ExportInstance("table", TableAddress(0)), ), ) return moduleinst
def instantiate_test_module(store): def test__func(store, arg): pass def test__func_i32(store, arg): pass def test__func_f32(store, arg): pass def test__func__i32(store, arg): pass def test__func__f32(store, arg): pass def test__func_i32_i32(store, arg): pass def test__func_i64_i64(store, arg): pass store.allocate_host_function(FunctionType((), ()), test__func) store.allocate_host_function(FunctionType((ValType.i32, ), ()), test__func_i32) store.allocate_host_function(FunctionType((ValType.f32, ), ()), test__func_f32) store.allocate_host_function(FunctionType((), (ValType.i32, )), test__func__i32) store.allocate_host_function(FunctionType((), (ValType.f32, )), test__func__f32) store.allocate_host_function( FunctionType((ValType.i32, ), (ValType.i32, )), test__func_i32_i32) store.allocate_host_function( FunctionType((ValType.i64, ), (ValType.i64, )), test__func_i64_i64) store.allocate_memory(MemoryType(1, None)) store.allocate_global(GlobalType(Mutability.const, ValType.i32), UInt32(666)) store.allocate_global(GlobalType(Mutability.const, ValType.f32), Float32(0.0)) store.allocate_table(TableType(Limits(10, None), FunctionAddress)) moduleinst = ModuleInstance( types=( FunctionType((), ()), FunctionType((ValType.i32, ), ()), FunctionType((ValType.f32, ), ()), FunctionType((), (ValType.i32, )), FunctionType((), (ValType.f32, )), FunctionType((ValType.i32, ), (ValType.i32, )), FunctionType((ValType.i64, ), (ValType.i64, )), ), func_addrs=tuple(FunctionAddress(idx) for idx in range(7)), table_addrs=(TableAddress(0), ), memory_addrs=(MemoryAddress(0), ), global_addrs=(GlobalAddress(0), GlobalAddress(1)), exports=( ExportInstance("func", FunctionAddress(0)), ExportInstance("func_i32", FunctionAddress(1)), ExportInstance("func_f32", FunctionAddress(2)), ExportInstance("func__i32", FunctionAddress(3)), ExportInstance("func__f32", FunctionAddress(4)), ExportInstance("func__i32_i32", FunctionAddress(5)), ExportInstance("func__i64_i64", FunctionAddress(6)), ExportInstance("memory-2-inf", MemoryAddress(0)), ExportInstance("global-i32", GlobalAddress(0)), ExportInstance("global-f32", GlobalAddress(1)), ExportInstance("table-10-inf", TableAddress(0)), ), ) return moduleinst
def instantiate_test_module(store: Store) -> ModuleInstance: def test__func(config: Configuration, args: Tuple[TValue, ...]) -> Tuple[TValue, ...]: return tuple() def test__func_i32(config: Configuration, args: Tuple[TValue, ...]) -> Tuple[TValue, ...]: return tuple() def test__func_f32(config: Configuration, args: Tuple[TValue, ...]) -> Tuple[TValue, ...]: return tuple() def test__func__i32(config: Configuration, args: Tuple[TValue, ...]) -> Tuple[TValue, ...]: return tuple() def test__func__f32(config: Configuration, args: Tuple[TValue, ...]) -> Tuple[TValue, ...]: return tuple() def test__func_i32_i32(config: Configuration, args: Tuple[TValue, ...]) -> Tuple[TValue, ...]: return tuple() def test__func_i64_i64(config: Configuration, args: Tuple[TValue, ...]) -> Tuple[TValue, ...]: return tuple() store.allocate_host_function(FunctionType((), ()), test__func) store.allocate_host_function(FunctionType((ValType.i32,), ()), test__func_i32) store.allocate_host_function(FunctionType((ValType.f32,), ()), test__func_f32) store.allocate_host_function(FunctionType((), (ValType.i32,)), test__func__i32) store.allocate_host_function(FunctionType((), (ValType.f32,)), test__func__f32) store.allocate_host_function( FunctionType((ValType.i32,), (ValType.i32,)), test__func_i32_i32, ) store.allocate_host_function( FunctionType((ValType.i64,), (ValType.i64,)), test__func_i64_i64, ) store.allocate_memory(MemoryType(numpy.uint32(1), None)) store.allocate_global(GlobalType(Mutability.const, ValType.i32), numpy.uint32(666)) store.allocate_global(GlobalType(Mutability.const, ValType.f32), numpy.float32(0.0)) store.allocate_table(TableType(Limits(numpy.uint32(10), None), FunctionAddress)) moduleinst = ModuleInstance( types=( FunctionType((), ()), FunctionType((ValType.i32,), ()), FunctionType((ValType.f32,), ()), FunctionType((), (ValType.i32,)), FunctionType((), (ValType.f32,)), FunctionType((ValType.i32,), (ValType.i32,)), FunctionType((ValType.i64,), (ValType.i64,)), ), func_addrs=tuple(FunctionAddress(idx) for idx in range(7)), table_addrs=(TableAddress(0),), memory_addrs=(MemoryAddress(0),), global_addrs=(GlobalAddress(0), GlobalAddress(1)), exports=( ExportInstance("func", FunctionAddress(0)), ExportInstance("func_i32", FunctionAddress(1)), ExportInstance("func_f32", FunctionAddress(2)), ExportInstance("func__i32", FunctionAddress(3)), ExportInstance("func__f32", FunctionAddress(4)), ExportInstance("func__i32_i32", FunctionAddress(5)), ExportInstance("func__i64_i64", FunctionAddress(6)), ExportInstance("memory-2-inf", MemoryAddress(0)), ExportInstance("global-i32", GlobalAddress(0)), ExportInstance("global-f32", GlobalAddress(1)), ExportInstance("table-10-inf", TableAddress(0)), ), ) return moduleinst
def test_validate_function_type_with_multiple_results(): function_type = FunctionType((), (ValType.i32, ValType.i32)) with pytest.raises(ValidationError, match="Function types may only have one result"): validate_function_type(function_type)
def test_validate_function_type_with_single_result(): function_type = FunctionType((), (ValType.i32, )) validate_function_type(function_type)
def test_validate_function_type_without_results(): function_type = FunctionType((), ()) validate_function_type(function_type)
import pytest from wasm.datatypes import ( FunctionType, ValType, ) @pytest.mark.parametrize( 'function_type_a,function_type_b,expected', ( ( FunctionType(tuple(), tuple()), FunctionType(tuple(), tuple()), True, ), ( FunctionType((ValType.i32, ), tuple()), FunctionType((ValType.i32, ), tuple()), True, ), ( FunctionType(tuple(), (ValType.i32, )), FunctionType(tuple(), (ValType.i32, )), True, ), ( FunctionType((ValType.i32, ), tuple()), FunctionType(tuple(), (ValType.i32, )), False, ),