From fb55e748563600977ded4b165bcff93d75455cb2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mika=C3=ABl=20Cluseau?= <mcluseau@isi.nc>
Date: Wed, 13 Jun 2018 13:26:00 +1100
Subject: [PATCH] feat: optionnaly limit /hosts access by token

---
 http.go | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/http.go b/http.go
index 7e5866b..c6ddf08 100644
--- a/http.go
+++ b/http.go
@@ -14,11 +14,29 @@ import (
 )
 
 var (
+	hostsToken = flag.String("hosts-token", "", "Token to give to access /hosts (open is none)")
+
 	reHost = regexp.MustCompile("^/hosts/([^/]+)/([^/]+)$")
 
 	trustXFF = flag.Bool("trust-xff", true, "Trust the X-Forwarded-For header")
 )
 
+func authorizeHosts(r *http.Request) bool {
+	if *hostsToken == "" {
+		// access is open
+		return true
+	}
+
+	reqToken := r.Header.Get("Authorization")
+
+	return reqToken == "Bearer "+*hostsToken
+}
+
+func forbidden(w http.ResponseWriter, r *http.Request) {
+	log.Printf("denied access to %s from %s", r.RequestURI, r.RemoteAddr)
+	http.Error(w, "Forbidden", http.StatusForbidden)
+}
+
 func serveHostByIP(w http.ResponseWriter, r *http.Request) {
 	host, cfg := hostByIP(w, r)
 	if host == nil {
@@ -64,6 +82,11 @@ func hostByIP(w http.ResponseWriter, r *http.Request) (*clustersconfig.Host, *cl
 }
 
 func serveHosts(w http.ResponseWriter, r *http.Request) {
+	if !authorizeHosts(r) {
+		forbidden(w, r)
+		return
+	}
+
 	cfg, err := readConfig()
 	if err != nil {
 		http.Error(w, "", http.StatusServiceUnavailable)
@@ -79,6 +102,11 @@ func serveHosts(w http.ResponseWriter, r *http.Request) {
 }
 
 func serveHost(w http.ResponseWriter, r *http.Request) {
+	if !authorizeHosts(r) {
+		forbidden(w, r)
+		return
+	}
+
 	match := reHost.FindStringSubmatch(r.URL.Path)
 	if match == nil {
 		http.NotFound(w, r)