diff options
author | Ron Yorston <rmy@pobox.com> | 2021-01-23 13:22:33 +0000 |
---|---|---|
committer | Denys Vlasenko <vda.linux@googlemail.com> | 2021-02-03 20:43:07 +0100 |
commit | 6ad38d66d08a2646c6423b967a790656ae026c95 (patch) | |
tree | a8866dbecafeb1679135008d7e52f245adb0e8e7 | |
parent | c849e72c0b216578f96b763529c5e58056d0662c (diff) | |
download | busybox-w32-6ad38d66d08a2646c6423b967a790656ae026c95.tar.gz busybox-w32-6ad38d66d08a2646c6423b967a790656ae026c95.tar.bz2 busybox-w32-6ad38d66d08a2646c6423b967a790656ae026c95.zip |
libbb: code shrink fgets_str
Use a NULL value of maxsz_p to indicate to xmalloc_fgets_internal()
that the caller doesn't care about the maximum size of the buffer.
This allows the default maximum size to be set once in
xmalloc_fgets_internal() instead of separately in each caller.
function old new delta
xmalloc_fgets_internal 273 287 +14
xmalloc_fgets_str 30 9 -21
xmalloc_fgetline_str 33 12 -21
xmalloc_fgets_str_len 38 10 -28
------------------------------------------------------------------------------
(add/remove: 0/0 grow/shrink: 1/3 up/down: 14/-70) Total: -56 bytes
Signed-off-by: Ron Yorston <rmy@pobox.com>
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
-rw-r--r-- | libbb/fgets_str.c | 17 |
1 files changed, 5 insertions, 12 deletions
diff --git a/libbb/fgets_str.c b/libbb/fgets_str.c index 1a7f2e9e0..c884ef8af 100644 --- a/libbb/fgets_str.c +++ b/libbb/fgets_str.c | |||
@@ -17,7 +17,7 @@ static char *xmalloc_fgets_internal(FILE *file, const char *terminating_string, | |||
17 | int linebufsz = 0; | 17 | int linebufsz = 0; |
18 | int idx = 0; | 18 | int idx = 0; |
19 | int ch; | 19 | int ch; |
20 | size_t maxsz = *maxsz_p; | 20 | size_t maxsz = maxsz_p ? *maxsz_p : INT_MAX - 4095; |
21 | 21 | ||
22 | while (1) { | 22 | while (1) { |
23 | ch = fgetc(file); | 23 | ch = fgetc(file); |
@@ -53,7 +53,8 @@ static char *xmalloc_fgets_internal(FILE *file, const char *terminating_string, | |||
53 | /* Grow/shrink *first*, then store NUL */ | 53 | /* Grow/shrink *first*, then store NUL */ |
54 | linebuf = xrealloc(linebuf, idx + 1); | 54 | linebuf = xrealloc(linebuf, idx + 1); |
55 | linebuf[idx] = '\0'; | 55 | linebuf[idx] = '\0'; |
56 | *maxsz_p = idx; | 56 | if (maxsz_p) |
57 | *maxsz_p = idx; | ||
57 | return linebuf; | 58 | return linebuf; |
58 | } | 59 | } |
59 | 60 | ||
@@ -63,23 +64,15 @@ static char *xmalloc_fgets_internal(FILE *file, const char *terminating_string, | |||
63 | * Return NULL if EOF is reached immediately. */ | 64 | * Return NULL if EOF is reached immediately. */ |
64 | char* FAST_FUNC xmalloc_fgets_str(FILE *file, const char *terminating_string) | 65 | char* FAST_FUNC xmalloc_fgets_str(FILE *file, const char *terminating_string) |
65 | { | 66 | { |
66 | size_t maxsz = INT_MAX - 4095; | 67 | return xmalloc_fgets_internal(file, terminating_string, 0, NULL); |
67 | return xmalloc_fgets_internal(file, terminating_string, 0, &maxsz); | ||
68 | } | 68 | } |
69 | 69 | ||
70 | char* FAST_FUNC xmalloc_fgets_str_len(FILE *file, const char *terminating_string, size_t *maxsz_p) | 70 | char* FAST_FUNC xmalloc_fgets_str_len(FILE *file, const char *terminating_string, size_t *maxsz_p) |
71 | { | 71 | { |
72 | size_t maxsz; | ||
73 | |||
74 | if (!maxsz_p) { | ||
75 | maxsz = INT_MAX - 4095; | ||
76 | maxsz_p = &maxsz; | ||
77 | } | ||
78 | return xmalloc_fgets_internal(file, terminating_string, 0, maxsz_p); | 72 | return xmalloc_fgets_internal(file, terminating_string, 0, maxsz_p); |
79 | } | 73 | } |
80 | 74 | ||
81 | char* FAST_FUNC xmalloc_fgetline_str(FILE *file, const char *terminating_string) | 75 | char* FAST_FUNC xmalloc_fgetline_str(FILE *file, const char *terminating_string) |
82 | { | 76 | { |
83 | size_t maxsz = INT_MAX - 4095; | 77 | return xmalloc_fgets_internal(file, terminating_string, 1, NULL); |
84 | return xmalloc_fgets_internal(file, terminating_string, 1, &maxsz); | ||
85 | } | 78 | } |