使用golang goftp 写一个ftp 服务器,只支持上传功能,其他删除。

1,关于


ftp 是个很老的技术了。
项目需要搭建一个ftp服务器。但是还是想做点程序实现些特定的功能。
需要监听到服务上传文件啥的。

2,项目


从网上找了一个lib库:
https://github.com/goftp/server
非常方便。要是懒得自己写。还有个gui界面的。
https://github.com/goftp/ftpd
直接下载编译就行了:
还有界面。

3,代码


但是不需要那么多的功能。
只需要做一个简单的服务就行。
main.go

package main

import (
    "github.com/goftp/server"
    "os"
    "fmt"
    "github.com/lunny/log"
    "github.com/Unknwon/goconfig"
)

func main() {
    //读取配置文件:
    cfg, _ := goconfig.LoadConfigFile("ftp.conf")
    //端口
    port, _ := cfg.Int("server", "port")
    //使用简单的登录验证,用户名密码写死。
    authUser, _ := cfg.GetValue("auth", "user")
    authPassword, _ := cfg.GetValue("auth", "password")
    //权限:
    permOwner, _ := cfg.GetValue("perm", "owner")
    permGroup, _ := cfg.GetValue("perm", "group")
    //上传目录dir
    rootPath, _ := cfg.GetValue("server", "dir")
    _, err := os.Lstat(rootPath)
    if os.IsNotExist(err) {
        os.MkdirAll(rootPath, os.ModePerm)
    } else if err != nil {
        fmt.Println(err)
        return
    }
    //设置权限。
    factory := &dirver.PutFileOnlyDriverFactory{
        rootPath,
        server.NewSimplePerm(permOwner, permGroup),
    }

    opt := &server.ServerOpts{
        Name:    "go",
        Factory: factory,
        Port:    port,
        Auth:    &server.SimpleAuth{authUser, authPassword},
    }
    // start ftp server
    ftpServer := server.NewServer(opt)
    log.Info("FTP Server", "1.0")
    err = ftpServer.ListenAndServe()
    if err != nil {
        log.Fatal("Error starting server:", err)
    }
}

其中驱动,自己根据官方的重写了下:

package dirver

import (
    "errors"
    "fmt"
    "io"
    "os"
    "path/filepath"
    "strings"
    "github.com/goftp/server"
    "github.com/lunny/log"
)

type PutFileOnlyDriver struct {
    RootPath string
    server.Perm
}

type FileInfo struct {
    os.FileInfo

    mode  os.FileMode
    owner string
    group string
}

func (f *FileInfo) Mode() os.FileMode {
    return f.mode
}

func (f *FileInfo) Owner() string {
    return f.owner
}

func (f *FileInfo) Group() string {
    return f.group
}

func (driver *PutFileOnlyDriver) realPath(path string) string {
    paths := strings.Split(path, "/")
    return filepath.Join(append([]string{driver.RootPath}, paths...)...)
}

func (driver *PutFileOnlyDriver) Init(conn *server.Conn) {
    //driver.conn = conn
}

func (driver *PutFileOnlyDriver) ChangeDir(path string) error {
    rPath := driver.realPath(path)
    f, err := os.Lstat(rPath)
    if err != nil {
        return err
    }
    if f.IsDir() {
        return nil
    }
    return errors.New("Not a directory")
}

func (driver *PutFileOnlyDriver) Stat(path string) (server.FileInfo, error) {
    basepath := driver.realPath(path)
    rPath, err := filepath.Abs(basepath)
    if err != nil {
        return nil, err
    }
    f, err := os.Lstat(rPath)
    if err != nil {
        return nil, err
    }
    mode, err := driver.Perm.GetMode(path)
    if err != nil {
        return nil, err
    }
    if f.IsDir() {
        mode |= os.ModeDir
    }
    owner, err := driver.Perm.GetOwner(path)
    if err != nil {
        return nil, err
    }
    group, err := driver.Perm.GetGroup(path)
    if err != nil {
        return nil, err
    }
    return &FileInfo{f, mode, owner, group}, nil
}

func (driver *PutFileOnlyDriver) ListDir(path string, callback func(server.FileInfo) error) error {
    return nil
}

func (driver *PutFileOnlyDriver) DeleteDir(path string) error {
    return errors.New("Not a directory")
}

func (driver *PutFileOnlyDriver) DeleteFile(path string) error {
    return errors.New("Not a file")
}

func (driver *PutFileOnlyDriver) Rename(fromPath string, toPath string) error {
    return errors.New("Not a file")
}

func (driver *PutFileOnlyDriver) MakeDir(path string) error {
    return errors.New("Not a directory")
}

func (driver *PutFileOnlyDriver) GetFile(path string, offset int64) (int64, io.ReadCloser, error) {
    return 0, nil, errors.New("Not a file")
}

func (driver *PutFileOnlyDriver) PutFile(destPath string, data io.Reader, appendData bool) (int64, error) {
    log.Info("############### PutFile destPath : ", destPath)
    rPath := driver.realPath(destPath)
    var isExist bool
    f, err := os.Lstat(rPath)
    if err == nil {
        isExist = true
        if f.IsDir() {
            return 0, errors.New("A dir has the same name")
        }
    } else {
        if os.IsNotExist(err) {
            isExist = false
        } else {
            return 0, errors.New(fmt.Sprintln("Put File error:", err))
        }
    }

    if appendData && !isExist {
        appendData = false
    }

    if !appendData {
        if isExist {
            err = os.Remove(rPath)
            if err != nil {
                return 0, err
            }
        }
        f, err := os.Create(rPath)
        if err != nil {
            return 0, err
        }
        defer f.Close()
        bytes, err := io.Copy(f, data)
        if err != nil {
            return 0, err
        }
        return bytes, nil
    }

    of, err := os.OpenFile(rPath, os.O_APPEND|os.O_RDWR, 0660)
    if err != nil {
        return 0, err
    }
    defer of.Close()

    _, err = of.Seek(0, os.SEEK_END)
    if err != nil {
        return 0, err
    }

    bytes, err := io.Copy(of, data)
    if err != nil {
        return 0, err
    }

    return bytes, nil
}

type PutFileOnlyDriverFactory struct {
    RootPath string
    server.Perm
}

func (factory *PutFileOnlyDriverFactory) NewDriver() (server.Driver, error) {
    return &PutFileOnlyDriver{factory.RootPath, factory.Perm}, nil
}

把几个接口屏蔽了。其中配置文件:

[server]
port=21
dir=file

[auth]
user=ftpuser
password=ftpuser

[perm]
owner=root
group=root

4,总结


其中上传方法:

func (driver *PutFileOnlyDriver) PutFile(destPath string, data io.Reader, appendData bool) (int64, error) {

就可以监控ftp上传的文件了。非常方便。
虽然是个老的技术。但是通过golang,可以非常的方便的搭建一个ftp服务器。

本文的原文连接是:
https://blog.csdn.net/freewebsys/article/details/82048257

博主地址是:http://blog.csdn.net/freewebsys

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值