Add 'crates/wiggle/' from commit 'cd484e49932d8dd8f1bd1a002e0717ad8bff07fb'

git-subtree-dir: crates/wiggle
git-subtree-mainline: 2ead747f48
git-subtree-split: cd484e4993

crates/wiggle/tests/handles.rs

@@ -0,0 +1,74 @@
+use proptest::prelude::*;
+use wiggle_runtime::{GuestError, GuestMemory, GuestType};
+use wiggle_test::{impl_errno, HostMemory, MemArea, WasiCtx};
+
+const FD_VAL: u32 = 123;
+
+wiggle::from_witx!({
+    witx: ["tests/handles.witx"],
+    ctx: WasiCtx,
+});
+
+impl_errno!(types::Errno);
+
+impl<'a> handle_examples::HandleExamples for WasiCtx<'a> {
+    fn fd_create(&self) -> Result<types::Fd, types::Errno> {
+        Ok(types::Fd::from(FD_VAL))
+    }
+    fn fd_consume(&self, fd: types::Fd) -> Result<(), types::Errno> {
+        println!("FD_CONSUME {}", fd);
+        if fd == types::Fd::from(FD_VAL) {
+            Ok(())
+        } else {
+            Err(types::Errno::InvalidArg)
+        }
+    }
+}
+
+#[derive(Debug)]
+struct HandleExercise {
+    pub return_loc: MemArea,
+}
+
+impl HandleExercise {
+    pub fn test(&self) {
+        let ctx = WasiCtx::new();
+        let host_memory = HostMemory::new();
+
+        let e = handle_examples::fd_create(&ctx, &host_memory, self.return_loc.ptr as i32);
+
+        assert_eq!(e, types::Errno::Ok.into(), "fd_create error");
+
+        let h_got: u32 = host_memory
+            .ptr(self.return_loc.ptr)
+            .read()
+            .expect("return ref_mut");
+
+        assert_eq!(h_got, 123, "fd_create return val");
+
+        let e = handle_examples::fd_consume(&ctx, &host_memory, h_got as i32);
+
+        assert_eq!(e, types::Errno::Ok.into(), "fd_consume error");
+
+        let e = handle_examples::fd_consume(&ctx, &host_memory, h_got as i32 + 1);
+
+        assert_eq!(
+            e,
+            types::Errno::InvalidArg.into(),
+            "fd_consume invalid error"
+        );
+    }
+
+    pub fn strat() -> BoxedStrategy<Self> {
+        (HostMemory::mem_area_strat(types::Fd::guest_size()))
+            .prop_map(|return_loc| HandleExercise { return_loc })
+            .boxed()
+    }
+}
+
+proptest! {
+    #[test]
+    fn handle_exercise(e in HandleExercise::strat()) {
+        e.test()
+    }
+}