to md5 fingerprint

Author: gy2006

core/src/main/java/com/flowci/core/common/helper/CipherHelper.java

@@ -23,25 +23,21 @@
 import com.jcraft.jsch.JSch;
 import com.jcraft.jsch.JSchException;
 import com.jcraft.jsch.KeyPair;
+import sun.security.util.DerInputStream;
+import sun.security.util.DerValue;
+
+import javax.crypto.Cipher;
+import javax.crypto.spec.SecretKeySpec;
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.math.BigInteger;
 import java.nio.ByteBuffer;
 import java.nio.charset.StandardCharsets;
-import java.security.KeyFactory;
-import java.security.NoSuchAlgorithmException;
-import java.security.PrivateKey;
-import java.security.PublicKey;
+import java.security.*;
 import java.security.spec.InvalidKeySpecException;
 import java.security.spec.RSAPrivateCrtKeySpec;
 import java.security.spec.RSAPublicKeySpec;
 import java.util.Base64;
-import javax.crypto.Cipher;
-import javax.crypto.spec.SecretKeySpec;
-
-import org.apache.commons.lang.NotImplementedException;
-import sun.security.util.DerInputStream;
-import sun.security.util.DerValue;
 
 public abstract class CipherHelper {
 
@@ -102,12 +98,29 @@ public static String decrypt(String encrypted, String privateKey) {
             }
         }
 
-        public static String fingerprintMd5(String publicKey) {
-            throw new NotImplementedException();
+        public static String fingerprintMd5(String publicKey) throws NoSuchAlgorithmException {
+            String derFormat = publicKey.split(" ")[1].trim();
+            MessageDigest messageDigest = MessageDigest.getInstance("MD5");
+            byte[] digest = messageDigest.digest(Base64.getDecoder().decode(derFormat));
+            final StringBuilder toRet = new StringBuilder();
+            for (int i = 0; i < digest.length; i++) {
+                if (i != 0) {
+                    toRet.append(":");
+                }
+
+                int b = digest[i] & 0xff;
+                String hex = Integer.toHexString(b);
+
+                if (hex.length() == 1) {
+                    toRet.append("0");
+                }
+                toRet.append(hex);
+            }
+            return toRet.toString();
         }
 
         private static PrivateKey toPrivateKey(String key)
-            throws NoSuchAlgorithmException, InvalidKeySpecException, IOException {
+                throws NoSuchAlgorithmException, InvalidKeySpecException, IOException {
             KeyFactory keyFactory = KeyFactory.getInstance("RSA");
 
             String content = key.replaceAll("\\n", "").replace(RsaPrivateKeyStart, "").replace(RsaPrivateKeyEnd, "");
@@ -127,7 +140,7 @@ private static PrivateKey toPrivateKey(String key)
             BigInteger crtCoef = seq[8].getBigInteger();
 
             RSAPrivateCrtKeySpec keySpec =
-                new RSAPrivateCrtKeySpec(modulus, publicExp, privateExp, prime1, prime2, exp1, exp2, crtCoef);
+                    new RSAPrivateCrtKeySpec(modulus, publicExp, privateExp, prime1, prime2, exp1, exp2, crtCoef);
 
             return keyFactory.generatePrivate(keySpec);
         }
@@ -136,7 +149,7 @@ private static PrivateKey toPrivateKey(String key)
          * from <type><space><base64data><space><comment> to public key
          */
         private static PublicKey toPublicKey(String sshPublicKey)
-            throws NoSuchAlgorithmException, InvalidKeySpecException {
+                throws NoSuchAlgorithmException, InvalidKeySpecException {
             String[] line = sshPublicKey.trim().split(" ", 3);
             String type = line[0];
             String content = line[1];

core/src/test/java/com/flowci/core/test/common/CipherHelperTest.java

@@ -27,6 +27,7 @@
 
 import java.io.IOException;
 import java.io.InputStream;
+import java.security.NoSuchAlgorithmException;
 
 import static com.flowci.core.common.helper.CipherHelper.RSA.fingerprintMd5;
 
@@ -61,7 +62,7 @@ public void should_encrypt_decrypt_by_rsa() {
     }
 
     @Test
-    public void should_create_public_key_fingerprint() throws IOException {
+    public void should_create_public_key_fingerprint() throws IOException, NoSuchAlgorithmException {
         InputStream in = CipherHelper.class.getClassLoader().getResourceAsStream("pk_fingerprint");
         String publicKey = StringHelper.toString(in);
         Assert.assertEquals("09:e6:ce:d3:ba:a3:ee:75:9e:96:7b:55:12:85:c6:4e", fingerprintMd5(publicKey));