mirror of
https://github.com/drakkan/sftpgo.git
synced 2025-12-07 14:50:55 +03:00
add support for data transfer bandwidth limits
with total limit or separate settings for uploads and downloads and overrides based on the client's IP address. Limits can be reset using the REST API Signed-off-by: Nicola Murino <nicola.murino@gmail.com>
This commit is contained in:
@@ -49,6 +49,9 @@ func (f *httpdFile) Read(p []byte) (n int, err error) {
|
||||
n, err = f.reader.Read(p)
|
||||
atomic.AddInt64(&f.BytesSent, int64(n))
|
||||
|
||||
if err == nil {
|
||||
err = f.CheckRead()
|
||||
}
|
||||
if err != nil && err != io.EOF {
|
||||
f.TransferError(err)
|
||||
return
|
||||
@@ -70,8 +73,8 @@ func (f *httpdFile) Write(p []byte) (n int, err error) {
|
||||
n, err = f.writer.Write(p)
|
||||
atomic.AddInt64(&f.BytesReceived, int64(n))
|
||||
|
||||
if f.MaxWriteSize > 0 && err == nil && atomic.LoadInt64(&f.BytesReceived) > f.MaxWriteSize {
|
||||
err = common.ErrQuotaExceeded
|
||||
if err == nil {
|
||||
err = f.CheckWrite()
|
||||
}
|
||||
if err != nil {
|
||||
f.TransferError(err)
|
||||
|
||||
Reference in New Issue
Block a user