select: fix 64-bit timeout truncation on pre-time64 kernels

[musl] / src / stdio / getdelim.c

diff --git a/src/stdio/getdelim.c b/src/stdio/getdelim.c
index f770d20..df11444 100644 (file)
--- a/src/stdio/getdelim.c
+++ b/src/stdio/getdelim.c
@@ -1,59 +1,83 @@
 #include "stdio_impl.h"
+#include <string.h>
+#include <stdlib.h>
+#include <inttypes.h>
+#include <errno.h>
 
-#define MIN(a,b) ((a)<(b) ? (a) : (b))
-
-ssize_t getdelim(char **s, size_t *n, int delim, FILE *f)
+ssize_t getdelim(char **restrict s, size_t *restrict n, int delim, FILE *restrict f)
 {
        char *tmp;
        unsigned char *z;
        size_t k;
        size_t i=0;
+       int c;
+
+       FLOCK(f);
 
        if (!n || !s) {
+               f->mode |= f->mode-1;
+               f->flags |= F_ERR;
+               FUNLOCK(f);
                errno = EINVAL;
                return -1;
        }
 
        if (!*s) *n=0;
 
-       FLOCK(f);
-
-       while (!feof(f)) {
-               z = memchr(f->rpos, delim, f->rend - f->rpos);
-               k = z ? z - f->rpos + 1 : f->rend - f->rpos;
+       for (;;) {
+               if (f->rpos != f->rend) {
+                       z = memchr(f->rpos, delim, f->rend - f->rpos);
+                       k = z ? z - f->rpos + 1 : f->rend - f->rpos;
+               } else {
+                       z = 0;
+                       k = 0;
+               }
                if (i+k >= *n) {
-                       if (k >= SIZE_MAX-i) goto oom;
-                       *n = i+k+1;
-                       if (*n < SIZE_MAX/2) *n *= 2;
-                       tmp = realloc(*s, *n);
+                       size_t m = i+k+2;
+                       if (!z && m < SIZE_MAX/4) m += m/2;
+                       tmp = realloc(*s, m);
                        if (!tmp) {
-                               *n = i+k+1;
-                               tmp = realloc(*s, *n);
-                               if (!tmp) goto oom;
+                               m = i+k+2;
+                               tmp = realloc(*s, m);
+                               if (!tmp) {
+                                       /* Copy as much as fits and ensure no
+                                        * pushback remains in the FILE buf. */
+                                       k = *n-i;
+                                       memcpy(*s+i, f->rpos, k);
+                                       f->rpos += k;
+                                       f->mode |= f->mode-1;
+                                       f->flags |= F_ERR;
+                                       FUNLOCK(f);
+                                       errno = ENOMEM;
+                                       return -1;
+                               }
                        }
                        *s = tmp;
+                       *n = m;
+               }
+               if (k) {
+                       memcpy(*s+i, f->rpos, k);
+                       f->rpos += k;
+                       i += k;
                }
-               memcpy(*s+i, f->rpos, k);
-               f->rpos += k;
-               i += k;
                if (z) break;
-               __underflow(f);
+               if ((c = getc_unlocked(f)) == EOF) {
+                       if (!i || !feof(f)) {
+                               FUNLOCK(f);
+                               return -1;
+                       }
+                       break;
+               }
+               /* If the byte read by getc won't fit without growing the
+                * output buffer, push it back for next iteration. */
+               if (i+1 >= *n) *--f->rpos = c;
+               else if (((*s)[i++] = c) == delim) break;
        }
        (*s)[i] = 0;
-       if (feof(f) || ferror(f)) {
-               FUNLOCK(f);
-               return -1;
-       }
 
        FUNLOCK(f);
 
-       if (i > SSIZE_MAX) {
-               errno = EOVERFLOW;
-               return -1;
-       }
-
        return i;
-oom:
-       errno = ENOMEM;
-       return -1;
 }
+
+weak_alias(getdelim, __getdelim);