From d8c8bf9964ec4ac733359db6f54b084b476d32ef Mon Sep 17 00:00:00 2001
From: n0vad3v <n0vad3v@riseup.net>
Date: Tue, 25 Feb 2020 22:12:24 +0800
Subject: [PATCH] Fix read file type bug.

---
 webp-server.go | 34 +++++++++++++++++++++++++++++-----
 1 file changed, 29 insertions(+), 5 deletions(-)

diff --git a/webp-server.go b/webp-server.go
index 75888bb..f8facff 100644
--- a/webp-server.go
+++ b/webp-server.go
@@ -10,6 +10,7 @@ import (
 	"image/png"
 	"io/ioutil"
 	"log"
+	"net/http"
 	"os"
 	"path"
 	"strconv"
@@ -32,11 +33,19 @@ var configPath string
 func webpEncoder(p1, p2 string, quality float32) {
 	var buf bytes.Buffer
 	var img image.Image
-	data, _ := ioutil.ReadFile(p1)
-	if strings.Contains(p1, "jpg") || strings.Contains(p1, "jpeg") {
-		img, _ = jpeg.Decode(bytes.NewReader(data))
-	} else if strings.Contains(p1, "png") {
-		img, _ = png.Decode(bytes.NewReader(data))
+
+	file_data, err := os.Open(p1)
+	if err != nil {
+		panic(err)
+	}
+	img_data, _ := ioutil.ReadFile(p1)
+	defer file_data.Close()
+
+	contentType, err := GetFileContentType(file_data)
+	if strings.Contains(contentType, "jpg") || strings.Contains(contentType, "jpeg") {
+		img, _ = jpeg.Decode(bytes.NewReader(img_data))
+	} else if strings.Contains(contentType, "png") {
+		img, _ = png.Decode(bytes.NewReader(img_data))
 	}
 
 	if err := webp.Encode(&buf, img, &webp.Options{Lossless: true, Quality: quality}); err != nil {
@@ -188,3 +197,18 @@ func Find(slice []string, val string) (int, bool) {
 	}
 	return -1, false
 }
+
+func GetFileContentType(out *os.File) (string, error) {
+
+	buffer := make([]byte, 512)
+
+	_, err := out.Read(buffer)
+	if err != nil {
+		return "", err
+	}
+
+	// content-type by returning "application/octet-stream" if no others seemed to match.
+	contentType := http.DetectContentType(buffer)
+
+	return contentType, nil
+}