使用Golang为ZIP文件添加密码保护

ZIP文件是一种常用的文件压缩格式，可以将多个文件和文件夹打包成一个单独的文件，以减小文件大小和便于传输。有时我们需要为ZIP文件添加密码保护，以确保其安全性。本文将介绍如何使用Golang为ZIP文件添加密码保护。

1. 导入Golang Archive/zip包

要进行ZIP文件处理，我们需要导入Golang Archive/zip包。在代码中添加以下导入语句：

import "archive/zip"

2. 创建ZIP文件并设置密码

首先，我们需要创建一个ZIP文件并设置密码。可以通过调用`Create`方法来创建ZIP文件对象，并使用`SetPassword`方法来设置密码。以下是一个示例代码：

func CreateAndSetPassword(filePath, password string) error {
  file, err := os.Create(filePath)
  if err != nil {
    return err
  }
  defer file.Close()

  writer := zip.NewWriter(file)
  writer.SetPassword(password)

  // 添加文件到ZIP，并设置密码
  ...
}

3. 添加文件到ZIP并设置密码

一旦我们创建了ZIP文件对象并设置了密码，我们可以通过调用`Create`方法来向ZIP文件中添加文件，并使用`SetPassword`方法为每个文件设置密码。以下是一个示例代码：

func AddFileWithPassword(writer *zip.Writer, file, password string) error {
  // 创建ZIP文件的文件头
  header, err := zip.FileInfoHeader(fileInfo)
  if err != nil {
    return err
  }
  header.Method = zip.Deflate

  header.SetPassword(password)

  // 创建ZIP文件的Writer
  writer, err := writer.CreateHeader(header)
  if err != nil {
    return err
  }

  // 将文件内容写入到ZIP文件
  ...
}

4. 保存并关闭ZIP文件

完成向ZIP文件中添加文件后，我们需要调用`Close`方法来保存ZIP文件并关闭ZIP文件对象。以下是一个示例代码：

func SaveAndCloseZIP(writer *zip.Writer) error {
  err := writer.Close()
  if err != nil {
    return err
  }

  return nil
}

5. 完整示例代码

下面是一个完整的示例代码，展示了如何使用Golang为ZIP文件添加密码保护：

import (
  "archive/zip"
  "io"
  "os"
  "path/filepath"
)

func CreateAndSetPassword(filePath, password string) error {
  file, err := os.Create(filePath)
  if err != nil {
    return err
  }
  defer file.Close()

  writer := zip.NewWriter(file)
  writer.SetPassword(password)

  // 添加文件到ZIP，并设置密码
  err = AddFileWithPassword(writer, "file.txt", password)
  if err != nil {
    return err
  }

  // 保存并关闭ZIP文件
  err = SaveAndCloseZIP(writer)
  if err != nil {
    return err
  }

  return nil
}

func AddFileWithPassword(writer *zip.Writer, file, password string) error {
  f, err := os.Open(file)
  if err != nil {
    return err
  }
  defer f.Close()

  stat, err := f.Stat()
  if err != nil {
    return err
  }

  header, err := zip.FileInfoHeader(stat)
  if err != nil {
    return err
  }
  header.Method = zip.Deflate

  header.SetPassword(password)

  w, err := writer.CreateHeader(header)
  if err != nil {
    return err
  }

  _, err = io.Copy(w, f)
  if err != nil {
    return err
  }

  return nil
}

func SaveAndCloseZIP(writer *zip.Writer) error {
  err := writer.Close()
  if err != nil {
    return err
  }

  return nil
}

func main() {
  err := CreateAndSetPassword("files.zip", "password123")
  if err != nil {
    fmt.Println("Error:", err)
    return
  }

  fmt.Println("ZIP file created successfully.")
}

结论

通过使用Golang Archive/zip包，我们可以轻松地为ZIP文件添加密码保护。上述示例代码演示了创建ZIP文件并设置密码的基本步骤。根据需要，您可以添加更多的文件和设置更复杂的密码策略来满足您的需求。