All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.wildfly.security.ssl.MechanismDatabase Maven / Gradle / Ivy

There is a newer version: 2.6.0.Final
Show newest version
/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2014 Red Hat, Inc., and individual contributors
 * as indicated by the @author tags.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.wildfly.security.ssl;

import static org.wildfly.security.ssl.ElytronMessages.log;

import java.io.IOError;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.EnumMap;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Pattern;

class MechanismDatabase {
    private static final MechanismDatabase INSTANCE = new MechanismDatabase("MechanismDatabase.properties");
    private static final MechanismDatabase TLS13_INSTANCE = new MechanismDatabase("TLS13MechanismDatabase.properties", true);

    private final Map entriesByStdName;
    private final Map entriesByOSSLName;
    private final Entry[][] algorithmsById;
    private final boolean isTLS13;

    static MechanismDatabase getInstance() {
        return INSTANCE;
    }

    static MechanismDatabase getTLS13Instance() {
        return TLS13_INSTANCE;
    }

    MechanismDatabase(String databaseFileName) {
        this(databaseFileName, false);
    }

    MechanismDatabase(String databaseFileName, boolean isTLS13) {
        this.isTLS13 = isTLS13;
        // load and initialize database properties
        final LinkedProperties properties = new LinkedProperties();

        try (InputStream stream = getClass().getResourceAsStream(databaseFileName)) {
            try (InputStreamReader reader = new InputStreamReader(stream, StandardCharsets.UTF_8)) {
                properties.load(reader);
            }
        } catch (IOException e) {
            throw new IOError(e);
        }

        Pattern p = Pattern.compile("\\s*,\\s*");
        final Map entriesByStdName = new LinkedHashMap<>();
        final Map entriesByOSSLName = new LinkedHashMap<>();
        final Map> entriesByKeyExchange = new EnumMap>(KeyAgreement.class);
        final Map> entriesByAuthentication = new EnumMap>(Authentication.class);
        final Map> entriesByEncryption = new EnumMap>(Encryption.class);
        final Map> entriesByDigest = new EnumMap>(Digest.class);
        final Map> entriesByProtocol = new EnumMap>(Protocol.class);
        final Map> entriesByLevel = new EnumMap>(SecurityLevel.class);

        final Map aliases = new HashMap<>();
        final Entry[][] algorithms = new Entry[256][];

        for (Map.Entry mapEntry : properties.stringMapEntries()) {
            final String name = mapEntry.getKey();
            final String rawValue = mapEntry.getValue();
            final String[] strings = p.split(rawValue);
            if (strings.length == 1 && strings[0].startsWith("alias:")) {
                String target = strings[0].substring(6);
                aliases.put(name, target);
                if (name.startsWith("TLS_")) {
                    aliases.put("SSL_" + name.substring(4), target);
                }
            } else if (strings.length != 11 && strings.length != 13) {
                log.warnInvalidStringCountForMechanismDatabaseEntry(name);
            } else {
                boolean ok = true;
                final String[] openSslNames = strings[0].split("\\|");
                final KeyAgreement key = KeyAgreement.forName(strings[1]);
                if (! isTLS13 && key == null) {
                    log.warnInvalidKeyExchangeForMechanismDatabaseEntry(strings[1], name);
                    ok = false;
                }
                final Authentication auth = Authentication.forName(strings[2]);
                if ((! isTLS13 && auth == null) || (isTLS13 && ! strings[2].equals("ANY"))) {
                    log.warnInvalidAuthenticationForMechanismDatabaseEntry(strings[2], name);
                    ok = false;
                }
                final Encryption enc = Encryption.forName(strings[3]);
                if (enc == null) {
                    log.warnInvalidEncryptionForMechanismDatabaseEntry(strings[3], name);
                    ok = false;
                }
                final Digest digest = Digest.forName(strings[4]);
                if (digest == null) {
                    log.warnInvalidDigestForMechanismDatabaseEntry(strings[4], name);
                    ok = false;
                }
                final Protocol prot = Protocol.forName(strings[5]);
                if ((prot == null) || (isTLS13 ^ prot.equals(Protocol.TLSv1_3))) {
                    log.warnInvalidProtocolForMechanismDatabaseEntry(strings[5], name);
                    ok = false;
                }
                final boolean export = Boolean.parseBoolean(strings[6]);
                final SecurityLevel level = SecurityLevel.forName(strings[7]);
                if (level == null) {
                    log.warnInvalidLevelForMechanismDatabaseEntry(strings[7], name);
                    ok = false;
                }
                final boolean fips = Boolean.parseBoolean(strings[8]);
                int strBits;
                try {
                    strBits = Integer.parseInt(strings[9], 10);
                } catch (NumberFormatException ignored) {
                    log.warnInvalidStrengthBitsForMechanismDatabaseEntry(strings[9], name);
                    strBits = 0;
                    ok = false;
                }
                int algBits;
                try {
                    algBits = Integer.parseInt(strings[10], 10);
                } catch (NumberFormatException ignored) {
                    log.warnInvalidAlgorithmBitsForMechanismDatabaseEntry(strings[10], name);
                    algBits = 0;
                    ok = false;
                }
                int byte1, byte2;
                try {
                    byte1 = Integer.parseInt(strings[11], 16);
                    byte2 = Integer.parseInt(strings[12], 16);
                } catch (ArrayIndexOutOfBoundsException | NumberFormatException ignored) {
                    // ignore the entry
                    byte1 = -1;
                    byte2 = -1;
                }
                if (ok) {
                    final Entry entry = new Entry(name, Arrays.asList(openSslNames), new ArrayList(0), key, auth, enc, digest, prot, export, level, fips, strBits, algBits);
                    if (entriesByStdName.containsKey(name)) {
                        log.warnInvalidDuplicateMechanismDatabaseEntry(name);
                    } else {
                        entriesByStdName.put(name, entry);
                    }
                    for (String openSslName : openSslNames) {
                        if (entriesByOSSLName.containsKey(openSslName)) {
                            log.warnInvalidDuplicateOpenSslStyleAliasForMechanismDatabaseEntry(openSslName, name, entriesByOSSLName.get(openSslName).getName());
                        } else {
                            entriesByOSSLName.put(openSslName, entry);
                        }
                        if (key == KeyAgreement.DHE) {
                            final String newKey = join("-", replaceEdh(openSslName.split("-")));
                            if (!entriesByOSSLName.containsKey(newKey)) {
                                entriesByOSSLName.put(newKey, entry);
                            }
                        }
                    }

                    if (! isTLS13) {
                        addTo(entriesByKeyExchange, key, entry);
                        addTo(entriesByAuthentication, auth, entry);
                    }
                    addTo(entriesByEncryption, enc, entry);
                    addTo(entriesByDigest, digest, entry);
                    addTo(entriesByProtocol, prot, entry);
                    addTo(entriesByLevel, level, entry);
                    if (byte1 != -1 && byte2 != -1) {
                        Entry[] row = algorithms[byte1];
                        if (row == null) {
                            row = algorithms[byte1] = new Entry[256];
                        }
                        row[byte2] = entry;
                    }
                    if (name.startsWith("TLS_")) {
                        aliases.put("SSL_" + name.substring(4), name);
                    }
                }
            }
        }

        for (Map.Entry entry : aliases.entrySet()) {
            final String name = entry.getKey();
            String value = entry.getValue();

            while (aliases.containsKey(value)) {
                // Just skip to the end, intermediate aliases will get their own turn in the for loop.
                value = aliases.get(value);
            }

            if (entriesByStdName.containsKey(name)) {
                log.warnInvalidDuplicateMechanismDatabaseEntry(name);
            } else if (! entriesByStdName.containsKey(value)) {
                log.warnInvalidAliasForMissingMechanismDatabaseEntry(value, name);
            } else {
                final Entry dbEntry = entriesByStdName.get(value);
                dbEntry.getAliases().add(name);
                entriesByStdName.put(name, dbEntry);
            }
        }

        this.entriesByStdName = entriesByStdName;
        this.entriesByOSSLName = entriesByOSSLName;
        this.algorithmsById = algorithms;
    }

