Fix waf detection

web/fixtures/default_scan_engines.yaml

@@ -3,6 +3,7 @@
   fields:
     engine_name: Full Scan
     subdomain_discovery: true
+    waf_detection: true
     dir_file_fuzz: true
     port_scan: true
     fetch_url: true
@@ -36,6 +37,7 @@
   fields:
     engine_name: Subdomain Only Scan
     subdomain_discovery: true
+    waf_detection: true
     dir_file_fuzz: false
     port_scan: false
     fetch_url: false
@@ -69,6 +71,7 @@
   fields:
     engine_name: OSINT Only
     subdomain_discovery: false
+    waf_detection: false
     dir_file_fuzz: false
     port_scan: false
     fetch_url: false
@@ -102,6 +105,7 @@
   fields:
     engine_name: Vulnerability Scan Only
     subdomain_discovery: true
+    waf_detection: false
     dir_file_fuzz: false
     port_scan: false
     fetch_url: false
@@ -135,6 +139,7 @@
   fields:
     engine_name: reNgine Recommended
     subdomain_discovery: true
+    waf_detection: true
     dir_file_fuzz: true
     port_scan: false
     fetch_url: false

web/reNgine/tasks.py

@@ -214,6 +214,7 @@ def initiate_scan(
 	task.scan_status = 1
 	task.start_scan_date = timezone.now()
 	task.subdomain_discovery = True if engine_object.subdomain_discovery else False
+	task.waf_detection = True if engine_object.waf_detection else False
 	task.dir_file_fuzz = True if engine_object.dir_file_fuzz else False
 	task.port_scan = True if engine_object.port_scan else False
 	task.fetch_url = True if engine_object.fetch_url else False

web/startScan/migrations/0029_scanhistory_waf_detection.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.4 on 2022-07-10 18:13
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('startScan', '0028_auto_20220710_1140'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='scanhistory',
+            name='waf_detection',
+            field=models.BooleanField(default=False, null=True),
+        ),
+    ]

web/startScan/models.py

@@ -23,6 +23,7 @@ class ScanHistory(models.Model):
 	scan_type = models.ForeignKey(EngineType, on_delete=models.CASCADE)
 	celery_id = models.CharField(max_length=100, blank=True)
 	subdomain_discovery = models.BooleanField(null=True, default=False)
+	waf_detection = models.BooleanField(null=True, default=False)
 	dir_file_fuzz = models.BooleanField(null=True, default=False)
 	port_scan = models.BooleanField(null=True, default=False)
 	fetch_url = models.BooleanField(null=True, default=False)

web/startScan/templates/startScan/_items/scanEngine_select.html

@@ -32,6 +32,18 @@
                 {% endif %}
               </td>
             </tr>
+            <tr>
+              <td>
+                <p class="mr-2 mt-1">WAF Detection</>
+              </td>
+              <td>
+                {% if engine.waf_detection %}
+                  {% include 'base/_items/green_tick.html' %}
+                {% else %}
+                  {% include 'base/_items/red_cross.html' %}
+                {% endif %}
+              </td>
+            </tr>
             <tr>
               <td>
                 <p class="mr-2 mt-1">Screenshot</>