mirror of
https://github.com/roddhjav/apparmor.d.git
synced 2024-11-15 16:03:51 +01:00
108 lines
2.5 KiB
Go
108 lines
2.5 KiB
Go
// apparmor.d - Full set of apparmor profiles
|
|
// Copyright (C) 2021-2024 Alexandre Pujol <alexandre@pujol.io>
|
|
// SPDX-License-Identifier: GPL-2.0-only
|
|
|
|
package directive
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/roddhjav/apparmor.d/pkg/paths"
|
|
"github.com/roddhjav/apparmor.d/pkg/prebuild/cfg"
|
|
)
|
|
|
|
var (
|
|
// Define the directive keyword globally
|
|
Keyword = "#aa:"
|
|
|
|
// Build the profiles with the following directive applied
|
|
Directives = map[string]Directive{}
|
|
|
|
regDirective = regexp.MustCompile(`(?m).*` + Keyword + `([a-z]*)( .*)?`)
|
|
)
|
|
|
|
// Main directive interface
|
|
type Directive interface {
|
|
cfg.BaseInterface
|
|
Apply(opt *Option, profile string) (string, error)
|
|
}
|
|
|
|
// Directive options
|
|
type Option struct {
|
|
Name string
|
|
ArgMap map[string]string
|
|
ArgList []string
|
|
File *paths.Path
|
|
Raw string
|
|
}
|
|
|
|
func NewOption(file *paths.Path, match []string) *Option {
|
|
if len(match) != 3 {
|
|
panic(fmt.Sprintf("Invalid directive: %v", match))
|
|
}
|
|
argList := strings.Fields(match[2])
|
|
argMap := map[string]string{}
|
|
for _, t := range argList {
|
|
tmp := strings.Split(t, "=")
|
|
if len(tmp) < 2 {
|
|
argMap[tmp[0]] = ""
|
|
} else {
|
|
argMap[tmp[0]] = tmp[1]
|
|
}
|
|
}
|
|
return &Option{
|
|
Name: match[1],
|
|
ArgMap: argMap,
|
|
ArgList: argList,
|
|
File: file,
|
|
Raw: match[0],
|
|
}
|
|
}
|
|
|
|
// Clean removes selected directive line from input string.
|
|
// Useful to remove directive text applied on some condition only
|
|
func (o *Option) Clean(input string) string {
|
|
return strings.Replace(input, o.Raw, o.cleanKeyword(o.Raw), 1)
|
|
}
|
|
|
|
// cleanKeyword removes the dirextive keywork (#aa:...) from the input string
|
|
func (o *Option) cleanKeyword(input string) string {
|
|
reg := regexp.MustCompile(`\s*` + Keyword + o.Name + `( .*)?$`)
|
|
return reg.ReplaceAllString(input, "")
|
|
}
|
|
|
|
// Check if the directive is inline or if it is a paragraph
|
|
func (o *Option) IsInline() bool {
|
|
inline := true
|
|
tmp := strings.Split(o.Raw, Keyword)
|
|
if len(tmp) >= 1 {
|
|
left := strings.TrimSpace(tmp[0])
|
|
if len(left) == 0 {
|
|
inline = false
|
|
}
|
|
}
|
|
return inline
|
|
}
|
|
|
|
func RegisterDirective(d Directive) {
|
|
Directives[d.Name()] = d
|
|
}
|
|
|
|
func Run(file *paths.Path, profile string) (string, error) {
|
|
var err error
|
|
for _, match := range regDirective.FindAllStringSubmatch(profile, -1) {
|
|
opt := NewOption(file, match)
|
|
drtv, ok := Directives[opt.Name]
|
|
if !ok {
|
|
return "", fmt.Errorf("Unknown directive '%s' in %s", opt.Name, opt.File)
|
|
}
|
|
profile, err = drtv.Apply(opt, profile)
|
|
if err != nil {
|
|
return "", fmt.Errorf("%s %s: %w", drtv.Name(), opt.File, err)
|
|
}
|
|
}
|
|
return profile, nil
|
|
}
|