-
Notifications
You must be signed in to change notification settings - Fork 995
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fixes #37936 - Invalidate jwt for any user or users(API)
- Loading branch information
1 parent
fb512ee
commit 8f69cb2
Showing
5 changed files
with
117 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,58 @@ | ||
module Api | ||
module V2 | ||
class RegistrationTokensController < V2::BaseController | ||
include Foreman::Controller::UsersMixin | ||
|
||
include Foreman::Controller::Parameters::User | ||
include Foreman::Controller::AutoCompleteSearch | ||
before_action :authenticate, :only => [:invalidate_jwt_tokens, :invalidate_jwt] | ||
|
||
def resource_class | ||
User | ||
end | ||
|
||
def find_resource(permission = :view_users) | ||
editing_self? ? User.find(User.current.id) : User.authorized(permission).except_hidden.find(params[:id]) | ||
end | ||
|
||
def action_permission | ||
case params[:action] | ||
when 'invalidate_jwt_tokens', 'invalidate_jwt' | ||
'edit' | ||
else | ||
super | ||
end | ||
end | ||
|
||
api :DELETE, '/users/:id/registration_tokens', N_("Invalidate all registration tokens for a specific user.") | ||
description <<-DOC | ||
The user you specify will no longer be able to register hosts by using their JWTs. | ||
DOC | ||
param :id, String, :desc => N_("ID of the user"), :required => true | ||
|
||
def invalidate_jwt | ||
@user = find_resource(:edit_users) | ||
unless @user | ||
raise ::Foreman::Exception.new(N_("No record found for %s"), params[:id]) | ||
end | ||
@user.jwt_secret&.destroy | ||
process_success _("Successfully invalidated registration tokens for %s.\n" % @user.login) | ||
end | ||
|
||
api :DELETE, "/registration_tokens", N_("Invalidate all registration tokens for multiple users.") | ||
param :search, String, :desc => N_("URL-encoded search query that selects users for which registration tokens will be invalidated. Search query example: id ^ (2, 4, 6)"), :required => true | ||
description <<-DOC | ||
The users you specify will no longer be able to register hosts by using their JWTs. | ||
DOC | ||
|
||
def invalidate_jwt_tokens | ||
raise ::Foreman::Exception.new(N_("Please provide search parameter")) if params[:search].blank? | ||
@users = resource_scope_for_index(:permission => :edit_users).except_hidden.uniq | ||
if @users.blank? | ||
raise ::Foreman::Exception.new(N_("No record found for search '%s'"), params[:search]) end | ||
JwtSecret.where(user_id: @users).destroy_all | ||
process_success _("Successfully invalidated registration tokens for %s.\n" % @users.pluck(:login).to_sentence) | ||
end | ||
end | ||
end | ||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
46 changes: 46 additions & 0 deletions
46
test/controllers/api/v2/registration_tokens_controller_test.rb
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,46 @@ | ||
require 'test_helper' | ||
|
||
class Api::V2::RegistrationTokensControllerTest < ActionController::TestCase | ||
test 'user shall invalidate tokens for self' do | ||
user = users(:one) | ||
FactoryBot.create(:jwt_secret, token: 'test_jwt_secret', user: user) | ||
delete :invalidate_jwt, params: { :id => user.id.to_s}, session: set_session_user(user) | ||
user.reload | ||
assert_response :success | ||
assert_nil user.jwt_secret | ||
end | ||
|
||
test 'user with edit permission should be able to invalidate jwt for another user' do | ||
setup_user 'edit', 'users' | ||
user = users(:one) | ||
FactoryBot.create(:jwt_secret, token: 'test_jwt_secret', user: user) | ||
delete :invalidate_jwt_tokens, params: { :search => "id ^ (#{user.id})"}, session: set_session_user(User.current) | ||
user.reload | ||
assert_response :success | ||
assert_nil user.jwt_secret | ||
end | ||
|
||
test 'user without edit permission should not be able to invalidate jwt for another user' do | ||
User.current = users(:one) | ||
user = users(:two) | ||
FactoryBot.create(:jwt_secret, token: 'test_jwt_secret', user: user) | ||
delete :invalidate_jwt_tokens, params: { :search => "id ^ (#{user.id})"}, session: set_session_user(User.current) | ||
user.reload | ||
assert_response :forbidden | ||
assert_not_nil user.jwt_secret | ||
response = JSON.parse(@response.body) | ||
assert_equal "Missing one of the required permissions: edit_users", response['error']['details'] | ||
end | ||
|
||
test 'invalidating jwt should fail without search params' do | ||
setup_user 'edit', 'users' | ||
user = users(:two) | ||
FactoryBot.create(:jwt_secret, token: 'test_jwt_secret', user: user) | ||
delete :invalidate_jwt_tokens, session: set_session_user(User.current) | ||
user.reload | ||
assert_response :error | ||
assert_not_nil user.jwt_secret | ||
response = JSON.parse(@response.body) | ||
assert_equal "ERF42-7534 [Foreman::Exception]: Please provide search parameter", response['error']['message'] | ||
end | ||
end |