mirror of
https://github.com/roddhjav/apparmor.d.git
synced 2024-12-27 23:48:00 +01:00
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
// apparmor.d - Full set of apparmor profiles
|
|
// Copyright (C) 2023-2024 Alexandre Pujol <alexandre@pujol.io>
|
|
// SPDX-License-Identifier: GPL-2.0-only
|
|
|
|
package integration
|
|
|
|
import (
|
|
"archive/tar"
|
|
"compress/gzip"
|
|
"fmt"
|
|
"io"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/roddhjav/apparmor.d/pkg/paths"
|
|
)
|
|
|
|
// Either or not to extract the file
|
|
func toExtrat(name string, subfolders []string) bool {
|
|
for _, subfolder := range subfolders {
|
|
if strings.HasPrefix(name, subfolder) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// Extract part of an archive to a destination directory
|
|
func extratTo(src *paths.Path, dst *paths.Path, subfolders []string) error {
|
|
gzIn, err := src.Open()
|
|
if err != nil {
|
|
return fmt.Errorf("opening %s: %w", src, err)
|
|
}
|
|
defer gzIn.Close()
|
|
|
|
in, err := gzip.NewReader(gzIn)
|
|
if err != nil {
|
|
return fmt.Errorf("decoding %s: %w", src, err)
|
|
}
|
|
defer in.Close()
|
|
|
|
if err := dst.MkdirAll(); err != nil {
|
|
return fmt.Errorf("creating %s: %w", src, err)
|
|
}
|
|
|
|
tarIn := tar.NewReader(in)
|
|
for {
|
|
header, err := tarIn.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if header.Typeflag == tar.TypeReg {
|
|
if !toExtrat(header.Name, subfolders) {
|
|
continue
|
|
}
|
|
path := dst.Join(filepath.Base(header.Name))
|
|
file, err := path.Create()
|
|
if err != nil {
|
|
return fmt.Errorf("creating %s: %w", file.Name(), err)
|
|
}
|
|
if _, err := io.Copy(file, tarIn); err != nil {
|
|
return fmt.Errorf("extracting %s: %w", file.Name(), err)
|
|
}
|
|
file.Close()
|
|
}
|
|
}
|
|
return nil
|
|
}
|