gitlab-org--gitlab-foss/lib/api/api_guard.rb

127 lines
4 KiB
Ruby
Raw Normal View History

2014-12-19 09:15:29 -05:00
# Guard API with OAuth 2.0 Access Token
require 'rack/oauth2'
module API
module APIGuard
extend ActiveSupport::Concern
2014-12-19 09:15:29 -05:00
included do |base|
# OAuth2 Resource Server Authentication
use Rack::OAuth2::Server::Resource::Bearer, 'The API' do |request|
# The authenticator only fetches the raw token string
2014-12-19 09:15:29 -05:00
# Must yield access token to store it in the env
request.access_token
end
2014-12-19 09:15:29 -05:00
helpers HelperMethods
2014-12-19 09:15:29 -05:00
install_error_responders(base)
end
2014-12-19 09:15:29 -05:00
class_methods do
# Set the authorization scope(s) allowed for an API endpoint.
#
# A call to this method maps the given scope(s) to the current API
# endpoint class. If this method is called multiple times on the same class,
# the scopes are all aggregated.
def allow_access_with_scope(scopes, options = {})
Array(scopes).each do |scope|
allowed_scopes << Scope.new(scope, options)
end
end
def allowed_scopes
@scopes ||= []
end
end
# Helper Methods for Grape Endpoint
module HelperMethods
2017-11-17 04:09:56 -05:00
include Gitlab::Auth::UserAuthFinders
2017-11-13 10:27:30 -05:00
2017-10-12 08:38:39 -04:00
def find_current_user!
user = find_user_from_sources
2017-10-12 08:38:39 -04:00
return unless user
2017-10-12 08:38:39 -04:00
forbidden!('User is blocked') unless Gitlab::UserAccess.new(user).allowed? && user.can?(:access_api)
user
end
def find_user_from_sources
find_user_from_access_token || find_user_from_warden
end
2017-11-07 13:17:41 -05:00
private
# An array of scopes that were registered (using `allow_access_with_scope`)
# for the current endpoint class. It also returns scopes registered on
# `API::API`, since these are meant to apply to all API routes.
def scopes_registered_for_endpoint
@scopes_registered_for_endpoint ||=
begin
endpoint_classes = [options[:for].presence, ::API::API].compact
endpoint_classes.reduce([]) do |memo, endpoint|
if endpoint.respond_to?(:allowed_scopes)
memo.concat(endpoint.allowed_scopes)
else
memo
end
end
end
end
2014-12-19 09:15:29 -05:00
end
module ClassMethods
private
2014-12-19 09:15:29 -05:00
def install_error_responders(base)
2017-11-16 11:03:19 -05:00
error_classes = [Gitlab::Auth::MissingTokenError,
Gitlab::Auth::TokenNotFoundError,
Gitlab::Auth::ExpiredError,
Gitlab::Auth::RevokedError,
Gitlab::Auth::InsufficientScopeError]
2014-12-19 09:15:29 -05:00
base.__send__(:rescue_from, *error_classes, oauth2_bearer_token_error_handler) # rubocop:disable GitlabSecurity/PublicSend
end
def oauth2_bearer_token_error_handler
2017-03-31 11:11:28 -04:00
proc do |e|
response =
case e
2017-11-16 11:03:19 -05:00
when Gitlab::Auth::MissingTokenError
Rack::OAuth2::Server::Resource::Bearer::Unauthorized.new
2017-11-16 11:03:19 -05:00
when Gitlab::Auth::TokenNotFoundError
Rack::OAuth2::Server::Resource::Bearer::Unauthorized.new(
:invalid_token,
"Bad Access Token.")
2017-11-16 11:03:19 -05:00
when Gitlab::Auth::ExpiredError
Rack::OAuth2::Server::Resource::Bearer::Unauthorized.new(
:invalid_token,
"Token is expired. You can either do re-authorization or token refresh.")
2017-11-16 11:03:19 -05:00
when Gitlab::Auth::RevokedError
Rack::OAuth2::Server::Resource::Bearer::Unauthorized.new(
:invalid_token,
"Token was revoked. You have to re-authorize from the user.")
2017-11-16 11:03:19 -05:00
when Gitlab::Auth::InsufficientScopeError
# FIXME: ForbiddenError (inherited from Bearer::Forbidden of Rack::Oauth2)
# does not include WWW-Authenticate header, which breaks the standard.
Rack::OAuth2::Server::Resource::Bearer::Forbidden.new(
:insufficient_scope,
Rack::OAuth2::Server::Resource::ErrorMethods::DEFAULT_DESCRIPTION[:insufficient_scope],
{ scope: e.scopes })
end
response.finish
end
end
2014-12-19 09:15:29 -05:00
end
end
2015-02-02 22:30:09 -05:00
end