1 files changed, 78 insertions, 0 deletions
diff --git a/stdio-common/tst-getline-enomem.c b/stdio-common/tst-getline-enomem.c
new file mode 100644
index 0000000000..7fc70ea9b5
--- /dev/null
+++ b/stdio-common/tst-getline-enomem.c
@@ -0,0 +1,78 @@
+/* Test getline: ENOMEM on allocation failure.
+   Copyright (C) 2024 Free Software Foundation, Inc.
+   This file is part of the GNU C Library.
+
+   The GNU C Library is free software; you can redistribute it and/or
+   modify it under the terms of the GNU Lesser General Public
+   License as published by the Free Software Foundation; either
+   version 2.1 of the License, or (at your option) any later version.
+
+   The GNU C Library is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+   Lesser General Public License for more details.
+
+   You should have received a copy of the GNU Lesser General Public
+   License along with the GNU C Library; if not, see
+   <https://www.gnu.org/licenses/>.  */
+
+#include <errno.h>
+#include <mcheck.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/resource.h>
+
+#include <support/check.h>
+#include <support/test-driver.h>
+
+/* Produce a stream of test data based on data in COOKIE (ignored),
+   storing up to SIZE bytes in BUF.  */
+
+static ssize_t
+io_read (void *cookie, char *buf, size_t size)
+{
+  memset (buf, 'x', size);
+  return size;
+}
+
+/* Set up a test stream with fopencookie.  */
+
+static FILE *
+open_test_stream (void)
+{
+  static cookie_io_functions_t io_funcs = { .read = io_read };
+  static int cookie;
+  FILE *fp = fopencookie (&cookie, "r", io_funcs);
+  TEST_VERIFY_EXIT (fp != NULL);
+  return fp;
+}
+
+int
+do_test (void)
+{
+  FILE *fp;
+  char *lineptr = NULL;
+  size_t size = 0;
+  ssize_t ret;
+  mtrace ();
+  /* Test ENOMEM (and error indicator for stream set) for memory
+     allocation failure.  */
+  verbose_printf ("Testing memory allocation failure\n");
+  fp = open_test_stream ();
+  struct rlimit limit;
+  TEST_VERIFY_EXIT (getrlimit (RLIMIT_AS, &limit) == 0);
+  limit.rlim_cur = 32 * 1024 * 1024;
+  TEST_VERIFY_EXIT (setrlimit (RLIMIT_AS, &limit) == 0);
+  errno = 0;
+  ret = getline (&lineptr, &size, fp);
+  TEST_COMPARE (ret, -1);
+  TEST_COMPARE (errno, ENOMEM);
+  TEST_COMPARE (!!ferror (fp), 1);
+  TEST_COMPARE (feof (fp), 0);
+  free (lineptr);
+  fclose (fp);
+  return 0;
+}
+
+#include <support/test-driver.c>