mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-09-18 15:57:44 +02:00
107 lines
4.3 KiB
Python
107 lines
4.3 KiB
Python
from targetApp.models import Domain
|
|
from startScan.models import ScanHistory, WayBackEndPoint, ScannedHost, VulnerabilityScan
|
|
|
|
from django.shortcuts import render, redirect
|
|
from django.http import HttpResponse
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.dispatch import receiver
|
|
from django.contrib.auth.signals import user_logged_out, user_logged_in
|
|
from django.contrib import messages
|
|
from django.contrib.auth import update_session_auth_hash
|
|
from django.contrib.auth.forms import PasswordChangeForm
|
|
from django.db.models import Count, Value, CharField, Q
|
|
|
|
|
|
def index(request):
|
|
domain_count = Domain.objects.all().count()
|
|
endpoint_count = WayBackEndPoint.objects.all().count()
|
|
scan_count = ScanHistory.objects.all().count()
|
|
subdomain_count = ScannedHost.objects.all().count()
|
|
alive_count = \
|
|
ScannedHost.objects.all().exclude(http_status__exact=0).count()
|
|
endpoint_alive_count = \
|
|
WayBackEndPoint.objects.filter(http_status__exact=200).count()
|
|
on_going_scan_count = \
|
|
ScanHistory.objects.filter(scan_status=1).count()
|
|
recent_completed_scans = ScanHistory.objects.all().order_by(
|
|
'-last_scan_date').filter(Q(scan_status=0) | Q(scan_status=2) | Q(scan_status=3))[:4]
|
|
currently_scanning = ScanHistory.objects.filter(scan_status=1)[:4]
|
|
pending_scans = ScanHistory.objects.filter(scan_status=-1)[:4]
|
|
info_count = VulnerabilityScan.objects.filter(severity=0).count()
|
|
low_count = VulnerabilityScan.objects.filter(severity=1).count()
|
|
medium_count = VulnerabilityScan.objects.filter(severity=2).count()
|
|
high_count = VulnerabilityScan.objects.filter(severity=3).count()
|
|
critical_count = VulnerabilityScan.objects.filter(severity=4).count()
|
|
total_vul_count = info_count + low_count + \
|
|
medium_count + high_count + critical_count
|
|
# most_vulnerable_target = Domain.objects.exclude(
|
|
# scanhistory__scannedhost__vulnerabilityscan__severity=0).annotate(
|
|
# num_vul=Count(
|
|
# 'scanhistory__scannedhost__vulnerabilityscan__name',
|
|
# distinct=True)).order_by('-num_vul')[
|
|
# :5]
|
|
most_vulnerable_target = Domain.objects.annotate(num_vul=Count(
|
|
'scanhistory__scannedhost__vulnerabilityscan__name')).order_by('-num_vul')[:7]
|
|
most_common_vulnerability = VulnerabilityScan.objects.values("name", "severity").exclude(
|
|
severity=0).annotate(count=Count('name')).order_by("-count")[:7]
|
|
context = {
|
|
'dashboard_data_active': 'true',
|
|
'domain_count': domain_count,
|
|
'endpoint_count': endpoint_count,
|
|
'scan_count': scan_count,
|
|
'subdomain_count': subdomain_count,
|
|
'alive_count': alive_count,
|
|
'endpoint_alive_count': endpoint_alive_count,
|
|
'on_going_scan_count': on_going_scan_count,
|
|
'recent_completed_scans': recent_completed_scans,
|
|
'pending_scans': pending_scans,
|
|
'currently_scanning': currently_scanning,
|
|
'info_count': info_count,
|
|
'low_count': low_count,
|
|
'medium_count': medium_count,
|
|
'high_count': high_count,
|
|
'critical_count': critical_count,
|
|
'most_vulnerable_target': most_vulnerable_target,
|
|
'most_common_vulnerability': most_common_vulnerability,
|
|
'total_vul_count': total_vul_count,
|
|
}
|
|
return render(request, 'dashboard/index.html', context)
|
|
|
|
|
|
def profile(request):
|
|
if request.method == 'POST':
|
|
form = PasswordChangeForm(request.user, request.POST)
|
|
if form.is_valid():
|
|
user = form.save()
|
|
update_session_auth_hash(request, user)
|
|
messages.success(
|
|
request,
|
|
'Your password was successfully changed!')
|
|
return redirect('profile')
|
|
else:
|
|
messages.error(request, 'Please correct the error below.')
|
|
else:
|
|
form = PasswordChangeForm(request.user)
|
|
return render(request, 'dashboard/profile.html', {
|
|
'form': form
|
|
})
|
|
|
|
|
|
@receiver(user_logged_out)
|
|
def on_user_logged_out(sender, request, **kwargs):
|
|
messages.add_message(
|
|
request,
|
|
messages.INFO,
|
|
'You have been successfully logged out. Thank you ' +
|
|
'for using reNgine.')
|
|
|
|
|
|
@receiver(user_logged_in)
|
|
def on_user_logged_in(sender, request, **kwargs):
|
|
messages.add_message(
|
|
request,
|
|
messages.INFO,
|
|
'Hi @' +
|
|
request.user.username +
|
|
' welcome back!')
|