X-Git-Url: https://asedeno.scripts.mit.edu/gitweb/?a=blobdiff_plain;f=sshbn.c;fp=sshbn.c;h=42923ac071055167d721b6c183ea2823ed5a30ea;hb=c46da2f079cd28503f7b3fac5734bf6863858257;hp=b40781ff2996dc65fe9659e4d15ef8921d1f52f3;hpb=0acc74d711638c583d67c187a1b731b36a075d9f;p=PuTTY.git

diff --git a/sshbn.c b/sshbn.c
index b40781ff..42923ac0 100644
--- a/sshbn.c
+++ b/sshbn.c
@@ -1236,18 +1236,18 @@ Bignum bignum_from_bytes(const unsigned char *data, int nbytes)
 Bignum bignum_random_in_range(const Bignum lower, const Bignum upper)
 {
     Bignum ret = NULL;
+    unsigned char *bytes;
     int upper_len = bignum_bitcount(upper);
     int upper_bytes = upper_len / 8;
     int upper_bits = upper_len % 8;
     if (upper_bits) ++upper_bytes;
 
+    bytes = snewn(upper_bytes, unsigned char);
     do {
-        unsigned char *bytes;
         int i;
 
         if (ret) freebn(ret);
 
-        bytes = snewn(upper_bytes, unsigned char);
         for (i = 0; i < upper_bytes; ++i)
         {
             bytes[i] = (unsigned char)random_byte();
@@ -1260,6 +1260,7 @@ Bignum bignum_random_in_range(const Bignum lower, const Bignum upper)
 
         ret = bignum_from_bytes(bytes, upper_bytes);
     } while (bignum_cmp(ret, lower) < 0 || bignum_cmp(ret, upper) > 0);
+    sfree(bytes);
 
     return ret;
 }