Bagaimanakah saya boleh menyahzip fail dalam Go dengan cekap dan mencegah ralat biasa?

Linda Hamilton
Lepaskan: 2024-11-11 06:41:02
asal
834 orang telah melayarinya

How can I unzip files in Go efficiently and prevent common errors?

Nyahzip Fail Dengan Mudah dengan Go

Menyahzip fail dalam Go sepatutnya mudah, tetapi kadangkala anda menghadapi halangan. Mari kita terokai penyelesaian mudah untuk menyahzip fail dengan berkesan.

Kod Permulaan:

Anda membentangkan coretan kod berikut:

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
}
Salin selepas log masuk

Penyelesaian Dioptimumkan:

Untuk menambah baik kod awal, pertimbangkan peningkatan berikut:

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
}
Salin selepas log masuk

Penambahbaikan:

  • Penciptaan Direktori Destinasi: Jika direktori destinasi tidak wujud, ia dibuat sebelum menyahzip.
  • Penutupan untuk Deskriptor Fail: Untuk mengelakkan isu timbunan tangguh, penutupan akan ditutup proses pengekstrakan dan penulisan fail.
  • Pencegahan ZipSlip: Semakan memastikan fail yang diekstrak kekal dalam folder destinasi yang ditetapkan.
  • Pengendalian Ralat: Tutup () operasi dikendalikan dengan panik ralat() untuk amalan terbaik.

Atas ialah kandungan terperinci Bagaimanakah saya boleh menyahzip fail dalam Go dengan cekap dan mencegah ralat biasa?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

sumber:php.cn
Kenyataan Laman Web ini
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn
Artikel terbaru oleh pengarang
Tutorial Popular
Lagi>
Muat turun terkini
Lagi>
kesan web
Kod sumber laman web
Bahan laman web
Templat hujung hadapan