-
Notifications
You must be signed in to change notification settings - Fork 282
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[ELY-2112] Automatic registration of client side / JVM wide default S…
…SLContext
- Loading branch information
Showing
14 changed files
with
638 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
117 changes: 117 additions & 0 deletions
117
.../java/org/wildfly/security/auth/client/WildFlyElytronClientDefaultSSLContextProvider.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,117 @@ | ||
/* | ||
* JBoss, Home of Professional Open Source. | ||
* Copyright 2021 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.auth.client; | ||
|
||
import org.kohsuke.MetaInfServices; | ||
import org.wildfly.client.config.ConfigXMLParseException; | ||
import org.wildfly.security.auth.client._private.ElytronMessages; | ||
import static org.wildfly.security.auth.client._private.ElytronMessages.log; | ||
|
||
import java.io.FileNotFoundException; | ||
import java.security.GeneralSecurityException; | ||
import java.security.NoSuchAlgorithmException; | ||
import java.security.Provider; | ||
import java.util.List; | ||
import java.util.Map; | ||
|
||
/** | ||
* Provider that loads Elytron client configuration and provides default SSLContext which can be returned with SSLContext.getDefault() call. | ||
* Default SSLContext is the configured SSL context that does not have any specific rule when it should be used, so it matches all rules. | ||
*/ | ||
@MetaInfServices(value = Provider.class) | ||
public final class WildFlyElytronClientDefaultSSLContextProvider extends Provider { | ||
|
||
private static final long serialVersionUID = -8281186085283177185L; | ||
public static final String ELYTRON_CLIENT_DEFAULT_SSL_CONTEXT_PROVIDER_NAME = "WildFlyElytronClientDefaultSSLContextProvider"; | ||
|
||
/** | ||
* WildFlyElytronClientDefaultSSLContextProvider that uses Elytron client configuration found on classpath. | ||
*/ | ||
public WildFlyElytronClientDefaultSSLContextProvider() { | ||
this(null); | ||
} | ||
|
||
/** | ||
* WildFlyElytronClientDefaultSSLContextProvider that uses Elytron client configuration found on provided path. | ||
* | ||
* @param configPath path to Elytron client configuration path | ||
*/ | ||
public WildFlyElytronClientDefaultSSLContextProvider(String configPath) { | ||
super(ELYTRON_CLIENT_DEFAULT_SSL_CONTEXT_PROVIDER_NAME, 1.0, "Elytron client provider for default SSLContext"); | ||
putService(new ClientSSLContextProviderService(this, "SSLContext", "Default", "org.wildfly.security.auth.client.provider.WildFlyElytronClientDefaultSSLContextSpi", null, null, configPath)); | ||
} | ||
|
||
private static final class ClientSSLContextProviderService extends Provider.Service { | ||
String configPath; | ||
// this is Integer because we need to count the number of times entered | ||
// entered.get()==2 means we requested this provider second time, creating a loop, so we throw an sslContextForSecurityProviderCreatesInfiniteLoop exception | ||
// AuthenticationContextConfigurationClient receives sslContextForSecurityProviderCreatesInfiniteLoop exception during obtaining of default SSL context and will therefore request default SSL context from other providers | ||
// after default SSL context from other provider is returned, we must check the entered variable again and throw an exception to inform users that this provider was unsuccessful because of invalid configuration | ||
private final ThreadLocal<Integer> entered = new ThreadLocal<>(); | ||
|
||
ClientSSLContextProviderService(Provider provider, String type, String algorithm, String className, List<String> aliases, | ||
Map<String, String> attributes, String configPath) { | ||
super(provider, type, algorithm, className, aliases, attributes); | ||
this.configPath = configPath; | ||
} | ||
|
||
/** | ||
* There is a risk of looping if the Elytron client configuration is invalid. | ||
* Loop will happen when Elytron client provider has configured default SSL context to be SSLContext::getDefault. | ||
* Entered variable counts the number of entrances in order to avoid this loop. | ||
* When it is equal or higher than 2 the NoSuchAlgorithmException will be thrown. | ||
* When this exception is encountered, JVM tries to obtain default SSLContext from providers of lower priority | ||
* and returns it to Elytron client as the default SSL context. | ||
* Since we do not want to wrap the SSL context from other provider with this provider, we will throw an exception again | ||
* which makes JVM escape this provider entirely and continue in the list of other providers. | ||
*/ | ||
@Override | ||
public Object newInstance(Object ignored) throws NoSuchAlgorithmException { | ||
Integer enteredCountTmp = entered.get(); | ||
entered.set(enteredCountTmp == null ? 1 : enteredCountTmp + 1); | ||
if (entered.get() >= 2) { | ||
// we do not do clean up entered variable here because it is needed for the second check and possible throwing of second exception below | ||
throw ElytronMessages.log.sslContextForSecurityProviderCreatesInfiniteLoop(); | ||
} | ||
|
||
WildFlyElytronClientDefaultSSLContextSpi sslContext; | ||
try { | ||
if (configPath == null) { | ||
sslContext = new WildFlyElytronClientDefaultSSLContextSpi(AuthenticationContext.captureCurrent()); | ||
} else { | ||
sslContext = new WildFlyElytronClientDefaultSSLContextSpi(this.configPath); | ||
} | ||
// if we had an exception previously, then default SSLContext was still returned from other security provider of lower priority in | ||
// AuthenticationContextConfigurationClient#getSSLContextFactory method. | ||
// Since we do not want to wrap SSLContext from other provider with this provider, we need to check entered variable again | ||
// and throw an exception which makes JVM ignore this provider and probe other providers again | ||
if (entered.get() >= 2) { | ||
throw ElytronMessages.log.sslContextForSecurityProviderCreatesInfiniteLoop(); | ||
} | ||
} catch (ConfigXMLParseException | GeneralSecurityException e) { | ||
if (e.getCause() instanceof FileNotFoundException) { | ||
throw log.clientConfigurationFileNotFound(); | ||
} | ||
throw new NoSuchAlgorithmException(e); | ||
} finally { | ||
entered.remove(); | ||
} | ||
return sslContext; | ||
} | ||
} | ||
} |
139 changes: 139 additions & 0 deletions
139
.../main/java/org/wildfly/security/auth/client/WildFlyElytronClientDefaultSSLContextSpi.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,139 @@ | ||
/* | ||
* JBoss, Home of Professional Open Source. | ||
* Copyright 2021 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.auth.client; | ||
|
||
import org.wildfly.client.config.ConfigXMLParseException; | ||
import org.wildfly.common.Assert; | ||
import org.wildfly.security.auth.client._private.ElytronMessages; | ||
|
||
import javax.net.ssl.KeyManager; | ||
import javax.net.ssl.SSLContext; | ||
import javax.net.ssl.SSLContextSpi; | ||
import javax.net.ssl.SSLEngine; | ||
import javax.net.ssl.SSLServerSocketFactory; | ||
import javax.net.ssl.SSLSessionContext; | ||
import javax.net.ssl.SSLSocketFactory; | ||
import javax.net.ssl.TrustManager; | ||
import java.io.File; | ||
import java.net.URI; | ||
import java.net.URISyntaxException; | ||
import java.nio.file.Paths; | ||
import java.security.AccessController; | ||
import java.security.GeneralSecurityException; | ||
import java.security.PrivilegedAction; | ||
import java.security.SecureRandom; | ||
|
||
/** | ||
* SSLContextSpi that is used by WildFlyElytronClientDefaultSSLContextProvider | ||
*/ | ||
public class WildFlyElytronClientDefaultSSLContextSpi extends SSLContextSpi { | ||
|
||
private SSLContext configuredDefaultClientSSLContext; | ||
|
||
/** | ||
* SSLContextSpi used by WildFlyElytronClientDefaultSSLContextProvider that uses AuthenticationContext found on the classpath to retrieve default SSLContext. | ||
*/ | ||
public WildFlyElytronClientDefaultSSLContextSpi() throws GeneralSecurityException { | ||
this(AuthenticationContext.captureCurrent()); | ||
} | ||
|
||
/** | ||
* SSLContextSpi used by WildFlyElytronClientDefaultSSLContextProvider that uses Elytron client configuration from provided path to retrieve default SSLContext. | ||
* | ||
* @param configPath path to the Elytron client configuration file | ||
*/ | ||
public WildFlyElytronClientDefaultSSLContextSpi(String configPath) throws GeneralSecurityException, ConfigXMLParseException { | ||
this(ElytronXmlParser.parseAuthenticationClientConfiguration(configPathUrlToUri(configPath)).create()); | ||
} | ||
|
||
/** | ||
* SSLContextSpi used by WildFlyElytronClientDefaultSSLContextProvider that uses Elytron client configuration from provided authentication context. | ||
* | ||
* @param authenticationContext authentication context used to retrieve default SSLContext | ||
*/ | ||
public WildFlyElytronClientDefaultSSLContextSpi(AuthenticationContext authenticationContext) throws GeneralSecurityException { | ||
Assert.checkNotNullParam("authenticationContext", authenticationContext); | ||
|
||
AuthenticationContextConfigurationClient AUTH_CONTEXT_CLIENT = AccessController.doPrivileged((PrivilegedAction<AuthenticationContextConfigurationClient>) AuthenticationContextConfigurationClient::new); | ||
this.configuredDefaultClientSSLContext = AUTH_CONTEXT_CLIENT.getSSLContext(authenticationContext); | ||
} | ||
|
||
@Override | ||
protected void engineInit(KeyManager[] keyManagers, TrustManager[] trustManagers, SecureRandom secureRandom) { | ||
// ignore | ||
} | ||
|
||
@Override | ||
protected SSLSocketFactory engineGetSocketFactory() { | ||
return this.configuredDefaultClientSSLContext.getSocketFactory(); | ||
} | ||
|
||
@Override | ||
protected SSLServerSocketFactory engineGetServerSocketFactory() { | ||
return this.configuredDefaultClientSSLContext.getServerSocketFactory(); | ||
} | ||
|
||
@Override | ||
protected SSLEngine engineCreateSSLEngine() { | ||
return this.configuredDefaultClientSSLContext.createSSLEngine(); | ||
} | ||
|
||
@Override | ||
protected SSLEngine engineCreateSSLEngine(String s, int i) { | ||
return this.configuredDefaultClientSSLContext.createSSLEngine(s, i); | ||
} | ||
|
||
@Override | ||
protected SSLSessionContext engineGetServerSessionContext() { | ||
return this.configuredDefaultClientSSLContext.getServerSessionContext(); | ||
} | ||
|
||
@Override | ||
protected SSLSessionContext engineGetClientSessionContext() { | ||
return this.configuredDefaultClientSSLContext.getClientSessionContext(); | ||
} | ||
|
||
/** | ||
* Source: A utility method taken from https://github.com/wildfly/wildfly-client-config/blob/master/src/main/java/org/wildfly/client/config/ClientConfiguration.java on March 2022 | ||
*/ | ||
static URI configPathUrlToUri(String wildFlyConfig) { | ||
if (wildFlyConfig == null || wildFlyConfig.isEmpty()) { | ||
throw ElytronMessages.log.clientConfigurationFileNotValid(); | ||
} | ||
if (File.separator.equals("\\") && wildFlyConfig.contains("\\")) { // we are on the windows and path is for windows | ||
File f = new File(wildFlyConfig); | ||
return f.toPath().toUri(); | ||
} else { | ||
try { | ||
URI uri = new URI(wildFlyConfig); | ||
if (!uri.isAbsolute()) { // URI does not include schema | ||
if (uri.getPath().charAt(0) != File.separatorChar && uri.getPath().charAt(0) != '/') { // relative path | ||
String userDir = System.getProperty("user.dir").replace(File.separatorChar, '/'); | ||
return Paths.get(userDir, uri.getPath()).toUri(); | ||
} else { // absolute path | ||
return Paths.get(uri.getPath()).toUri(); | ||
} | ||
} | ||
return uri; | ||
} catch (URISyntaxException e) { | ||
// no config file there | ||
return null; | ||
} | ||
} | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
46 changes: 46 additions & 0 deletions
46
...client/src/test/java/org/wildfly/security/auth/client/DefaultSSLContextEmptyPathTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,46 @@ | ||
/* | ||
* JBoss, Home of Professional Open Source. | ||
* Copyright 2021 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.auth.client; | ||
|
||
import org.junit.Assert; | ||
import org.junit.Test; | ||
|
||
import javax.net.ssl.SSLContext; | ||
import java.security.NoSuchAlgorithmException; | ||
import java.security.Security; | ||
|
||
/** | ||
* Test that default SSLContext provider will throw an exception when configured file path is empty | ||
*/ | ||
public class DefaultSSLContextEmptyPathTest { | ||
|
||
@Test(expected = IllegalArgumentException.class) | ||
public void defaultSSLContextNonexistentConfigFileTest() { | ||
Security.insertProviderAt(new WildFlyElytronClientDefaultSSLContextProvider(""), 1); | ||
Assert.assertNotNull(Security.getProvider("WildFlyElytronClientDefaultSSLContextProvider")); | ||
AuthenticationContext authenticationContext = AuthenticationContext.captureCurrent(); | ||
authenticationContext.run(() -> { | ||
try { | ||
SSLContext.getDefault(); | ||
} catch (NoSuchAlgorithmException e) { | ||
Assert.fail("Obtaining default SSL context from provider with invalid path threw incorrect exception"); | ||
} | ||
}); | ||
} | ||
} |
Oops, something went wrong.