3 Commits

Author SHA1 Message Date
Drew Noel 45d00f92d2 Merge e83c466a33 into 9049c8476a 2024-08-15 13:29:15 +00:00
Drew Noel e83c466a33 Return errno strings, clean up comment 2024-08-15 09:21:50 -04:00
Drew Noel 1297d284e9 Switch connect_ex result checks to errno lookups 2024-07-22 12:54:13 -04:00
+3 -2
View File
@@ -21,6 +21,7 @@
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
"""
import errno
import multiprocessing
import os
import queue
@@ -442,10 +443,10 @@ class DHEat:
# out.d("Creating socket (%u of %u already exist)..." % (len(socket_dict), concurrent_sockets), write_now=True)
ret = s.connect_ex((target_ip_address, aconf.port))
num_attempted_connections += 1
if ret in [0, 115]: # Check if connection is successful or EINPROGRESS.
if ret in [0, errno.EINPROGRESS]:
socket_dict[s] = now
else:
out.d("connect_ex() returned: %d" % ret, write_now=True)
out.d("connect_ex() returned: %s (%d)" % (os.strerror(ret), ret), write_now=True)
# out.d("Calling select() on %u sockets..." % len(socket_dict), write_now=True)
socket_list: List[socket.socket] = [*socket_dict] # Get a list of sockets from the dictionary.