Add a bit of safeguard against malicious attacks

Author: matteosuppo

killbrowser.go

@@ -1,7 +1,7 @@
 package main
 
 import (
-	"log"
+	"errors"
 	"net/http"
 
 	"github.com/gin-gonic/gin"
@@ -17,25 +17,31 @@ func killBrowserHandler(c *gin.Context) {
 
 	c.BindJSON(&data)
 
-	command, err := findBrowser(data.Process)
+	if data.Process != "chrome" && data.Process != "chrom" {
+		c.JSON(http.StatusBadRequest, errors.New("You can't kill the process"+data.Process))
+		return
+	}
 
-	log.Println(command)
+	command, err := findBrowser(data.Process)
 
 	if err != nil {
 		c.JSON(http.StatusInternalServerError, err.Error())
+		return
 	}
 
 	if data.Action == "kill" || data.Action == "restart" {
 		_, err := killBrowser(data.Process)
 		if err != nil {
 			c.JSON(http.StatusInternalServerError, err.Error())
+			return
 		}
 	}
 
 	if data.Action == "restart" {
 		_, err := startBrowser(command, data.URL)
 		if err != nil {
 			c.JSON(http.StatusInternalServerError, err.Error())
+			return
 		}
 	}