Fix std feature in wasm-encoder (#1961)

* Fix `std` feature in `wasm-encoder`

* Update Cargo.lock

* Some updates

* Fix `std` features

* Test `no_std` works for `wasm-encoder` in CI

Author: ark0f

.github/workflows/main.yml

@@ -259,6 +259,8 @@ jobs:
       - run: cargo check --no-default-features -p wat --features component-model
       - run: cargo check --no-default-features -p wat --features dwarf
       - run: cargo check --no-default-features -p wasm-encoder
+      - run: cargo check --no-default-features -p wasm-encoder --target x86_64-unknown-none
+      - run: cargo check --no-default-features -p wasm-encoder --target x86_64-unknown-none --features wasmparser,component-model
       - run: cargo check --no-default-features -p wasm-encoder --features component-model
       - run: cargo check --no-default-features -p wasm-encoder --features wasmparser
       - run: cargo check --no-default-features -p wasm-smith

Cargo.lock

@@ -78,7 +78,7 @@ comfy-table = { version = "7.1.3", default-features = false }
 criterion = { version = "0.5.1", default-features = false }
 env_logger = "0.11"
 indexmap = { version = "2.7.0", default-features = false }
-leb128 = "0.2.4"
+leb128fmt = { version = "0.1.0", default-features = false }
 libfuzzer-sys = "0.4.0"
 log = "0.4.17"
 num_cpus = "1.13"
@@ -103,12 +103,12 @@ gimli = "0.30.0"
 id-arena = "2"
 
 wasm-compose = { version = "0.224.0", path = "crates/wasm-compose" }
-wasm-encoder = { version = "0.224.0", path = "crates/wasm-encoder", default-features = false, features = ["std"] }
+wasm-encoder = { version = "0.224.0", path = "crates/wasm-encoder", default-features = false }
 wasm-metadata = { version = "0.224.0", path = "crates/wasm-metadata" }
 wasm-mutate = { version = "0.224.0", path = "crates/wasm-mutate" }
 wasm-shrink = { version = "0.224.0", path = "crates/wasm-shrink" }
 wasm-smith = { version = "0.224.0", path = "crates/wasm-smith" }
-wasmparser = { version = "0.224.0", path = "crates/wasmparser", default-features = false, features = ['std', 'simd'] }
+wasmparser = { version = "0.224.0", path = "crates/wasmparser", default-features = false, features = ['simd'] }
 wasmprinter = { version = "0.224.0", path = "crates/wasmprinter", default-features = false }
 wast = { version = "224.0.0", path = "crates/wast", default-features = false }
 wat = { version = "1.224.0", path = "crates/wat", default-features = false }
@@ -129,7 +129,7 @@ wat = { workspace = true, features = ['dwarf', 'component-model'] }
 termcolor = { workspace = true }
 
 # Dependencies of `validate`
-wasmparser = { workspace = true, optional = true, features = ['component-model', 'simd'] }
+wasmparser = { workspace = true, optional = true, features = ['std', 'component-model', 'simd'] }
 rayon = { workspace = true, optional = true }
 bitflags = { workspace = true, optional = true }
 
@@ -150,7 +150,7 @@ wasm-shrink = { workspace = true, features = ["clap"], optional = true }
 wasm-mutate = { workspace = true, features = ["clap"], optional = true }
 
 # Dependencies of `strip`
-wasm-encoder = { workspace = true, optional = true }
+wasm-encoder = { workspace = true, features = ["std"], optional = true }
 regex = { version = "1.6.0", optional = true }
 
 # Dependencies of `compose`

Cargo.toml

@@ -25,7 +25,7 @@ arbitrary = { workspace = true, features = ["derive"] }
 wasm-mutate = { workspace = true }
 wasm-shrink = { workspace = true }
 wasm-smith = { workspace = true }
-wasmparser = { workspace = true }
+wasmparser = { workspace = true, features = ["std"] }
 wasmprinter = { workspace = true }
 wast = { workspace = true }
 wat = { workspace = true }

crates/c-api/Cargo.toml

@@ -16,8 +16,8 @@ workspace = true
 
 [dependencies]
 wat = { workspace = true }
-wasm-encoder = { workspace = true, features = ['wasmparser', 'component-model'] }
-wasmparser = { workspace = true, features = ['validate', 'component-model', 'features'] }
+wasm-encoder = { workspace = true, features = ['std', 'wasmparser', 'component-model'] }
+wasmparser = { workspace = true, features = ['std', 'validate', 'component-model', 'features'] }
 indexmap = { workspace = true, features = ["serde"] }
 anyhow = { workspace = true }
 serde = { workspace = true }

crates/wasm-compose/Cargo.toml

@@ -20,7 +20,7 @@ all-features = true
 workspace = true
 
 [dependencies]
-leb128 = { workspace = true }
+leb128fmt.workspace = true
 
 # Enable this dependency to get a bunch of `From<wasmparser::Foo> for
 # wasm_encoder::Foo` impls.
@@ -36,7 +36,7 @@ wasmprinter = { workspace = true }
 default = ['std', 'component-model']
 
 # A feature which enables implementations of `std::error::Error` as appropriate.
-std = []
+std = ["wasmparser?/std"]
 
 # On-by-default: conditional support for emitting components in addition to
 # core modules.

crates/wasm-encoder/Cargo.toml

@@ -4001,10 +4001,7 @@ impl ConstExpr {
         if prefix != 0xd2 {
             return None;
         }
-        leb128::read::unsigned(&mut &self.bytes[1..])
-            .ok()?
-            .try_into()
-            .ok()
+        leb128fmt::decode_uint_slice::<u32, 32>(&self.bytes[1..], &mut 0).ok()
     }
 }
 

