1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
17 "git.arvados.org/arvados.git/lib/controller/rpc"
18 "git.arvados.org/arvados.git/lib/ctrlctx"
19 "git.arvados.org/arvados.git/sdk/go/arvados"
20 "git.arvados.org/arvados.git/sdk/go/auth"
21 "git.arvados.org/arvados.git/sdk/go/httpserver"
24 type loginController interface {
25 Login(ctx context.Context, opts arvados.LoginOptions) (arvados.LoginResponse, error)
26 Logout(ctx context.Context, opts arvados.LogoutOptions) (arvados.LogoutResponse, error)
27 UserAuthenticate(ctx context.Context, options arvados.UserAuthenticateOptions) (arvados.APIClientAuthorization, error)
30 func chooseLoginController(cluster *arvados.Cluster, railsProxy *railsProxy) loginController {
31 wantGoogle := cluster.Login.Google.Enable
32 wantOpenIDConnect := cluster.Login.OpenIDConnect.Enable
33 wantSSO := cluster.Login.SSO.Enable
34 wantPAM := cluster.Login.PAM.Enable
35 wantLDAP := cluster.Login.LDAP.Enable
36 wantTest := cluster.Login.Test.Enable
38 case 1 != countTrue(wantGoogle, wantOpenIDConnect, wantSSO, wantPAM, wantLDAP, wantTest):
39 return errorLoginController{
40 error: errors.New("configuration problem: exactly one of Login.Google, Login.OpenIDConnect, Login.SSO, Login.PAM, Login.LDAP, and Login.Test must be enabled"),
43 return &oidcLoginController{
45 RailsProxy: railsProxy,
46 Issuer: "https://accounts.google.com",
47 ClientID: cluster.Login.Google.ClientID,
48 ClientSecret: cluster.Login.Google.ClientSecret,
49 UseGooglePeopleAPI: cluster.Login.Google.AlternateEmailAddresses,
51 EmailVerifiedClaim: "email_verified",
53 case wantOpenIDConnect:
54 return &oidcLoginController{
56 RailsProxy: railsProxy,
57 Issuer: cluster.Login.OpenIDConnect.Issuer,
58 ClientID: cluster.Login.OpenIDConnect.ClientID,
59 ClientSecret: cluster.Login.OpenIDConnect.ClientSecret,
60 EmailClaim: cluster.Login.OpenIDConnect.EmailClaim,
61 EmailVerifiedClaim: cluster.Login.OpenIDConnect.EmailVerifiedClaim,
62 UsernameClaim: cluster.Login.OpenIDConnect.UsernameClaim,
65 return &ssoLoginController{railsProxy}
67 return &pamLoginController{Cluster: cluster, RailsProxy: railsProxy}
69 return &ldapLoginController{Cluster: cluster, RailsProxy: railsProxy}
71 return &testLoginController{Cluster: cluster, RailsProxy: railsProxy}
73 return errorLoginController{
74 error: errors.New("BUG: missing case in login controller setup switch"),
79 func countTrue(vals ...bool) int {
81 for _, val := range vals {
89 // Login and Logout are passed through to the wrapped railsProxy;
90 // UserAuthenticate is rejected.
91 type ssoLoginController struct{ *railsProxy }
93 func (ctrl *ssoLoginController) UserAuthenticate(ctx context.Context, opts arvados.UserAuthenticateOptions) (arvados.APIClientAuthorization, error) {
94 return arvados.APIClientAuthorization{}, httpserver.ErrorWithStatus(errors.New("username/password authentication is not available"), http.StatusBadRequest)
97 type errorLoginController struct{ error }
99 func (ctrl errorLoginController) Login(context.Context, arvados.LoginOptions) (arvados.LoginResponse, error) {
100 return arvados.LoginResponse{}, ctrl.error
102 func (ctrl errorLoginController) Logout(context.Context, arvados.LogoutOptions) (arvados.LogoutResponse, error) {
103 return arvados.LogoutResponse{}, ctrl.error
105 func (ctrl errorLoginController) UserAuthenticate(context.Context, arvados.UserAuthenticateOptions) (arvados.APIClientAuthorization, error) {
106 return arvados.APIClientAuthorization{}, ctrl.error
109 func noopLogout(cluster *arvados.Cluster, opts arvados.LogoutOptions) (arvados.LogoutResponse, error) {
110 target := opts.ReturnTo
112 if cluster.Services.Workbench2.ExternalURL.Host != "" {
113 target = cluster.Services.Workbench2.ExternalURL.String()
115 target = cluster.Services.Workbench1.ExternalURL.String()
118 return arvados.LogoutResponse{RedirectLocation: target}, nil
121 func createAPIClientAuthorization(ctx context.Context, conn *rpc.Conn, rootToken string, authinfo rpc.UserSessionAuthInfo) (resp arvados.APIClientAuthorization, err error) {
122 ctxRoot := auth.NewContext(ctx, &auth.Credentials{Tokens: []string{rootToken}})
123 newsession, err := conn.UserSessionCreate(ctxRoot, rpc.UserSessionCreateOptions{
124 // Send a fake ReturnTo value instead of the caller's
125 // opts.ReturnTo. We won't follow the resulting
126 // redirect target anyway.
127 ReturnTo: ",https://none.invalid",
133 target, err := url.Parse(newsession.RedirectLocation)
137 token := target.Query().Get("api_token")
138 tx, err := ctrlctx.CurrentTx(ctx)
143 if strings.Contains(token, "/") {
144 tokenparts := strings.Split(token, "/")
145 if len(tokenparts) >= 3 {
146 tokensecret = tokenparts[2]
149 var exp sql.NullString
151 err = tx.QueryRowxContext(ctx, "select uuid, api_token, expires_at, scopes from api_client_authorizations where api_token=$1", tokensecret).Scan(&resp.UUID, &resp.APIToken, &exp, &scopes)
155 resp.ExpiresAt = exp.String
157 err = json.Unmarshal(scopes, &resp.Scopes)
159 return resp, fmt.Errorf("unmarshal scopes: %s", err)