如何使用 Go 的 zip 包简化并安全地进行文件解压缩？

Go 中简单有效的文件解压

Go 中解压缩 ZIP 档案可以是一个简单的过程。正如原始代码所示，一种方法涉及手动处理存档中每个文件的解压缩。

但是，有一种更无缝、更有效的方法可以使用以下更新的解决方案来完成此操作：

func Unzip(src, dest string) error {
    r, err := zip.OpenReader(src)
    if err != nil {
        return err
    }
    defer func() {
        if err := r.Close(); err != nil {
            panic(err)
        }
    }()

    os.MkdirAll(dest, 0755)

    // Closure to wrap file extraction and writing, avoiding deferred stack overflow
    extractAndWriteFile := func(f *zip.File) error {
        rc, err := f.Open()
        if err != nil {
            return err
        }
        defer func() {
            if err := rc.Close(); err != nil {
                panic(err)
            }
        }()

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

        // Check for directory traversal (ZipSlip) vulnerability
        if !strings.HasPrefix(path, filepath.Clean(dest)+string(os.PathSeparator)) {
            return fmt.Errorf("illegal file path: %s", path)
        }

        if f.FileInfo().IsDir() {
            os.MkdirAll(path, f.Mode())
        } else {
            os.MkdirAll(filepath.Dir(path), f.Mode())
            f, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
            if err != nil {
                return err
            }
            defer func() {
                if err := f.Close(); err != nil {
                    panic(err)
                }
            }()

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

        return nil
    }

    for _, f := range r.File {
        err := extractAndWriteFile(f)
        if err != nil {
            return err
        }
    }

    return nil
}

这个更新的解决方案提供了几个优点：

• 处理的延迟 Close() 错误：它可以优雅地处理关闭文件读取器 (rc) 和关闭文件读取器时的潜在错误。存档读取器 (r)，即使在遇到故障时也能确保正确的清理。
• 闭包包装的文件提取：每个文件的提取和写入过程都封装在一个闭包中，消除了需要堆叠和清理每个文件的多个 defer .Close() 调用。
• 目录创建：该解决方案确保目标目录在提取文件之前存在，防止任何提取失败。
• ZipSlip 预防： 它包括安全检查，通过验证文件是否未提取到预期目标目录之外来防止潜在的目录遍历 (ZipSlip) 漏洞。

通过这些增强功能，此代码提供了一种简单、健壮且安全的方法来在 Go 中解压缩文件。

以上是如何使用 Go 的 zip 包简化并安全地进行文件解压缩？的详细内容。更多信息请关注PHP中文网其他相关文章！