From 34a4ebfeaa26c27a4ec031ba70383ec8d27489a0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Motiejus=20Jak=C5=A1tys?= <motiejus@jakstys.lt>
Date: Thu, 19 Aug 2021 16:32:22 +0300
Subject: [PATCH] additional error check

---
 main.go          | 2 +-
 rootfs/rootfs.go | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index ee99746..73c7801 100644
--- a/main.go
+++ b/main.go
@@ -31,7 +31,7 @@ func main() {
 
 	c := &command{flattener: rootfs.Flatten, Stdout: os.Stdout}
 	if err := c.execute(os.Args[1], os.Args[2]); err != nil {
-		fmt.Printf("Error: %v", err)
+		fmt.Printf("Error: %v\n", err)
 		os.Exit(1)
 	}
 	os.Exit(0)
diff --git a/rootfs/rootfs.go b/rootfs/rootfs.go
index 80cf636..f8aee15 100644
--- a/rootfs/rootfs.go
+++ b/rootfs/rootfs.go
@@ -52,6 +52,9 @@ func Flatten(rd io.ReadSeeker, w io.Writer) (_err error) {
 		if err == io.EOF {
 			break
 		}
+		if err != nil {
+			return err
+		}
 		if hdr.Typeflag != tar.TypeReg {
 			continue
 		}