fix closed the remote connection on pull causes service panic

Signed-off-by: zhangguanzhang <zhangguanzhang@qq.com>

1 files changed, 11 insertions, 2 deletions

diff --git a/pkg/channel/writer.go b/pkg/channel/writer.go
index dbb38e416..28c9d7de5 100644
--- a/pkg/channel/writer.go
+++ b/pkg/channel/writer.go
@@ -1,6 +1,7 @@
 package channel
 
 import (
+	"fmt"
 	"io"
 	"sync"
 
@@ -32,16 +33,24 @@ func (w *writeCloser) Chan() <-chan []byte {
 }
 
 // Write method for WriteCloser
-func (w *writeCloser) Write(b []byte) (int, error) {
+func (w *writeCloser) Write(b []byte) (bLen int, err error) {
+	// https://github.com/containers/podman/issues/7896
+	// when podman-remote pull image, if it was killed, the server will panic: send on closed channel
+	// so handle it
+	defer func() {
+		if rErr := recover(); rErr != nil {
+			err = fmt.Errorf("%s", rErr)
+		}
+	}()
 	if w == nil || w.ch == nil {
 		return 0, errors.New("use channel.NewWriter() to initialize a WriteCloser")
 	}
 
 	w.mux.Lock()
+	defer w.mux.Unlock()
 	buf := make([]byte, len(b))
 	copy(buf, b)
 	w.ch <- buf
-	w.mux.Unlock()
 
 	return len(b), nil
 }