From 2c57d32ca1e1c651c23ddfb57572a6531ff30b09 Mon Sep 17 00:00:00 2001 From: Assar Westerlund Date: Mon, 29 Jan 2001 08:36:45 +0000 Subject: [PATCH] test two generated mallocs for success git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@9577 ec53bebd-3082-4978-b11e-865c3cabbd6b --- lib/asn1/gen_decode.c | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/lib/asn1/gen_decode.c b/lib/asn1/gen_decode.c index 242485b2b..8dc45b740 100644 --- a/lib/asn1/gen_decode.c +++ b/lib/asn1/gen_decode.c @@ -1,5 +1,5 @@ /* - * Copyright (c) 1997 - 2000 Kungliga Tekniska Högskolan + * Copyright (c) 1997 - 2001 Kungliga Tekniska Högskolan * (Royal Institute of Technology, Stockholm, Sweden). * All rights reserved. * @@ -131,7 +131,8 @@ decode_type (const char *name, const Type *t) if (0 && m->type->type == TType){ if(m->optional) fprintf (codefile, - "%s = malloc(sizeof(*%s));\n", s, s); + "%s = malloc(sizeof(*%s));\n" + "if(%s == NULL) return ENOMEM;\n", s, s, s); fprintf (codefile, "e = decode_seq_%s(p, len, %d, %d, %s, &l);\n", m->type->symbol->gen_name, @@ -179,8 +180,8 @@ decode_type (const char *name, const Type *t) "return ASN1_BAD_FORMAT;\n"); if (m->optional) fprintf (codefile, - "%s = malloc(sizeof(*%s));\n", - s, s); + "%s = malloc(sizeof(*%s));\n" + "if(%s == NULL) return ENOMEM;\n", s, s, s); decode_type (s, m->type); fprintf (codefile, "if(dce_fix){\n"