IT评测·应用市场-qidao123.com

标题: 说说验证码功能的实现 [打印本页]

作者: 曹旭辉    时间: 2023-6-7 01:06
标题: 说说验证码功能的实现
前言

大家好,我是 god23bin,今天说说验证码功能的实现,相信大家都经常接触到验证码的,毕竟平时上网也能遇到各种验证码,需要我们输入验证码进行验证我们是人类,而不是机器人。
验证码有多种类型,比如图片验证码、短信验证码和邮件验证码等等,虽说多种类型,图片也好,短信也好,邮件也好,都是承载验证码的载体,最主要的核心就是一个验证码的生成、存储和校验。
本篇文章就从这几个方面出发说说验证码,废话不多说,下面开始正文。
实现思路

验证码验证的功能,其实现思路还是挺简单的,不论是图片验证码、短信验证码还是邮件验证码,无非就以下几点:
验证码的生成

首先,需要知道的就是验证码的生成,这就涉及到生成验证码的算法,可以自己纯手写,也可以使用人家提供的工具,这里我就介绍下面 4 种生成验证码的方式。
1. 纯原生手写生成文本验证码

需求:随机产生一个 n 位的验证码,每位可能是数字、大写字母、小写字母。
实现:本质就是随机生成字符串,字符串可包含数字、大写字母、小写字母。
准备一个包含数字、大写字母、小写字母的字符串,借助 Random 类,循环 n 次随机获取字符串的下标,就能拼接出一个随机字符组成的字符串了。
  1. package cn.god23bin.demo.util;
  2. import java.util.Random;
  3. public class MyCaptchaUtil {
  4.         /**
  5.      * 生成 n 位验证码
  6.      * @param n 位数
  7.      * @return n 位验证码
  8.      **/
  9.     public static String generateCode(int n) {
  10.         String chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
  11.         StringBuilder sb = new StringBuilder();
  12.         Random random = new Random();
  13.         for (int i = 0; i < n; i++) {
  14.             int index = random.nextInt(chars.length());
  15.             sb.append(chars.charAt(index));
  16.         }
  17.         return sb.toString();
  18.     }
  19.    
  20. }
复制代码
2. 纯原生手写生成图片验证码

实现:使用 Java 的 awt 和 swing 库来生成图片验证码。下面使用 BufferedImage 类创建一个指定大小的图片,然后随机生成 n 个字符,将其画在图片上,将生成的字符和图片验证码放到哈希表返回。后续我们就可以拿到验证码的文本值,并且可以将图片验证码输出到指定的输出流中。
  1. package cn.god23bin.demo.util;
  2. import java.awt.*;
  3. import java.awt.image.BufferedImage;
  4. import java.util.HashMap;
  5. import java.util.Map;
  6. public class MyCaptchaUtil {
  7.         /**
  8.      * 生成 n 位的图片验证码
  9.      * @param n 位数
  10.      * @return 哈希表,code 获取文本验证码,img 获取 BufferedImage 图片对象
  11.      **/
  12.     public static Map<String, Object> generateCodeImage(int n) {
  13.         int width = 100, height = 50;
  14.         BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
  15.         Graphics2D g = image.createGraphics();
  16.         g.setColor(Color.LIGHT_GRAY);
  17.         g.fillRect(0, 0, width, height);
  18.         String chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
  19.         Random random = new Random();
  20.         StringBuilder sb = new StringBuilder();
  21.         for (int i = 0; i < n; i++) {
  22.             int index = random.nextInt(chars.length());
  23.             char c = chars.charAt(index);
  24.             sb.append(c);
  25.             g.setColor(new Color(random.nextInt(255), random.nextInt(255), random.nextInt(255)));
  26.             g.setFont(new Font("Arial", Font.BOLD, 25));
  27.             g.drawString(Character.toString(c), 20 + i * 15, 25);
  28.         }
  29.         Map<String, Object> res = new HashMap<>();
  30.         res.put("code", sb.toString());
  31.         res.put("img", image);
  32.         return res;
  33.     }
  34.    
  35. }
