From 1297d284e912d0d0fd3e1ceb90bd9ee2bff7c6ce Mon Sep 17 00:00:00 2001
From: Drew Noel <drew.noel@aon.com>
Date: Mon, 22 Jul 2024 12:54:13 -0400
Subject: [PATCH] Switch connect_ex result checks to errno lookups

---
 src/ssh_audit/dheat.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/ssh_audit/dheat.py b/src/ssh_audit/dheat.py
index 8700040..135cdb0 100644
--- a/src/ssh_audit/dheat.py
+++ b/src/ssh_audit/dheat.py
@@ -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,7 +443,7 @@ 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]:  # Check if connection is successful or EINPROGRESS.
                     socket_dict[s] = now
                 else:
                     out.d("connect_ex() returned: %d" % ret, write_now=True)