Wrap value to 2**53 if 0

crossbario · Apr 13, 2024 · f084051 · f084051
1 parent c12ce18
commit f084051
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autobahn/util.py b/autobahn/util.py
@@ -279,7 +279,7 @@ def __next__(self):
 
 
 # 8 byte mask with 53 LSBs set (WAMP requires IDs from [1, 2**53]
-_WAMP_ID_MASK = 2 ** 53 - 1
+_WAMP_ID_MASK = struct.unpack("@Q", b"\x00\x1f\xff\xff\xff\xff\xff\xff")[0]
 
 
 def rid():
@@ -298,7 +298,7 @@ def rid():
     :returns: A random integer ID.
     :rtype: int
     """
-    return (struct.unpack("@Q", os.urandom(8))[0] & _WAMP_ID_MASK) + 1
+    return struct.unpack("@Q", os.urandom(8))[0] & _WAMP_ID_MASK or 2 ** 53
 
 
 # noinspection PyShadowingBuiltins