Added forgotten mem check.

1 files changed, 7 insertions, 1 deletions

diff --git a/src/Backtrace/main.c b/src/Backtrace/main.c
index c1078c0b..9be93ca5 100644
--- a/src/Backtrace/main.c
+++ b/src/Backtrace/main.c
@@ -148,12 +148,18 @@ int main(int argc, char **argv)
 
     if (size > FILE_SIZE_LIMIT)
     {
-      fprintf(stderr, "Input file too big (%zd). Maximum size is %zd", 
+      fprintf(stderr, "Input file too big (%zd). Maximum size is %zd.\n", 
               size, FILE_SIZE_LIMIT);
       exit(EX_IOERR);
     }
 
     bttext = malloc(size + 1);
+    if (!bttext)
+    {
+      fputs("malloc failed", stderr);
+      exit(EX_OSERR);
+    }
+
     if (1 != fread(bttext, size, 1, fp))
     {
       fprintf(stderr, "Unable to read from '%s'.\n", arguments.filename);