Ophestra Umiker
7f29b37a32
Generic setup payload encoder adapted from fshim. Signed-off-by: Ophestra Umiker <cat@ophivana.moe>
43 lines
841 B
Go
43 lines
841 B
Go
package proc
|
|
|
|
import (
|
|
"encoding/gob"
|
|
"errors"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
var (
|
|
ErrNotSet = errors.New("environment variable not set")
|
|
ErrInvalid = errors.New("bad file descriptor")
|
|
)
|
|
|
|
func Setup(extraFiles *[]*os.File) (int, *gob.Encoder, error) {
|
|
if r, w, err := os.Pipe(); err != nil {
|
|
return -1, nil, err
|
|
} else {
|
|
fd := 3 + len(*extraFiles)
|
|
*extraFiles = append(*extraFiles, r)
|
|
return fd, gob.NewEncoder(w), nil
|
|
}
|
|
}
|
|
|
|
func Receive(key string, e any) (func() error, error) {
|
|
var setup *os.File
|
|
|
|
if s, ok := os.LookupEnv(key); !ok {
|
|
return nil, ErrNotSet
|
|
} else {
|
|
if fd, err := strconv.Atoi(s); err != nil {
|
|
return nil, err
|
|
} else {
|
|
setup = os.NewFile(uintptr(fd), "setup")
|
|
if setup == nil {
|
|
return nil, ErrInvalid
|
|
}
|
|
}
|
|
}
|
|
|
|
return func() error { return setup.Close() }, gob.NewDecoder(setup).Decode(e)
|
|
}
|