From 668404b049a76d022cdd7460aeb61f8ff1e267c4 Mon Sep 17 00:00:00 2001
From: nocturn9x <hackhab@gmail.com>
Date: Mon, 31 May 2021 11:11:56 +0200
Subject: [PATCH] Minor fixes to SSL

---
 giambio/core.py   | 1 +
 giambio/socket.py | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/giambio/core.py b/giambio/core.py
index 3c77d3c..ba5ec6e 100644
--- a/giambio/core.py
+++ b/giambio/core.py
@@ -18,6 +18,7 @@ limitations under the License.
 
 # Import libraries and internal resources
 import types
+import socket
 from itertools import chain
 from timeit import default_timer
 from giambio.context import TaskManager
diff --git a/giambio/socket.py b/giambio/socket.py
index 3f56ab6..eefdb21 100644
--- a/giambio/socket.py
+++ b/giambio/socket.py
@@ -19,9 +19,9 @@ import socket as builtin_socket
 from giambio.run import get_event_loop
 from giambio.exceptions import ResourceClosed
 from giambio.traps import want_write, want_read
+from ssl import SSLWantReadError, SSLWantWriteError
 
-# TODO: Take into account SSLWantReadError and SSLWantWriteError
-IOInterrupt = (BlockingIOError, InterruptedError)
+IOInterrupt = (BlockingIOError, InterruptedError, SSLWantReadError, SSLWantWriteError)
 
 
 class AsyncSocket: