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