mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-24 14:56:41 +00:00
a181dd0ebc
* refactor(gallery): move under core/ Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * fix(unarchive): do not allow symlinks Signed-off-by: Ettore Di Giacinto <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
package utils
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/mholt/archiver/v3"
|
|
)
|
|
|
|
func IsArchive(file string) bool {
|
|
uaIface, err := archiver.ByExtension(file)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
_, ok := uaIface.(archiver.Unarchiver)
|
|
return ok
|
|
}
|
|
|
|
func ExtractArchive(archive, dst string) error {
|
|
uaIface, err := archiver.ByExtension(archive)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
un, ok := uaIface.(archiver.Unarchiver)
|
|
if !ok {
|
|
return fmt.Errorf("format specified by source filename is not an archive format: %s (%T)", archive, uaIface)
|
|
}
|
|
|
|
mytar := &archiver.Tar{
|
|
OverwriteExisting: true,
|
|
MkdirAll: true,
|
|
ImplicitTopLevelFolder: false,
|
|
ContinueOnError: true,
|
|
}
|
|
|
|
switch v := uaIface.(type) {
|
|
case *archiver.Tar:
|
|
uaIface = mytar
|
|
case *archiver.TarBrotli:
|
|
v.Tar = mytar
|
|
case *archiver.TarBz2:
|
|
v.Tar = mytar
|
|
case *archiver.TarGz:
|
|
v.Tar = mytar
|
|
case *archiver.TarLz4:
|
|
v.Tar = mytar
|
|
case *archiver.TarSz:
|
|
v.Tar = mytar
|
|
case *archiver.TarXz:
|
|
v.Tar = mytar
|
|
case *archiver.TarZstd:
|
|
v.Tar = mytar
|
|
}
|
|
|
|
err = archiver.Walk(archive, func(f archiver.File) error {
|
|
if f.FileInfo.Mode()&os.ModeSymlink != 0 {
|
|
return fmt.Errorf("archive contains a symlink")
|
|
}
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return un.Unarchive(archive, dst)
|
|
}
|