finish

main.go

@@ -11,9 +11,9 @@ import (
 	"image/png"
 	"log"
 	"net/http"
+	
 	"runtime/debug"
 	"strconv"
-	"strings"
 	"time"
 )
 
@@ -68,9 +68,9 @@ func embedLSB(img image.Image, message string) image.Image {
 	for i := 0; i < bounds.Dx(); i++ {
 		for j := 0; j < bounds.Dy(); j++ {
 			r, g, b, a := img.At(i, j).RGBA()
-			pixel := [4]uint8{uint8(r >> 8), uint8(g >> 8), uint8(b >> 8), uint8(a >> 8)}
+			pixel := [3]uint8{uint8(r >> 8), uint8(g >> 8), uint8(b >> 8)}
 
-			for k := 0; k < 4; k++ { // Loop untuk setiap channel warna (RGBA)
+			for k := 0; k < 3; k++ { // Loop untuk setiap channel warna (RGB)
 				if dataIndex < panjangPesan {
 					isiPixel := intToBinaryString(int(pixel[k]))
 					isiPixel = isiPixel[:7] + string(binaryPesan[dataIndex])
@@ -80,7 +80,7 @@ func embedLSB(img image.Image, message string) image.Image {
 				}
 			}
 
-			newImg.SetRGBA(i, j, color.RGBA{R: pixel[0], G: pixel[1], B: pixel[2], A: pixel[3]})
+			newImg.Set(i, j, color.RGBA{pixel[0], pixel[1], pixel[2], uint8(a >> 8)})
 		}
 	}
 	return newImg
@@ -94,10 +94,10 @@ func extractLSB(img image.Image) string {
 
 	for i := 0; i < bounds.Dx(); i++ {
 		for j := 0; j < bounds.Dy(); j++ {
-			r, g, b, a := img.At(i, j).RGBA()
-			pixel := [4]uint8{uint8(r >> 8), uint8(g >> 8), uint8(b >> 8), uint8(a >> 8)}
+			r, g, b, _ := img.At(i, j).RGBA()
+			pixel := [3]uint8{uint8(r >> 8), uint8(g >> 8), uint8(b >> 8)}
 
-			for k := 0; k < 4; k++ { // Loop untuk setiap channel warna (RGBA)
+			for k := 0; k < 3; k++ { // Loop untuk setiap channel warna (RGB)
 				binaryPixel := intToBinaryString(int(pixel[k]))
 				binaryMessage += string(binaryPixel[7])
 				dataIndex++
@@ -126,49 +126,27 @@ func uploadHandler(w http.ResponseWriter, r *http.Request) {
 	if r.Method == http.MethodPost {
 		start := time.Now()
 
-		file, fileHeader, err := r.FormFile("image")
+		file, _, err := r.FormFile("image")
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusBadRequest)
 			return
 		}
 		defer file.Close()
 
-		var img image.Image
-		var imgFormat string
-
-		switch strings.ToLower(filepath.Ext(fileHeader.Filename)) {
-		case ".png":
-			img, imgFormat, err = image.Decode(file)
-			if err != nil {
-				http.Error(w, err.Error(), http.StatusBadRequest)
-				return
-			}
-		case ".jpg", ".jpeg":
-			img, err = jpeg.Decode(file)
-			if err != nil {
-				http.Error(w, err.Error(), http.StatusBadRequest)
-				return
-			}
-			imgFormat = "jpeg"
-		default:
-			http.Error(w, "Unsupported file format", http.StatusBadRequest)
+		img, _, err := image.Decode(file)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusBadRequest)
 			return
 		}
 
 		message := r.FormValue("message")
 		binaryPesan := pesanKeBinary(message)
 
-		// Embed message
 		newImg := embedLSB(img, binaryPesan)
 		embedDuration := time.Since(start).Seconds()
 
 		var buf bytes.Buffer
-		switch imgFormat {
-		case "jpeg":
-			err = jpeg.Encode(&buf, newImg, nil)
-		case "png":
-			err = png.Encode(&buf, newImg)
-		}
+		err = png.Encode(&buf, newImg)
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusInternalServerError)
 			return
@@ -185,36 +163,21 @@ func uploadHandler(w http.ResponseWriter, r *http.Request) {
 		http.ServeFile(w, r, "templates/index.html")
 	}
 }
+
 func extractHandler(w http.ResponseWriter, r *http.Request) {
 	if r.Method == http.MethodPost {
 		start := time.Now()
 
-		file, fileHeader, err := r.FormFile("image")
+		file, _, err := r.FormFile("image")
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusBadRequest)
 			return
 		}
 		defer file.Close()
 
-		var img image.Image
-		var imgFormat string
-
-		switch strings.ToLower(filepath.Ext(fileHeader.Filename)) {
-		case ".png":
-			img, imgFormat, err = image.Decode(file)
-			if err != nil {
-				http.Error(w, err.Error(), http.StatusBadRequest)
-				return
-			}
-		case ".jpg", ".jpeg":
-			img, err = jpeg.Decode(file)
-			if err != nil {
-				http.Error(w, err.Error(), http.StatusBadRequest)
-				return
-			}
-			imgFormat = "jpeg"
-		default:
-			http.Error(w, "Unsupported file format", http.StatusBadRequest)
+		img, _, err := image.Decode(file)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusBadRequest)
 			return
 		}
 

templates/index.html

@@ -39,6 +39,7 @@
             type="file"
             id="image"
             name="image"
+            accept="image/png"
             class="mt-1 block w-full border border-gray-300 rounded-md shadow-sm focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm"
           />
         </div>