Check return value from GetConsoleMode/SetConsoleMode

tartley · Aug 25, 2022 · 7ec896d · 7ec896d
1 parent 7d3d3b9
commit 7ec896d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/colorama/win32.py b/colorama/win32.py
@@ -171,8 +171,12 @@ def GetConsoleMode(stream_id):
         handle = _GetStdHandle(stream_id)
         mode = wintypes.DWORD()
         success = _GetConsoleMode(handle, byref(mode))
+        if not success:
+            raise ctypes.WinError()
         return mode.value
 
     def SetConsoleMode(stream_id, mode):
         handle = _GetStdHandle(stream_id)
-        return _SetConsoleMode(handle, mode)
+        success = _SetConsoleMode(handle, mode)
+        if not success:
+            raise ctypes.WinError()