core/src/main/java/org/bouncycastle/crypto/digests/ShortenedDigest.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

package org.bouncycastle.crypto.digests;

import org.bouncycastle.crypto.ExtendedDigest;

/**
 * Wrapper class that reduces the output length of a particular digest to
 * only the first n bytes of the digest function.
 */
public class ShortenedDigest 
    implements ExtendedDigest
{
    private ExtendedDigest baseDigest;
    private int            length;
    
    /**
     * Base constructor.
     * 
     * @param baseDigest underlying digest to use.
     * @param length length in bytes of the output of doFinal.
     * @exception IllegalArgumentException if baseDigest is null, or length is greater than baseDigest.getDigestSize().
     */
    public ShortenedDigest(
        ExtendedDigest baseDigest,
        int            length)
    {
        if (baseDigest == null)
        {
            throw new IllegalArgumentException("baseDigest must not be null");
        }
        
        if (length > baseDigest.getDigestSize())
        {
            throw new IllegalArgumentException("baseDigest output not large enough to support length");
        }
        
        this.baseDigest = baseDigest;
        this.length = length;
    }
    
    public String getAlgorithmName()
    {
        return baseDigest.getAlgorithmName() + "(" + length * 8 + ")";
    }

    public int getDigestSize()
    {
        return length;
    }

    public void update(byte in)
    {
        baseDigest.update(in);
    }

    public void update(byte[] in, int inOff, int len)
    {
        baseDigest.update(in, inOff, len);
    }

    public int doFinal(byte[] out, int outOff)
    {
        byte[] tmp = new byte[baseDigest.getDigestSize()];
        
        baseDigest.doFinal(tmp, 0);
        
        System.arraycopy(tmp, 0, out, outOff, length);
        
        return length;
    }

    public void reset()
    {
        baseDigest.reset();
    }

    public int getByteLength()
    {
        return baseDigest.getByteLength();
    }
}