admin
2020-05-20 98b1a0affd69bbe63223c21fdd2c404e8bedfccb
utils/src/main/java/com/qcloud/cmq/CMQTool.java
@@ -1,74 +1,74 @@
package com.qcloud.cmq;
import java.io.UnsupportedEncodingException;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
public class CMQTool {
   private static char[] b64c = new char[] { 'A', 'B', 'C', 'D',
         'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q',
         'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd',
         'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q',
         'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3',
         '4', '5', '6', '7', '8', '9', '+', '/' };
    private static final String CONTENT_CHARSET = "UTF-8";
    private static final String HMAC_ALGORITHM = "HmacSHA1";
    private static final String HMAC_SHA256_ALGORITHM = "HmacSHA256";
   public static String base64_encode(byte[] data) {
      StringBuffer sb = new StringBuffer();
      int len = data.length;
      int i = 0;
      int b1, b2, b3;
      while (i < len) {
         b1 = data[i++] & 0xff;
         if (i == len) {
            sb.append(b64c[b1 >>> 2]);
            sb.append(b64c[(b1 & 0x3) << 4]);
            sb.append("==");
            break;
         }
         b2 = data[i++] & 0xff;
         if (i == len) {
            sb.append(b64c[b1 >>> 2]);
            sb.append(b64c[((b1 & 0x03) << 4)
                  | ((b2 & 0xf0) >>> 4)]);
            sb.append(b64c[(b2 & 0x0f) << 2]);
            sb.append("=");
            break;
         }
         b3 = data[i++] & 0xff;
         sb.append(b64c[b1 >>> 2]);
         sb.append(b64c[((b1 & 0x03) << 4)
               | ((b2 & 0xf0) >>> 4)]);
         sb.append(b64c[((b2 & 0x0f) << 2)
               | ((b3 & 0xc0) >>> 6)]);
         sb.append(b64c[b3 & 0x3f]);
      }
      return sb.toString();
   }
   public static String sign(String src, String key,String method)
          throws NoSuchAlgorithmException, UnsupportedEncodingException, InvalidKeyException
    {
      Mac mac ;
      if( method == "sha1")
      {
           mac = Mac.getInstance(HMAC_ALGORITHM);
      }
      else
      {
         mac = Mac.getInstance(HMAC_SHA256_ALGORITHM);
      }
        SecretKeySpec secretKey = new SecretKeySpec(key.getBytes(CONTENT_CHARSET), mac.getAlgorithm());
        mac.init(secretKey);
        byte[] digest = mac.doFinal(src.getBytes(CONTENT_CHARSET));
        return new String(base64_encode(digest));
    }
}
package com.qcloud.cmq;
import java.io.UnsupportedEncodingException;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
public class CMQTool {
   private static char[] b64c = new char[] { 'A', 'B', 'C', 'D',
         'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q',
         'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd',
         'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q',
         'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3',
         '4', '5', '6', '7', '8', '9', '+', '/' };
    private static final String CONTENT_CHARSET = "UTF-8";
    private static final String HMAC_ALGORITHM = "HmacSHA1";
    private static final String HMAC_SHA256_ALGORITHM = "HmacSHA256";
   public static String base64_encode(byte[] data) {
      StringBuffer sb = new StringBuffer();
      int len = data.length;
      int i = 0;
      int b1, b2, b3;
      while (i < len) {
         b1 = data[i++] & 0xff;
         if (i == len) {
            sb.append(b64c[b1 >>> 2]);
            sb.append(b64c[(b1 & 0x3) << 4]);
            sb.append("==");
            break;
         }
         b2 = data[i++] & 0xff;
         if (i == len) {
            sb.append(b64c[b1 >>> 2]);
            sb.append(b64c[((b1 & 0x03) << 4)
                  | ((b2 & 0xf0) >>> 4)]);
            sb.append(b64c[(b2 & 0x0f) << 2]);
            sb.append("=");
            break;
         }
         b3 = data[i++] & 0xff;
         sb.append(b64c[b1 >>> 2]);
         sb.append(b64c[((b1 & 0x03) << 4)
               | ((b2 & 0xf0) >>> 4)]);
         sb.append(b64c[((b2 & 0x0f) << 2)
               | ((b3 & 0xc0) >>> 6)]);
         sb.append(b64c[b3 & 0x3f]);
      }
      return sb.toString();
   }
   public static String sign(String src, String key,String method)
          throws NoSuchAlgorithmException, UnsupportedEncodingException, InvalidKeyException
    {
      Mac mac ;
      if( method == "sha1")
      {
           mac = Mac.getInstance(HMAC_ALGORITHM);
      }
      else
      {
         mac = Mac.getInstance(HMAC_SHA256_ALGORITHM);
      }
        SecretKeySpec secretKey = new SecretKeySpec(key.getBytes(CONTENT_CHARSET), mac.getAlgorithm());
        mac.init(secretKey);
        byte[] digest = mac.doFinal(src.getBytes(CONTENT_CHARSET));
        return new String(base64_encode(digest));
    }
}