From 95f13965dc4223497fedf85ec3dc4d286996cc39 Mon Sep 17 00:00:00 2001
From: swee <meow@swee.codes>
Date: Wed, 5 Feb 2025 17:27:15 -0800
Subject: [PATCH] Update server.py

---
 server.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/server.py b/server.py
index 817fe20..32a5eb0 100644
--- a/server.py
+++ b/server.py
@@ -179,14 +179,20 @@ sockets = {}
 sockets_ssl = {}
 # Open the specified non-SSL sockets.
 for i in restrict_ip.split(" "):
-    sockets[i] = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    if ":" in i:
+        sockets[i] = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
+    else:
+        sockets[i] = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
     sockets[i].setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
     sockets[i].settimeout(None)
     sockets[i].bind((i,6667))
     sockets[i].listen(1)
 if ssl_option:
     for i in restrict_ip.split(" "):
-        sockets_ssl[i] = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        if ":" in i:
+            sockets_ssl[i] = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
+        else:
+            sockets_ssl[i] = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         sockets_ssl[i].setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
         sockets_ssl[i].settimeout(None)
         sockets_ssl[i].bind((i,6697))