[TODO] Make use of the SetRequestInformationMechanismFactory to set t…

…he request information during HTTP authentication
Skyllarr · Nov 20, 2023 · f9cf847 · f9cf847
1 parent 2178f92
commit f9cf847
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/elytron/src/main/java/org/wildfly/extension/elytron/HttpServerDefinitions.java b/elytron/src/main/java/org/wildfly/extension/elytron/HttpServerDefinitions.java
@@ -57,6 +57,7 @@
 import org.wildfly.security.http.util.SecurityProviderServerMechanismFactory;
 import org.wildfly.security.http.util.ServiceLoaderServerMechanismFactory;
 import org.wildfly.security.http.util.SetMechanismInformationMechanismFactory;
+import org.wildfly.security.http.util.SetRequestInformationMechanismFactory;
 import org.wildfly.security.http.util.SocketAddressCallbackServerMechanismFactory;
 
 /**
@@ -150,6 +151,7 @@ protected ValueSupplier<HttpServerAuthenticationMechanismFactory> getValueSuppli
                 final Map<String, String> propertiesMap = PROPERTIES.unwrap(context, model);
                 return () -> {
                     HttpServerAuthenticationMechanismFactory factory = factoryInjector.getValue();
+                    factory = new SetRequestInformationMechanismFactory(factory);
                     factory = new SocketAddressCallbackServerMechanismFactory(factory);
                     factory = new SetMechanismInformationMechanismFactory(factory);
                     factory = finalFilter != null ? new FilterServerMechanismFactory(factory, finalFilter) : factory;
@@ -192,7 +194,7 @@ protected ValueSupplier<HttpServerAuthenticationMechanismFactory> getValueSuppli
                     if ( findProviderService(actualProviders, serviceFilter) == null ) {
                         throw ROOT_LOGGER.noSuitableProvider(HttpServerAuthenticationMechanismFactory.class.getSimpleName());
                     }
-                    return new SocketAddressCallbackServerMechanismFactory(new SetMechanismInformationMechanismFactory(new SecurityProviderServerMechanismFactory(actualProviders)));
+                    return new SetRequestInformationMechanismFactory(new SocketAddressCallbackServerMechanismFactory(new SetMechanismInformationMechanismFactory(new SecurityProviderServerMechanismFactory(actualProviders))));
                 };
             }
 
@@ -216,7 +218,7 @@ protected ValueSupplier<HttpServerAuthenticationMechanismFactory> getValueSuppli
                     try {
                         ClassLoader classLoader = doPrivileged((PrivilegedExceptionAction<ClassLoader>) () -> resolveClassLoader(module));
 
-                        return new SocketAddressCallbackServerMechanismFactory(new SetMechanismInformationMechanismFactory(new ServiceLoaderServerMechanismFactory(classLoader)));
+                        return new SetRequestInformationMechanismFactory(new SocketAddressCallbackServerMechanismFactory(new SetMechanismInformationMechanismFactory(new ServiceLoaderServerMechanismFactory(classLoader))));
                     } catch (Exception e) {
                         throw new StartException(e);
                     }