1 package org
.argeo
.cms
.security
;
3 import java
.io
.IOException
;
4 import java
.io
.InputStream
;
5 import java
.io
.OutputStream
;
7 import java
.math
.BigInteger
;
8 import java
.net
.InetAddress
;
9 import java
.nio
.file
.Files
;
10 import java
.nio
.file
.Path
;
11 import java
.security
.GeneralSecurityException
;
12 import java
.security
.KeyPair
;
13 import java
.security
.KeyPairGenerator
;
14 import java
.security
.KeyStore
;
15 import java
.security
.KeyStoreException
;
16 import java
.security
.PrivateKey
;
17 import java
.security
.SecureRandom
;
18 import java
.security
.Security
;
19 import java
.security
.cert
.Certificate
;
20 import java
.security
.cert
.CertificateException
;
21 import java
.security
.cert
.X509Certificate
;
22 import java
.util
.Arrays
;
23 import java
.util
.Date
;
25 import javax
.security
.auth
.x500
.X500Principal
;
27 import org
.argeo
.api
.cms
.CmsConstants
;
28 import org
.argeo
.api
.cms
.CmsLog
;
29 import org
.argeo
.cms
.internal
.runtime
.KernelConstants
;
30 import org
.bouncycastle
.asn1
.pkcs
.PrivateKeyInfo
;
31 import org
.bouncycastle
.cert
.X509CertificateHolder
;
32 import org
.bouncycastle
.cert
.X509v3CertificateBuilder
;
33 import org
.bouncycastle
.cert
.jcajce
.JcaX509CertificateConverter
;
34 import org
.bouncycastle
.cert
.jcajce
.JcaX509v3CertificateBuilder
;
35 import org
.bouncycastle
.jce
.provider
.BouncyCastleProvider
;
36 import org
.bouncycastle
.openssl
.PEMParser
;
37 import org
.bouncycastle
.openssl
.jcajce
.JcaPEMKeyConverter
;
38 import org
.bouncycastle
.openssl
.jcajce
.JceOpenSSLPKCS8DecryptorProviderBuilder
;
39 import org
.bouncycastle
.operator
.ContentSigner
;
40 import org
.bouncycastle
.operator
.InputDecryptorProvider
;
41 import org
.bouncycastle
.operator
.OperatorCreationException
;
42 import org
.bouncycastle
.operator
.jcajce
.JcaContentSignerBuilder
;
43 import org
.bouncycastle
.pkcs
.PKCS8EncryptedPrivateKeyInfo
;
44 import org
.bouncycastle
.pkcs
.PKCSException
;
47 * Utilities around private keys and certificate, mostly wrapping BouncyCastle
50 public class PkiUtils
{
51 private final static CmsLog log
= CmsLog
.getLog(PkiUtils
.class);
53 public final static String PKCS12
= "PKCS12";
54 public static final String DEFAULT_KEYSTORE_PATH
= KernelConstants
.DIR_NODE
+ '/' + CmsConstants
.NODE
+ ".p12";
56 public static final String DEFAULT_PEM_KEY_PATH
= KernelConstants
.DIR_NODE
+ '/' + CmsConstants
.NODE
+ ".key";
58 public static final String DEFAULT_PEM_CERT_PATH
= KernelConstants
.DIR_NODE
+ '/' + CmsConstants
.NODE
+ ".crt";
60 private final static String SECURITY_PROVIDER
;
62 Security
.addProvider(new BouncyCastleProvider());
63 SECURITY_PROVIDER
= "BC";
66 public static X509Certificate
generateSelfSignedCertificate(KeyStore keyStore
, X500Principal x500Principal
,
67 int keySize
, char[] keyPassword
) {
69 KeyPairGenerator kpGen
= KeyPairGenerator
.getInstance("RSA", SECURITY_PROVIDER
);
70 kpGen
.initialize(keySize
, new SecureRandom());
71 KeyPair pair
= kpGen
.generateKeyPair();
72 Date notBefore
= new Date(System
.currentTimeMillis() - 10000);
73 Date notAfter
= new Date(System
.currentTimeMillis() + 365 * 24L * 3600 * 1000);
74 BigInteger serial
= BigInteger
.valueOf(System
.currentTimeMillis());
75 X509v3CertificateBuilder certGen
= new JcaX509v3CertificateBuilder(x500Principal
, serial
, notBefore
,
76 notAfter
, x500Principal
, pair
.getPublic());
77 ContentSigner sigGen
= new JcaContentSignerBuilder("SHA256WithRSAEncryption").setProvider(SECURITY_PROVIDER
)
78 .build(pair
.getPrivate());
79 X509Certificate cert
= new JcaX509CertificateConverter().setProvider(SECURITY_PROVIDER
)
80 .getCertificate(certGen
.build(sigGen
));
81 cert
.checkValidity(new Date());
82 cert
.verify(cert
.getPublicKey());
84 keyStore
.setKeyEntry(x500Principal
.getName(), pair
.getPrivate(), keyPassword
, new Certificate
[] { cert
});
86 } catch (GeneralSecurityException
| OperatorCreationException e
) {
87 throw new RuntimeException("Cannot generate self-signed certificate", e
);
91 public static KeyStore
getKeyStore(Path keyStoreFile
, char[] keyStorePassword
, String keyStoreType
) {
93 KeyStore store
= KeyStore
.getInstance(keyStoreType
, SECURITY_PROVIDER
);
94 if (Files
.exists(keyStoreFile
)) {
95 try (InputStream fis
= Files
.newInputStream(keyStoreFile
)) {
96 store
.load(fis
, keyStorePassword
);
102 } catch (GeneralSecurityException
| IOException e
) {
103 throw new RuntimeException("Cannot load keystore " + keyStoreFile
, e
);
107 public static void saveKeyStore(Path keyStoreFile
, char[] keyStorePassword
, KeyStore keyStore
) {
109 try (OutputStream fis
= Files
.newOutputStream(keyStoreFile
)) {
110 keyStore
.store(fis
, keyStorePassword
);
112 } catch (GeneralSecurityException
| IOException e
) {
113 throw new RuntimeException("Cannot save keystore " + keyStoreFile
, e
);
117 // public static byte[] pemToPKCS12(final String keyFile, final String cerFile, final String password)
118 // throws Exception {
119 // // Get the private key
120 // FileReader reader = new FileReader(keyFile);
122 // PEMReader pem = new PemReader(reader, new PasswordFinder() {
124 // public char[] getPassword() {
125 // return password.toCharArray();
129 // PrivateKey key = ((KeyPair) pem.readObject()).getPrivate();
134 // // Get the certificate
135 // reader = new FileReader(cerFile);
136 // pem = new PEMReader(reader);
138 // X509Certificate cert = (X509Certificate) pem.readObject();
143 // // Put them into a PKCS12 keystore and write it to a byte[]
144 // ByteArrayOutputStream bos = new ByteArrayOutputStream();
145 // KeyStore ks = KeyStore.getInstance("PKCS12");
147 // ks.setKeyEntry("alias", (Key) key, password.toCharArray(), new java.security.cert.Certificate[] { cert });
148 // ks.store(bos, password.toCharArray());
150 // return bos.toByteArray();
153 public static void loadPem(KeyStore keyStore
, Reader key
, char[] keyPassword
, Reader cert
) {
154 PrivateKey privateKey
= loadPemPrivateKey(key
, keyPassword
);
155 X509Certificate certificate
= loadPemCertificate(cert
);
157 keyStore
.setKeyEntry(certificate
.getSubjectX500Principal().getName(), privateKey
, keyPassword
,
158 new java
.security
.cert
.Certificate
[] { certificate
});
159 } catch (KeyStoreException e
) {
160 throw new RuntimeException("Cannot store PEM certificate", e
);
164 public static PrivateKey
loadPemPrivateKey(Reader reader
, char[] keyPassword
) {
165 try (PEMParser pemParser
= new PEMParser(reader
)) {
166 JcaPEMKeyConverter converter
= new JcaPEMKeyConverter().setProvider("BC");
167 Object object
= pemParser
.readObject();
168 PrivateKeyInfo privateKeyInfo
;
169 if (object
instanceof PKCS8EncryptedPrivateKeyInfo
) {
170 if (keyPassword
== null)
171 throw new IllegalArgumentException("A key password is required");
172 InputDecryptorProvider decProv
= new JceOpenSSLPKCS8DecryptorProviderBuilder().build(keyPassword
);
173 privateKeyInfo
= ((PKCS8EncryptedPrivateKeyInfo
) object
).decryptPrivateKeyInfo(decProv
);
174 } else if (object
instanceof PrivateKeyInfo
) {
175 privateKeyInfo
= (PrivateKeyInfo
) object
;
177 throw new IllegalArgumentException("Unsupported format for private key");
179 return converter
.getPrivateKey(privateKeyInfo
);
180 } catch (IOException
| OperatorCreationException
| PKCSException e
) {
181 throw new RuntimeException("Cannot read private key", e
);
185 public static X509Certificate
loadPemCertificate(Reader reader
) {
186 try (PEMParser pemParser
= new PEMParser(reader
)) {
187 X509CertificateHolder certHolder
= (X509CertificateHolder
) pemParser
.readObject();
188 X509Certificate cert
= new JcaX509CertificateConverter().setProvider(SECURITY_PROVIDER
)
189 .getCertificate(certHolder
);
191 } catch (IOException
| CertificateException e
) {
192 throw new RuntimeException("Cannot read private key", e
);
196 public static void main(String
[] args
) throws Exception
{
197 final String ALGORITHM
= "RSA";
198 final String provider
= "BC";
199 SecureRandom secureRandom
= new SecureRandom();
200 long begin
= System
.currentTimeMillis();
201 for (int i
= 512; i
< 1024; i
= i
+ 2) {
203 KeyPairGenerator keyGen
= KeyPairGenerator
.getInstance(ALGORITHM
, provider
);
204 keyGen
.initialize(i
, secureRandom
);
205 keyGen
.generateKeyPair();
206 } catch (Exception e
) {
207 System
.err
.println(i
+ " : " + e
.getMessage());
210 System
.out
.println((System
.currentTimeMillis() - begin
) + " ms");
212 // // String text = "a";
214 // "testtesttesttesttesttesttesttesttesttesttesttesttesttesttest";
216 // System.out.println(text);
217 // PrivateKey privateKey;
218 // PublicKey publicKey;
219 // char[] password = "changeit".toCharArray();
220 // String alias = "CN=test";
221 // KeyStore keyStore = KeyStore.getInstance("pkcs12");
222 // File p12file = new File("test.p12");
224 // if (!p12file.exists()) {
225 // keyStore.load(null);
226 // generateSelfSignedCertificate(keyStore, new X500Principal(alias),
228 // try (OutputStream out = new FileOutputStream(p12file)) {
229 // keyStore.store(out, password);
232 // try (InputStream in = new FileInputStream(p12file)) {
233 // keyStore.load(in, password);
234 // privateKey = (PrivateKey) keyStore.getKey(alias, password);
235 // publicKey = keyStore.getCertificateChain(alias)[0].getPublicKey();
238 // // final KeyPairGenerator keyGen =
239 // // KeyPairGenerator.getInstance(ALGORITHM);
240 // // keyGen.initialize(4096, new SecureRandom());
241 // // long begin = System.currentTimeMillis();
242 // // key = keyGen.generateKeyPair();
243 // // System.out.println((System.currentTimeMillis() - begin) + " ms");
244 // // keyStore.load(null);
245 // // keyStore.setKeyEntry("test", key.getPrivate(), password, null);
246 // // try(OutputStream out=new FileOutputStream(p12file)) {
247 // // keyStore.store(out, password);
249 // // privateKey = key.getPrivate();
250 // // publicKey = key.getPublic();
252 // Cipher encrypt = Cipher.getInstance(ALGORITHM);
253 // encrypt.init(Cipher.ENCRYPT_MODE, publicKey);
254 // byte[] encrypted = encrypt.doFinal(text.getBytes());
255 // String encryptedBase64 =
256 // Base64.getEncoder().encodeToString(encrypted);
257 // System.out.println(encryptedBase64);
258 // byte[] encryptedFromBase64 =
259 // Base64.getDecoder().decode(encryptedBase64);
261 // Cipher decrypt = Cipher.getInstance(ALGORITHM);
262 // decrypt.init(Cipher.DECRYPT_MODE, privateKey);
263 // byte[] decrypted = decrypt.doFinal(encryptedFromBase64);
264 // System.out.println(new String(decrypted));
265 // } catch (Exception e) {
266 // e.printStackTrace();
271 public static void createSelfSignedKeyStore(Path keyStorePath
, char[] keyStorePassword
, String keyStoreType
) {
272 // for (Provider provider : Security.getProviders())
273 // System.out.println(provider.getName());
274 // File keyStoreFile = keyStorePath.toFile();
275 char[] keyPwd
= Arrays
.copyOf(keyStorePassword
, keyStorePassword
.length
);
276 if (!Files
.exists(keyStorePath
)) {
278 Files
.createDirectories(keyStorePath
.getParent());
279 KeyStore keyStore
= getKeyStore(keyStorePath
, keyStorePassword
, keyStoreType
);
280 generateSelfSignedCertificate(keyStore
,
281 new X500Principal("CN=" + InetAddress
.getLocalHost().getHostName() + ",OU=UNSECURE,O=UNSECURE"),
283 saveKeyStore(keyStorePath
, keyStorePassword
, keyStore
);
284 if (log
.isDebugEnabled())
285 log
.debug("Created self-signed unsecure keystore " + keyStorePath
);
286 } catch (Exception e
) {
288 if (Files
.size(keyStorePath
) == 0)
289 Files
.delete(keyStorePath
);
290 } catch (IOException e1
) {
293 log
.error("Cannot create keystore " + keyStorePath
, e
);
296 throw new IllegalStateException("Keystore " + keyStorePath
+ " already exists");