wa-lang.org/wazero@v1.0.2/internal/gojs/fs.go

package gojs

import (
	"context"
	"fmt"
	"io"
	"io/fs"
	"os"
	"syscall"

	"wa-lang.org/wazero/api"
	"wa-lang.org/wazero/internal/wasm"
)

var (
	// jsfs = js.Global().Get("fs") // fs_js.go init
	//
	// js.fsCall conventions:
	// * funcWrapper callback is the last parameter
	//   * arg0 is error and up to one result in arg1
	jsfs = newJsVal(refJsfs, "fs").
		addProperties(map[string]interface{}{
			"constants": jsfsConstants, // = jsfs.Get("constants") // init
		}).
		addFunction("open", &jsfsOpen{}).
		addFunction("stat", &jsfsStat{}).
		addFunction("fstat", &jsfsFstat{}).
		addFunction("lstat", &jsfsStat{}). // because fs.FS doesn't support symlink
		addFunction("close", &jsfsClose{}).
		addFunction("read", &jsfsRead{}).
		addFunction("write", &jsfsWrite{}).
		addFunction("readdir", &jsfsReaddir{})

	// TODO: stub all these with syscall.ENOSYS
	//	* _, err := fsCall("mkdir", path, perm) // syscall.Mkdir
	//	* _, err := fsCall("unlink", path) // syscall.Unlink
	//	* _, err := fsCall("rmdir", path) // syscall.Rmdir
	//	* _, err := fsCall("chmod", path, mode) // syscall.Chmod
	//	* _, err := fsCall("fchmod", fd, mode) // syscall.Fchmod
	//	* _, err := fsCall("chown", path, uint32(uid), uint32(gid)) // syscall.Chown
	//	* _, err := fsCall("fchown", fd, uint32(uid), uint32(gid)) // syscall.Fchown
	//	* _, err := fsCall("lchown", path, uint32(uid), uint32(gid)) // syscall.Lchown
	//	* _, err := fsCall("utimes", path, atime, mtime) // syscall.UtimesNano
	//	* _, err := fsCall("rename", from, to) // syscall.Rename
	//	* _, err := fsCall("truncate", path, length) // syscall.Truncate
	//	* _, err := fsCall("ftruncate", fd, length) // syscall.Ftruncate
	//	* dst, err := fsCall("readlink", path) // syscall.Readlink
	//	* _, err := fsCall("link", path, link) // syscall.Link
	//	* _, err := fsCall("symlink", path, link) // syscall.Symlink
	//	* _, err := fsCall("fsync", fd) // syscall.Fsync

	// jsfsConstants = jsfs Get("constants") // fs_js.go init
	jsfsConstants = newJsVal(refJsfsConstants, "constants").
			addProperties(map[string]interface{}{
			"O_WRONLY": oWRONLY,
			"O_RDWR":   oRDWR,
			"O_CREAT":  oCREAT,
			"O_TRUNC":  oTRUNC,
			"O_APPEND": oAPPEND,
			"O_EXCL":   oEXCL,
		})

	// oWRONLY = jsfsConstants Get("O_WRONLY").Int() // fs_js.go init
	oWRONLY = api.EncodeF64(float64(os.O_WRONLY))

	// oRDWR = jsfsConstants Get("O_RDWR").Int() // fs_js.go init
	oRDWR = api.EncodeF64(float64(os.O_RDWR))

	// o CREAT = jsfsConstants Get("O_CREAT").Int() // fs_js.go init
	oCREAT = api.EncodeF64(float64(os.O_CREATE))

	// oTRUNC = jsfsConstants Get("O_TRUNC").Int() // fs_js.go init
	oTRUNC = api.EncodeF64(float64(os.O_TRUNC))

	// oAPPEND = jsfsConstants Get("O_APPEND").Int() // fs_js.go init
	oAPPEND = api.EncodeF64(float64(os.O_APPEND))

	// oEXCL = jsfsConstants Get("O_EXCL").Int() // fs_js.go init
	oEXCL = api.EncodeF64(float64(os.O_EXCL))
)

// jsfsOpen implements fs.Open
//
//	jsFD /* Int */, err := fsCall("open", path, flags, perm)
type jsfsOpen struct{}

// invoke implements jsFn.invoke
func (*jsfsOpen) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	name := args[0].(string)
	flags := toUint32(args[1]) // flags are derived from constants like oWRONLY
	perm := toUint32(args[2])
	callback := args[3].(funcWrapper)

	fd, err := syscallOpen(ctx, mod, name, flags, perm)
	return callback.invoke(ctx, mod, refJsfs, err, fd) // note: error first
}

// jsfsStat is used for syscall.Stat
//
//	jsSt, err := fsCall("stat", path)
type jsfsStat struct{}

// invoke implements jsFn.invoke
func (*jsfsStat) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	name := args[0].(string)
	callback := args[1].(funcWrapper)

	stat, err := syscallStat(ctx, mod, name)
	return callback.invoke(ctx, mod, refJsfs, err, stat) // note: error first
}

