Go语言中FTP文件传输的安全性考量与安全替代方案

go get golang.org/x/crypto/ssh
go get github.com/pkg/sftp

package main

import (
    "fmt"
    "io"
    "log"
    "os"
    "path/filepath"

    "github.com/pkg/sftp"
    "golang.org/x/crypto/ssh"
)

func main() {
    // SFTP服务器配置
    host := "your_sftp_server.com" // 替换为你的SFTP服务器地址
    port := "22"                   // SFTP默认端口
    user := "your_username"        // 替换为你的用户名
    password := "your_password"    // 替换为你的密码

    // SSH客户端配置
    config := &ssh.ClientConfig{
        User: user,
        Auth: []ssh.AuthMethod{
            ssh.Password(password),
            // 如果使用密钥认证，可以配置：
            // ssh.PublicKeys(parsePrivateKey("path/to/your/private_key")),
        },
        // 注意：ssh.InsecureIgnoreHostKey() 在生产环境中非常不安全！
        // 它会跳过主机密钥验证，使你的连接容易受到中间人攻击。
        // 在生产环境中，应使用 ssh.FixedHostKey(hostKey) 或 ssh.KnownHosts(knownHostsPath)
        // 来验证服务器的主机密钥。
        HostKeyCallback: ssh.InsecureIgnoreHostKey(),
        // Timeout: 10 * time.Second, // 可选：设置连接超时
    }

    // 连接到SSH服务器
    addr := net.JoinHostPort(host, port)
    conn, err := ssh.Dial("tcp", addr, config)
    if err != nil {
        log.Fatalf("无法连接SSH服务器 %s: %v", addr, err)
    }
    defer conn.Close()
    fmt.Printf("成功连接到SSH服务器: %s\n", addr)

    // 创建SFTP客户端
    client, err := sftp.NewClient(conn)
    if err != nil {
        log.Fatalf("无法创建SFTP客户端: %v", err)
    }
    defer client.Close()
    fmt.Println("SFTP客户端已创建。")

    // --- 上传文件示例 ---
    localFilePath := "local_upload_test.txt"
    remoteUploadPath := "/remote/path/to/uploaded_file.txt" // 远程服务器上的目标路径

    // 创建一个本地测试文件
    if err := os.WriteFile(localFilePath, []byte("Hello from Go SFTP upload!"), 0644); err != nil {
        log.Fatalf("无法创建本地文件 '%s': %v", localFilePath, err)
    }
    defer os.Remove(localFilePath) // 结束后删除本地测试文件

    localFile, err := os.Open(localFilePath)
    if err != nil {
        log.Fatalf("无法打开本地文件 '%s': %v", localFilePath, err)
    }
    defer localFile.Close()

    remoteFile, err := client.Create(remoteUploadPath)
    if err != nil {
        log.Fatalf("无法创建远程文件 '%s': %v", remoteUploadPath, err)
    }
    defer remoteFile.Close()

    if _, err := io.Copy(remoteFile, localFile); err != nil {
        log.Fatalf("上传文件 '%s' 到 '%s' 失败: %v", localFilePath, remoteUploadPath, err)
    }
    fmt.Printf("文件 '%s' 已成功上传到 '%s'\n", localFilePath, remoteUploadPath)

    // --- 下载文件示例 ---
    remoteDownloadPath := remoteUploadPath // 假设下载刚刚上传的文件
    localDownloadPath := "local_download_test.txt"

    srcFile, err := client.Open(remoteDownloadPath)
    if err != nil {
        log.Fatalf("无法打开远程文件 '%s' 进行下载: %v", remoteDownloadPath, err)
    }
    defer srcFile.Close()

    dstFile, err := os.Create(localDownloadPath)
    if err != nil {
        log.Fatalf("无法创建本地文件 '%s' 进行下载: %v", localDownloadPath, err)
    }
    defer dstFile.Close()
    defer os.Remove(localDownloadPath) // 结束后删除本地下载文件

    if _, err := io.Copy(dstFile, srcFile); err != nil {
        log.Fatalf("下载文件 '%s' 到 '%s' 失败: %v", remoteDownloadPath, localDownloadPath, err)
    }
    fmt.Printf("文件 '%s' 已成功下载到 '%s'\n", remoteDownloadPath, localDownloadPath)

    // 验证下载内容 (可选)
    downloadedContent, err := os.ReadFile(localDownloadPath)
    if err != nil {
        log.Fatalf("无法读取下载文件内容: %v", err)
    }
    fmt.Printf("下载文件内容: %s\n", string(downloadedContent))
}

// parsePrivateKey 是一个辅助函数，用于解析SSH私钥文件
// func parsePrivateKey(path string) ssh.Signer {
//  key, err := os.ReadFile(path)
//  if err != nil {
//      log.Fatalf("无法读取私钥文件 '%s': %v", path, err)
//  }
//  signer, err := ssh.ParsePrivateKey(key)
//  if err != nil {
//      log.Fatalf("无法解析私钥: %v", err)
//  }
//  return signer
// }