From cfed6fdddfc5d1dafc943e01a825e6fbb001f93a Mon Sep 17 00:00:00 2001 From: Renaud Fivet Date: Tue, 24 Mar 2015 17:49:12 +0800 Subject: [PATCH] Extra allocation check during initialisation. --- main.c | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/main.c b/main.c index 354bbee..020cf51 100644 --- a/main.c +++ b/main.c @@ -275,7 +275,12 @@ int main(int argc, char **argv) unqname(bname); /* set this to inactive */ - bp = bfind(bname, TRUE, 0); + bp = bfind( bname, TRUE, 0) ; + if( bp == NULL) { + fputs( "Buffer creation failed!\n", stderr) ; + exit( EXIT_FAILURE) ; + } + strncpy( bp->b_fname, argv[ carg], sizeof bp->b_fname - 1) ; /* max filename length limited to NFILEN - 1 (79) */ bp->b_fname[ sizeof bp->b_fname - 1] = 0 ; bp->b_active = FALSE;