sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-01-23 15:54:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixes #1550

Browse Source
This commit is contained in:
Miroslav Stampar 2015-11-20 16:52:59 +01:00
parent 7fa9c8e938
commit a5489516eb
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lib/core/option.py
View File

@ -1027,12 +1027,13 @@ def _setSocketPreConnect():
def _():
while kb.threadContinue:
try:
for address in socket._ready:
if len(socket._ready[address]) < SOCKET_PRE_CONNECT_QUEUE_SIZE:
s = socket.socket()
for key in socket._ready:
if len(socket._ready[key]) < SOCKET_PRE_CONNECT_QUEUE_SIZE:
family, type, proto, address = key
s = socket.socket(family, type, proto)
s._connect(address)
with kb.locks.socket:
socket._ready[address].append(s._sock)
socket._ready[key].append(s._sock)
except socket.error:
pass
finally:
@ -1040,11 +1041,12 @@ def _setSocketPreConnect():
def connect(self, address):
found = False
key = (self.family, self.type, self.proto, address)
with kb.locks.socket:
if address not in socket._ready:
socket._ready[address] = []
if len(socket._ready[address]) > 0:
self._sock = socket._ready[address].pop(0)
if key not in socket._ready:
socket._ready[key] = []
if len(socket._ready[key]) > 0:
self._sock = socket._ready[key].pop(0)
found = True
if not found:
self._connect(address)