[Fix] Fix runserver for PyCharm debug

.github/workflows/mobsf-test.yml

@@ -65,6 +65,7 @@ jobs:
         poetry run python manage.py makemigrations
         poetry run python manage.py makemigrations StaticAnalyzer
         poetry run python manage.py migrate
+        poetry run python manage.py collectstatic --noinput
         poetry run python manage.py create_roles
 
     - uses: actions/setup-java@v4

.gitignore

@@ -78,4 +78,5 @@ mobsf/secret
 mobsf/StaticAnalyzer/migrations
 mobsf/MobSF/windows_vm_priv_key.asc
 mobsf/setup_done.txt
+mobsf/staticfiles
 TODO.md

manage.py

@@ -6,13 +6,20 @@
 
 warnings.filterwarnings('ignore', category=UserWarning, module='cffi')
 
-if __name__ == '__main__':
-    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'mobsf.MobSF.settings')
 
-    from django.core.management import execute_from_command_line
-    if 'runserver' in sys.argv:
-        print('We do not allow debug server anymore. '
-              'Please follow official docs: '
-              'https://mobsf.github.io/docs/')
-        sys.exit(0)
+def main():
+    """Run administrative tasks."""
+    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'mobsf.MobSF.settings')
+    try:
+        from django.core.management import execute_from_command_line
+    except ImportError as exc:
+        raise ImportError(
+            "Couldn't import Django. Are you sure it's installed and "
+            "available on your PYTHONPATH environment variable? Did you "
+            "forget to activate a virtual environment?"
+        ) from exc
     execute_from_command_line(sys.argv)
+
+
+if __name__ == '__main__':
+    main()

mobsf/MobSF/settings.py

@@ -192,23 +192,17 @@
     'mobsf.MobSF',
     'mobsf.MalwareAnalyzer',
 )
-MIDDLEWARE_CLASSES = (
+MIDDLEWARE = (
     'django.middleware.security.SecurityMiddleware',
     'whitenoise.middleware.WhiteNoiseMiddleware',
     'django.middleware.common.CommonMiddleware',
     'django.middleware.csrf.CsrfViewMiddleware',
-    'django.contrib.auth.middleware.AuthenticationMiddleware',
-    'django.contrib.auth.middleware.SessionAuthenticationMiddleware',
-    'django.contrib.messages.middleware.MessageMiddleware',
-    'django.middleware.clickjacking.XFrameOptionsMiddleware',
-    'django_ratelimit.middleware.RatelimitMiddleware',
-)
-MIDDLEWARE = (
     'mobsf.MobSF.views.api.api_middleware.RestApiAuthMiddleware',
     'django.contrib.sessions.middleware.SessionMiddleware',
     'django.contrib.auth.middleware.AuthenticationMiddleware',
     'django.contrib.messages.middleware.MessageMiddleware',
-
+    'django.middleware.clickjacking.XFrameOptionsMiddleware',
+    'django_ratelimit.middleware.RatelimitMiddleware',
 )
 ROOT_URLCONF = 'mobsf.MobSF.urls'
 WSGI_APPLICATION = 'mobsf.MobSF.wsgi.application'
@@ -240,8 +234,16 @@
 MEDIA_ROOT = os.path.join(BASE_DIR, 'uploads')
 MEDIA_URL = '/uploads/'
 STATIC_URL = '/static/'
-STATIC_ROOT = os.path.join(BASE_DIR, 'static')
-STATICFILES_STORAGE = 'whitenoise.storage.CompressedStaticFilesStorage'
+STATIC_ROOT = os.path.join(BASE_DIR, 'staticfiles')
+STATICFILES_DIRS = [os.path.join(BASE_DIR, "static")]
+STORAGES = {
+    "default": {
+        "BACKEND": "django.core.files.storage.FileSystemStorage",
+    },
+    "staticfiles": {
+        "BACKEND": "whitenoise.storage.CompressedManifestStaticFilesStorage",
+    },
+}
 # 256MB limit for file uploads
 DATA_UPLOAD_MAX_MEMORY_SIZE = 256 * 1024 * 1024
 # 400MB per file limit for uncompressed files

mobsf/MobSF/wsgi.py

@@ -11,14 +11,8 @@
 
 from django.core.wsgi import get_wsgi_application
 
-from whitenoise import WhiteNoise
-
-from . import settings
-
 
 warnings.filterwarnings('ignore', category=UserWarning, module='cffi')
 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'mobsf.MobSF.settings')
 
-static = os.path.join(settings.BASE_DIR, 'static')
-application = WhiteNoise(get_wsgi_application(),
-                         root=static, prefix='static/')
+application = get_wsgi_application()

mobsf/StaticAnalyzer/tests.py

@@ -6,6 +6,7 @@
 
 from mobsf.MobSF.init import api_key
 
