Release 0.101.57

Author: wh1te909

src/api/agents.js

@@ -265,3 +265,134 @@ export async function wakeUpWOL(agent_id) {
   const { data } = await axios.post(`${baseUrl}/${agent_id}/wol/`);
   return data;
 }
+
+export async function fetchAgentRegistry(
+  agent_id,
+  path,
+  page = 1,
+  hiveSearch = false,
+) {
+  try {
+    const { data } = await axios.get(`${baseUrl}/${agent_id}/registry/`, {
+      params: { path: `${path}`, page, page_size: hiveSearch ? 100000 : 400 },
+    });
+    return data;
+  } catch (e) {
+    console.error(e);
+    throw e;
+  }
+}
+
+export async function deleteRegistryKey(agent_id, path) {
+  try {
+    const { data } = await axios.delete(
+      `${baseUrl}/${agent_id}/registry/delete-key/`,
+      {
+        params: { path: `${path}` },
+      },
+    );
+    return data;
+  } catch (e) {
+    console.error(e);
+  }
+}
+
+export async function createRegistryKey(agent_id, path) {
+  try {
+    const { data } = await axios.post(
+      `${baseUrl}/${agent_id}/registry/create-key/`,
+      { path },
+    );
+    return data;
+  } catch (e) {
+    console.error(e);
+    throw e;
+  }
+}
+
+export async function renameRegistryKey(agent_id, old_path, new_path) {
+  try {
+    const { data } = await axios.post(
+      `${baseUrl}/${agent_id}/registry/rename-key/`,
+      { old_path, new_path },
+    );
+    return data;
+  } catch (e) {
+    console.error(e);
+    throw e;
+  }
+}
+
+export async function deleteRegistryValue(agent_id, path, name) {
+  try {
+    const { data } = await axios.delete(
+      `${baseUrl}/${agent_id}/registry/delete-value/`,
+      {
+        params: { path, name },
+      },
+    );
+    return data;
+  } catch (e) {
+    console.error("Failed to delete value:", e);
+    throw e;
+  }
+}
+
+export async function renameRegistryValue(agentId, path, oldName, newName) {
+  try {
+    const { data } = await axios.post(
+      `${baseUrl}/${agentId}/registry/rename-value/`,
+      {
+        path,
+        old_name: oldName,
+        new_name: newName,
+      },
+    );
+    return data;
+  } catch (e) {
+    console.error("Failed to rename value:", e);
+    throw e;
+  }
+}
+
+export async function modifyRegistryValue(
+  agentId,
+  path,
+  name,
+  type,
+  dataValue,
+) {
+  try {
+    const { data } = await axios.post(
+      `${baseUrl}/${agentId}/registry/modify-value/`,
+      {
+        path,
+        name,
+        type,
+        data: dataValue,
+      },
+    );
+    return data;
+  } catch (e) {
+    console.error("Failed to modify registry value:", e);
+    throw e;
+  }
+}
+
+export async function createRegistryValue(agentId, path, name, type, data) {
+  const formData = new FormData();
+  formData.append("path", path);
+  formData.append("name", name);
+  formData.append("type", type);
+  formData.append("data", data);
+  try {
+    const { data } = await axios.post(
+      `${baseUrl}/${agentId}/registry/create-value/`,
+      formData,
+    );
+    return data;
+  } catch (e) {
+    console.error("Failed to create registry value:", e);
+    throw e;
+  }
+}

src/assets/binary_type.png

@@ -129,6 +129,10 @@
                 v-model="localRole.can_recover_agents"
                 label="Recover Agents"
               />
+              <q-checkbox
+                v-model="localRole.can_use_registry"
+                label="Use Registry"
+              />
             </div>
           </q-card-section>
           <div class="text-subtitle2">Core</div>
@@ -552,6 +556,7 @@ export default {
           // reporting perms
           can_view_reports: false,
           can_manage_reports: false,
+          can_use_registry: false,
         });
 
     const loading = ref(false);

src/assets/string_type.png

@@ -0,0 +1,105 @@
+<template>
+  <q-card-section class="q-px-xs q-pb-md q-pt-xs" v-if="hasText">
+    <script-output-copy-clip label="Live Output" :data="outputText" />
+    <q-separator class="q-my-sm" />
+  </q-card-section>
+  <div class="command-stream" ref="streamContainer" v-if="hasText">
+    <div class="terminal">
+      <pre class="mt-0">{{ outputText }}</pre>
+    </div>
+  </div>
+</template>
+
+<script setup lang="ts">
+import {
+  ref,
+  onMounted,
+  onUnmounted,
+  watchEffect,
+  nextTick,
+  computed,
+} from "vue";
+import { useTemplateRef } from "vue";
+import { useAgentCmdWSConnection } from "@/websocket/agent";
+import ScriptOutputCopyClip from "@/components/scripts/ScriptOutputCopyClip.vue";
+import { uid } from "quasar";
+
+const props = defineProps({
+  agentId: { type: String, required: true },
+  cmd: { type: String, required: true },
+  shell: { type: String, required: true },
+  custom_shell: { type: String, required: false },
+  timeout: { type: Number, default: 10 },
+});
+
+// emits
+const emit = defineEmits(["updateOutput", "streamLoaded", "streamClosed"]);
+
+const cmdId = uid();
+const { send, data, reset, close, status } = useAgentCmdWSConnection(
+  props.agentId,
+  cmdId,
+);
+
+const outputText = ref("");
+const streamContainer = useTemplateRef<HTMLElement>("streamContainer");
+let firstChunk = false;
+
+const hasText = computed(() => outputText.value.trim() !== "");
+
+watchEffect(() => {
+  if (data.value.length) {
+    outputText.value = data.value.map((msg) => msg.output).join("\n");
+    emit("updateOutput", outputText.value);
+
+    if (!firstChunk && data.value.length > 0) {
+      firstChunk = true;
+      emit("streamLoaded");
+    }
+
+    nextTick(() => {
+      if (streamContainer.value) {
+        streamContainer.value.scrollTop = streamContainer.value.scrollHeight;
+      }
+    });
+  }
+});
+
+watchEffect(() => {
+  if (status.value === "CLOSED") emit("streamClosed");
+});
+
+onMounted(() => {
+  outputText.value = "";
+  reset();
+  send(
+    JSON.stringify({
+      shell: props.shell,
+      cmd: props.cmd,
+      timeout: props.timeout,
+      run_as_user: false,
+      custom_shell: props.custom_shell,
+      stream: true,
+      cmd_id: cmdId,
+    }),
+  );
+});
+
+onUnmounted(close);
+</script>
+
+<style scoped>
+.command-stream {
+  font-family: monospace;
+  background: #191818;
+  color: #fff;
+  padding: 0 10px;
+  height: 30vh;
+  overflow-y: auto;
+  border: 1px solid #ccc;
+  border-radius: 4px;
+}
+.terminal {
+  white-space: pre-wrap;
+}
+</style>