mirror of
https://github.com/pgpainless/pgpainless.git
synced 2024-12-25 04:17:59 +01:00
Only check message integrity once
This commit is contained in:
parent
baeb12918c
commit
b40476c965
1 changed files with 11 additions and 0 deletions
|
@ -11,12 +11,17 @@ import javax.annotation.Nonnull;
|
|||
import org.bouncycastle.openpgp.PGPEncryptedData;
|
||||
import org.bouncycastle.openpgp.PGPException;
|
||||
import org.pgpainless.exception.ModificationDetectionException;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
public class IntegrityProtectedInputStream extends InputStream {
|
||||
|
||||
private static final Logger LOGGER = LoggerFactory.getLogger(IntegrityProtectedInputStream.class);
|
||||
|
||||
private final InputStream inputStream;
|
||||
private final PGPEncryptedData encryptedData;
|
||||
private final ConsumerOptions options;
|
||||
private boolean closed = false;
|
||||
|
||||
public IntegrityProtectedInputStream(InputStream inputStream, PGPEncryptedData encryptedData, ConsumerOptions options) {
|
||||
this.inputStream = inputStream;
|
||||
|
@ -36,11 +41,17 @@ public class IntegrityProtectedInputStream extends InputStream {
|
|||
|
||||
@Override
|
||||
public void close() throws IOException {
|
||||
if (closed) {
|
||||
return;
|
||||
}
|
||||
closed = true;
|
||||
|
||||
if (encryptedData.isIntegrityProtected() && !options.isIgnoreMDCErrors()) {
|
||||
try {
|
||||
if (!encryptedData.verify()) {
|
||||
throw new ModificationDetectionException();
|
||||
}
|
||||
LOGGER.debug("Integrity Protection check passed");
|
||||
} catch (PGPException e) {
|
||||
throw new IOException("Failed to verify integrity protection", e);
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue