@@ -753,15 +753,19 @@ class _(dict):
753753 page = unicode (page , errors = "ignore" )
754754 else :
755755 page = getUnicode (page )
756- socket .setdefaulttimeout (conf .timeout )
757756
758- for function in kb .preprocessFunctions :
759- try :
760- page , responseHeaders , code = function (page , responseHeaders , code )
761- except Exception as ex :
762- errMsg = "error occurred while running preprocess "
763- errMsg += "function '%s' ('%s')" % (function .func_name , getSafeExString (ex ))
764- raise SqlmapGenericException (errMsg )
757+ for function in kb .preprocessFunctions :
758+ try :
759+ page , responseHeaders , code = function (page , responseHeaders , code )
760+ except Exception as ex :
761+ errMsg = "error occurred while running preprocess "
762+ errMsg += "function '%s' ('%s')" % (function .func_name , getSafeExString (ex ))
763+ raise SqlmapGenericException (errMsg )
764+
765+ threadData .lastPage = page
766+ threadData .lastCode = code
767+
768+ socket .setdefaulttimeout (conf .timeout )
765769
766770 processResponse (page , responseHeaders , status )
767771
@@ -1309,10 +1313,9 @@ def _(value):
13091313 page , headers , code = Connect .getPage (url = _ (kb .secondReq [0 ]), post = _ (kb .secondReq [2 ]), method = kb .secondReq [1 ], cookie = kb .secondReq [3 ], silent = silent , auxHeaders = dict (auxHeaders , ** dict (kb .secondReq [4 ])), response = response , raise404 = False , ignoreTimeout = timeBasedCompare , refreshing = True )
13101314
13111315 threadData .lastQueryDuration = calculateDeltaSeconds (start )
1312- threadData .lastPage = page
1313- threadData .lastCode = code
13141316
1315- kb .originalCode = kb .originalCode or code
1317+ kb .originalCode = code if kb .originalCode is None else kb .originalCode
1318+ kb .originalPage = page if kb .originalPage is None else kb .originalPage
13161319
13171320 if kb .testMode :
13181321 kb .testQueryCount += 1
0 commit comments