diff --git a/libio/fmemopen.c b/libio/fmemopen.c index 3ab3e8d..1addfac 100644 --- a/libio/fmemopen.c +++ b/libio/fmemopen.c @@ -149,6 +149,7 @@ __fmemopen (void *buf, size_t len, const char *mode) { cookie_io_functions_t iof; fmemopen_cookie_t *c; + FILE *ret; c = (fmemopen_cookie_t *) calloc (sizeof (fmemopen_cookie_t), 1); if (c == NULL) @@ -209,7 +210,16 @@ __fmemopen (void *buf, size_t len, const char *mode) iof.seek = fmemopen_seek; iof.close = fmemopen_close; - return _IO_fopencookie (c, mode, iof); + ret = _IO_fopencookie (c, mode, iof); + + if (__glibc_unlikely (ret == NULL)) + { + /* BZ #18757 -- set EINVAL */ + __set_errno (EINVAL); + free (c); + } + + return ret; } libc_hidden_def (__fmemopen) versioned_symbol (libc, __fmemopen, fmemopen, GLIBC_2_22); diff --git a/libio/oldfmemopen.c b/libio/oldfmemopen.c index 8e35672..40432d1 100644 --- a/libio/oldfmemopen.c +++ b/libio/oldfmemopen.c @@ -204,6 +204,7 @@ __old_fmemopen (void *buf, size_t len, const char *mode) { cookie_io_functions_t iof; fmemopen_cookie_t *c; + FILE *ret; if (__glibc_unlikely (len == 0)) { @@ -259,7 +260,15 @@ __old_fmemopen (void *buf, size_t len, const char *mode) iof.seek = fmemopen_seek; iof.close = fmemopen_close; - return _IO_fopencookie (c, mode, iof); + ret = _IO_fopencookie (c, mode, iof); + if (__glibc_unlikely (ret == NULL)) + { + /* BZ 18757 -- set EINVAL */ + __set_errno (EINVAL); + free (c); + } + + return ret; } compat_symbol (libc, __old_fmemopen, fmemopen, GLIBC_2_2); #endif diff --git a/libio/test-fmemopen.c b/libio/test-fmemopen.c index 63ca89f..4d15b36 100644 --- a/libio/test-fmemopen.c +++ b/libio/test-fmemopen.c @@ -24,6 +24,28 @@ static char buffer[] = "foobar"; #include static int +do_bz18757 (void) +{ + char c = 0; + FILE *stream; + + errno = 0; + stream = fmemopen (&c, 1, "?"); + if (stream == NULL) + { + if (errno == EINVAL) + return 0; + + printf ("FAIL: errno = %i, but wanted EINVAL (%i)\n", errno, EINVAL); + return 1; + } + + printf ("FAIL: stream = %p, but wanted NULL\n", stream); + fclose (stream); + return 2; +} + +static int do_test (void) { int ch; @@ -44,7 +66,7 @@ do_test (void) fclose (stream); - return ret; + return ret + do_bz18757 (); } #define TEST_FUNCTION do_test ()