1 /*
   2  * Copyright (c) 2018, 2019, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 import jdk.jpackage.internal.Arguments;
  25 import jdk.jpackage.internal.DeployParams;
  26 import jdk.jpackage.internal.PackagerException;
  27 import java.io.File;
  28 
  29 /*
  30  * @test
  31  * @bug 8211285
  32  * @summary DeployParamsTest
  33  * @modules jdk.jpackage
  34  * @modules jdk.jpackage/jdk.jpackage.internal
  35  * @run main/othervm -Xmx512m DeployParamsTest
  36  */
  37 public class DeployParamsTest {
  38 
  39     private static File testRoot = null;
  40 
  41     private static void setUp() {
  42         testRoot = new File("deployParamsTest");
  43         System.out.println("DeployParamsTest: " + testRoot.getAbsolutePath());
  44         testRoot.mkdir();
  45     }
  46 
  47     private static void tearDown() {
  48         if (testRoot != null) {
  49             testRoot.delete();
  50         }
  51     }
  52 
  53     private static void testValidateAppName1() throws Exception {
  54         DeployParams params = getParamsAppName();
  55 
  56         setAppName(params, "Test");
  57         params.validate();
  58 
  59         setAppName(params, "Test Name");
  60         params.validate();
  61 
  62         setAppName(params, "Test - Name !!!");
  63         params.validate();
  64     }
  65 
  66     private static void testValidateAppName2() throws Exception {
  67         DeployParams params = getParamsAppName();
  68 
  69         setAppName(params, "Test\nName");
  70         appName2TestHelper(params);
  71 
  72         setAppName(params, "Test\rName");
  73         appName2TestHelper(params);
  74 
  75         setAppName(params, "TestName\\");
  76         appName2TestHelper(params);
  77 
  78         setAppName(params, "Test \" Name");
  79         appName2TestHelper(params);
  80     }
  81 
  82     private static void appName2TestHelper(DeployParams params) throws Exception {
  83         try {
  84             params.validate();
  85         } catch (PackagerException pe) {
  86             if (!pe.getMessage().equals("Error: Invalid character found in --name argument")) {
  87                 throw new Exception("Unexpected PackagerException received: " + pe);
  88             }
  89 
  90             return; // Done
  91         }
  92 
  93         throw new Exception("Expecting PackagerException");
  94     }
  95 
  96     // Returns deploy params initialized to pass all validation, except for
  97     // app name
  98     private static DeployParams getParamsAppName() {
  99         DeployParams params = new DeployParams();
 100 
 101         params.setOutput(testRoot);
 102         params.addResource(testRoot, new File(testRoot, "test.jar"));
 103         params.addBundleArgument(Arguments.CLIOptions.APPCLASS.getId(), "TestClass");
 104         params.addBundleArgument(Arguments.CLIOptions.MAIN_JAR.getId(), "test.jar");
 105 
 106         return params;
 107     }
 108 
 109     private static void setAppName(DeployParams params, String appName) {
 110         params.addBundleArgument(Arguments.CLIOptions.NAME.getId(), appName);
 111     }
 112 
 113     public static void main(String[] args) throws Exception {
 114         setUp();
 115 
 116         try {
 117             testValidateAppName1();
 118             testValidateAppName2();
 119         } finally {
 120             tearDown();
 121         }
 122     }
 123 
 124 }