Go语言实现安全高效的ZIP文件解压教程

package main

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

// Unzip 函数用于将ZIP文件解压到指定目录
func Unzip(src, dest string) error {
    // 1. 打开ZIP文件
    r, err := zip.OpenReader(src)
    if err != nil {
        return fmt.Errorf("无法打开ZIP文件 %s: %w", src, err)
    }
    // 确保在函数退出时关闭ZIP读取器
    defer func() {
        if err := r.Close(); err != nil {
            // 最佳实践：处理关闭时的错误，通常panic或记录日志
            panic(fmt.Errorf("关闭ZIP文件读取器失败: %w", err))
        }
    }()

    // 2. 创建目标解压目录（如果不存在）
    // 0755 权限表示所有者可读写执行，组用户和其他用户可读执行
    if err := os.MkdirAll(dest, 0755); err != nil {
        return fmt.Errorf("无法创建目标目录 %s: %w", dest, err)
    }

    // 3. 遍历ZIP文件中的每个条目
    for _, f := range r.File {
        // 4. 定义一个内部函数来处理单个文件/目录的解压和写入
        // 这样做可以有效管理defer调用，避免在大量文件时堆栈溢出
        err := extractAndWriteFile(f, dest)
        if err != nil {
            return fmt.Errorf("处理文件 %s 失败: %w", f.Name, err)
        }
    }

    return nil
}

// extractAndWriteFile 辅助函数，负责解压并写入单个文件或创建目录
func extractAndWriteFile(f *zip.File, dest string) error {
    // 打开ZIP文件中的当前文件内容
    rc, err := f.Open()
    if err != nil {
        return fmt.Errorf("无法打开ZIP文件中的文件 %s: %w", f.Name, err)
    }
    // 确保在文件处理完毕后关闭文件读取器
    defer func() {
        if err := rc.Close(); err != nil {
            panic(fmt.Errorf("关闭文件读取器失败: %w", err))
        }
    }()

    // 构建目标文件的完整路径
    path := filepath.Join(dest, f.Name)

    // **安全考量：ZipSlip防护**
    // 检查解压路径是否超出目标目录范围，防止目录遍历攻击
    if !strings.HasPrefix(path, filepath.Clean(dest)+string(os.PathSeparator)) {
        return fmt.Errorf("非法文件路径（ZipSlip攻击风险）: %s", path)
    }

    // 如果是目录，则创建目录
    if f.FileInfo().IsDir() {
        if err := os.MkdirAll(path, f.Mode()); err != nil {
            return fmt.Errorf("无法创建目录 %s: %w", path, err)
        }
    } else {
        // 如果是文件，确保其父目录存在
        if err := os.MkdirAll(filepath.Dir(path), f.Mode()); err != nil {
            return fmt.Errorf("无法创建文件 %s 的父目录: %w", path, err)
        }
        // 创建并打开目标文件用于写入
        outFile, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
        if err != nil {
            return fmt.Errorf("无法创建/打开文件 %s: %w", path, err)
        }
        // 确保在文件写入完毕后关闭文件句柄
        defer func() {
            if err := outFile.Close(); err != nil {
                panic(fmt.Errorf("关闭文件句柄失败: %w", err))
            }
        }()

        // 将ZIP文件中的内容复制到目标文件
        if _, err := io.Copy(outFile, rc); err != nil {
            return fmt.Errorf("复制文件内容失败 %s: %w", path, err)
        }
    }
    return nil
}

func main() {
    // 示例用法
    // 假设有一个名为 "example.zip" 的文件在当前目录
    // 将其解压到 "unzipped_files" 目录
    srcZip := "example.zip" // 替换为你的ZIP文件路径
    destDir := "unzipped_files"

    // 创建一个示例ZIP文件用于测试
    if err := createExampleZip(srcZip); err != nil {
        fmt.Printf("创建示例ZIP文件失败: %v\n", err)
        return
    }

    fmt.Printf("开始解压 %s 到 %s...\n", srcZip, destDir)
    err := Unzip(srcZip, destDir)
    if err != nil {
        fmt.Printf("解压失败: %v\n", err)
    } else {
        fmt.Printf("解压成功！文件已解压到 %s\n", destDir)
    }

    // 清理示例文件和目录
    // os.Remove(srcZip)
    // os.RemoveAll(destDir)
}

// createExampleZip 用于创建测试用的ZIP文件
func createExampleZip(zipName string) error {
    outFile, err := os.Create(zipName)
    if err != nil {
        return err
    }
    defer outFile.Close()

    zipWriter := zip.NewWriter(outFile)
    defer zipWriter.Close()

    // 添加一个文件
    file1, err := zipWriter.Create("test_dir/file1.txt")
    if err != nil {
        return err
    }
    _, err = file1.Write([]byte("This is the content of file1."))
    if err != nil {
        return err
    }

    // 添加另一个文件
    file2, err := zipWriter.Create("file2.txt")
    if err != nil {
        return err
    }
    _, err = file2.Write([]byte("This is the content of file2."))
    if err != nil {
        return err
    }

    // 添加一个空目录
    _, err = zipWriter.Create("empty_dir/")
    if err != nil {
        return err
    }

    fmt.Printf("已创建示例ZIP文件: %s\n", zipName)
    return nil
}

path := filepath.Join(dest, f.Name)

// 检查ZipSlip (目录遍历)
if !strings.HasPrefix(path, filepath.Clean(dest)+string(os.PathSeparator)) {
    return fmt.Errorf("非法文件路径（ZipSlip攻击风险）: %s", path)
}