Mirrors/sftpgo
1
0
Fork 0
mirror of https://github.com/drakkan/sftpgo.git synced 2025-12-07 23:00:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

scp: close transfers before sending upload errors

Browse Source 
This change should fix the random failure in TestSCPTransferQuotaLimits
because the quota is already updated when the scp command ends.

Signed-off-by: Nicola Murino <nicola.murino@gmail.com>
This commit is contained in:
Nicola Murino
2023-12-21 18:03:07 +01:00
parent d396c24ad4
commit 61b5f97bf2
5 changed files with 37 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
internal/sftpd/scp.go
View File

@@ -192,15 +192,15 @@ func (c *scpCommand) getUploadFileData(sizeToRead int64, transfer *transfer) err
for {
n, err := c.connection.channel.Read(buf)
if err != nil {
c.sendErrorMessage(transfer.Fs, err)
transfer.TransferError(err)
transfer.Close()
c.sendErrorMessage(transfer.Fs, err)
return err
}
_, err = transfer.WriteAt(buf[:n], sizeToRead-remaining)
if err != nil {
c.sendErrorMessage(transfer.Fs, err)
transfer.Close()
c.sendErrorMessage(transfer.Fs, err)
return err
}
remaining -= int64(n)