services/api/app/controllers/user_sessions_controller.rb

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: AGPL-3.0
   4
   5 class UserSessionsController < ApplicationController
   6   before_action :require_auth_scope, :only => [ :destroy ]
   7
   8   skip_before_action :set_cors_headers
   9   skip_before_action :find_object_by_uuid
  10   skip_before_action :render_404_if_no_object
  11
  12   respond_to :html
  13
  14   # omniauth callback method
  15   def create
  16     omniauth = request.env['omniauth.auth']
  17
  18     identity_url_ok = (omniauth['info']['identity_url'].length > 0) rescue false
  19     unless identity_url_ok
  20       # Whoa. This should never happen.
  21       logger.error "UserSessionsController.create: omniauth object missing/invalid"
  22       logger.error "omniauth: "+omniauth.pretty_inspect
  23
  24       return redirect_to login_failure_url
  25     end
  26
  27     # Only local users can create sessions, hence uuid_like_pattern
  28     # here.
  29     user = User.unscoped.where('identity_url = ? and uuid like ?',
  30                                omniauth['info']['identity_url'],
  31                                User.uuid_like_pattern).first
  32     if not user
  33       # Check for permission to log in to an existing User record with
  34       # a different identity_url
  35       Link.where("link_class = ? and name = ? and tail_uuid = ? and head_uuid like ?",
  36                  'permission',
  37                  'can_login',
  38                  omniauth['info']['email'],
  39                  User.uuid_like_pattern).each do |link|
  40         if prefix = link.properties['identity_url_prefix']
  41           if prefix == omniauth['info']['identity_url'][0..prefix.size-1]
  42             user = User.find_by_uuid(link.head_uuid)
  43             break if user
  44           end
  45         end
  46       end
  47     end
  48
  49     if not user
  50       # New user registration
  51       user = User.new(:email => omniauth['info']['email'],
  52                       :first_name => omniauth['info']['first_name'],
  53                       :last_name => omniauth['info']['last_name'],
  54                       :identity_url => omniauth['info']['identity_url'],
  55                       :is_active => Rails.configuration.new_users_are_active,
  56                       :owner_uuid => system_user_uuid)
  57       if omniauth['info']['username']
  58         user.set_initial_username(requested: omniauth['info']['username'])
  59       end
  60       act_as_system_user do
  61         user.save or raise Exception.new(user.errors.messages)
  62       end
  63     else
  64       user.email = omniauth['info']['email']
  65       user.first_name = omniauth['info']['first_name']
  66       user.last_name = omniauth['info']['last_name']
  67       if user.identity_url.nil?
  68         # First login to a pre-activated account
  69         user.identity_url = omniauth['info']['identity_url']
  70       end
  71
  72       while (uuid = user.redirect_to_user_uuid)
  73         user = User.unscoped.where(uuid: uuid).first
  74         if !user
  75           raise Exception.new("identity_url #{omniauth['info']['identity_url']} redirects to nonexistent uuid #{uuid}")
  76         end
  77       end
  78     end
  79
  80     # For the benefit of functional and integration tests:
  81     @user = user
  82
  83     # prevent ArvadosModel#before_create and _update from throwing
  84     # "unauthorized":
  85     Thread.current[:user] = user
  86
  87     user.save or raise Exception.new(user.errors.messages)
  88
  89     omniauth.delete('extra')
  90
  91     # Give the authenticated user a cookie for direct API access
  92     session[:user_id] = user.id
  93     session[:api_client_uuid] = nil
  94     session[:api_client_trusted] = true # full permission to see user's secrets
  95
  96     @redirect_to = root_path
  97     if params.has_key?(:return_to)
  98       # return_to param's format is 'remote,return_to_url'. This comes from login()
  99       # encoding the remote=zbbbb parameter passed by a client asking for a salted
 100       # token.
 101       remote, return_to_url = params[:return_to].split(',', 2)
 102       if remote !~ /^[0-9a-z]{5}$/ && remote != ""
 103         return send_error 'Invalid remote cluster id', status: 400
 104       end
 105       remote = nil if remote == ''
 106       return send_api_token_to(return_to_url, user, remote)
 107     end
 108     redirect_to @redirect_to
 109   end
 110
 111   # Omniauth failure callback
 112   def failure
 113     flash[:notice] = params[:message]
 114   end
 115
 116   # logout - Clear our rack session BUT essentially redirect to the provider
 117   # to clean up the Devise session from there too !
 118   def logout
 119     session[:user_id] = nil
 120
 121     flash[:notice] = 'You have logged off'
 122     return_to = params[:return_to] || root_url
 123     redirect_to "#{Rails.configuration.sso_provider_url}/users/sign_out?redirect_uri=#{CGI.escape return_to}"
 124   end
 125
 126   # login - Just bounce to /auth/joshid. The only purpose of this function is
 127   # to save the return_to parameter (if it exists; see the application
 128   # controller). /auth/joshid bypasses the application controller.
 129   def login
 130     if params[:remote] !~ /^[0-9a-z]{5}$/ && !params[:remote].nil?
 131       return send_error 'Invalid remote cluster id', status: 400
 132     end
 133     if current_user and params[:return_to]
 134       # Already logged in; just need to send a token to the requesting
 135       # API client.
 136       #
 137       # FIXME: if current_user has never authorized this app before,
 138       # ask for confirmation here!
 139
 140       return send_api_token_to(params[:return_to], current_user, params[:remote])
 141     end
 142     p = []
 143     p << "auth_provider=#{CGI.escape(params[:auth_provider])}" if params[:auth_provider]
 144     if params[:return_to]
 145       # Encode remote param inside callback's return_to, so that we'll get it on
 146       # create() after login.
 147       remote_param = params[:remote].nil? ? '' : params[:remote]
 148       p << "return_to=#{CGI.escape(remote_param + ',' + params[:return_to])}"
 149     end
 150     redirect_to "/auth/joshid?#{p.join('&')}"
 151   end
 152
 153   def send_api_token_to(callback_url, user, remote=nil)
 154     # Give the API client a token for making API calls on behalf of
 155     # the authenticated user
 156
 157     # Stub: automatically register all new API clients
 158     api_client_url_prefix = callback_url.match(%r{^.*?://[^/]+})[0] + '/'
 159     act_as_system_user do
 160       @api_client = ApiClient.
 161         find_or_create_by(url_prefix: api_client_url_prefix)
 162     end
 163
 164     @api_client_auth = ApiClientAuthorization.
 165       new(user: user,
 166           api_client: @api_client,
 167           created_by_ip_address: remote_ip,
 168           scopes: ["all"])
 169     @api_client_auth.save!
 170
 171     if callback_url.index('?')
 172       callback_url += '&'
 173     else
 174       callback_url += '?'
 175     end
 176     if remote.nil?
 177       token = @api_client_auth.token
 178     else
 179       token = @api_client_auth.salted_token(remote: remote)
 180     end
 181     callback_url += 'api_token=' + token
 182     redirect_to callback_url
 183   end
 184
 185   def cross_origin_forbidden
 186     send_error 'Forbidden', status: 403
 187   end
 188 end