9-SpringSecurity：登录时的图片验证码

原创已于 2024-04-11 14:42:38 修改 · 836 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#java #开发语言

于 2024-04-11 14:42:37 首次发布

2024年程序员学习专栏收录该内容

187 篇文章

订阅专栏

public class LoginController {

// 模拟数据库或缓存，存储验证码

private Map<String, String> captchaStore = new ConcurrentHashMap<>();

private final AuthenticationManagerBuilder authenticationManagerBuilder;

public LoginController(AuthenticationManagerBuilder authenticationManagerBuilder) {

this.authenticationManagerBuilder = authenticationManagerBuilder;

}

@GetMapping(“/login”)

public String loginPage(Model model) {

// gif类型

// GifCaptcha captcha = new GifCaptcha(130, 48);

// 中文类型

// ChineseCaptcha captcha = new ChineseCaptcha(130, 48);

// 中文gif类型

// ChineseGifCaptcha captcha = new ChineseGifCaptcha(130, 48);

// 算术类型

ArithmeticCaptcha captcha = new ArithmeticCaptcha(130, 48);

// 几位数运算(默认是两位)

captcha.setLen(2);

// 获取生成的文本

String code = captcha.text().equals(“0.0”) ? “0” : captcha.text();

// 验证码对应的UUID

String uuid = UUID.randomUUID().toString();

log.info(“Captcha generated: {}, UUID generated:{}”, code, uuid);

// 保存生成的验证码信息

captchaStore.put(uuid, code);

// 返回验证码信息

model.addAttribute(“img”, captcha.toBase64());

model.addAttribute(“uuid”, uuid);

return “login”;

}

@PostMapping(“/signin”)

@ResponseBody

public String signin(FormUser formUser, HttpServletRequest request) throws Exception {

log.info(“formUser: {}”, formUser);

// 查询验证码

String code = (String) captchaStore.get(formUser.getUuid());

// 清除验证码

captchaStore.remove(formUser.getUuid());

// 核验验证码

if (StringUtils.isEmptyOrWhitespace(code)) {

throw new Exception(“EXPIRED”);

}

if (StringUtils.isEmptyOrWhitespace(formUser.getCode()) || !formUser.getCode().equalsIgnoreCase(code)) {

throw new Exception(“INVALID”);

}

// 核验用户名密码

UsernamePasswordAuthenticationToken authenticationToken = new UsernamePasswordAuthenticationToken(formUser.getUsername(), formUser.getPassword());

Authentication authentication = authenticationManagerBuilder.getObject().authenticate(authenticationToken);

SecurityContextHolder.getContext().setAuthentication(authentication);

log.info(“authentication: {}”, authentication);

return SecurityContextHolder.getContext().getAuthentication().getPrincipal().toString();

}

@RequestMapping(value = “/”)

@ResponseBody

public String loginSuccess() {

log.info(SecurityContextHolder.getContext().getAuthentication().toString());

return SecurityContextHolder.getContext().getAuthentication().toString();

}

@GetMapping(value = “/user/add”)

@ResponseBody

public String accessResource1() {

return " Access Resource 1: Add User";

}

@GetMapping(value = “/user/query”)

@ResponseBody

public String accessResource2() {

return " Access Resource 2: Query User";

}

用户信息

为了集中焦点在本篇的主题验证码上，避免引入其他复杂性，这里采用内存型用户信息来演示，关于从数据库中获取用户信息，可参考6-SpringSecurity：数据库存储用户信息。

@Component

public class CustomUserDetailsService implements UserDetailsService {

@Override

public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {

return User.withUsername(“dev”).password(new BCryptPasswordEncoder().encode(“123”)).authorities(“p1”, “p2”).build();

}

安全配置

关于 SpringSecurity 的配置如下，主要是对资源进行保护，完成权限核验：

@Configuration

public class SecurityConfig extends WebSecurityConfigurerAdapter {

@Bean

public PasswordEncoder passwordEncoder() {

return new BCryptPasswordEncoder();

}

@Override

protected void configure(HttpSecurity http) throws Exception {

http.authorizeRequests()

.antMatchers(“/user/add”).hasAuthority(“p1”)

.antMatchers(“/user/**”).authenticated()

.anyRequest().permitAll() // Let other request pass

.and()

.csrf().disable() // turn off csrf, or will be 403 forbidden

.formLogin()

.loginPage(“/login”);

}

前端页面

用户名

密码

验证码

Note:

获取验证码的请求应该放行，这里 .anyRequest().permitAll() 包含了登录页的其他URL。
以上仅实现了 SpringSecurity 集成验证码的核心逻辑，实际使用时，密码甚至用户名应该密文传输：前端加密，后端解密，再交由 SpringSecurity 进行认证；
前端页面点击验证码刷新可自行实现；