Add a SSO HTTP proxy for legacy services

The proxy wraps endpoints (exposed as separate HTTP hosts) with single-sign-on
authentication.

cmd/sso-proxy/main.go

+package main
+
+import (
+	"context"
+	"flag"
+	"io/ioutil"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"strings"
+	"syscall"
+	"time"
+
+	"gopkg.in/yaml.v2"
+
+	"git.autistici.org/id/go-sso/proxy"
+)
+
+var (
+	addr       = flag.String("addr", ":5003", "address to listen on")
+	configFile = flag.String("config", "/etc/sso/proxy.yml", "path of config file")
+)
+
+func loadConfig() (*proxy.Configuration, error) {
+	// Read YAML config.
+	data, err := ioutil.ReadFile(*configFile)
+	if err != nil {
+		return nil, err
+	}
+	var config proxy.Configuration
+	if err := yaml.Unmarshal(data, &config); err != nil {
+		return nil, err
+	}
+	return &config, nil
+}
+
+// Set defaults for command-line flags using variables from the environment.
+func setFlagDefaultsFromEnv() {
+	flag.VisitAll(func(f *flag.Flag) {
+		envVar := "SSOPROXY_" + strings.ToUpper(strings.Replace(f.Name, "-", "_", -1))
+		if value := os.Getenv(envVar); value != "" {
+			f.DefValue = value
+			f.Value.Set(value)
+		}
+	})
+}
+
+func main() {
+	setFlagDefaultsFromEnv()
+	flag.Parse()
+
+	config, err := loadConfig()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	h, err := proxy.NewProxy(config)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	srv := &http.Server{
+		Addr:    *addr,
+		Handler: h,
+	}
+
+	sigCh := make(chan os.Signal, 1)
+	go func() {
+		<-sigCh
+		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+		_ = srv.Shutdown(ctx)
+		_ = srv.Close()
+	}()
+	signal.Notify(sigCh, syscall.SIGTERM, syscall.SIGINT)
+
+	if err := srv.ListenAndServe(); err != nil {
+		log.Fatal(err)
+	}
+}

debian/control

@@ -10,3 +10,9 @@ Architecture: any
 Depends: ${shlibs:Depends}, ${misc:Depends}, auth-server
 Description: Single-Sign-On server.
  Single-Sign-On server, integrated with git.autistici.org/id/auth.
+
+Package: sso-proxy
+Architecture: any
+Depends: ${shlibs:Depends}, ${misc:Depends}
+Description: Single-Sign-On HTTP proxy.
+ Single-Sign-On HTTP proxy.

debian/rules

@@ -8,7 +8,7 @@ export DH_GOLANG_EXCLUDES = vendor
 	dh $@ --with systemd --with golang --buildsystem golang
 
 override_dh_install:
-	rm -fr $(CURDIR)/debian/sso-server/usr/share/gocode
+	rm -fr $(CURDIR)/debian/tmp/usr/share/gocode
 	dh_install
 
 override_dh_systemd_enable:

debian/sso-proxy.default

+ADDR=:5003

debian/sso-proxy.install

+usr/bin/sso-proxy

debian/sso-proxy.postinst

+#!/bin/sh
+
+set -e
+
+case "$1" in
+configure)
+    addgroup --system --quiet sso-proxy
+    adduser --system --no-create-home --home /run/sso-proxy \
+      --disabled-password --disabled-login \
+      --quiet --ingroup sso-proxy sso-proxy
+    ;;
+esac
+
+#DEBHELPER#
+
+exit 0

debian/sso-proxy.service

+[Unit]
+Description=SSO Proxy
+
+[Service]
+User=sso-proxy
+Group=sso-proxy
+EnvironmentFile=-/etc/default/sso-proxy
+ExecStart=/usr/bin/sso-proxy --addr $ADDR
+Restart=always
+
+[Install]
+WantedBy=multi-user.target
+

debian/sso-server.install

+usr/bin/sso-server

httpsso/handler.go

