Basic SFTP support, Issue #521

This commit is contained in:
Jack Schmidt 2016-11-12 18:36:08 -05:00 committed by Nick Craig-Wood
parent 27e18b6efa
commit b9a8315696
2 changed files with 422 additions and 0 deletions

View file

@ -12,6 +12,7 @@ import (
_ "github.com/ncw/rclone/local"
_ "github.com/ncw/rclone/onedrive"
_ "github.com/ncw/rclone/s3"
_ "github.com/ncw/rclone/sftp"
_ "github.com/ncw/rclone/swift"
_ "github.com/ncw/rclone/yandex"
)

421
sftp/sftp.go Normal file
View file

@ -0,0 +1,421 @@
// Package sftp provides a filesystem interface using github.com/pkg/sftp
package sftp
import (
"io"
"net"
"os"
"path"
"path/filepath"
"sync"
"time"
"golang.org/x/crypto/ssh"
"golang.org/x/crypto/ssh/agent"
"github.com/ncw/rclone/fs"
"github.com/pkg/sftp"
)
func init() {
fsi := &fs.RegInfo{
Name: "sftp",
Description: "SSH/SFTP Connection",
NewFs: NewFs,
Options: []fs.Option{{
Name: "host",
Help: "SSH host to connect to",
Optional: false,
Examples: []fs.OptionExample{{
Value: "example.com",
Help: "Connect to example.com",
}},
}, {
Name: "user",
Help: "SSH username, leave blank for current username, " + os.Getenv("USER"),
Optional: true,
}, {
Name: "port",
Help: "SSH port",
Optional: true,
}, {
Name: "pass",
Help: "SSH password, leave blank to use ssh-agent",
Optional: true,
IsPassword: true,
}},
}
fs.Register(fsi)
}
type stringer interface {
String() string
}
func debug(o stringer, msg string) {
fs.Debug(o, msg)
}
// Fs stores the interface to the remote SFTP files
type Fs struct {
name string
root string
features *fs.Features // optional features
url string
sshClient *ssh.Client
sftpClient *sftp.Client
}
// Object is a remote SFTP file that has been stat'd (so it exists, but is not necessarily open for reading)
type Object struct {
fs *Fs
remote string
info os.FileInfo
}
// ObjectReader holds the sftp.File interface to a remote SFTP file opened for reading
type ObjectReader struct {
object *Object
sftpFile *sftp.File
}
// NewFs creates a new Fs object from the name and root. It connects to
// the host specified in the config file.
func NewFs(name, root string) (fs.Fs, error) {
user := fs.ConfigFileGet(name, "user")
host := fs.ConfigFileGet(name, "host")
port := fs.ConfigFileGet(name, "port")
pass := fs.ConfigFileGet(name, "pass")
if root == "" {
root = "."
}
if user == "" {
user = os.Getenv("USER")
}
if port == "" {
port = "22"
}
config := &ssh.ClientConfig{
User: user,
Auth: []ssh.AuthMethod{},
}
if pass == "" {
if sshAgent, err := net.Dial("unix", os.Getenv("SSH_AUTH_SOCK")); err == nil {
sshAgentClient := agent.NewClient(sshAgent)
signers, _ := sshAgentClient.Signers()
for i, signer := range signers {
if 2*i < len(signers) {
signers[i] = signers[len(signers)-i-1]
signers[len(signers)-i-1] = signer
}
}
config.Auth = append(config.Auth, ssh.PublicKeys(signers...))
}
} else {
clearpass, err := fs.Reveal(pass)
if err != nil {
return nil, err
}
config.Auth = append(config.Auth, ssh.Password(clearpass))
}
if sshClient, err := ssh.Dial("tcp", host+":"+port, config); err != nil {
return nil, err
} else if sftpClient, err := sftp.NewClient(sshClient); err != nil {
_ = sshClient.Close()
return nil, err
} else {
f := &Fs{
name: name,
root: root,
sshClient: sshClient,
sftpClient: sftpClient,
url: "sftp://" + user + "@" + host + ":" + port + "/" + root,
}
f.features = (&fs.Features{}).Fill(f)
return f, nil
}
}
// Name returns the configured name of the file system
func (f *Fs) Name() string {
return f.name
}
// Root returns the root for the filesystem
func (f *Fs) Root() string {
return f.root
}
// String returns the URL for the filesystem
func (f *Fs) String() string {
return f.url
}
// Features returns the optional features of this Fs
func (f *Fs) Features() *fs.Features {
return f.features
}
// Precision is the remote sftp file system's modtime precision, which we have no way of knowing. We estimate at 1s
func (f *Fs) Precision() time.Duration {
return time.Second
}
// NewObject creates a new remote sftp file object
func (f *Fs) NewObject(remote string) (fs.Object, error) {
debug(f, "New '"+remote+"'")
info, err := f.sftpClient.Stat(f.sftpClient.Join(f.root, remote))
if err != nil {
return nil, err
}
object := &Object{
fs: f,
remote: remote,
info: info,
}
return object, nil
}
func (f *Fs) list(out fs.ListOpts, dirs string, name string, info os.FileInfo, level int, done *sync.WaitGroup) {
debug(f, "list '"+f.sftpClient.Join(dirs, name)+"'")
defer done.Done()
if info.IsDir() {
if out.IncludeDirectory(info.Name()) {
dir := &fs.Dir{
Name: info.Name(),
When: info.ModTime(),
Bytes: -1,
Count: -1,
}
if level >= out.Level() {
return
}
if infos, err := f.sftpClient.ReadDir(f.sftpClient.Join(f.root, dirs, name)); err == nil {
dir.Count = int64(len(infos))
out.AddDir(dir)
done.Add(len(infos))
for _, newInfo := range infos {
go f.list(out, f.sftpClient.Join(dirs, name), newInfo.Name(), newInfo, level+1, done)
}
}
}
} else {
file := &Object{
fs: f,
remote: f.sftpClient.Join(dirs, info.Name()),
info: info,
}
out.Add(file)
}
}
// List the files and directories starting at <dir>
func (f *Fs) List(out fs.ListOpts, dir string) {
debug(f, "List '"+dir+"'")
if dir == "" {
dir = "."
}
var done sync.WaitGroup
if info, _ := f.sftpClient.Stat(f.sftpClient.Join(f.root, dir)); info != nil {
done.Add(1)
f.list(out, "", ".", info, 0, &done)
}
debug(f, "List--waiting")
done.Wait()
out.Finished()
}
// Put data from <in> into a new remote sftp file object described by <src.Remote()> and <src.ModTime()>
func (f *Fs) Put(in io.Reader, src fs.ObjectInfo) (fs.Object, error) {
debug(f, "Put '"+src.Remote()+"'")
_ = f.mkdir(f.sftpClient.Join(f.root, filepath.Dir(src.Remote())))
file, err := f.sftpClient.Create(f.sftpClient.Join(f.root, src.Remote()))
if err != nil {
return nil, err
}
_, err = file.ReadFrom(in)
if err != nil {
return nil, err
}
o, err := f.NewObject(src.Remote())
if err != nil {
return nil, err
}
err = o.SetModTime(src.ModTime())
if err != nil {
return nil, err
}
return o, nil
}
func (f *Fs) mkdir(path string) error {
debug(f, "mkdir '"+path+"'")
parent := filepath.Dir(path)
if parent != "." && parent != "/" {
_ = f.mkdir(parent)
}
return f.sftpClient.Mkdir(path)
}
// Mkdir makes the root directory of the Fs object
func (f *Fs) Mkdir(dir string) error {
root := path.Join(f.root, dir)
debug(f, "Mkdir '"+root+"'")
o, _ := f.NewObject("")
if o == nil {
return f.mkdir(root)
}
return nil
}
// Rmdir removes the root directory of the Fs object
func (f *Fs) Rmdir(dir string) error {
root := path.Join(f.root, dir)
debug(f, "Rmdir '"+root+"'")
return f.sftpClient.Remove(f.root)
}
// Move renames a remote sftp file object
func (f *Fs) Move(src fs.Object, remote string) (fs.Object, error) {
debug(f, "Move '"+src.Remote()+"' to '"+remote+"'")
err := f.sftpClient.Rename(
f.sftpClient.Join(f.root, src.Remote()),
f.sftpClient.Join(f.root, remote))
if err != nil {
return nil, err
}
dstObj, err := f.NewObject(remote)
return dstObj, err
}
// Hashes returns fs.HashNone to indicate remote hashing is unavailable
func (f *Fs) Hashes() fs.HashSet {
return fs.HashSet(fs.HashNone)
}
// Fs is the filesystem this remote sftp file object is located within
func (o *Object) Fs() fs.Info {
return o.fs
}
// String returns the URL to the remote SFTP file
func (o *Object) String() string {
if o == nil {
return "<nil>"
}
return o.fs.url + "/" + o.remote
}
// Remote the name of the remote SFTP file, relative to the fs root
func (o *Object) Remote() string {
return o.remote
}
// Hash returns "" since SFTP (in Go or OpenSSH) doesn't support remote calculation of hashes
func (o *Object) Hash(r fs.HashType) (string, error) {
debug(o.fs, "Hash '"+o.remote+"'")
return "", fs.ErrHashUnsupported
}
// Size returns the size in bytes of the remote sftp file
func (o *Object) Size() int64 {
debug(o.fs, "Size '"+o.remote+"'")
return o.info.Size()
}
// ModTime returns the modification time of the remote sftp file
func (o *Object) ModTime() time.Time {
debug(o.fs, "ModTime '"+o.remote+"'")
return o.info.ModTime()
}
// SetModTime sets the modification and access time to the specified time
func (o *Object) SetModTime(modTime time.Time) error {
debug(o.fs, "SetModTime '"+o.remote+"'")
err := o.fs.sftpClient.Chtimes(o.fs.sftpClient.Join(o.fs.root, o.remote), modTime, modTime)
if err != nil {
return err
}
o.info, err = o.fs.sftpClient.Stat(o.fs.sftpClient.Join(o.fs.root, o.remote))
return err
}
// Storable returns whether the remote sftp file is a regular file (not a directory, symbolic link, block device, character device, named pipe, etc)
func (o *Object) Storable() bool {
debug(o.fs, "Storable '"+o.remote+"'?")
return o.info.Mode().IsRegular()
}
// Read from a remote sftp file object reader
func (file *ObjectReader) Read(p []byte) (n int, err error) {
debug(file.object.fs, "Read '"+file.object.remote+"'")
n, err = file.sftpFile.Read(p)
return n, err
}
// Close a reader of a remote sftp file
func (file *ObjectReader) Close() (err error) {
debug(file.object.fs, "Close '"+file.object.remote+"'")
err = file.sftpFile.Close()
return err
}
// Open a remote sftp file object for reading. Seek is supported
func (o *Object) Open(options ...fs.OpenOption) (in io.ReadCloser, err error) {
debug(o.fs, "Open '"+o.remote+"'")
var offset int64
offset = 0
for _, option := range options {
switch x := option.(type) {
case *fs.SeekOption:
offset = x.Offset
default:
if option.Mandatory() {
fs.Log(o, "Unsupported mandatory option: %v", option)
}
}
}
sftpFile, err := o.fs.sftpClient.Open(o.fs.sftpClient.Join(o.fs.root, o.remote))
if err != nil {
return nil, err
}
if offset > 0 {
off, err := sftpFile.Seek(offset, 0)
if err != nil || off != offset {
return nil, err
}
}
in = &ObjectReader{
object: o,
sftpFile: sftpFile,
}
return in, nil
}
// Update a remote sftp file using the data <in> and ModTime from <src>
func (o *Object) Update(in io.Reader, src fs.ObjectInfo) error {
debug(o.fs, "Update '"+o.remote+"'")
file, err := o.fs.sftpClient.Create(o.fs.sftpClient.Join(o.fs.root, o.remote))
if err == nil {
_, err = file.ReadFrom(in)
if err != nil {
return err
}
err = o.SetModTime(src.ModTime())
return err
}
return err
}
// Remove a remote sftp file object
func (o *Object) Remove() error {
debug(o.fs, "Remove '"+o.remote+"'")
return o.fs.sftpClient.Remove(o.fs.sftpClient.Join(o.fs.root, o.remote))
}
// Check the interfaces are satisfied
var (
_ fs.Fs = &Fs{}
_ fs.Mover = &Fs{}
_ fs.Object = &Object{}
)