复制代码
我们可以写一个获取验证码的接口,以二进制流输出返回给前端,前端可以直接使用 img 标签来显示我们返回的图片,只需在 src 属性赋值我们的获取验证码接口。
  1. @RequestMapping("/captcha")
  2. @RestController
  3. public class CaptchaController {
  4.     @GetMapping("/code/custom")
  5.     public void getCode(HttpServletResponse response) {
  6.         Map<String, Object> map = MyCaptchaUtil.generateCodeImage(5);
  7.         System.out.println(map.get("code"));
  8.         BufferedImage img = (BufferedImage) map.get("img");
  9.         // 设置响应头,防止缓存
  10.         response.setHeader("Cache-Control", "no-store, no-cache");
  11.         response.setContentType("image/png");
  12.         try {
  13.             ImageIO.write(img, "png", response.getOutputStream());
  14.         } catch (IOException e) {
  15.             e.printStackTrace();
  16.         }
  17.     }
  18. }
复制代码
3. 使用 Hutool 工具生成图形验证码

引入依赖:可以单独引入验证码模块或者全部模块都引入
  1. <dependency>
  2.     <groupId>cn.hutool</groupId>
  3.     <artifactId>hutool-captcha</artifactId>
  4.     <version>5.8.15</version>
  5. </dependency>
  6. <dependency>
  7.     <groupId>cn.hutool</groupId>
  8.     <artifactId>hutool-all</artifactId>
  9.     <version>5.8.15</version>
  10. </dependency>
复制代码
  1. // 设置图形验证码的宽和高,同时生成了验证码,可以通过 lineCaptcha.getCode() 获取文本验证码
  2. LineCaptcha lineCaptcha = CaptchaUtil.createLineCaptcha(200, 100);
复制代码
  1. // 设置图形验证码的宽、高、验证码字符数、干扰元素个数
  2. CircleCaptcha captcha = CaptchaUtil.createCircleCaptcha(200, 100, 4, 20);
复制代码
  1. // 定义图形验证码的宽、高、验证码字符数、干扰线宽度
  2. ShearCaptcha captcha = CaptchaUtil.createShearCaptcha(200, 100, 4, 4);
复制代码

获取验证码接口:
  1. @RequestMapping("/captcha")
  2. @RestController
  3. public class CaptchaController {
  4.     @GetMapping("/code/hutool")
  5.     public void getCodeByHutool(HttpServletResponse response) {
  6.         LineCaptcha lineCaptcha = CaptchaUtil.createLineCaptcha(200, 100);
  7.         System.out.println("线段干扰的验证码:" + lineCaptcha.getCode());
  8.         // 设置响应头,防止缓存
  9.         response.setHeader("Cache-Control", "no-store, no-cache");
  10.         response.setContentType("image/png");
  11.         try {
  12.             lineCaptcha.write(response.getOutputStream());
  13.         } catch (IOException e) {
  14.             e.printStackTrace();
  15.         }
  16.     }
  17. }
复制代码
4. 使用 Kaptcha 生成验证码

Kaptcha 是谷歌的一个生成验证码工具包,我们简单配置其属性就可以实现验证码的验证功能。
引入依赖项:它只有一个版本:2.3.2
  1. <dependency>
  2.     <groupId>com.github.penggle</groupId>
  3.     <artifactId>kaptcha</artifactId>
  4.     <version>2.3.2</version>
  5. </dependency>
