Update status on task timeout (#2454)

Author: ajinabraham

mobsf/StaticAnalyzer/views/common/async_task.py

@@ -4,6 +4,7 @@
     timedelta,
 )
 
+from django.dispatch import receiver
 from django.utils import timezone
 from django.shortcuts import render
 from django.conf import settings
@@ -14,6 +15,7 @@
 from django.views.decorators.http import require_http_methods
 
 from django_q.tasks import async_task
+from django_q.signals import post_execute
 
 from mobsf.StaticAnalyzer.models import (
     EnqueuedTask,
@@ -30,6 +32,19 @@
 logger = logging.getLogger(__name__)
 
 
+@receiver(post_execute)
+def detect_timeout(sender, task, **kwargs):
+    """Detect scan task timeout."""
+    if 'Task exceeded maximum timeout' in task['result']:
+        task_id = task['id']
+        EnqueuedTask.objects.filter(task_id=task_id).update(
+            app_name='Failed',
+            completed_at=timezone.now(),
+            status='Scan Timeout',
+        )
+        logger.error('Task %s exceeded maximum timeout', task_id)
+
+
 def async_analysis(checksum, app_name, func, *args):
     """Async Analysis Task."""
     # Check if the task is already completed