Add wasi function required to run ripgrep

Signed-off-by: Ádám László Kulcsár <[email protected]>

Author: kulcsaradam

src/wasi/WASI.cpp

@@ -289,6 +289,14 @@ void WASI::fd_seek(ExecutionState& state, Value* argv, Value* result, Instance*
     result[0] = Value(uvwasi_fd_seek(WASI::g_uvwasi, fd, fileDelta, whence, file_size));
 }
 
+void WASI::fd_filestat_get(ExecutionState& state, Value* argv, Value* result, Instance* instance)
+{
+    uint32_t fd = argv[0].asI32();
+    uvwasi_filestat_t* buf = reinterpret_cast<uvwasi_filestat_t*>(get_memory_pointer(instance, argv[1], sizeof(uvwasi_filestat_t)));
+
+    result[0] = Value(uvwasi_fd_filestat_get(WASI::g_uvwasi, fd, buf));
+}
+
 void WASI::fd_advise(ExecutionState& state, Value* argv, Value* result, Instance* instance)
 {
     uint32_t fd = argv[0].asI32();
@@ -433,6 +441,11 @@ void WASI::random_get(ExecutionState& state, Value* argv, Value* result, Instanc
     result[0] = Value(uvwasi_random_get(WASI::g_uvwasi, buf, length));
 }
 
+void WASI::sched_yield(ExecutionState& state, Value* argv, Value* result, Instance* instance)
+{
+    result[0] = Value(uvwasi_sched_yield(WASI::g_uvwasi));
+}
+
 } // namespace Walrus
 
 #endif

src/wasi/WASI.h

@@ -51,6 +51,7 @@ class WASI {
     F(fd_fdstat_set_flags, I32I32_RI32)                    \
     F(fd_prestat_get, I32I32_RI32)                         \
     F(fd_prestat_dir_name, I32I32I32_RI32)                 \
+    F(fd_filestat_get, I32I32_RI32)                        \
     F(fd_seek, I32I64I32I32_RI32)                          \
     F(fd_advise, I32I64I64I32_RI32)                        \
     F(path_open, I32I32I32I32I32I64I64I32I32_RI32)         \
@@ -62,7 +63,8 @@ class WASI {
     F(path_unlink_file, I32I32I32_RI32)                    \
     F(poll_oneoff, I32I32I32I32_RI32)                      \
     F(environ_get, I32I32_RI32)                            \
-    F(environ_sizes_get, I32I32_RI32)
+    F(environ_sizes_get, I32I32_RI32)                      \
+    F(sched_yield, RI32)
 
 #define ERRORS(ERR)                                                   \
     ERR(success, "no error")                                          \

test/wasi/fd_filestat_get.wast

@@ -0,0 +1,59 @@
+(module
+  (import "wasi_snapshot_preview1" "path_open" (func $path_open (param i32 i32 i32 i32 i32 i64 i64 i32 i32) (result i32)))
+  (import "wasi_snapshot_preview1" "fd_close" (func $fd_close (param i32) (result i32)))
+  (import "wasi_snapshot_preview1" "fd_filestat_get" (func $fd_filestat_get (param i32 i32) (result i32)))
+
+  (memory 1 1)
+  (data (i32.const 100) "./fd_filestat_get.wast")
+
+  (func (export "fd_filestat_get") (result i32 i32 i32)
+    i32.const 3 ;; Directory file descriptior, by default 3 is the first opened directory
+    i32.const 1 ;; uvwasi_lookupflags_t: UVWASI_LOOKUP_SYMLINK_FOLLOW
+    i32.const 100 ;; Offset of file name in memory
+    i32.const 23 ;; Length of file name
+    i32.const 0 ;; uvwasi_oflags_t: none
+    i64.const 0x202000 ;; base uvwasi_rights_t: UVWASI_RIGHT_PATH_OPEN, UVWASI_RIGHT_FD_FILESTAT_GET
+    i64.const 0x202000 ;; inherited uvwasi_rights_t: UVWASI_RIGHT_PATH_OPEN, UVWASI_RIGHT_FD_FILESTAT_GET
+    i32.const 0 ;; uvwasi_fdflags_t: none
+    i32.const 0 ;; Offset to store at the opened file descriptor in memory
+    call $path_open
+
+	i32.const 0
+	i32.load
+	i32.const 1000
+	call $fd_filestat_get
+
+    i32.const 0
+    i32.load
+    call $fd_close
+  )
+
+  (func (export "fd_filestat_get_bad") (result i32 i32 i32 i32)
+    i32.const 3 ;; Directory file descriptior, by default 3 is the first opened directory
+    i32.const 1 ;; uvwasi_lookupflags_t: UVWASI_LOOKUP_SYMLINK_FOLLOW
+    i32.const 100 ;; Offset of file name in memory
+    i32.const 23 ;; Length of file name
+    i32.const 0 ;; uvwasi_oflags_t: none
+    i64.const 0x202000 ;; base uvwasi_rights_t: UVWASI_RIGHT_PATH_OPEN, UVWASI_RIGHT_FD_FILESTAT_GET
+    i64.const 0x202000 ;; inherited uvwasi_rights_t: UVWASI_RIGHT_PATH_OPEN, UVWASI_RIGHT_FD_FILESTAT_GET
+    i32.const 0 ;; uvwasi_fdflags_t: none
+    i32.const 0 ;; Offset to store at the opened file descriptor in memory
+    call $path_open
+
+	i32.const 0
+	i32.load
+	i32.const 65535
+	call $fd_filestat_get
+
+	i32.const -1
+	i32.const 1000
+	call $fd_filestat_get
+
+    i32.const 0
+    i32.load
+    call $fd_close
+  )
+)
+
+(assert_return (invoke "fd_filestat_get") (i32.const 0) (i32.const 0)  (i32.const 0))
+(assert_return (invoke "fd_filestat_get_bad") (i32.const 0) (i32.const 28) (i32.const 8) (i32.const 0))

test/wasi/sched_yield.wast

@@ -0,0 +1,8 @@
+(module
+    (import "wasi_snapshot_preview1" "sched_yield" (func $sched_yield (result i32)))
+	(func (export "sched_yield") (result i32)
+		call $sched_yield
+	)
+)
+
+(assert_return (invoke "sched_yield") (i32.const 0))