-
Notifications
You must be signed in to change notification settings - Fork 462
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #6010 from yersan/WFCORE-6825-WFCORE-6833
[CVE-2024-4029] [WFCORE-6825] [WFCORE-6833] Options to control resources used for incomming connections and XNIO upgrade
- Loading branch information
Showing
4 changed files
with
247 additions
and
3 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
156 changes: 156 additions & 0 deletions
156
...test/java/org/wildfly/core/test/standalone/mgmt/ManagementInterfaceResourcesTestCase.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,156 @@ | ||
| /* | ||
| * Copyright The WildFly Authors | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| */ | ||
|
|
||
| package org.wildfly.core.test.standalone.mgmt; | ||
|
|
||
| import static org.jboss.as.test.shared.TimeoutUtil.adjust; | ||
| import static org.junit.Assert.assertTrue; | ||
|
|
||
| import java.io.IOException; | ||
| import java.net.InetSocketAddress; | ||
| import java.net.Socket; | ||
| import java.net.SocketAddress; | ||
| import java.util.logging.Level; | ||
| import java.util.logging.Logger; | ||
|
|
||
| import jakarta.inject.Inject; | ||
|
|
||
| import org.jboss.as.test.integration.management.util.CLIWrapper; | ||
| import org.jboss.as.test.shared.TestSuiteEnvironment; | ||
| import org.junit.Test; | ||
| import org.junit.runner.RunWith; | ||
| import org.wildfly.common.function.ExceptionRunnable; | ||
| import org.wildfly.core.testrunner.ServerControl; | ||
| import org.wildfly.core.testrunner.ServerController; | ||
| import org.wildfly.core.testrunner.WildFlyRunner; | ||
|
|
||
| /** | ||
| * Test case to test resource limits and clean up of management interface connections. | ||
| * | ||
| * @author <a href="mailto:darran.lofthouse@jboss.com">Darran Lofthouse</a> | ||
| */ | ||
| @RunWith(WildFlyRunner.class) | ||
| @ServerControl(manual = true) | ||
| public class ManagementInterfaceResourcesTestCase { | ||
| private static final Logger LOG = Logger.getLogger(ManagementInterfaceResourcesTestCase.class.getName()); | ||
|
|
||
| private static final String BACKLOG_PROPERTY = "org.wildfly.management.backlog"; | ||
| private static final String CONNECTION_HIGH_WATER_PROPERTY = "org.wildfly.management.connection-high-water"; | ||
| private static final String CONNECTION_LOW_WATER_PROPERTY = "org.wildfly.management.connection-low-water"; | ||
| private static final String NO_REQUEST_TIMEOUT_PROPERTY = "org.wildfly.management.no-request-timeout"; | ||
|
|
||
| @Inject | ||
| protected static ServerController controller; | ||
|
|
||
| /** | ||
| * Test that the management interface will not accept new connections when the number of active connections reaches the | ||
| * high water mark. After the number of open connections has been reduced to the low watermark it will test that connections | ||
| * are accepted again. | ||
| */ | ||
| @Test | ||
| public void testWatermarks() throws Exception { | ||
| runTest(60000, () -> { | ||
| String mgmtAddress = TestSuiteEnvironment.getServerAddress(); | ||
| int mgmtPort = TestSuiteEnvironment.getServerPort(); | ||
| LOG.info(mgmtAddress + ":" + mgmtPort); | ||
| SocketAddress targetAddress = new InetSocketAddress(mgmtAddress, mgmtPort); | ||
|
|
||
| int socketsOpened = 0; | ||
| boolean oneFailed = false; | ||
| Socket[] sockets = new Socket[9]; | ||
| for (int i = 0 ; i < 9 ; i++) { | ||
| LOG.info("Opening socket " + i + " socketsOpened=" + socketsOpened); | ||
| try { | ||
| sockets[i] = new Socket(); | ||
| sockets[i].connect(targetAddress, 5000); | ||
| socketsOpened++; | ||
| } catch (IOException e) { | ||
| LOG.log(Level.SEVERE, "Probably an expected exception trying to open a new connection", e); | ||
| assertTrue("Less sockets than low watermark opened.", socketsOpened > 3); | ||
| oneFailed = true; | ||
| } | ||
| } | ||
| assertTrue("Opening of one socket was expected to fail.", oneFailed); | ||
|
|
||
| // Now close the connections and we should be able to connect again. | ||
| for (int i = 0 ; i < socketsOpened ; i++) { | ||
| sockets[i].close(); | ||
| } | ||
|
|
||
| Socket goodSocket = new Socket(); | ||
| // This needs a reasonable time to give the server time to respond to the closed connections. | ||
| goodSocket.connect(targetAddress, 10000); | ||
| goodSocket.close(); | ||
| }); | ||
| } | ||
|
|
||
| @Test | ||
| public void testTimeout() throws Exception { | ||
| runTest(10000, () -> { | ||
| String mgmtAddress = TestSuiteEnvironment.getServerAddress(); | ||
| int mgmtPort = TestSuiteEnvironment.getServerPort(); | ||
| SocketAddress targetAddress = new InetSocketAddress(mgmtAddress, mgmtPort); | ||
|
|
||
| int socketsOpened = 0; | ||
| boolean oneFailed = false; | ||
| Socket[] sockets = new Socket[9]; | ||
| for (int i = 0 ; i < 9 ; i++) { | ||
| LOG.info("Opening socket " + i + " socketsOpened=" + socketsOpened); | ||
| try { | ||
| sockets[i] = new Socket(); | ||
| sockets[i].connect(targetAddress, 5000); | ||
| socketsOpened++; | ||
| } catch (IOException e) { | ||
| LOG.log(Level.SEVERE, "Probably an expected exception trying to open a new connection", e); | ||
| assertTrue("Less sockets than low watermark opened.", socketsOpened > 3); | ||
| oneFailed = true; | ||
| } | ||
| } | ||
| assertTrue("Opening of one socket was expected to fail.", oneFailed); | ||
|
|
||
| // Notice that the exception received when we tried to open a new socket could have been a timeout (SocketTimeoutException) | ||
| // or a connection refused (IOException). It depends on the OS and the network configuration. | ||
| // So, we could also have had 5000ms for each bad socket that triggered a SocketTimeoutException. | ||
| Thread.sleep(adjust(12000)); | ||
|
|
||
| Socket goodSocket = new Socket(); | ||
| // This needs to be longer than 500ms to give the server time to respond to the closed connections. | ||
| goodSocket.connect(targetAddress, 10000); | ||
| goodSocket.close(); | ||
|
|
||
| // Clean up remaining sockets | ||
| for (int i = 0 ; i < socketsOpened ; i++) { | ||
| sockets[i].close(); | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| private void runTest(int noRequestTimeout, ExceptionRunnable<Exception> test) throws Exception { | ||
| controller.startInAdminMode(); | ||
| try (CLIWrapper cli = new CLIWrapper(true)) { | ||
| cli.sendLine(String.format("/system-property=%s:add(value=%d)", BACKLOG_PROPERTY, 2)); | ||
| cli.sendLine(String.format("/system-property=%s:add(value=%d)", CONNECTION_HIGH_WATER_PROPERTY, 6)); | ||
| cli.sendLine(String.format("/system-property=%s:add(value=%d)", CONNECTION_LOW_WATER_PROPERTY, 3)); | ||
| cli.sendLine(String.format("/system-property=%s:add(value=%d)", NO_REQUEST_TIMEOUT_PROPERTY, noRequestTimeout)); | ||
| } | ||
|
|
||
| try { | ||
| controller.reload(); | ||
|
|
||
| test.run(); | ||
| } finally { | ||
| controller.reload(); | ||
|
|
||
| try (CLIWrapper cli = new CLIWrapper(true)) { | ||
| cli.sendLine(String.format("/system-property=%s:remove()", BACKLOG_PROPERTY)); | ||
| cli.sendLine(String.format("/system-property=%s:remove()", CONNECTION_HIGH_WATER_PROPERTY)); | ||
| cli.sendLine(String.format("/system-property=%s:remove()", CONNECTION_LOW_WATER_PROPERTY)); | ||
| cli.sendLine(String.format("/system-property=%s:remove()", NO_REQUEST_TIMEOUT_PROPERTY)); | ||
| } | ||
| controller.stop(); | ||
| } | ||
| } | ||
|
|
||
| } |