wazuh-kibana-app/public/controllers/agents.js
2018-04-12 11:58:15 +02:00

483 lines
18 KiB
JavaScript

const app = require('ui/modules').get('app/wazuh', []);
const beautifier = require('plugins/wazuh/utils/json-beautifier');
app.controller('agentsController',
function ($scope, $location, $q, $rootScope, appState, genericReq, apiReq, AgentsAutoComplete, errorHandler, metricService) {
// Timestamp for visualizations at controller's startup
if(!$rootScope.visTimestamp) {
$rootScope.visTimestamp = new Date().getTime();
if(!$rootScope.$$phase) $rootScope.$digest();
}
$rootScope.page = 'agents';
$scope.extensions = appState.getExtensions().extensions;
$scope.agentsAutoComplete = AgentsAutoComplete;
// Check the url hash and retrieve the tabView information
if ($location.search().tabView){
$scope.tabView = $location.search().tabView;
} else { // If tabView doesn't exist, default it to 'panels'
$scope.tabView = "panels";
$location.search("tabView", "panels");
}
// Check the url hash and retrivew the tab information
if ($location.search().tab){
$scope.tab = $location.search().tab;
} else { // If tab doesn't exist, default it to 'general'
$scope.tab = "general";
$location.search("tab", "general");
// Now we initialize the implicitFilter
$rootScope.currentImplicitFilter = "";
}
// Metrics Audit
const metricsAudit = {
auditNewFiles : '[vis-id="\'Wazuh-App-Agents-Audit-New-files-metric\'"]',
auditReadFiles : '[vis-id="\'Wazuh-App-Agents-Audit-Read-files-metric\'"]',
auditModifiedFiles: '[vis-id="\'Wazuh-App-Agents-Audit-Modified-files-metric\'"]',
auditRemovedFiles : '[vis-id="\'Wazuh-App-Agents-Audit-Removed-files-metric\'"]'
}
// Metrics Vulnerability Detector
const metricsVulnerability = {
vulnCritical: '[vis-id="\'Wazuh-App-Agents-VULS-Metric-Critical-severity\'"]',
vulnHigh : '[vis-id="\'Wazuh-App-Agents-VULS-Metric-High-severity\'"]',
vulnMedium : '[vis-id="\'Wazuh-App-Agents-VULS-Metric-Medium-severity\'"]',
vulnLow : '[vis-id="\'Wazuh-App-Agents-VULS-Metric-Low-severity\'"]'
}
// Metrics Scap
const metricsScap = {
scapLastScore : '[vis-id="\'Wazuh-App-Agents-OSCAP-Last-score\'"]',
scapHighestScore: '[vis-id="\'Wazuh-App-Agents-OSCAP-Higher-score-metric\'"]',
scapLowestScore : '[vis-id="\'Wazuh-App-Agents-OSCAP-Lower-score-metric\'"]'
}
// Metrics Virustotal
const metricsVirustotal = {
virusMalicious: '[vis-id="\'Wazuh-App-Agents-Virustotal-Total-Malicious\'"]',
virusPositives: '[vis-id="\'Wazuh-App-Agents-Virustotal-Total-Positives\'"]',
virusTotal : '[vis-id="\'Wazuh-App-Agents-Virustotal-Total\'"]'
}
$rootScope.tabVisualizations = {
general : 7,
fim : 8,
pm : 4,
vuls : 7,
oscap : 13,
audit : 15,
pci : 3,
virustotal : 6,
configuration: 0
};
// Object for matching nav items and Wazuh groups
const tabFilters = {
general : { group: '' },
fim : { group: 'syscheck' },
pm : { group: 'rootcheck' },
vuls : { group: 'vulnerability-detector' },
oscap : { group: 'oscap' },
audit : { group: 'audit' },
pci : { group: 'pci_dss' },
virustotal: { group: 'virustotal' }
};
const checkMetrics = (tab, subtab) => {
metricService.destroyWatchers();
if(subtab === 'panels'){
switch (tab) {
case 'audit':
metricService.createWatchers(metricsAudit);
break;
case 'vuls':
metricService.createWatchers(metricsVulnerability);
break;
case 'oscap':
metricService.createWatchers(metricsScap);
break;
case 'virustotal':
metricService.createWatchers(metricsVirustotal);
break;
}
}
if(!$rootScope.$$phase) $rootScope.$digest();
}
// Switch subtab
$scope.switchSubtab = subtab => {
if($scope.tabView === subtab) return;
if(subtab === 'panels' && $scope.tab !== 'configuration'){
if(!$rootScope.visTimestamp) {
$rootScope.visTimestamp = new Date().getTime();
if(!$rootScope.$$phase) $rootScope.$digest();
}
// Create current tab visualizations
genericReq.request('GET',`/api/wazuh-elastic/create-vis/agents-${$scope.tab}/${$rootScope.visTimestamp}/${appState.getCurrentPattern()}`)
.then(() => {
// Render visualizations
$rootScope.$broadcast('updateVis');
checkMetrics($scope.tab, 'panels');
})
.catch(error => errorHandler.handle(error, 'Agents'));
} else {
checkMetrics($scope.tab, subtab);
}
}
// Switch tab
$scope.switchTab = tab => {
if ($scope.tab === tab) return;
if(!$rootScope.visTimestamp) {
$rootScope.visTimestamp = new Date().getTime();
if(!$rootScope.$$phase) $rootScope.$digest();
}
if(tab !== 'configuration') {
// Create current tab visualizations
genericReq.request('GET',`/api/wazuh-elastic/create-vis/agents-${tab}/${$rootScope.visTimestamp}/${appState.getCurrentPattern()}`)
.then(() => {
// Render visualizations
$rootScope.$broadcast('updateVis');
checkMetrics(tab, 'panels');
// Deleting app state traces in the url
$location.search('_a', null);
})
.catch(error => errorHandler.handle(error, 'Agents'));
}
};
// Watchers
// We watch the resultState provided by the discover
$scope.$watch('tabView', () => {
$location.search('tabView', $scope.tabView);
if ($rootScope.ownHandlers) {
for (let h of $rootScope.ownHandlers) {
h._scope.$destroy();
}
}
$rootScope.ownHandlers = [];
$rootScope.loadedVisualizations = [];
});
$scope.$watch('tab', () => {
$location.search('tab', $scope.tab);
$scope.tabView = 'panels';
if ($rootScope.ownHandlers) {
for (let h of $rootScope.ownHandlers) {
h._scope.$destroy();
}
}
$rootScope.ownHandlers = [];
$rootScope.loadedVisualizations = [];
// Update the implicit filter
if (typeof tabFilters[$scope.tab] !== 'undefined' && tabFilters[$scope.tab].group === "") $rootScope.currentImplicitFilter = "";
else $rootScope.currentImplicitFilter = (typeof tabFilters[$scope.tab] !== 'undefined') ? tabFilters[$scope.tab].group : '';
if($scope.tab === 'configuration'){
firstLoad();
}
});
// Agent data
$scope.getAgentStatusClass = (agentStatus) => agentStatus === "Active" ? "teal" : "red";
$scope.formatAgentStatus = (agentStatus) => {
return ['Active','Disconnected'].includes(agentStatus) ? agentStatus : 'Never connected';
};
const calculateMinutes = (start,end) => {
let time = new Date(start);
let endTime = new Date(end);
let minutes = ((endTime - time) / 1000) / 60;
return minutes;
}
const validateRootCheck = () => {
$scope.agent.rootcheck.duration = 'Unknown';
if ($scope.agent.rootcheck.end && $scope.agent.rootcheck.start) {
$scope.agent.rootcheck.duration = ((new Date($scope.agent.rootcheck.end) - new Date($scope.agent.rootcheck.start))/1000)/60;
$scope.agent.rootcheck.duration = Math.round($scope.agent.rootcheck.duration * 100) / 100;
if($scope.agent.rootcheck.duration <= 0){
$scope.agent.rootcheck.inProgress = true;
}
} else {
if (!$scope.agent.rootcheck.end) {
$scope.agent.rootcheck.end = 'Unknown';
}
if (!$scope.agent.rootcheck.start){
$scope.agent.rootcheck.start = 'Unknown';
}
}
}
const validateSysCheck = () => {
$scope.agent.syscheck.duration = 'Unknown';
if ($scope.agent.syscheck.end && $scope.agent.syscheck.start) {
$scope.agent.syscheck.duration = ((new Date($scope.agent.syscheck.end) - new Date($scope.agent.syscheck.start))/1000)/60;
$scope.agent.syscheck.duration = Math.round($scope.agent.syscheck.duration * 100) / 100;
if($scope.agent.syscheck.duration <= 0){
$scope.agent.syscheck.inProgress = true;
}
} else {
if (!$scope.agent.syscheck.end) {
$scope.agent.syscheck.end = 'Unknown';
}
if (!$scope.agent.syscheck.start){
$scope.agent.syscheck.start = 'Unknown';
}
}
}
/** Prevents from double agent and come from autocomplete */
let lastAgent = null;
const checkDouble = id => {
if(lastAgent && lastAgent !== id){
$rootScope.agentsAutoCompleteFired = true;
if(!$rootScope.$$phase) $rootScope.$digest();
}
}
$scope.getAgent = async (newAgentId,fromAutocomplete) => {
try {
if($scope.tab === 'configuration'){
return $scope.getAgentConfig(newAgentId);
}
let id = null;
// They passed an id
if (newAgentId) {
id = newAgentId;
checkDouble(id);
$location.search('agent', id);
} else {
if ($location.search().agent && !$rootScope.globalAgent) { // There's one in the url
id = $location.search().agent;
checkDouble(id);
} else { // We pick the one in the rootScope
id = $rootScope.globalAgent;
checkDouble(id);
$location.search('agent', id);
delete $rootScope.globalAgent;
}
}
if (id === '000' && $scope.tab === 'configuration') {
$scope.tab = 'general';
$scope.switchTab('general');
}
const data = await Promise.all([
apiReq.request('GET', `/agents/${id}`, {}),
apiReq.request('GET', `/syscheck/${id}/last_scan`, {}),
apiReq.request('GET', `/rootcheck/${id}/last_scan`, {})
]);
// Agent
$scope.agent = data[0].data.data;
lastAgent = data[0].data.data.id;
if ($scope.agent.os) {
$scope.agentOS = $scope.agent.os.name + ' ' + $scope.agent.os.version;
}
else { $scope.agentOS = 'Unkwnown' };
// Syscheck
$scope.agent.syscheck = data[1].data.data;
validateSysCheck();
// Rootcheck
$scope.agent.rootcheck = data[2].data.data;
validateRootCheck();
if(!$scope.$$phase) $scope.$digest();
return;
} catch (error) {
errorHandler.handle(error,'Agents');
if(!$rootScope.$$phase) $rootScope.$digest();
}
};
$scope.goGroups = agent => {
$rootScope.globalAgent = agent;
$scope.agentsAutoComplete.reset();
if($rootScope.ownHandlers) {
for(let h of $rootScope.ownHandlers){
h._scope.$destroy();
}
}
if(metricService.hasItems()) metricService.destroyWatchers();
$rootScope.ownHandlers = [];
$rootScope.comeFrom = 'agents';
$location.search('_a',null);
$location.search('tab', 'groups');
$location.path('/manager');
};
$scope.analizeAgents = async search => {
try {
$scope.agentsAutoComplete.filters = [];
await $scope.agentsAutoComplete.addFilter('search',search);
if(!$scope.$$phase) $scope.$digest();
return $scope.agentsAutoComplete.items;
} catch (error) {
errorHandler.handle(error,'Agents');
if(!$rootScope.$$phase) $rootScope.$digest();
}
}
//Load
try {
if($scope.tab !== 'configuration'){
$scope.getAgent();
}
$scope.agentsAutoComplete.nextPage('');
} catch (e) {
errorHandler.handle('Unexpected exception loading controller','Agents');
if(!$rootScope.$$phase) $rootScope.$digest();
}
//Destroy
$scope.$on("$destroy", () => {
$scope.agentsAutoComplete.reset();
if($rootScope.ownHandlers) {
for(let h of $rootScope.ownHandlers){
h._scope.$destroy();
}
}
if(metricService.hasItems()) metricService.destroyWatchers();
$rootScope.ownHandlers = [];
});
//PCI tab
let tabs = [];
genericReq.request('GET', '/api/wazuh-api/pci/all')
.then((data) => {
for(let key in data.data){
tabs.push({
"title": key,
"content": data.data[key]
});
}
})
.catch(error => {
errorHandler.handle(error,'Agents');
if(!$rootScope.$$phase) $rootScope.$digest();
});
$scope.tabs = tabs;
$scope.selectedIndex = 0;
$scope.isArray = angular.isArray;
const getAgent = newAgentId => {
// They passed an id
if (newAgentId) {
$location.search('agent', newAgentId);
}
};
$scope.getAgentConfig = newAgentId => {
getAgent(newAgentId);
firstLoad();
}
$scope.goGroup = () => {
$rootScope.globalAgent = $scope.agent;
$rootScope.comeFrom = 'agents';
$location.search('tab', 'groups');
$location.path('/manager');
};
const firstLoad = async () => {
try{
$scope.configurationError = false;
$scope.load = true;
let id;
if ($location.search().agent && !$rootScope.globalAgent) { // There's one in the url
id = $location.search().agent;
} else { // We pick the one in the rootScope
id = $rootScope.globalAgent;
$location.search('agent', id);
delete $rootScope.globalAgent;
}
let data = await apiReq.request('GET', `/agents/${id}`, {});
$scope.agent = data.data.data;
$scope.groupName = $scope.agent.group;
if(!$scope.groupName){
$scope.configurationError = true;
$scope.load = false;
if($scope.agent.id === '000'){
$scope.configurationError = false;
$scope.tab = 'general';
$scope.switchTab('general');
}
if(!$scope.$$phase) $scope.$digest();
return;
}
data = await apiReq.request('GET', `/agents/groups/${$scope.groupName}/configuration`, {});
$scope.groupConfiguration = data.data.data.items[0];
$scope.rawJSON = beautifier.prettyPrint(data.data.data.items);
data = await Promise.all([
apiReq.request('GET', `/agents/groups?search=${$scope.groupName}`, {}),
apiReq.request('GET', `/agents/groups/${$scope.groupName}`, {})
]);
let filtered = data[0].data.data.items.filter(item => item.name === $scope.groupName);
$scope.groupMergedSum = (filtered.length) ? filtered[0].merged_sum : 'Unknown';
filtered = data[1].data.data.items.filter(item => item.id === $scope.agent.id);
$scope.agentMergedSum = (filtered.length) ? filtered[0].merged_sum : 'Unknown';
$scope.isSynchronized = (($scope.agentMergedSum === $scope.groupMergedSum) && !([$scope.agentMergedSum,$scope.groupMergedSum].includes('Unknown')) ) ? true : false;
$scope.load = false;
if(!$scope.$$phase) $scope.$digest();
return;
} catch (error){
errorHandler.handle(error,'Agents');
if(!$rootScope.$$phase) $rootScope.$digest();
}
}
/** End of agent configuration */
if($scope.tab !== 'configuration'){
// Create visualizations for controller's first execution
genericReq.request('GET',`/api/wazuh-elastic/create-vis/agents-${$scope.tab}/${$rootScope.visTimestamp}/${appState.getCurrentPattern()}`)
.then(() => {
// Render visualizations
$rootScope.$broadcast('updateVis');
checkMetrics($scope.tab,'panels');
})
.catch(error => errorHandler.handle(error, 'Agents'));
}
});