1
1
mirror of https://github.com/quexten/goldwarden.git synced 2024-12-30 14:55:36 +03:00
goldwarden/agent/processsecurity/unix.go

60 lines
1.1 KiB
Go
Raw Normal View History

2023-08-24 04:22:03 +03:00
//go:build linux || freebsd
package processsecurity
2023-12-23 10:37:17 +03:00
import (
"fmt"
"github.com/godbus/dbus/v5"
"golang.org/x/sys/unix"
)
2023-08-24 04:22:03 +03:00
func DisableDumpable() error {
return unix.Prctl(unix.PR_SET_DUMPABLE, 0, 0, 0, 0)
}
2023-12-23 10:37:17 +03:00
func MonitorLocks(onlock func()) error {
bus, err := dbus.SessionBus()
if err != nil {
return err
}
err = bus.AddMatchSignal(dbus.WithMatchInterface("org.gnome.ScreenSaver"))
if err != nil {
return err
}
err = bus.AddMatchSignal(dbus.WithMatchMember("org.freedesktop.ScreenSaver"))
if err != nil {
return err
}
signals := make(chan *dbus.Signal, 10)
bus.Signal(signals)
for {
select {
case message := <-signals:
fmt.Println("Message:", message)
fmt.Println("name ", message.Name)
if message.Name == "org.gnome.ScreenSaver.ActiveChanged" {
if len(message.Body) == 0 {
continue
}
locked, err := message.Body[0].(bool)
if err || locked {
onlock()
}
}
if message.Name == "org.freedesktop.ScreenSaver.ActiveChanged" {
if len(message.Body) == 0 {
continue
}
locked, err := message.Body[0].(bool)
if err || locked {
onlock()
}
}
}
}
return nil
}