d/freebsd-nq
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Handle partial writes

Browse Source 
Reported by:	ed
Reviewed by:	cognet
Approved by:	cognet
This commit is contained in:
Pietro Cerutti 2017-06-13 13:26:50 +00:00
parent dadfd1ed33
commit e9dd5a0c73
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
usr.bin/yes/yes.c
View File

@ -55,6 +55,8 @@ main(int argc, char **argv)
char * exp = y;
size_t buflen = 0;
size_t explen = sizeof(y);
size_t more;
ssize_t ret;
if (caph_limit_stdio() < 0 || (cap_enter() < 0 && errno != ENOSYS))
err(1, "capsicum");
@ -77,8 +79,10 @@ main(int argc, char **argv)
explen = buflen;
}
while (write(STDOUT_FILENO, exp, explen) > 0)
;
more = explen;
while ((ret = write(STDOUT_FILENO, exp + (explen - more), more)) > 0)
if ((more -= ret) == 0)
more = explen;
err(1, "stdout");
/*NOTREACHED*/