Bảo mật mùa xuân: Đặt GrantedAuthorities

Question

Có cách nào để đặt List<GrantedAuthority> trong đối tượng Authentication/UserDetailsImpl không? Trong ứng dụng của tôi, tôi có hai lớp bảo mật, một lớp để đăng nhập (sử dụng trình xác thực đăng nhập tùy chỉnh của tôi, trong lớp tôi đặt đối tượng Authentication sử dụng UsernamePasswordAuthenticationToken) và một cho "câu hỏi thách thức" nơi người dùng được nhắc trả lời một câu hỏi cụ thể.

Điều tôi muốn làm là thêm GrantedAuthority vào List<GrantedAuthority> hiện tại, được tạo trong quá trình đăng nhập, sau khi người dùng trả lời câu hỏi thách thức.

Điều này có khả thi không?

Answer 1

bạn có thể làm điều đó với đoạn mã sau:

Collection<SimpleGrantedAuthority> oldAuthorities = (Collection<SimpleGrantedAuthority>)SecurityContextHolder.getContext().getAuthentication().getAuthorities(); 
SimpleGrantedAuthority authority = new SimpleGrantedAuthority("ROLE_ANOTHER"); 
List<SimpleGrantedAuthority> updatedAuthorities = new ArrayList<SimpleGrantedAuthority>(); 
updatedAuthorities.add(authority); 
updatedAuthorities.addAll(oldAuthorities); 

SecurityContextHolder.getContext().setAuthentication(
     new UsernamePasswordAuthenticationToken(
       SecurityContextHolder.getContext().getAuthentication().getPrincipal(), 
       SecurityContextHolder.getContext().getAuthentication().getCredentials(), 
       updatedAuthorities) 
); 

Answer 2

Phương thức UserDetails.getAuthorities() chỉ trả về đối tượng Collection<GrantedAuthority>. Bạn có thể sử dụng phương thức Collection thích hợp để thêm quyền hạn mới của bạn vào bộ sưu tập đó.

Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal(); 
if (principal instanceof UserDetails) { 
    ((UserDetails) principal).getAuthorities().add(New GrantedAuthorityImpl("ROLE_FOO")); 
} 

Selah.