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

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

import java.util.HashMap;
import org.bouncycastle.crypto.AsymmetricCipherKeyPair;
import java.security.PrivateKey;
import java.security.PublicKey;
import org.bouncycastle.pqc.crypto.mlkem.MLKEMPrivateKeyParameters;
import org.bouncycastle.pqc.crypto.mlkem.MLKEMPublicKeyParameters;
import java.security.KeyPair;
import org.bouncycastle.jcajce.util.SpecUtil;
import org.bouncycastle.pqc.jcajce.spec.KyberParameterSpec;
import org.bouncycastle.crypto.KeyGenerationParameters;
import java.security.InvalidAlgorithmParameterException;
import java.security.spec.AlgorithmParameterSpec;
import org.bouncycastle.util.Strings;
import org.bouncycastle.crypto.CryptoServicesRegistrar;
import org.bouncycastle.pqc.crypto.mlkem.MLKEMParameters;
import java.security.SecureRandom;
import org.bouncycastle.pqc.crypto.mlkem.MLKEMKeyPairGenerator;
import org.bouncycastle.pqc.crypto.mlkem.MLKEMKeyGenerationParameters;
import java.util.Map;
import java.security.KeyPairGenerator;

public class KyberKeyPairGeneratorSpi extends KeyPairGenerator
{
    private static Map parameters;
    MLKEMKeyGenerationParameters param;
    MLKEMKeyPairGenerator engine;
    SecureRandom random;
    boolean initialised;
    private MLKEMParameters kyberParameters;
    
    public KyberKeyPairGeneratorSpi() {
        super("KYBER");
        this.engine = new MLKEMKeyPairGenerator();
        this.random = CryptoServicesRegistrar.getSecureRandom();
        this.initialised = false;
        this.kyberParameters = null;
    }
    
    protected KyberKeyPairGeneratorSpi(final MLKEMParameters kyberParameters) {
        super(Strings.toUpperCase(kyberParameters.getName()));
        this.engine = new MLKEMKeyPairGenerator();
        this.random = CryptoServicesRegistrar.getSecureRandom();
        this.initialised = false;
        this.kyberParameters = kyberParameters;
    }
    
    @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 || !KyberKeyPairGeneratorSpi.parameters.containsKey(nameFromParams)) {
            throw new InvalidAlgorithmParameterException("invalid ParameterSpec: " + obj);
        }
        final MLKEMParameters mlkemParameters = KyberKeyPairGeneratorSpi.parameters.get(nameFromParams);
        this.param = new MLKEMKeyGenerationParameters(secureRandom, mlkemParameters);
        if (this.kyberParameters != null && !mlkemParameters.getName().equals(this.kyberParameters.getName())) {
            throw new InvalidAlgorithmParameterException("key pair generator locked to " + Strings.toUpperCase(this.kyberParameters.getName()));
        }
        this.engine.init(this.param);
        this.initialised = true;
    }
    
    private static String getNameFromParams(final AlgorithmParameterSpec algorithmParameterSpec) {
        if (algorithmParameterSpec instanceof KyberParameterSpec) {
            return ((KyberParameterSpec)algorithmParameterSpec).getName();
        }
        return Strings.toLowerCase(SpecUtil.getNameFrom(algorithmParameterSpec));
    }
    
    @Override
    public KeyPair generateKeyPair() {
        if (!this.initialised) {
            if (this.kyberParameters != null) {
                this.param = new MLKEMKeyGenerationParameters(this.random, this.kyberParameters);
            }
            else {
                this.param = new MLKEMKeyGenerationParameters(this.random, MLKEMParameters.ml_kem_1024);
            }
            this.engine.init(this.param);
            this.initialised = true;
        }
        final AsymmetricCipherKeyPair generateKeyPair = this.engine.generateKeyPair();
        return new KeyPair(new BCKyberPublicKey((MLKEMPublicKeyParameters)generateKeyPair.getPublic()), new BCKyberPrivateKey((MLKEMPrivateKeyParameters)generateKeyPair.getPrivate()));
    }
    
    static {
        (KyberKeyPairGeneratorSpi.parameters = new HashMap()).put(KyberParameterSpec.kyber512.getName(), MLKEMParameters.ml_kem_512);
        KyberKeyPairGeneratorSpi.parameters.put(KyberParameterSpec.kyber768.getName(), MLKEMParameters.ml_kem_768);
        KyberKeyPairGeneratorSpi.parameters.put(KyberParameterSpec.kyber1024.getName(), MLKEMParameters.ml_kem_1024);
    }
    
    public static class Kyber1024 extends KyberKeyPairGeneratorSpi
    {
        public Kyber1024() {
            super(MLKEMParameters.ml_kem_1024);
        }
    }
    
    public static class Kyber512 extends KyberKeyPairGeneratorSpi
    {
        public Kyber512() {
            super(MLKEMParameters.ml_kem_512);
        }
    }
    
    public static class Kyber768 extends KyberKeyPairGeneratorSpi
    {
        public Kyber768() {
            super(MLKEMParameters.ml_kem_768);
        }
    }
}