// syscallStat is like syscall.Stat
func syscallStat(ctx context.Context, mod api.Module, name string) (*jsSt, error) {
	fsc := mod.(*wasm.CallContext).Sys.FS(ctx)
	if fd, err := fsc.OpenFile(ctx, name); err != nil {
		return nil, err
	} else {
		defer fsc.CloseFile(ctx, fd)
		return syscallFstat(ctx, mod, fd)
	}
}

// jsfsStat is used for syscall.Open
//
//	stat, err := fsCall("fstat", fd); err == nil && stat.Call("isDirectory").Bool()
type jsfsFstat struct{}

// invoke implements jsFn.invoke
func (*jsfsFstat) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	fd := toUint32(args[0])
	callback := args[1].(funcWrapper)

	fstat, err := syscallFstat(ctx, mod, fd)
	return callback.invoke(ctx, mod, refJsfs, err, fstat) // note: error first
}

// syscallFstat is like syscall.Fstat
func syscallFstat(ctx context.Context, mod api.Module, fd uint32) (*jsSt, error) {
	fsc := mod.(*wasm.CallContext).Sys.FS(ctx)
	if f, ok := fsc.OpenedFile(ctx, fd); !ok {
		return nil, syscall.EBADF
	} else if stat, err := f.File.Stat(); err != nil {
		return nil, err
	} else {
		ret := &jsSt{}
		ret.isDir = stat.IsDir()
		// TODO ret.dev=stat.Sys
		ret.mode = uint32(stat.Mode())
		ret.size = uint32(stat.Size())
		ret.mtimeMs = uint32(stat.ModTime().UnixMilli())
		return ret, nil
	}
}

// jsfsClose is used for syscall.Close
//
//	_, err := fsCall("close", fd)
type jsfsClose struct{}

// invoke implements jsFn.invoke
func (*jsfsClose) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	fd := toUint32(args[0])
	callback := args[1].(funcWrapper)

	err := syscallClose(ctx, mod, fd)
	return callback.invoke(ctx, mod, refJsfs, err, true) // note: error first
}

// syscallClose is like syscall.Close
func syscallClose(ctx context.Context, mod api.Module, fd uint32) (err error) {
	fsc := mod.(*wasm.CallContext).Sys.FS(ctx)
	if ok := fsc.CloseFile(ctx, fd); !ok {
		err = syscall.EBADF // already closed
	}
	return
}

// jsfsRead is used in syscall.Read and syscall.Pread, called by
// src/internal/poll/fd_unix.go poll.Read.
//
//	n, err := fsCall("read", fd, buf, 0, len(b), nil)
type jsfsRead struct{}

// invoke implements jsFn.invoke
func (*jsfsRead) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	fd := toUint32(args[0])
	buf, ok := args[1].(*byteArray)
	if !ok {
		return nil, fmt.Errorf("arg[1] is %v not a []byte", args[1])
	}
	offset := toUint32(args[2])
	byteCount := toUint32(args[3])
	fOffset := args[4] // nil unless Pread
	callback := args[5].(funcWrapper)

	n, err := syscallRead(ctx, mod, fd, fOffset, buf.slice[offset:offset+byteCount])
	return callback.invoke(ctx, mod, refJsfs, err, n) // note: error first
}

// syscallRead is like syscall.Read
func syscallRead(ctx context.Context, mod api.Module, fd uint32, offset interface{}, p []byte) (n uint32, err error) {
	r := fdReader(ctx, mod, fd)
	if r == nil {
		err = syscall.EBADF
	}

	if offset != nil {
		if s, ok := r.(io.Seeker); ok {
			if _, err := s.Seek(toInt64(offset), io.SeekStart); err != nil {
				return 0, err
			}
		} else {
			return 0, syscall.ENOTSUP
		}
	}

	if nRead, e := r.Read(p); e == nil || e == io.EOF {
		// fs_js.go cannot parse io.EOF so coerce it to nil.
		// See https://github.com/golang/go/issues/43913
		n = uint32(nRead)
	} else {
		err = e
	}
	return
}

// jsfsWrite is used in syscall.Write and syscall.Pwrite.
//
// Notably, offset is non-nil in Pwrite.
//
//	n, err := fsCall("write", fd, buf, 0, len(b), nil)
type jsfsWrite struct{}

// invoke implements jsFn.invoke
func (*jsfsWrite) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	fd := toUint32(args[0])
	buf, ok := args[1].(*byteArray)
	if !ok {
		return nil, fmt.Errorf("arg[1] is %v not a []byte", args[1])
	}
	offset := toUint32(args[2])
	byteCount := toUint32(args[3])
	fOffset := args[4] // nil unless Pread
	callback := args[5].(funcWrapper)

	if byteCount > 0 { // empty is possible on EOF
		n, err := syscallWrite(ctx, mod, fd, fOffset, buf.slice[offset:offset+byteCount])
		return callback.invoke(ctx, mod, refJsfs, err, n) // note: error first
	}
	return callback.invoke(ctx, mod, refJsfs, nil, refValueZero)
}

