org.bouncycastle.cms.RecipientOperator Maven / Gradle / Ivy
package org.bouncycastle.cms;
import java.io.InputStream;
import java.io.OutputStream;
import org.bouncycastle.operator.InputAEADDecryptor;
import org.bouncycastle.operator.InputDecryptor;
import org.bouncycastle.operator.MacCalculator;
import org.bouncycastle.util.io.TeeInputStream;
public class RecipientOperator
{
private final Object operator;
public RecipientOperator(InputDecryptor decryptor)
{
this.operator = decryptor;
}
public RecipientOperator(MacCalculator macCalculator)
{
this.operator = macCalculator;
}
public InputStream getInputStream(InputStream dataIn)
{
if (operator instanceof InputDecryptor)
{
return ((InputDecryptor)operator).getInputStream(dataIn);
}
else
{
return new TeeInputStream(dataIn, ((MacCalculator)operator).getOutputStream());
}
}
public boolean isAEADBased()
{
return operator instanceof InputAEADDecryptor;
}
public OutputStream getAADStream()
{
return ((InputAEADDecryptor)operator).getAADStream();
}
public boolean isMacBased()
{
return operator instanceof MacCalculator;
}
public byte[] getMac()
{
return ((MacCalculator)operator).getMac();
}
}