249 lines
10 KiB
Java
249 lines
10 KiB
Java
/*
|
|
* Copyright 2021 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.key.parsing;
|
|
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
|
|
import java.io.ByteArrayInputStream;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.net.URISyntaxException;
|
|
import java.net.URL;
|
|
import java.nio.charset.StandardCharsets;
|
|
import java.nio.file.Files;
|
|
import java.security.InvalidAlgorithmParameterException;
|
|
import java.security.NoSuchAlgorithmException;
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
|
|
import org.bouncycastle.openpgp.PGPException;
|
|
import org.bouncycastle.openpgp.PGPPublicKeyRing;
|
|
import org.bouncycastle.openpgp.PGPPublicKeyRingCollection;
|
|
import org.bouncycastle.openpgp.PGPSecretKeyRing;
|
|
import org.bouncycastle.openpgp.PGPSecretKeyRingCollection;
|
|
import org.bouncycastle.openpgp.PGPUtil;
|
|
import org.junit.jupiter.api.Test;
|
|
import org.pgpainless.PGPainless;
|
|
import org.pgpainless.implementation.ImplementationFactory;
|
|
import org.pgpainless.key.collection.PGPKeyRingCollection;
|
|
import org.pgpainless.key.util.KeyRingUtils;
|
|
|
|
class KeyRingReaderTest {
|
|
|
|
@Test
|
|
public void assertThatPGPUtilsDetectAsciiArmoredData() throws IOException, PGPException {
|
|
InputStream inputStream = getClass().getClassLoader().getResourceAsStream("pub_keys_10_pieces.asc");
|
|
|
|
InputStream possiblyArmored = PGPUtil.getDecoderStream(PGPUtil.getDecoderStream(inputStream));
|
|
|
|
PGPPublicKeyRingCollection collection = new PGPPublicKeyRingCollection(
|
|
possiblyArmored, ImplementationFactory.getInstance().getKeyFingerprintCalculator());
|
|
assertEquals(10, collection.size());
|
|
}
|
|
|
|
@Test
|
|
void publicKeyRingCollectionFromStream() throws IOException, PGPException {
|
|
InputStream inputStream = getClass().getClassLoader().getResourceAsStream("pub_keys_10_pieces.asc");
|
|
PGPPublicKeyRingCollection rings = PGPainless.readKeyRing().publicKeyRingCollection(inputStream);
|
|
assertEquals(10, rings.size());
|
|
}
|
|
|
|
@Test
|
|
void publicKeyRingCollectionFromNotArmoredStream() throws IOException, PGPException,
|
|
InvalidAlgorithmParameterException, NoSuchAlgorithmException {
|
|
Collection<PGPPublicKeyRing> collection = new ArrayList<>();
|
|
|
|
for (int i = 0; i < 10; i++) {
|
|
PGPSecretKeyRing secretKeys = PGPainless.generateKeyRing().simpleEcKeyRing("user_" + i + "@encrypted.key");
|
|
collection.add(KeyRingUtils.publicKeyRingFrom(secretKeys));
|
|
}
|
|
|
|
PGPPublicKeyRingCollection originalRings = new PGPPublicKeyRingCollection(collection);
|
|
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
|
originalRings.encode(out);
|
|
|
|
ByteArrayInputStream inputStream = new ByteArrayInputStream(out.toByteArray());
|
|
PGPPublicKeyRingCollection parsedRings = PGPainless.readKeyRing().publicKeyRingCollection(inputStream);
|
|
assertEquals(10, parsedRings.size());
|
|
}
|
|
|
|
@Test
|
|
void publicKeyRingCollectionFromString() throws IOException, PGPException, URISyntaxException {
|
|
URL resource = getClass().getClassLoader().getResource("pub_keys_10_pieces.asc");
|
|
String armoredString = new String(Files.readAllBytes(new File(resource.toURI()).toPath()));
|
|
InputStream inputStream = new ByteArrayInputStream(armoredString.getBytes(StandardCharsets.UTF_8));
|
|
PGPPublicKeyRingCollection rings = PGPainless.readKeyRing().publicKeyRingCollection(inputStream);
|
|
assertEquals(10, rings.size());
|
|
}
|
|
|
|
@Test
|
|
void publicKeyRingCollectionFromBytes() throws IOException, PGPException, URISyntaxException {
|
|
URL resource = getClass().getClassLoader().getResource("pub_keys_10_pieces.asc");
|
|
byte[] bytes = Files.readAllBytes(new File(resource.toURI()).toPath());
|
|
InputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
|
|
PGPPublicKeyRingCollection rings = PGPainless.readKeyRing().publicKeyRingCollection(byteArrayInputStream);
|
|
assertEquals(10, rings.size());
|
|
}
|
|
|
|
/**
|
|
* One armored pub key
|
|
*/
|
|
@Test
|
|
void parsePublicKeysSingleArmored() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(1, getPgpPublicKeyRingsFromResource("single_pub_key_armored.asc").size());
|
|
}
|
|
|
|
/**
|
|
* One binary pub key
|
|
*/
|
|
@Test
|
|
void parsePublicKeysSingleBinary() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(1, getPgpPublicKeyRingsFromResource("single_pub_key_binary.key").size());
|
|
}
|
|
|
|
/**
|
|
* Many armored pub keys with a single -----BEGIN PGP PUBLIC KEY BLOCK-----...-----END PGP PUBLIC KEY BLOCK-----
|
|
*/
|
|
@Test
|
|
void parsePublicKeysMultiplyArmoredSingleHeader() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPgpPublicKeyRingsFromResource("10_pub_keys_armored_single_header.asc").size());
|
|
}
|
|
|
|
/**
|
|
* Many armored pub keys where each has own -----BEGIN PGP PUBLIC KEY BLOCK-----...-----END PGP PUBLIC KEY BLOCK-----
|
|
*/
|
|
@Test
|
|
void parsePublicKeysMultiplyArmoredOwnHeader() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPgpPublicKeyRingsFromResource("10_pub_keys_armored_own_header.asc").size());
|
|
}
|
|
|
|
/**
|
|
* Many armored pub keys where each has own -----BEGIN PGP PUBLIC KEY BLOCK-----...-----END PGP PUBLIC KEY BLOCK-----.
|
|
* Each of those blocks can have a different count of keys.
|
|
*/
|
|
@Test
|
|
void parsePublicKeysMultiplyArmoredOwnWithSingleHeader() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPgpPublicKeyRingsFromResource("10_pub_keys_armored_own_with_single_header.asc").size());
|
|
}
|
|
|
|
/**
|
|
* Many binary pub keys
|
|
*/
|
|
@Test
|
|
void parsePublicKeysMultiplyBinary() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPgpPublicKeyRingsFromResource("10_pub_keys_binary.key").size());
|
|
}
|
|
|
|
/**
|
|
* One armored private key
|
|
*/
|
|
@Test
|
|
void parseSecretKeysSingleArmored() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(1, getPgpSecretKeyRingsFromResource("single_prv_key_armored.asc").size());
|
|
}
|
|
|
|
/**
|
|
* One binary private key
|
|
*/
|
|
@Test
|
|
void parseSecretKeysSingleBinary() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(1, getPgpSecretKeyRingsFromResource("single_prv_key_binary.key").size());
|
|
}
|
|
|
|
/**
|
|
* Many armored private keys with a single
|
|
* -----BEGIN PGP PRIVATE KEY BLOCK-----...-----END PGP PRIVATE KEY BLOCK-----
|
|
*/
|
|
@Test
|
|
void parseSecretKeysMultiplyArmoredSingleHeader() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPgpSecretKeyRingsFromResource("10_prv_keys_armored_single_header.asc").size());
|
|
}
|
|
|
|
/**
|
|
* Many armored private keys where each has own -----BEGIN PGP PRIVATE KEY BLOCK-----...-----END PGP PRIVATE KEY BLOCK-----
|
|
*/
|
|
@Test
|
|
void parseSecretKeysMultiplyArmoredOwnHeader() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPgpSecretKeyRingsFromResource("10_prv_keys_armored_own_header.asc").size());
|
|
}
|
|
|
|
/**
|
|
* Many armored private keys where each has own -----BEGIN PGP PRIVATE KEY BLOCK-----...-----END PGP PRIVATE KEY BLOCK-----.
|
|
* Each of those blocks can have a different count of keys.
|
|
*/
|
|
@Test
|
|
void parseSecretKeysMultiplyArmoredOwnWithSingleHeader() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPgpSecretKeyRingsFromResource("10_prv_keys_armored_own_with_single_header.asc").size());
|
|
}
|
|
|
|
/**
|
|
* Many binary private keys
|
|
*/
|
|
@Test
|
|
void parseSecretKeysMultiplyBinary() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPgpSecretKeyRingsFromResource("10_prv_keys_binary.key").size());
|
|
}
|
|
|
|
/**
|
|
* Many armored keys(private or pub) where each has own -----BEGIN PGP ... KEY BLOCK-----...-----END PGP ... KEY BLOCK-----
|
|
*/
|
|
@Test
|
|
void parseKeysMultiplyArmoredOwnHeader() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPGPKeyRingsFromResource("10_prv_and_pub_keys_armored_own_header.asc").size());
|
|
}
|
|
|
|
/**
|
|
* Many armored keys(private or pub) where each has own -----BEGIN PGP ... KEY BLOCK-----...-----END PGP ... KEY BLOCK-----
|
|
* Each of those blocks can have a different count of keys.
|
|
*/
|
|
@Test
|
|
void parseKeysMultiplyArmoredOwnWithSingleHeader() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPGPKeyRingsFromResource("10_prv_and_pub_keys_armored_own_with_single_header.asc").size());
|
|
}
|
|
|
|
/**
|
|
* Many binary keys(private or pub)
|
|
*/
|
|
@Test
|
|
void parseKeysMultiplyBinary() throws IOException, PGPException, URISyntaxException {
|
|
assertEquals(10, getPGPKeyRingsFromResource("10_prv_and_pub_keys_binary.key").size());
|
|
}
|
|
|
|
private InputStream getFileInputStreamFromResource(String fileName) throws IOException, URISyntaxException {
|
|
URL resource = getClass().getClassLoader().getResource(fileName);
|
|
assert resource != null;
|
|
return new FileInputStream(new File(resource.toURI()));
|
|
}
|
|
|
|
private PGPKeyRingCollection getPGPKeyRingsFromResource(String fileName)
|
|
throws IOException, URISyntaxException, PGPException {
|
|
return PGPainless.readKeyRing().keyRingCollection(getFileInputStreamFromResource(fileName), true);
|
|
}
|
|
|
|
private PGPPublicKeyRingCollection getPgpPublicKeyRingsFromResource(String fileName)
|
|
throws IOException, URISyntaxException, PGPException {
|
|
return PGPainless.readKeyRing().publicKeyRingCollection(getFileInputStreamFromResource(fileName));
|
|
}
|
|
|
|
private PGPSecretKeyRingCollection getPgpSecretKeyRingsFromResource(String fileName)
|
|
throws IOException, URISyntaxException, PGPException {
|
|
return PGPainless.readKeyRing().secretKeyRingCollection(getFileInputStreamFromResource(fileName));
|
|
}
|
|
}
|