com.clickntap.utils.SecurityUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of click_framework Show documentation
Java Framework based on Spring Framework, Freemarker and Simplicity
The newest version!
package com.clickntap.utils;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.StringReader;
import java.security.MessageDigest;
import java.security.spec.AlgorithmParameterSpec;
import java.util.Base64;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;

import org.apache.commons.codec.binary.Hex;
import org.apache.tools.ant.util.ReaderInputStream;

public class SecurityUtils {
  private static final String DEFAULT_IV = "0123456789abcdef";
  private static final String DEFAULT_CIPHER_ALGORITHM = "AES/CTR/NoPadding";
  private static final String DEFAULT_ALGORITHM = "AES";

  public static byte[] decrypt(byte[] input, byte[] key, String algorithm, String cipherAlgorithm, AlgorithmParameterSpec paramSpec) throws Exception {
    SecretKeySpec secretKey = new SecretKeySpec(key, algorithm);
    Cipher cipher = Cipher.getInstance(cipherAlgorithm);
    cipher.init(Cipher.DECRYPT_MODE, secretKey, paramSpec);
    return cipher.doFinal(input);
  }

  public static byte[] decrypt(byte[] input, byte[] key) throws Exception {
    return decrypt(input, key, DEFAULT_ALGORITHM, DEFAULT_CIPHER_ALGORITHM, new IvParameterSpec(DEFAULT_IV.getBytes()));
  }

  public static byte[] encrypt(byte[] input, byte[] key, String algorithm, String cipherAlgorithm, AlgorithmParameterSpec paramSpec) throws Exception {
    SecretKeySpec secretKey = new SecretKeySpec(key, algorithm);
    Cipher cipher = Cipher.getInstance(cipherAlgorithm);
    cipher.init(Cipher.ENCRYPT_MODE, secretKey, paramSpec);
    return cipher.doFinal(input);
  }

  public static byte[] encrypt(byte[] input, byte[] key) throws Exception {
    return encrypt(input, key, DEFAULT_ALGORITHM, DEFAULT_CIPHER_ALGORITHM, new IvParameterSpec(DEFAULT_IV.getBytes()));
  }

  public static SecretKey generateKey(int size) throws Exception {
    return generateKey(size, DEFAULT_ALGORITHM);
  }

  public static SecretKey generateKey(int size, String algorithm) throws Exception {
    KeyGenerator kgen = KeyGenerator.getInstance(algorithm);
    kgen.init(size);
    return kgen.generateKey();
  }

  private static String encode(String s, String encoding) throws Exception {
    MessageDigest md = MessageDigest.getInstance(encoding);
    md.reset();
    md.update(s.getBytes(ConstUtils.UTF_8));
    return new String(Hex.encodeHex(md.digest()));
  }

  public static String md5(String s) throws Exception {
    StringReader reader = new StringReader(s);
    ReaderInputStream in = new ReaderInputStream(reader);
    String md5 = md5(in);
    in.close();
    reader.close();
    return md5;
  }

  public static String sha1(String s) throws Exception {
    return encode(s, "SHA-1");
  }

  public static String sha256(String s) throws Exception {
    return encode(s, "SHA-256");
  }

  public static String base64enc(byte[] s) throws Exception {
    return new String(Base64.getEncoder().encode(s));
  }

  public static byte[] base64dec(String s) throws Exception {
    return Base64.getDecoder().decode(s.getBytes(ConstUtils.UTF_8));
  }

  public static String md5(File f) throws Exception {
    String md5;
    InputStream in = new FileInputStream(f);
    md5 = SecurityUtils.md5(in);
    in.close();
    return md5;
  }

  public static String md5(InputStream in) throws Exception {
    byte[] buffer = new byte[1024];
    MessageDigest md = MessageDigest.getInstance("MD5");
    int numRead;
    do {
      numRead = in.read(buffer);
      if (numRead > 0) {
        md.update(buffer, 0, numRead);
      }
    } while (numRead != -1);
    return new String(Hex.encodeHex(md.digest()));
  }
}