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

package org.bouncycastle.jcajce.provider.asymmetric.edec;

import org.bouncycastle.util.encoders.Hex;
import org.bouncycastle.asn1.ASN1ObjectIdentifier;
import org.bouncycastle.internal.asn1.edec.EdECObjectIdentifiers;
import org.bouncycastle.asn1.pkcs.PrivateKeyInfo;
import org.bouncycastle.crypto.params.Ed448PublicKeyParameters;
import org.bouncycastle.crypto.params.X25519PublicKeyParameters;
import org.bouncycastle.crypto.params.X448PublicKeyParameters;
import org.bouncycastle.asn1.x509.AlgorithmIdentifier;
import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo;
import java.security.spec.X509EncodedKeySpec;
import java.security.PublicKey;
import java.security.PrivateKey;
import org.bouncycastle.jcajce.interfaces.EdDSAPublicKey;
import org.bouncycastle.jcajce.interfaces.XDHPublicKey;
import org.bouncycastle.jcajce.spec.RawEncodedKeySpec;
import org.bouncycastle.crypto.util.OpenSSHPublicKeyUtil;
import org.bouncycastle.crypto.params.Ed25519PublicKeyParameters;
import org.bouncycastle.util.Arrays;
import org.bouncycastle.jcajce.spec.OpenSSHPublicKeySpec;
import java.io.IOException;
import java.security.spec.InvalidKeySpecException;
import org.bouncycastle.crypto.params.AsymmetricKeyParameter;
import org.bouncycastle.crypto.util.OpenSSHPrivateKeyUtil;
import org.bouncycastle.crypto.params.Ed25519PrivateKeyParameters;
import org.bouncycastle.asn1.ASN1Primitive;
import org.bouncycastle.asn1.ASN1OctetString;
import org.bouncycastle.asn1.ASN1Sequence;
import org.bouncycastle.jcajce.spec.OpenSSHPrivateKeySpec;
import java.security.spec.KeySpec;
import java.security.InvalidKeyException;
import java.security.Key;
import org.bouncycastle.jcajce.provider.util.AsymmetricKeyInfoConverter;
import org.bouncycastle.jcajce.provider.asymmetric.util.BaseKeyFactorySpi;

public class KeyFactorySpi extends BaseKeyFactorySpi implements AsymmetricKeyInfoConverter
{
    static final byte[] x448Prefix;
    static final byte[] x25519Prefix;
    static final byte[] Ed448Prefix;
    static final byte[] Ed25519Prefix;
    private static final byte x448_type = 111;
    private static final byte x25519_type = 110;
    private static final byte Ed448_type = 113;
    private static final byte Ed25519_type = 112;
    String algorithm;
    private final boolean isXdh;
    private final int specificBase;
    
    public KeyFactorySpi(final String algorithm, final boolean isXdh, final int specificBase) {
        this.algorithm = algorithm;
        this.isXdh = isXdh;
        this.specificBase = specificBase;
    }
    
    @Override
    protected Key engineTranslateKey(final Key key) throws InvalidKeyException {
        throw new InvalidKeyException("key type unknown");
    }
    
    @Override
    protected KeySpec engineGetKeySpec(final Key key, final Class clazz) throws InvalidKeySpecException {
        if (clazz.isAssignableFrom(OpenSSHPrivateKeySpec.class) && key instanceof BCEdDSAPrivateKey) {
            try {
                return new OpenSSHPrivateKeySpec(OpenSSHPrivateKeyUtil.encodePrivateKey(new Ed25519PrivateKeyParameters(ASN1OctetString.getInstance(ASN1Primitive.fromByteArray(ASN1OctetString.getInstance(ASN1Sequence.getInstance(key.getEncoded()).getObjectAt(2)).getOctets())).getOctets())));
            }
            catch (final IOException ex) {
                throw new InvalidKeySpecException(ex.getMessage(), ex.getCause());
            }
        }
        if (clazz.isAssignableFrom(OpenSSHPublicKeySpec.class) && key instanceof BCEdDSAPublicKey) {
            try {
                final byte[] encoded = key.getEncoded();
                if (!Arrays.areEqual(KeyFactorySpi.Ed25519Prefix, 0, KeyFactorySpi.Ed25519Prefix.length, encoded, 0, encoded.length - 32)) {
                    throw new InvalidKeySpecException("Invalid Ed25519 public key encoding");
                }
                return new OpenSSHPublicKeySpec(OpenSSHPublicKeyUtil.encodePublicKey(new Ed25519PublicKeyParameters(encoded, KeyFactorySpi.Ed25519Prefix.length)));
            }
            catch (final IOException ex2) {
                throw new InvalidKeySpecException(ex2.getMessage(), ex2.getCause());
            }
        }
        if (clazz.isAssignableFrom(RawEncodedKeySpec.class)) {
            if (key instanceof XDHPublicKey) {
                return new RawEncodedKeySpec(((XDHPublicKey)key).getUEncoding());
            }
            if (key instanceof EdDSAPublicKey) {
                return new RawEncodedKeySpec(((EdDSAPublicKey)key).getPointEncoding());
            }
        }
        return super.engineGetKeySpec(key, clazz);
    }
    
