From ac9bb7eda530dca8bd59bb900bbdfdb12e907586 Mon Sep 17 00:00:00 2001 From: Markus Friedl Date: Tue, 25 Dec 2001 18:49:57 +0000 Subject: be more careful on allocation --- usr.bin/ssh/key.c | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'usr.bin') diff --git a/usr.bin/ssh/key.c b/usr.bin/ssh/key.c index 1013578e1b0..3a0ed046bdf 100644 --- a/usr.bin/ssh/key.c +++ b/usr.bin/ssh/key.c @@ -32,7 +32,7 @@ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "includes.h" -RCSID("$OpenBSD: key.c,v 1.36 2001/12/19 07:18:56 deraadt Exp $"); +RCSID("$OpenBSD: key.c,v 1.37 2001/12/25 18:49:56 markus Exp $"); #include @@ -63,6 +63,8 @@ key_new(int type) rsa = RSA_new(); rsa->n = BN_new(); rsa->e = BN_new(); + if (rsa == NULL || rsa->n == NULL || rsa->e == NULL) + fatal("key_new: malloc failure"); k->rsa = rsa; break; case KEY_DSA: @@ -71,6 +73,9 @@ key_new(int type) dsa->q = BN_new(); dsa->g = BN_new(); dsa->pub_key = BN_new(); + if (dsa == NULL || dsa->p == NULL || dsa->q == NULL || + dsa->g == NULL || dsa->pub_key == NULL) + fatal("key_new: malloc failure"); k->dsa = dsa; break; case KEY_UNSPEC: -- cgit v1.2.3