属性类 package com.example.demo.util; import java.awt.image.BufferedImage; public class VerifyImage { // 原图 BufferedImage srcImage; // 抠图后的图 BufferedImage cutImage; // 滑块坐标点 Integer XPosition; Integer YPosition; public VerifyImage(BufferedImage srcImage, BufferedImage cutImage, Integer XPosition, Integer YPosition) { this.srcImage = srcImage; this.cutImage = cutImage; this.XPosition = XPosition; this.YPosition = YPosition; } public BufferedImage getSrcImage() { return srcImage; } public void setSrcImage(BufferedImage srcImage) { this.srcImage = srcImage; } public BufferedImage getCutImage() { return cutImage; } public void setCutImage(BufferedImage cutImage) { this.cutImage = cutImage; } public Integer getXPosition() { return XPosition; } public void setXPosition(Integer XPosition) { this.XPosition = XPosition; } public Integer getYPosition() { return YPosition; } public void setYPosition(Integer YPosition) { this.YPosition = YPosition; } }
生成验证码图片工具类 package com.example.demo.util; import org.springframework.stereotype.Component; import sun.misc.BASE64Decoder; import sun.misc.BASE64Encoder; import javax.imageio.ImageIO; import java.awt.*; import java.awt.image.BufferedImage; import java.io.*; import java.util.ArrayList; import java.util.List; import java.util.Random; @Component public class VerifyImageUtil { /** * 源文件宽度 */ private static int ORI_WIDTH = 300; /** * 源文件高度 */ private static int ORI_HEIGHT = 150; /** * 模板图宽度 */ private static int CUT_WIDTH = 50; /** * 模板图高度 */ private static int CUT_HEIGHT = 50; /** * 抠图凸起圆心 */ private static int circleR = 5; /** * 抠图内部矩形填充大小 */ private static int RECTANGLE_PADDING = 8; /** * 抠图的边框宽度 */ private static int SLIDER_IMG_OUT_PADDING = 1; /** * 根据传入的路径生成指定验证码图片 * * @param filePath * @return * @throws IOException */ public VerifyImage getVerifyImage(String filePath) throws IOException { BufferedImage srcImage = ImageIO.read(new File(filePath)); int locationX = CUT_WIDTH + new Random().nextInt(srcImage.getWidth() - CUT_WIDTH * 3); int locationY = CUT_HEIGHT + new Random().nextInt(srcImage.getHeight() - CUT_HEIGHT) / 2; BufferedImage markImage = new BufferedImage(CUT_WIDTH,CUT_HEIGHT,BufferedImage.TYPE_4BYTE_ABGR); int[][] data = getBlockData(); cutImgByTemplate(srcImage, markImage, data, locationX, locationY); return new VerifyImage(srcImage,markImage,locationX,locationY); } /** * 生成随机滑块形状 * <p> * 0 透明像素 * 1 滑块像素 * 2 阴影像素 * @return int[][] */ private static int[][] getBlockData() { int[][] data = new int[CUT_WIDTH][CUT_HEIGHT]; Random random = new Random(); //(x-a)²+(y-b)²=r² //x中心位置左右5像素随机 double x1 = RECTANGLE_PADDING + (CUT_WIDTH - 2 * RECTANGLE_PADDING) / 2.0 - 5 + random.nextInt(10); //y 矩形上边界半径-1像素移动 double y1_top = RECTANGLE_PADDING - random.nextInt(3); double y1_bottom = CUT_HEIGHT - RECTANGLE_PADDING + random.nextInt(3); double y1 = random.nextInt(2) == 1 ? y1_top : y1_bottom; double x2_right = CUT_WIDTH - RECTANGLE_PADDING - circleR + random.nextInt(2 * circleR - 4); double x2_left = RECTANGLE_PADDING + circleR - 2 - random.nextInt(2 * circleR - 4); double x2 = random.nextInt(2) == 1 ? x2_right : x2_left; double y2 = RECTANGLE_PADDING + (CUT_HEIGHT - 2 * RECTANGLE_PADDING) / 2.0 - 4 + random.nextInt(10); double po = Math.pow(circleR, 2); for (int i = 0; i < CUT_WIDTH; i++) { for (int j = 0; j < CUT_HEIGHT; j++) { //矩形区域 boolean fill; if ((i >= RECTANGLE_PADDING && i < CUT_WIDTH - RECTANGLE_PADDING) && (j >= RECTANGLE_PADDING && j < CUT_HEIGHT - RECTANGLE_PADDING)) { data[i][j] = 1; fill = true; } else { data[i][j] = 0; fill = false; } //凸出区域 double d3 = Math.pow(i - x1, 2) + Math.pow(j - y1, 2); if (d3 < po) { data[i][j] = 1; } else { if (!fill) { data[i][j] = 0; } } //凹进区域 double d4 = Math.pow(i - x2, 2) + Math.pow(j - y2, 2); if (d4 < po) { data[i][j] = 0; } } } //边界阴影 for (int i = 0; i < CUT_WIDTH; i++) { for (int j = 0; j < CUT_HEIGHT; j++) { //四个正方形边角处理 for (int k = 1; k <= SLIDER_IMG_OUT_PADDING; k++) { //左上、右上 if (i >= RECTANGLE_PADDING - k && i < RECTANGLE_PADDING && ((j >= RECTANGLE_PADDING - k && j < RECTANGLE_PADDING) || (j >= CUT_HEIGHT - RECTANGLE_PADDING - k && j < CUT_HEIGHT - RECTANGLE_PADDING +1))) { data[i][j] = 2; } //左下、右下 if (i >= CUT_WIDTH - RECTANGLE_PADDING + k - 1 && i < CUT_WIDTH - RECTANGLE_PADDING + 1) { for (int n = 1; n <= SLIDER_IMG_OUT_PADDING; n++) { if (((j >= RECTANGLE_PADDING - n && j < RECTANGLE_PADDING) || (j >= CUT_HEIGHT - RECTANGLE_PADDING - n && j <= CUT_HEIGHT - RECTANGLE_PADDING ))) { data[i][j] = 2; } } } } if (data[i][j] == 1 && j - SLIDER_IMG_OUT_PADDING > 0 && data[i][j - SLIDER_IMG_OUT_PADDING] == 0) { data[i][j - SLIDER_IMG_OUT_PADDING] = 2; } if (data[i][j] == 1 && j + SLIDER_IMG_OUT_PADDING > 0 && j + SLIDER_IMG_OUT_PADDING < CUT_HEIGHT && data[i][j + SLIDER_IMG_OUT_PADDING] == 0) { data[i][j + SLIDER_IMG_OUT_PADDING] = 2; } if (data[i][j] == 1 && i - SLIDER_IMG_OUT_PADDING > 0 && data[i - SLIDER_IMG_OUT_PADDING][j] == 0) { data[i - SLIDER_IMG_OUT_PADDING][j] = 2; } if (data[i][j] == 1 && i + SLIDER_IMG_OUT_PADDING > 0 && i + SLIDER_IMG_OUT_PADDING < CUT_WIDTH && data[i + SLIDER_IMG_OUT_PADDING][j] == 0) { data[i + SLIDER_IMG_OUT_PADDING][j] = 2; } } } return data; } /** * 裁剪区块 * 根据生成的滑块形状，对原图和裁剪块进行变色处理 * @param oriImage 原图 * @param targetImage 裁剪图 * @param blockImage 滑块 * @param x 裁剪点x * @param y 裁剪点y */ private static void cutImgByTemplate(BufferedImage oriImage, BufferedImage targetImage, int[][] blockImage, int x, int y) { for (int i = 0; i < CUT_WIDTH; i++) { for (int j = 0; j < CUT_HEIGHT; j++) { int _x = x + i; int _y = y + j; int rgbFlg = blockImage[i][j]; int rgb_ori = oriImage.getRGB(_x, _y); // 原图中对应位置变色处理 if (rgbFlg == 1) { //抠图上复制对应颜色值 targetImage.setRGB(i,j, rgb_ori); //原图对应位置颜色变化 oriImage.setRGB(_x, _y, Color.LIGHT_GRAY.getRGB()); } else if (rgbFlg == 2) { targetImage.setRGB(i, j, Color.WHITE.getRGB()); oriImage.setRGB(_x, _y, Color.GRAY.getRGB()); }else if(rgbFlg == 0){ //int alpha = 0; targetImage.setRGB(i, j, rgb_ori & 0x00ffffff); } } } } /** * 随机获取一张图片对象 * @param path * @return * @throws IOException */ public static BufferedImage getRandomImage(String path) throws IOException { File files = new File(path); File[] fileList = files.listFiles(); List<String> fileNameList = new ArrayList<>(); if (fileList!=null && fileList.length!=0){ for (File tempFile:fileList){ if (tempFile.isFile() && tempFile.getName().endsWith(".jpg")){ fileNameList.add(tempFile.getAbsolutePath().trim()); } } } Random random = new Random(); File imageFile = new File(fileNameList.get(random.nextInt(fileNameList.size()))); return ImageIO.read(imageFile); } /** * 将IMG输出为文件 * @param image * @param file * @throws Exception */ public static void writeImg(BufferedImage image, String file) throws Exception { byte[] imagedata = null; ByteArrayOutputStream bao=new ByteArrayOutputStream(); ImageIO.write(image,"png",bao); imagedata = bao.toByteArray(); FileOutputStream out = new FileOutputStream(new File(file)); out.write(imagedata); out.close(); } /** * 将图片转换为BASE64 * @param image * @return * @throws IOException */ public static String getImageBASE64(BufferedImage image) throws IOException { ByteArrayOutputStream out = new ByteArrayOutputStream(); ImageIO.write(image,"png",out); //转成byte数组 byte[] bytes = out.toByteArray(); BASE64Encoder encoder = new BASE64Encoder(); //生成BASE64编码 return encoder.encode(bytes); } /** * 将BASE64字符串转换为图片 * @param base64String * @return */ public static BufferedImage base64StringToImage(String base64String) { try { BASE64Decoder decoder=new BASE64Decoder(); byte[] bytes1 = decoder.decodeBuffer(base64String); ByteArrayInputStream bais = new ByteArrayInputStream(bytes1); return ImageIO.read(bais); } catch (IOException e) { e.printStackTrace(); } return null; } }
在controller中进行测试 package com.example.demo.controller; import com.example.demo.util.VerifyImage; import com.example.demo.util.VerifyImageUtil; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.core.io.Resource; import org.springframework.core.io.support.PathMatchingResourcePatternResolver; import org.springframework.core.io.support.ResourcePatternResolver; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.bind.annotation.RestController; import javax.imageio.ImageIO; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpSession; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.io.OutputStream; import java.util.Random; @RestController public class TestController { @Autowired VerifyImageUtil verifyImageUtil; // 定义一个Ant模式通配符的Resource查找器，可以用来查找类路径下或者文件系统中的资源。 ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver(); /** * 返回裁剪后的图及坐标 * * @param httpServletRequest * @param httpServletResponse * @throws IOException */ @RequestMapping("getCutImage") @ResponseBody public void getCutImage(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse) throws IOException { // 设置响应的信息类型 httpServletResponse.setContentType("image/png"); // 得到图片文件夹中所有路径 Resource[] resources = resolver.getResources("classpath*:/verifyImg/*"); int ranNum = new Random().nextInt(resources.length); // 通过随机的图片路径得到验证码图片 VerifyImage verifyImage = verifyImageUtil.getVerifyImage(String.valueOf(resources[ranNum].getFile())); ByteArrayOutputStream baosCutImage = new ByteArrayOutputStream(); ImageIO.write(verifyImage.getCutImage(), "png", baosCutImage); baosCutImage.flush(); byte[] imageInByteCutImage = baosCutImage.toByteArray(); baosCutImage.close(); //// session处理 HttpSession mySession = httpServletRequest.getSession(); mySession.setAttribute("ImageX", verifyImage.getXPosition()); mySession.setAttribute("ImageY", verifyImage.getYPosition()); System.out.println("ImageX:" + mySession.getAttribute("ImageX")); System.out.println("ImageY:" + mySession.getAttribute("ImageY")); OutputStream stream = httpServletResponse.getOutputStream(); stream.write(imageInByteCutImage); // 清空 stream.flush(); // 关闭 stream.close(); } /** * 返回原图 * * @param httpServletRequest * @param httpServletResponse * @throws IOException */ @RequestMapping("getSrcImage") @ResponseBody public void getSrcImage(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse) throws IOException { // 设置响应的信息类型 httpServletResponse.setContentType("image/png"); // 得到图片文件夹中所有路径 Resource[] resources = resolver.getResources("classpath*:/verifyImg/*"); int ranNum = new Random().nextInt(resources.length); // 通过随机的图片路径得到验证码图片 VerifyImage verifyImage = verifyImageUtil.getVerifyImage(String.valueOf(resources[ranNum].getFile())); ByteArrayOutputStream baosSrcImage = new ByteArrayOutputStream(); ImageIO.write(verifyImage.getSrcImage(), "png", baosSrcImage); baosSrcImage.flush(); byte[] imageInByteSrcImage = baosSrcImage.toByteArray(); baosSrcImage.close(); OutputStream stream = httpServletResponse.getOutputStream(); stream.write(imageInByteSrcImage); stream.flush(); // 清空 stream.flush(); // 关闭 stream.close(); } /** * 得到图片验证的坐标，进行有效性校对 * * @param x * @param y * @param httpServletRequest * @param httpServletResponse * @return */ @RequestMapping("verifyImage") public String verifyImage(String x, String y, HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse) { HttpSession httpSession = httpServletRequest.getSession(); String ImageX = String.valueOf(httpSession.getAttribute("ImageX")); String ImageY = String.valueOf(httpSession.getAttribute("ImageY")); // 计算验证图片坐标的误差值 int absX = Math.abs(Integer.parseInt(x) - Integer.parseInt(ImageX)); int absY = Math.abs(Integer.parseInt(y) - Integer.parseInt(ImageY)); if (absX < 6 && absY < 6) { return "验证码正确"; } return "验证码错误"; } }