复制代码
简单看看 kaptcha 属性:
属性描述默认值kaptcha.border图片边框,合法值:yes , noyeskaptcha.border.color边框颜色,合法值: r,g,b (and optional alpha) 或者 white,black,blue.blackkaptcha.border.thickness边框厚度,合法值:>01kaptcha.image.width图片宽200kaptcha.image.height图片高50kaptcha.producer.impl图片实现类com.google.code.kaptcha.impl.DefaultKaptchakaptcha.textproducer.impl文本实现类com.google.code.kaptcha.text.impl.DefaultTextCreatorkaptcha.textproducer.char.string文本集合,验证码值从此集合中获取abcde2345678gfynmnpwxkaptcha.textproducer.char.length验证码长度5kaptcha.textproducer.font.names字体Arial, Courierkaptcha.textproducer.font.size字体大小40pxkaptcha.textproducer.font.color字体颜色,合法值: r,g,b  或者 white,black,blue.blackkaptcha.textproducer.char.space文字间隔2kaptcha.noise.impl干扰实现类com.google.code.kaptcha.impl.DefaultNoisekaptcha.noise.color干扰颜色,合法值: r,g,b 或者 white,black,blue.blackkaptcha.obscurificator.impl图片样式: 水纹com.google.code.kaptcha.impl.WaterRipple 鱼眼com.google.code.kaptcha.impl.FishEyeGimpy 阴影com.google.code.kaptcha.impl.ShadowGimpycom.google.code.kaptcha.impl.WaterRipplekaptcha.background.impl背景实现类com.google.code.kaptcha.impl.DefaultBackgroundkaptcha.background.clear.from背景颜色渐变,开始颜色light greykaptcha.background.clear.to背景颜色渐变,结束颜色whitekaptcha.word.impl文字渲染器com.google.code.kaptcha.text.impl.DefaultWordRendererkaptcha.session.keysession keyKAPTCHA_SESSION_KEYkaptcha.session.datesession dateKAPTCHA_SESSION_DATE简单配置下 Kaptcha:
  1. package cn.god23bin.demo.config;
  2. import com.google.code.kaptcha.impl.DefaultKaptcha;
  3. import com.google.code.kaptcha.util.Config;
  4. import org.springframework.context.annotation.Bean;
  5. import org.springframework.context.annotation.Configuration;
  6. import java.util.Properties;
  7. @Configuration
  8. public class KaptchaConfig {
  9.     /**
  10.      * 配置生成图片验证码的bean
  11.      * @return
  12.      */
  13.     @Bean(name = "kaptchaProducer")
  14.     public DefaultKaptcha getKaptchaBean() {
  15.         DefaultKaptcha defaultKaptcha = new DefaultKaptcha();
  16.         Properties properties = new Properties();
  17.         properties.setProperty("kaptcha.border", "no");
  18.         properties.setProperty("kaptcha.textproducer.font.color", "black");
  19.         properties.setProperty("kaptcha.textproducer.char.space", "4");
  20.         properties.setProperty("kaptcha.textproducer.char.length", "4");
  21.         properties.setProperty("kaptcha.textproducer.char.string", "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789");
  22.         Config config = new Config(properties);
  23.         defaultKaptcha.setConfig(config);
  24.         return defaultKaptcha;
  25.     }
  26. }
复制代码
也是和 Hutool 一样,很简单就能生成验证码了。如下:
  1. // 生成文字验证码
  2. String text = kaptchaProducer.createText();
  3. // 生成图片验证码
  4. BufferedImage image = kaptchaProducer.createImage(text);
复制代码
获取验证码接口:
  1. @RequestMapping("/captcha")
  2. @RestController
  3. public class CaptchaController {
  4.     @Autowired
  5.     private Producer kaptchaProducer;
  6.     @GetMapping("/code/kaptcha")
  7.     public void getCodeByKaptcha(HttpServletResponse response) {
  8.         // 生成文字验证码
  9.         String text = kaptchaProducer.createText();
  10.         System.out.println("文字验证码:" + text);
  11.         // 生成图片验证码
  12.         BufferedImage image = kaptchaProducer.createImage(text);
  13.         // 设置响应头,防止缓存
  14.         response.setHeader("Cache-Control", "no-store, no-cache");
  15.         response.setContentType("image/jpeg");
  16.         try {
  17.             ImageIO.write(image, "jpg", response.getOutputStream());
  18.         } catch (IOException e) {
  19.             e.printStackTrace();
  20.         }
  21.     }
  22. }
复制代码
验证码的存储与校验

上面的验证码的生成,就仅仅是生成验证码,并没有将验证码存储在后端,所以现在我们需要做的是:将验证码存储起来,便于后续的校验对比。
那么存储到什么地方呢?如果你没接触过 Redis,那么第一次的想法可能就是存储到关系型数据库中,比如 MySQL。想当年,我最开始的想法就是这样哈哈哈。
不过,目前用得最多的就是将验证码存储到 Redis 中,好处就是减少了数据库的压力,加快了验证码的读取效率,还能轻松设置验证码的过期时间。
简单配置 Redis

