1 /* 2 * Copyright (c) 2015, 2016, 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 /** 25 * Test that shebang handling works properly. 26 * 27 * @test 28 * @option -scripting 29 * @run 30 */ 31 32 // The test generates three different JavaScript source files. The first two 33 // are generated at the beginning of the test and do not change. 34 // * a.js 35 // print("A: " + arguments) 36 // * b.js 37 // #!<path_to_jjs> -lalelu -- ignore 38 // print("B: " + arguments) 39 // 40 // The third file, shebang.js, is generated differently for each particular 41 // test case, containing different shebang lines and one statement: 42 // * shebang.js 43 // #!<path_to_jjs> <shebang_line> 44 // print("S: " + arguments) 45 // 46 // The path_to_jjs is extracted from the environment based on JAVA_HOME, so the 47 // latter must be set properly. 48 // 49 // Each shebang.js is run four times, in all possible combinations of values 50 // from the following two axes: 51 // * without passing any arguments, and passing the arguments 'a.js' and 52 // '"hello world"' (the latter being a quoted string); 53 // * run via jjs, and via direct shell execution (using shebang). 54 55 var pseudosheb = "#!${jjs} -lalelu -- ignore", 56 System = Java.type('java.lang.System'), 57 Paths = Java.type('java.nio.file.Paths'), 58 Files = Java.type('java.nio.file.Files'), 59 Opt = Java.type('java.nio.file.StandardOpenOption'), 60 Arrays = Java.type('java.util.Arrays') 61 62 var sep = Java.type('java.io.File').separator, 63 win = System.getProperty("os.name").startsWith("Windows"), 64 jjsName = "jjs" + (win ? ".exe" : ""), 65 javaHome = System.getProperty("java.home") 66 67 var jjs = javaHome + "/../bin/".replace(/\//g, sep) + jjsName 68 if (!Files.exists(Paths.get(jjs))) { 69 jjs = javaHome + "/bin/".replace(/\//g, sep) + jjsName 70 } 71 72 // Create and cwd to a temporary directory. 73 74 var tmpdir = Files.createTempDirectory(null), 75 tmp = tmpdir.toAbsolutePath().toString(), 76 curpwd = $ENV.PWD 77 78 $ENV.PWD = tmp 79 80 // Test cases. Each case is documented with the expected output for the four 81 // different executions. 82 83 var shebs = [ 84 // No arguments on the shebang line. 85 // noargs jjs/shebang -> no output but "S" prefix 86 // args jjs/shebang -> output the arguments with "S" prefix 87 "", 88 // One interpreter argument. 89 // noargs jjs/shebang -> no output but "S" prefix 90 // args jjs/shebang -> output the arguments with "S" prefix 91 "--language=es6", 92 // Two interpreter arguments. 93 // noargs jjs/shebang -> no output but "S" prefix 94 // args jjs/shebang -> output the arguments with "S" prefix 95 "--language=es6 -scripting", 96 // One interpreter argument and a JavaScript file without shebang. 97 // (For shebang execution, this is a pathological example, as the 98 // JavaScript file passed as a shebang argument will be analyzed and 99 // shebang mode will not be entered.) 100 // noargs jjs -> no output but "S" prefix 101 // args jjs -> output the arguments with "S" prefix 102 // noargs shebang -> no output but "A" and "S" prefixes 103 // args shebang -> output "A", "S", and "A" prefixes, then the error 104 // message: 105 // "java.io.IOException: hello world is not a file" 106 "-scripting a.js", 107 // One interpreter argument and a JavaScript file with shebang. (This 108 // is another pathological example, as it will force shebang mode, 109 // leading to all subsequent arguments, including shebang.js, being 110 // treated as arguments to the script b.js.) 111 // noargs jjs -> no output but the "S" prefix 112 // args jjs -> output the arguments with "S" prefix 113 // noargs shebang -> output shebang.js with "B" prefix 114 // args shebang -> output shebang.js and the arguments with "B" 115 // prefix 116 "-scripting b.js" 117 ] 118 119 function write(file, lines) { 120 Files.write(Paths.get(tmp, file), Arrays.asList(lines), Opt.CREATE, Opt.WRITE) 121 } 122 123 function insn(name) { 124 return "print('${name}:' + arguments)" 125 } 126 127 function run(viajjs, name, arg1, arg2) { 128 var prefix = viajjs ? "${jjs} -scripting " : win ? 'sh -c "' : '', 129 suffix = viajjs ? '' : win ? '"' : '' 130 $EXEC("${prefix}./shebang.js ${arg1} ${arg2}${suffix}") 131 print("* ${name} via ${viajjs ? 'jjs' : 'shebang'}") 132 print($OUT.trim()) 133 print($ERR.trim()) 134 } 135 136 write('a.js', insn('A')) 137 write('b.js', [pseudosheb, insn('B')]) 138 139 shebs.forEach(function(sheb) { 140 var shebang = "#!${jjs} ${sheb}" 141 print("<<< ${sheb} >>>") 142 write('shebang.js', [shebang, insn('S')]) 143 $EXEC('chmod +x shebang.js') 144 run(false, 'noargs', '', '') 145 run(true, 'noargs', '', '') 146 run(false, 'withargs', 'a.js', "'hello world'") 147 run(true, 'withargs', 'a.js', "'hello world'") 148 $EXEC('rm shebang.js') 149 }) 150 151 // Cleanup. 152 153 $EXEC('rm a.js b.js') 154 $ENV.PWD = curpwd 155 Files.delete(tmpdir)