Backport winrm panic workaround

Signed-off-by: Kimmo Lehto <klehto@mirantis.com>
k0sproject · Jun 5, 2024 · d79e782 · d79e782
1 parent 566086e
commit d79e782
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/winrm.go b/winrm.go
@@ -198,6 +198,16 @@ type Command struct {
 
 // Wait blocks until the command finishes
 func (c *Command) Wait() error {
+	defer func() {
+		if r := recover(); err == nil && r != nil {
+			if strings.Contains(fmt.Sprint(r), "close of closed channel") {
+				log.Debugf("recovered from a panic in Command.Wait: %v", r)
+			} else {
+				panic(r)
+			}
+		}
+	}()
+
 	defer c.sh.Close()
 	defer c.cmd.Close()