should solve the problem with timeout problems in time-based payloads

This commit is contained in:
Miroslav Stampar 2010-12-20 16:45:41 +00:00
parent 364bc8e7d4
commit c948bced61

View File

@ -88,6 +88,7 @@ class Connect:
raise404 = kwargs.get('raise404', True) raise404 = kwargs.get('raise404', True)
auxHeaders = kwargs.get('auxHeaders', None) auxHeaders = kwargs.get('auxHeaders', None)
response = kwargs.get('response', False) response = kwargs.get('response', False)
ignoreTimeout = kwargs.get('ignoreTimeout', False)
page = "" page = ""
cookieStr = "" cookieStr = ""
@ -287,7 +288,7 @@ class Connect:
if "BadStatusLine" not in tbMsg: if "BadStatusLine" not in tbMsg:
warnMsg += " or proxy" warnMsg += " or proxy"
if silent: if silent or (ignoreTimeout and "timeout" in tbMsg):
return None, None return None, None
elif kb.retriesCount < conf.retries: elif kb.retriesCount < conf.retries:
kb.retriesCount += 1 kb.retriesCount += 1
@ -406,7 +407,7 @@ class Connect:
start = time.time() start = time.time()
if not content and not response and kb.nullConnection: if kb.nullConnection and not content and not response and not timeBasedCompare:
if kb.nullConnection == NULLCONNECTION.HEAD: if kb.nullConnection == NULLCONNECTION.HEAD:
method = HTTPMETHOD.HEAD method = HTTPMETHOD.HEAD
elif kb.nullConnection == NULLCONNECTION.RANGE: elif kb.nullConnection == NULLCONNECTION.RANGE:
@ -423,7 +424,7 @@ class Connect:
pageLength = int(headers['Content-Range'][headers['Content-Range'].find('/') + 1:]) pageLength = int(headers['Content-Range'][headers['Content-Range'].find('/') + 1:])
if not pageLength: if not pageLength:
page, headers = Connect.getPage(url=uri, get=get, post=post, cookie=cookie, ua=ua, silent=silent, method=method, auxHeaders=auxHeaders, response=response, raise404=raise404) page, headers = Connect.getPage(url=uri, get=get, post=post, cookie=cookie, ua=ua, silent=silent, method=method, auxHeaders=auxHeaders, response=response, raise404=raise404, ignoreTimeout=timeBasedCompare)
kb.lastQueryDuration = calculateDeltaSeconds(start) kb.lastQueryDuration = calculateDeltaSeconds(start)