preparation for fuzz testing

Author: wasm-forge

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ic-wasi-polyfill"
-version = "0.8.2"
+version = "0.8.3"
 edition = "2024"
 keywords = ["ic", "wasi", "wasi-polyfill"]
 description = "The project provides polyfill implementation of *wasi_snapshot_preview1* functions using IC System API."

test/canisters/fs_tests/dfx.json

@@ -4,7 +4,8 @@
       "candid": "src/fs_tests_backend/fs_tests_backend.did",
       "package": "fs_tests_backend",
       "wasm": "target/wasm32-wasip1/release/fs_tests_backend_nowasi.wasm",
-      "type": "rust"
+      "build": "scripts/build.sh",
+      "type": "custom"
     }
   },
   "defaults": {

…st/canisters/fs_tests/scripts/install.sh test/canisters/fs_tests/scripts/build.shtest/canisters/fs_tests/scripts/install.sh renamed to test/canisters/fs_tests/scripts/build.sh test/canisters/fs_tests/scripts/install.sh renamed to test/canisters/fs_tests/scripts/build.sh

@@ -11,7 +11,7 @@ ic-wasm $RELEASE_DIR/fs_tests_backend.wasm -o $RELEASE_DIR/fs_tests_backend_meta
 wasi2ic $RELEASE_DIR/fs_tests_backend_meta.wasm $RELEASE_DIR/fs_tests_backend_nowasi.wasm
 
 
-dfx canister create fs_tests_backend
+#dfx canister create fs_tests_backend
 
-dfx canister install --mode reinstall fs_tests_backend --wasm $RELEASE_DIR/fs_tests_backend_nowasi.wasm -y
+#dfx canister install --mode reinstall fs_tests_backend --wasm $RELEASE_DIR/fs_tests_backend_nowasi.wasm -y
 

test/canisters/fs_tests/scripts/test_read.sh

@@ -1,10 +1,14 @@
 #!/bin/sh
 
 # try reading a file if it exists, then 
-dfx canister call fs_tests_backend read_file '(".test_hidden/testdir/test.txt")'
+#dfx canister call fs_tests_backend read_file '(".test_hidden/testdir/test.txt")'
 
 # write some content into file
 dfx canister call fs_tests_backend write_file '(".test_hidden/testdir/test.txt", "some test content")'
 
 # try reading again
 dfx canister call fs_tests_backend read_file '(".test_hidden/testdir/test.txt")'
+
+# compute hash
+dfx canister call fs_tests_backend compute_file_hash '(".test_hidden/testdir/test.txt")'
+

test/canisters/fs_tests/src/fs_tests_backend/Cargo.toml

@@ -8,10 +8,18 @@ edition = "2021"
 [lib]
 crate-type = ["cdylib"]
 
+[[bin]]
+name = "fs_test"
+path = "src/main.rs"
+
 [dependencies]
 candid = "0.10"
 ic-cdk = "0.18.3"
 ic-stable-structures = "0.6.7"
 
 ic-wasi-polyfill = {path="../../../../../../ic-wasi-polyfill"}
 #ic-wasi-polyfill = "0.5"
+
+sha2 = "0.10"
+anyhow = "1.0.98"
+

test/canisters/fs_tests/src/fs_tests_backend/fs_tests_backend.did

@@ -1,7 +1,10 @@
 service : {
+    "greet": (text) -> (text) query;
+    "basic_fs_test": () -> ();
+
+    "compute_file_hash": (text) -> (text);
 
     "read_file": (text) -> (text);
 
     "write_file": (text, text) -> ();
-
 }

test/canisters/fs_tests/src/fs_tests_backend/src/lib.rs

@@ -1,15 +1,39 @@
 use ic_stable_structures::{
-    memory_manager::{MemoryId, MemoryManager},
     DefaultMemoryImpl, Memory,
+    memory_manager::{MemoryId, MemoryManager},
 };
 
+#[ic_cdk::query]
+fn greet(name: String) -> String {
+    format!("Hello, {}!", name)
+}
+
+pub struct SimpleRng {
+    state: u64,
+}
+
+impl SimpleRng {
+    pub fn new(seed: u64) -> Self {
+        Self { state: seed }
+    }
+
+    pub fn next(&mut self) -> u64 {
+        // Constants from Numerical Recipes
+        const A: u64 = 6364136223846793005;
+        const C: u64 = 1;
+
+        self.state = self.state.wrapping_mul(A).wrapping_add(C);
+        self.state
+    }
+}
+
 use std::{
     cell::RefCell,
     fs::{self},
 };
 
 const PROFILING: MemoryId = MemoryId::new(100);
-const WASI_MEMORY_ID: MemoryId = MemoryId::new(1);
+const WASI_MEMORY_ID: MemoryId = MemoryId::new(11);
 
 thread_local! {
     static MEMORY_MANAGER: RefCell<MemoryManager<DefaultMemoryImpl>> =
@@ -36,12 +60,48 @@ fn init() {
     });
 }
 
-#[ic_cdk::post_upgrade]
-fn post_upgrade() {
-    profiling_init();
+use sha2::{Digest, Sha256};
+use std::fs::File;
+use std::io::BufReader;
+use std::io::Read;
+use std::io::Write;
 
-    let wasi_memory = MEMORY_MANAGER.with(|m| m.borrow().get(WASI_MEMORY_ID));
-    ic_wasi_polyfill::init_with_memory(&[0u8; 32], &[], wasi_memory);
+#[ic_cdk::update]
+fn compute_file_hash(path: String) -> String {
+    let file = File::open(path).expect("I/O error");
+    let mut reader = BufReader::new(file);
+    let mut hasher = Sha256::new();
+    let mut buffer = [0u8; 4096];
+
+    loop {
+        let n = reader.read(&mut buffer).expect("I/O error");
+
+        if n == 0 {
+            break;
+        }
+        hasher.update(&buffer[..n]);
+    }
+
+    let hash = hasher.finalize();
+
+    format!("{:x}", hash)
+}
+
+#[ic_cdk::update]
+pub fn basic_fs_test() {
+    let file_path = "test_file.txt";
+    let test_content = "Hello, Rust!";
+
+    let mut file = File::create(file_path).expect("Failed to create file");
+    file.write_all(test_content.as_bytes())
+        .expect("Failed to write to file");
+
+    let mut read_content = String::new();
+    let mut file = File::open(file_path).expect("Failed to open file");
+    file.read_to_string(&mut read_content)
+        .expect("Failed to read file");
+
+    assert_eq!(read_content, test_content);
 }
 
 #[ic_cdk::update]

test/canisters/fs_tests/src/fs_tests_backend/src/main.rs

@@ -0,0 +1,5 @@
+mod lib;
+
+fn main() {
+    lib::basic_fs_test();
+}