Check result of write (thans to Alexander Kapshuk <alexander.kapshuk@…

…gmail)
shuowpro · Sep 19, 2016 · 89826f4 · 89826f4
1 parent 912575a
commit 89826f4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/cat.c b/cat.c
@@ -9,8 +9,12 @@ cat(int fd)
 {
   int n;
 
-  while((n = read(fd, buf, sizeof(buf))) > 0)
-    write(1, buf, n);
+  while((n = read(fd, buf, sizeof(buf))) > 0) {
+    if (write(1, buf, n) != n) {
+      printf(1, "cat: write error\n");
+      exit();
+    }
+  }
   if(n < 0){
     printf(1, "cat: read error\n");
     exit();