csharp

dafny-lang · Nov 7, 2023 · b3b4a1e · b3b4a1e
1 parent 39ceb39
commit b3b4a1e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/interop/cs/DRandomUniformPowerOfTwo.cs b/src/interop/cs/DRandomUniformPowerOfTwo.cs
@@ -11,7 +11,13 @@ namespace UniformPowerOfTwo_mImplementation {
     public class DRandomUniformPowerOfTwo : Coin_mInterface.DRandomCoin {
 
       /// Generates a uniformly random BigInteger between 0 (inclusive) and 2^bitLength (exclusive)
-      private static BigInteger UniformPowerOfTwo(int bitLength) {
+      private static BigInteger UniformPowerOfTwo(BigInteger n) {
+        if (n <= BigInteger.Zero) {
+          throw new ArgumentException("n must be positive");
+        }
+
+        int bitLength = System.Convert.ToInt32(n.GetBitLength());
+
         if (bitLength < 1) {
           return BigInteger.Zero;
         }