    @Override
    protected PrivateKey engineGeneratePrivate(final KeySpec keySpec) throws InvalidKeySpecException {
        if (!(keySpec instanceof OpenSSHPrivateKeySpec)) {
            return super.engineGeneratePrivate(keySpec);
        }
        final AsymmetricKeyParameter privateKeyBlob = OpenSSHPrivateKeyUtil.parsePrivateKeyBlob(((OpenSSHPrivateKeySpec)keySpec).getEncoded());
        if (privateKeyBlob instanceof Ed25519PrivateKeyParameters) {
            return new BCEdDSAPrivateKey(privateKeyBlob);
        }
        throw new IllegalStateException("openssh private key not Ed25519 private key");
    }
    
    @Override
    protected PublicKey engineGeneratePublic(final KeySpec keySpec) throws InvalidKeySpecException {
        if (keySpec instanceof X509EncodedKeySpec) {
            byte[] array = ((X509EncodedKeySpec)keySpec).getEncoded();
            if (this.specificBase == 0 || this.specificBase == array[8]) {
                if (array[9] == 5 && array[10] == 0) {
                    final SubjectPublicKeyInfo instance = SubjectPublicKeyInfo.getInstance(array);
                    final SubjectPublicKeyInfo subjectPublicKeyInfo = new SubjectPublicKeyInfo(new AlgorithmIdentifier(instance.getAlgorithm().getAlgorithm()), instance.getPublicKeyData().getBytes());
                    try {
                        array = subjectPublicKeyInfo.getEncoded("DER");
                    }
                    catch (final IOException ex) {
                        throw new InvalidKeySpecException("attempt to reconstruct key failed: " + ex.getMessage());
                    }
                }
                switch (array[8]) {
                    case 111: {
                        return new BCXDHPublicKey(KeyFactorySpi.x448Prefix, array);
                    }
                    case 110: {
                        return new BCXDHPublicKey(KeyFactorySpi.x25519Prefix, array);
                    }
                    case 113: {
                        return new BCEdDSAPublicKey(KeyFactorySpi.Ed448Prefix, array);
                    }
                    case 112: {
                        return new BCEdDSAPublicKey(KeyFactorySpi.Ed25519Prefix, array);
                    }
                    default: {
                        return super.engineGeneratePublic(keySpec);
                    }
                }
            }
        }
        else if (keySpec instanceof RawEncodedKeySpec) {
            final byte[] encoded = ((RawEncodedKeySpec)keySpec).getEncoded();
            switch (this.specificBase) {
                case 111: {
                    return new BCXDHPublicKey(new X448PublicKeyParameters(encoded));
                }
                case 110: {
                    return new BCXDHPublicKey(new X25519PublicKeyParameters(encoded));
                }
                case 113: {
                    return new BCEdDSAPublicKey(new Ed448PublicKeyParameters(encoded));
                }
                case 112: {
                    return new BCEdDSAPublicKey(new Ed25519PublicKeyParameters(encoded));
                }
                default: {
                    throw new InvalidKeySpecException("factory not a specific type, cannot recognise raw encoding");
                }
            }
        }
        else if (keySpec instanceof OpenSSHPublicKeySpec) {
            final AsymmetricKeyParameter publicKey = OpenSSHPublicKeyUtil.parsePublicKey(((OpenSSHPublicKeySpec)keySpec).getEncoded());
            if (publicKey instanceof Ed25519PublicKeyParameters) {
                return new BCEdDSAPublicKey(new byte[0], ((Ed25519PublicKeyParameters)publicKey).getEncoded());
            }
            throw new IllegalStateException("openssh public key not Ed25519 public key");
        }
        return super.engineGeneratePublic(keySpec);
    }
    
