org.civis.blockchain.ssm.client.crypto.KeyPairReader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ssm-sdk-client Show documentation
A SDK HTTP Client implementation to interact with signing state machines chaincode developped for Hyperleger Fabric.
The newest version!
package org.civis.blockchain.ssm.client.crypto;

import org.bouncycastle.asn1.pkcs.RSAPrivateKey;
import org.bouncycastle.crypto.CryptoException;
import org.bouncycastle.util.io.pem.PemObject;
import org.bouncycastle.util.io.pem.PemReader;
import org.civis.blockchain.ssm.client.Utils.FileUtils;

import java.io.IOException;
import java.io.Reader;
import java.security.*;
import java.security.interfaces.RSAPublicKey;
import java.security.spec.InvalidKeySpecException;
import java.security.spec.RSAPrivateCrtKeySpec;
import java.security.spec.X509EncodedKeySpec;

public class KeyPairReader {

    public static KeyPair loadKeyPair(String filename) throws CryptoException {
        PrivateKey priv = loadPrivateKey(filename);
        PublicKey pub = loadPublicKey(filename);
        return new KeyPair(pub, priv);
    }

    public static PrivateKey loadPrivateKey(String filename) throws CryptoException {
        try {
            PemObject pem = getPemObject(filename);
            RSAPrivateKey key = RSAPrivateKey.getInstance(pem.getContent());
            KeyFactory kf = KeyFactory.getInstance("RSA");
            RSAPrivateCrtKeySpec privSpec = new RSAPrivateCrtKeySpec(key.getModulus(), key.getPublicExponent(), key.getPrivateExponent(), key.getPrime1(), key.getPrime2(), key.getExponent1(), key.getExponent2(), key.getCoefficient());
            return kf.generatePrivate(privSpec);
        } catch (InvalidKeySpecException e) {
            throw new CryptoException("Private key can't be loaded", e);
        } catch (NoSuchAlgorithmException e) {
            throw new CryptoException("Private key can't be loaded", e);
        } catch (IOException e) {
            throw new CryptoException("Private key can't be loaded", e);
        }
    }

    public static PublicKey loadPublicKey(String filename) throws CryptoException {
        try {
            if (!filename.endsWith(".pub")) {
                filename = filename.concat(".pub");
            }
            PemObject pem = getPemObject(filename);
            byte[] pubKeyBytes = pem.getContent();

            X509EncodedKeySpec pubSpec = new X509EncodedKeySpec(pubKeyBytes);
            KeyFactory kf = KeyFactory.getInstance("RSA");
            RSAPublicKey pubKey = (RSAPublicKey) kf.generatePublic(pubSpec);
            return pubKey;
        } catch (InvalidKeySpecException e) {
            throw new CryptoException("Public key can't be loaded", e);
        } catch (NoSuchAlgorithmException e) {
            throw new CryptoException("Public key can't be loaded", e);
        } catch (IOException e) {
            throw new CryptoException("Public key can't be loaded", e);
        }
    }


    public static PublicKey fromByteArray(byte[] pub) throws CryptoException {
        try {
            return KeyFactory.getInstance("RSA").generatePublic(new X509EncodedKeySpec(pub));
        } catch (InvalidKeySpecException e) {
            throw new CryptoException("Public key can't be loaded", e);
        } catch (NoSuchAlgorithmException e) {
            throw new CryptoException("Public key can't be loaded", e);
        }
    }

    public static KeyPair generateRSAKey() throws CryptoException {
        try {
            KeyPairGenerator kpg = KeyPairGenerator.getInstance("RSA");
            kpg.initialize(2048);
            return kpg.generateKeyPair();
        } catch (NoSuchAlgorithmException e) {
            throw new CryptoException("RSA key can't be generated", e);
        }
    }

    private static PemObject getPemObject(String filename) throws IOException {
        Reader reader = FileUtils.getReader(filename);
        PemReader rpem = new PemReader(reader);
        return rpem.readPemObject();
    }

}