commit ab82db706f3f49525ec194c8bd9655c5323240b7
author Jean-Paul Calderone <exarkun@boson> Thu Mar 06 00:09:31 2008 -0500
committer Jean-Paul Calderone <exarkun@boson> Thu Mar 06 00:09:31 2008 -0500
tree c1f623c091b69d9d69d91d2394777eff48c94396
parent 1bfdea0bb91a7b8cb22d0d3ccfd92abad44b24a5

slightly better PKey invalid bit tests, and a fix for older versions of OpenSSL

src/crypto/pkey.c

diff --git a/src/crypto/pkey.c b/src/crypto/pkey.c
index 07fe423..0c2bea6 100644
--- a/src/crypto/pkey.c
+++ b/src/crypto/pkey.c
@@ -50,6 +50,10 @@
     switch (type)
     {
         case crypto_TYPE_RSA:
+            if (bits <= 0) {
+                PyErr_SetString(PyExc_ValueError, "Invalid number of bits");
+                return NULL;
+            }
             if ((rsa = RSA_generate_key(bits, 0x10001, NULL, NULL)) == NULL)
                 FAIL();
             if (!EVP_PKEY_assign_RSA(self->pkey, rsa))