如何在 Go 中高效解壓縮檔案並防止常見錯誤？

使用 Go 輕鬆解壓縮檔案

在 Go 中解壓縮檔案應該是一件輕而易舉的事，但有時你會遇到障礙。讓我們探索一種有效解壓縮檔案的簡單解決方案。

初始代碼：

您提供了以下代碼片段：

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

    for _, f := range r.File {
        rc, err := f.Open()
        if err != nil {
            return err
        }
        defer rc.Close()

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

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

    return nil
}

優化解決方案：

要改進初始程式碼，請考慮以下內容增強功能：

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 address file descriptors issue
    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 ZipSlip (Directory traversal)
        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
}

改進：

• 目標目錄建立：如果目標目錄不存在，則先前已建立正在解壓縮。
• 檔案關閉描述符： 為了避免延遲堆疊問題，閉包包裝了檔案擷取和寫入過程。
• ZipSlip 預防： 檢查確保提取的檔案保留在指定的目標資料夾中。
• 錯誤處理： Close() 操作最好使用錯誤panic() 來處理練習。

以上是如何在 Go 中高效解壓縮檔案並防止常見錯誤？的詳細內容。更多資訊請關注PHP中文網其他相關文章！