引入 Redis 依赖项:
我们使用 Spring Data Redis,它提供了 RedisTemplate 和 StringRedisTemplate 模板类,简化了我们使用 Java 进行 Redis 的操作。
  1. <dependency>
  2.     <groupId>org.springframework.boot</groupId>
  3.     <artifactId>spring-boot-starter-data-redis</artifactId>
  4. </dependency>
复制代码
简单配置下 Redis:
  1. spring:
  2.   redis:
  3.     host: localhost
  4.     port: 6379
  5.     database: 1
  6.     timeout: 5000
复制代码
  1. @Configuration
  2. public class RedisConfig extends CachingConfigurerSupport {
  3.     @Bean
  4.     public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
  5.         // 大多数情况,都是选用<String, Object>
  6.         RedisTemplate<String, Object> template = new RedisTemplate<>();
  7.         template.setConnectionFactory(redisConnectionFactory);
  8.         // 使用JSON的序列化对象,对数据 key 和 value 进行序列化转换
  9.         Jackson2JsonRedisSerializer<Object> jackson2JsonRedisSerializer = new Jackson2JsonRedisSerializer<>(Object.class);
  10.         // ObjectMapper 是 Jackson 的一个工作类,作用是将 JSON 转成 Java 对象,即反序列化。或将 Java 对象转成 JSON,即序列化
  11.         ObjectMapper mapper = new ObjectMapper();
  12.         // 设置序列化时的可见性,第一个参数是选择序列化哪些属性,比如时序列化 setter? 还是 filed? 第二个参数是选择哪些修饰符权限的属性来序列化,比如 private 或者 public,这里的 any 是指对所有权限修饰的属性都可见(可序列化)
  13.         mapper.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
  14.         jackson2JsonRedisSerializer.setObjectMapper(mapper);
  15.         // 设置 RedisTemplate 模板的序列化方式为 jacksonSeial
  16.         template.setDefaultSerializer(jackson2JsonRedisSerializer);
  17.         return template;
  18.     }
  19.    
  20. }
复制代码
将验证码存储到 Redis

将验证码存储到 Redis 设置 5 分钟的过期时间,Redis 是 Key Value 这种形式存储的,所以需要约定好 Key 的命名规则。
命名的时候,为了区分为每个用户生成的验证码,所以需要一个标识,刚好可以通过当前请求的 HttpSession 中的 SessionID 作为唯一标识,拼接到 Key 的名称中。
当然,也不一定使用 SessionID 作为唯一标识,如果能知道其他的,也可以用其他的作为标识,比如拼接用户的手机号。
实现:
  1. @RequestMapping("/captcha")
  2. @RestController
  3. public class CaptchaController {
  4.     @Autowired
  5.     private Producer kaptchaProducer;
  6.     @Autowired
  7.     private RedisTemplate<String, Object> redisTemplate;
  8.     @GetMapping("/code")
  9.     public void getCode(HttpServletRequest request, HttpServletResponse response) {
  10.         // 生成文字验证码
  11.         String text = kaptchaProducer.createText();
  12.         System.out.println("文字验证码:" + text);
  13.         // 生成图片验证码
  14.         BufferedImage image = kaptchaProducer.createImage(text);
  15.         // 存储到 Redis 设置 5 分钟的过期时间
  16.         // 约定好存储的 Key 的命名规则,这里使用 code_sessionId_type_1 表示图形验证码
  17.         // Code_sessionId_Type_1:分为 3 部分,code 表明是验证码,sessionId 表明是给哪个用户的验证码,type_n 表明验证码类型,n 为 1 表示图形验证码,2 表示短信验证码,3 表示邮件验证码
  18.         String key = "code_" + request.getSession().getId() + "_type_1";
  19.         redisTemplate.opsForValue().set(key, text, 5, TimeUnit.SECONDS);
  20.         response.setHeader("Cache-Control", "no-store, no-cache");
  21.         response.setContentType("image/jpeg");
  22.         try {
  23.             ImageIO.write(image, "jpg", response.getOutputStream());
  24.         } catch (IOException e) {
  25.             e.printStackTrace();
  26.         }
  27.     }
  28. }
