one more update related to the redirection mechanism

lib/core/option.py

@@ -1469,6 +1469,7 @@ def __setKnowledgeBaseAttributes(flushAll=True):
     kb.pageTemplates = dict()
     kb.previousMethod = None
     kb.orderByColumns = None
+    kb.originalCode = None
     kb.originalPage = None
     kb.originalTimeDelay = None
 

lib/request/connect.py

@@ -309,7 +309,7 @@ class Connect:
                   else kb.originalPage if kb.redirectChoice == REDIRECTION.ORIGINAL\
                   else conn.read()
                 skipLogTraffic = kb.redirectChoice != REDIRECTION.FOLLOW
-                code = conn.redcode
+                code = conn.redcode if kb.redirectChoice != REDIRECTION.ORIGINAL else kb.originalCode
             else:
                 page = conn.read()
 
@@ -517,6 +517,7 @@ class Connect:
         page = None
         pageLength = None
         uri = None
+        code = None
 
         if not place:
             place = kb.injection.place or PLACE.GET
@@ -700,6 +701,8 @@ class Connect:
 
         threadData.lastQueryDuration = calculateDeltaSeconds(start)
 
+        kb.originalCode = kb.originalCode or code
+
         if kb.testMode:
             kb.testQueryCount += 1