From 61e58aa14da2f9ce45b49d38b889a00ef5a3b1d1 Mon Sep 17 00:00:00 2001 From: Ophestra Date: Sat, 8 Mar 2025 23:24:39 +0900 Subject: helper/proc: expose setup file Signed-off-by: Ophestra --- helper/proc/fd.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'helper') 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) } -- cgit v1.3.1