Fix more potential seg-faults in gprof.

[deliverable/binutils-gdb.git] / gprof / source.c

diff --git a/gprof/source.c b/gprof/source.c
index 3943f33b1969cbdf3053a4d98a0e05c4cc9df470..6a977a6fa8ed05d1c352fe72e53e488cc75ba98a 100644 (file)
--- a/gprof/source.c
+++ b/gprof/source.c
@@ -1,6 +1,6 @@
 /* source.c - Keep track of source files.
 
-   Copyright 2000, 2001, 2002, 2004, 2007 Free Software Foundation, Inc.
+   Copyright (C) 2000-2016 Free Software Foundation, Inc.
 
    This file is part of GNU Binutils.
 
@@ -237,7 +237,7 @@ annotate_source (Source_File *sf, unsigned int max_width,
       fprintf (ofp, _("*** File %s:\n"), sf->name);
     }
 
-  annotation = xmalloc (max_width + 1);
+  annotation = (char *) xmalloc (max_width + 1);
   line_num = 1;
   new_line = TRUE;
 
@@ -259,5 +259,6 @@ annotate_source (Source_File *sf, unsigned int max_width,
     }
 
   free (annotation);
+  fclose (ifp);
   return ofp;
 }