    static String[] replaceEdh(String... strings) {
        for (int i = 0, stringsLength = strings.length; i < stringsLength; i++) {
            if ("EDH".equals(strings[i])) {
                strings[i] = "DHE";
            } else if ("DHE".equals(strings[i])) {
                strings[i] = "EDH";
            }
        }
        return strings;
    }

    static String join(String joiner, String... strings) {
        StringBuilder b = new StringBuilder();
        final int length = strings.length;
        int i = 0;
        if (length == 0) return "";
        for (;;) {
            b.append(strings[i ++]);
            if (i == length) {
                return b.toString();
            }
            b.append(joiner);
        }
    }

    static  void addTo(Map> map, T item, Entry entry) {
        List list = map.get(item);
        if (list == null) {
            list = new ArrayList<>(1);
            map.put(item, list);
        }
        list.add(entry);
    }

    Entry getCipherSuite(final String cipherSuite) {
        return entriesByStdName.get(cipherSuite);
    }

    Entry getCipherSuiteOpenSSLName(final String cipherSuite) {
        return entriesByOSSLName.get(cipherSuite);
    }

    Entry getCipherSuiteById(final int byte1, final int byte2) {
        if (byte1 < 0 || byte1 > 255 || byte2 < 0 || byte2 > 255) {
            return null;
        }
        final Entry[] row = algorithmsById[byte1];
        if (row == null) {
            return null;
        }
        return row[byte2];
    }

