fix the error when providing a file name

lazygyu · Dec 9, 2022 · c3a401c · c3a401c
1 parent 009e6bb
commit c3a401c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -46,11 +46,16 @@ func main() {
     filename := args[0]
 
     // open file
-    f, err := os.Open(filename)
+    fd, err := os.Open(filename)
     if err != nil {
       log.Fatal("cannot open the file")
     }
-    defer f.Close()
+    fStat, err := fd.Stat()
+    if err != nil || fStat.Size() == 0 {
+      log.Fatal("Invalid file")
+    }
+    defer fd.Close()
+    f = bufio.NewReader(fd)
   }
 
 	// load img