Add secure middleware to enable more header options

Gopkg.toml

@@ -4,8 +4,8 @@
 #
 
 [[constraint]]
-  name = "github.com/18F/hmacauth"
-  version = "~1.0.1"
+  name = "github.com/mbland/hmacauth"
+  version = "~1.0.2"
 
 [[constraint]]
   name = "github.com/BurntSushi/toml"
@@ -36,7 +36,7 @@
   name = "google.golang.org/api"
 
 [[constraint]]
-  name = "gopkg.in/fsnotify.v1"
+  name = "github.com/fsnotify/fsnotify"
   version = "~1.2.0"
 
 [[constraint]]

main.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/BurntSushi/toml"
 	"github.com/mreiferson/go-options"
+	"github.com/unrolled/secure"
 )
 
 func main() {
@@ -21,6 +22,8 @@ func main() {
 	upstreams := StringArray{}
 	skipAuthRegex := StringArray{}
 	googleGroups := StringArray{}
+	httpAllowedHosts := StringArray{}
+	httpHostsProxyHeaders := StringArray{}
 
 	config := flagSet.String("config", "", "path to config file")
 	showVersion := flagSet.Bool("version", false, "print version string")
@@ -81,6 +84,25 @@ func main() {
 
 	flagSet.String("signature-key", "", "GAP-Signature request signature key (algorithm:secretkey)")
 
+	// These are options that allow you to tune various parameters for https://github.com/unrolled/secure
+	flagSet.Var(&httpAllowedHosts, "httpAllowedHosts", "a list of fully qualified domain names that are allowed. Default is empty list, which allows any and all host names.")
+	flagSet.Var(&httpHostsProxyHeaders, "httpHostsProxyHeaders", "a set of header keys that may hold a proxied hostname value for the request.")
+	flagSet.Bool("httpSSLRedirect", false, "If set to true, then only allow HTTPS requests. Default is false.")
+	flagSet.Bool("httpSSLTemporaryRedirect", false, "If true, then a 302 will be used while redirecting. Default is false (301).")
+	flagSet.String("httpSSLHost", "", "the host name that is used to redirect HTTP requests to HTTPS. Default is \"\", which indicates to use the same host.")
+	flagSet.Int64("httpSTSSeconds", 0, "The max-age of the Strict-Transport-Security header. Default is 0, which would NOT include the header.")
+	flagSet.Bool("httpSTSIncludeSubdomains", false, "If set to true, the 'includeSubdomains' will be appended to the Strict-Transport-Security header. Default is false.")
+	flagSet.Bool("httpSTSPreload", false, "If set to true, the 'preload' flag will be appended to the Strict-Transport-Security header. Default is false.")
+	flagSet.Bool("httpForceSTSHeader", false, "STS header is only included when the connection is HTTPS. If you want to force it to always be added, set to true. Default is false.")
+	flagSet.Bool("httpFrameDeny", false, "If set to true, adds the X-Frame-Options header with the value of 'DENY'. Default is false.")
+	flagSet.String("httpCustomFrameOptionsValue", "", "allows the X-Frame-Options header value to be set with a custom value. This overrides the FrameDeny option. Default is \"\".")
+	flagSet.Bool("httpContentTypeNosniff", false, "If true, adds the X-Content-Type-Options header with the value 'nosniff'. Default is false.")
+	flagSet.Bool("httpBrowserXssFilter", false, "If true, adds the X-XSS-Protection header with the value '1; mode=block'. Default is false.")
+	flagSet.String("httpCustomBrowserXssValue", "", "Allows the X-XSS-Protection header value to be set with a custom value. This overrides the BrowserXssFilter option. Default is \"\".")
+	flagSet.String("httpContentSecurityPolicy", "", "Allows the Content-Security-Policy header value to be set with a custom value. Default is \"\". Passing a template string will replace '$NONCE' with a dynamic nonce value of 16 bytes for each request which can be later retrieved using the Nonce function.")
+	flagSet.String("httpPublicKey", "", "Implements HPKP to prevent MITM attacks with forged certificates. Default is \"\".")
+	flagSet.String("httpReferrerPolicy", "", "Allows the Referrer-Policy header with the value to be set with a custom value. Default is \"\".")
+
 	flagSet.Parse(os.Args[1:])
 
 	if *showVersion {
@@ -106,24 +128,44 @@ func main() {
 		os.Exit(1)
 	}
 	validator := NewValidator(opts.EmailDomains, opts.AuthenticatedEmailsFile)
-	oauthproxy := NewOAuthProxy(opts, validator)
+	bareproxy := NewOAuthProxy(opts, validator)
 
 	if len(opts.EmailDomains) != 0 && opts.AuthenticatedEmailsFile == "" {
 		if len(opts.EmailDomains) > 1 {
-			oauthproxy.SignInMessage = fmt.Sprintf("Authenticate using one of the following domains: %v", strings.Join(opts.EmailDomains, ", "))
+			bareproxy.SignInMessage = fmt.Sprintf("Authenticate using one of the following domains: %v", strings.Join(opts.EmailDomains, ", "))
 		} else if opts.EmailDomains[0] != "*" {
-			oauthproxy.SignInMessage = fmt.Sprintf("Authenticate using %v", opts.EmailDomains[0])
+			bareproxy.SignInMessage = fmt.Sprintf("Authenticate using %v", opts.EmailDomains[0])
 		}
 	}
 
 	if opts.HtpasswdFile != "" {
 		log.Printf("using htpasswd file %s", opts.HtpasswdFile)
-		oauthproxy.HtpasswdFile, err = NewHtpasswdFromFile(opts.HtpasswdFile)
-		oauthproxy.DisplayHtpasswdForm = opts.DisplayHtpasswdForm
+		bareproxy.HtpasswdFile, err = NewHtpasswdFromFile(opts.HtpasswdFile)
+		bareproxy.DisplayHtpasswdForm = opts.DisplayHtpasswdForm
 		if err != nil {
 			log.Fatalf("FATAL: unable to open %s %s", opts.HtpasswdFile, err)
 		}
 	}
+	secureMiddleware := secure.New(secure.Options{
+		AllowedHosts:            opts.HttpAllowedHosts,
+		HostsProxyHeaders:       opts.HttpHostsProxyHeaders,
+		SSLRedirect:             opts.HttpSSLRedirect,
+		SSLTemporaryRedirect:    opts.HttpSSLTemporaryRedirect,
+		SSLHost:                 opts.HttpSSLHost,
+		STSSeconds:              opts.HttpSTSSeconds,
+		STSIncludeSubdomains:    opts.HttpSTSIncludeSubdomains,
+		STSPreload:              opts.HttpSTSPreload,
+		ForceSTSHeader:          opts.HttpForceSTSHeader,
+		FrameDeny:               opts.HttpFrameDeny,
+		CustomFrameOptionsValue: opts.HttpCustomFrameOptionsValue,
+		ContentTypeNosniff:      opts.HttpContentTypeNosniff,
+		BrowserXssFilter:        opts.HttpBrowserXssFilter,
+		CustomBrowserXssValue:   opts.HttpCustomBrowserXssValue,
+		ContentSecurityPolicy:   opts.HttpContentSecurityPolicy,
+		PublicKey:               opts.HttpPublicKey,
+		ReferrerPolicy:          opts.HttpReferrerPolicy,
+	})
+	oauthproxy := secureMiddleware.Handler(bareproxy)
 
 	s := &Server{
 		Handler: LoggingHandler(os.Stdout, oauthproxy, opts.RequestLogging, opts.RequestLoggingFormat),

oauthproxy_test.go

@@ -18,6 +18,7 @@ import (
 	"github.com/bitly/oauth2_proxy/providers"
 	"github.com/mbland/hmacauth"
 	"github.com/stretchr/testify/assert"
+	"github.com/unrolled/secure"
 )
 
 func init() {
@@ -836,3 +837,73 @@ func TestRequestSignaturePostRequest(t *testing.T) {
 	assert.Equal(t, 200, st.rw.Code)
 	assert.Equal(t, st.rw.Body.String(), "signatures match")
 }
+
+func TestHttpBrowserXssFilterTrue(t *testing.T) {
+	opts := NewOptions()
+	opts.ClientID = "bazquux"
+	opts.ClientSecret = "foobar"
+	opts.CookieSecret = "xyzzyplugh"
+	opts.HttpBrowserXssFilter = true
+	opts.Validate()
+
+	bareproxy := NewOAuthProxy(opts, func(string) bool { return true })
+	secureMiddleware := secure.New(secure.Options{
+		AllowedHosts:            opts.HttpAllowedHosts,
+		HostsProxyHeaders:       opts.HttpHostsProxyHeaders,
+		SSLRedirect:             opts.HttpSSLRedirect,
+		SSLTemporaryRedirect:    opts.HttpSSLTemporaryRedirect,
+		SSLHost:                 opts.HttpSSLHost,
+		STSSeconds:              opts.HttpSTSSeconds,
+		STSIncludeSubdomains:    opts.HttpSTSIncludeSubdomains,
+		STSPreload:              opts.HttpSTSPreload,
+		ForceSTSHeader:          opts.HttpForceSTSHeader,
+		FrameDeny:               opts.HttpFrameDeny,
+		CustomFrameOptionsValue: opts.HttpCustomFrameOptionsValue,
+		ContentTypeNosniff:      opts.HttpContentTypeNosniff,
+		BrowserXssFilter:        opts.HttpBrowserXssFilter,
+		CustomBrowserXssValue:   opts.HttpCustomBrowserXssValue,
+		ContentSecurityPolicy:   opts.HttpContentSecurityPolicy,
+		PublicKey:               opts.HttpPublicKey,
+		ReferrerPolicy:          opts.HttpReferrerPolicy,
+	})
+	proxy := secureMiddleware.Handler(bareproxy)
+	rw := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", "/", nil)
+	proxy.ServeHTTP(rw, req)
+	assert.Equal(t, "1; mode=block", rw.HeaderMap.Get("X-XSS-Protection"))
+}
+
+func TestHttpBrowserXssFilterFalseBydefault(t *testing.T) {
+	opts := NewOptions()
+	opts.ClientID = "bazquux"
+	opts.ClientSecret = "foobar"
+	opts.CookieSecret = "xyzzyplugh"
+	opts.Validate()
+
+	bareproxy := NewOAuthProxy(opts, func(string) bool { return true })
+	secureMiddleware := secure.New(secure.Options{
+		AllowedHosts:            opts.HttpAllowedHosts,
+		HostsProxyHeaders:       opts.HttpHostsProxyHeaders,
+		SSLRedirect:             opts.HttpSSLRedirect,
+		SSLTemporaryRedirect:    opts.HttpSSLTemporaryRedirect,
+		SSLHost:                 opts.HttpSSLHost,
+		STSSeconds:              opts.HttpSTSSeconds,
+		STSIncludeSubdomains:    opts.HttpSTSIncludeSubdomains,
+		STSPreload:              opts.HttpSTSPreload,
+		ForceSTSHeader:          opts.HttpForceSTSHeader,
+		FrameDeny:               opts.HttpFrameDeny,
+		CustomFrameOptionsValue: opts.HttpCustomFrameOptionsValue,
+		ContentTypeNosniff:      opts.HttpContentTypeNosniff,
+		BrowserXssFilter:        opts.HttpBrowserXssFilter,
+		CustomBrowserXssValue:   opts.HttpCustomBrowserXssValue,
+		ContentSecurityPolicy:   opts.HttpContentSecurityPolicy,
+		PublicKey:               opts.HttpPublicKey,
+		ReferrerPolicy:          opts.HttpReferrerPolicy,
+	})
+	proxy := secureMiddleware.Handler(bareproxy)
+	rw := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", "/", nil)
+	proxy.ServeHTTP(rw, req)
+	assert.Equal(t, false, opts.HttpBrowserXssFilter)
+	assert.Equal(t, "", rw.HeaderMap.Get("X-XSS-Protection"))
+}

options.go

@@ -79,6 +79,25 @@ type Options struct {
 
 	SignatureKey string `flag:"signature-key" cfg:"signature_key" env:"OAUTH2_PROXY_SIGNATURE_KEY"`
 
+	// These are options that allow you to tune various parameters for https://github.com/unrolled/secure
+	HttpAllowedHosts            []string `flag:"httpAllowedHosts" cfg:"httpAllowedHosts"`
+	HttpHostsProxyHeaders       []string `flag:"httpHostsProxyHeaders" cfg:"httpHostsProxyHeaders"`
+	HttpSSLRedirect             bool     `flag:"httpSSLRedirect" cfg:"httpSSLRedirect"`
+	HttpSSLTemporaryRedirect    bool     `flag:"httpSSLTemporaryRedirect" cfg:"httpSSLTemporaryRedirect"`
+	HttpSSLHost                 string   `flag:"httpSSLHost" cfg:"httpSSLHost"`
+	HttpSTSSeconds              int64    `flag:"httpSTSSeconds" cfg:"httpSTSSeconds"`
+	HttpSTSIncludeSubdomains    bool     `flag:"httpSTSIncludeSubdomains" cfg:"httpSTSIncludeSubdomains"`
+	HttpSTSPreload              bool     `flag:"httpSTSPreload" cfg:"httpSTSPreload"`
+	HttpForceSTSHeader          bool     `flag:"httpForceSTSHeader" cfg:"httpForceSTSHeader"`
+	HttpFrameDeny               bool     `flag:"httpFrameDeny" cfg:"httpFrameDeny"`
+	HttpCustomFrameOptionsValue string   `flag:"httpCustomFrameOptionsValue" cfg:"httpCustomFrameOptionsValue"`
+	HttpContentTypeNosniff      bool     `flag:"httpContentTypeNosniff" cfg:"httpContentTypeNosniff"`
+	HttpBrowserXssFilter        bool     `flag:"httpBrowserXssFilter" cfg:"httpBrowserXssFilter"`
+	HttpCustomBrowserXssValue   string   `flag:"httpCustomBrowserXssValue" cfg:"httpCustomBrowserXssValue"`
+	HttpContentSecurityPolicy   string   `flag:"httpContentSecurityPolicy" cfg:"httpContentSecurityPolicy"`
+	HttpPublicKey               string   `flag:"httpPublicKey" cfg:"httpPublicKey"`
+	HttpReferrerPolicy          string   `flag:"httpReferrerPolicy" cfg:"httpReferrerPolicy"`
+
 	// internal values that are set after config validation
 	redirectURL   *url.URL
 	proxyURLs     []*url.URL

test.sh

@@ -11,4 +11,4 @@ for pkg in $(go list ./... | grep -v '/vendor/' ); do
     echo "go test -v -race $pkg"
     GOMAXPROCS=4 go test -v -timeout 90s0s -race "$pkg" || EXIT_CODE=1
 done
-exit $EXIT_CODE
+exit $EXIT_CODE

watcher.go

@@ -8,7 +8,7 @@ import (
 	"path/filepath"
 	"time"
 
-	"gopkg.in/fsnotify.v1"
+	"github.com/fsnotify/fsnotify"
 )
 
 func WaitForReplacement(filename string, op fsnotify.Op,