Update ElGamal.py

bug for 7680-bit key
serengil · May 30, 2024 · 7c93ebb · 7c93ebb
1 parent d24108a
commit 7c93ebb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lightphe/cryptosystems/ElGamal.py b/lightphe/cryptosystems/ElGamal.py
@@ -51,7 +51,8 @@ def generate_keys(self, key_size: int):
 
         # picking a generator g
         # g = random.randint(2, int(math.sqrt(p))) # reaches int limit for 3072-bit key
-        g = int(random.uniform(2, float(decimal.Decimal(p).sqrt())))
+        # g = int(random.uniform(2, float(decimal.Decimal(p).sqrt())))
+        g = random.randint(2, int(decimal.Decimal(p).sqrt()))
 
         # picking a private key x
         x = random.randint(1, p - 2)