← Back to branch summary

~brian-thomason/+junk/bouncycastle

« back to all changes in this revision

Viewing changes to src/org/bouncycastle/cms/RecipientOperator.java

  • Committer: Brian Thomason
  • Date: 2011-12-20 17:20:32 UTC
  • Revision ID: brian.thomason@canonical.com-20111220172032-rdtm13jgdxtksacr
Initial import

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/org/bouncycastle/cms/RecipientOperator.java
 
1
package org.bouncycastle.cms;
 
2
 
 
3
import java.io.InputStream;
 
4
 
 
5
import org.bouncycastle.asn1.x509.AlgorithmIdentifier;
 
6
import org.bouncycastle.operator.InputDecryptor;
 
7
import org.bouncycastle.operator.MacCalculator;
 
8
import org.bouncycastle.util.io.TeeInputStream;
 
9
 
 
10
public class RecipientOperator
 
11
{
 
12
    private final AlgorithmIdentifier algorithmIdentifier;
 
13
    private final Object operator;
 
14
 
 
15
    public RecipientOperator(InputDecryptor decryptor)
 
16
    {
 
17
        this.algorithmIdentifier = decryptor.getAlgorithmIdentifier();
 
18
        this.operator = decryptor;
 
19
    }
 
20
 
 
21
    public RecipientOperator(MacCalculator macCalculator)
 
22
    {
 
23
        this.algorithmIdentifier = macCalculator.getAlgorithmIdentifier();
 
24
        this.operator = macCalculator;
 
25
    }
 
26
 
 
27
    public InputStream getInputStream(InputStream dataIn)
 
28
    {
 
29
        if (operator instanceof InputDecryptor)
 
30
        {
 
31
            return ((InputDecryptor)operator).getInputStream(dataIn);
 
32
        }
 
33
        else
 
34
        {
 
35
            return new TeeInputStream(dataIn, ((MacCalculator)operator).getOutputStream());
 
36
        }
 
37
    }
 
38
 
 
39
    public boolean isMacBased()
 
40
    {
 
41
        return operator instanceof MacCalculator;
 
42
    }
 
43
 
 
44
    public byte[] getMac()
 
45
    {
 
46
        return ((MacCalculator)operator).getMac();
 
47
    }
 
48
}