From d5d775fb4e5881f43a90b8dd237bc433bb1834b9 Mon Sep 17 00:00:00 2001
From: dushixiang <798148596@qq.com>
Date: Sun, 14 Feb 2021 14:43:29 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dsftp=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E5=A4=9A=E5=87=BA=E9=83=A8=E5=88=86=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E5=AD=97=E7=AC=A6=E7=9A=84bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/api/session.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/pkg/api/session.go b/pkg/api/session.go
index 2f3bbc7..619d4d4 100644
--- a/pkg/api/session.go
+++ b/pkg/api/session.go
@@ -267,11 +267,15 @@ func SessionUploadEndpoint(c echo.Context) error {
 
 		buf := make([]byte, 1024)
 		for {
-			n, _ := src.Read(buf)
-			if n == 0 {
-				break
+			n, err := src.Read(buf)
+			if err != nil {
+				if err != io.EOF {
+					logrus.Warnf("文件上传错误 %v", err)
+				} else {
+					break
+				}
 			}
-			_, _ = dstFile.Write(buf)
+			_, _ = dstFile.Write(buf[:n])
 		}
 		return Success(c, nil)
 	} else if "rdp" == session.Protocol {