223 lines
6.9 KiB
Go
223 lines
6.9 KiB
Go
package dbus_test
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"git.ophivana.moe/cat/fortify/dbus"
|
|
)
|
|
|
|
var samples = []dbusTestCase{
|
|
{
|
|
"org.chromium.Chromium", &dbus.Config{
|
|
See: nil,
|
|
Talk: []string{"org.freedesktop.Notifications", "org.freedesktop.FileManager1", "org.freedesktop.ScreenSaver",
|
|
"org.freedesktop.secrets", "org.kde.kwalletd5", "org.kde.kwalletd6", "org.gnome.SessionManager"},
|
|
Own: []string{"org.chromium.Chromium.*", "org.mpris.MediaPlayer2.org.chromium.Chromium.*",
|
|
"org.mpris.MediaPlayer2.chromium.*"},
|
|
Call: map[string]string{"org.freedesktop.portal.*": "*"},
|
|
Broadcast: map[string]string{"org.freedesktop.portal.*": "@/org/freedesktop/portal/*"},
|
|
Log: false,
|
|
Filter: true,
|
|
}, false, false,
|
|
[2]string{"unix:path=/run/user/1971/bus", "/tmp/fortify.1971/12622d846cc3fe7b4c10359d01f0eb47/bus"},
|
|
[]string{
|
|
"unix:path=/run/user/1971/bus",
|
|
"/tmp/fortify.1971/12622d846cc3fe7b4c10359d01f0eb47/bus",
|
|
"--filter",
|
|
"--talk=org.freedesktop.Notifications",
|
|
"--talk=org.freedesktop.FileManager1",
|
|
"--talk=org.freedesktop.ScreenSaver",
|
|
"--talk=org.freedesktop.secrets",
|
|
"--talk=org.kde.kwalletd5",
|
|
"--talk=org.kde.kwalletd6",
|
|
"--talk=org.gnome.SessionManager",
|
|
"--own=org.chromium.Chromium.*",
|
|
"--own=org.mpris.MediaPlayer2.org.chromium.Chromium.*",
|
|
"--own=org.mpris.MediaPlayer2.chromium.*",
|
|
"--call=org.freedesktop.portal.*=*",
|
|
"--broadcast=org.freedesktop.portal.*=@/org/freedesktop/portal/*",
|
|
},
|
|
},
|
|
{
|
|
"org.chromium.Chromium+", &dbus.Config{
|
|
See: nil,
|
|
Talk: []string{"org.bluez", "org.freedesktop.Avahi", "org.freedesktop.UPower"},
|
|
Own: nil,
|
|
Call: nil,
|
|
Broadcast: nil,
|
|
Log: false,
|
|
Filter: true,
|
|
}, false, false,
|
|
[2]string{"unix:path=/run/dbus/system_bus_socket", "/tmp/fortify.1971/12622d846cc3fe7b4c10359d01f0eb47/system_bus_socket"},
|
|
[]string{"unix:path=/run/dbus/system_bus_socket",
|
|
"/tmp/fortify.1971/12622d846cc3fe7b4c10359d01f0eb47/system_bus_socket",
|
|
"--filter",
|
|
"--talk=org.bluez",
|
|
"--talk=org.freedesktop.Avahi",
|
|
"--talk=org.freedesktop.UPower",
|
|
},
|
|
},
|
|
|
|
{
|
|
"dev.vencord.Vesktop", &dbus.Config{
|
|
See: nil,
|
|
Talk: []string{"org.freedesktop.Notifications", "org.kde.StatusNotifierWatcher"},
|
|
Own: []string{"dev.vencord.Vesktop.*", "org.mpris.MediaPlayer2.dev.vencord.Vesktop.*"},
|
|
Call: map[string]string{"org.freedesktop.portal.*": "*"},
|
|
Broadcast: map[string]string{"org.freedesktop.portal.*": "@/org/freedesktop/portal/*"},
|
|
Log: false,
|
|
Filter: true,
|
|
}, false, false,
|
|
[2]string{"unix:path=/run/user/1971/bus", "/tmp/fortify.1971/34c24f16a0d791d28835ededaf446033/bus"},
|
|
[]string{
|
|
"unix:path=/run/user/1971/bus",
|
|
"/tmp/fortify.1971/34c24f16a0d791d28835ededaf446033/bus",
|
|
"--filter",
|
|
"--talk=org.freedesktop.Notifications",
|
|
"--talk=org.kde.StatusNotifierWatcher",
|
|
"--own=dev.vencord.Vesktop.*",
|
|
"--own=org.mpris.MediaPlayer2.dev.vencord.Vesktop.*",
|
|
"--call=org.freedesktop.portal.*=*",
|
|
"--broadcast=org.freedesktop.portal.*=@/org/freedesktop/portal/*"},
|
|
},
|
|
|
|
{
|
|
"moe.ophivana.CrashTestDummy", &dbus.Config{
|
|
See: []string{"moe.ophivana.CrashTestDummy1"},
|
|
Talk: []string{"org.freedesktop.Notifications"},
|
|
Own: []string{"moe.ophivana.CrashTestDummy.*", "org.mpris.MediaPlayer2.moe.ophivana.CrashTestDummy.*"},
|
|
Call: map[string]string{"org.freedesktop.portal.*": "*"},
|
|
Broadcast: map[string]string{"org.freedesktop.portal.*": "@/org/freedesktop/portal/*"},
|
|
Log: true,
|
|
Filter: true,
|
|
}, false, false,
|
|
[2]string{"unix:path=/run/user/1971/bus", "/tmp/fortify.1971/5da7845287a936efbc2fa75d7d81e501/bus"},
|
|
[]string{
|
|
"unix:path=/run/user/1971/bus",
|
|
"/tmp/fortify.1971/5da7845287a936efbc2fa75d7d81e501/bus",
|
|
"--filter",
|
|
"--see=moe.ophivana.CrashTestDummy1",
|
|
"--talk=org.freedesktop.Notifications",
|
|
"--own=moe.ophivana.CrashTestDummy.*",
|
|
"--own=org.mpris.MediaPlayer2.moe.ophivana.CrashTestDummy.*",
|
|
"--call=org.freedesktop.portal.*=*",
|
|
"--broadcast=org.freedesktop.portal.*=@/org/freedesktop/portal/*",
|
|
"--log"},
|
|
},
|
|
{
|
|
"moe.ophivana.CrashTestDummy1", &dbus.Config{
|
|
See: []string{"moe.ophivana.CrashTestDummy"},
|
|
Talk: []string{"org.freedesktop.Notifications"},
|
|
Own: []string{"moe.ophivana.CrashTestDummy1.*", "org.mpris.MediaPlayer2.moe.ophivana.CrashTestDummy1.*"},
|
|
Call: map[string]string{"org.freedesktop.portal.*": "*"},
|
|
Broadcast: map[string]string{"org.freedesktop.portal.*": "@/org/freedesktop/portal/*"},
|
|
Log: true,
|
|
Filter: true,
|
|
}, false, true,
|
|
[2]string{"unix:path=/run/user/1971/bus", "/tmp/fortify.1971/5da7845287a936efbc2fa75d7d81e501/bus"},
|
|
[]string{
|
|
"unix:path=/run/user/1971/bus",
|
|
"/tmp/fortify.1971/5da7845287a936efbc2fa75d7d81e501/bus",
|
|
"--filter",
|
|
"--see=moe.ophivana.CrashTestDummy",
|
|
"--talk=org.freedesktop.Notifications",
|
|
"--own=moe.ophivana.CrashTestDummy1.*",
|
|
"--own=org.mpris.MediaPlayer2.moe.ophivana.CrashTestDummy1.*",
|
|
"--call=org.freedesktop.portal.*=*",
|
|
"--broadcast=org.freedesktop.portal.*=@/org/freedesktop/portal/*",
|
|
"--log"},
|
|
},
|
|
}
|
|
|
|
type dbusTestCase struct {
|
|
id string
|
|
c *dbus.Config
|
|
wantErr bool
|
|
wantErrF bool
|
|
bus [2]string
|
|
want []string
|
|
}
|
|
|
|
var (
|
|
testCasesV []dbusTestCase
|
|
testCasePairsV map[string][2]dbusTestCase
|
|
|
|
testCaseOnce sync.Once
|
|
)
|
|
|
|
func testCases() []dbusTestCase {
|
|
testCaseOnce.Do(testCaseGenerate)
|
|
return testCasesV
|
|
}
|
|
|
|
func testCasePairs() map[string][2]dbusTestCase {
|
|
testCaseOnce.Do(testCaseGenerate)
|
|
return testCasePairsV
|
|
}
|
|
|
|
func injectNulls(t *[]string) {
|
|
f := make([]string, len(*t))
|
|
for i := range f {
|
|
f[i] = "\x00" + (*t)[i] + "\x00"
|
|
}
|
|
*t = f
|
|
}
|
|
|
|
func testCaseGenerate() {
|
|
// create null-injected test cases
|
|
testCasesV = make([]dbusTestCase, len(samples)*2)
|
|
for i := range samples {
|
|
testCasesV[i] = samples[i]
|
|
testCasesV[len(samples)+i] = samples[i]
|
|
testCasesV[len(samples)+i].c = new(dbus.Config)
|
|
*testCasesV[len(samples)+i].c = *samples[i].c
|
|
|
|
// inject nulls
|
|
fi := &testCasesV[len(samples)+i]
|
|
fi.wantErr = true
|
|
fi.c = &*fi.c
|
|
|
|
injectNulls(&fi.c.See)
|
|
injectNulls(&fi.c.Talk)
|
|
injectNulls(&fi.c.Own)
|
|
}
|
|
|
|
// enumerate test case pairs
|
|
var pc int
|
|
for _, tc := range samples {
|
|
if tc.id != "" {
|
|
pc++
|
|
}
|
|
}
|
|
testCasePairsV = make(map[string][2]dbusTestCase, pc)
|
|
for i, tc := range testCasesV {
|
|
if tc.id == "" {
|
|
continue
|
|
}
|
|
|
|
// skip already enumerated system bus test
|
|
if tc.id[len(tc.id)-1] == '+' {
|
|
continue
|
|
}
|
|
|
|
ftp := [2]dbusTestCase{tc}
|
|
|
|
// system proxy tests always place directly after its user counterpart with id ending in +
|
|
if i+1 < len(testCasesV) && testCasesV[i+1].id[len(testCasesV[i+1].id)-1] == '+' {
|
|
// attach system bus config
|
|
ftp[1] = testCasesV[i+1]
|
|
|
|
// check for misplaced/mismatching tests
|
|
if ftp[0].wantErr != ftp[1].wantErr || ftp[0].id+"+" != ftp[1].id {
|
|
panic("mismatching session/system pairing")
|
|
}
|
|
}
|
|
|
|
k := tc.id
|
|
if tc.wantErr {
|
|
k = "malformed_" + k
|
|
}
|
|
testCasePairsV[k] = ftp
|
|
}
|
|
}
|