Fix _countdown_handler not invoking timeout_change; Fix value is valu…

…e < 0 (#2287)
Gallopsled · Oct 29, 2023 · 3ff37a8 · 3ff37a8
1 parent 045b8c2
commit 3ff37a8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pwnlib/timeout.py b/pwnlib/timeout.py
@@ -30,9 +30,11 @@ def __enter__(self):
             self.obj._stop = min(self.obj._stop, self.old_stop)
 
         self.obj._timeout = self.timeout
+        self.obj.timeout_change()
     def __exit__(self, *a):
         self.obj._timeout = self.old_timeout
         self.obj._stop    = self.old_stop
+        self.obj.timeout_change()
 
 class _local_handler(object):
     def __init__(self, obj, timeout):
@@ -157,7 +159,7 @@ def _get_timeout_seconds(self, value):
         else:
             value = float(value)
 
-            if value is value < 0:
+            if value < 0:
                 raise AttributeError("timeout: Timeout cannot be negative")
 
             if value > self.maximum: