Free a malloc()ed ungetc buffer.

[gnulib.git] / lib / fpurge.c

diff --git a/lib/fpurge.c b/lib/fpurge.c
index c4572fa..0e8c164 100644 (file)
--- a/lib/fpurge.c
+++ b/lib/fpurge.c
@@ -29,6 +29,12 @@ fpurge (FILE *fp)
 #if defined _IO_ferror_unlocked     /* GNU libc, BeOS */
   fp->_IO_read_end = fp->_IO_read_ptr;
   fp->_IO_write_ptr = fp->_IO_write_base;
+  /* Avoid memory leak when there is an active ungetc buffer.  */
+  if (fp->_IO_save_base != NULL)
+    {
+      free (fp->_IO_save_base);
+      fp->_IO_save_base = NULL;
+    }
   return 0;
 #elif defined __sferror             /* FreeBSD, NetBSD, OpenBSD, MacOS X, Cygwin */
   fp->_p = fp->_bf._base;
@@ -36,6 +42,20 @@ fpurge (FILE *fp)
   fp->_w = ((fp->_flags & (__SLBF | __SNBF)) == 0 /* fully buffered? */
            ? fp->_bf._size
            : 0);
+  /* Avoid memory leak when there is an active ungetc buffer.  */
+# if defined __NetBSD__ || defined __OpenBSD__ /* NetBSD, OpenBSD */
+   /* See <http://cvsweb.netbsd.org/bsdweb.cgi/src/lib/libc/stdio/fileext.h?rev=HEAD&content-type=text/x-cvsweb-markup>
+      and <http://www.openbsd.org/cgi-bin/cvsweb/src/lib/libc/stdio/fileext.h?rev=HEAD&content-type=text/x-cvsweb-markup> */
+#  define fp_ub ((struct { struct __sbuf _ub; } *) fp->_ext._base)->_ub
+# else                                         /* FreeBSD, MacOS X, Cygwin */
+#  define fp_ub fp->_ub
+# endif
+  if (fp_ub._base != NULL)
+    {
+      if (fp_ub._base != fp->_ubuf)
+       free (fp_ub._base);
+      fp_ub._base = NULL;
+    }
   return 0;
 #elif defined _IOERR                /* AIX, HP-UX, IRIX, OSF/1, Solaris, mingw */
   fp->_ptr = fp->_base;
@@ -43,6 +63,6 @@ fpurge (FILE *fp)
     fp->_cnt = 0;
   return 0;
 #else
- #error "Please port gnulib fpurge.c to your platform!"
+ #error "Please port gnulib fpurge.c to your platform! Look at the definitions of fflush, setvbuf and ungetc on your system, then report this to bug-gnulib."
 #endif
 }