crates/wasm-encoder/src/core/code.rs

@@ -135,25 +135,29 @@ impl Encode for usize {
 
 impl Encode for u32 {
     fn encode(&self, sink: &mut Vec<u8>) {
-        leb128::write::unsigned(sink, (*self).into()).unwrap();
+        let (value, pos) = leb128fmt::encode_u32(*self).unwrap();
+        sink.extend_from_slice(&value[..pos]);
     }
 }
 
 impl Encode for i32 {
     fn encode(&self, sink: &mut Vec<u8>) {
-        leb128::write::signed(sink, (*self).into()).unwrap();
+        let (value, pos) = leb128fmt::encode_s32(*self).unwrap();
+        sink.extend_from_slice(&value[..pos]);
     }
 }
 
 impl Encode for u64 {
     fn encode(&self, sink: &mut Vec<u8>) {
-        leb128::write::unsigned(sink, *self).unwrap();
+        let (value, pos) = leb128fmt::encode_u64(*self).unwrap();
+        sink.extend_from_slice(&value[..pos]);
     }
 }
 
 impl Encode for i64 {
     fn encode(&self, sink: &mut Vec<u8>) {
-        leb128::write::signed(sink, *self).unwrap();
+        let (value, pos) = leb128fmt::encode_s64(*self).unwrap();
+        sink.extend_from_slice(&value[..pos]);
     }
 }
 
@@ -200,8 +204,8 @@ where
 }
 
 fn encoding_size(n: u32) -> usize {
-    let mut buf = [0u8; 5];
-    leb128::write::unsigned(&mut &mut buf[..], n.into()).unwrap()
+    let (_value, pos) = leb128fmt::encode_u32(n).unwrap();
+    pos
 }
 
 fn encode_section(sink: &mut Vec<u8>, count: u32, bytes: &[u8]) {
@@ -214,6 +218,11 @@ fn encode_section(sink: &mut Vec<u8>, count: u32, bytes: &[u8]) {
 mod test {
     use super::*;
 
+    #[test]
+    fn test_encoding_size() {
+        assert_eq!(encoding_size(624485), 3);
+    }
+
     #[test]
     fn it_encodes_an_empty_module() {
         let bytes = Module::new().finish();

crates/wasm-encoder/src/lib.rs

@@ -13,8 +13,8 @@ workspace = true
 [dependencies]
 clap = { workspace = true, optional = true }
 anyhow = { workspace = true }
-wasmparser = { workspace = true, features = ['component-model', 'hash-collections'] }
-wasm-encoder = { workspace = true, features = ['component-model'] }
+wasmparser = { workspace = true, features = ['std', 'component-model', 'hash-collections'] }
+wasm-encoder = { workspace = true, features = ['std', 'component-model'] }
 indexmap = { workspace = true, features = ["serde"] }
 serde = { workspace = true }
 serde_derive = { workspace = true }

crates/wasm-metadata/Cargo.toml

@@ -14,7 +14,7 @@ num_cpus = { workspace = true }
 rand = { workspace = true }
 wasm-mutate = { workspace = true }
 wasmprinter = { workspace = true }
-wasmparser = { workspace = true, features = ['validate'] }
+wasmparser = { workspace = true, features = ['std', 'validate'] }
 wasmtime = { workspace = true }
 env_logger = { workspace = true }
 itertools = "0.13.0"