diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/main/java/org/apache/hadoop/yarn/service/utils/ServiceApiUtil.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/main/java/org/apache/hadoop/yarn/service/utils/ServiceApiUtil.java index 3780c99c24f..5e7a335f931 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/main/java/org/apache/hadoop/yarn/service/utils/ServiceApiUtil.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/main/java/org/apache/hadoop/yarn/service/utils/ServiceApiUtil.java @@ -250,7 +250,7 @@ public class ServiceApiUtil { public static void validateJvmOpts(String jvmOpts) throws IllegalArgumentException { - Pattern pattern = Pattern.compile("[!~#?@*&%${}()<>\\[\\]|\"\\/,`;]"); + Pattern pattern = Pattern.compile("[!~#?@*&%${}()<>\\[\\]|\",`;]"); Matcher matcher = pattern.matcher(jvmOpts); if (matcher.find()) { throw new IllegalArgumentException( diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/test/java/org/apache/hadoop/yarn/service/utils/TestServiceApiUtil.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/test/java/org/apache/hadoop/yarn/service/utils/TestServiceApiUtil.java index a93f3d92796..2a611717f2c 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/test/java/org/apache/hadoop/yarn/service/utils/TestServiceApiUtil.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/test/java/org/apache/hadoop/yarn/service/utils/TestServiceApiUtil.java @@ -46,10 +46,12 @@ import java.util.Collections; import java.util.List; import static org.assertj.core.api.Assertions.assertThat; +import static org.apache.hadoop.test.LambdaTestUtils.intercept; import static org.apache.hadoop.yarn.service.conf.RestApiConstants.DEFAULT_UNLIMITED_LIFETIME; import static org.apache.hadoop.yarn.service.exceptions.RestApiErrorMessages.*; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNotNull; +import static org.junit.Assert.fail; /** * Test for ServiceApiUtil helper methods. @@ -766,10 +768,19 @@ public class TestServiceApiUtil extends ServiceTestUtils { Assert.assertTrue(thread.isAlive()); } - @Test(expected = IllegalArgumentException.class) - public void testJvmOpts() { - String jvmOpts = "`ping -c 3 example.com`"; - ServiceApiUtil.validateJvmOpts(jvmOpts); + @Test + public void testJvmOpts() throws Exception { + String invalidJvmOpts = "`ping -c 3 example.com`"; + intercept(IllegalArgumentException.class, + "Invalid character in yarn.service.am.java.opts.", + () -> ServiceApiUtil.validateJvmOpts(invalidJvmOpts)); + String validJvmOpts = "-Dyarn.service.am.java.opts=-Xmx768m " + + "-Djava.security.auth.login.config=/opt/hadoop/etc/jaas-zk.conf"; + try { + ServiceApiUtil.validateJvmOpts(validJvmOpts); + } catch (Exception ex) { + fail("Invalid character in yarn.service.am.java.opts."); + } } public static Service createExampleApplication() {