用go写一个加密文件和解密文件的工具

好的，我可以为你提供一个基本的加密文件和解密文件的工具的代码，使用Go语言编写。

首先，我们需要导入一些必要的包，包括`crypto/aes`、`crypto/cipher`、`crypto/rand`、`encoding/hex`、`flag`、`fmt`和`io/ioutil`。代码如下：

package main

import (
    "crypto/aes"
    "crypto/cipher"
    "crypto/rand"
    "encoding/hex"
    "flag"
    "fmt"
    "io/ioutil"
)

然后，我们需要定义一些变量来处理命令行参数。我们使用`flag`包来处理这些参数。我们需要定义两个命令行参数，一个用于加密文件，另一个用于解密文件。代码如下：

var (
    encryptFile string
    decryptFile string
)

func init() {
    flag.StringVar(&encryptFile, "encrypt", "", "file to encrypt")
    flag.StringVar(&decryptFile, "decrypt", "", "file to decrypt")
    flag.Parse()
}

然后，我们需要定义一个函数来生成随机密钥。我们使用`crypto/rand`包来生成随机字节数组。代码如下：

func generateKey() []byte {
    key := make([]byte, 32)
    if _, err := rand.Read(key); err != nil {
        panic(err)
    }
    return key
}

接下来，我们需要定义一个函数来加密文件。我们使用`crypto/aes`包来创建一个新的AES加密块，并使用随机生成的密钥来初始化加密块。然后，我们使用`crypto/cipher`包来创建一个新的加密器，将加密器应用于文件内容，并将加密后的文件写入新文件中。代码如下：

func encryptFileWithKey(key []byte, inputFile string, outputFile string) error {
    block, err := aes.NewCipher(key)
    if err != nil {
        return err
    }

    fileData, err := ioutil.ReadFile(inputFile)
    if err != nil {
        return err
    }

    ciphertext := make([]byte, aes.BlockSize+len(fileData))
    iv := ciphertext[:aes.BlockSize]
    if _, err := rand.Read(iv); err != nil {
        return err
    }

    stream := cipher.NewCTR(block, iv)
    stream.XORKeyStream(ciphertext[aes.BlockSize:], fileData)

    return ioutil.WriteFile(outputFile, ciphertext, 0644)
}

最后，我们需要定义一个函数来解密文件。我们使用`crypto/aes`包创建一个新的AES加密块，并使用相同的密钥和IV来初始化加密块。然后，我们使用`crypto/cipher`包创建一个新的解密器，并将解密器应用于文件内容。最后，我们将解密后的文件写入新文件。代码如下：

func decryptFileWithKey(key []byte, inputFile string, outputFile string) error {
    fileData, err := ioutil.ReadFile(inputFile)
    if err != nil {
        return err
    }

    iv := fileData[:aes.BlockSize]
    ciphertext := fileData[aes.BlockSize:]

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

    stream := cipher.NewCTR(block, iv)
    stream.XORKeyStream(ciphertext, ciphertext)

    return ioutil.WriteFile(outputFile, ciphertext, 0644)
}

最后，我们需要在`main`函数中调用这些函数来加密或解密文件。代码如下：

func main() {
    if encryptFile != "" {
        key := generateKey()
        fmt.Println("Encryption key:", hex.EncodeToString(key))
        if err := encryptFileWithKey(key, encryptFile, encryptFile+".enc"); err != nil {
            panic(err)
        }
        fmt.Println("File encrypted successfully!")
    }

    if decryptFile != "" {
        var key []byte
        fmt.Print("Enter encryption key: ")
        fmt.Scanln(&key)
        if err := decryptFileWithKey(key, decryptFile, decryptFile+".dec"); err != nil {
            panic(err)
        }
        fmt.Println("File decrypted successfully!")
    }
}

现在，我们已经完成了一个简单的加密文件和解密文件的工具。完整代码如下：