77 lines
2.7 KiB
Go
77 lines
2.7 KiB
Go
// Package sys includes constants and types used by both public and internal APIs.
|
|
package sys
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
)
|
|
|
|
// These two special exit codes are reserved by wazero for context Cancel and Timeout integrations.
|
|
// The assumption here is that well-behaving Wasm programs won't use these two exit codes.
|
|
const (
|
|
// ExitCodeContextCanceled corresponds to context.Canceled and returned by ExitError.ExitCode in that case.
|
|
ExitCodeContextCanceled uint32 = 0xffffffff
|
|
// ExitCodeDeadlineExceeded corresponds to context.DeadlineExceeded and returned by ExitError.ExitCode in that case.
|
|
ExitCodeDeadlineExceeded uint32 = 0xefffffff
|
|
)
|
|
|
|
// ExitError is returned to a caller of api.Function when api.Module CloseWithExitCode was invoked,
|
|
// or context.Context passed to api.Function Call was canceled or reached the Timeout.
|
|
//
|
|
// ExitCode zero value means success while any other value is an error.
|
|
//
|
|
// Here's an example of how to get the exit code:
|
|
//
|
|
// main := module.ExportedFunction("main")
|
|
// if err := main(ctx); err != nil {
|
|
// if exitErr, ok := err.(*sys.ExitError); ok {
|
|
// // If your main function expects to exit, this could be ok if Code == 0
|
|
// }
|
|
// --snip--
|
|
//
|
|
// Note: While possible the reason of this was "proc_exit" from "wasi_snapshot_preview1", it could be from other host
|
|
// functions, for example an AssemblyScript's abort handler, or any arbitrary caller of CloseWithExitCode.
|
|
//
|
|
// See https://github.com/WebAssembly/WASI/blob/main/phases/snapshot/docs.md#proc_exit and
|
|
// https://www.assemblyscript.org/concepts.html#special-imports
|
|
//
|
|
// Note: In the case of context cancellation or timeout, the api.Module from which the api.Function created is closed.
|
|
type ExitError struct {
|
|
moduleName string
|
|
exitCode uint32
|
|
}
|
|
|
|
func NewExitError(moduleName string, exitCode uint32) *ExitError {
|
|
return &ExitError{moduleName: moduleName, exitCode: exitCode}
|
|
}
|
|
|
|
// ModuleName is the api.Module that was closed.
|
|
func (e *ExitError) ModuleName() string {
|
|
return e.moduleName
|
|
}
|
|
|
|
// ExitCode returns zero on success, and an arbitrary value otherwise.
|
|
func (e *ExitError) ExitCode() uint32 {
|
|
return e.exitCode
|
|
}
|
|
|
|
// Error implements the error interface.
|
|
func (e *ExitError) Error() string {
|
|
switch e.exitCode {
|
|
case ExitCodeContextCanceled:
|
|
return fmt.Sprintf("module %q closed with %s", e.moduleName, context.Canceled)
|
|
case ExitCodeDeadlineExceeded:
|
|
return fmt.Sprintf("module %q closed with %s", e.moduleName, context.DeadlineExceeded)
|
|
default:
|
|
return fmt.Sprintf("module %q closed with exit_code(%d)", e.moduleName, e.exitCode)
|
|
}
|
|
}
|
|
|
|
// Is allows use via errors.Is
|
|
func (e *ExitError) Is(err error) bool {
|
|
if target, ok := err.(*ExitError); ok {
|
|
return e.moduleName == target.moduleName && e.exitCode == target.exitCode
|
|
}
|
|
return false
|
|
}
|