// 
// Decompiled by Procyon v0.6.0
// 

package com.nimbusds.jose.crypto.impl;

import com.nimbusds.jose.KeyLengthException;
import javax.crypto.spec.SecretKeySpec;
import javax.crypto.SecretKey;
import com.nimbusds.jose.shaded.jcip.Immutable;

@Immutable
public final class CompositeKey
{
    private final SecretKey inputKey;
    private final SecretKey macKey;
    private final SecretKey encKey;
    private final int truncatedMacLength;
    
    public CompositeKey(final SecretKey inputKey) throws KeyLengthException {
        this.inputKey = inputKey;
        final byte[] secretKeyBytes = inputKey.getEncoded();
        if (secretKeyBytes.length == 32) {
            this.macKey = new SecretKeySpec(secretKeyBytes, 0, 16, "HMACSHA256");
            this.encKey = new SecretKeySpec(secretKeyBytes, 16, 16, "AES");
            this.truncatedMacLength = 16;
        }
        else if (secretKeyBytes.length == 48) {
            this.macKey = new SecretKeySpec(secretKeyBytes, 0, 24, "HMACSHA384");
            this.encKey = new SecretKeySpec(secretKeyBytes, 24, 24, "AES");
            this.truncatedMacLength = 24;
        }
        else {
            if (secretKeyBytes.length != 64) {
                throw new KeyLengthException("Unsupported AES/CBC/PKCS5Padding/HMAC-SHA2 key length, must be 256, 384 or 512 bits");
            }
            this.macKey = new SecretKeySpec(secretKeyBytes, 0, 32, "HMACSHA512");
            this.encKey = new SecretKeySpec(secretKeyBytes, 32, 32, "AES");
            this.truncatedMacLength = 32;
        }
    }
    
    public SecretKey getInputKey() {
        return this.inputKey;
    }
    
    public SecretKey getMACKey() {
        return this.macKey;
    }
    
    public int getTruncatedMACByteLength() {
        return this.truncatedMacLength;
    }
    
    public SecretKey getAESKey() {
        return this.encKey;
    }
}