1#include "stdio_impl.h"
2#include <errno.h>
3#include <inttypes.h>
4#include <string.h>
5
6#define MIN(a, b) ((a) < (b) ? (a) : (b))
7
8ssize_t getdelim(char** restrict s, size_t* restrict n, int delim, FILE* restrict f) {
9    char* tmp;
10    unsigned char* z;
11    size_t k;
12    size_t i = 0;
13    int c;
14
15    FLOCK(f);
16
17    if (!n || !s) {
18        f->flags |= F_ERR;
19        FUNLOCK(f);
20        errno = EINVAL;
21        return -1;
22    }
23
24    if (!*s)
25        *n = 0;
26
27    for (;;) {
28        z = memchr(f->rpos, delim, f->rend - f->rpos);
29        k = z ? z - f->rpos + 1 : f->rend - f->rpos;
30        if (i + k + 1 >= *n) {
31            if (k >= SIZE_MAX / 2 - i)
32                goto oom;
33            size_t m = i + k + 2;
34            if (!z && m < SIZE_MAX / 4)
35                m += m / 2;
36            tmp = realloc(*s, m);
37            if (!tmp) {
38                m = i + k + 2;
39                tmp = realloc(*s, m);
40                if (!tmp)
41                    goto oom;
42            }
43            *s = tmp;
44            *n = m;
45        }
46        memcpy(*s + i, f->rpos, k);
47        f->rpos += k;
48        i += k;
49        if (z)
50            break;
51        if ((c = getc_unlocked(f)) == EOF) {
52            if (!i || !feof(f)) {
53                FUNLOCK(f);
54                return -1;
55            }
56            break;
57        }
58        if (((*s)[i++] = c) == delim)
59            break;
60    }
61    (*s)[i] = 0;
62
63    FUNLOCK(f);
64
65    return i;
66oom:
67    f->flags |= F_ERR;
68    FUNLOCK(f);
69    errno = ENOMEM;
70    return -1;
71}
72
73weak_alias(getdelim, __getdelim);
74