// syscallWrite is like syscall.Write
func syscallWrite(ctx context.Context, mod api.Module, fd uint32, offset interface{}, p []byte) (n uint32, err error) {
	if writer := fdWriter(ctx, mod, fd); writer == nil {
		err = syscall.EBADF
	} else if nWritten, e := writer.Write(p); e == nil || e == io.EOF {
		// fs_js.go cannot parse io.EOF so coerce it to nil.
		// See https://github.com/golang/go/issues/43913
		n = uint32(nWritten)
	} else {
		err = e
	}
	return
}

// jsfsReaddir is used in syscall.Open
//
//	dir, err := fsCall("readdir", path)
//		dir.Length(), dir.Index(i).String()
type jsfsReaddir struct{}

// invoke implements jsFn.invoke
func (*jsfsReaddir) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	name := args[0].(string)
	callback := args[1].(funcWrapper)

	stat, err := syscallReaddir(ctx, mod, name)
	return callback.invoke(ctx, mod, refJsfs, err, stat) // note: error first
}

func syscallReaddir(ctx context.Context, mod api.Module, name string) (*objectArray, error) {
	fsc := mod.(*wasm.CallContext).Sys.FS(ctx)
	fd, err := fsc.OpenFile(ctx, name)
	if err != nil {
		return nil, err
	}
	defer fsc.CloseFile(ctx, fd)

	if f, ok := fsc.OpenedFile(ctx, fd); !ok {
		return nil, syscall.EBADF
	} else if d, ok := f.File.(fs.ReadDirFile); !ok {
		return nil, syscall.ENOTDIR
	} else if l, err := d.ReadDir(-1); err != nil {
		return nil, err
	} else {
		entries := make([]interface{}, 0, len(l))
		for _, e := range l {
			entries = append(entries, e.Name())
		}
		return &objectArray{entries}, nil
	}
}

type returnZero struct{}

// invoke implements jsFn.invoke
func (*returnZero) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	return refValueZero, nil
}

type returnSliceOfZero struct{}

// invoke implements jsFn.invoke
func (*returnSliceOfZero) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	return &objectArray{slice: []interface{}{refValueZero}}, nil
}

type returnArg0 struct{}

// invoke implements jsFn.invoke
func (*returnArg0) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	return args[0], nil
}

// cwd for fs.Open syscall.Getcwd in fs_js.go
type cwd struct{}

// invoke implements jsFn.invoke
func (*cwd) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	return getState(ctx).cwd, nil
}

// chdir for fs.Open syscall.Chdir in fs_js.go
type chdir struct{}

// invoke implements jsFn.invoke
func (*chdir) invoke(ctx context.Context, mod api.Module, args ...interface{}) (interface{}, error) {
	path := args[0].(string)

	// TODO: refactor so that sys has path-based ops, also needed in WASI.
	fsc := mod.(*wasm.CallContext).Sys.FS(ctx)
	if fd, err := fsc.OpenFile(ctx, path); err != nil {
		return nil, syscall.ENOENT
	} else if f, ok := fsc.OpenedFile(ctx, fd); !ok {
		return nil, syscall.ENOENT
	} else if s, err := f.File.Stat(); err != nil {
		fsc.CloseFile(ctx, fd)
		return nil, syscall.ENOENT
	} else if !s.IsDir() {
		fsc.CloseFile(ctx, fd)
		return nil, syscall.ENOTDIR
	} else {
		getState(ctx).cwd = path
		return nil, nil
	}
}

// jsSt is pre-parsed from fs_js.go setStat to avoid thrashing
type jsSt struct {
	isDir bool

	dev     uint32
	ino     uint32
	mode    uint32
	nlink   uint32
	uid     uint32
	gid     uint32
	rdev    uint32
	size    uint32
	blksize uint32
	blocks  uint32
	atimeMs uint32
	mtimeMs uint32
	ctimeMs uint32
}

// get implements jsGet.get
func (s *jsSt) get(_ context.Context, propertyKey string) interface{} {
	switch propertyKey {
	case "dev":
		return s.dev
	case "ino":
		return s.ino
	case "mode":
		return s.mode
	case "nlink":
		return s.nlink
	case "uid":
		return s.uid
	case "gid":
		return s.gid
	case "rdev":
		return s.rdev
	case "size":
		return s.size
	case "blksize":
		return s.blksize
	case "blocks":
		return s.blocks
	case "atimeMs":
		return s.atimeMs
	case "mtimeMs":
		return s.mtimeMs
	case "ctimeMs":
		return s.ctimeMs
	}
	panic(fmt.Sprintf("TODO: stat.%s", propertyKey))
}

// call implements jsCall.call
func (s *jsSt) call(ctx context.Context, mod api.Module, this ref, method string, args ...interface{}) (interface{}, error) {
	if method == "isDirectory" {
		return s.isDir, nil
	}
	panic(fmt.Sprintf("TODO: stat.%s", method))
}