oauth2_proxy/main.go

117 lines
5.0 KiB
Go
Raw Normal View History

2012-12-11 01:59:23 +00:00
package main
import (
"flag"
2012-12-17 18:38:33 +00:00
"fmt"
2012-12-11 01:59:23 +00:00
"log"
2014-08-07 20:16:39 +00:00
"os"
2015-03-20 03:03:00 +00:00
"runtime"
2012-12-11 01:59:23 +00:00
"strings"
"time"
2012-12-11 01:59:23 +00:00
2014-11-09 19:51:10 +00:00
"github.com/BurntSushi/toml"
"github.com/mreiferson/go-options"
2012-12-11 01:59:23 +00:00
)
func main() {
2015-05-21 03:23:48 +00:00
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
2015-05-21 06:50:21 +00:00
flagSet := flag.NewFlagSet("oauth2_proxy", flag.ExitOnError)
2014-11-09 19:51:10 +00:00
emailDomains := StringArray{}
2014-11-09 19:51:10 +00:00
upstreams := StringArray{}
skipAuthRegex := StringArray{}
2014-11-09 19:51:10 +00:00
config := flagSet.String("config", "", "path to config file")
showVersion := flagSet.Bool("version", false, "print version string")
flagSet.String("http-address", "127.0.0.1:4180", "[http://]<addr>:<port> or unix://<path> to listen on for HTTP clients")
2015-06-08 01:51:47 +00:00
flagSet.String("https-address", ":443", "<addr>:<port> to listen on for HTTPS clients")
flagSet.String("tls-cert", "", "path to certificate file")
flagSet.String("tls-key", "", "path to private key file")
2014-11-09 19:51:10 +00:00
flagSet.String("redirect-url", "", "the OAuth Redirect URL. ie: \"https://internalapp.yourcompany.com/oauth2/callback\"")
flagSet.Var(&upstreams, "upstream", "the http url(s) of the upstream endpoint. If multiple, routing is based on path")
flagSet.Bool("pass-basic-auth", true, "pass HTTP Basic Auth, X-Forwarded-User and X-Forwarded-Email information to upstream")
flagSet.Bool("pass-access-token", false, "pass OAuth access_token to upstream via X-Forwarded-Access-Token header")
2015-03-17 19:15:15 +00:00
flagSet.Bool("pass-host-header", true, "pass the request Host Header to upstream")
flagSet.Var(&skipAuthRegex, "skip-auth-regex", "bypass authentication for requests path's that match (may be given multiple times)")
2014-11-09 19:51:10 +00:00
flagSet.Var(&emailDomains, "email-domain", "authenticate emails with the specified domain (may be given multiple times). Use * to authenticate any email")
2015-05-21 03:23:48 +00:00
flagSet.String("github-org", "", "restrict logins to members of this organisation")
flagSet.String("github-team", "", "restrict logins to members of this team")
2015-05-21 06:50:21 +00:00
flagSet.String("client-id", "", "the OAuth Client ID: ie: \"123456.apps.googleusercontent.com\"")
2014-11-09 19:51:10 +00:00
flagSet.String("client-secret", "", "the OAuth Client Secret")
flagSet.String("authenticated-emails-file", "", "authenticate against emails via file (one per line)")
flagSet.String("htpasswd-file", "", "additionally authenticate against a htpasswd file. Entries must be created with \"htpasswd -s\" for SHA encryption")
flagSet.Bool("display-htpasswd-form", true, "display username / password login form if an htpasswd file is provided")
2015-03-17 22:06:06 +00:00
flagSet.String("custom-templates-dir", "", "path to custom html templates")
flagSet.String("proxy-prefix", "/oauth2", "the url root path that this proxy should be nested under (e.g. /<oauth2>/sign_in)")
flagSet.String("cookie-name", "_oauth2_proxy", "the name of the cookie that the oauth_proxy creates")
2014-11-09 19:51:10 +00:00
flagSet.String("cookie-secret", "", "the seed string for secure cookies")
2014-11-10 02:07:02 +00:00
flagSet.String("cookie-domain", "", "an optional cookie domain to force cookies to (ie: .yourcompany.com)*")
2014-11-09 19:51:10 +00:00
flagSet.Duration("cookie-expire", time.Duration(168)*time.Hour, "expire timeframe for cookie")
flagSet.Duration("cookie-refresh", time.Duration(0), "refresh the cookie when less than this much time remains before expiration; 0 to disable")
2015-03-18 03:13:45 +00:00
flagSet.Bool("cookie-secure", true, "set secure (HTTPS) cookie flag")
flagSet.Bool("cookie-httponly", true, "set HttpOnly cookie flag")
2014-11-09 19:51:10 +00:00
flagSet.Bool("request-logging", true, "Log requests to stdout")
2015-06-08 01:51:47 +00:00
flagSet.String("provider", "google", "OAuth provider")
flagSet.String("login-url", "", "Authentication endpoint")
flagSet.String("redeem-url", "", "Token redemption endpoint")
flagSet.String("profile-url", "", "Profile access endpoint")
2015-05-08 21:13:35 +00:00
flagSet.String("validate-url", "", "Access token validation endpoint")
flagSet.String("scope", "", "Oauth scope specification")
2014-11-09 19:51:10 +00:00
flagSet.Parse(os.Args[1:])
2014-11-10 02:07:02 +00:00
if *showVersion {
2015-05-21 06:50:21 +00:00
fmt.Printf("oauth2_proxy v%s (built with %s)\n", VERSION, runtime.Version())
2014-11-10 02:07:02 +00:00
return
}
2014-11-09 19:51:10 +00:00
opts := NewOptions()
2014-11-15 04:06:07 +00:00
cfg := make(EnvOptions)
2014-11-09 19:51:10 +00:00
if *config != "" {
_, err := toml.DecodeFile(*config, &cfg)
if err != nil {
log.Fatalf("ERROR: failed to load config file %s - %s", *config, err)
}
}
2014-11-15 04:06:07 +00:00
cfg.LoadEnvForStruct(opts)
2014-11-09 19:51:10 +00:00
options.Resolve(opts, flagSet, cfg)
2012-12-11 01:59:23 +00:00
2014-11-09 19:51:10 +00:00
err := opts.Validate()
2012-12-11 01:59:23 +00:00
if err != nil {
2014-11-09 19:51:10 +00:00
log.Printf("%s", err)
os.Exit(1)
2012-12-11 01:59:23 +00:00
}
validator := NewValidator(opts.EmailDomains, opts.AuthenticatedEmailsFile)
2014-11-09 19:51:10 +00:00
oauthproxy := 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, ", "))
} else if opts.EmailDomains[0] != "*" {
oauthproxy.SignInMessage = fmt.Sprintf("Authenticate using %v", opts.EmailDomains[0])
2014-11-09 05:26:52 +00:00
}
2012-12-11 01:59:23 +00:00
}
2014-11-09 19:51:10 +00:00
if opts.HtpasswdFile != "" {
2014-11-10 02:07:02 +00:00
log.Printf("using htpasswd file %s", opts.HtpasswdFile)
2014-11-09 19:51:10 +00:00
oauthproxy.HtpasswdFile, err = NewHtpasswdFromFile(opts.HtpasswdFile)
oauthproxy.DisplayHtpasswdForm = opts.DisplayHtpasswdForm
2012-12-17 18:38:33 +00:00
if err != nil {
2014-11-09 19:51:10 +00:00
log.Fatalf("FATAL: unable to open %s %s", opts.HtpasswdFile, err)
2012-12-17 18:38:33 +00:00
}
2012-12-11 01:59:23 +00:00
}
2014-11-09 19:51:10 +00:00
2015-06-08 01:51:47 +00:00
s := &Server{
Handler: LoggingHandler(os.Stdout, oauthproxy, opts.RequestLogging),
Opts: opts,
}
2015-06-08 01:51:47 +00:00
s.ListenAndServe()
2012-12-11 01:59:23 +00:00
}