    @Override
    public PrivateKey generatePrivate(final PrivateKeyInfo privateKeyInfo) throws IOException {
        final ASN1ObjectIdentifier algorithm = privateKeyInfo.getPrivateKeyAlgorithm().getAlgorithm();
        if (this.isXdh) {
            if ((this.specificBase == 0 || this.specificBase == 111) && algorithm.equals(EdECObjectIdentifiers.id_X448)) {
                return new BCXDHPrivateKey(privateKeyInfo);
            }
            if ((this.specificBase == 0 || this.specificBase == 110) && algorithm.equals(EdECObjectIdentifiers.id_X25519)) {
                return new BCXDHPrivateKey(privateKeyInfo);
            }
        }
        else if (algorithm.equals(EdECObjectIdentifiers.id_Ed448) || algorithm.equals(EdECObjectIdentifiers.id_Ed25519)) {
            if ((this.specificBase == 0 || this.specificBase == 113) && algorithm.equals(EdECObjectIdentifiers.id_Ed448)) {
                return new BCEdDSAPrivateKey(privateKeyInfo);
            }
            if ((this.specificBase == 0 || this.specificBase == 112) && algorithm.equals(EdECObjectIdentifiers.id_Ed25519)) {
                return new BCEdDSAPrivateKey(privateKeyInfo);
            }
        }
        throw new IOException("algorithm identifier " + algorithm + " in key not recognized");
    }
    
    @Override
    public PublicKey generatePublic(final SubjectPublicKeyInfo subjectPublicKeyInfo) throws IOException {
        final ASN1ObjectIdentifier algorithm = subjectPublicKeyInfo.getAlgorithm().getAlgorithm();
        if (this.isXdh) {
            if ((this.specificBase == 0 || this.specificBase == 111) && algorithm.equals(EdECObjectIdentifiers.id_X448)) {
                return new BCXDHPublicKey(subjectPublicKeyInfo);
            }
            if ((this.specificBase == 0 || this.specificBase == 110) && algorithm.equals(EdECObjectIdentifiers.id_X25519)) {
                return new BCXDHPublicKey(subjectPublicKeyInfo);
            }
        }
        else if (algorithm.equals(EdECObjectIdentifiers.id_Ed448) || algorithm.equals(EdECObjectIdentifiers.id_Ed25519)) {
            if ((this.specificBase == 0 || this.specificBase == 113) && algorithm.equals(EdECObjectIdentifiers.id_Ed448)) {
                return new BCEdDSAPublicKey(subjectPublicKeyInfo);
            }
            if ((this.specificBase == 0 || this.specificBase == 112) && algorithm.equals(EdECObjectIdentifiers.id_Ed25519)) {
                return new BCEdDSAPublicKey(subjectPublicKeyInfo);
            }
        }
        throw new IOException("algorithm identifier " + algorithm + " in key not recognized");
    }
    
    static {
        x448Prefix = Hex.decode("3042300506032b656f033900");
        x25519Prefix = Hex.decode("302a300506032b656e032100");
        Ed448Prefix = Hex.decode("3043300506032b6571033a00");
        Ed25519Prefix = Hex.decode("302a300506032b6570032100");
    }
    
    public static class Ed25519 extends KeyFactorySpi
    {
        public Ed25519() {
            super("Ed25519", false, 112);
        }
    }
    
    public static class Ed448 extends KeyFactorySpi
    {
        public Ed448() {
            super("Ed448", false, 113);
        }
    }
    
    public static class EdDSA extends KeyFactorySpi
    {
        public EdDSA() {
            super("EdDSA", false, 0);
        }
    }
    
    public static class X25519 extends KeyFactorySpi
    {
        public X25519() {
            super("X25519", true, 110);
        }
    }
    
    public static class X448 extends KeyFactorySpi
    {
        public X448() {
            super("X448", true, 111);
        }
    }
    
    public static class XDH extends KeyFactorySpi
    {
        public XDH() {
            super("XDH", true, 0);
        }
    }
}