RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-10-31 17:32:44 +00:00
Code Issues Activity Actions

/bin/cp: Handle partial write() case.

Browse source
This commit is contained in:
Andreas Kling 2019-01-27 10:04:53 +01:00
parent 32755aeb60
commit c4c475b49d
1 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

16
Userland/cp.cpp
View file

@ -29,12 +29,18 @@ int main(int argc, char** argv)
}
if (nread == 0)
break;
ssize_t nwritten = write(dst_fd, buffer, nread);
if (nwritten < 0) {
perror("write dst");
return 1;
ssize_t remaining_to_write = nread;
char* bufptr = buffer;
while (remaining_to_write) {
ssize_t nwritten = write(dst_fd, bufptr, remaining_to_write);
if (nwritten < 0) {
perror("write dst");
return 1;
}
assert(nwritten > 0);
remaining_to_write -= nwritten;
bufptr += nwritten;
}
assert(nwritten != 0);
}
close(src_fd);
close(dst_fd);