-
Notifications
You must be signed in to change notification settings - Fork 263
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[ELY-2583] Make requestURI and Source-Address available from RealmSuc…
…cessfulAuthenticationEvent and RealmFailedAuthenticationEvent
- Loading branch information
Showing
5 changed files
with
306 additions
and
2 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
131 changes: 131 additions & 0 deletions
131
tests/base/src/test/java/org/wildfly/security/auth/server/RealmEventTest.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,131 @@ | ||
| /* | ||
| * JBoss, Home of Professional Open Source. | ||
| * Copyright 2023 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.server; | ||
|
|
||
| import org.junit.Test; | ||
| import org.wildfly.security.auth.callback.MechanismInformationCallback; | ||
| import org.wildfly.security.auth.permission.LoginPermission; | ||
| import org.wildfly.security.authz.Attributes; | ||
| import org.wildfly.security.authz.MapAttributes; | ||
| import org.wildfly.security.credential.PasswordCredential; | ||
| import org.wildfly.security.password.PasswordFactory; | ||
| import org.wildfly.security.password.WildFlyElytronPasswordProvider; | ||
| import org.wildfly.security.password.interfaces.ClearPassword; | ||
| import org.wildfly.security.password.spec.ClearPasswordSpec; | ||
| import org.wildfly.security.permission.PermissionVerifier; | ||
|
|
||
| import javax.security.auth.callback.CallbackHandler; | ||
| import javax.security.auth.callback.UnsupportedCallbackException; | ||
| import java.io.IOException; | ||
| import java.net.URI; | ||
| import java.net.URISyntaxException; | ||
| import java.security.NoSuchAlgorithmException; | ||
| import java.security.Provider; | ||
| import java.security.Security; | ||
| import java.security.spec.InvalidKeySpecException; | ||
|
|
||
| import static org.junit.Assert.fail; | ||
| import static org.wildfly.security.authz.RoleDecoder.KEY_SOURCE_ADDRESS; | ||
|
|
||
| public class RealmEventTest { | ||
|
|
||
| private static SecurityDomain usersDomain; | ||
| private static TestCustomRealm.CustomRealm usersRealm; | ||
| private static final Provider provider = WildFlyElytronPasswordProvider.getInstance(); | ||
|
|
||
| private ServerAuthenticationContext setupAndGetServerAuthenticationContext() throws IOException, UnsupportedCallbackException { | ||
| Security.addProvider(provider); | ||
|
|
||
| usersRealm = new TestCustomRealm.CustomRealm(); | ||
| SecurityDomain.Builder builder = SecurityDomain.builder(); | ||
| builder.addRealm("users", usersRealm).build(); | ||
| builder.setDefaultRealmName("users"); | ||
| builder.setPermissionMapper((permissionMappable, roles) -> PermissionVerifier.from(new LoginPermission())); | ||
| usersDomain = builder.build(); | ||
|
|
||
| ServerAuthenticationContext serverAuthenticationContext = usersDomain.createNewAuthenticationContext(); | ||
| serverAuthenticationContext.addRuntimeAttributes(createRuntimeAttributesWithSourceAddress()); | ||
|
|
||
| MechanismInformation mechanismInformation = new MechanismInformation() { | ||
| @Override | ||
| public String getMechanismType() { | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public String getMechanismName() { | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public String getHostName() { | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public String getProtocol() { | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public URI getRequestURI() { | ||
| try { | ||
| return new URI("http://www.testURI.com"); | ||
| } catch (URISyntaxException e) { | ||
| throw new RuntimeException(e); | ||
| } | ||
| } | ||
| }; | ||
|
|
||
| CallbackHandler callbackHandler = serverAuthenticationContext.createCallbackHandler(); | ||
| callbackHandler.handle(new MechanismInformationCallback[]{new MechanismInformationCallback(mechanismInformation)}); | ||
| return serverAuthenticationContext; | ||
| } | ||
|
|
||
| @Test | ||
| public void testRealmSuccessfulAuthenticationEvent() throws IOException, UnsupportedCallbackException { | ||
| ServerAuthenticationContext serverAuthenticationContext = setupAndGetServerAuthenticationContext(); | ||
| try { | ||
| serverAuthenticationContext.setAuthenticationName("myadmin"); | ||
| serverAuthenticationContext.addPublicCredential(new PasswordCredential( | ||
| PasswordFactory.getInstance(ClearPassword.ALGORITHM_CLEAR).generatePassword( | ||
| new ClearPasswordSpec("mypassword".toCharArray())))); | ||
|
|
||
| serverAuthenticationContext.authorize(); | ||
| } catch (RealmUnavailableException | InvalidKeySpecException | NoSuchAlgorithmException e) { | ||
| fail(); | ||
| } | ||
| serverAuthenticationContext.succeed(); | ||
| } | ||
|
|
||
| @Test | ||
| public void testRealmFailedAuthenticationEvent() throws NoSuchAlgorithmException, IOException, UnsupportedCallbackException, InvalidKeySpecException { | ||
| ServerAuthenticationContext serverAuthenticationContext = setupAndGetServerAuthenticationContext(); | ||
| serverAuthenticationContext.setAuthenticationName("myadmin"); | ||
| serverAuthenticationContext.addPublicCredential(new PasswordCredential( | ||
| PasswordFactory.getInstance(ClearPassword.ALGORITHM_CLEAR).generatePassword( | ||
| new ClearPasswordSpec("wrongPassword".toCharArray())))); | ||
| serverAuthenticationContext.fail(); | ||
| } | ||
|
|
||
| private Attributes createRuntimeAttributesWithSourceAddress() { | ||
| MapAttributes runtimeAttributes = new MapAttributes(); | ||
| runtimeAttributes.addFirst(KEY_SOURCE_ADDRESS, "10.12.14.16"); | ||
| return runtimeAttributes; | ||
| } | ||
| } |
112 changes: 112 additions & 0 deletions
112
tests/base/src/test/java/org/wildfly/security/auth/server/TestCustomRealm.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,112 @@ | ||
| /* | ||
| * JBoss, Home of Professional Open Source. | ||
| * Copyright 2023 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.server; | ||
|
|
||
| import org.wildfly.security.auth.SupportLevel; | ||
| import org.wildfly.security.auth.server.event.RealmEvent; | ||
| import org.wildfly.security.auth.server.event.RealmFailedAuthenticationEvent; | ||
| import org.wildfly.security.auth.server.event.RealmSuccessfulAuthenticationEvent; | ||
| import org.wildfly.security.credential.Credential; | ||
| import org.wildfly.security.evidence.Evidence; | ||
| import org.wildfly.security.evidence.PasswordGuessEvidence; | ||
|
|
||
| import java.security.Principal; | ||
| import java.security.spec.AlgorithmParameterSpec; | ||
| import java.util.Arrays; | ||
|
|
||
| import static org.junit.Assert.assertEquals; | ||
| import static org.junit.Assert.fail; | ||
|
|
||
| public class TestCustomRealm { | ||
|
|
||
| public static class CustomRealm implements SecurityRealm { | ||
|
|
||
| // this realm does not allow acquiring credentials | ||
| public SupportLevel getCredentialAcquireSupport(Class<? extends Credential> credentialType, String algorithmName, | ||
| AlgorithmParameterSpec parameterSpec) throws RealmUnavailableException { | ||
| return SupportLevel.UNSUPPORTED; | ||
| } | ||
|
|
||
| // this realm will be able to verify password evidences only | ||
| public SupportLevel getEvidenceVerifySupport(Class<? extends Evidence> evidenceType, String algorithmName) | ||
| throws RealmUnavailableException { | ||
| return PasswordGuessEvidence.class.isAssignableFrom(evidenceType) ? SupportLevel.POSSIBLY_SUPPORTED : SupportLevel.UNSUPPORTED; | ||
| } | ||
|
|
||
| public RealmIdentity getRealmIdentity(final Principal principal) throws RealmUnavailableException { | ||
|
|
||
| if ("myadmin".equals(principal.getName())) { // identity "myadmin" will have password "mypassword" | ||
| return new RealmIdentity() { | ||
| public Principal getRealmIdentityPrincipal() { | ||
| return principal; | ||
| } | ||
|
|
||
| public SupportLevel getCredentialAcquireSupport(Class<? extends Credential> credentialType, | ||
| String algorithmName, AlgorithmParameterSpec parameterSpec) throws RealmUnavailableException { | ||
| return SupportLevel.UNSUPPORTED; | ||
| } | ||
|
|
||
| public <C extends Credential> C getCredential(Class<C> credentialType) throws RealmUnavailableException { | ||
| return null; | ||
| } | ||
|
|
||
| public SupportLevel getEvidenceVerifySupport(Class<? extends Evidence> evidenceType, String algorithmName) { | ||
| return PasswordGuessEvidence.class.isAssignableFrom(evidenceType) ? SupportLevel.SUPPORTED : SupportLevel.UNSUPPORTED; | ||
| } | ||
|
|
||
| // evidence will be accepted if it is password "mypassword" | ||
| public boolean verifyEvidence(Evidence evidence) { | ||
| if (evidence instanceof PasswordGuessEvidence) { | ||
| PasswordGuessEvidence guess = (PasswordGuessEvidence) evidence; | ||
| try { | ||
| return Arrays.equals("mypassword".toCharArray(), guess.getGuess()); | ||
|
|
||
| } finally { | ||
| guess.destroy(); | ||
| } | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| public boolean exists() { | ||
| return true; | ||
| } | ||
| }; | ||
| } | ||
| return RealmIdentity.NON_EXISTENT; | ||
| } | ||
|
|
||
| @Override | ||
| public void handleRealmEvent(RealmEvent event) { | ||
| if (event instanceof RealmSuccessfulAuthenticationEvent) { | ||
| assertEquals("10.12.14.16", ((RealmSuccessfulAuthenticationEvent) event).getAuthorizationIdentity().getRuntimeAttributes().get("Source-Address").get(0)); | ||
| assertEquals("testURI", ((RealmSuccessfulAuthenticationEvent) event).getAuthorizationIdentity().getRuntimeAttributes().get("Request-URI").get(0)); | ||
| assertEquals("myadmin", ((RealmSuccessfulAuthenticationEvent) event).getRealmIdentity().getRealmIdentityPrincipal()); | ||
| } | ||
| if (event instanceof RealmFailedAuthenticationEvent) { | ||
| try { | ||
| assertEquals("10.12.14.16", ((RealmFailedAuthenticationEvent) event).getRealmIdentity().getAuthorizationIdentity().getRuntimeAttributes().get("Source-Address").get(0)); | ||
| assertEquals("testURI", ((RealmFailedAuthenticationEvent) event).getRealmIdentity().getAuthorizationIdentity().getRuntimeAttributes().get("Request-URI").get(0)); | ||
| assertEquals("myadmin", ((RealmSuccessfulAuthenticationEvent) event).getRealmIdentity().getRealmIdentityPrincipal()); | ||
| } catch (RealmUnavailableException e) { | ||
| fail("RealmFailedAuthenticationEvent should have runtime attributes associated"); | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } |