apparmor.d/pkg/prebuild/directive/dbus.go

// apparmor.d - Full set of apparmor profiles
// Copyright (C) 2021-2024 Alexandre Pujol <alexandre@pujol.io>
// SPDX-License-Identifier: GPL-2.0-only

// Dbus directive
//
// Example of supported directive:
// #aa:dbus own bus=session name=org.freedesktop.FileManager1
// #aa:dbus talk name=org.freedesktop.login1 label=systemd-logind
//
// See https://apparmor.pujol.io/development/dbus/#dbus-directive
//

package directive

import (
	"fmt"
	"strings"

	"github.com/roddhjav/apparmor.d/pkg/aa"
	"github.com/roddhjav/apparmor.d/pkg/prebuild"
)

type Dbus struct {
	prebuild.Base
}

func init() {
	RegisterDirective(&Dbus{
		Base: prebuild.Base{
			Keyword: "dbus",
			Msg:     "Dbus directive applied",
			Help: []string{
				"own bus=<bus> name=<name> [interface=AARE] [path=AARE]",
				"talk bus=<bus> name=<name> label=<profile> [interface=AARE] [path=AARE]",
			},
		}},
	)
}

func (d Dbus) Apply(opt *Option, profile string) (string, error) {
	var r aa.Rules

	action, err := d.sanityCheck(opt)
	if err != nil {
		return "", err
	}
	name := opt.File.Base()
	if len(name) > 15 {
		name = name[:15]
	}
	switch action {
	case "own":
		r = d.own(opt.ArgMap, name)
	case "talk":
		r = d.talk(opt.ArgMap, name)
	}

	aa.IndentationLevel = strings.Count(
		strings.SplitN(opt.Raw, Keyword, 1)[0], aa.Indentation,
	)
	generatedDbus := r.String()
	lenDbus := len(generatedDbus)
	generatedDbus = generatedDbus[:lenDbus-1]
	profile = strings.Replace(profile, opt.Raw, generatedDbus, -1)
	return profile, nil
}

func (d Dbus) sanityCheck(opt *Option) (string, error) {
	if len(opt.ArgList) < 1 {
		return "", fmt.Errorf("Unknown dbus action: %s in %s", opt.Name, opt.File)
	}
	action := opt.ArgList[0]
	if action != "own" && action != "talk" {
		return "", fmt.Errorf("Unknown dbus action: %s in %s", opt.Name, opt.File)
	}

	if _, present := opt.ArgMap["name"]; !present {
		return "", fmt.Errorf("Missing name for 'dbus: %s' in %s", action, opt.File)
	}
	if _, present := opt.ArgMap["bus"]; !present {
		return "", fmt.Errorf("Missing bus for '%s' in %s", opt.ArgMap["name"], opt.File)
	}
	if _, present := opt.ArgMap["label"]; !present && action == "talk" {
		return "", fmt.Errorf("Missing label for '%s' in %s", opt.ArgMap["name"], opt.File)
	}

	// Set default values
	if _, present := opt.ArgMap["path"]; !present {
		opt.ArgMap["path"] = "/" + strings.Replace(opt.ArgMap["name"], ".", "/", -1) + "{,/**}"
	}
	opt.ArgMap["name"] += "{,.*}"
	return action, nil
}

func getInterfaces(rules map[string]string) []string {
	var interfaces []string
	if _, present := rules["interface"]; present {
		interfaces = []string{rules["interface"]}
	} else {
		interfaces = []string{rules["name"]}
	}

	if _, present := rules["interface+"]; present {
		interfaces = append(interfaces, rules["interface+"])
	}
	return interfaces
}

