1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
13 "git.arvados.org/arvados.git/lib/controller/rpc"
14 "git.arvados.org/arvados.git/sdk/go/arvados"
15 "git.arvados.org/arvados.git/sdk/go/auth"
16 "git.arvados.org/arvados.git/sdk/go/httpserver"
19 type loginController interface {
20 Login(ctx context.Context, opts arvados.LoginOptions) (arvados.LoginResponse, error)
21 Logout(ctx context.Context, opts arvados.LogoutOptions) (arvados.LogoutResponse, error)
22 UserAuthenticate(ctx context.Context, options arvados.UserAuthenticateOptions) (arvados.APIClientAuthorization, error)
25 func chooseLoginController(cluster *arvados.Cluster, railsProxy *railsProxy) loginController {
26 wantGoogle := cluster.Login.Google.Enable
27 wantOpenIDConnect := cluster.Login.OpenIDConnect.Enable
28 wantSSO := cluster.Login.SSO.Enable
29 wantPAM := cluster.Login.PAM.Enable
30 wantLDAP := cluster.Login.LDAP.Enable
32 case wantGoogle && !wantOpenIDConnect && !wantSSO && !wantPAM && !wantLDAP:
33 return &oidcLoginController{
35 RailsProxy: railsProxy,
36 Issuer: "https://accounts.google.com",
37 ClientID: cluster.Login.Google.ClientID,
38 ClientSecret: cluster.Login.Google.ClientSecret,
39 UseGooglePeopleAPI: cluster.Login.Google.AlternateEmailAddresses,
41 EmailVerifiedClaim: "email_verified",
43 case !wantGoogle && wantOpenIDConnect && !wantSSO && !wantPAM && !wantLDAP:
44 return &oidcLoginController{
46 RailsProxy: railsProxy,
47 Issuer: cluster.Login.OpenIDConnect.Issuer,
48 ClientID: cluster.Login.OpenIDConnect.ClientID,
49 ClientSecret: cluster.Login.OpenIDConnect.ClientSecret,
50 EmailClaim: cluster.Login.OpenIDConnect.EmailClaim,
51 EmailVerifiedClaim: cluster.Login.OpenIDConnect.EmailVerifiedClaim,
52 UsernameClaim: cluster.Login.OpenIDConnect.UsernameClaim,
54 case !wantGoogle && !wantOpenIDConnect && wantSSO && !wantPAM && !wantLDAP:
55 return &ssoLoginController{railsProxy}
56 case !wantGoogle && !wantOpenIDConnect && !wantSSO && wantPAM && !wantLDAP:
57 return &pamLoginController{Cluster: cluster, RailsProxy: railsProxy}
58 case !wantGoogle && !wantOpenIDConnect && !wantSSO && !wantPAM && wantLDAP:
59 return &ldapLoginController{Cluster: cluster, RailsProxy: railsProxy}
61 return errorLoginController{
62 error: errors.New("configuration problem: exactly one of Login.Google, Login.OpenIDConnect, Login.SSO, Login.PAM, and Login.LDAP must be enabled"),
67 // Login and Logout are passed through to the wrapped railsProxy;
68 // UserAuthenticate is rejected.
69 type ssoLoginController struct{ *railsProxy }
71 func (ctrl *ssoLoginController) UserAuthenticate(ctx context.Context, opts arvados.UserAuthenticateOptions) (arvados.APIClientAuthorization, error) {
72 return arvados.APIClientAuthorization{}, httpserver.ErrorWithStatus(errors.New("username/password authentication is not available"), http.StatusBadRequest)
75 type errorLoginController struct{ error }
77 func (ctrl errorLoginController) Login(context.Context, arvados.LoginOptions) (arvados.LoginResponse, error) {
78 return arvados.LoginResponse{}, ctrl.error
80 func (ctrl errorLoginController) Logout(context.Context, arvados.LogoutOptions) (arvados.LogoutResponse, error) {
81 return arvados.LogoutResponse{}, ctrl.error
83 func (ctrl errorLoginController) UserAuthenticate(context.Context, arvados.UserAuthenticateOptions) (arvados.APIClientAuthorization, error) {
84 return arvados.APIClientAuthorization{}, ctrl.error
87 func noopLogout(cluster *arvados.Cluster, opts arvados.LogoutOptions) (arvados.LogoutResponse, error) {
88 target := opts.ReturnTo
90 if cluster.Services.Workbench2.ExternalURL.Host != "" {
91 target = cluster.Services.Workbench2.ExternalURL.String()
93 target = cluster.Services.Workbench1.ExternalURL.String()
96 return arvados.LogoutResponse{RedirectLocation: target}, nil
99 func createAPIClientAuthorization(ctx context.Context, conn *rpc.Conn, rootToken string, authinfo rpc.UserSessionAuthInfo) (arvados.APIClientAuthorization, error) {
100 ctxRoot := auth.NewContext(ctx, &auth.Credentials{Tokens: []string{rootToken}})
101 resp, err := conn.UserSessionCreate(ctxRoot, rpc.UserSessionCreateOptions{
102 // Send a fake ReturnTo value instead of the caller's
103 // opts.ReturnTo. We won't follow the resulting
104 // redirect target anyway.
105 ReturnTo: ",https://none.invalid",
109 return arvados.APIClientAuthorization{}, err
111 target, err := url.Parse(resp.RedirectLocation)
113 return arvados.APIClientAuthorization{}, err
115 token := target.Query().Get("api_token")
116 return conn.APIClientAuthorizationCurrent(auth.NewContext(ctx, auth.NewCredentials(token)), arvados.GetOptions{})