-
Notifications
You must be signed in to change notification settings - Fork 40
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Showing
6 changed files
with
320 additions
and
22 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
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,125 @@ | ||
| /* | ||
| * Copyright (c) 2023, Azul Systems, Inc. All rights reserved. | ||
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. | ||
| * | ||
| * This code is free software; you can redistribute it and/or modify it | ||
| * under the terms of the GNU General Public License version 2 only, as | ||
| * published by the Free Software Foundation. | ||
| * | ||
| * This code is distributed in the hope that it will be useful, but WITHOUT | ||
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | ||
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | ||
| * version 2 for more details (a copy is included in the LICENSE file that | ||
| * accompanied this code). | ||
| * | ||
| * You should have received a copy of the GNU General Public License version | ||
| * 2 along with this work; if not, write to the Free Software Foundation, | ||
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | ||
| * | ||
| * Please contact Azul Systems, 385 Moffett Park Drive, Suite 115, Sunnyvale | ||
| * CA 94089 USA or visit www.azul.com if you need additional information or | ||
| * have any questions. | ||
| */ | ||
|
|
||
| import jdk.test.lib.containers.docker.DockerTestUtils; | ||
| import jdk.test.lib.crac.CracBuilder; | ||
| import jdk.test.lib.crac.CracProcess; | ||
| import jdk.test.lib.crac.CracTest; | ||
| import jdk.test.lib.crac.CracTestArg; | ||
| import jdk.test.lib.process.OutputAnalyzer; | ||
|
|
||
| import static jdk.test.lib.Asserts.assertEquals; | ||
| import static jdk.test.lib.Asserts.assertLessThan; | ||
|
|
||
| import java.util.Arrays; | ||
|
|
||
| /* | ||
| * @test ContainerPidAdjustmentTest | ||
| * @summary The test checks that process PID is adjusted with the specified value, when checkpointing in a container. Default min PID value is 128. | ||
| * @requires (os.family == "linux") | ||
| * @library /test/lib | ||
| * @build ContainerPidAdjustmentTest | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest true false INF true 128 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest true true 1 false 1 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest true true 100 true 100 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false false INF false 1 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 1 false 1 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 1 true 1 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 200 true 200 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 1000 false 1000 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 2000 true 2000 1000 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 0 true -1 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true -10 true -1 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true blabla true -1 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 5000000 true -1 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 5000000 true -1 4194200 | ||
| * @run driver/timeout=60 jdk.test.lib.crac.CracTest false true 4194303 true -1 | ||
| */ | ||
| public class ContainerPidAdjustmentTest implements CracTest { | ||
| @CracTestArg(0) | ||
| boolean runDirectly; | ||
|
|
||
| @CracTestArg(1) | ||
| boolean needSetMinPid; | ||
|
|
||
| @CracTestArg(2) | ||
| String lastPid; | ||
|
|
||
| @CracTestArg(3) | ||
| boolean usePrivilegedContainer; | ||
|
|
||
| @CracTestArg(4) | ||
| long expectedLastPid; | ||
|
|
||
| @CracTestArg(value = 5, optional = true) | ||
| String lastPidSetup; | ||
|
|
||
| final private String CURRENT_PID_MESSAGE = "Current PID = "; | ||
|
|
||
| @Override | ||
| public void test() throws Exception { | ||
| if (!DockerTestUtils.canTestDocker()) { | ||
| return; | ||
| } | ||
| CracBuilder builder = new CracBuilder() | ||
| .inDockerImage("pid-adjustment") | ||
| .runContainerDirectly(runDirectly) | ||
| .containerUsePrivileged(usePrivilegedContainer); | ||
| if (needSetMinPid) { | ||
| builder.vmOption("-XX:CRaCMinPid=" + lastPid); | ||
| } | ||
| if (0 > expectedLastPid) { | ||
| builder.captureOutput(true); | ||
| } | ||
| if (null != lastPidSetup) { | ||
| // Set up the initial last pid, | ||
| // create a non-privileged user, | ||
| // and force spinning the last pid running checkpoint under the user. | ||
| builder | ||
| .containerSetup(Arrays.asList("bash", "-c", "useradd the_user && echo " + lastPidSetup + " >/proc/sys/kernel/ns_last_pid")) | ||
| .dockerCheckpointOptions(Arrays.asList("-u", "the_user")); | ||
| } | ||
|
|
||
| if (0 < expectedLastPid) { | ||
| builder.startCheckpoint().waitForSuccess(); | ||
| } else { | ||
| final int expectedExitValue = (int)java.lang.Math.abs(expectedLastPid); | ||
| CracProcess process = builder.startCheckpoint(); | ||
| final int exitValue = process.waitFor(); | ||
| assertEquals(expectedExitValue, exitValue, "Process returned unexpected exit code: " + exitValue); | ||
| OutputAnalyzer oa = process.outputAnalyzer(); | ||
| oa.shouldNotContain(CURRENT_PID_MESSAGE); | ||
| if (null != lastPidSetup) { | ||
| oa.shouldContain("spin_last_pid: Invalid argument (" + lastPid + ")"); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void exec() throws Exception { | ||
| System.out.println(CURRENT_PID_MESSAGE + ProcessHandle.current().pid()); | ||
| assertLessThan((long)0, expectedLastPid, "Shouldn't happen"); | ||
| assertLessThan(expectedLastPid, ProcessHandle.current().pid()); | ||
| } | ||
| } |
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
Oops, something went wrong.