xref: /wasmtime-44.0.1/tests/all/instance.rs (revision 331b0dee)
1 use anyhow::Result;
2 use wasmtime::*;
3 
4 #[test]
5 fn wrong_import_numbers() -> Result<()> {
6     let mut store = Store::<()>::default();
7     let module = Module::new(store.engine(), r#"(module (import "" "" (func)))"#)?;
8 
9     assert!(Instance::new(&mut store, &module, &[]).is_err());
10     let func = Func::wrap(&mut store, || {});
11     assert!(Instance::new(&mut store, &module, &[func.clone().into(), func.into()]).is_err());
12     Ok(())
13 }
14 
15 #[test]
16 fn initializes_linear_memory() -> Result<()> {
17     // Test for https://github.com/bytecodealliance/wasmtime/issues/2784
18     let wat = r#"
19         (module
20             (memory (export "memory") 2)
21             (data (i32.const 0) "Hello World!")
22         )"#;
23     let module = Module::new(&Engine::default(), wat)?;
24 
25     let mut store = Store::new(module.engine(), ());
26     let instance = Instance::new(&mut store, &module, &[])?;
27     let memory = instance.get_memory(&mut store, "memory").unwrap();
28 
29     let mut bytes = [0; 12];
30     memory.read(&store, 0, &mut bytes)?;
31     assert_eq!(bytes, "Hello World!".as_bytes());
32     Ok(())
33 }
34 
35 #[test]
36 fn linear_memory_limits() -> Result<()> {
37     // this test will allocate 4GB of virtual memory space, and may not work in
38     // situations like CI QEMU emulation where it triggers SIGKILL.
39     if std::env::var("WASMTIME_TEST_NO_HOG_MEMORY").is_ok() {
40         return Ok(());
41     }
42     test(&Engine::default())?;
43     test(&Engine::new(Config::new().allocation_strategy(
44         InstanceAllocationStrategy::Pooling {
45             strategy: PoolingAllocationStrategy::NextAvailable,
46             module_limits: ModuleLimits {
47                 memory_pages: 65536,
48                 ..ModuleLimits::default()
49             },
50             instance_limits: InstanceLimits::default(),
51         },
52     ))?)?;
53     return Ok(());
54 
55     fn test(engine: &Engine) -> Result<()> {
56         let wat = r#"
57         (module
58             (memory 65534)
59 
60             (func (export "grow")  (result i32)
61                 i32.const 1
62                 memory.grow)
63             (func (export "size")  (result i32)
64                 memory.size)
65         )
66     "#;
67         let module = Module::new(engine, wat)?;
68 
69         let mut store = Store::new(engine, ());
70         let instance = Instance::new(&mut store, &module, &[])?;
71         let size = instance.get_typed_func::<(), i32, _>(&mut store, "size")?;
72         let grow = instance.get_typed_func::<(), i32, _>(&mut store, "grow")?;
73 
74         assert_eq!(size.call(&mut store, ())?, 65534);
75         assert_eq!(grow.call(&mut store, ())?, 65534);
76         assert_eq!(size.call(&mut store, ())?, 65535);
77         assert_eq!(grow.call(&mut store, ())?, 65535);
78         assert_eq!(size.call(&mut store, ())?, 65536);
79         assert_eq!(grow.call(&mut store, ())?, -1);
80         assert_eq!(size.call(&mut store, ())?, 65536);
81         Ok(())
82     }
83 }
84