From 28c412eb49c0284e06e7ff9e02218a75a21c51d6 Mon Sep 17 00:00:00 2001 From: Assar Westerlund Date: Tue, 11 Nov 1997 20:49:40 +0000 Subject: [PATCH] (krb5_build_ap_req): check return value from malloc. git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@3941 ec53bebd-3082-4978-b11e-865c3cabbd6b --- lib/krb5/build_ap_req.c | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/lib/krb5/build_ap_req.c b/lib/krb5/build_ap_req.c index 9e7742ba0..5fd6a4246 100644 --- a/lib/krb5/build_ap_req.c +++ b/lib/krb5/build_ap_req.c @@ -47,6 +47,7 @@ krb5_build_ap_req (krb5_context context, krb5_data authenticator, krb5_data *ret) { + krb5_error_code ret = 0; AP_REQ ap; Ticket t; size_t len; @@ -71,9 +72,12 @@ krb5_build_ap_req (krb5_context context, ret->length = length_AP_REQ(&ap); ret->data = malloc(ret->length); - encode_AP_REQ((unsigned char *)ret->data + ret->length - 1, - ret->length, &ap, &len); + if(ret->data == NULL) + ret = ENOMEM; + else + encode_AP_REQ((unsigned char *)ret->data + ret->length - 1, + ret->length, &ap, &len); free_AP_REQ(&ap); - return 0; + return ret; }