+package httpsso
+
+import (
+	"encoding/gob"
+	"encoding/hex"
+	"io"
+	"math/rand"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+
+	"github.com/gorilla/sessions"
+
+	"git.autistici.org/id/go-sso"
+	"git.autistici.org/id/go-sso/httputil"
+)
+
+type authSession struct {
+	*httputil.ExpiringSession
+
+	Auth     bool
+	Username string
+}
+
+var authSessionLifetime = 1 * time.Hour
+
+func init() {
+	gob.Register(&authSession{})
+}
+
+// SSOWrapper protects http handlers with single-sign-on authentication.
+type SSOWrapper struct {
+	v              sso.Validator
+	sessionAuthKey []byte
+	sessionEncKey  []byte
+	serverURL      string
+}
+
+// NewSSOWrapper returns a new SSOWrapper that will authenticate users
+// on the specified login service.
+func NewSSOWrapper(serverURL string, pkey []byte, domain string, sessionAuthKey, sessionEncKey []byte) (*SSOWrapper, error) {
+	v, err := sso.NewValidator(pkey, domain)
+	if err != nil {
+		return nil, err
+	}
+
+	return &SSOWrapper{
+		v:              v,
+		serverURL:      serverURL,
+		sessionAuthKey: sessionAuthKey,
+		sessionEncKey:  sessionEncKey,
+	}, nil
+}
+
+// Wrap a http.Handler with authentication and access control.
+// Currently only a simple form of group-based ACLs is supported.
+func (s *SSOWrapper) Wrap(h http.Handler, service string, groups []string) http.Handler {
+	svcPath := pathFromService(service)
+	store := sessions.NewCookieStore(s.sessionAuthKey, s.sessionEncKey)
+	store.Options = &sessions.Options{
+		HttpOnly: true,
+		Secure:   true,
+		MaxAge:   0,
+		Path:     svcPath,
+	}
+
+	return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+		session, _ := store.Get(req, "sso")
+
+		switch strings.TrimPrefix(req.URL.Path, svcPath) {
+		case "sso_login":
+			s.handleLogin(w, req, session, service, groups)
+
+		case "sso_logout":
+			s.handleLogout(w, req, session)
+
+		default:
+			if auth, ok := session.Values["a"].(*authSession); ok && auth.Valid() && auth.Auth {
+				req.Header.Set("X-Authenticated-User", auth.Username)
+
+				h.ServeHTTP(w, req)
+				return
+			}
+
+			s.redirectToLogin(w, req, session, service, groups)
+		}
+	})
+}
+
+func (s *SSOWrapper) handleLogin(w http.ResponseWriter, req *http.Request, session *sessions.Session, service string, groups []string) {
+	t := req.FormValue("t")
+	d := req.FormValue("d")
+
+	// Pop the nonce from the session.
+	nonce, ok := session.Values["nonce"].(string)
+	if !ok || nonce == "" {
+		http.Error(w, "Missing nonce", http.StatusBadRequest)
+		return
+	}
+	delete(session.Values, "nonce")
+
+	tkt, err := s.v.Validate(t, nonce, service, groups)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	// Authenticate the user.
+	session.Values["a"] = &authSession{
+		ExpiringSession: httputil.NewExpiringSession(authSessionLifetime),
+		Auth:            true,
+		Username:        tkt.User,
+	}
+	if err := sessions.Save(req, w); err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	http.Redirect(w, req, d, http.StatusFound)
+}
+
+func (s *SSOWrapper) handleLogout(w http.ResponseWriter, req *http.Request, session *sessions.Session) {
+	session.Options.MaxAge = -1
+	if err := sessions.Save(req, w); err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	w.Header().Set("Content-Type", "text/plain")
+	io.WriteString(w, "OK")
+}
+
+// Redirect to the SSO server.
+func (s *SSOWrapper) redirectToLogin(w http.ResponseWriter, req *http.Request, session *sessions.Session, service string, groups []string) {
+	// Generate a random nonce and store it in the local session.
+	nonce := makeUniqueNonce()
+	session.Values["nonce"] = nonce
+	if err := sessions.Save(req, w); err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	v := make(url.Values)
+	v.Set("s", service)
+	v.Set("d", req.URL.String())
+	v.Set("n", nonce)
+	v.Set("g", strings.Join(groups, ","))
+	loginURL := s.serverURL + "?" + v.Encode()
+	http.Redirect(w, req, loginURL, http.StatusFound)
+}
+
+// Extract the URL path from the service specification. The result
+// will have both a leading and a trailing slash.
+func pathFromService(service string) string {
+	i := strings.IndexRune(service, '/')
+	if i < 0 {
+		return ""
+	}
+	return service[i:]
+}
+
+func makeUniqueNonce() string {
+	var b [8]byte
+	if _, err := rand.Read(b[:]); err != nil {
+		panic(err)
+	}
+	return hex.EncodeToString(b[:])
+}

httputil/session.go

