PGPainless/pgpainless
PGPainless/pgpainless
1
0
Fork 0
mirror of https://github.com/pgpainless/pgpainless.git synced 2025-09-14 20:59:39 +02:00
Code Issues Releases Wiki Activity

Added multithreading implementation

Browse source
This commit is contained in:
DenBond7 2023-04-20 14:48:53 +03:00
parent 82212a65aa
commit 8bfbf5520a
No known key found for this signature in database
GPG key ID: F74FC4E6441BA8C3
1 changed files with 46 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

49
pgpainless-core/src/test/java/investigations/PgpDecryptMultiThreadsTest.java
View file

@ -9,13 +9,21 @@
package investigations;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Collections;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import org.bouncycastle.openpgp.PGPException;
import org.bouncycastle.openpgp.PGPPublicKeyRingCollection;
import org.bouncycastle.openpgp.PGPSecretKeyRing;
import org.bouncycastle.openpgp.PGPSecretKeyRingCollection;
import org.bouncycastle.util.io.Streams;
import org.junit.jupiter.api.Test;
@ -87,15 +95,50 @@ public class PgpDecryptMultiThreadsTest {
private static final String PASSPHRASE = "android";
private static final String ORIGINAL_TEXT = "Some text";
private static final PGPSecretKeyRing secretKeyRing;
static {
try {
secretKeyRing = PGPainless.readKeyRing().secretKeyRing(
SENDER_PRIVATE_KEY
);
} catch (IOException e) {
throw new RuntimeException(e);
}
}
@Test
public void testDecryptionInMultiThreads() throws PGPException, IOException {
public void testDecryptionInMultiThreads() throws InterruptedException {
AtomicInteger atomicInteger = new AtomicInteger();
int numberOfThreads = 10;
int numberOfAttempts = 1000;
ExecutorService service = Executors.newFixedThreadPool(numberOfThreads);
CountDownLatch latch = new CountDownLatch(numberOfThreads);
for (int i = 0; i < numberOfThreads; i++) {
service.submit(() -> {
for (int j = 0; j < numberOfAttempts; j++) {
try {
doDecryption();
atomicInteger.incrementAndGet();
} catch (Exception e) {
e.printStackTrace();
}
}
latch.countDown();
});
}
assertTrue(latch.await(300, TimeUnit.SECONDS));
assertEquals(numberOfThreads * numberOfAttempts, atomicInteger.get());
}
private static void doDecryption() throws IOException, PGPException {
SecretKeyRingProtector secretKeyRingProtector =
SecretKeyRingProtector.unlockAnyKeyWith(Passphrase.fromPassword(PASSPHRASE));
PGPPublicKeyRingCollection pgpPublicKeyRingCollection = PGPainless.readKeyRing().publicKeyRingCollection(
SENDER_PUBLIC_KEY + "\n" + RECEIVER_PUBLIC_KEY);
PGPSecretKeyRingCollection secretKeyRingCollection =
PGPainless.readKeyRing().secretKeyRingCollection(SENDER_PRIVATE_KEY);
new PGPSecretKeyRingCollection(Collections.singletonList(secretKeyRing));
try (DecryptionStream decryptionStream = PGPainless.decryptAndOrVerify()
.onInputStream(new ByteArrayInputStream(ENCRYPTED_TEXT.getBytes()))