2020-09-01 03:14:04 +00:00
|
|
|
package sftp
|
|
|
|
|
|
|
|
import (
|
|
|
|
"io"
|
|
|
|
"os"
|
|
|
|
)
|
|
|
|
|
2021-01-10 22:25:39 +00:00
|
|
|
const (
|
|
|
|
// Extends the default SFTP server to return a quota exceeded error to the client.
|
|
|
|
//
|
|
|
|
// @see https://tools.ietf.org/id/draft-ietf-secsh-filexfer-13.txt
|
|
|
|
ErrSSHQuotaExceeded = fxerr(15)
|
|
|
|
)
|
2020-09-01 03:14:04 +00:00
|
|
|
|
2021-01-10 22:43:27 +00:00
|
|
|
type ListerAt []os.FileInfo
|
|
|
|
|
2020-09-01 03:14:04 +00:00
|
|
|
// Returns the number of entries copied and an io.EOF error if we made it to the end of the file list.
|
|
|
|
// Take a look at the pkg/sftp godoc for more information about how this function should work.
|
|
|
|
func (l ListerAt) ListAt(f []os.FileInfo, offset int64) (int, error) {
|
|
|
|
if offset >= int64(len(l)) {
|
|
|
|
return 0, io.EOF
|
|
|
|
}
|
|
|
|
|
|
|
|
if n := copy(f, l[offset:]); n < len(f) {
|
|
|
|
return n, io.EOF
|
|
|
|
} else {
|
|
|
|
return n, nil
|
|
|
|
}
|
|
|
|
}
|
2021-01-10 22:25:39 +00:00
|
|
|
|
2021-01-10 22:43:27 +00:00
|
|
|
type fxerr uint32
|
|
|
|
|
2021-01-10 22:25:39 +00:00
|
|
|
func (e fxerr) Error() string {
|
|
|
|
switch e {
|
|
|
|
case ErrSSHQuotaExceeded:
|
|
|
|
return "Quota Exceeded"
|
|
|
|
default:
|
|
|
|
return "Failure"
|
|
|
|
}
|
|
|
|
}
|