feat(aa-log): format rule before print.

This commit is contained in:
Alexandre Pujol 2023-10-01 19:00:39 +01:00
parent 352c444ae6
commit b99bb8da46
No known key found for this signature in database
GPG Key ID: C5469996F0DF68EC
7 changed files with 73 additions and 34 deletions

View File

@ -68,8 +68,9 @@ func aaLog(logger string, path string, profile string) error {
if rules { if rules {
profiles := aaLogs.ParseToProfiles() profiles := aaLogs.ParseToProfiles()
for _, profile := range profiles { for _, profile := range profiles {
profile.Sort()
profile.MergeRules() profile.MergeRules()
profile.Sort()
profile.Format()
fmt.Print(profile.String() + "\n") fmt.Print(profile.String() + "\n")
} }
} else { } else {

View File

@ -4,10 +4,6 @@
package aa package aa
import (
"strings"
)
type File struct { type File struct {
Qualifier Qualifier
Path string Path string
@ -26,28 +22,19 @@ func FileFromLog(log map[string]string) ApparmorRule {
func (r *File) Less(other any) bool { func (r *File) Less(other any) bool {
o, _ := other.(*File) o, _ := other.(*File)
letterR := "" letterR := getLetterIn(fileAlphabet, r.Path)
letterO := "" letterO := getLetterIn(fileAlphabet, o.Path)
for _, letter := range fileAlphabet {
if strings.HasPrefix(r.Path, letter) {
letterR = letter
}
if strings.HasPrefix(o.Path, letter) {
letterO = letter
}
}
if fileWeights[letterR] == fileWeights[letterO] || letterR == "" || letterO == "" { if fileWeights[letterR] == fileWeights[letterO] || letterR == "" || letterO == "" {
if r.Path == o.Path { if r.Qualifier.Equals(o.Qualifier) {
if r.Qualifier.Equals(o.Qualifier) { if r.Path == o.Path {
if r.Access == o.Access { if r.Access == o.Access {
return r.Target < o.Target return r.Target < o.Target
} }
return r.Access < o.Access return r.Access < o.Access
} }
return r.Qualifier.Less(o.Qualifier) return r.Path < o.Path
} }
return r.Path < o.Path return r.Qualifier.Less(o.Qualifier)
} }
return fileWeights[letterR] < fileWeights[letterO] return fileWeights[letterR] < fileWeights[letterO]
} }

View File

@ -116,11 +116,8 @@ func (p *AppArmorProfile) AddRule(log map[string]string) {
} }
} }
func typeToValue(i reflect.Type) string {
return strings.ToLower(strings.TrimPrefix(i.String(), "*aa."))
}
// Sort the rules in the profile // Sort the rules in the profile
// Follow: https://apparmor.pujol.io/development/guidelines/#guidelines
func (p *AppArmorProfile) Sort() { func (p *AppArmorProfile) Sort() {
sort.Slice(p.Rules, func(i, j int) bool { sort.Slice(p.Rules, func(i, j int) bool {
typeOfI := reflect.TypeOf(p.Rules[i]) typeOfI := reflect.TypeOf(p.Rules[i])
@ -163,3 +160,33 @@ func (p *AppArmorProfile) MergeRules() {
} }
} }
} }
// Format the profile for better readability before printing it
// Follow: https://apparmor.pujol.io/development/guidelines/#the-file-block
func (p *AppArmorProfile) Format() {
hasOwnedRule := false
for i := len(p.Rules) - 1; i > 0; i-- {
j := i - 1
typeOfI := reflect.TypeOf(p.Rules[i])
typeOfJ := reflect.TypeOf(p.Rules[j])
// File rule
if typeOfI == reflect.TypeOf((*File)(nil)) && typeOfJ == reflect.TypeOf((*File)(nil)) {
letterI := getLetterIn(fileAlphabet, p.Rules[i].(*File).Path)
letterJ := getLetterIn(fileAlphabet, p.Rules[j].(*File).Path)
// Add prefix before rule path to align with other rule
if p.Rules[i].(*File).Owner {
hasOwnedRule = true
} else if hasOwnedRule {
p.Rules[i].(*File).Prefix = " "
}
if letterI != letterJ {
// Add a new empty line between Files rule of different type
hasOwnedRule = false
p.Rules = append(p.Rules[:i], append([]ApparmorRule{&Rule{}}, p.Rules[i:]...)...)
}
}
}
}

View File

@ -13,6 +13,8 @@ type Qualifier struct {
Owner bool Owner bool
NoNewPrivs bool NoNewPrivs bool
FileInherit bool FileInherit bool
Prefix string
Padding string
} }
func NewQualifierFromLog(log map[string]string) Qualifier { func NewQualifierFromLog(log map[string]string) Qualifier {
@ -47,13 +49,13 @@ func NewQualifierFromLog(log map[string]string) Qualifier {
} }
func (r Qualifier) Less(other Qualifier) bool { func (r Qualifier) Less(other Qualifier) bool {
if r.Audit == other.Audit { if r.Owner == other.Owner {
if r.AccessType == other.AccessType { if r.Audit == other.Audit {
return r.Owner return r.AccessType < other.AccessType
} }
return r.AccessType < other.AccessType return r.Audit
} }
return r.Audit return other.Owner
} }
func (r Qualifier) Equals(other Qualifier) bool { func (r Qualifier) Equals(other Qualifier) bool {

View File

@ -142,6 +142,10 @@ func typeOf(i any) string {
return strings.TrimPrefix(reflect.TypeOf(i).String(), "*aa.") return strings.TrimPrefix(reflect.TypeOf(i).String(), "*aa.")
} }
func typeToValue(i reflect.Type) string {
return strings.ToLower(strings.TrimPrefix(i.String(), "*aa."))
}
func indent(s string) string { func indent(s string) string {
return indentation + s return indentation + s
} }
@ -149,3 +153,12 @@ func indent(s string) string {
func indentDbus(s string) string { func indentDbus(s string) string {
return indentation + " " + s return indentation + " " + s
} }
func getLetterIn(alphabet []string, in string) string {
for _, letter := range alphabet {
if strings.HasPrefix(in, letter) {
return letter
}
}
return ""
}

View File

@ -19,10 +19,7 @@
{{ end -}} {{ end -}}
{{- range .Variables -}} {{- range .Variables -}}
{{ "@{" }}{{ .Name }}{{ "} = " }} {{ "@{" }}{{ .Name }}{{ "} = " }}{{ join .Values }}
{{- range .Values -}}
{{ . }}{{ " " }}
{{- end }}
{{ end -}} {{ end -}}
{{- "profile" -}} {{- "profile" -}}
@ -43,6 +40,10 @@
{{- $oldtype := "" -}} {{- $oldtype := "" -}}
{{- range .Rules -}} {{- range .Rules -}}
{{- $type := typeof . -}} {{- $type := typeof . -}}
{{- if eq $type "Rule" -}}
{{- "\n" -}}
{{- continue -}}
{{- end -}}
{{- if and (ne $type $oldtype) (ne $oldtype "") -}} {{- if and (ne $type $oldtype) (ne $oldtype "") -}}
{{- "\n" -}} {{- "\n" -}}
{{- end -}} {{- end -}}
@ -217,7 +218,12 @@
{{- if eq $type "File" -}} {{- if eq $type "File" -}}
{{- template "qualifier" . -}} {{- template "qualifier" . -}}
{{ .Path }}{{ " " }}{{ .Access }} {{- .Path -}}
{{- " " -}}
{{- with .Padding -}}
{{ . }}
{{- end -}}
{{- .Access -}}
{{- with .Target -}} {{- with .Target -}}
{{ " -> " }}{{ . }} {{ " -> " }}{{ . }}
{{- end -}} {{- end -}}

View File

@ -1,4 +1,7 @@
{{- define "qualifier" -}} {{- define "qualifier" -}}
{{- with .Prefix -}}
{{ . }}
{{- end -}}
{{- if .Owner -}} {{- if .Owner -}}
{{- "owner " -}} {{- "owner " -}}
{{- end -}} {{- end -}}