From 13a0ea225fb89fe1928303faf9fa23afe355ab97 Mon Sep 17 00:00:00 2001 From: Tavian Barnes Date: Fri, 12 Mar 2010 17:17:06 -0500 Subject: Check for malloc() failures. --- dimension/tokenize.c | 2 ++ 1 file changed, 2 insertions(+) (limited to 'dimension/tokenize.c') diff --git a/dimension/tokenize.c b/dimension/tokenize.c index bccccc9..2bead0b 100644 --- a/dimension/tokenize.c +++ b/dimension/tokenize.c @@ -226,6 +226,8 @@ dmnsn_include_buffer(int token, dmnsn_token_buffer *prev, dmnsn_error(DMNSN_SEVERITY_HIGH, "Couldn't allocate space for filename."); char *localdir = dirname(filename_copy); char *local_include = malloc(strlen(localdir) + strlen(include) + 2); + if (!local_include) + dmnsn_error(DMNSN_SEVERITY_HIGH, "Couldn't allocate space for filename."); strcpy(local_include, localdir); strcat(local_include, "/"); strcat(local_include, include); -- cgit v1.2.3