diff options
Diffstat (limited to 'helper')
| -rw-r--r-- | helper/proc/fd.go | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/helper/proc/fd.go b/helper/proc/fd.go index e99abb38..abf8f36a 100644 --- a/helper/proc/fd.go +++ b/helper/proc/fd.go @@ -25,7 +25,7 @@ func Setup(extraFiles *[]*os.File) (int, *gob.Encoder, error) { // Receive retrieves payload pipe fd from the environment, // receives its payload and returns the Close method of the pipe. -func Receive(key string, e any) (func() error, error) { +func Receive(key string, e any, v **os.File) (func() error, error) { var setup *os.File if s, ok := os.LookupEnv(key); !ok { @@ -38,8 +38,11 @@ func Receive(key string, e any) (func() error, error) { if setup == nil { return nil, ErrInvalid } + if v != nil { + *v = setup + } } } - return func() error { return setup.Close() }, gob.NewDecoder(setup).Decode(e) + return setup.Close, gob.NewDecoder(setup).Decode(e) } |