1 files changed, 151 insertions, 0 deletions
diff --git a/gnulib-tests/test-getndelim2.c b/gnulib-tests/test-getndelim2.c
new file mode 100644
index 0000000..bedbb03
--- /dev/null
+++ b/gnulib-tests/test-getndelim2.c
@@ -0,0 +1,151 @@
+/* Test of getndelim2() function.
+   Copyright (C) 2008-2022 Free Software Foundation, Inc.
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 3, or (at your option)
+   any later version.
+
+   This program 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 General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, see <https://www.gnu.org/licenses/>.  */
+
+/* Written by Eric Blake <ebb9@byu.net>, 2008.  */
+
+#include <config.h>
+
+#include "getndelim2.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "macros.h"
+
+int
+main (void)
+{
+  FILE *f;
+  char *line = NULL;
+  size_t len = 0;
+  ssize_t result;
+
+  /* Create test file.  */
+  f = fopen ("test-getndelim2.txt", "wb+");
+  if (!f || fwrite ("a\nbc\nd\0f", 1, 8, f) != 8)
+    {
+      fputs ("Failed to create sample file.\n", stderr);
+      remove ("test-getndelim2.txt");
+      return 1;
+    }
+  rewind (f);
+
+  /* Unlimited lines.  */
+
+  /* Test initial allocation, which must include trailing NUL.  */
+  result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', '\n', f);
+  ASSERT (result == 2);
+  ASSERT (strcmp (line, "a\n") == 0);
+  ASSERT (2 < len);
+
+  /* Test growth of buffer, must not leak.  */
+  free (line);
+  line = malloc (1);
+  len = 0;
+  result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, EOF, '\n', f);
+  ASSERT (result == 3);
+  ASSERT (strcmp (line, "bc\n") == 0);
+  ASSERT (3 < len);
+
+  /* Test embedded NULs and EOF behavior.  */
+  result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', EOF, f);
+  ASSERT (result == 3);
+  ASSERT (memcmp (line, "d\0f", 4) == 0);
+  ASSERT (3 < len);
+
+  result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', EOF, f);
+  ASSERT (result == -1);
+
+  /* Using offset and nmax.  */
+
+  /* Limit growth to four bytes, including NUL, but parse to next 'd'.  */
+  free (line);
+  rewind (f);
+  line = malloc (8);
+  memset (line, 'e', 8);
+  len = 8;
+  result = getndelim2 (&line, &len, 6, 10, 'd', 'd', f);
+  ASSERT (result == 3);
+  ASSERT (10 == len);
+  ASSERT (strcmp (line, "eeeeeea\nb") == 0);
+
+  /* No change if offset larger than limit.  */
+  result = getndelim2 (&line, &len, len, 1, EOF, EOF, f);
+  ASSERT (result == -1);
+  ASSERT (10 == len);
+  ASSERT (strcmp (line, "eeeeeea\nb") == 0);
+
+  /* Consume to end of file, including embedded NUL.  */
+  result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, EOF, EOF, f);
+  ASSERT (result == 2);
+  ASSERT (10 == len);
+  ASSERT (memcmp (line, "\0f\0eeea\nb", 10) == 0);
+
+  result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', '\r', f);
+  ASSERT (result == -1);
+
+  /* Larger file size.  */
+  rewind (f);
+  {
+    int i;
+    for (i = 0; i < 16; i++)
+      fprintf (f, "%500x%c", i + 0u, i % 2 ? '\n' : '\r');
+  }
+  rewind (f);
+  {
+    char buffer[502];
+
+    result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', '\r', f);
+    ASSERT (result == 501);
+    ASSERT (501 < len);
+    memset (buffer, ' ', 499);
+    buffer[499] = '0';
+    buffer[500] = '\r';
+    buffer[501] = '\0';
+    ASSERT (strcmp (buffer, line) == 0);
+
+    result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', '\r', f);
+    ASSERT (result == 501);
+    ASSERT (501 < len);
+    buffer[499] = '1';
+    buffer[500] = '\n';
+    ASSERT (strcmp (buffer, line) == 0);
+
+    result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, 'g', 'f', f);
+    ASSERT (result == 501 * 14 - 1);
+    ASSERT (501 * 14 <= len);
+    buffer[499] = 'f';
+    buffer[500] = '\0';
+    ASSERT (strcmp (buffer, line + 501 * 13) == 0);
+
+    result = getndelim2 (&line, &len, 501 * 14 - 1, GETNLINE_NO_LIMIT,
+                         EOF, EOF, f);
+    ASSERT (result == 1);
+    buffer[500] = '\n';
+    ASSERT (strcmp (buffer, line + 501 * 13) == 0);
+
+    result = getndelim2 (&line, &len, 501 * 14 - 1, GETNLINE_NO_LIMIT,
+                         EOF, EOF, f);
+    buffer[500] = '\0';
+    ASSERT (strcmp (buffer, line + 501 * 13) == 0);
+    ASSERT (result == -1);
+  }
+
+  fclose (f);
+  remove ("test-getndelim2.txt");
+  return 0;
+}