Java实现验证码生成工具类-CheckSumBuilder

向右看齐 2023-02-21 06:35 2阅读 0赞

### Java实现验证码生成工具类-CheckSumBuilder ###

*  背景
 *  代码实现
 *  单元测试

# 背景 #

一版在Web系统登录、表单提交及资源下载等关键功能为了防止黑客遍历破解都会用到验证码功能，下面就来了解一下Java如何简单实现验证码的生成。

# 代码实现 #

CheckSumBuilder.java

package com.utils;
    
    import java.security.MessageDigest;
    
    /** * 功能说明：验证码生成工具类 * 修改说明： * @author zheng * @date 2020年6月29日 下午1:33:26 * @version 0.1 */
    public class CheckSumBuilder { 
        /** * 功能说明：计算并获取CheckSum * 修改说明： * @author zheng * @date 2020年6月29日 下午1:33:49 * @param appSecret 密码 * @param nonce 随机串 * @param curTime 当前时间戳 * @return 返回生成的验证码 */
        public static String getCheckSum(String appSecret, String nonce, String curTime) { 
            return encode("sha1", appSecret + nonce + curTime);
        }
    
        /** * 功能说明：对参数进行MD5加密 * 修改说明： * @author zheng * @date 2020年6月29日 下午1:34:49 * @param requestBody 要加密的内容 * @return 返回MD5加密后的字符串 */
        public static String getMD5(String requestBody) { 
            return encode("md5", requestBody);
        }
    
        /** * 功能说明：使用指定加密算法对字符串进行加密 * 修改说明： * @author zheng * @date 2020年6月29日 下午1:35:45 * @param algorithm 加密算法 * @param value 要加密的字符串 * @return 返回加密后的字符串 */
        private static String encode(String algorithm, String value) { 
            if (value == null) { 
                return null;
            }
            try { 
                MessageDigest messageDigest
                        = MessageDigest.getInstance(algorithm);
                messageDigest.update(value.getBytes());
                return getFormattedText(messageDigest.digest());
            } catch (Exception e) { 
                throw new RuntimeException(e);
            }
        }
    
        /** * 功能说明：把字节数组格式化为16进制字符串 * 修改说明： * @author zheng * @date 2020年6月29日 下午1:37:10 * @param bytes 字节数组 * @return 返回格式化后的字符串 */
        private static String getFormattedText(byte[] bytes) { 
            int len = bytes.length;
            StringBuilder buf = new StringBuilder(len * 2);
            for (int j = 0; j < len; j++) { 
                buf.append(HEX_DIGITS[(bytes[j] >> 4) & 0x0f]);
                buf.append(HEX_DIGITS[bytes[j] & 0x0f]);
            }
            return buf.toString();
        }
    
        private static final char[] HEX_DIGITS = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
    }

# 单元测试 #

CheckSumBuilderTest.java

package com.utils;
    
    import java.util.Date;
    import java.util.Random;
    
    import org.junit.Test;
    
    /** * 功能说明：CheckSumBuilder单元测试类 * 修改说明： * @author zheng * @date 2020年6月29日 下午1:54:27 * @version 0.1 */
    public class CheckSumBuilderTest { 
    	
    	/** * 功能说明：验证码生成单元测试方法 * 修改说明： * @author zheng * @date 2020年6月29日 下午1:49:32 */
    	@Test
    	public void testGetCheckSum() { 
    		Random r = new Random();
    		Date now = new Date();
    		String appSecret = "zheng123";
    		String nonce = "" + r.nextLong();
    		String curTime = "" + now.getTime();
    		String checkSum = CheckSumBuilder.getCheckSum(appSecret, nonce, curTime);
    		System.out.println("checkSum = [" + checkSum.substring(0, 5) + "]");
    	}
    }