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

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

import java.util.HashMap;
import org.bouncycastle.crypto.AsymmetricCipherKeyPair;
import java.security.PrivateKey;
import java.security.PublicKey;
import org.bouncycastle.pqc.crypto.saber.SABERPrivateKeyParameters;
import org.bouncycastle.pqc.crypto.saber.SABERPublicKeyParameters;
import java.security.KeyPair;
import org.bouncycastle.util.Strings;
import org.bouncycastle.jcajce.util.SpecUtil;
import org.bouncycastle.pqc.jcajce.spec.SABERParameterSpec;
import java.security.InvalidAlgorithmParameterException;
import org.bouncycastle.crypto.KeyGenerationParameters;
import org.bouncycastle.pqc.crypto.saber.SABERParameters;
import java.security.spec.AlgorithmParameterSpec;
import org.bouncycastle.crypto.CryptoServicesRegistrar;
import java.security.SecureRandom;
import org.bouncycastle.pqc.crypto.saber.SABERKeyPairGenerator;
import org.bouncycastle.pqc.crypto.saber.SABERKeyGenerationParameters;
import java.util.Map;
import java.security.KeyPairGenerator;

public class SABERKeyPairGeneratorSpi extends KeyPairGenerator
{
    private static Map parameters;
    SABERKeyGenerationParameters param;
    SABERKeyPairGenerator engine;
    SecureRandom random;
    boolean initialised;
    
    public SABERKeyPairGeneratorSpi() {
        super("SABER");
        this.engine = new SABERKeyPairGenerator();
        this.random = CryptoServicesRegistrar.getSecureRandom();
        this.initialised = false;
    }
    
    @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) {
            this.param = new SABERKeyGenerationParameters(secureRandom, (SABERParameters)SABERKeyPairGeneratorSpi.parameters.get(nameFromParams));
            this.engine.init(this.param);
            this.initialised = true;
            return;
        }
        throw new InvalidAlgorithmParameterException("invalid ParameterSpec: " + obj);
    }
    
    private static String getNameFromParams(final AlgorithmParameterSpec algorithmParameterSpec) {
        if (algorithmParameterSpec instanceof SABERParameterSpec) {
            return ((SABERParameterSpec)algorithmParameterSpec).getName();
        }
        return Strings.toLowerCase(SpecUtil.getNameFrom(algorithmParameterSpec));
    }
    
    @Override
    public KeyPair generateKeyPair() {
        if (!this.initialised) {
            this.param = new SABERKeyGenerationParameters(this.random, SABERParameters.firesaberkem256r3);
            this.engine.init(this.param);
            this.initialised = true;
        }
        final AsymmetricCipherKeyPair generateKeyPair = this.engine.generateKeyPair();
        return new KeyPair(new BCSABERPublicKey((SABERPublicKeyParameters)generateKeyPair.getPublic()), new BCSABERPrivateKey((SABERPrivateKeyParameters)generateKeyPair.getPrivate()));
    }
    
    static {
        (SABERKeyPairGeneratorSpi.parameters = new HashMap()).put(SABERParameterSpec.lightsaberkem128r3.getName(), SABERParameters.lightsaberkem128r3);
        SABERKeyPairGeneratorSpi.parameters.put(SABERParameterSpec.saberkem128r3.getName(), SABERParameters.saberkem128r3);
        SABERKeyPairGeneratorSpi.parameters.put(SABERParameterSpec.firesaberkem128r3.getName(), SABERParameters.firesaberkem128r3);
        SABERKeyPairGeneratorSpi.parameters.put(SABERParameterSpec.lightsaberkem192r3.getName(), SABERParameters.lightsaberkem192r3);
        SABERKeyPairGeneratorSpi.parameters.put(SABERParameterSpec.saberkem192r3.getName(), SABERParameters.saberkem192r3);
        SABERKeyPairGeneratorSpi.parameters.put(SABERParameterSpec.firesaberkem192r3.getName(), SABERParameters.firesaberkem192r3);
        SABERKeyPairGeneratorSpi.parameters.put(SABERParameterSpec.lightsaberkem256r3.getName(), SABERParameters.lightsaberkem256r3);
        SABERKeyPairGeneratorSpi.parameters.put(SABERParameterSpec.saberkem256r3.getName(), SABERParameters.saberkem256r3);
        SABERKeyPairGeneratorSpi.parameters.put(SABERParameterSpec.firesaberkem256r3.getName(), SABERParameters.firesaberkem256r3);
    }
}