Add basic auth

Author: triole

src/argparse.go

@@ -19,12 +19,13 @@ var (
 )
 
 var CLI struct {
-	Port          int    `help:"port where to serve" short:p default:9999`
-	ResponseDelay []int  `help:"server response delay in ms, use twice to define range for a random value before each request" short:r`
-	JSONLog       bool   `help:"enable json log, instead of text one" short:j`
-	LogFile       string `help:"log file" short:l default:/dev/stdout`
-	Verbose       bool   `help:"verbose mode, print full response data set" short:v`
-	VersionFlag   bool   `help:"display version" short:V`
+	Port          int      `help:"port where to serve" short:p default:9999`
+	BasicAuth     []string `help:"enable basic auth" short:b placeholder:USER:PASS sep::`
+	ResponseDelay []int    `help:"server response delay in ms, use twice to define range for a random value before each request" short:r`
+	JSONLog       bool     `help:"enable json log, instead of text one" short:j`
+	LogFile       string   `help:"log file" short:l default:/dev/stdout`
+	Verbose       bool     `help:"verbose mode, print full response data set" short:v`
+	VersionFlag   bool     `help:"display version" short:V`
 }
 
 func parseArgs() {

src/main.go

@@ -17,6 +17,7 @@ type tConf struct {
 	Bind          string
 	Port          int
 	ResponseDelay []int
+	BasicAuth     []string
 }
 
 func main() {
@@ -25,6 +26,7 @@ func main() {
 	conf := tConf{
 		Port:          CLI.Port,
 		ResponseDelay: CLI.ResponseDelay,
+		BasicAuth:     CLI.BasicAuth,
 	}
 
 	lg = logging.Init(CLI.LogFile, CLI.JSONLog)

src/server.go

@@ -41,6 +41,24 @@ func (h *handler) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
 			) * time.Millisecond,
 		)
 	}
+
+	if len(CLI.BasicAuth) > 0 {
+		user, pass, ok := req.BasicAuth()
+		if !ok {
+			fmt.Println("Error parsing basic auth")
+			resp.WriteHeader(401)
+			return
+		}
+		if user != CLI.BasicAuth[0] || pass != CLI.BasicAuth[1] {
+			lg.LogError("Credentials incorrect", logrus.Fields{
+				"user": user,
+				"pass": pass,
+			})
+			resp.WriteHeader(401)
+			return
+		}
+	}
+
 	responseCode := parseResponseCode(req.URL.String())
 	resp.WriteHeader(responseCode)
 	data := tResponse{