482 lines
17 KiB
JavaScript
482 lines
17 KiB
JavaScript
// DNS Reconnaissance Tool - Enhanced Frontend JavaScript with Real-time Updates
|
|
|
|
class ReconTool {
|
|
constructor() {
|
|
this.currentScanId = null;
|
|
this.pollInterval = null;
|
|
this.liveDataInterval = null;
|
|
this.currentReport = null;
|
|
this.init();
|
|
}
|
|
|
|
init() {
|
|
this.bindEvents();
|
|
this.setupRealtimeElements();
|
|
}
|
|
|
|
setupRealtimeElements() {
|
|
// Create live discovery container if it doesn't exist
|
|
if (!document.getElementById('liveDiscoveries')) {
|
|
const progressSection = document.getElementById('progressSection');
|
|
const liveDiv = document.createElement('div');
|
|
liveDiv.id = 'liveDiscoveries';
|
|
liveDiv.innerHTML = `
|
|
<div class="live-discoveries" style="display: none;">
|
|
<h3>🔍 Live Discoveries</h3>
|
|
<div class="stats-grid">
|
|
<div class="stat-item">
|
|
<span class="stat-label">Hostnames:</span>
|
|
<span id="liveHostnames" class="stat-value">0</span>
|
|
</div>
|
|
<div class="stat-item">
|
|
<span class="stat-label">IP Addresses:</span>
|
|
<span id="liveIPs" class="stat-value">0</span>
|
|
</div>
|
|
<div class="stat-item">
|
|
<span class="stat-label">DNS Records:</span>
|
|
<span id="liveDNS" class="stat-value">0</span>
|
|
</div>
|
|
<div class="stat-item">
|
|
<span class="stat-label">Certificates:</span>
|
|
<span id="liveCerts" class="stat-value">0</span>
|
|
</div>
|
|
<div class="stat-item">
|
|
<span class="stat-label">Shodan Results:</span>
|
|
<span id="liveShodan" class="stat-value">0</span>
|
|
</div>
|
|
<div class="stat-item">
|
|
<span class="stat-label">VirusTotal:</span>
|
|
<span id="liveVT" class="stat-value">0</span>
|
|
</div>
|
|
</div>
|
|
<div class="discoveries-list">
|
|
<h4>📋 Recent Discoveries</h4>
|
|
<div id="recentHostnames" class="discovery-section">
|
|
<strong>Hostnames:</strong>
|
|
<div class="hostname-list"></div>
|
|
</div>
|
|
<div id="recentIPs" class="discovery-section">
|
|
<strong>IP Addresses:</strong>
|
|
<div class="ip-list"></div>
|
|
</div>
|
|
<div id="activityLog" class="discovery-section">
|
|
<strong>Activity Log:</strong>
|
|
<div class="activity-list"></div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
`;
|
|
progressSection.appendChild(liveDiv);
|
|
}
|
|
}
|
|
|
|
bindEvents() {
|
|
// Start scan button
|
|
document.getElementById('startScan').addEventListener('click', () => {
|
|
this.startScan();
|
|
});
|
|
|
|
// New scan button
|
|
document.getElementById('newScan').addEventListener('click', () => {
|
|
this.resetToForm();
|
|
});
|
|
|
|
// Report view toggles
|
|
document.getElementById('showJson').addEventListener('click', () => {
|
|
this.showReport('json');
|
|
});
|
|
|
|
document.getElementById('showText').addEventListener('click', () => {
|
|
this.showReport('text');
|
|
});
|
|
|
|
// Download buttons
|
|
document.getElementById('downloadJson').addEventListener('click', () => {
|
|
this.downloadReport('json');
|
|
});
|
|
|
|
document.getElementById('downloadText').addEventListener('click', () => {
|
|
this.downloadReport('text');
|
|
});
|
|
|
|
// Enter key in target field
|
|
document.getElementById('target').addEventListener('keypress', (e) => {
|
|
if (e.key === 'Enter') {
|
|
this.startScan();
|
|
}
|
|
});
|
|
}
|
|
|
|
async startScan() {
|
|
const target = document.getElementById('target').value.trim();
|
|
|
|
if (!target) {
|
|
alert('Please enter a target domain or hostname');
|
|
return;
|
|
}
|
|
|
|
const scanData = {
|
|
target: target,
|
|
max_depth: parseInt(document.getElementById('maxDepth').value),
|
|
shodan_key: document.getElementById('shodanKey').value.trim() || null,
|
|
virustotal_key: document.getElementById('virustotalKey').value.trim() || null
|
|
};
|
|
|
|
try {
|
|
// Show progress section
|
|
this.showProgressSection();
|
|
this.updateProgress(0, 'Starting scan...');
|
|
|
|
console.log('🚀 Starting scan with data:', scanData);
|
|
|
|
const response = await fetch('/api/scan', {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
},
|
|
body: JSON.stringify(scanData)
|
|
});
|
|
|
|
if (!response.ok) {
|
|
throw new Error(`HTTP error! status: ${response.status}`);
|
|
}
|
|
|
|
const result = await response.json();
|
|
|
|
if (result.error) {
|
|
throw new Error(result.error);
|
|
}
|
|
|
|
this.currentScanId = result.scan_id;
|
|
console.log('✅ Scan started with ID:', this.currentScanId);
|
|
|
|
this.startPolling();
|
|
this.startLiveDataPolling();
|
|
|
|
} catch (error) {
|
|
console.error('❌ Failed to start scan:', error);
|
|
this.showError(`Failed to start scan: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
startPolling() {
|
|
// Poll every 2 seconds for status updates
|
|
this.pollInterval = setInterval(() => {
|
|
this.checkScanStatus();
|
|
}, 2000);
|
|
|
|
// Also check immediately
|
|
this.checkScanStatus();
|
|
}
|
|
|
|
startLiveDataPolling() {
|
|
// Poll every 3 seconds for live data updates
|
|
this.liveDataInterval = setInterval(() => {
|
|
this.updateLiveData();
|
|
}, 3000);
|
|
|
|
// Show the live discoveries section
|
|
const liveSection = document.querySelector('.live-discoveries');
|
|
if (liveSection) {
|
|
liveSection.style.display = 'block';
|
|
}
|
|
|
|
// Also update immediately
|
|
this.updateLiveData();
|
|
}
|
|
|
|
async checkScanStatus() {
|
|
if (!this.currentScanId) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const response = await fetch(`/api/scan/${this.currentScanId}/status`);
|
|
|
|
if (!response.ok) {
|
|
throw new Error(`HTTP error! status: ${response.status}`);
|
|
}
|
|
|
|
const status = await response.json();
|
|
|
|
if (status.error) {
|
|
throw new Error(status.error);
|
|
}
|
|
|
|
// Update progress
|
|
this.updateProgress(status.progress, status.message);
|
|
|
|
// Update live stats
|
|
if (status.live_stats) {
|
|
this.updateLiveStats(status.live_stats);
|
|
}
|
|
|
|
// Check if completed
|
|
if (status.status === 'completed') {
|
|
console.log('✅ Scan completed');
|
|
this.stopPolling();
|
|
await this.loadScanReport();
|
|
} else if (status.status === 'error') {
|
|
this.stopPolling();
|
|
throw new Error(status.error || 'Scan failed');
|
|
}
|
|
|
|
} catch (error) {
|
|
console.error('❌ Error checking scan status:', error);
|
|
this.stopPolling();
|
|
this.showError(`Error checking scan status: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
async updateLiveData() {
|
|
if (!this.currentScanId) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const response = await fetch(`/api/scan/${this.currentScanId}/live-data`);
|
|
|
|
if (!response.ok) {
|
|
return; // Silently fail for live data
|
|
}
|
|
|
|
const data = await response.json();
|
|
|
|
if (data.error) {
|
|
return; // Silently fail for live data
|
|
}
|
|
|
|
// Update live discoveries
|
|
this.updateLiveDiscoveries(data);
|
|
|
|
} catch (error) {
|
|
// Silently fail for live data updates
|
|
console.debug('Live data update failed:', error);
|
|
}
|
|
}
|
|
|
|
updateLiveStats(stats) {
|
|
// Update the live statistics counters
|
|
const statElements = {
|
|
'liveHostnames': stats.hostnames || 0,
|
|
'liveIPs': stats.ip_addresses || 0,
|
|
'liveDNS': stats.dns_records || 0,
|
|
'liveCerts': stats.certificates || 0,
|
|
'liveShodan': stats.shodan_results || 0,
|
|
'liveVT': stats.virustotal_results || 0
|
|
};
|
|
|
|
Object.entries(statElements).forEach(([elementId, value]) => {
|
|
const element = document.getElementById(elementId);
|
|
if (element) {
|
|
element.textContent = value;
|
|
|
|
// Add a brief highlight effect when value changes
|
|
if (element.textContent !== value.toString()) {
|
|
element.style.backgroundColor = '#ff9900';
|
|
setTimeout(() => {
|
|
element.style.backgroundColor = '';
|
|
}, 1000);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
updateLiveDiscoveries(data) {
|
|
// Update hostnames list
|
|
const hostnameList = document.querySelector('#recentHostnames .hostname-list');
|
|
if (hostnameList && data.hostnames) {
|
|
// Show last 10 hostnames
|
|
const recentHostnames = data.hostnames.slice(-10);
|
|
hostnameList.innerHTML = recentHostnames.map(hostname =>
|
|
`<span class="discovery-item">${hostname}</span>`
|
|
).join('');
|
|
}
|
|
|
|
// Update IP addresses list
|
|
const ipList = document.querySelector('#recentIPs .ip-list');
|
|
if (ipList && data.ip_addresses) {
|
|
// Show last 10 IPs
|
|
const recentIPs = data.ip_addresses.slice(-10);
|
|
ipList.innerHTML = recentIPs.map(ip =>
|
|
`<span class="discovery-item">${ip}</span>`
|
|
).join('');
|
|
}
|
|
|
|
// Update activity log
|
|
const activityList = document.querySelector('#activityLog .activity-list');
|
|
if (activityList && data.latest_discoveries) {
|
|
const activities = data.latest_discoveries.slice(-5); // Last 5 activities
|
|
activityList.innerHTML = activities.map(activity => {
|
|
const time = new Date(activity.timestamp * 1000).toLocaleTimeString();
|
|
return `<div class="activity-item">[${time}] ${activity.message}</div>`;
|
|
}).join('');
|
|
}
|
|
}
|
|
|
|
async loadScanReport() {
|
|
try {
|
|
console.log('📄 Loading scan report...');
|
|
const response = await fetch(`/api/scan/${this.currentScanId}/report`);
|
|
|
|
if (!response.ok) {
|
|
throw new Error(`HTTP error! status: ${response.status}`);
|
|
}
|
|
|
|
const report = await response.json();
|
|
|
|
if (report.error) {
|
|
throw new Error(report.error);
|
|
}
|
|
|
|
this.currentReport = report;
|
|
console.log('✅ Report loaded successfully');
|
|
this.showResultsSection();
|
|
this.showReport('text'); // Default to text view
|
|
|
|
} catch (error) {
|
|
console.error('❌ Error loading report:', error);
|
|
this.showError(`Error loading report: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
stopPolling() {
|
|
if (this.pollInterval) {
|
|
clearInterval(this.pollInterval);
|
|
this.pollInterval = null;
|
|
}
|
|
if (this.liveDataInterval) {
|
|
clearInterval(this.liveDataInterval);
|
|
this.liveDataInterval = null;
|
|
}
|
|
}
|
|
|
|
showProgressSection() {
|
|
document.getElementById('scanForm').style.display = 'none';
|
|
document.getElementById('progressSection').style.display = 'block';
|
|
document.getElementById('resultsSection').style.display = 'none';
|
|
}
|
|
|
|
showResultsSection() {
|
|
document.getElementById('scanForm').style.display = 'none';
|
|
document.getElementById('progressSection').style.display = 'none';
|
|
document.getElementById('resultsSection').style.display = 'block';
|
|
|
|
// Hide live discoveries in results section
|
|
const liveSection = document.querySelector('.live-discoveries');
|
|
if (liveSection) {
|
|
liveSection.style.display = 'none';
|
|
}
|
|
}
|
|
|
|
resetToForm() {
|
|
this.stopPolling();
|
|
this.currentScanId = null;
|
|
this.currentReport = null;
|
|
|
|
document.getElementById('scanForm').style.display = 'block';
|
|
document.getElementById('progressSection').style.display = 'none';
|
|
document.getElementById('resultsSection').style.display = 'none';
|
|
|
|
// Hide live discoveries
|
|
const liveSection = document.querySelector('.live-discoveries');
|
|
if (liveSection) {
|
|
liveSection.style.display = 'none';
|
|
}
|
|
|
|
// Clear form
|
|
document.getElementById('target').value = '';
|
|
document.getElementById('shodanKey').value = '';
|
|
document.getElementById('virustotalKey').value = '';
|
|
document.getElementById('maxDepth').value = '2';
|
|
}
|
|
|
|
updateProgress(percentage, message) {
|
|
const progressFill = document.getElementById('progressFill');
|
|
const progressMessage = document.getElementById('progressMessage');
|
|
|
|
progressFill.style.width = `${percentage || 0}%`;
|
|
progressMessage.textContent = message || 'Processing...';
|
|
}
|
|
|
|
showError(message) {
|
|
// Update progress section to show error
|
|
this.updateProgress(0, `Error: ${message}`);
|
|
|
|
// Also alert the user
|
|
alert(`Error: ${message}`);
|
|
}
|
|
|
|
showReport(type) {
|
|
if (!this.currentReport) {
|
|
return;
|
|
}
|
|
|
|
const reportContent = document.getElementById('reportContent');
|
|
const showJsonBtn = document.getElementById('showJson');
|
|
const showTextBtn = document.getElementById('showText');
|
|
|
|
if (type === 'json') {
|
|
// Show JSON report
|
|
try {
|
|
// The json_report should already be a string from the server
|
|
let jsonData;
|
|
if (typeof this.currentReport.json_report === 'string') {
|
|
jsonData = JSON.parse(this.currentReport.json_report);
|
|
} else {
|
|
jsonData = this.currentReport.json_report;
|
|
}
|
|
reportContent.textContent = JSON.stringify(jsonData, null, 2);
|
|
} catch (e) {
|
|
console.error('Error parsing JSON report:', e);
|
|
reportContent.textContent = this.currentReport.json_report;
|
|
}
|
|
|
|
showJsonBtn.classList.add('active');
|
|
showTextBtn.classList.remove('active');
|
|
} else {
|
|
// Show text report
|
|
reportContent.textContent = this.currentReport.text_report;
|
|
|
|
showTextBtn.classList.add('active');
|
|
showJsonBtn.classList.remove('active');
|
|
}
|
|
}
|
|
|
|
downloadReport(type) {
|
|
if (!this.currentReport) {
|
|
return;
|
|
}
|
|
|
|
let content, filename, mimeType;
|
|
|
|
if (type === 'json') {
|
|
content = typeof this.currentReport.json_report === 'string'
|
|
? this.currentReport.json_report
|
|
: JSON.stringify(this.currentReport.json_report, null, 2);
|
|
filename = `recon-report-${this.currentScanId}.json`;
|
|
mimeType = 'application/json';
|
|
} else {
|
|
content = this.currentReport.text_report;
|
|
filename = `recon-report-${this.currentScanId}.txt`;
|
|
mimeType = 'text/plain';
|
|
}
|
|
|
|
// Create download link
|
|
const blob = new Blob([content], { type: mimeType });
|
|
const url = window.URL.createObjectURL(blob);
|
|
const a = document.createElement('a');
|
|
a.href = url;
|
|
a.download = filename;
|
|
document.body.appendChild(a);
|
|
a.click();
|
|
window.URL.revokeObjectURL(url);
|
|
document.body.removeChild(a);
|
|
}
|
|
}
|
|
|
|
// Initialize the application when DOM is loaded
|
|
document.addEventListener('DOMContentLoaded', () => {
|
|
console.log('🌐 DNS Reconnaissance Tool initialized');
|
|
new ReconTool();
|
|
}); |