Check tcp port

urlchecker_bot.py

@@ -5,6 +5,7 @@ from urlchecker_platform import logPlatform
 from urlchecker_status import logStatus
 from urlchecker_dns import getResolverDict, expandDomainList, getServerIpDict
 from urlchecker_http import getUrlHostname
+from urlchecker_network import isTcpPortOpen
 
 
 __version__ = "0.0.3"
@@ -46,14 +47,21 @@ class WebBot:
         domain_list = expandDomainList(domain_list)
         print(domain_list)
 
-        # Check if domain have an A record
-        server_id_dict = getServerIpDict(
+        # Get the list of server to check
+        server_ip_dict = getServerIpDict(
             self._db, status_id, resolver_dict, domain_list, "A"
         )
-        print(server_id_dict)
 
         # Check TCP port for the list of IP found
-        # If https ok, check SSL certificate
+        # XXX For now, check http/https only
+        server_ip_list = [x for x in server_ip_dict.keys()]
+        for server_ip in server_ip_list:
+            isTcpPortOpen(self._db, server_ip, 80, status_id)
+            isTcpPortOpen(self._db, server_ip, 443, status_id)
+        print(server_ip_dict)
+
+        # XXX If https ok, check SSL certificate
+
         # Check HTTP Status
 
     def stop(self):

urlchecker_network.py

+import socket
+import errno
+
+TIMEOUT = 2
+
+
 def logNetwork(db, ip, transport, port, state, status_id):
 
     with db._db.atomic():
@@ -25,3 +31,28 @@ def logNetwork(db, ip, transport, port, state, status_id):
                 state=state,
             )
         return previous_entry.id
+
+
+def isTcpPortOpen(db, ip, port, status_id):
+    is_open = False
+    sock = socket.socket()
+    sock.settimeout(TIMEOUT)
+    try:
+        sock.connect((ip, port))
+        state = "open"
+        is_open = True
+    except ConnectionRefusedError:
+        state = "closed"
+    except socket.timeout:
+        state = "filtered"
+    except OSError as e:
+        if e.errno == errno.EHOSTUNREACH:
+            # OSError: [Errno 113] No route to host
+            state = "filtered"
+        else:
+            raise
+
+    sock.close()
+
+    logNetwork(db, ip, "TCP", port, state, status_id)
+    return is_open