func (d Dbus) own(rules map[string]string, name string) aa.Rules {
	interfaces := getInterfaces(rules)

	res := aa.Rules{
		&aa.Unix{
			Access: []string{"bind"}, Type: "stream",
			Address: `@@{udbus}/bus/` + name + `/` + rules["bus"],
		},
		&aa.Dbus{
			Access: []string{"bind"}, Bus: rules["bus"], Name: rules["name"],
		},
	}

	// Interfaces
	for _, iface := range interfaces {
		res = append(res,
			&aa.Dbus{
				Access: []string{"receive"}, Bus: rules["bus"], Path: rules["path"],
				Interface: iface,
				PeerName:  `"@{busname}"`,
			},
			&aa.Dbus{
				Access: []string{"send"}, Bus: rules["bus"], Path: rules["path"],
				Interface: iface,
				PeerName:  `"{@{busname},org.freedesktop.DBus}"`,
			},
		)
	}

	res = append(res,
		// DBus.Properties
		&aa.Dbus{
			Access: []string{"send", "receive"}, Bus: rules["bus"], Path: rules["path"],
			Interface: "org.freedesktop.DBus.Properties",
			Member:    "{Get,GetAll,Set,PropertiesChanged}",
			PeerName:  `"{@{busname},org.freedesktop.DBus}"`,
		},

		// DBus.Introspectable
		&aa.Dbus{
			Access: []string{"receive"}, Bus: rules["bus"], Path: rules["path"],
			Interface: "org.freedesktop.DBus.Introspectable",
			Member:    "Introspect",
			PeerName:  `"@{busname}"`,
		},

		// DBus.ObjectManager
		&aa.Dbus{
			Access: []string{"receive"}, Bus: rules["bus"], Path: rules["path"],
			Interface: "org.freedesktop.DBus.ObjectManager",
			Member:    "GetManagedObjects",
			PeerName:  `"{@{busname},` + rules["name"] + `}"`,
		},
		&aa.Dbus{
			Access: []string{"send"}, Bus: rules["bus"], Path: rules["path"],
			Interface: "org.freedesktop.DBus.ObjectManager",
			Member:    "{InterfacesAdded,InterfacesRemoved}",
			PeerName:  `"{@{busname},org.freedesktop.DBus}"`,
		},
	)
	return res
}

func (d Dbus) talk(rules map[string]string, name string) aa.Rules {
	interfaces := getInterfaces(rules)

	res := aa.Rules{
		&aa.Unix{
			Access: []string{"bind"}, Type: "stream",
			Address: `@@{udbus}/bus/` + name + `/` + rules["bus"],
		},
	}

	// Interfaces
	for _, iface := range interfaces {
		res = append(res, &aa.Dbus{
			Access: []string{"send", "receive"}, Bus: rules["bus"], Path: rules["path"],
			Interface: iface,
			PeerName:  `"{@{busname},` + rules["name"] + `}"`, PeerLabel: rules["label"],
		})
	}

	res = append(res,
		// DBus.Properties
		&aa.Dbus{
			Access: []string{"send", "receive"}, Bus: rules["bus"], Path: rules["path"],
			Interface: "org.freedesktop.DBus.Properties",
			Member:    "{Get,GetAll,Set,PropertiesChanged}",
			PeerName:  `"{@{busname},` + rules["name"] + `}"`, PeerLabel: rules["label"],
		},

		// DBus.Introspectable
		&aa.Dbus{
			Access: []string{"send"}, Bus: rules["bus"], Path: rules["path"],
			Interface: "org.freedesktop.DBus.Introspectable",
			Member:    "Introspect",
			PeerName:  `"{@{busname},` + rules["name"] + `}"`, PeerLabel: rules["label"],
		},

		// DBus.ObjectManager
		&aa.Dbus{
			Access: []string{"send"}, Bus: rules["bus"], Path: rules["path"],
			Interface: "org.freedesktop.DBus.ObjectManager",
			Member:    "GetManagedObjects",
			PeerName:  `"{@{busname},` + rules["name"] + `}"`, PeerLabel: rules["label"],
		},
		&aa.Dbus{
			Access: []string{"receive"}, Bus: rules["bus"], Path: rules["path"],
			Interface: "org.freedesktop.DBus.ObjectManager",
			Member:    "{InterfacesAdded,InterfacesRemoved}",
			PeerName:  `"{@{busname},` + rules["name"] + `}"`, PeerLabel: rules["label"],
		},
	)
	return res
}