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

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

import java.util.HashMap;
import org.bouncycastle.jcajce.util.SpecUtil;
import org.bouncycastle.crypto.AsymmetricCipherKeyPair;
import java.security.PrivateKey;
import java.security.PublicKey;
import org.bouncycastle.pqc.crypto.sphincsplus.SPHINCSPlusPrivateKeyParameters;
import org.bouncycastle.pqc.crypto.sphincsplus.SPHINCSPlusPublicKeyParameters;
import java.security.KeyPair;
import java.security.InvalidAlgorithmParameterException;
import java.security.spec.AlgorithmParameterSpec;
import org.bouncycastle.crypto.KeyGenerationParameters;
import org.bouncycastle.pqc.crypto.sphincsplus.SPHINCSPlusParameters;
import org.bouncycastle.util.Strings;
import org.bouncycastle.pqc.jcajce.spec.SPHINCSPlusParameterSpec;
import org.bouncycastle.crypto.CryptoServicesRegistrar;
import java.security.SecureRandom;
import org.bouncycastle.pqc.crypto.sphincsplus.SPHINCSPlusKeyPairGenerator;
import org.bouncycastle.pqc.crypto.sphincsplus.SPHINCSPlusKeyGenerationParameters;
import java.util.Map;
import java.security.KeyPairGenerator;

public class SPHINCSPlusKeyPairGeneratorSpi extends KeyPairGenerator
{
    private static Map parameters;
    SPHINCSPlusKeyGenerationParameters param;
    SPHINCSPlusKeyPairGenerator engine;
    SecureRandom random;
    boolean initialised;
    
    public SPHINCSPlusKeyPairGeneratorSpi() {
        super("SPHINCS+");
        this.engine = new SPHINCSPlusKeyPairGenerator();
        this.random = CryptoServicesRegistrar.getSecureRandom();
        this.initialised = false;
    }
    
    protected SPHINCSPlusKeyPairGeneratorSpi(final SPHINCSPlusParameterSpec sphincsPlusParameterSpec) {
        super("SPHINCS+-" + Strings.toUpperCase(sphincsPlusParameterSpec.getName()));
        this.engine = new SPHINCSPlusKeyPairGenerator();
        this.random = CryptoServicesRegistrar.getSecureRandom();
        this.initialised = false;
        this.param = new SPHINCSPlusKeyGenerationParameters(this.random, SPHINCSPlusKeyPairGeneratorSpi.parameters.get(sphincsPlusParameterSpec.getName()));
        this.engine.init(this.param);
        this.initialised = true;
    }
    
