如何使用Java生成图像验证码

验证码生成是一个常见的需求，通常用于防止自动化脚本或机器人提交表单。在Java中，可以使用多种方式生成验证码，包括生成图像验证码和短信验证码。下面我将介绍如何使用Java生成图像验证码。

图像验证码生成

图像验证码通常包括随机生成的数字或字母，并以图像的形式呈现给用户。

使用Java生成图像验证码的示例：

package com.conch.hrx.cb.other;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.Random;

public class CaptchaGenerator {

    private static final int WIDTH = 120;
    private static final int HEIGHT = 40;
    private static final int CODE_COUNT = 4;
    private static final int LINE_COUNT = 20;

    // CODE_SEQUENCE数组定义了可能的字符集
    private static final char[] CODE_SEQUENCE = { 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J',
            'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
            '1', '2', '3', '4', '5', '6', '7', '8', '9' };

    private static Random random = new Random();

    public static void main(String[] args) {
        String captcha = generateCaptcha();
        System.out.println("Generated Captcha: " + captcha);
        saveImage(captcha, "captcha.png");
    }

    /**
     * 方法生成一个包含随机字符的字符串
     * @return String 生成的字符串
     */
    public static String generateCaptcha() {
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < CODE_COUNT; i++) {
            String ch = String.valueOf(CODE_SEQUENCE[random.nextInt(CODE_SEQUENCE.length)]);
            sb.append(ch);
        }
        return sb.toString();
    }

    /**
     * 方法将验证码保存到文件中
     * @param captcha 验证码
     */
    public static void saveImage(String captcha, String fileName) {
        BufferedImage image = new BufferedImage(WIDTH, HEIGHT, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = image.createGraphics();
        g.setColor(getRandomColor());
        g.fillRect(0, 0, WIDTH, HEIGHT);
        g.setFont(new Font("Arial", Font.BOLD, 20));

        // 绘制验证码
        for (int i = 0; i < CODE_COUNT; i++) {
            g.setColor(getRandomColor());
            g.drawString(String.valueOf(captcha.charAt(i)), 20 * i + 10, 25);
        }

        // 绘制干扰线
        for (int i = 0; i < LINE_COUNT; i++) {
            int x = random.nextInt(WIDTH);
            int y = random.nextInt(HEIGHT);
            int xl = random.nextInt(20);
            int yl = random.nextInt(20);
            g.setColor(getRandomColor());
            g.drawLine(x, y, x + xl, y + yl);
        }

        g.dispose();

        try {
            // 使用ImageIO.write()方法将生成的图像保存为PNG文件
            ImageIO.write(image, "PNG", new File(fileName));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private static Color getRandomColor() {
        return new Color(random.nextInt(220), random.nextInt(220), random.nextInt(220));
    }
}

生成的验证码图片：

 

 使用 HappyCaptcha 生成验证码实现

HappyCaptcha 是一个简单的 Java 库，用于生成验证码图片。它可以帮助你快速生成带有随机字符的验证码图片，并且支持一些简单的配置选项。以下是如何使用 HappyCaptcha 生成验证码的步骤：

1. 添加 HappyCaptcha 依赖

首先，你需要在项目中添加 HappyCaptcha 的依赖。如果你使用的是 Maven，可以在 pom.xml 中添加以下依赖：

<dependency>
    <groupId>com.github.whvcse</groupId>
    <artifactId>happy-captcha</artifactId>
    <version>1.0.0</version>
</dependency>

2. 使用 HappyCaptcha 生成验证码

HappyCaptcha 提供了一个简单的方式来生成验证码图片。以下是一个基本的使用示例：

import com.github.whvcse.happy.captcha.HappyCaptcha;
import com.github.whvcse.happy.captcha.exception.HappyCaptchaException;
import com.github.whvcse.happy.captcha.type.CaptchaType;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

public class CaptchaServlet extends javax.servlet.http.HttpServlet {

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
        // 创建一个 HappyCaptcha 实例
        HappyCaptcha happyCaptcha = new HappyCaptcha();
        happyCaptcha.setLength(4); // 设置验证码长度
        happyCaptcha.setType(CaptchaType.NUMBER_AND_LETTER); // 设置验证码类型

        try {
            // out 方法用于将验证码图片输出到 HTTP 响应中
            happyCaptcha.out(req, resp);
        } catch (HappyCaptchaException e) {
            e.printStackTrace();
            resp.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, "Failed to generate captcha");
        }
    }
}

3. 验证用户输入

在用户提交表单时，你需要验证用户输入的验证码是否正确。以下是如何验证用户输入的示例：

@Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws IOException {
    String userInput = req.getParameter("captcha"); // 获取用户输入的验证码
    String sessionCaptcha = (String) req.getSession().getAttribute("captcha"); // 获取存储在Session中的验证码

    // 在用户提交表单时，从 Session 中获取存储的验证码并与用户输入进行比较
    if (sessionCaptcha != null && sessionCaptcha.equalsIgnoreCase(userInput)) {
        // 验证码正确，继续处理
        resp.getWriter().write("Captcha is correct!");
    } else {
        // 验证码错误
        resp.getWriter().write("Captcha is incorrect!");
    }
}

注意事项

• Session 管理：确保在生成验证码时将验证码存储在 Session 中，以便在验证时可以获取到。
• 安全性：验证码应该仅在验证后使用一次，并在验证后从 Session 中移除，以防止重放攻击。
• 验证码类型：CaptchaType 提供了多种验证码类型，包括数字、字母和数字字母混合。