135 Star 507 Fork 122

GVP悬镜安全/OpenSCA-cli

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
tar.go 2.04 KB
一键复制 编辑 原始数据 按行查看 历史
骆天奇 提交于 2023-12-25 21:19 +08:00 . fix: tar decomporess file not found
package walk
import (
"archive/tar"
"compress/bzip2"
"compress/gzip"
"context"
"io"
"os"
"path/filepath"
"strings"
"github.com/xmirrorsecurity/opensca-cli/v3/opensca/logs"
)
func xtar(ctx context.Context, filter ExtractFileFilter, input, output string) bool {
if !checkFileExt(input, ".tar") {
return false
}
f, err := os.Open(input)
if err != nil {
logs.Warn(err)
return false
}
defer f.Close()
fr := tar.NewReader(f)
for {
select {
case <-ctx.Done():
return false
default:
}
fh, err := fr.Next()
if err == io.EOF {
break
}
if err != nil {
logs.Warn(err)
break
}
fp := filepath.Join(output, fh.Name)
// avoid zip slip
if !strings.HasPrefix(fp, filepath.Clean(output)+string(os.PathSeparator)) {
logs.Warn("Invalid file path: %s", fp)
continue
}
if fh.Typeflag == tar.TypeDir {
os.MkdirAll(fp, 0755)
continue
}
if filter != nil && !filter(fp) {
continue
}
os.MkdirAll(filepath.Dir(fp), 0777)
fw, err := os.Create(fp)
if err != nil {
logs.Warn(err)
continue
}
io.Copy(fw, fr)
fw.Close()
}
return true
}
func xgz(input, output string) bool {
if !checkFileHead(input, M_GZ) {
return false
}
f, err := os.Open(input)
if err != nil {
logs.Warn(err)
return false
}
defer f.Close()
fr, err := gzip.NewReader(f)
if err != nil {
logs.Warn(err)
return false
}
defer fr.Close()
fp := filepath.Join(output, strings.TrimSuffix(filepath.Base(input), filepath.Ext(input)))
fw, err := os.Create(fp)
if err != nil {
logs.Warn(err)
return false
}
_, err = io.Copy(fw, fr)
fw.Close()
return err == nil
}
func xbz2(input, output string) bool {
if !checkFileHead(input, M_BZ2) {
return false
}
f, err := os.Open(input)
if err != nil {
logs.Warn(err)
return false
}
defer f.Close()
fr := bzip2.NewReader(f)
fp := filepath.Join(output, strings.TrimSuffix(filepath.Base(input), filepath.Ext(input)))
fw, err := os.Create(fp)
if err != nil {
logs.Warn(err)
return false
}
_, err = io.Copy(fw, fr)
fw.Close()
return err == nil
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/XmirrorSecurity/OpenSCA-cli.git
git@gitee.com:XmirrorSecurity/OpenSCA-cli.git
XmirrorSecurity
OpenSCA-cli
OpenSCA-cli
master

搜索帮助