cmd/airliftd: fix serving small files (resolves #35)

Resolved a misuse of io.ReadFull used to detect MIME type. io.EOF may be returned if file is smaller than provided buffer.
moshee · Mar 21, 2021 · a15ef7c · a15ef7c
1 parent d352ba2
commit a15ef7c
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/cmd/airliftd/server.go b/cmd/airliftd/server.go
@@ -443,10 +443,20 @@ func getFile(g *gas.Gas) (int, gas.Outputter) {
 	if err != nil {
 		return 500, out.Error(g, err)
 	}
-
 	defer f.Close()
 
-	buf := make([]byte, 512)
+	fi, err := f.Stat()
+	if err != nil {
+		return 500, out.Error(g, err)
+	}
+
+	bufsize := 512
+	fsize := int(fi.Size())
+	if fsize < bufsize {
+		bufsize = fsize
+	}
+
+	buf := make([]byte, bufsize)
 	_, err = io.ReadFull(f, buf)
 	if err != nil {
 		return 500, out.Error(g, err)