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

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

import java.util.HashMap;
import org.bouncycastle.crypto.AsymmetricCipherKeyPair;
import java.security.PrivateKey;
import java.security.PublicKey;
import org.bouncycastle.pqc.crypto.falcon.FalconPrivateKeyParameters;
import org.bouncycastle.pqc.crypto.falcon.FalconPublicKeyParameters;
import java.security.KeyPair;
import org.bouncycastle.jcajce.util.SpecUtil;
import org.bouncycastle.pqc.jcajce.spec.FalconParameterSpec;
import org.bouncycastle.crypto.KeyGenerationParameters;
import org.bouncycastle.util.Strings;
import java.security.InvalidAlgorithmParameterException;
import java.security.spec.AlgorithmParameterSpec;
import org.bouncycastle.crypto.CryptoServicesRegistrar;
import java.security.SecureRandom;
import org.bouncycastle.pqc.crypto.falcon.FalconKeyPairGenerator;
import org.bouncycastle.pqc.crypto.falcon.FalconKeyGenerationParameters;
import org.bouncycastle.pqc.crypto.falcon.FalconParameters;
import java.util.Map;
import java.security.KeyPairGenerator;

public class FalconKeyPairGeneratorSpi extends KeyPairGenerator
{
    private static Map parameters;
    private final FalconParameters falconParameters;
    FalconKeyGenerationParameters param;
    FalconKeyPairGenerator engine;
    SecureRandom random;
    boolean initialised;
    
    public FalconKeyPairGeneratorSpi() {
        super("FALCON");
        this.engine = new FalconKeyPairGenerator();
        this.random = CryptoServicesRegistrar.getSecureRandom();
        this.initialised = false;
        this.falconParameters = null;
    }
    
    protected FalconKeyPairGeneratorSpi(final FalconParameters falconParameters) {
        super(falconParameters.getName());
        this.engine = new FalconKeyPairGenerator();
        this.random = CryptoServicesRegistrar.getSecureRandom();
        this.initialised = false;
        this.falconParameters = falconParameters;
    }
    
    @Override
    public void initialize(final int n, final SecureRandom secureRandom) {
        throw new IllegalArgumentException("use AlgorithmParameterSpec");
    }
    
    @Override
    public void initialize(final AlgorithmParameterSpec obj, final SecureRandom secureRandom) throws InvalidAlgorithmParameterException {
        final String nameFromParams = getNameFromParams(obj);
        if (nameFromParams == null || !FalconKeyPairGeneratorSpi.parameters.containsKey(nameFromParams)) {
            throw new InvalidAlgorithmParameterException("invalid ParameterSpec: " + obj);
        }
        final FalconParameters falconParameters = FalconKeyPairGeneratorSpi.parameters.get(nameFromParams);
        this.param = new FalconKeyGenerationParameters(secureRandom, falconParameters);
        if (this.falconParameters != null && !falconParameters.getName().equals(this.falconParameters.getName())) {
            throw new InvalidAlgorithmParameterException("key pair generator locked to " + Strings.toUpperCase(this.falconParameters.getName()));
        }
        this.engine.init(this.param);
        this.initialised = true;
    }
    
    private static String getNameFromParams(final AlgorithmParameterSpec algorithmParameterSpec) {
        if (algorithmParameterSpec instanceof FalconParameterSpec) {
            return ((FalconParameterSpec)algorithmParameterSpec).getName();
        }
        return Strings.toLowerCase(SpecUtil.getNameFrom(algorithmParameterSpec));
    }
    
    @Override
    public KeyPair generateKeyPair() {
        if (!this.initialised) {
            if (this.falconParameters != null) {
                this.param = new FalconKeyGenerationParameters(this.random, this.falconParameters);
            }
            else {
                this.param = new FalconKeyGenerationParameters(this.random, FalconParameters.falcon_512);
            }
            this.engine.init(this.param);
            this.initialised = true;
        }
        final AsymmetricCipherKeyPair generateKeyPair = this.engine.generateKeyPair();
        return new KeyPair(new BCFalconPublicKey((FalconPublicKeyParameters)generateKeyPair.getPublic()), new BCFalconPrivateKey((FalconPrivateKeyParameters)generateKeyPair.getPrivate()));
    }
    
    static {
        (FalconKeyPairGeneratorSpi.parameters = new HashMap()).put(FalconParameterSpec.falcon_512.getName(), FalconParameters.falcon_512);
        FalconKeyPairGeneratorSpi.parameters.put(FalconParameterSpec.falcon_1024.getName(), FalconParameters.falcon_1024);
    }
    
    public static class Falcon1024 extends FalconKeyPairGeneratorSpi
    {
        public Falcon1024() {
            super(FalconParameters.falcon_1024);
        }
    }
    
    public static class Falcon512 extends FalconKeyPairGeneratorSpi
    {
        public Falcon512() {
            super(FalconParameters.falcon_512);
        }
    }
}