fix: break an infinite loop on file transfer

smallnest · Dec 5, 2023 · a069a9a · a069a9a
1 parent 2442c5b
commit a069a9a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/client/xclient.go b/client/xclient.go
@@ -1196,6 +1196,8 @@ loop:
 	for {
 		select {
 		case <-ctx.Done():
+			err = ctx.Err()
+			break loop
 		default:
 			if tb != nil {
 				tb.Wait(FileTransferBufferSize)
@@ -1257,6 +1259,8 @@ loop:
 	for {
 		select {
 		case <-ctx.Done():
+			err = ctx.Err()
+			break loop
 		default:
 			n, er := r.Read(buf)
 			if n > 0 {