first commit

This commit is contained in:
Kevin Muñoz 2024-06-24 08:38:51 -05:00
commit aa1317ce11
No known key found for this signature in database
GPG Key ID: 3CA0B9DF1BE7CE09
4 changed files with 419 additions and 0 deletions

9
.gitignore vendored Normal file
View File

@ -0,0 +1,9 @@
/.venv
/.static
/__pycache__
/templates/index.html~
/templates/result.html
/templates/scan.html
/templates/upload.html
app.py~
app2.py

76
app.py Normal file
View File

@ -0,0 +1,76 @@
from flask import Flask, render_template, request, jsonify
from flask_socketio import SocketIO, emit
import subprocess
import uuid
import os
import requests
app = Flask(__name__)
socketio = SocketIO(app)
@app.route('/', methods=['GET', 'POST'])
def index():
return render_template('index.html')
@app.route('/upload', methods=['POST'])
def upload_file():
file = request.files.get('file')
if file:
file_path = os.path.join('/tmp', f"{uuid.uuid4()}_{file.filename}")
file.save(file_path)
# Ejecuta el escaneo en un hilo separado para no bloquear la aplicación
socketio.start_background_task(target=scan_file, file_path=file_path)
return 'Archivo subido exitosamente: ' + file.filename
return 'No se recibió ningún archivo', 400
@app.route('/scan_url', methods=['POST'])
def scan_url():
url = request.json.get('url')
if not url:
return jsonify({'error': 'No URL provided'}), 400
# Añadir esquema si falta
if not url.startswith(('http://', 'https://')):
url = 'http://' + url
try:
# Descargar el contenido de la URL a un archivo temporal
response = requests.get(url)
if response.status_code != 200:
return jsonify({'error': f'Error al descargar la URL: {response.status_code}'}), 400
file_path = os.path.join('/tmp', f"{uuid.uuid4()}_downloaded_content")
with open(file_path, 'wb') as temp_file:
temp_file.write(response.content)
# Ejecutar el escaneo en un hilo separado para no bloquear la aplicación
socketio.start_background_task(target=scan_file, file_path=file_path)
return jsonify({'message': f'Archivo descargado y guardado como {file_path}'}), 200
except Exception as e:
return jsonify({'error': str(e)}), 500
def scan_file(file_path):
try:
scan_command = ["clamscan", "-r", file_path]
process = subprocess.Popen(scan_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
for line in iter(process.stdout.readline, ''):
if line.strip():
socketio.emit('scan_output', {'data': line.strip()})
socketio.emit('scan_output', {'data': '--- Escaneo completado ---'})
process.stdout.close()
process.stderr.close()
process.wait()
except Exception as e:
socketio.emit('scan_output', {'data': f'Error: {str(e)}'})
finally:
if os.path.exists(file_path):
os.remove(file_path)
if __name__ == '__main__':
socketio.run(app, debug=True)

21
requirements.txt Normal file
View File

@ -0,0 +1,21 @@
bidict
blinker
certifi
charset-normalizer
clamd
click
Flask
Flask-SocketIO
h11
idna
itsdangerous
Jinja2
MarkupSafe
pyClamd
python-engineio
python-socketio
requests
simple-websocket
urllib3
Werkzeug
wsproto

313
templates/index.html Normal file
View File

@ -0,0 +1,313 @@
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Condor Business Solutions SecureScan</title>
<script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.4.1/socket.io.js"></script>
<link href="https://stackpath.bootstrapcdn.com/bootstrap/4.5.2/css/bootstrap.min.css" rel="stylesheet">
<script src="https://cdn.jsdelivr.net/npm/particles.js@2.0.0"></script>
<style>
/* General styling */
body {
background-color: #1e1e2f;
color: #dcdcdc;
font-family: 'Roboto', sans-serif;
overflow: hidden; /* Para que no aparezca la barra de scroll */
}
h1 {
color: #00d1b2;
margin-bottom: 30px;
text-align: center;
}
.container {
max-width: 600px;
margin: 40px auto;
padding: 20px;
background: rgba(41, 41, 61, 0.85); /* Fondo semi-transparente */
border-radius: 10px;
box-shadow: 0 0 15px rgba(0, 0, 0, 0.5);
position: relative; /* Necesario para el contenedor */
z-index: 1;
}
.form-control-file, .form-control, .btn {
border-radius: 50px;
margin-bottom: 15px;
}
.btn-primary, .btn-secondary {
width: 100%;
transition: all 0.3s ease;
}
.btn-primary:hover, .btn-secondary:hover {
background-color: #00b89c;
}
.btn-primary {
background-color: #00d1b2;
border: none;
}
.btn-secondary {
background-color: #007bff;
border: none;
}
.progress {
height: 20px;
border-radius: 50px;
overflow: hidden;
}
.progress-bar {
transition: width 0.4s ease;
}
.terminal {
height: 400px;
overflow-y: scroll;
background-color: #1a1a2e;
color: #00d1b2;
padding: 20px;
border-radius: 10px;
box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.5);
font-family: 'Courier New', monospace;
}
/* Scrollbar styling */
.terminal::-webkit-scrollbar {
width: 10px;
}
.terminal::-webkit-scrollbar-thumb {
background-color: #00d1b2;
border-radius: 10px;
}
/* Responsive design */
@media (max-width: 768px) {
.container {
width: 90%;
}
}
/* Fullscreen background for particles */
#particles-js {
position: absolute;
width: 100%;
height: 100%;
top: 0;
left: 0;
z-index: 0; /* Debajo del contenido principal */
}
</style>
</head>
<body>
<div id="particles-js"></div>
<div class="container mt-4">
<h1>Condor Business Solutions CyberScan|Terminal Web de Escaneo</h1>
<input type="file" id="fileInput" class="form-control-file mb-3">
<button id="startScanBtn" class="btn btn-primary">Comenzar Escaneo</button>
<input type="text" id="urlInput" class="form-control" placeholder="Introduce una URL para escanear">
<button id="startUrlScanBtn" class="btn btn-secondary">Escanear URL</button>
<div id="progress" class="progress mt-3" style="display: none;">
<div id="progressBar" class="progress-bar bg-success" role="progressbar" style="width: 0%;" aria-valuenow="0" aria-valuemin="0" aria-valuemax="100"></div>
</div>
<div id="scanOutput" class="terminal mt-4"></div>
<div id="signaturesUsed" class="mt-3" style="display: none;">
<strong>Número de firmas utilizadas:</strong> <span id="signaturesCount"></span>
</div>
</div>
<script type="text/javascript">
var socket = io.connect('http://' + document.domain + ':' + location.port);
document.getElementById('startScanBtn').onclick = function() {
var fileInput = document.getElementById('fileInput');
var file = fileInput.files[0];
if (!file) {
appendToTerminal('<span style="color: red;">Por favor selecciona un archivo.</span>');
return;
}
appendToTerminal('<span style="color: green;">Subiendo archivo: ' + file.name + '</span>');
var formData = new FormData();
formData.append('file', file);
// Mostrar la barra de progreso y ocultar el botón
document.getElementById('progress').style.display = 'block';
document.getElementById('startScanBtn').style.display = 'none';
fetch('/upload', {
method: 'POST',
body: formData
})
.then(response => response.text())
.then(data => {
appendToTerminal('<span style="color: green;">' + data + '</span>');
// Restablecer la visibilidad del botón después del escaneo
document.getElementById('startScanBtn').style.display = 'block';
document.getElementById('progress').style.display = 'none';
document.getElementById('progressBar').style.width = '0%';
})
.catch(error => {
appendToTerminal('<span style="color: red;">Error al subir el archivo: ' + error + '</span>');
// Restablecer la visibilidad del botón en caso de error
document.getElementById('startScanBtn').style.display = 'block';
document.getElementById('progress').style.display = 'none';
document.getElementById('progressBar').style.width = '0%';
});
};
document.getElementById('startUrlScanBtn').onclick = function() {
var urlInput = document.getElementById('urlInput').value;
if (!urlInput) {
appendToTerminal('<span style="color: red;">Por favor introduce una URL.</span>');
return;
}
// Asegurar que la URL tenga un esquema
if (!urlInput.startsWith('http://') && !urlInput.startsWith('https://')) {
urlInput = 'http://' + urlInput;
}
appendToTerminal('<span style="color: green;">Escaneando URL: ' + urlInput + '</span>');
// Mostrar la barra de progreso y ocultar el botón
document.getElementById('progress').style.display = 'block';
document.getElementById('startUrlScanBtn').style.display = 'none';
fetch('/scan_url', {
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({ url: urlInput })
})
.then(response => response.json())
.then(data => {
if (data.error) {
appendToTerminal('<span style="color: red;">Error: ' + data.error + '</span>');
} else {
appendToTerminal('<span style="color: green;">' + data.message + '</span>');
}
// Restablecer la visibilidad del botón después del escaneo
document.getElementById('startUrlScanBtn').style.display = 'block';
document.getElementById('progress').style.display = 'none';
document.getElementById('progressBar').style.width = '0%';
})
.catch(error => {
appendToTerminal('<span style="color: red;">Error al escanear la URL: ' + error + '</span>');
// Restablecer la visibilidad del botón en caso de error
document.getElementById('startUrlScanBtn').style.display = 'block';
document.getElementById('progress').style.display = 'none';
document.getElementById('progressBar').style.width = '0%';
});
};
socket.on('scan_output', function(data) {
appendToTerminal(data.data);
// Parsear porcentaje de progreso si está presente
var progressMatch = data.data.match(/(\d+\.\d+)%/);
if (progressMatch) {
var percent = parseFloat(progressMatch[1]);
updateProgressBar(percent);
}
// Mostrar el número de firmas utilizadas (filtrando solo el número)
var signaturesMatch = data.data.match(/Known viruses: (\d+)/);
if (signaturesMatch) {
document.getElementById('signaturesUsed').style.display = 'block';
document.getElementById('signaturesCount').textContent = signaturesMatch[1];
}
});
function appendToTerminal(message) {
var scanOutput = document.getElementById('scanOutput');
var newLine = document.createElement('div');
newLine.innerHTML = message;
scanOutput.appendChild(newLine);
scanOutput.scrollTop = scanOutput.scrollHeight;
}
function updateProgressBar(percent) {
var progressBar = document.getElementById('progressBar');
progressBar.style.width = percent + '%';
progressBar.setAttribute('aria-valuenow', percent);
progressBar.innerHTML = percent.toFixed(2) + '%';
}
// Configurar particles.js
particlesJS('particles-js', {
"particles": {
"number": {
"value": 80,
"density": {
"enable": true,
"value_area": 800
}
},
"color": {
"value": "#ffffff"
},
"shape": {
"type": "circle",
"stroke": {
"width": 0,
"color": "#000000"
},
"polygon": {
"nb_sides": 5
},
},
"opacity": {
"value": 0.5,
"random": false,
"anim": {
"enable": false,
"speed": 1,
"opacity_min": 0.1,
"sync": false
}
},
"size": {
"value": 3,
"random": true,
"anim": {
"enable": false,
"speed": 40,
"size_min": 0.1,
"sync": false
}
},
"line_linked": {
"enable": true,
"distance": 150,
"color": "#ffffff",
"opacity": 0.4,
"width": 1
},
"move": {
"enable": true,
"speed": 6,
"direction": "none",
"random": false,
"straight": false,
"out_mode": "out",
"attract": {
"enable": false,
"rotateX": 600,
"rotateY": 1200
}
}
},
"interactivity": {
"detect_on": "canvas",
"events": {
"onhover": {
"enable": false
},
"onclick": {
"enable": false
},
"resize": true
},
},
"retina_detect": true
});
</script>
</body>
</html>