复制代码
上面代码中有一个额外的设计就是,由于发送的验证码有多种类型(图形验证码、短信验证码、邮件验证码),所以加多了一个 type_n 来标识当前存储的验证码是什么类型的,方便以后出现问题快速定位。
实际上,这里的命名规则,可以根据你的具体需求来定制,又比如说,登录的时候需要验证码、注册的时候也需要验证码、修改用户密码的时候也需要验证码,为了便于出现问题进行定位,也可以继续加多一个标识 when_n,n 为 1 表示注册、n 为 2 表示登录,以此类推。
校验

我们模拟登录的时候进行验证码的校验,使用一个 LoginDTO 对象来接收前端的登录相关的参数。
  1. package cn.god23bin.demo.model.domain.dto;
  2. import lombok.Data;
  3. @Data
  4. public class LoginDTO {
  5.     private String username;
  6.     private String password;
  7.     /**
  8.      * 验证码
  9.      */
  10.     private String code;
  11. }
复制代码
写一个登录接口,登录的过程中,校验用户输入的验证码。
  1. @RequestMapping("/user")
  2. @RestController
  3. public class UserController {
  4.     @Autowired
  5.     private RedisTemplate<String, Object> redisTemplate;
  6.     @PostMapping("/login")
  7.     public Result<String> login(@RequestBody LoginDTO loginDTO, HttpServletRequest request) {
  8.         if (!"root".equals(loginDTO.getUsername()) || !"123456".equals(loginDTO.getPassword())) {
  9.             return Result.fail("登录失败!账号或密码不正确!");
  10.         }
  11.         // 校验用户输入的验证码
  12.         String code = loginDTO.getCode();
  13.         String codeInRedis = (String) redisTemplate.opsForValue().get("code_" + request.getSession().getId() + "_type_1");
  14.         if (!code.equals(codeInRedis)) {
  15.             return Result.fail("验证码不正确!");
  16.         }
  17.         return Result.ok("登录成功!");
  18.     }
  19. }
复制代码
至此,便完成了验证码功能的实现。
获取验证码的安全设计

验证码功能的实现现在是OK的,但还有一点需要注意,那就是防止验证码被随意调用获取,或者被大量调用。如果不做限制,那么谁都能调用,就非常大的可能会被攻击了。
我们上面实现的验证码功能是图形验证码,是校验用户从图形验证码中看到后输入的数字字母组合跟后端生成的组合是否是一致的。对于图形验证码,到这里就可以了,不用限制(当然想限制也可以)。但是对于短信验证码,就还不可以。我们需要额外考虑一些防刷机制,以保障系统的安全性和可靠性(因为发短信是要钱的啊!)。
对于短信来说,一种常见的攻击方式是「短信轰炸」,攻击者通过自动批量提交手机号码、模拟IP等手段,对系统进行大规模的短信请求,从而消耗资源或干扰正常业务。为了应对这种情况,我们需要设计一些防刷机制。
防刷机制

目前我了解到的防刷机制有下面几种,如果你有别的方法,欢迎评论说出来噢!
至于这些机制的实现,有机会再写写,你感兴趣的话可以自己去操作试试!
总结

本篇文字就说了验证码功能的实现思路和实现,包括验证码的生成、存储、展示和校验。
最后我们也说了验证码的防刷机制,这是需要考虑的,这里的防刷机制对于使用大量不同手机号、不同 IP 地址是没效果的,依旧可以暴刷。所以这部分内容还是有待研究的。也欢迎大家在评论区说出你的看法!
最后的最后

希望各位屏幕前的靓仔靓女们给个三连!你轻轻地点了个赞,那将在我的心里世界增添一颗明亮而耀眼的星!
咱们下期再见!

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!




欢迎光临 IT评测·应用市场-qidao123.com (https://dis.qidao123.com/) Powered by Discuz! X3.4