sqlite3/vfs_file.go

package sqlite3

import (
	"context"
	"os"
	"time"

	"github.com/tetratelabs/wazero/api"
)

const (
	// These need to match the offsets asserted in os.c
	vfsFileIDOffset          = 4
	vfsFileLockOffset        = 8
	vfsFileLockTimeoutOffset = 12
)

func (vfsFileMethods) NewID(ctx context.Context, file *os.File) uint32 {
	vfs := ctx.Value(vfsKey{}).(*vfsState)

	// Find an empty slot.
	for id, ptr := range vfs.files {
		if ptr == nil {
			vfs.files[id] = file
			return uint32(id)
		}
	}

	// Add a new slot.
	vfs.files = append(vfs.files, file)
	return uint32(len(vfs.files) - 1)
}

func (vfsFileMethods) Open(ctx context.Context, mod api.Module, pFile uint32, file *os.File) {
	mem := memory{mod}
	id := vfsFile.NewID(ctx, file)
	mem.writeUint32(pFile+vfsFileIDOffset, id)
}

func (vfsFileMethods) Close(ctx context.Context, mod api.Module, pFile uint32) error {
	mem := memory{mod}
	id := mem.readUint32(pFile + vfsFileIDOffset)
	vfs := ctx.Value(vfsKey{}).(*vfsState)
	file := vfs.files[id]
	vfs.files[id] = nil
	return file.Close()
}

func (vfsFileMethods) GetOS(ctx context.Context, mod api.Module, pFile uint32) *os.File {
	mem := memory{mod}
	id := mem.readUint32(pFile + vfsFileIDOffset)
	vfs := ctx.Value(vfsKey{}).(*vfsState)
	return vfs.files[id]
}

func (vfsFileMethods) GetLock(ctx context.Context, mod api.Module, pFile uint32) vfsLockState {
	mem := memory{mod}
	return vfsLockState(mem.readUint8(pFile + vfsFileLockOffset))
}

func (vfsFileMethods) SetLock(ctx context.Context, mod api.Module, pFile uint32, lock vfsLockState) {
	mem := memory{mod}
	mem.writeUint8(pFile+vfsFileLockOffset, uint8(lock))
}

func (vfsFileMethods) GetLockTimeout(ctx context.Context, mod api.Module, pFile uint32) time.Duration {
	mem := memory{mod}
	return time.Duration(mem.readUint32(pFile+vfsFileLockTimeoutOffset)) * time.Millisecond
}