package com.util.MD5;

import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Arrays;

public class MD5Util {

	private static final String HEX_NUMS_STR = "0123456789ABCDEF";
	private static final Integer SALT_LENGTH = 12;

	/**
	 * 将16进制字符串转换成字节数组
	 * 
	 * @param hex
	 * @return
	 */
	private static byte[] hexStringToByte(String hex) {
		int len = (hex.length() / 2);
		byte[] result = new byte[len];
		char[] hexChars = hex.toCharArray();
		for (int i = 0; i < len; i++) {
			int pos = i * 2;
			result[i] = (byte) (HEX_NUMS_STR.indexOf(hexChars[pos]) << 4 | HEX_NUMS_STR.indexOf(hexChars[pos + 1]));
		}
		return result;
	}

	/**
	 * 将指定byte数组转换成16进制字符串
	 * 
	 * @param b
	 * @return
	 */
	private static String byteToHexString(byte[] b) {
		StringBuffer hexString = new StringBuffer();
		for (int i = 0; i < b.length; i++) {
			String hex = Integer.toHexString(b[i] & 0xFF);
			if (hex.length() == 1) {
				hex = '0' + hex;
			}
			hexString.append(hex.toUpperCase());
		}
		return hexString.toString();
	}

	/**
	 * 验证是否合法
	 * 
	 * @param originalString
	 * @param denseString
	 * @return
	 * @throws NoSuchAlgorithmException
	 * @throws UnsupportedEncodingException
	 */
	public static boolean validation(String originalString, String denseString) throws NoSuchAlgorithmException, UnsupportedEncodingException {
		//将16进制字符串格式口令转换成字节数组  
		byte[] pwdInDb = hexStringToByte(denseString);
		//声明盐变量  
		byte[] salt = new byte[SALT_LENGTH];
		//将盐从数据库中保存的口令字节数组中提取出来  
		System.arraycopy(pwdInDb, 0, salt, 0, SALT_LENGTH);
		//创建消息摘要对象  
		MessageDigest md = MessageDigest.getInstance("MD5");
		//将盐数据传入消息摘要对象  
		md.update(salt);
		//将口令的数据传给消息摘要对象  
		md.update(originalString.getBytes("UTF-8"));
		//生成输入口令的消息摘要  
		byte[] digest = md.digest();
		//声明一个保存数据库中口令消息摘要的变量  
		byte[] digestInDb = new byte[pwdInDb.length - SALT_LENGTH];
		//取得数据库中口令的消息摘要  
		System.arraycopy(pwdInDb, SALT_LENGTH, digestInDb, 0, digestInDb.length);
		//比较根据输入口令生成的消息摘要和数据库中消息摘要是否相同  
		System.out.println(byteToHexString(denseString.getBytes()));
		if (Arrays.equals(digest, digestInDb)) {
			//口令正确返回口令匹配消息  
			return true;
		} else {
			//口令不正确返回口令不匹配消息  
			return false;
		}
	}

	/**
	 * 获得加密后的16进制形式
	 * 
	 * @param originalString
	 * @return
	 * @throws NoSuchAlgorithmException
	 * @throws UnsupportedEncodingException
	 */
	public static String getEncryptedPwd(String originalString) {
		byte[] pwd = null;
		try {
			//声明加密后的口令数组变量  
			//随机数生成器  
			SecureRandom random = new SecureRandom();
			//声明盐数组变量  
			byte[] salt = new byte[SALT_LENGTH];
			//将随机数放入盐变量中  
			random.nextBytes(salt);

			//声明消息摘要对象  
			MessageDigest md = null;
			//创建消息摘要  
			md = MessageDigest.getInstance("MD5");
			//将盐数据传入消息摘要对象  
			md.update(salt);
			//将口令的数据传给消息摘要对象  
			md.update(originalString.getBytes("UTF-8"));
			//获得消息摘要的字节数组  
			byte[] digest = md.digest();

			//因为要在口令的字节数组中存放盐，所以加上盐的字节长度  
			pwd = new byte[digest.length + SALT_LENGTH];
			//将盐的字节拷贝到生成的加密口令字节数组的前12个字节，以便在验证口令时取出盐  
			System.arraycopy(salt, 0, pwd, 0, SALT_LENGTH);
			//将消息摘要拷贝到加密口令字节数组从第13个字节开始的字节  
			System.arraycopy(digest, 0, pwd, SALT_LENGTH, digest.length);
			//将字节数组格式加密后的口令转化为16进制字符串格式的口令  
		} catch (NoSuchAlgorithmException e) {

		} catch (UnsupportedEncodingException e) {

		}
		return byteToHexString(pwd);
	}

	public final static String MD5Passwd(String s) {
		char hexDigits[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F' };
		try {
			byte[] btInput = s.getBytes();
			// 获得MD5摘要算法的 MessageDigest 对象
			MessageDigest mdInst = MessageDigest.getInstance("MD5");
			// 使用指定的字节更新摘要
			mdInst.update(btInput);
			// 获得密文
			byte[] md = mdInst.digest();
			// 把密文转换成十六进制的字符串形式
			int j = md.length;
			char str[] = new char[j * 2];
			int k = 0;
			for (int i = 0; i < j; i++) {
				byte byte0 = md[i];
				str[k++] = hexDigits[byte0 >>> 4 & 0xf];
				str[k++] = hexDigits[byte0 & 0xf];
			}
			return new String(str);
		} catch (Exception e) {
			e.printStackTrace();
			return null;
		}
	}
}
