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

httpd/webdav: use a custom listener with read and write deadlines

Browse Source
This commit is contained in:
Nicola Murino
2021-05-07 20:41:20 +02:00
parent d4bfc3f6b5
commit 4ea4202b99
12 changed files with 133 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
webdavd/webdavd_test.go
View File

@@ -2348,6 +2348,17 @@ func checkBasicFunc(client *gowebdav.Client) error {
return err
}
func checkFileSize(remoteDestPath string, expectedSize int64, client *gowebdav.Client) error {
info, err := client.Stat(remoteDestPath)
if err != nil {
return err
}
if info.Size() != expectedSize {
return fmt.Errorf("uploaded file size does not match, actual: %v, expected: %v", info.Size(), expectedSize)
}
return nil
}
func uploadFile(localSourcePath string, remoteDestPath string, expectedSize int64, client *gowebdav.Client) error {
srcFile, err := os.Open(localSourcePath)
if err != nil {
@@ -2359,12 +2370,10 @@ func uploadFile(localSourcePath string, remoteDestPath string, expectedSize int6
return err
}
if expectedSize > 0 {
info, err := client.Stat(remoteDestPath)
err = checkFileSize(remoteDestPath, expectedSize, client)
if err != nil {
return err
}
if info.Size() != expectedSize {
return fmt.Errorf("uploaded file size does not match, actual: %v, expected: %v", info.Size(), expectedSize)
time.Sleep(1 * time.Second)
return checkFileSize(remoteDestPath, expectedSize, client)
}
}
return nil