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

package org.bouncycastle.pqc.jcajce.provider.mayo;

import java.security.spec.AlgorithmParameterSpec;
import java.security.SignatureException;
import org.bouncycastle.pqc.crypto.mayo.MayoPrivateKeyParameters;
import org.bouncycastle.crypto.params.ParametersWithRandom;
import java.security.PrivateKey;
import org.bouncycastle.crypto.CipherParameters;
import java.security.InvalidKeyException;
import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo;
import java.security.PublicKey;
import org.bouncycastle.util.Strings;
import org.bouncycastle.pqc.crypto.mayo.MayoParameters;
import java.security.SecureRandom;
import org.bouncycastle.pqc.crypto.mayo.MayoSigner;
import java.io.ByteArrayOutputStream;
import java.security.Signature;

public class SignatureSpi extends Signature
{
    private final ByteArrayOutputStream bOut;
    private final MayoSigner signer;
    private SecureRandom random;
    private final MayoParameters parameters;
    
    protected SignatureSpi(final MayoSigner signer) {
        super("Mayo");
        this.bOut = new ByteArrayOutputStream();
        this.signer = signer;
        this.parameters = null;
    }
    
    protected SignatureSpi(final MayoSigner signer, final MayoParameters parameters) {
        super(Strings.toUpperCase(parameters.getName()));
        this.parameters = parameters;
        this.bOut = new ByteArrayOutputStream();
        this.signer = signer;
    }
    
    @Override
    protected void engineInitVerify(PublicKey publicKey) throws InvalidKeyException {
        if (!(publicKey instanceof BCMayoPublicKey)) {
            try {
                publicKey = new BCMayoPublicKey(SubjectPublicKeyInfo.getInstance(publicKey.getEncoded()));
            }
            catch (final Exception ex) {
                throw new InvalidKeyException("unknown public key passed to Mayo: " + ex.getMessage());
            }
        }
        final BCMayoPublicKey bcMayoPublicKey = (BCMayoPublicKey)publicKey;
        if (this.parameters != null) {
            final String upperCase = Strings.toUpperCase(this.parameters.getName());
            if (!upperCase.equals(bcMayoPublicKey.getAlgorithm())) {
                throw new InvalidKeyException("signature configured for " + upperCase);
            }
        }
        this.signer.init(false, bcMayoPublicKey.getKeyParams());
    }
    
    @Override
    protected void engineInitSign(final PrivateKey privateKey, final SecureRandom random) throws InvalidKeyException {
        this.random = random;
        this.engineInitSign(privateKey);
    }
    
    @Override
    protected void engineInitSign(final PrivateKey privateKey) throws InvalidKeyException {
        if (privateKey instanceof BCMayoPrivateKey) {
            final BCMayoPrivateKey bcMayoPrivateKey = (BCMayoPrivateKey)privateKey;
            final MayoPrivateKeyParameters keyParams = bcMayoPrivateKey.getKeyParams();
            if (this.parameters != null) {
                final String upperCase = Strings.toUpperCase(this.parameters.getName());
                if (!upperCase.equals(bcMayoPrivateKey.getAlgorithm())) {
                    throw new InvalidKeyException("signature configured for " + upperCase);
                }
            }
            if (this.random != null) {
                this.signer.init(true, new ParametersWithRandom(keyParams, this.random));
            }
            else {
                this.signer.init(true, keyParams);
            }
            return;
        }
        throw new InvalidKeyException("unknown private key passed to Mayo");
    }
    
    @Override
    protected void engineUpdate(final byte b) throws SignatureException {
        this.bOut.write(b);
    }
    
    @Override
    protected void engineUpdate(final byte[] b, final int off, final int len) throws SignatureException {
        this.bOut.write(b, off, len);
    }
    
    @Override
    protected byte[] engineSign() throws SignatureException {
        try {
            final byte[] byteArray = this.bOut.toByteArray();
            this.bOut.reset();
            return this.signer.generateSignature(byteArray);
        }
        catch (final Exception ex) {
            throw new SignatureException(ex.toString());
        }
    }
    
    @Override
    protected boolean engineVerify(final byte[] array) throws SignatureException {
        final byte[] byteArray = this.bOut.toByteArray();
        this.bOut.reset();
        return this.signer.verifySignature(byteArray, array);
    }
    
    @Override
    protected void engineSetParameter(final AlgorithmParameterSpec algorithmParameterSpec) {
        throw new UnsupportedOperationException("engineSetParameter unsupported");
    }
    
    @Override
    @Deprecated
    protected void engineSetParameter(final String s, final Object o) {
        throw new UnsupportedOperationException("engineSetParameter unsupported");
    }
    
    @Override
    @Deprecated
    protected Object engineGetParameter(final String s) {
        throw new UnsupportedOperationException("engineSetParameter unsupported");
    }
    
    public static class Base extends SignatureSpi
    {
        public Base() {
            super(new MayoSigner());
        }
    }
    
    public static class Mayo1 extends SignatureSpi
    {
        public Mayo1() {
            super(new MayoSigner(), MayoParameters.mayo1);
        }
    }
    
    public static class Mayo2 extends SignatureSpi
    {
        public Mayo2() {
            super(new MayoSigner(), MayoParameters.mayo2);
        }
    }
    
    public static class Mayo3 extends SignatureSpi
    {
        public Mayo3() {
            super(new MayoSigner(), MayoParameters.mayo3);
        }
    }
    
    public static class Mayo5 extends SignatureSpi
    {
        public Mayo5() {
            super(new MayoSigner(), MayoParameters.mayo5);
        }
    }
}