RustPython
diff --git a/‎Lib/test/test_hashlib.py‎
Lines changed: 0 additions & 2 deletions b/‎Lib/test/test_hashlib.py‎
Lines changed: 0 additions & 2 deletions
diff --git a/‎crates/stdlib/src/hashlib.rs‎
Lines changed: 9 additions & 2 deletions b/‎crates/stdlib/src/hashlib.rs‎
Lines changed: 9 additions & 2 deletions
@@ -295,8 +295,6 @@ def test_hexdigest(self):
                 self.assertIsInstance(h.digest(), bytes)
                 self.assertEqual(hexstr(h.digest()), h.hexdigest())
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_digest_length_overflow(self):
         # See issue #34922
         large_sizes = (2**29, 2**32-10, 2**32+10, 2**61, 2**64-10, 2**64+10)
 
@@ -83,9 +83,16 @@ pub mod _hashlib {
     }
 
     impl XofDigestArgs {
+        // Match CPython's SHAKE output guard in Modules/sha3module.c.
+        const MAX_SHAKE_OUTPUT_LENGTH: usize = 1 << 29;
+
         fn length(&self, vm: &VirtualMachine) -> PyResult<usize> {
-            usize::try_from(self.length)
-                .map_err(|_| vm.new_value_error("length must be non-negative"))
+            let length = usize::try_from(self.length)
+                .map_err(|_| vm.new_value_error("length must be non-negative"))?;
+            if length >= Self::MAX_SHAKE_OUTPUT_LENGTH {
+                return Err(vm.new_value_error("length is too large"));
+            }
+            Ok(length)
         }
     }
Original file line number	Diff line number	Diff line change
`@@ -83,9 +83,16 @@ pub mod _hashlib {`
`83`	`83`	`}`
`84`	`84`
`85`	`85`	`impl XofDigestArgs {`
	`86`	`+ // Match CPython's SHAKE output guard in Modules/sha3module.c.`
	`87`	`+ const MAX_SHAKE_OUTPUT_LENGTH: usize = 1 << 29;`
	`88`	`+`
`86`	`89`	`fn length(&self, vm: &VirtualMachine) -> PyResult<usize> {`
`87`		`- usize::try_from(self.length)`
`88`		`- .map_err(\|_\| vm.new_value_error("length must be non-negative"))`
	`90`	`+ let length = usize::try_from(self.length)`
	`91`	`+ .map_err(\|_\| vm.new_value_error("length must be non-negative"))?;`
	`92`	`+ if length >= Self::MAX_SHAKE_OUTPUT_LENGTH {`
	`93`	`+ return Err(vm.new_value_error("length is too large"));`
	`94`	`+ }`
	`95`	`+ Ok(length)`
`89`	`96`	`}`
`90`	`97`	`}`
`91`	`98`