如何用Golang实现文件加密传输 结合crypto包实现AES加密

package main

import (
    "crypto/aes"
    "crypto/cipher"
    "crypto/rand"
    "fmt"
    "io"
    "io/ioutil"
    "log"
    "os"
)

// generateRandomKey 生成随机密钥
func generateRandomKey() ([]byte, error) {
    key := make([]byte, 32) // AES-256 密钥长度为 32 字节
    _, err := rand.Read(key)
    if err != nil {
        return nil, err
    }
    return key, nil
}

// encryptFile 使用 AES 加密文件
func encryptFile(key []byte, inputFile string, outputFile string) error {
    plaintext, err := ioutil.ReadFile(inputFile)
    if err != nil {
        return err
    }

    block, err := aes.NewCipher(key)
    if err != nil {
        return err
    }

    // 创建 GCM 加密模式
    aesGCM, err := cipher.NewGCM(block)
    if err != nil {
        return err
    }

    // 创建 nonce
    nonce := make([]byte, aesGCM.NonceSize())
    if _, err = io.ReadFull(rand.Reader, nonce); err != nil {
        return err
    }

    // 加密数据
    ciphertext := aesGCM.Seal(nonce, nonce, plaintext, nil)

    // 写入加密后的文件
    err = ioutil.WriteFile(outputFile, ciphertext, 0600)
    if err != nil {
        return err
    }

    return nil
}

// decryptFile 使用 AES 解密文件
func decryptFile(key []byte, inputFile string, outputFile string) error {
    ciphertext, err := ioutil.ReadFile(inputFile)
    if err != nil {
        return err
    }

    block, err := aes.NewCipher(key)
    if err != nil {
        return err
    }

    aesGCM, err := cipher.NewGCM(block)
    if err != nil {
        return err
    }

    nonceSize := aesGCM.NonceSize()
    if len(ciphertext) < nonceSize {
        return fmt.Errorf("ciphertext too short")
    }

    nonce, ciphertext := ciphertext[:nonceSize], ciphertext[nonceSize:]

    plaintext, err := aesGCM.Open(nil, []byte(nonce), []byte(ciphertext), nil)
    if err != nil {
        return err
    }

    err = ioutil.WriteFile(outputFile, plaintext, 0600)
    if err != nil {
        return err
    }

    return nil
}

func main() {
    // 生成随机密钥
    key, err := generateRandomKey()
    if err != nil {
        log.Fatal(err)
    }

    // 示例文件名
    inputFile := "original.txt"
    encryptedFile := "encrypted.txt"
    decryptedFile := "decrypted.txt"

    // 创建示例文件
    err = ioutil.WriteFile(inputFile, []byte("This is a secret message!"), 0644)
    if err != nil {
        log.Fatal(err)
    }

    // 加密文件
    err = encryptFile(key, inputFile, encryptedFile)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("File encrypted successfully!")

    // 解密文件
    err = decryptFile(key, encryptedFile, decryptedFile)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("File decrypted successfully!")

    // 清理示例文件
    os.Remove(inputFile)
    os.Remove(encryptedFile)
    os.Remove(decryptedFile)
}