+from django.contrib.auth import get_user_model
 from django.conf import settings
 from django.http import HttpResponse
 from django.test import Client, TestCase
@@ -24,6 +25,12 @@ def static_analysis_test():
         uploaded = []
         logger.info('Running Upload Test')
         http_client = Client()
+
+        # Create Test User
+        user_model = get_user_model()
+        user_model.objects.create_user(username='mobsf_test', password='mobsf_test', is_superuser=True, is_staff=True)
+        http_client.login(username='mobsf_test', password='mobsf_test')
+
         apk_dir = os.path.join(settings.BASE_DIR, 'StaticAnalyzer/test_files/')
         for filename in os.listdir(apk_dir):
             if not filename.endswith(EXTS):

mobsf/static/jsTree/custom.css

@@ -1,5 +1,3 @@
-{% load static %}
-
 html,
 body {
 	height: 100% ;
@@ -23,7 +21,7 @@ div#fileframe {
 
 div#contentframe {
 	background-color: #FFFFFF ;
-	background-image: url( "/static/jsTree/themes/default/content_background.gif" ) ;
+	background-image: url(themes/default/content_background.gif) ;
 	background-position: left top ;
 	background-repeat: repeat-y ;
 	float: left ;

run.bat

@@ -7,4 +7,7 @@ goto :run
 SET conf="0.0.0.0:8000 [::]:8000"
 :run
 echo Running MobSF on %conf%
+
+poetry run python manage.py collectstatic --noinput
+
 poetry run waitress-serve --listen=%conf% --threads=10 --channel-timeout=3600 mobsf.MobSF.wsgi:application

run.sh

@@ -31,5 +31,7 @@ else
     PORT='8000'
 fi
 
-python3 -m poetry run gunicorn -b ${IP}:${PORT} mobsf.MobSF.wsgi:application --workers=1 --threads=10 --timeout=3600 \
+poetry run python manage.py collectstatic --noinput
+
+poetry run gunicorn -b ${IP}:${PORT} mobsf.MobSF.wsgi:application --workers=1 --threads=10 --timeout=3600 \
     --log-level=critical --log-file=- --access-logfile=- --error-logfile=- --capture-output

scripts/clean.sh

@@ -27,7 +27,7 @@ if [[ "$VAL" =~ ^[yY]$ ]]; then
     echo 'Cleaning up MobSF directories and files...'
 
     # Remove files from key directories
-    rm -rf ./mobsf/{uploads,downloads,StaticAnalyzer/migrations,DynamicAnalyzer/migrations,MobSF/migrations}/*
+    rm -rf ./mobsf/{uploads,downloads,staticfiles,StaticAnalyzer/migrations,DynamicAnalyzer/migrations,MobSF/migrations}/*
 
     echo 'Removing Python bytecode and cache files'
     find ./ -type f -name "*.pyc" -o -name "*.pyo" -delete

scripts/entrypoint.sh

@@ -4,6 +4,9 @@ set -e
 python3 manage.py makemigrations && \
 python3 manage.py makemigrations StaticAnalyzer && \
 python3 manage.py migrate
+
+python3 manage.py collectstatic --noinput
+
 set +e
 python3 manage.py createsuperuser --noinput --email ""
 set -e

setup.bat

@@ -69,8 +69,15 @@ where python >nul 2>&1 && (
   poetry run python manage.py makemigrations
   poetry run python manage.py makemigrations StaticAnalyzer
   poetry run python manage.py migrate
+
+  echo [INSTALL] Collect Static files
+  poetry run python manage.py collectstatic --noinput
+
+  echo [INSTALL] Creating Super User
   poetry run python manage.py createsuperuser --noinput --email ""
   poetry run python manage.py create_roles
+
+  echo [INSTALL] wkhtmltopdf is not installed
   echo Download and Install wkhtmltopdf for PDF Report Generation - https://wkhtmltopdf.org/downloads.html
   echo [INSTALL] Installation Complete
   exit /b 0

setup.sh

@@ -53,9 +53,15 @@ export DJANGO_SUPERUSER_PASSWORD=mobsf
 python3 -m poetry run python manage.py makemigrations
 python3 -m poetry run python manage.py makemigrations StaticAnalyzer
 python3 -m poetry run python manage.py migrate
+
+echo '[INSTALL] Collect Static files'
+python3 -m poetry run python manage.py collectstatic --noinput
+
+echo '[INSTALL] Creating Superuser'
 python3 -m poetry run python manage.py createsuperuser --noinput --email ""
 python3 -m poetry run python manage.py create_roles
 
+echo '[INSTALL] Checking wkhtmltopdf'
 # Check for wkhtmltopdf
 if ! command -v wkhtmltopdf &>/dev/null; then
     echo 'Download and Install wkhtmltopdf for PDF Report Generation - https://wkhtmltopdf.org/downloads.html'