From 7f204256251fa30e0526c3b5ff84d0d99bb7eac1 Mon Sep 17 00:00:00 2001
From: zekiahmetbayar <zekiahmetbayar1@gmail.com>
Date: Thu, 26 Oct 2023 14:10:31 +0300
Subject: [PATCH] feature: Basic auth for web console

---
 cmd/asynqmon/main.go | 52 +++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 51 insertions(+), 1 deletion(-)

diff --git a/cmd/asynqmon/main.go b/cmd/asynqmon/main.go
index a5e8d01..8e8b86c 100644
--- a/cmd/asynqmon/main.go
+++ b/cmd/asynqmon/main.go
@@ -2,6 +2,8 @@ package main
 
 import (
 	"bytes"
+	"crypto/sha256"
+	"crypto/subtle"
 	"crypto/tls"
 	"flag"
 	"fmt"
@@ -25,6 +27,11 @@ type Config struct {
 	// Server port
 	Port int
 
+	// Basic auth options
+	EnableBasicAuth   bool
+	BasicAuthUsername string
+	BasicAuthPassword string
+
 	// Redis connection options
 	RedisAddr         string
 	RedisDB           int
@@ -72,6 +79,9 @@ func parseFlags(progname string, args []string) (cfg *Config, output string, err
 	flags.BoolVar(&conf.EnableMetricsExporter, "enable-metrics-exporter", getEnvOrDefaultBool("ENABLE_METRICS_EXPORTER", false), "enable prometheus metrics exporter to expose queue metrics")
 	flags.StringVar(&conf.PrometheusServerAddr, "prometheus-addr", getEnvDefaultString("PROMETHEUS_ADDR", ""), "address of prometheus server to query time series")
 	flags.BoolVar(&conf.ReadOnly, "read-only", getEnvOrDefaultBool("READ_ONLY", false), "restrict to read-only mode")
+	flags.BoolVar(&conf.EnableBasicAuth, "enable-basic-auth", getEnvOrDefaultBool("ENABLE_BASIC_AUTH", false), "enable basic auth for web console")
+	flags.StringVar(&conf.BasicAuthUsername, "basic-auth-username", getEnvDefaultString("BASIC_AUTH_USERNAME", "administrator"), "username for web console's basic auth")
+	flags.StringVar(&conf.BasicAuthPassword, "basic-auth-password", getEnvDefaultString("BASIC_AUTH_PASSWORD", "Passw0rd!!!"), "password for web console's basic auth")
 
 	err = flags.Parse(args)
 	if err != nil {
@@ -159,8 +169,18 @@ func main() {
 	c := cors.New(cors.Options{
 		AllowedMethods: []string{"GET", "POST", "DELETE"},
 	})
+
 	mux := http.NewServeMux()
-	mux.Handle("/", c.Handler(h))
+	// Is basic auth enabled for web console
+	switch cfg.EnableBasicAuth {
+	case true:
+		// Serve with basic auth
+		mux.Handle("/", basicAuth(c.Handler(h), cfg.BasicAuthUsername, cfg.BasicAuthPassword))
+	default:
+		// Serve without basic auth
+		mux.Handle("/", c.Handler(h))
+	}
+
 	if cfg.EnableMetricsExporter {
 		// Using NewPedanticRegistry here to test the implementation of Collectors and Metrics.
 		reg := prometheus.NewPedanticRegistry()
@@ -187,6 +207,36 @@ func main() {
 	log.Fatal(srv.ListenAndServe())
 }
 
+// Basic auth
+func basicAuth(next http.Handler, expectedUsername, expectedPassword string) http.HandlerFunc {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		// Get username and password from user
+		username, password, ok := r.BasicAuth()
+		if ok {
+			// Get hash of username and password input
+			usernameHash := sha256.Sum256([]byte(username))
+			passwordHash := sha256.Sum256([]byte(password))
+
+			// Get hash of expected username and password
+			expectedUsernameHash := sha256.Sum256([]byte(expectedUsername))
+			expectedPasswordHash := sha256.Sum256([]byte(expectedPassword))
+
+			// Check username and passwords are match
+			usernameMatch := (subtle.ConstantTimeCompare(usernameHash[:], expectedUsernameHash[:]) == 1)
+			passwordMatch := (subtle.ConstantTimeCompare(passwordHash[:], expectedPasswordHash[:]) == 1)
+
+			if usernameMatch && passwordMatch {
+				// Serve
+				next.ServeHTTP(w, r)
+				return
+			}
+		}
+		// Set error, unauthorized
+		w.Header().Set("WWW-Authenticate", `Basic realm="restricted", charset="UTF-8"`)
+		http.Error(w, "Unauthorized", http.StatusUnauthorized)
+	})
+}
+
 func payloadFormatterFunc(cfg *Config) func(string, []byte) string {
 	return func(taskType string, payload []byte) string {
 		payloadStr := asynqmon.DefaultPayloadFormatter.FormatPayload(taskType, payload)