    boolean isTLS13() {
        return isTLS13;
    }

    static final class Entry {
        private final String name;
        private final List openSslNames;
        private final List aliases;
        private final KeyAgreement keyAgreement;
        private final Authentication authentication;
        private final Encryption encryption;
        private final Digest digest;
        private final Protocol protocol;
        private final boolean export;
        private final SecurityLevel level;
        private final boolean fips;
        private final int strengthBits;
        private final int algorithmBits;

        Entry(final String name, final List openSslNames, final List aliases, final KeyAgreement keyAgreement, final Authentication authentication, final Encryption encryption, final Digest digest, final Protocol protocol, final boolean export, final SecurityLevel level, final boolean fips, final int strengthBits, final int algorithmBits) {
            this.name = name;
            this.openSslNames = openSslNames;
            this.aliases = aliases;
            this.keyAgreement = keyAgreement;
            this.authentication = authentication;
            this.encryption = encryption;
            this.digest = digest;
            this.protocol = protocol;
            this.export = export;
            this.level = level;
            this.fips = fips;
            this.strengthBits = strengthBits;
            this.algorithmBits = algorithmBits;
        }

        public String getName() {
            return name;
        }

        public List getOpenSslNames() {
            return openSslNames;
        }

        public List getAliases() {
            return aliases;
        }

        public KeyAgreement getKeyAgreement() {
            return keyAgreement;
        }

        public Authentication getAuthentication() {
            return authentication;
        }

        public Encryption getEncryption() {
            return encryption;
        }

        public Digest getDigest() {
            return digest;
        }

        public Protocol getProtocol() {
            return protocol;
        }

        public boolean isExport() {
            return export;
        }

        public SecurityLevel getLevel() {
            return level;
        }

        public boolean isFips() {
            return fips;
        }

        public int getStrengthBits() {
            return strengthBits;
        }

        public int getAlgorithmBits() {
            return algorithmBits;
        }

        public String toString() {
            return getName() + "/" + String.join("|", openSslNames);
        }
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy