Merge pull request #176 from yg3630536/feature/add-rsa-ecrypt

* add code for rsa
WeBankBlockchain · Apr 24, 2020 · 39ea83c · 39ea83c
2 parents ac21454 + 67f7654
commit 39ea83c
Show file tree

Hide file tree

Showing 7 changed files with 175 additions and 3 deletions.
diff --git a/src/main/java/com/webank/weid/suite/crypto/AesCryptService.java b/src/main/java/com/webank/weid/suite/crypto/AesCryptService.java
@@ -34,7 +34,6 @@
 import org.slf4j.LoggerFactory;
 
 import com.webank.weid.exception.EncodeSuiteException;
-import com.webank.weid.suite.encode.CipherEncodeProcessor;
 import com.webank.weid.suite.entity.CryptType;
 
 /**
@@ -44,7 +43,7 @@
  */
 public class AesCryptService implements CryptService {
 
-    private static final Logger logger = LoggerFactory.getLogger(CipherEncodeProcessor.class);
+    private static final Logger logger = LoggerFactory.getLogger(AesCryptService.class);
 
     private static final String KEY_ALGORITHM = CryptType.AES.name();
 

diff --git a/src/main/java/com/webank/weid/suite/crypto/CryptServiceFactory.java b/src/main/java/com/webank/weid/suite/crypto/CryptServiceFactory.java
@@ -40,6 +40,7 @@ public class CryptServiceFactory {
 
     static {
         cryptServiceMap.put(CryptType.AES.name(), new AesCryptService());
+        cryptServiceMap.put(CryptType.RSA.name(), new RsaCryptService());
     }
 
     /**

diff --git a/src/main/java/com/webank/weid/suite/crypto/KeyGenerator.java b/src/main/java/com/webank/weid/suite/crypto/KeyGenerator.java
@@ -19,6 +19,16 @@
 
 package com.webank.weid.suite.crypto;
 
+import java.nio.charset.StandardCharsets;
+import java.security.KeyPair;
+import java.security.KeyPairGenerator;
+import java.security.NoSuchAlgorithmException;
+import java.security.SecureRandom;
+
+import org.bouncycastle.util.encoders.Base64;
+
+import com.webank.weid.suite.entity.Asymmetrickey;
+import com.webank.weid.suite.entity.CryptType;
 import com.webank.weid.util.DataToolUtils;
 
 /**
@@ -27,6 +37,8 @@
  *
  */
 public class KeyGenerator {
+
+    public static final int DEFAULT_KEY_SIZE = 1024;
 
     /**
      * 使用UUID作为秘钥.
@@ -35,4 +47,30 @@ public class KeyGenerator {
     public static String getKey() {
         return DataToolUtils.getUuId32();   
     }
+
+    /**
+     * 生成RSA非对称加密密钥.
+     * @return 返回Asymmetrickey 非对此秘钥
+     * @throws NoSuchAlgorithmException 找不到Algorithm异常
+     */
+    public static Asymmetrickey getKeyForRsa() throws NoSuchAlgorithmException {
+        // KeyPairGenerator类用于生成公钥和私钥对，基于RSA算法生成对象
+        KeyPairGenerator keyPairGen = KeyPairGenerator.getInstance(CryptType.RSA.name());
+        // 初始化密钥对生成器，密钥大小为96-1024位
+        keyPairGen.initialize(DEFAULT_KEY_SIZE, new SecureRandom());
+        // 生成一个密钥对，保存在keyPair中
+        KeyPair keyPair = keyPairGen.generateKeyPair();
+        String pub = new String(
+            Base64.encode(keyPair.getPublic().getEncoded()), 
+            StandardCharsets.UTF_8
+        );
+        String pri = new String(
+            Base64.encode(keyPair.getPrivate().getEncoded()),
+            StandardCharsets.UTF_8
+        );
+        Asymmetrickey key = new Asymmetrickey();
+        key.setPrivavteKey(pri);
+        key.setPublicKey(pub);
+        return key;
+    }
 }
diff --git a/src/main/java/com/webank/weid/suite/crypto/RsaCryptService.java b/src/main/java/com/webank/weid/suite/crypto/RsaCryptService.java
@@ -0,0 +1,58 @@
+package com.webank.weid.suite.crypto;
+
+import java.nio.charset.StandardCharsets;
+import java.security.KeyFactory;
+import java.security.PrivateKey;
+import java.security.PublicKey;
+import java.security.spec.PKCS8EncodedKeySpec;
+import java.security.spec.X509EncodedKeySpec;
+
+import javax.crypto.Cipher;
+
+import org.bouncycastle.util.encoders.Base64;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.webank.weid.exception.EncodeSuiteException;
+import com.webank.weid.suite.entity.CryptType;
+
+public class RsaCryptService implements CryptService {
+
+    private static final Logger logger = LoggerFactory.getLogger(RsaCryptService.class);
+
+    private static final String KEY_ALGORITHM = CryptType.RSA.name();
+
+    @Override
+    public String encrypt(String data, String publicKey) throws EncodeSuiteException {
+        try {
+            byte[] pubByte = Base64.decode(publicKey);
+            PublicKey pub = KeyFactory.getInstance(KEY_ALGORITHM)
+                .generatePublic(new X509EncodedKeySpec(pubByte));
+            Cipher cipher = Cipher.getInstance(KEY_ALGORITHM);
+            cipher.init(Cipher.ENCRYPT_MODE, pub);
+            byte[] encrypted = cipher.doFinal(data.getBytes(StandardCharsets.UTF_8));
+            return new String(Base64.encode(encrypted), StandardCharsets.UTF_8);
+        } catch (Exception e) {
+            logger.error("RSA encrypt error, please check the log.", e);
+            throw new EncodeSuiteException();
+        }
+    }
+
+    @Override
+    public String decrypt(String data, String privateKey) throws EncodeSuiteException {
+        try {
+            // 64位解码加密后的字符串
+            byte[] inputByte = Base64.decode(data.getBytes(StandardCharsets.UTF_8));
+            byte[] priByte = Base64.decode(privateKey);
+            PrivateKey priKey = KeyFactory.getInstance(KEY_ALGORITHM)
+                .generatePrivate(new PKCS8EncodedKeySpec(priByte));
+            // RSA解密
+            Cipher cipher = Cipher.getInstance(KEY_ALGORITHM);
+            cipher.init(Cipher.DECRYPT_MODE, priKey);
+            return new String(cipher.doFinal(inputByte), StandardCharsets.UTF_8);
+        } catch (Exception e) {
+            logger.error("RAS decrypt error, please check the log.", e);
+            throw new EncodeSuiteException();
+        }
+    }
+}
diff --git a/src/main/java/com/webank/weid/suite/entity/Asymmetrickey.java b/src/main/java/com/webank/weid/suite/entity/Asymmetrickey.java
@@ -0,0 +1,29 @@
+/*
+ *       Copyright© (2018-2020) WeBank Co., Ltd.
+ *
+ *       This file is part of weid-java-sdk.
+ *
+ *       weid-java-sdk is free software: you can redistribute it and/or modify
+ *       it under the terms of the GNU Lesser General Public License as published by
+ *       the Free Software Foundation, either version 3 of the License, or
+ *       (at your option) any later version.
+ *
+ *       weid-java-sdk is distributed in the hope that it will be useful,
+ *       but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *       GNU Lesser General Public License for more details.
+ *
+ *       You should have received a copy of the GNU Lesser General Public License
+ *       along with weid-java-sdk.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package com.webank.weid.suite.entity;
+
+import lombok.Data;
+
+@Data
+public class Asymmetrickey {
+
+    private String privavteKey;
+    private String publicKey;
+}
diff --git a/src/main/java/com/webank/weid/suite/entity/CryptType.java b/src/main/java/com/webank/weid/suite/entity/CryptType.java
@@ -25,5 +25,5 @@
  *
  */
 public enum CryptType {
-    AES;
+    AES, RSA;
 }
diff --git a/src/test/java/com/webank/weid/util/TestCrypt.java b/src/test/java/com/webank/weid/util/TestCrypt.java
@@ -0,0 +1,47 @@
+package com.webank.weid.util;
+
+import java.security.NoSuchAlgorithmException;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.webank.weid.suite.crypto.CryptServiceFactory;
+import com.webank.weid.suite.crypto.KeyGenerator;
+import com.webank.weid.suite.entity.Asymmetrickey;
+import com.webank.weid.suite.entity.CryptType;
+
+public class TestCrypt {
+
+    private static final Logger logger = LoggerFactory.getLogger(TestCrypt.class);
+
+    @Test
+    public void testAes() {
+        String key = KeyGenerator.getKey();
+        logger.info("key: {}", key);
+        String original = "{\"name\":\"zhangsan\",age:12}";
+        logger.info("original: {}", original);
+        String encrypt = CryptServiceFactory.getCryptService(CryptType.AES).encrypt(original, key);
+        logger.info("encrypt: {}", encrypt);
+        String decrypt = CryptServiceFactory.getCryptService(CryptType.AES).decrypt(encrypt, key);
+        logger.info("decrypt: {}", decrypt);
+        Assert.assertEquals(original, decrypt);
+    }
+
+    @Test
+    public void testRsa() throws NoSuchAlgorithmException {
+        Asymmetrickey key = KeyGenerator.getKeyForRsa();
+        logger.info("pub key: {}", key.getPublicKey());
+        logger.info("pri key: {}", key.getPrivavteKey());
+        String original = "{\"name\":\"zhangsan\",age:12}";
+        logger.info("original: {}", original);
+        String encrypt = CryptServiceFactory.getCryptService(CryptType.RSA)
+            .encrypt(original, key.getPublicKey());
+        logger.info("encrypt: {}", encrypt);
+        String decrypt = CryptServiceFactory.getCryptService(CryptType.RSA)
+            .decrypt(encrypt, key.getPrivavteKey());
+        logger.info("decrypt: {}", decrypt);
+        Assert.assertEquals(original, decrypt);
+    }
+}