2020-10-25 19:54:03 +01:00
|
|
|
/*
|
|
|
|
* Copyright 2020 Paul Schaub.
|
|
|
|
*
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
|
|
|
package org.pgpainless.util;
|
|
|
|
|
|
|
|
import java.io.ByteArrayInputStream;
|
|
|
|
import java.io.ByteArrayOutputStream;
|
|
|
|
import java.io.IOException;
|
2020-11-18 12:20:59 +01:00
|
|
|
import java.io.InputStream;
|
2021-05-08 14:02:44 +02:00
|
|
|
import java.util.Iterator;
|
2020-10-25 19:54:03 +01:00
|
|
|
|
|
|
|
import org.bouncycastle.bcpg.ArmoredOutputStream;
|
2021-05-08 14:02:44 +02:00
|
|
|
import org.bouncycastle.openpgp.PGPKeyRing;
|
2020-10-25 19:54:03 +01:00
|
|
|
import org.bouncycastle.openpgp.PGPPublicKeyRing;
|
|
|
|
import org.bouncycastle.openpgp.PGPSecretKeyRing;
|
|
|
|
import org.bouncycastle.util.io.Streams;
|
2021-05-08 14:02:44 +02:00
|
|
|
import org.pgpainless.key.OpenPgpV4Fingerprint;
|
2020-10-25 19:54:03 +01:00
|
|
|
|
2020-11-20 12:01:39 +01:00
|
|
|
public class ArmorUtils {
|
2020-10-25 19:54:03 +01:00
|
|
|
|
|
|
|
public static String toAsciiArmoredString(PGPSecretKeyRing secretKeys) throws IOException {
|
2021-05-08 14:02:44 +02:00
|
|
|
MultiMap<String, String> header = keyToHeader(secretKeys);
|
|
|
|
return toAsciiArmoredString(secretKeys.getEncoded(), header);
|
2020-11-18 12:20:59 +01:00
|
|
|
}
|
2020-10-25 19:54:03 +01:00
|
|
|
|
2020-11-18 12:20:59 +01:00
|
|
|
public static String toAsciiArmoredString(PGPPublicKeyRing publicKeys) throws IOException {
|
2021-05-08 14:02:44 +02:00
|
|
|
MultiMap<String, String> header = keyToHeader(publicKeys);
|
|
|
|
return toAsciiArmoredString(publicKeys.getEncoded(), header);
|
|
|
|
}
|
|
|
|
|
|
|
|
private static MultiMap<String, String> keyToHeader(PGPKeyRing keyRing) {
|
|
|
|
MultiMap<String, String> header = new MultiMap<>();
|
|
|
|
OpenPgpV4Fingerprint fingerprint = new OpenPgpV4Fingerprint(keyRing);
|
|
|
|
Iterator<String> userIds = keyRing.getPublicKey().getUserIDs();
|
|
|
|
|
|
|
|
header.put("Comment", fingerprint.prettyPrint());
|
|
|
|
if (userIds.hasNext()) {
|
|
|
|
header.put("Comment", userIds.next());
|
|
|
|
}
|
|
|
|
return header;
|
2020-11-18 12:20:59 +01:00
|
|
|
}
|
2020-10-25 19:54:03 +01:00
|
|
|
|
2020-11-18 12:20:59 +01:00
|
|
|
public static String toAsciiArmoredString(byte[] bytes) throws IOException {
|
2021-05-08 14:02:44 +02:00
|
|
|
return toAsciiArmoredString(bytes, null);
|
|
|
|
}
|
|
|
|
|
|
|
|
public static String toAsciiArmoredString(byte[] bytes, MultiMap<String, String> additionalHeaderValues) throws IOException {
|
|
|
|
return toAsciiArmoredString(new ByteArrayInputStream(bytes), additionalHeaderValues);
|
2020-10-25 19:54:03 +01:00
|
|
|
}
|
|
|
|
|
2020-11-18 12:20:59 +01:00
|
|
|
public static String toAsciiArmoredString(InputStream inputStream) throws IOException {
|
2021-05-08 14:02:44 +02:00
|
|
|
return toAsciiArmoredString(inputStream, null);
|
|
|
|
}
|
|
|
|
|
|
|
|
public static String toAsciiArmoredString(InputStream inputStream, MultiMap<String, String> additionalHeaderValues) throws IOException {
|
2020-10-25 19:54:03 +01:00
|
|
|
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
2021-02-19 19:50:36 +01:00
|
|
|
ArmoredOutputStream armor = ArmoredOutputStreamFactory.get(out);
|
2021-05-08 14:02:44 +02:00
|
|
|
if (additionalHeaderValues != null) {
|
|
|
|
for (String header : additionalHeaderValues.keySet()) {
|
|
|
|
for (String value : additionalHeaderValues.get(header)) {
|
|
|
|
armor.addHeader(header, value);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-11-18 12:20:59 +01:00
|
|
|
Streams.pipeAll(inputStream, armor);
|
2020-10-25 19:54:03 +01:00
|
|
|
armor.close();
|
|
|
|
|
|
|
|
return out.toString();
|
|
|
|
}
|
|
|
|
}
|