+
+
@@ -482,16 +586,23 @@
Agents
agentFieldConfig: {},
deployCommands: [],
filteredCommands: [],
- pageOpen: true,
+ bootstrapAbilities: [],
+ deadmanAbilities: [],
isAddingBootstrapAbility: false,
+ isAddingDeadmanAbility: false,
abilitySearchQuery: [],
abilitySearchResults: [],
+ isHoveringAgentCount: false,
+
// Modals
showDeployModal: false,
showConfigModal: false,
showDetailsModal: false,
+ showConfirmModal: false,
+ confirmText: '',
+ confirmFunction: () => {},
// Input validation
requiredFieldsConfig: ['sleep_min', 'sleep_max', 'watchdog', 'untrusted_timer', 'implant_name'],
@@ -502,10 +613,12 @@
Agents
initPage() {
apiV2('GET', '/api/v2/agents').then((agents) => {
this.agents = agents;
- this.agents.sort((a, b) => new Date(b.last_seen).getTime() - new Date(a.last_seen).getTime());
+ this.sortAgents();
return apiV2('GET', '/api/v2/config/agents');
}).then((agentConfig) => {
this.agentConfig = agentConfig;
+ this.bootstrapAbilities = this.agentConfig.bootstrap_abilities || [];
+ this.deadmanAbilities = this.agentConfig.deadman_abilities || [];
return apiV2('GET', '/api/v2/abilities');
}).then(async (allAbilities) => {
this.abilities = allAbilities;
@@ -528,13 +641,17 @@
Agents
refreshAgents() {
apiV2('GET', '/api/v2/agents').then((agents) => {
this.agents = agents;
- this.agents.sort((a, b) => new Date(b.last_seen).getTime() - new Date(a.last_seen).getTime())
+ this.sortAgents();
}).catch((error) => {
toast('Error refreshing agents', false);
console.error(error);
});
},
+ sortAgents() {
+ this.agents.sort((a, b) => new Date(b.last_seen.replace(/-/g, "/")).getTime() - new Date(a.last_seen.replace(/-/g, "/")).getTime());
+ },
+
selectAbility() {
this.selectedAbility = this.deploymentAbilities.find((ability) => ability.ability_id === this.selectedAbilityId);
this.platforms = [...new Set(this.selectedAbility.executors.map((executor) => executor.platform))];
@@ -620,20 +737,30 @@
Agents
},
addBootstrapAbility(id) {
- this.agentConfig.bootstrap_abilities.push(id);
+ this.bootstrapAbilities.push(id);
this.abilitySearchQuery = '';
this.abilitySearchResults = [];
this.isAddingBootstrapAbility = false;
},
- getAgentStatus(agent) {
- let lastSeen = Date.parse(agent.last_seen);
- let isAlive = Date.now() - lastSeen < (agent.sleep_max * 1000) + 500;
+ addDeadmanAbility(id) {
+ this.deadmanAbilities.push(id);
+ this.abilitySearchQuery = '';
+ this.abilitySearchResults = [];
+ this.isAddingDeadmanAbility = false;
+ },
- if (agent.sleep_max === 3 && agent.sleep_min === 3 && agent.watchdog === 1) {
- return (Date.now() - lastSeen > 60000) ? 'dead' : 'pending kill'
+ getAgentStatus(agent) {
+ if (!agent.last_seen) return '';
+ let lastSeen = new Date(agent.last_seen).getTime();
+ let msSinceSeen = Date.now() - lastSeen;
+ // Give a buffer of 1 minute to mark an agent dead
+ let isAlive = (msSinceSeen < (agent.sleep_max * 1000));
+
+ if (msSinceSeen <= 60000 && agent.sleep_min === 3 && agent.sleep_max === 3 && agent.watchdog === 1) {
+ return 'pending kill'
} else {
- return isAlive ? 'alive' : 'dead';
+ return msSinceSeen <= 60000 || isAlive ? 'alive' : 'dead';
}
},
@@ -653,7 +780,11 @@
Agents
this.fieldErrorsConfig = validateInputs(this.agentConfig, this.requiredFieldsConfig);
if (this.fieldErrorsConfig.length) return;
- let reqBody = { ...this.agentConfig };
+ let reqBody = {
+ ...this.agentConfig,
+ bootstrap_abilities: this.bootstrapAbilities,
+ deadman_abilities: this.deadmanAbilities
+ };
delete reqBody.deployments;
apiV2('PATCH', '/api/v2/config/agents', reqBody).then((response) => {
this.showConfigModal = false;
@@ -667,29 +798,67 @@
Agents
removeAgent(paw, index) {
apiV2('DELETE', `/api/v2/agents/${paw}`).then(() => {
this.agents.splice(index, 1);
- toast('Agent has been removed, but may reappear if it beacons again.', true);
}).catch((error) => {
toast('Error removing the agent', false);
console.error(error);
});
},
- killAgent(agent) {
- if (confirm(`Are you sure you want to kill agent "${agent.paw}"?`)) {
- let reqBody = {
- watchdog: 1,
- sleep_min: 3,
- sleep_max: 3
- };
+ confirmAlt(msg, fn) {
+ this.confirmText = msg;
+ this.confirmFunction = fn;
+ this.showConfirmModal = true;
+ },
+
+ removeDeadAgents() {
+ this.agents.forEach((agent, index) => {
+ if (this.getAgentStatus(agent) === 'dead') {
+ this.removeAgent(agent.paw, index);
+ }
+ });
+ },
+ removeAllAgents() {
+ this.agents.forEach((agent, index) => {
+ this.removeAgent(agent.paw, index);
+ });
+ },
+
+ killAgent(agent) {
+ let reqBody = {
+ watchdog: 1,
+ sleep_min: 3,
+ sleep_max: 3
+ };
+
+ apiV2('PATCH', `/api/v2/agents/${agent.paw}`, reqBody).then((response) => {
+ toast('Agent will be killed after its next beacon.', true);
+ }).catch((error) => {
+ toast('Error killing the agent', false);
+ console.error(error);
+ });
+ this.showDetailsModal = false;
+ },
+
+ killAllAgents() {
+ let reqBody = {
+ watchdog: 1,
+ sleep_min: 3,
+ sleep_max: 3
+ };
+ let numKilled = 0;
+
+ this.agents.forEach((agent, index) => {
apiV2('PATCH', `/api/v2/agents/${agent.paw}`, reqBody).then((response) => {
- toast('Agent will be killed after its next beacon.', true);
+ numKilled++;
+ if (numKilled === this.agents.length) {
+ toast('All agents have been configured to be killed by next beacon.', true);
+ }
}).catch((error) => {
toast('Error killing the agent', false);
console.error(error);
});
- this.showDetailsModal = false;
- }
+ });
}
};
}
@@ -720,6 +889,13 @@
Agents
width: 80px;
}
+ .is-white {
+ background-color: #636363;
+ }
+ .is-white:hover {
+ background-color: #505050 !important;
+ }
+
.cmd-copy-button {
position: absolute;
top: 10px;