system/dbus: drop proxy output beyond threshold

This prevents xdg-dbus-proxy from running the priv process out of memory.

Signed-off-by: Ophestra <cat@gensokyo.uk>

1 files changed, 279 insertions, 0 deletions

diff --git a/system/dbus_test.go b/system/dbus_test.go
new file mode 100644
index 00000000..d5d053b7
--- /dev/null
+++ b/system/dbus_test.go
@@ -0,0 +1,279 @@
+package system
+
+import (
+	"reflect"
+	"slices"
+	"strings"
+	"syscall"
+	"testing"
+)
+
+func TestLinePrefixWriter(t *testing.T) {
+	testCases := []struct {
+		name    string
+		prefix  string
+		f       func(w func(s string))
+		wantErr []error
+		wantPt  []string
+		want    []string
+		wantExt []string
+		wantBuf string
+	}{
+		{"nop", "(nop) ", func(func(string)) {}, nil, nil, nil, nil, ""},
+
+		{"partial", "(break) ", func(w func(string)) {
+			w("C-65533: -> ")
+		}, nil, nil, nil, nil,
+			"C-65533: -> "},
+
+		{"break", "(break) ", func(w func(string)) {
+			w("C-65533: -> ")
+			w("org.freedesktop.DBus fake ListNames\n")
+		}, nil, nil, []string{
+			"C-65533: -> org.freedesktop.DBus fake ListNames",
+		}, nil, ""},
+
+		{"break pt", "(break pt) ", func(w func(string)) {
+			w("init: ")
+			w("received setup parameters\n")
+		}, nil, []string{
+			"init: received setup parameters",
+		}, nil, nil, ""},
+
+		{"threshold", "(threshold) ", func(w func(s string)) {
+			w(string(make([]byte, lpwSizeThreshold)))
+			w("\n")
+		}, []error{nil, syscall.ENOMEM}, nil, nil, nil,
+			string(make([]byte, lpwSizeThreshold))},
+
+		{"threshold multi", "(threshold multi) ", func(w func(s string)) {
+			w(":3\n")
+			w(string(make([]byte, lpwSizeThreshold-3)))
+			w("\n")
+		}, []error{nil, nil, syscall.ENOMEM}, nil, []string{
+			":3",
+		}, nil, string(make([]byte, lpwSizeThreshold-3))},
+
+		{"threshold multi partial", "(threshold multi partial) ", func(w func(s string)) {
+			w(":3\n")
+			w(string(make([]byte, lpwSizeThreshold-2)))
+		}, []error{nil, syscall.ENOMEM}, nil, []string{
+			":3",
+		}, []string{
+			"dropped 16777215 bytes of output",
+		}, string(make([]byte, lpwSizeThreshold-2))},
+
+		{"threshold exact", "(threshold exact) ", func(w func(s string)) {
+			w(string(make([]byte, lpwSizeThreshold-1)))
+			w("\n")
+		}, nil, nil, []string{
+			string(make([]byte, lpwSizeThreshold-1)),
+		}, nil, ""},
+
+		{"sample", "(dbus) ", func(w func(s string)) {
+			w("init: received setup parameters\n")
+			w(`init: mounting "/nix/store/5gml2l2cj28yvyfyzblzjy1laqpxmyzd-libselinux-3.8.1/lib" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/5gml2l2cj28yvyfyzblzjy1laqpxmyzd-libselinux-3.8.1/lib" on "/sysroot/nix/store/5gml2l2cj28yvyfyzblzjy1laqpxmyzd-libselinux-3.8.1/lib" flags 0x4005` + "\n")
+			w(`init: mounting "/nix/store/bcs094l67dlbqf7idxxbljp293zms9mh-util-linux-minimal-2.41-lib/lib" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/bcs094l67dlbqf7idxxbljp293zms9mh-util-linux-minimal-2.41-lib/lib" on "/sysroot/nix/store/bcs094l67dlbqf7idxxbljp293zms9mh-util-linux-minimal-2.41-lib/lib" flags 0x4005` + "\n")
+			w(`init: mounting "/nix/store/jl19fdc7gdxqz9a1s368r9d15vpirnqy-zlib-1.3.1/lib" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/jl19fdc7gdxqz9a1s368r9d15vpirnqy-zlib-1.3.1/lib" on "/sysroot/nix/store/jl19fdc7gdxqz9a1s368r9d15vpirnqy-zlib-1.3.1/lib" flags 0x4005` + "\n")
+			w(`init: mounting "/nix/store/rnn29mhynsa4ncmk0fkcrdr29n0j20l4-libffi-3.4.8/lib" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/rnn29mhynsa4ncmk0fkcrdr29n0j20l4-libffi-3.4.8/lib" on "/sysroot/nix/store/rnn29mhynsa4ncmk0fkcrdr29n0j20l4-libffi-3.4.8/lib" flags 0x4005` + "\n")
+			w(`init: mounting "/nix/store/vvp8hlss3d5q6hn0cifq04jrpnp6bini-pcre2-10.44/lib" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/vvp8hlss3d5q6hn0cifq04jrpnp6bini-pcre2-10.44/lib" on "/sysroot/nix/store/vvp8hlss3d5q6hn0cifq04jrpnp6bini-pcre2-10.44/lib" flags 0x4005` + "\n")
+			w(`init: mounting "/nix/store/y3nxdc2x8hwivppzgx5hkrhacsh87l21-glib-2.84.3/lib" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/y3nxdc2x8hwivppzgx5hkrhacsh87l21-glib-2.84.3/lib" on "/sysroot/nix/store/y3nxdc2x8hwivppzgx5hkrhacsh87l21-glib-2.84.3/lib" flags 0x4005` + "\n")
+			w(`init: mounting "/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib" on "/sysroot/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib" flags 0x4005` + "\n")
+			w(`init: mounting "/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib64" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib" on "/sysroot/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib64" flags 0x4005` + "\n")
+			w(`init: mounting "/run/user/1000" flags 0x0` + "\n")
+			w(`init: resolved "/host/run/user/1000" on "/sysroot/run/user/1000" flags 0x4005` + "\n")
+			w(`init: mounting "/tmp/hakurei.0/99dd71ee2146369514e0d10783368f8f" flags 0x2` + "\n")
+			w(`init: resolved "/host/tmp/hakurei.0/99dd71ee2146369514e0d10783368f8f" on "/sysroot/tmp/hakurei.0/99dd71ee2146369514e0d10783368f8f" flags 0x4004` + "\n")
+			w(`init: mounting "/nix/store/d2divmq2d897amikcwpdx7zrbpddxxcl-xdg-dbus-proxy-0.1.6/bin" flags 0x0` + "\n")
+			w(`init: resolved "/host/nix/store/d2divmq2d897amikcwpdx7zrbpddxxcl-xdg-dbus-proxy-0.1.6/bin" on "/sysroot/nix/store/d2divmq2d897amikcwpdx7zrbpddxxcl-xdg-dbus-proxy-0.1.6/bin" flags 0x4005` + "\n")
+			w("init: resolving presets 0xf\n")
+			w("init: 68 filter rules loaded\n")
+			w("init: starting initial program /nix/store/d2divmq2d897amikcwpdx7zrbpddxxcl-xdg-dbus-proxy-0.1.6/bin/xdg-dbus-proxy\n")
+			w("C1: -> org.freedesktop.DBus call org.freedesktop.DBus.Hello at /org/freedesktop/DBus\n")
+			w("C-65536: -> org.freedesktop.DBus fake wildcarded AddMatch for org.freedesktop.portal\n")
+			w("C-65535: -> org.freedesktop.DBus fake AddMatch for org.freedesktop.Notifications\n")
+			w("C-65534: -> org.freedesktop.DBus fake GetNameOwner for org.freedesktop.Notifications\n")
+			w("C-65533: -> org.freedesktop.DBus fake ListNames\n")
+			w("B1: <- org.freedesktop.DBus return from C1\n")
+			w("B2: <- org.freedesktop.DBus signal org.freedesktop.DBus.NameAcquired at /org/freedesktop/DBus\n")
+			w("B3: <- org.freedesktop.DBus return from C-65536\n")
+			w("*SKIPPED*\n")
+			w("B4: <- org.freedesktop.DBus return from C-65535\n")
+			w("*SKIPPED*\n")
+			w("B5: <- org.freedesktop.DBus return error org.freedesktop.DBus.Error.NameHasNoOwner from C-65534\n")
+			w("*SKIPPED*\n")
+			w("B6: <- org.freedesktop.DBus return from C-65533\n")
+			w("C-65532: -> org.freedesktop.DBus fake GetNameOwner for org.freedesktop.DBus\n")
+			w("*SKIPPED*\n")
+			w("B7: <- org.freedesktop.DBus return from C-65532\n")
+			w("*SKIPPED*\n")
+			w("C2: -> org.freedesktop.DBus call org.freedesktop.DBus.AddMatch at /org/freedesktop/DBus\n")
+			w("C3: -> org.freedesktop.DBus call org.freedesktop.DBus.GetNameOwner at /org/freedesktop/DBus\n")
+			w("C4: -> org.freedesktop.DBus call org.freedesktop.DBus.AddMatch at /org/freedesktop/DBus\n")
+			w("C5: -> org.freedesktop.DBus call org.freedesktop.DBus.StartServiceByName at /org/freedesktop/DBus\n")
+			w("B8: <- org.freedesktop.DBus return from C2\n")
+			w("B9: <- org.freedesktop.DBus return error org.freedesktop.DBus.Error.NameHasNoOwner from C3\n")
+			w("B10: <- org.freedesktop.DBus return from C4\n")
+			w("B12: <- org.freedesktop.DBus signal org.freedesktop.DBus.NameOwnerChanged at /org/freedesktop/DBus\n")
+			w("B11: <- org.freedesktop.DBus return from C5\n")
+			w("C6: -> org.freedesktop.DBus call org.freedesktop.DBus.GetNameOwner at /org/freedesktop/DBus\n")
+			w("B13: <- org.freedesktop.DBus return from C6\n")
+			w("C7: -> :1.4 call org.freedesktop.Notifications.GetServerInformation at /org/freedesktop/Notifications\n")
+			w("B4: <- :1.4 return from C7\n")
+			w("C8: -> :1.4 call org.freedesktop.Notifications.GetServerInformation at /org/freedesktop/Notifications\n")
+			w("B5: <- :1.4 return from C8\n")
+			w("C9: -> :1.4 call org.freedesktop.Notifications.Notify at /org/freedesktop/Notifications\n")
+			w("B6: <- :1.4 return from C9\n")
+			w("C10: -> org.freedesktop.DBus call org.freedesktop.DBus.RemoveMatch at /org/freedesktop/DBus\n")
+			w("C11: -> org.freedesktop.DBus call org.freedesktop.DBus.RemoveMatch at /org/freedesktop/DBus\n")
+			w("B14: <- org.freedesktop.DBus return from C10\n")
+			w("B15: <- org.freedesktop.DBus return from C11\n")
+			w("init: initial process exited with code 0\n")
+		}, nil, []string{
+			"init: received setup parameters",
+			`init: mounting "/nix/store/5gml2l2cj28yvyfyzblzjy1laqpxmyzd-libselinux-3.8.1/lib" flags 0x0`,
+			`init: resolved "/host/nix/store/5gml2l2cj28yvyfyzblzjy1laqpxmyzd-libselinux-3.8.1/lib" on "/sysroot/nix/store/5gml2l2cj28yvyfyzblzjy1laqpxmyzd-libselinux-3.8.1/lib" flags 0x4005`,
+			`init: mounting "/nix/store/bcs094l67dlbqf7idxxbljp293zms9mh-util-linux-minimal-2.41-lib/lib" flags 0x0`,
+			`init: resolved "/host/nix/store/bcs094l67dlbqf7idxxbljp293zms9mh-util-linux-minimal-2.41-lib/lib" on "/sysroot/nix/store/bcs094l67dlbqf7idxxbljp293zms9mh-util-linux-minimal-2.41-lib/lib" flags 0x4005`,
+			`init: mounting "/nix/store/jl19fdc7gdxqz9a1s368r9d15vpirnqy-zlib-1.3.1/lib" flags 0x0`,
+			`init: resolved "/host/nix/store/jl19fdc7gdxqz9a1s368r9d15vpirnqy-zlib-1.3.1/lib" on "/sysroot/nix/store/jl19fdc7gdxqz9a1s368r9d15vpirnqy-zlib-1.3.1/lib" flags 0x4005`,
+			`init: mounting "/nix/store/rnn29mhynsa4ncmk0fkcrdr29n0j20l4-libffi-3.4.8/lib" flags 0x0`,
+			`init: resolved "/host/nix/store/rnn29mhynsa4ncmk0fkcrdr29n0j20l4-libffi-3.4.8/lib" on "/sysroot/nix/store/rnn29mhynsa4ncmk0fkcrdr29n0j20l4-libffi-3.4.8/lib" flags 0x4005`,
+			`init: mounting "/nix/store/vvp8hlss3d5q6hn0cifq04jrpnp6bini-pcre2-10.44/lib" flags 0x0`,
+			`init: resolved "/host/nix/store/vvp8hlss3d5q6hn0cifq04jrpnp6bini-pcre2-10.44/lib" on "/sysroot/nix/store/vvp8hlss3d5q6hn0cifq04jrpnp6bini-pcre2-10.44/lib" flags 0x4005`,
+			`init: mounting "/nix/store/y3nxdc2x8hwivppzgx5hkrhacsh87l21-glib-2.84.3/lib" flags 0x0`,
+			`init: resolved "/host/nix/store/y3nxdc2x8hwivppzgx5hkrhacsh87l21-glib-2.84.3/lib" on "/sysroot/nix/store/y3nxdc2x8hwivppzgx5hkrhacsh87l21-glib-2.84.3/lib" flags 0x4005`,
+			`init: mounting "/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib" flags 0x0`,
+			`init: resolved "/host/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib" on "/sysroot/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib" flags 0x4005`,
+			`init: mounting "/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib64" flags 0x0`,
+			`init: resolved "/host/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib" on "/sysroot/nix/store/zdpby3l6azi78sl83cpad2qjpfj25aqx-glibc-2.40-66/lib64" flags 0x4005`,
+			`init: mounting "/run/user/1000" flags 0x0`,
+			`init: resolved "/host/run/user/1000" on "/sysroot/run/user/1000" flags 0x4005`,
+			`init: mounting "/tmp/hakurei.0/99dd71ee2146369514e0d10783368f8f" flags 0x2`,
+			`init: resolved "/host/tmp/hakurei.0/99dd71ee2146369514e0d10783368f8f" on "/sysroot/tmp/hakurei.0/99dd71ee2146369514e0d10783368f8f" flags 0x4004`,
+			`init: mounting "/nix/store/d2divmq2d897amikcwpdx7zrbpddxxcl-xdg-dbus-proxy-0.1.6/bin" flags 0x0`,
+			`init: resolved "/host/nix/store/d2divmq2d897amikcwpdx7zrbpddxxcl-xdg-dbus-proxy-0.1.6/bin" on "/sysroot/nix/store/d2divmq2d897amikcwpdx7zrbpddxxcl-xdg-dbus-proxy-0.1.6/bin" flags 0x4005`,
+			"init: resolving presets 0xf",
+			"init: 68 filter rules loaded",
+			"init: starting initial program /nix/store/d2divmq2d897amikcwpdx7zrbpddxxcl-xdg-dbus-proxy-0.1.6/bin/xdg-dbus-proxy",
+
+			"init: initial process exited with code 0",
+		}, []string{
+			"C1: -> org.freedesktop.DBus call org.freedesktop.DBus.Hello at /org/freedesktop/DBus",
+			"C-65536: -> org.freedesktop.DBus fake wildcarded AddMatch for org.freedesktop.portal",
+			"C-65535: -> org.freedesktop.DBus fake AddMatch for org.freedesktop.Notifications",
+			"C-65534: -> org.freedesktop.DBus fake GetNameOwner for org.freedesktop.Notifications",
+			"C-65533: -> org.freedesktop.DBus fake ListNames",
+			"B1: <- org.freedesktop.DBus return from C1",
+			"B2: <- org.freedesktop.DBus signal org.freedesktop.DBus.NameAcquired at /org/freedesktop/DBus",
+			"B3: <- org.freedesktop.DBus return from C-65536",
+			"*SKIPPED*",
+			"B4: <- org.freedesktop.DBus return from C-65535",
+			"*SKIPPED*",
+			"B5: <- org.freedesktop.DBus return error org.freedesktop.DBus.Error.NameHasNoOwner from C-65534",
+			"*SKIPPED*",
+			"B6: <- org.freedesktop.DBus return from C-65533",
+			"C-65532: -> org.freedesktop.DBus fake GetNameOwner for org.freedesktop.DBus",
+			"*SKIPPED*",
+			"B7: <- org.freedesktop.DBus return from C-65532",
+			"*SKIPPED*",
+			"C2: -> org.freedesktop.DBus call org.freedesktop.DBus.AddMatch at /org/freedesktop/DBus",
+			"C3: -> org.freedesktop.DBus call org.freedesktop.DBus.GetNameOwner at /org/freedesktop/DBus",
+			"C4: -> org.freedesktop.DBus call org.freedesktop.DBus.AddMatch at /org/freedesktop/DBus",
+			"C5: -> org.freedesktop.DBus call org.freedesktop.DBus.StartServiceByName at /org/freedesktop/DBus",
+			"B8: <- org.freedesktop.DBus return from C2",
+			"B9: <- org.freedesktop.DBus return error org.freedesktop.DBus.Error.NameHasNoOwner from C3",
+			"B10: <- org.freedesktop.DBus return from C4",
+			"B12: <- org.freedesktop.DBus signal org.freedesktop.DBus.NameOwnerChanged at /org/freedesktop/DBus",
+			"B11: <- org.freedesktop.DBus return from C5",
+			"C6: -> org.freedesktop.DBus call org.freedesktop.DBus.GetNameOwner at /org/freedesktop/DBus",
+			"B13: <- org.freedesktop.DBus return from C6",
+			"C7: -> :1.4 call org.freedesktop.Notifications.GetServerInformation at /org/freedesktop/Notifications",
+			"B4: <- :1.4 return from C7",
+			"C8: -> :1.4 call org.freedesktop.Notifications.GetServerInformation at /org/freedesktop/Notifications",
+			"B5: <- :1.4 return from C8",
+			"C9: -> :1.4 call org.freedesktop.Notifications.Notify at /org/freedesktop/Notifications",
+			"B6: <- :1.4 return from C9",
+			"C10: -> org.freedesktop.DBus call org.freedesktop.DBus.RemoveMatch at /org/freedesktop/DBus",
+			"C11: -> org.freedesktop.DBus call org.freedesktop.DBus.RemoveMatch at /org/freedesktop/DBus",
+			"B14: <- org.freedesktop.DBus return from C10",
+			"B15: <- org.freedesktop.DBus return from C11",
+		}, nil, ""},
+	}
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			gotPt := make([]string, 0, len(tc.wantPt))
+			out := &linePrefixWriter{
+				prefix: tc.prefix,
+				println: func(v ...any) {
+					if len(v) != 1 {
+						t.Fatalf("invalid call to println: %#v", v)
+					}
+					gotPt = append(gotPt, v[0].(string))
+				},
+				buf: new(strings.Builder),
+			}
+
+			var pos int
+			tc.f(func(s string) {
+				if _, err := out.Write([]byte(s)); err != nil {
+					if tc.wantErr != nil {
+						if !reflect.DeepEqual(err, tc.wantErr[pos]) {
+							t.Fatalf("Write: error = %v, want %v", err, tc.wantErr[pos])
+						}
+					} else {
+						t.Fatalf("Write: unexpected error: %v", err)
+						return
+					}
+				}
+				pos++
+			})
+
+			if !slices.Equal(out.msg, tc.want) {
+				t.Errorf("msg: %#v, want %#v", out.msg, tc.want)
+			}
+
+			if out.buf.String() != tc.wantBuf {
+				t.Errorf("buf: %q, want %q", out.buf, tc.wantBuf)
+			}
+
+			wantPt := make([]string, len(tc.wantPt))
+			for i, m := range tc.wantPt {
+				wantPt[i] = tc.prefix + m
+			}
+			if !slices.Equal(gotPt, wantPt) {
+				t.Errorf("passthrough: %#v, want %#v", gotPt, wantPt)
+			}
+
+			wantDump := make([]string, len(tc.want)+len(tc.wantExt))
+			for i, m := range tc.want {
+				wantDump[i] = tc.prefix + m
+			}
+			for i, m := range tc.wantExt {
+				wantDump[len(tc.want)+i] = tc.prefix + m
+			}
+			t.Run("dump", func(t *testing.T) {
+				got := make([]string, 0, len(wantDump))
+				out.println = func(v ...any) {
+					if len(v) != 1 {
+						t.Fatalf("Dump: invalid call to println: %#v", v)
+					}
+					got = append(got, v[0].(string))
+				}
+				out.Dump()
+
+				if !slices.Equal(got, wantDump) {
+					t.Errorf("Dump: %#v, want %#v", got, wantDump)
+				}
+			})
+		})
+	}
+}