+package httputil
+
+import (
+	"encoding/gob"
+	"time"
+)
+
+// ExpiringSession is a session with server-side expiration check.
+// Session data is saved in signed, encrypted cookies in the
+// browser. We'd like these cookies to expire when a certain amount of
+// time passes, or when the user closes the browser. We trust the
+// browser for the latter, but we enforce time-based expiration on the
+// server.
+type ExpiringSession struct {
+	Expiry time.Time
+}
+
+// NewExpiringSession returns a session that is valid for the given
+// duration.
+func NewExpiringSession(ttl time.Duration) *ExpiringSession {
+	return &ExpiringSession{
+		Expiry: time.Now().Add(ttl),
+	}
+}
+
+// Valid returns true if the session has not expired yet.
+// It can be called with a nil receiver.
+func (e *ExpiringSession) Valid() bool {
+	return e != nil && time.Now().Before(e.Expiry)
+}
+
+func init() {
+	gob.Register(&ExpiringSession{})
+}
+

server/util_test.go → httputil/session_test.go

-package server
+package httputil
 
 import (
 	"bytes"
@@ -14,7 +14,7 @@ func TestExpiringSession(t *testing.T) {
 		Data string
 	}
 	s := &mySession{
-		ExpiringSession: newExpiringSession(60 * time.Second),
+		ExpiringSession: NewExpiringSession(60 * time.Second),
 		Data:            "data",
 	}
 

proxy/proxy.go

+package proxy
+
+import (
+	"crypto/tls"
+	"crypto/x509"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/http/httputil"
+	"net/url"
+
+	"github.com/gorilla/mux"
+
+	"git.autistici.org/id/go-sso/httpsso"
+)
+
+// Backend defines a single-host HTTP proxy to a set of upstream
+// backends.
+type Backend struct {
+	Host            string     `yaml:"host"`
+	Upstream        []string   `yaml:"upstream"`
+	ClientTLSConfig *TLSConfig `yaml:"client_tls"`
+	//ServerTLSConfig *TLSConfig `yaml:"server_tls"`
+
+	AllowedGroups []string `yaml:"allowed_groups"`
+}
+
+func (b *Backend) newHandler(ssow *httpsso.SSOWrapper) (http.Handler, error) {
+	// Setup upstream connections.
+	if len(b.Upstream) < 1 {
+		return nil, errors.New("no backends specified")
+	}
+
+	u := &url.URL{Scheme: "http", Host: b.Host}
+	if b.ClientTLSConfig != nil {
+		u.Scheme = "https"
+	}
+
+	proxy := httputil.NewSingleHostReverseProxy(u)
+
+	var tlsConfig *tls.Config
+	if b.ClientTLSConfig != nil {
+		var err error
+		tlsConfig, err = b.ClientTLSConfig.toClientConfig()
+		if err != nil {
+			return nil, err
+		}
+	}
+	proxy.Transport = newTransport(b.Upstream, tlsConfig)
+
+	h := ssow.Wrap(proxy, b.Host+"/", b.AllowedGroups)
+	return h, nil
+}
+
+// TLSConfig defines the TLS parameters for a client connection.
+type TLSConfig struct {
+	Cert string `yaml:"cert"`
+	Key  string `yaml:"key"`
+	CA   string `yaml:"ca"`
+}
+
+func (c *TLSConfig) toClientConfig() (*tls.Config, error) {
+	cert, err := tls.LoadX509KeyPair(c.Cert, c.Key)
+	if err != nil {
+		return nil, err
+	}
+
+	cas, err := loadCA(c.CA)
+	if err != nil {
+		return nil, err
+	}
+
+	return &tls.Config{
+		Certificates: []tls.Certificate{cert},
+		RootCAs:      cas,
+	}, nil
+}
+
+func loadCA(path string) (*x509.CertPool, error) {
+	data, err := ioutil.ReadFile(path)
+	if err != nil {
+		return nil, err
+	}
+	cas := x509.NewCertPool()
+	cas.AppendCertsFromPEM(data)
+	return cas, nil
+}
+
+// func buildServerTLSConfig(config *Configuration) (*tls.Config, error) {
+// 	var certs []tls.Certificate
+// 	for _, b := range config.Backends {
+// 		cert, err := tls.LoadX509KeyPair(b.ServerTLSConfig.Cert, b.ServerTLSConfig.Key)
+// 		if err != nil {
+// 			return nil, err
+// 		}
+// 		certs = append(certs, cert)
+// 	}
+
+// 	c := &tls.Config{
+// 		Certificates: certs,
+// 	}
+
+// 	if config.CA != "" {
+// 		cas, err := loadCA(config.CA)
+// 		if err != nil {
+// 			return nil, err
+// 		}
+// 		c.ClientAuth = tls.RequireAndVerifyClientCert
+// 		c.ClientCAs = cas
+// 	}
+
+// 	c.BuildNameToCertificate()
+
+// 	return c, nil
+// }
+
+// Configuration for the proxy.
+type Configuration struct {
+	SessionAuthKey []byte `yaml:"session_auth_key"`
+	SessionEncKey  []byte `yaml:"session_enc_key"`
+	CA             string `yaml:"ca"`
+
+	SSOLoginServerURL string `yaml:"sso_server_url"`
+	SSOPublicKeyFile  string `yaml:"sso_public_key_file"`
+	SSODomain         string `yaml:"sso_domain"`
+
+	Backends []*Backend `yaml:"backends"`
+}
+
+// Sanity checks for the configuration.
+func (c *Configuration) check() error {
+	switch len(c.SessionAuthKey) {
+	case 32, 64:
+	case 0:
+		return errors.New("session_auth_key is empty")
+	default:
+		return errors.New("session_auth_key must be a random string of 32 or 64 bytes")
+	}
+	switch len(c.SessionEncKey) {
+	case 16, 24, 43:
+	case 0:
+		return errors.New("session_enc_key is empty")
+	default:
+		return errors.New("session_enc_key must be a random string of 16, 24 or 32 bytes")
+	}
+	if c.SSOLoginServerURL == "" {
+		return errors.New("sso_server_url is empty")
+	}
+	if c.SSODomain == "" {
+		return errors.New("sso_domain is empty")
+	}
+	return nil
+}
+
+// NewProxy builds a SSO-protected multi-host handler with the
+// specified configuration.
+func NewProxy(config *Configuration) (http.Handler, error) {
+	if err := config.check(); err != nil {
+		return nil, err
+	}
+
+	pkey, err := ioutil.ReadFile(config.SSOPublicKeyFile)
+	if err != nil {
+		return nil, err
+	}
+
+	w, err := httpsso.NewSSOWrapper(config.SSOLoginServerURL, pkey, config.SSODomain, config.SessionAuthKey, config.SessionEncKey)
+	if err != nil {
+		return nil, err
+	}
+
+	r := mux.NewRouter()
+	for _, b := range config.Backends {
+		h, err := b.newHandler(w)
+		if err != nil {
+			return nil, fmt.Errorf("error for host %s: %v", b.Host, err)
+		}
+		r.Host(b.Host).Handler(h)
+	}
+	return r, nil
+}

proxy/transport.go

+package proxy
+
+import (
+	"crypto/tls"
+	"errors"
+	"log"
+	"math/rand"
+	"net"
+	"net/http"
+	"sort"
+	"sync"
+)
+
+func resolveIPs(hosts []string) []string {
+	var resolved []string
+	for _, hostport := range hosts {
+		host, port, err := net.SplitHostPort(hostport)
+		if err != nil {
+			log.Printf("error parsing %s: %v", hostport, err)
+			continue
+		}
+		hostIPs, err := net.LookupIP(host)
+		if err != nil {
+			log.Printf("error resolving %s: %v", host, err)
+			continue
+		}
+		for _, ip := range hostIPs {
+			resolved = append(resolved, net.JoinHostPort(ip.String(), port))
+		}
+	}
+	return resolved
+}
+
+type balancer struct {
+	mx   sync.Mutex
+	ips  []string
+	errs []uint64
+}
+
+func (b *balancer) incrError(index int) {
+	b.mx.Lock()
+	b.errs[index]++
+	b.mx.Unlock()
+}
+
+func (b *balancer) dial(network, addr string) (net.Conn, error) {
+	ips, err := b.pickIPs()
+	if err != nil {
+		return nil, err
+	}
+	for _, s := range ips {
+		conn, err := net.Dial(network, s.ip)
+		if err == nil {
+			return conn, nil
+		}
+		log.Printf("error connecting to %s: %v", s.ip, err)
+		b.incrError(s.index)
+	}
+	return nil, errors.New("all upstream connections failed")
+}
+
+type ipScore struct {
+	ip    string
+	score int
+	index int
+}
+
+type ipScoreList []ipScore
+
+func (l ipScoreList) Len() int           { return len(l) }
+func (l ipScoreList) Swap(i, j int)      { l[i], l[j] = l[j], l[i] }
+func (l ipScoreList) Less(i, j int) bool { return l[i].score < l[j].score }
+
+func shuffleScores(scores []ipScore) {
+	for i, j := range rand.Perm(len(scores)) {
+		scores[i], scores[j] = scores[j], scores[i]
+	}
+}