From 763e2359ae3e4e9e1177bc479c9171ddfc79a4b3 Mon Sep 17 00:00:00 2001
From: Kay0u <pierre@kayou.io>
Date: Sat, 11 Apr 2020 16:08:14 +0200
Subject: [PATCH] ipv6 support

---
 yunodiagnoser.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/yunodiagnoser.py b/yunodiagnoser.py
index 9272ac9..381ce2f 100644
--- a/yunodiagnoser.py
+++ b/yunodiagnoser.py
@@ -59,7 +59,10 @@ def check_rate_limit(key, now):
 
 async def check_port_is_open(ip, port):
 
-    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    if ":" in ip:
+        sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
+    else:
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
     sock.settimeout(2)
     result = sock.connect_ex((ip, port))
     sock.close()