Commit 27707be
committed
Fixes sqlmapproject#1416
1 parent aa2112b commit 27707be
1 file changed
Lines changed: 2 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1053,9 +1053,9 @@ def _randomizeParameter(paramString, randomParameter): | |||
| 1053 | 1053 | _, headers, code = Connect.getPage(url=uri, get=get, post=post, method=method, cookie=cookie, ua=ua, referer=referer, host=host, silent=silent, auxHeaders=auxHeaders, raise404=raise404, skipRead=(kb.nullConnection == NULLCONNECTION.SKIP_READ)) | |
| 1054 | 1054 | ||
| 1055 | 1055 | if headers: | |
| 1056 | - if kb.nullConnection in (NULLCONNECTION.HEAD, NULLCONNECTION.SKIP_READ) and HTTP_HEADER.CONTENT_LENGTH in headers: | ||
| 1056 | + if kb.nullConnection in (NULLCONNECTION.HEAD, NULLCONNECTION.SKIP_READ) and headers.get(HTTP_HEADER.CONTENT_LENGTH): | ||
| 1057 | 1057 | pageLength = int(headers[HTTP_HEADER.CONTENT_LENGTH]) | |
| 1058 | - elif kb.nullConnection == NULLCONNECTION.RANGE and HTTP_HEADER.CONTENT_RANGE in headers: | ||
| 1058 | + elif kb.nullConnection == NULLCONNECTION.RANGE and headers.get(HTTP_HEADER.CONTENT_RANGE): | ||
| 1059 | 1059 | pageLength = int(headers[HTTP_HEADER.CONTENT_RANGE][headers[HTTP_HEADER.CONTENT_RANGE].find('/') + 1:]) | |
| 1060 | 1060 | finally: | |
| 1061 | 1061 | kb.pageCompress = popValue() | |
0 commit comments