fix: fix build, improve error handling

Signed-off-by: Henry Gressmann <[email protected]>

Author: explodingcamera

CHANGELOG.md

@@ -11,11 +11,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Support for the custom memory page sizes proposal ([#22](https://github.com/explodingcamera/tinywasm/pull/22) by [@danielstuart14](https://github.com/danielstuart14))
 
-### Changed
+### Breaking Changes
 
-- **Breaking:**: New backwards-incompatible version of the twasm format based on `postcard` (thanks [@dragonnn](https://github.com/dragonnn))
-- **Breaking:**: `RefNull` has been removed and replaced with new `FuncRef` and `ExternRef` structs
+- New backwards-incompatible version of the twasm format based on `postcard` (thanks [@dragonnn](https://github.com/dragonnn))
+- `RefNull` has been removed and replaced with new `FuncRef` and `ExternRef` structs
 - Increased MSRV to 1.83.0
+- `tinywasm::Error` is now `non_exhaustive`, `Error::ParseError` has been rename to `Error::Parser` and `Error::Twasm` has been added.
 
 ### Fixed
 

crates/tinywasm/src/error.rs

@@ -1,13 +1,15 @@
 use alloc::string::{String, ToString};
 use alloc::vec::Vec;
 use core::{fmt::Display, ops::ControlFlow};
+use tinywasm_types::archive::TwasmError;
 use tinywasm_types::FuncType;
 
 #[cfg(feature = "parser")]
 pub use tinywasm_parser::ParseError;
 
 /// Errors that can occur for `TinyWasm` operations
 #[derive(Debug)]
+#[non_exhaustive]
 pub enum Error {
     /// A WebAssembly trap occurred
     Trap(Trap),
@@ -41,7 +43,10 @@ pub enum Error {
 
     #[cfg(feature = "parser")]
     /// A parsing error occurred
-    ParseError(ParseError),
+    Parser(ParseError),
+
+    /// A serialization error occurred
+    Twasm(TwasmError),
 }
 
 #[derive(Debug)]
@@ -169,6 +174,11 @@ impl From<LinkingError> for Error {
     }
 }
 
+impl From<TwasmError> for Error {
+    fn from(value: TwasmError) -> Self {
+        Self::Twasm(value)
+    }
+}
 impl From<Trap> for Error {
     fn from(value: Trap) -> Self {
         Self::Trap(value)
@@ -179,11 +189,12 @@ impl Display for Error {
     fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
         match self {
             #[cfg(feature = "parser")]
-            Self::ParseError(err) => write!(f, "error parsing module: {err:?}"),
+            Self::Parser(err) => write!(f, "error parsing module: {err:?}"),
 
             #[cfg(feature = "std")]
             Self::Io(err) => write!(f, "I/O error: {err}"),
 
+            Self::Twasm(err) => write!(f, "serialization error: {err}"),
             Self::Trap(trap) => write!(f, "trap: {trap}"),
             Self::Linker(err) => write!(f, "linking error: {err}"),
             Self::InvalidLabelType => write!(f, "invalid label type"),
@@ -238,7 +249,7 @@ impl core::error::Error for Error {}
 #[cfg(feature = "parser")]
 impl From<tinywasm_parser::ParseError> for Error {
     fn from(value: tinywasm_parser::ParseError) -> Self {
-        Self::ParseError(value)
+        Self::Parser(value)
     }
 }
 

crates/tinywasm/src/interpreter/no_std_floats.rs

@@ -1,34 +1,25 @@
 pub(super) trait NoStdFloatExt {
     fn round(self) -> Self;
-    fn abs(self) -> Self;
-    fn signum(self) -> Self;
     fn ceil(self) -> Self;
     fn floor(self) -> Self;
     fn trunc(self) -> Self;
     fn sqrt(self) -> Self;
-    fn copysign(self, other: Self) -> Self;
 }
 
 #[rustfmt::skip]
 impl NoStdFloatExt for f64 {
     #[inline] fn round(self) -> Self { libm::round(self) }
-    #[inline] fn abs(self) -> Self { libm::fabs(self) }
-    #[inline] fn signum(self) -> Self { libm::copysign(1.0, self) }
     #[inline] fn ceil(self) -> Self { libm::ceil(self) }
     #[inline] fn floor(self) -> Self { libm::floor(self) }
     #[inline] fn trunc(self) -> Self { libm::trunc(self) }
     #[inline] fn sqrt(self) -> Self { libm::sqrt(self) }
-    #[inline] fn copysign(self, other: Self) -> Self { libm::copysign(self, other) }
 }
 
 #[rustfmt::skip]
 impl NoStdFloatExt for f32 {
     #[inline] fn round(self) -> Self { libm::roundf(self) }
-    #[inline] fn abs(self) -> Self { libm::fabsf(self) }
-    #[inline] fn signum(self) -> Self { libm::copysignf(1.0, self) }
     #[inline] fn ceil(self) -> Self { libm::ceilf(self) }
     #[inline] fn floor(self) -> Self { libm::floorf(self) }
     #[inline] fn trunc(self) -> Self { libm::truncf(self) }
     #[inline] fn sqrt(self) -> Self { libm::sqrtf(self) }
-    #[inline] fn copysign(self, other: Self) -> Self { libm::copysignf(self, other) }
 }

crates/types/src/lib.rs

@@ -45,6 +45,18 @@ pub use value::*;
 #[cfg(feature = "archive")]
 pub mod archive;
 
+#[cfg(not(feature = "archive"))]
+pub mod archive {
+    #[derive(Debug)]
+    pub enum TwasmError {}
+    impl core::fmt::Display for TwasmError {
+        fn fmt(&self, _: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
+            Err(core::fmt::Error)
+        }
+    }
+    impl core::error::Error for TwasmError {}
+}
+
 /// A `TinyWasm` WebAssembly Module
 ///
 /// This is the internal representation of a WebAssembly module in `TinyWasm`.

examples/rust/Cargo.toml

@@ -22,6 +22,10 @@ std=["tinywasm/std"]
 name="hello"
 path="src/hello.rs"
 
+[[bin]]
+name="host_fn"
+path="src/host_fn.rs"
+
 [[bin]]
 name="print"
 path="src/print.rs"

examples/rust/build.sh

@@ -1,7 +1,7 @@
 #!/usr/bin/env bash
-cd "$(dirname "$0")"
+cd "$(dirname "$0")" || exit
 
-bins=("hello" "fibonacci" "print" "tinywasm" "argon2id")
+bins=("host_fn" "hello" "fibonacci" "print" "tinywasm" "argon2id")
 exclude_wat=("tinywasm")
 out_dir="./target/wasm32-unknown-unknown/wasm"
 dest_dir="out"

examples/rust/src/argon2id.rs

@@ -1,6 +1,6 @@
 #![no_main]
 
-#[no_mangle]
+#[unsafe(no_mangle)]
 pub extern "C" fn argon2id(m_cost: i32, t_cost: i32, p_cost: i32) -> i32 {
     let password = b"password";
     let salt = b"some random salt";

examples/rust/src/fibonacci.rs

@@ -1,6 +1,6 @@
 #![no_main]
 
-#[no_mangle]
+#[unsafe(no_mangle)]
 pub extern "C" fn fibonacci(n: i32) -> i32 {
     let mut sum = 0;
     let mut last = 0;
@@ -13,7 +13,7 @@ pub extern "C" fn fibonacci(n: i32) -> i32 {
     sum
 }
 
-#[no_mangle]
+#[unsafe(no_mangle)]
 pub extern "C" fn fibonacci_recursive(n: i32) -> i32 {
     if n <= 1 {
         return n;

examples/rust/src/hello.rs

@@ -1,23 +1,23 @@
 #![no_main]
 
 #[link(wasm_import_module = "env")]
-extern "C" {
+unsafe extern "C" {
     fn print_utf8(location: i64, len: i32);
 }
 
 const ARG: &[u8] = &[0u8; 100];
 
-#[no_mangle]
+#[unsafe(no_mangle)]
 pub unsafe extern "C" fn arg_ptr() -> i32 {
     ARG.as_ptr() as i32
 }
 
-#[no_mangle]
+#[unsafe(no_mangle)]
 pub unsafe extern "C" fn arg_size() -> i32 {
     ARG.len() as i32
 }
 
-#[no_mangle]
+#[unsafe(no_mangle)]
 pub unsafe extern "C" fn hello(len: i32) {
     let arg = core::str::from_utf8(&ARG[0..len as usize]).unwrap();
     let res = format!("Hello, {}!", arg).as_bytes().to_vec();

examples/rust/src/host_fn.rs

@@ -0,0 +1,11 @@
+#![no_main]
+
+#[link(wasm_import_module = "env")]
+unsafe extern "C" {
+    fn bar(left: i64, right: i32) -> i32;
+}
+
+#[unsafe(no_mangle)]
+pub fn foo() -> i32 {
+    unsafe { bar(1, 2) }
+}