修复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) buf := make([]byte, 1024)
for { for {
n, _ := src.Read(buf) n, err := src.Read(buf)
if n == 0 { if err != nil {
break if err != io.EOF {
logrus.Warnf("文件上传错误 %v", err)
} else {
break
}
} }
_, _ = dstFile.Write(buf) _, _ = dstFile.Write(buf[:n])
} }
return Success(c, nil) return Success(c, nil)
} else if "rdp" == session.Protocol { } else if "rdp" == session.Protocol {