修复sftp文件上传多出部分文件字符的bug

This commit is contained in:
dushixiang 2021-02-14 14:43:29 +08:00
parent 6f7ac9e462
commit d5d775fb4e

View File

@ -267,11 +267,15 @@ func SessionUploadEndpoint(c echo.Context) error {
buf := make([]byte, 1024)
for {
n, _ := src.Read(buf)
if n == 0 {
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 {