mirror of
https://git.hardenedbsd.org/hardenedbsd/HardenedBSD.git
synced 2025-01-01 00:18:15 +01:00
1) Fix hang at the end of line buffer (PR 95715)
2) Localize PR: 95715 Submitted by: Li-Lun Wang <llwang@infor.org> (slightly edited by me)
This commit is contained in:
parent
1fe5a490ab
commit
1107d0af30
@ -54,6 +54,7 @@ __FBSDID("$FreeBSD$");
|
||||
#include <errno.h>
|
||||
#include <fcntl.h>
|
||||
#include <limits.h>
|
||||
#include <locale.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
@ -84,6 +85,9 @@ main(int argc, char *argv[])
|
||||
random_type = RANDOM_TYPE_UNSET;
|
||||
random_exit = randomize_lines = random_type = unbuffer_output = 0;
|
||||
unique_output = 1;
|
||||
|
||||
(void)setlocale(LC_CTYPE, "");
|
||||
|
||||
while ((ch = getopt(argc, argv, "ef:hlruUw")) != -1)
|
||||
switch (ch) {
|
||||
case 'e':
|
||||
|
@ -93,7 +93,7 @@ rand_node_append(struct rand_node *n)
|
||||
int
|
||||
randomize_fd(int fd, int type, int unique, double denom)
|
||||
{
|
||||
u_char *buf, *p;
|
||||
u_char *buf;
|
||||
u_int numnode, j, selected, slen;
|
||||
struct rand_node *n, *prev;
|
||||
int bufleft, eof, fndstr, ret;
|
||||
@ -148,30 +148,29 @@ randomize_fd(int fd, int type, int unique, double denom)
|
||||
fndstr = 0;
|
||||
}
|
||||
} else {
|
||||
p = (u_char *)realloc(buf, buflen * 2);
|
||||
if (p == NULL)
|
||||
buflen *= 2;
|
||||
buf = (u_char *)realloc(buf, buflen);
|
||||
if (buf == NULL)
|
||||
err(1, "realloc");
|
||||
|
||||
buf = p;
|
||||
if (!eof) {
|
||||
len = read(fd, &buf[i], buflen);
|
||||
len = read(fd, &buf[i], buflen - i);
|
||||
if (len == -1)
|
||||
err(1, "read");
|
||||
else if (len == 0) {
|
||||
eof++;
|
||||
break;
|
||||
} else if (len < (ssize_t)(buflen - i))
|
||||
buflen = (size_t)len;
|
||||
buflen = i + (size_t)len;
|
||||
|
||||
bufleft = (int)len;
|
||||
}
|
||||
|
||||
buflen *= 2;
|
||||
}
|
||||
}
|
||||
|
||||
if ((type == RANDOM_TYPE_LINES && buf[i] == '\n') ||
|
||||
(type == RANDOM_TYPE_WORDS && isspace((int)buf[i])) ||
|
||||
(type == RANDOM_TYPE_WORDS && isspace(buf[i])) ||
|
||||
(eof && i == buflen - 1)) {
|
||||
make_token:
|
||||
n = rand_node_allocate();
|
||||
|
Loading…
Reference in New Issue
Block a user