Skip to content

Commit

Permalink
Backport 70f2238ba9a81ba0bb3fe6cbd98a553009992ecb
Browse files Browse the repository at this point in the history
  • Loading branch information
@ktakakuri
ktakakuri committed Jan 25, 2024
1 parent 552c686 commit 98cff0f
Showing 1 changed file with 96 additions and 46 deletions.
142 changes: 96 additions & 46 deletions jdk/test/sun/management/jmxremote/startstop/JMXStartStopTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,47 @@ public class JMXStartStopTest {

private static final boolean verbose = false;

/**
* Dynamically allocates two distinct ports using {@linkplain java.net.ServerSocket}
* It keeps each of those ports blocked until it is first accessed by its getter
*/
private static class PortAllocator {
private final int port1, port2;
private final ServerSocket ss1, ss2;
PortAllocator() {
try {
ss1 = new ServerSocket(0);
ss2 = new ServerSocket(0);
port1 = ss1.getLocalPort();
port2 = ss2.getLocalPort();
} catch (IOException e) {
throw new Error("Error while obtaining free ports", e);
}
}

public int getPort1() {
if (!ss1.isClosed()) {
try {
ss1.close();
} catch (IOException e) {
// just ignore
}
}
return port1;
}

public int getPort2() {
if (!ss2.isClosed()) {
try {
ss2.close();
} catch (IOException e) {
// just ignore
}
}
return port2;
}
}

private static void dbg_print(String msg){
if (verbose) {
System.out.println("DBG: " +msg);
Expand Down Expand Up @@ -407,29 +448,28 @@ private static void jcmd(String target, final Consumer<String> c, String ... com
private static final String CMD_STOP = "ManagementAgent.stop";
private static final String CMD_START= "ManagementAgent.start";
private static final String CMD_START_LOCAL = "ManagementAgent.start_local";
private static final int port1 = 50234;
private static final int port2 = 50235;

static void test_01() throws Exception {
// Run an app with JMX enabled stop it and
// restart on other port

System.out.println("**** Test one ****");
PortAllocator pa = new PortAllocator();

Something s = doSomething(
"test_01",
"-Dcom.sun.management.jmxremote.port=" + port1,
"-Dcom.sun.management.jmxremote.port=" + pa.getPort1(),
"-Dcom.sun.management.jmxremote.authenticate=false",
"-Dcom.sun.management.jmxremote.ssl=false");

try {
testConnect(port1);
testConnect(pa.getPort1());

jcmd(CMD_STOP);
testNoConnect(port1);
testNoConnect(pa.getPort1());

jcmd(CMD_START, "jmxremote.port=" + port2);
testConnect(port2);
jcmd(CMD_START, "jmxremote.port=" + pa.getPort2());
testConnect(pa.getPort2());
} finally {
s.stop();
}
Expand All @@ -442,14 +482,16 @@ static void test_02() throws Exception {
System.out.println("**** Test two ****");

Something s = doSomething("test_02");
PortAllocator pa = new PortAllocator();
try {
jcmd(CMD_START,
"jmxremote.port=" + port1,
"jmxremote.port=" + pa.getPort1(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false");

testConnect(port1);
testConnect(pa.getPort1());
} finally {
// debugPortUsage(pa);
s.stop();
}
}
Expand All @@ -461,23 +503,24 @@ static void test_03() throws Exception {
System.out.println("**** Test three ****");

Something s = doSomething("test_03");
PortAllocator pa = new PortAllocator();
try {
jcmd(CMD_START,
"jmxremote.port=" + port1,
"jmxremote.port=" + pa.getPort1(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false");

// Second agent shouldn't start
jcmd(CMD_START,
"jmxremote.port=" + port2,
"jmxremote.port=" + pa.getPort2(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false");

// First agent should connect
testConnect(port1);
testConnect(pa.getPort1());

// Second agent should not connect
testNoConnect(port2);
testNoConnect(pa.getPort2());
} finally {
s.stop();
}
Expand All @@ -490,15 +533,15 @@ static void test_04() throws Exception {
System.out.println("**** Test four ****");

Something s = doSomething("test_04");

PortAllocator pa = new PortAllocator();
try {
jcmd(CMD_START,
"jmxremote.port=" + port1,
"jmxremote.rmi.port=" + port2,
"jmxremote.port=" + pa.getPort1(),
"jmxremote.rmi.port=" + pa.getPort2(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false");

testConnect(port1, port2);
testConnect(pa.getPort1(), pa.getPort2());
} finally {
s.stop();
}
Expand All @@ -511,10 +554,11 @@ static void test_05() throws Exception {
System.out.println("**** Test five ****");

Something s = doSomething("test_05");
PortAllocator pa = new PortAllocator();
try {
jcmd(CMD_START_LOCAL);

testNoConnect(port1);
testNoConnect(pa.getPort1());
testConnectLocal(s.getPid());
} finally {
s.stop();
Expand All @@ -533,14 +577,14 @@ static void test_06() throws Exception {
System.out.println("**** Test six ****");

Something s = doSomething("test_06");

PortAllocator pa = new PortAllocator();
try {
jcmd(CMD_START,
"jmxremote.port=" + port1,
"jmxremote.port=" + pa.getPort1(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false");

testConnect(port1, port2);
testConnect(pa.getPort1(), pa.getPort2());

final boolean[] checks = new boolean[3];
jcmd(
Expand All @@ -550,7 +594,7 @@ static void test_06() throws Exception {
}
},
CMD_START,
"jmxremote.port=" + port1,
"jmxremote.port=" + pa.getPort1(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false");

Expand All @@ -561,7 +605,7 @@ static void test_06() throws Exception {
}
},
CMD_START,
"jmxremote.port=" + port2,
"jmxremote.port=" + pa.getPort2(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false");

Expand All @@ -578,15 +622,15 @@ static void test_06() throws Exception {
},
CMD_START,
"jmxremote.port=" + ss.getLocalPort(),
"jmxremote.rmi.port=" + port2,
"jmxremote.rmi.port=" + pa.getPort2(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false");
if (!checks[0]) {
throw new Exception("Starting agent on port " + port1 + " should " +
throw new Exception("Starting agent on port " + pa.getPort1() + " should " +
"report an invalid agent state");
}
if (!checks[1]) {
throw new Exception("Starting agent on poprt " + port2 + " should " +
throw new Exception("Starting agent on poprt " + pa.getPort2() + " should " +
"report an invalid agent state");
}
if (!checks[2]) {
Expand All @@ -609,16 +653,17 @@ private static void test_07() throws Exception {
"test_07",
"-Dcom.sun.management.jmxremote.authenticate=false",
"-Dcom.sun.management.jmxremote.ssl=true");
PortAllocator pa = new PortAllocator();

try {
testNoConnect(port1);
testNoConnect(pa.getPort1());
jcmd(
CMD_START,
"jmxremote.port=" + port2,
"jmxremote.port=" + pa.getPort2(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false"
);
testConnect(port2);
testConnect(pa.getPort2());
} finally {
s.stop();
}
Expand All @@ -631,28 +676,29 @@ static void test_08() throws Exception {
// make sure these properties overridden corectly

System.out.println("**** Test eight ****");
PortAllocator pa = new PortAllocator();

Something s = doSomething(
"test_08",
"-Dcom.sun.management.jmxremote.port=" + port1,
"-Dcom.sun.management.jmxremote.port=" + pa.getPort1(),
"-Dcom.sun.management.jmxremote.authenticate=false",
"-Dcom.sun.management.jmxremote.ssl=true");

try {
testNoConnect(port1);
testNoConnect(pa.getPort1());

jcmd(CMD_STOP);

testNoConnect(port1);
testNoConnect(pa.getPort1());

jcmd(
CMD_START,
"jmxremote.port=" + port2,
"jmxremote.port=" + pa.getPort2(),
"jmxremote.authenticate=false",
"jmxremote.ssl=false"
);

testConnect(port2);
testConnect(pa.getPort2());
} finally {
s.stop();
}
Expand All @@ -673,22 +719,23 @@ static void test_09() throws Exception {
TEST_SRC + File.separator + "management_cl.properties",
"-Dcom.sun.management.jmxremote.authenticate=false"
);
PortAllocator pa = new PortAllocator();

try {
testNoConnect(port1);
testNoConnect(pa.getPort1());

jcmd(CMD_STOP);

testNoConnect(port1);
testNoConnect(pa.getPort1());

jcmd(CMD_START,
"config.file=" + TEST_SRC + File.separator +
"management_jcmd.properties",
"jmxremote.authenticate=false",
"jmxremote.port=" + port2
"jmxremote.port=" + pa.getPort2()
);

testConnect(port2);
testConnect(pa.getPort2());
} finally {
s.stop();
}
Expand All @@ -702,29 +749,30 @@ static void test_10() throws Exception {
// make sure these properties overridden corectly

System.out.println("**** Test ten ****");
PortAllocator pa = new PortAllocator();

Something s = doSomething(
"test_10",
"-Dcom.sun.management.jmxremote.port=" + port1,
"-Dcom.sun.management.jmxremote.port=" + pa.getPort1(),
"-Dcom.sun.management.jmxremote.authenticate=false",
"-Dcom.sun.management.jmxremote.ssl=true");

try {
testNoConnect(port1);
testNoConnect(pa.getPort1());

jcmd(CMD_STOP);
jcmd(CMD_START,
"jmxremote.ssl=false",
"jmxremote.port=" + port1
"jmxremote.port=" + pa.getPort1()
);
testConnect(port1);

jcmd(CMD_STOP);
jcmd(CMD_START,
"jmxremote.port=" + port1
"jmxremote.port=" + pa.getPort1()
);

testNoConnect(port1);
testNoConnect(pa.getPort1());
} finally {
s.stop();
}
Expand All @@ -736,14 +784,15 @@ static void test_11() throws Exception {
// make sure local agent is not affected

System.out.println("**** Test eleven ****");
PortAllocator pa = new PortAllocator();

Something s = doSomething(
"test_11",
"-Dcom.sun.management.jmxremote.port=" + port1,
"-Dcom.sun.management.jmxremote.port=" + pa.getPort1(),
"-Dcom.sun.management.jmxremote.authenticate=false",
"-Dcom.sun.management.jmxremote.ssl=false");
try {
testConnect(port1);
testConnect(pa.getPort1());
jcmd(CMD_STOP);
testConnectLocal(s.getPid());
} finally {
Expand All @@ -758,9 +807,10 @@ static void test_12() throws Exception {
System.out.println("**** Test twelve ****");

Something s = doSomething("test_12");
PortAllocator pa = new PortAllocator();

try {
testNoConnect(port1);
testNoConnect(pa.getPort1());
jcmd(CMD_START + "_local");

testConnectLocal(s.getPid());
Expand Down

0 comments on commit 98cff0f

Please sign in to comment.