wazuh-kibana-app/public/controllers/overview.js
2018-06-07 18:37:18 +02:00

408 lines
15 KiB
JavaScript

/*
* Wazuh app - Overview controller
* Copyright (C) 2018 Wazuh, Inc.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* Find more information about this on the LICENSE file.
*/
import $ from 'jquery';
import * as modules from 'ui/modules'
import FilterHandler from './filter-handler'
const app = modules.get('app/wazuh', []);
app.controller('overviewController', function ($sce, $timeout, $scope, $location, $rootScope, appState, genericReq, errorHandler, apiReq, rawVisualizations, loadedVisualizations, tabVisualizations, discoverPendingUpdates, visHandlers, vis2png) {
$location.search('_a',null)
const filterHandler = new FilterHandler(appState.getCurrentPattern());
discoverPendingUpdates.removeAll();
rawVisualizations.removeAll();
tabVisualizations.removeAll();
loadedVisualizations.removeAll();
const currentApi = JSON.parse(appState.getCurrentAPI()).id;
const extensions = appState.getExtensions(currentApi);
$scope.extensions = extensions;
$scope.wzMonitoringEnabled = false;
// Tab names
$scope.tabNames = {
welcome : 'Welcome',
general : 'General',
fim : 'File integrity',
pm : 'Policy monitoring',
vuls : 'Vulnerabilities',
oscap : 'Open SCAP',
audit : 'Audit',
pci : 'PCI DSS',
gdpr : 'GDPR',
aws : 'AWS',
virustotal: 'VirusTotal'
}
// Metrics General
const metricsGeneral = {
totalAlerts: '[vis-id="\'Wazuh-App-Overview-General-Metric-alerts\'"]',
level12 : '[vis-id="\'Wazuh-App-Overview-General-Level-12-alerts\'"]',
authFailure: '[vis-id="\'Wazuh-App-Overview-General-Authentication-failure\'"]',
authSuccess: '[vis-id="\'Wazuh-App-Overview-General-Authentication-success\'"]'
}
// Metrics FIM
const metricsFim = {
fimAdded : '[vis-id="\'Wazuh-App-Overview-FIM-Added\'"]',
fimModified: '[vis-id="\'Wazuh-App-Overview-FIM-Modified\'"]',
fimDeleted : '[vis-id="\'Wazuh-App-Overview-FIM-Deleted\'"]'
}
// Metrics Audit
const metricsAudit = {
auditNewFiles : '[vis-id="\'Wazuh-App-Overview-Audit-New-files\'"]',
auditReadFiles : '[vis-id="\'Wazuh-App-Overview-Audit-Read-files\'"]',
auditModifiedFiles: '[vis-id="\'Wazuh-App-Overview-Audit-Modified-files\'"]',
auditRemovedFiles : '[vis-id="\'Wazuh-App-Overview-Audit-Removed-files\'"]'
}
// Metrics Vulnerability Detector
const metricsVulnerability = {
vulnCritical: '[vis-id="\'Wazuh-App-Overview-VULS-Metric-Critical-severity\'"]',
vulnHigh : '[vis-id="\'Wazuh-App-Overview-VULS-Metric-High-severity\'"]',
vulnMedium : '[vis-id="\'Wazuh-App-Overview-VULS-Metric-Medium-severity\'"]',
vulnLow : '[vis-id="\'Wazuh-App-Overview-VULS-Metric-Low-severity\'"]'
}
// Metrics Scap
const metricsScap = {
scapLastScore : '[vis-id="\'Wazuh-App-Overview-OSCAP-Last-score\'"]',
scapHighestScore: '[vis-id="\'Wazuh-App-Overview-OSCAP-Highest-score\'"]',
scapLowestScore : '[vis-id="\'Wazuh-App-Overview-OSCAP-Lowest-score\'"]'
}
// Metrics Virustotal
const metricsVirustotal = {
virusMalicious: '[vis-id="\'Wazuh-App-Overview-Virustotal-Total-Malicious\'"]',
virusPositives: '[vis-id="\'Wazuh-App-Overview-Virustotal-Total-Positives\'"]',
virusTotal : '[vis-id="\'Wazuh-App-Overview-Virustotal-Total\'"]'
}
// Metrics AWS
const metricsAws = {
awsLogins :'[vis-id="\'Wazuh-App-Overview-AWS-Metric-Successful-logins\'"]',
awsMostActiveUser:'[vis-id="\'Wazuh-App-Overview-AWS-Most-active-user\'"]',
awsAuthorized :'[vis-id="\'Wazuh-App-Overview-AWS-Metric-Authorize-security\'"]',
awsRevoked :'[vis-id="\'Wazuh-App-Overview-AWS-Metric-Revoke-security\'"]'
}
let tabHistory = [];
// Check the url hash and retrieve 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');
}
if($scope.tab !== 'welcome') tabHistory.push($scope.tab);
// Check the url hash and retrieve tab information
if ($location.search().tab) {
$scope.tab = $location.search().tab;
} else { // If tab doesn't exist, default it to 'welcome'
$scope.tab = 'welcome';
$location.search('tab', 'welcome');
}
// This object represents the number of visualizations per tab; used to show a progress bar
tabVisualizations.assign({
welcome : 0,
general : 11,
fim : 10,
pm : 5,
vuls : 8,
oscap : 14,
audit : 15,
pci : 6,
gdpr : 6,
aws : 10,
virustotal: 7
});
// Object for matching nav items and rules 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' },
gdpr : { group: 'gdpr' },
aws : { group: 'amazon' },
virustotal: { group: 'virustotal' }
};
let filters = []
const assignFilters = (tab, localChange) => {
try{
filters = [];
const isCluster = appState.getClusterInfo().status == 'enabled';
filters.push(filterHandler.managerQuery(
isCluster ?
appState.getClusterInfo().cluster :
appState.getClusterInfo().manager,
isCluster
))
if(tab !== 'general'){
if(tab === 'pci') {
filters.push(filterHandler.pciQuery())
} else if(tab === 'gdpr') {
filters.push(filterHandler.gdprQuery())
} else {
filters.push(filterHandler.ruleGroupQuery(tabFilters[tab].group));
}
}
$rootScope.$emit('wzEventFilters',{filters, localChange});
if(!$rootScope.$$listenerCount['wzEventFilters']){
$timeout(100)
.then(() => assignFilters(tab))
}
} catch(error) {
errorHandler.handle('An error occurred while creating custom filters for visualizations','Overview',true);
}
}
const generateMetric = id => {
let html = $(id).html();
if (typeof html !== 'undefined' && html.includes('<span')) {
if(typeof html.split('<span>')[1] !== 'undefined'){
return html.split('<span>')[1].split('</span')[0];
} else if(html.includes('table') && html.includes('cell-hover')){
let nonB = html.split('ng-non-bindable')[1];
if(nonB &&
nonB.split('>')[1] &&
nonB.split('>')[1].split('</')[0]
) {
return nonB.split('>')[1].split('</')[0];
}
}
}
return '';
}
const createMetrics = metricsObject => {
for(let key in metricsObject) {
$scope[key] = () => generateMetric(metricsObject[key]);
}
}
const checkMetrics = (tab, subtab) => {
if(subtab === 'panels'){
switch (tab) {
case 'general':
createMetrics(metricsGeneral);
break;
case 'fim':
createMetrics(metricsFim);
break;
case 'audit':
createMetrics(metricsAudit);
break;
case 'vuls':
createMetrics(metricsVulnerability);
break;
case 'oscap':
createMetrics(metricsScap);
break;
case 'virustotal':
createMetrics(metricsVirustotal);
break;
case 'aws':
createMetrics(metricsAws);
break;
}
}
}
// Switch subtab
$scope.switchSubtab = (subtab, force = false, sameTab = true, preserveDiscover = false) => {
if ($scope.tabView === subtab && !force) return;
visHandlers.removeAll();
discoverPendingUpdates.removeAll();
rawVisualizations.removeAll();
loadedVisualizations.removeAll();
$location.search('tabView', subtab);
const localChange = ((subtab === 'panels' && $scope.tabView === 'discover') ||
(subtab === 'discover' && $scope.tabView === 'panels')) && sameTab;
if(subtab === 'panels' && $scope.tabView === 'discover' && sameTab){
$rootScope.$emit('changeTabView',{tabView:$scope.tabView})
}
$scope.tabView = subtab;
if(subtab === 'panels' && $scope.tab !== 'welcome'){
// Create current tab visualizations
genericReq.request('GET',`/api/wazuh-elastic/create-vis/overview-${$scope.tab}/${appState.getCurrentPattern()}`)
.then(data => {
rawVisualizations.assignItems(data.data.raw);
assignFilters($scope.tab, localChange || preserveDiscover);
$rootScope.$emit('changeTabView',{tabView:subtab})
$rootScope.$broadcast('updateVis');
checkMetrics($scope.tab, 'panels');
})
.catch(error => errorHandler.handle(error, 'Overview'));
} else {
$rootScope.$emit('changeTabView',{tabView:$scope.tabView})
checkMetrics($scope.tab, subtab);
}
}
// Switch tab
$scope.switchTab = (tab,force = false) => {
if(tab !== 'welcome') tabHistory.push(tab);
if (tabHistory.length > 2) tabHistory = tabHistory.slice(-2);
tabVisualizations.setTab(tab);
if ($scope.tab === tab && !force) return;
const sameTab = $scope.tab === tab;
$location.search('tab', tab);
const preserveDiscover = tabHistory.length === 2 && tabHistory[0] === tabHistory[1];
$scope.tab = tab;
$scope.switchSubtab('panels', true, sameTab, preserveDiscover);
};
$scope.startVis2Png = async () => {
try {
if(vis2png.isWorking()){
errorHandler.handle('Report in progress', 'Reporting',true);
return;
}
$scope.reportBusy = true;
$rootScope.reportStatus = 'Generating report...0%'
if(!$rootScope.$$phase) $rootScope.$digest();
vis2png.clear();
const idArray = rawVisualizations.getList().map(item => {
const tmpHTMLElement = $(`#${item.id}`);
vis2png.assignHTMLItem(item.id,tmpHTMLElement)
return item.id;
});
const appliedFilters = visHandlers.getAppliedFilters();
const tab = $scope.tab;
const array = await vis2png.checkArray(idArray)
const name = `wazuh-overview-${tab}-${Date.now() / 1000 | 0}.pdf`
const data ={
array,
name,
title: `Overview ${tab}`,
filters: appliedFilters.filters,
time: appliedFilters.time,
searchBar: appliedFilters.searchBar,
tab,
section: 'overview'
};
const request = await genericReq.request('POST','/api/wazuh-api/report',data)
$scope.reportBusy = false;
$rootScope.reportStatus = false;
errorHandler.info('Success. Go to Management -> Reporting', 'Reporting')
return;
} catch (error) {
$scope.reportBusy = false;
$rootScope.reportStatus = false;
errorHandler.handle(error, 'Reporting')
}
}
$scope.$on('$destroy', () => {
discoverPendingUpdates.removeAll();
rawVisualizations.removeAll();
tabVisualizations.removeAll();
loadedVisualizations.removeAll();
visHandlers.removeAll();
});
$scope.switchTab($scope.tab,true);
//PCI tab
let pciTabs = [];
genericReq
.request('GET', '/api/wazuh-api/pci/all')
.then(data => {
for (let key in data.data) {
pciTabs.push({
"title": key,
"content": data.data[key]
});
}
})
.catch(error => errorHandler.handle(error, 'Overview'));
$scope.pciTabs = pciTabs;
$scope.selectedPciIndex = 0;
//GDPR tab
let gdprTabs = [];
genericReq
.request('GET', '/api/wazuh-api/gdpr/all')
.then(data => {
for (let key in data.data) {
gdprTabs.push({
"title": key,
"content": data.data[key]
});
}
})
.catch(error => errorHandler.handle(error, 'Overview'));
$scope.gdprTabs = gdprTabs;
$scope.selectedGdprIndex = 0;
genericReq.request('GET', '/api/wazuh-api/configuration', {})
.then(configuration => {
if(configuration && configuration.data && configuration.data.data) {
$scope.wzMonitoringEnabled = typeof configuration.data.data['wazuh.monitoring.enabled'] !== 'undefined' ?
!!configuration.data.data['wazuh.monitoring.enabled'] :
true;
if(!$scope.wzMonitoringEnabled){
apiReq.request('GET', '/agents/summary', { })
.then(data => {
if(data && data.data && data.data.data){
$scope.agentsCountActive = data.data.data.Active;
$scope.agentsCountDisconnected = data.data.data.Disconnected;
$scope.agentsCountNeverConnected = data.data.data['Never connected'];
$scope.agentsCountTotal = data.data.data.Total;
$scope.agentsCoverity = (data.data.data.Active / data.data.data.Total) * 100;
} else {
throw new Error('Error fetching /agents/summary from Wazuh API')
}
})
.catch(error => errorHandler.handle(error, 'Overview - Monitoring'))
}
} else {
$scope.wzMonitoringEnabled = true;
}
})
.catch(error => {
$scope.wzMonitoringEnabled = true
errorHandler.handle(error, 'Overview');
});
});