Move embedded remote helpers to cmd/safcm/

[safcm/safcm.git] / rpc / conn.go

diff --git a/rpc/conn.go b/rpc/conn.go
index 9786941c7e808966a9a76841bfd6e03981f81ba6..3f7ade144c1c2905254da49eb7924552aa3c362d 100644 (file)
--- a/rpc/conn.go
+++ b/rpc/conn.go
@@ -20,6 +20,7 @@ package rpc
 import (
        "bufio"
        "fmt"
+       "io/fs"
        "os/exec"
        "strings"
        "sync"
@@ -36,6 +37,8 @@ type Conn struct {
        sshRemote string
        sshOpts   []string
 
+       remoteHelpers fs.FS
+
        cmd  *exec.Cmd
        conn *safcm.GobConn
 }
@@ -124,9 +127,14 @@ func (c *Conn) wait() error {
 }
 
 // Kill forcefully terminates the connection. It's safe to call Kill (and
-// Wait) multiple times.
+// Wait) multiple times. Calling it before Dial*() was called will only close
+// the Events channel.
 func (c *Conn) Kill() error {
        if c.cmd == nil {
+               if c.events != nil {
+                       close(c.events)
+                       c.events = nil
+               }
                return fmt.Errorf("Dial*() not called or already terminated")
        }