numfmt: fix bug when handling broken pipe

uutils · Dec 31, 2022 · 6203822 · 6203822
1 parent 5d15476
commit 6203822
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/uu/numfmt/src/numfmt.rs b/src/uu/numfmt/src/numfmt.rs
@@ -54,10 +54,13 @@ where
     R: BufRead,
 {
     let mut lines = input.lines();
-    lines
-        .by_ref()
-        .take(options.header)
-        .for_each(|line| println!("{}", line.unwrap()));
+
+    for line in lines.by_ref().take(options.header) {
+        match line {
+            Ok(l) => println!("{}\n", l),
+            Err(e) => return Err(Box::new(NumfmtError::IoError(e.to_string()))),
+        };
+    }
 
     let mut has_failed_flag = false;