    @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 SPHINCSPlusKeyGenerationParameters(secureRandom, (SPHINCSPlusParameters)SPHINCSPlusKeyPairGeneratorSpi.parameters.get(nameFromParams));
            this.engine.init(this.param);
            this.initialised = true;
            return;
        }
        throw new InvalidAlgorithmParameterException("invalid ParameterSpec: " + obj);
    }
    
    @Override
    public KeyPair generateKeyPair() {
        if (!this.initialised) {
            this.param = new SPHINCSPlusKeyGenerationParameters(this.random, SPHINCSPlusParameters.sha2_256s);
            this.engine.init(this.param);
            this.initialised = true;
        }
        final AsymmetricCipherKeyPair generateKeyPair = this.engine.generateKeyPair();
        return new KeyPair(new BCSPHINCSPlusPublicKey((SPHINCSPlusPublicKeyParameters)generateKeyPair.getPublic()), new BCSPHINCSPlusPrivateKey((SPHINCSPlusPrivateKeyParameters)generateKeyPair.getPrivate()));
    }
    
    private static String getNameFromParams(final AlgorithmParameterSpec algorithmParameterSpec) {
        if (algorithmParameterSpec instanceof SPHINCSPlusParameterSpec) {
            return ((SPHINCSPlusParameterSpec)algorithmParameterSpec).getName();
        }
        return Strings.toLowerCase(SpecUtil.getNameFrom(algorithmParameterSpec));
    }
    
    static {
        (SPHINCSPlusKeyPairGeneratorSpi.parameters = new HashMap()).put(SPHINCSPlusParameterSpec.sha2_128f_robust.getName(), SPHINCSPlusParameters.sha2_128f_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_128s_robust.getName(), SPHINCSPlusParameters.sha2_128s_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_192f_robust.getName(), SPHINCSPlusParameters.sha2_192f_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_192s_robust.getName(), SPHINCSPlusParameters.sha2_192s_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_256f_robust.getName(), SPHINCSPlusParameters.sha2_256f_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_256s_robust.getName(), SPHINCSPlusParameters.sha2_256s_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_128f.getName(), SPHINCSPlusParameters.sha2_128f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_128s.getName(), SPHINCSPlusParameters.sha2_128s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_192f.getName(), SPHINCSPlusParameters.sha2_192f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_192s.getName(), SPHINCSPlusParameters.sha2_192s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_256f.getName(), SPHINCSPlusParameters.sha2_256f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.sha2_256s.getName(), SPHINCSPlusParameters.sha2_256s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_128f_robust.getName(), SPHINCSPlusParameters.shake_128f_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_128s_robust.getName(), SPHINCSPlusParameters.shake_128s_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_192f_robust.getName(), SPHINCSPlusParameters.shake_192f_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_192s_robust.getName(), SPHINCSPlusParameters.shake_192s_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_256f_robust.getName(), SPHINCSPlusParameters.shake_256f_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_256s_robust.getName(), SPHINCSPlusParameters.shake_256s_robust);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_128f.getName(), SPHINCSPlusParameters.shake_128f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_128s.getName(), SPHINCSPlusParameters.shake_128s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_192f.getName(), SPHINCSPlusParameters.shake_192f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_192s.getName(), SPHINCSPlusParameters.shake_192s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_256f.getName(), SPHINCSPlusParameters.shake_256f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.shake_256s.getName(), SPHINCSPlusParameters.shake_256s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_128f.getName(), SPHINCSPlusParameters.haraka_128f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_128s.getName(), SPHINCSPlusParameters.haraka_128s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_192f.getName(), SPHINCSPlusParameters.haraka_192f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_192s.getName(), SPHINCSPlusParameters.haraka_192s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_256f.getName(), SPHINCSPlusParameters.haraka_256f);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_256s.getName(), SPHINCSPlusParameters.haraka_256s);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_128f_simple.getName(), SPHINCSPlusParameters.haraka_128f_simple);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_128s_simple.getName(), SPHINCSPlusParameters.haraka_128s_simple);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_192f_simple.getName(), SPHINCSPlusParameters.haraka_192f_simple);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_192s_simple.getName(), SPHINCSPlusParameters.haraka_192s_simple);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_256f_simple.getName(), SPHINCSPlusParameters.haraka_256f_simple);
        SPHINCSPlusKeyPairGeneratorSpi.parameters.put(SPHINCSPlusParameterSpec.haraka_256s_simple.getName(), SPHINCSPlusParameters.haraka_256s_simple);
    }
    
    public static class Sha2_128f extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Sha2_128f() {
            super(SPHINCSPlusParameterSpec.sha2_128f);
        }
    }
    
    public static class Sha2_128s extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Sha2_128s() {
            super(SPHINCSPlusParameterSpec.sha2_128s);
        }
    }
    
    public static class Sha2_192f extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Sha2_192f() {
            super(SPHINCSPlusParameterSpec.sha2_192f);
        }
    }
    
    public static class Sha2_192s extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Sha2_192s() {
            super(SPHINCSPlusParameterSpec.sha2_192s);
        }
    }
    
    public static class Sha2_256f extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Sha2_256f() {
            super(SPHINCSPlusParameterSpec.sha2_256f);
        }
    }
    
    public static class Sha2_256s extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Sha2_256s() {
            super(SPHINCSPlusParameterSpec.sha2_256s);
        }
    }
    
    public static class Shake_128f extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Shake_128f() {
            super(SPHINCSPlusParameterSpec.shake_128f);
        }
    }
    
    public static class Shake_128s extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Shake_128s() {
            super(SPHINCSPlusParameterSpec.shake_128s);
        }
    }
    
    public static class Shake_192f extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Shake_192f() {
            super(SPHINCSPlusParameterSpec.shake_192f);
        }
    }
    
    public static class Shake_192s extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Shake_192s() {
            super(SPHINCSPlusParameterSpec.shake_192s);
        }
    }
    
    public static class Shake_256f extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Shake_256f() {
            super(SPHINCSPlusParameterSpec.shake_256f);
        }
    }
    
    public static class Shake_256s extends SPHINCSPlusKeyPairGeneratorSpi
    {
        public Shake_256s() {
            super(SPHINCSPlusParameterSpec.shake_256s);
        }
    }
}