Get body from django proper

AikidoSec · Aug 22, 2024 · 0d8c6ae · 0d8c6ae
1 parent 96df99a
commit 0d8c6ae
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/aikido_firewall/sources/django.py b/aikido_firewall/sources/django.py
@@ -17,9 +17,18 @@ def gen_aikido_middleware_function(former__middleware_chain):
     """
 
     def aikido_middleware_function(request):
-        context = Context(
-            req=request.META, raw_body=request.body.decode("utf-8"), source="django"
-        )
+        # Get a parsed body from Django :
+        body = request.POST.dict()
+        if len(body) == 0 and request.content_type == "application/json":
+            try:
+                body = json.loads(request.body)
+            except Exception:
+                pass
+        if len(body) == 0:
+            # E.g. XML Data
+            body = request.body
+
+        context = Context(req=request.META, body=body, source="django")
         context.set_as_current_context()
         request_handler(stage="init")