rt/launcher/src/main/java/org/apidesign/bck2brwsr/launcher/Bck2BrwsrLauncher.java
1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/rt/launcher/src/main/java/org/apidesign/bck2brwsr/launcher/Bck2BrwsrLauncher.java Fri Mar 22 17:03:32 2013 +0100
1.3 @@ -0,0 +1,574 @@
1.4 +/**
1.5 + * Back 2 Browser Bytecode Translator
1.6 + * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
1.7 + *
1.8 + * This program is free software: you can redistribute it and/or modify
1.9 + * it under the terms of the GNU General Public License as published by
1.10 + * the Free Software Foundation, version 2 of the License.
1.11 + *
1.12 + * This program is distributed in the hope that it will be useful,
1.13 + * but WITHOUT ANY WARRANTY; without even the implied warranty of
1.14 + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
1.15 + * GNU General Public License for more details.
1.16 + *
1.17 + * You should have received a copy of the GNU General Public License
1.18 + * along with this program. Look for COPYING file in the top folder.
1.19 + * If not, see http://opensource.org/licenses/GPL-2.0.
1.20 + */
1.21 +package org.apidesign.bck2brwsr.launcher;
1.22 +
1.23 +import java.io.Closeable;
1.24 +import java.io.File;
1.25 +import java.io.IOException;
1.26 +import java.io.InputStream;
1.27 +import java.io.InterruptedIOException;
1.28 +import java.io.OutputStream;
1.29 +import java.io.UnsupportedEncodingException;
1.30 +import java.io.Writer;
1.31 +import java.net.URI;
1.32 +import java.net.URISyntaxException;
1.33 +import java.net.URL;
1.34 +import java.util.ArrayList;
1.35 +import java.util.Arrays;
1.36 +import java.util.Enumeration;
1.37 +import java.util.LinkedHashSet;
1.38 +import java.util.List;
1.39 +import java.util.Set;
1.40 +import java.util.concurrent.BlockingQueue;
1.41 +import java.util.concurrent.CountDownLatch;
1.42 +import java.util.concurrent.LinkedBlockingQueue;
1.43 +import java.util.concurrent.TimeUnit;
1.44 +import java.util.logging.Level;
1.45 +import java.util.logging.Logger;
1.46 +import org.apidesign.bck2brwsr.launcher.InvocationContext.Resource;
1.47 +import org.apidesign.vm4brwsr.Bck2Brwsr;
1.48 +import org.glassfish.grizzly.PortRange;
1.49 +import org.glassfish.grizzly.http.server.HttpHandler;
1.50 +import org.glassfish.grizzly.http.server.HttpServer;
1.51 +import org.glassfish.grizzly.http.server.NetworkListener;
1.52 +import org.glassfish.grizzly.http.server.Request;
1.53 +import org.glassfish.grizzly.http.server.Response;
1.54 +import org.glassfish.grizzly.http.server.ServerConfiguration;
1.55 +import org.glassfish.grizzly.http.util.HttpStatus;
1.56 +
1.57 +/**
1.58 + * Lightweight server to launch Bck2Brwsr applications and tests.
1.59 + * Supports execution in native browser as well as Java's internal
1.60 + * execution engine.
1.61 + */
1.62 +final class Bck2BrwsrLauncher extends Launcher implements Closeable {
1.63 + private static final Logger LOG = Logger.getLogger(Bck2BrwsrLauncher.class.getName());
1.64 + private static final InvocationContext END = new InvocationContext(null, null, null);
1.65 + private final Set<ClassLoader> loaders = new LinkedHashSet<>();
1.66 + private final BlockingQueue<InvocationContext> methods = new LinkedBlockingQueue<>();
1.67 + private long timeOut;
1.68 + private final Res resources = new Res();
1.69 + private final String cmd;
1.70 + private Object[] brwsr;
1.71 + private HttpServer server;
1.72 + private CountDownLatch wait;
1.73 +
1.74 + public Bck2BrwsrLauncher(String cmd) {
1.75 + this.cmd = cmd;
1.76 + }
1.77 +
1.78 + @Override
1.79 + InvocationContext runMethod(InvocationContext c) throws IOException {
1.80 + loaders.add(c.clazz.getClassLoader());
1.81 + methods.add(c);
1.82 + try {
1.83 + c.await(timeOut);
1.84 + } catch (InterruptedException ex) {
1.85 + throw new IOException(ex);
1.86 + }
1.87 + return c;
1.88 + }
1.89 +
1.90 + public void setTimeout(long ms) {
1.91 + timeOut = ms;
1.92 + }
1.93 +
1.94 + public void addClassLoader(ClassLoader url) {
1.95 + this.loaders.add(url);
1.96 + }
1.97 +
1.98 + public void showURL(String startpage) throws IOException {
1.99 + if (!startpage.startsWith("/")) {
1.100 + startpage = "/" + startpage;
1.101 + }
1.102 + HttpServer s = initServer(".", true);
1.103 + int last = startpage.lastIndexOf('/');
1.104 + String simpleName = startpage.substring(last);
1.105 + s.getServerConfiguration().addHttpHandler(new Page(resources, startpage), simpleName);
1.106 + s.getServerConfiguration().addHttpHandler(new Page(resources, null), "/");
1.107 + try {
1.108 + launchServerAndBrwsr(s, simpleName);
1.109 + } catch (URISyntaxException | InterruptedException ex) {
1.110 + throw new IOException(ex);
1.111 + }
1.112 + }
1.113 +
1.114 + void showDirectory(File dir, String startpage) throws IOException {
1.115 + if (!startpage.startsWith("/")) {
1.116 + startpage = "/" + startpage;
1.117 + }
1.118 + HttpServer s = initServer(dir.getPath(), false);
1.119 + try {
1.120 + launchServerAndBrwsr(s, startpage);
1.121 + } catch (URISyntaxException | InterruptedException ex) {
1.122 + throw new IOException(ex);
1.123 + }
1.124 + }
1.125 +
1.126 + @Override
1.127 + public void initialize() throws IOException {
1.128 + try {
1.129 + executeInBrowser();
1.130 + } catch (InterruptedException ex) {
1.131 + final InterruptedIOException iio = new InterruptedIOException(ex.getMessage());
1.132 + iio.initCause(ex);
1.133 + throw iio;
1.134 + } catch (Exception ex) {
1.135 + if (ex instanceof IOException) {
1.136 + throw (IOException)ex;
1.137 + }
1.138 + if (ex instanceof RuntimeException) {
1.139 + throw (RuntimeException)ex;
1.140 + }
1.141 + throw new IOException(ex);
1.142 + }
1.143 + }
1.144 +
1.145 + private HttpServer initServer(String path, boolean addClasses) throws IOException {
1.146 + HttpServer s = HttpServer.createSimpleServer(path, new PortRange(8080, 65535));
1.147 +
1.148 + final ServerConfiguration conf = s.getServerConfiguration();
1.149 + if (addClasses) {
1.150 + conf.addHttpHandler(new VM(resources), "/bck2brwsr.js");
1.151 + conf.addHttpHandler(new Classes(resources), "/classes/");
1.152 + }
1.153 + return s;
1.154 + }
1.155 +
1.156 + private void executeInBrowser() throws InterruptedException, URISyntaxException, IOException {
1.157 + wait = new CountDownLatch(1);
1.158 + server = initServer(".", true);
1.159 + final ServerConfiguration conf = server.getServerConfiguration();
1.160 +
1.161 + class DynamicResourceHandler extends HttpHandler {
1.162 + private final InvocationContext ic;
1.163 + public DynamicResourceHandler(InvocationContext ic) {
1.164 + if (ic == null || ic.resources.isEmpty()) {
1.165 + throw new NullPointerException();
1.166 + }
1.167 + this.ic = ic;
1.168 + for (Resource r : ic.resources) {
1.169 + conf.addHttpHandler(this, r.httpPath);
1.170 + }
1.171 + }
1.172 +
1.173 + public void close() {
1.174 + conf.removeHttpHandler(this);
1.175 + }
1.176 +
1.177 + @Override
1.178 + public void service(Request request, Response response) throws Exception {
1.179 + for (Resource r : ic.resources) {
1.180 + if (r.httpPath.equals(request.getRequestURI())) {
1.181 + LOG.log(Level.INFO, "Serving HttpResource for {0}", request.getRequestURI());
1.182 + response.setContentType(r.httpType);
1.183 + copyStream(r.httpContent, response.getOutputStream(), null);
1.184 + }
1.185 + }
1.186 + }
1.187 + }
1.188 +
1.189 + conf.addHttpHandler(new Page(resources,
1.190 + "org/apidesign/bck2brwsr/launcher/harness.xhtml"
1.191 + ), "/execute");
1.192 +
1.193 + conf.addHttpHandler(new HttpHandler() {
1.194 + int cnt;
1.195 + List<InvocationContext> cases = new ArrayList<>();
1.196 + DynamicResourceHandler prev;
1.197 + @Override
1.198 + public void service(Request request, Response response) throws Exception {
1.199 + String id = request.getParameter("request");
1.200 + String value = request.getParameter("result");
1.201 + if (value != null && value.indexOf((char)0xC5) != -1) {
1.202 + value = toUTF8(value);
1.203 + }
1.204 +
1.205 +
1.206 + InvocationContext mi = null;
1.207 + int caseNmbr = -1;
1.208 +
1.209 + if (id != null && value != null) {
1.210 + LOG.log(Level.INFO, "Received result for case {0} = {1}", new Object[]{id, value});
1.211 + value = decodeURL(value);
1.212 + int indx = Integer.parseInt(id);
1.213 + cases.get(indx).result(value, null);
1.214 + if (++indx < cases.size()) {
1.215 + mi = cases.get(indx);
1.216 + LOG.log(Level.INFO, "Re-executing case {0}", indx);
1.217 + caseNmbr = indx;
1.218 + }
1.219 + } else {
1.220 + if (!cases.isEmpty()) {
1.221 + LOG.info("Re-executing test cases");
1.222 + mi = cases.get(0);
1.223 + caseNmbr = 0;
1.224 + }
1.225 + }
1.226 +
1.227 + if (prev != null) {
1.228 + prev.close();
1.229 + prev = null;
1.230 + }
1.231 +
1.232 + if (mi == null) {
1.233 + mi = methods.take();
1.234 + caseNmbr = cnt++;
1.235 + }
1.236 + if (mi == END) {
1.237 + response.getWriter().write("");
1.238 + wait.countDown();
1.239 + cnt = 0;
1.240 + LOG.log(Level.INFO, "End of data reached. Exiting.");
1.241 + return;
1.242 + }
1.243 +
1.244 + if (!mi.resources.isEmpty()) {
1.245 + prev = new DynamicResourceHandler(mi);
1.246 + }
1.247 +
1.248 + cases.add(mi);
1.249 + final String cn = mi.clazz.getName();
1.250 + final String mn = mi.methodName;
1.251 + LOG.log(Level.INFO, "Request for {0} case. Sending {1}.{2}", new Object[]{caseNmbr, cn, mn});
1.252 + response.getWriter().write("{"
1.253 + + "className: '" + cn + "', "
1.254 + + "methodName: '" + mn + "', "
1.255 + + "request: " + caseNmbr
1.256 + );
1.257 + if (mi.html != null) {
1.258 + response.getWriter().write(", html: '");
1.259 + response.getWriter().write(encodeJSON(mi.html));
1.260 + response.getWriter().write("'");
1.261 + }
1.262 + response.getWriter().write("}");
1.263 + }
1.264 + }, "/data");
1.265 +
1.266 + this.brwsr = launchServerAndBrwsr(server, "/execute");
1.267 + }
1.268 +
1.269 + private static String encodeJSON(String in) {
1.270 + StringBuilder sb = new StringBuilder();
1.271 + for (int i = 0; i < in.length(); i++) {
1.272 + char ch = in.charAt(i);
1.273 + if (ch < 32 || ch == '\'' || ch == '"') {
1.274 + sb.append("\\u");
1.275 + String hs = "0000" + Integer.toHexString(ch);
1.276 + hs = hs.substring(hs.length() - 4);
1.277 + sb.append(hs);
1.278 + } else {
1.279 + sb.append(ch);
1.280 + }
1.281 + }
1.282 + return sb.toString();
1.283 + }
1.284 +
1.285 + @Override
1.286 + public void shutdown() throws IOException {
1.287 + methods.offer(END);
1.288 + for (;;) {
1.289 + int prev = methods.size();
1.290 + try {
1.291 + if (wait != null && wait.await(timeOut, TimeUnit.MILLISECONDS)) {
1.292 + break;
1.293 + }
1.294 + } catch (InterruptedException ex) {
1.295 + throw new IOException(ex);
1.296 + }
1.297 + if (prev == methods.size()) {
1.298 + LOG.log(
1.299 + Level.WARNING,
1.300 + "Timeout and no test has been executed meanwhile (at {0}). Giving up.",
1.301 + methods.size()
1.302 + );
1.303 + break;
1.304 + }
1.305 + LOG.log(Level.INFO,
1.306 + "Timeout, but tests got from {0} to {1}. Trying again.",
1.307 + new Object[]{prev, methods.size()}
1.308 + );
1.309 + }
1.310 + stopServerAndBrwsr(server, brwsr);
1.311 + }
1.312 +
1.313 + static void copyStream(InputStream is, OutputStream os, String baseURL, String... params) throws IOException {
1.314 + for (;;) {
1.315 + int ch = is.read();
1.316 + if (ch == -1) {
1.317 + break;
1.318 + }
1.319 + if (ch == '$' && params.length > 0) {
1.320 + int cnt = is.read() - '0';
1.321 + if (cnt == 'U' - '0') {
1.322 + os.write(baseURL.getBytes("UTF-8"));
1.323 + } else {
1.324 + if (cnt >= 0 && cnt < params.length) {
1.325 + os.write(params[cnt].getBytes("UTF-8"));
1.326 + } else {
1.327 + os.write('$');
1.328 + os.write(cnt + '0');
1.329 + }
1.330 + }
1.331 + } else {
1.332 + os.write(ch);
1.333 + }
1.334 + }
1.335 + }
1.336 +
1.337 + private Object[] launchServerAndBrwsr(HttpServer server, final String page) throws IOException, URISyntaxException, InterruptedException {
1.338 + server.start();
1.339 + NetworkListener listener = server.getListeners().iterator().next();
1.340 + int port = listener.getPort();
1.341 +
1.342 + URI uri = new URI("http://localhost:" + port + page);
1.343 + LOG.log(Level.INFO, "Showing {0}", uri);
1.344 + if (cmd == null) {
1.345 + try {
1.346 + LOG.log(Level.INFO, "Trying Desktop.browse on {0} {2} by {1}", new Object[] {
1.347 + System.getProperty("java.vm.name"),
1.348 + System.getProperty("java.vm.vendor"),
1.349 + System.getProperty("java.vm.version"),
1.350 + });
1.351 + java.awt.Desktop.getDesktop().browse(uri);
1.352 + LOG.log(Level.INFO, "Desktop.browse successfully finished");
1.353 + return null;
1.354 + } catch (UnsupportedOperationException ex) {
1.355 + LOG.log(Level.INFO, "Desktop.browse not supported: {0}", ex.getMessage());
1.356 + LOG.log(Level.FINE, null, ex);
1.357 + }
1.358 + }
1.359 + {
1.360 + String cmdName = cmd == null ? "xdg-open" : cmd;
1.361 + String[] cmdArr = {
1.362 + cmdName, uri.toString()
1.363 + };
1.364 + LOG.log(Level.INFO, "Launching {0}", Arrays.toString(cmdArr));
1.365 + final Process process = Runtime.getRuntime().exec(cmdArr);
1.366 + return new Object[] { process, null };
1.367 + }
1.368 + }
1.369 + private static String toUTF8(String value) throws UnsupportedEncodingException {
1.370 + byte[] arr = new byte[value.length()];
1.371 + for (int i = 0; i < arr.length; i++) {
1.372 + arr[i] = (byte)value.charAt(i);
1.373 + }
1.374 + return new String(arr, "UTF-8");
1.375 + }
1.376 +
1.377 + private static String decodeURL(String s) {
1.378 + for (;;) {
1.379 + int pos = s.indexOf('%');
1.380 + if (pos == -1) {
1.381 + return s;
1.382 + }
1.383 + int i = Integer.parseInt(s.substring(pos + 1, pos + 2), 16);
1.384 + s = s.substring(0, pos) + (char)i + s.substring(pos + 2);
1.385 + }
1.386 + }
1.387 +
1.388 + private void stopServerAndBrwsr(HttpServer server, Object[] brwsr) throws IOException {
1.389 + if (brwsr == null) {
1.390 + return;
1.391 + }
1.392 + Process process = (Process)brwsr[0];
1.393 +
1.394 + server.stop();
1.395 + InputStream stdout = process.getInputStream();
1.396 + InputStream stderr = process.getErrorStream();
1.397 + drain("StdOut", stdout);
1.398 + drain("StdErr", stderr);
1.399 + process.destroy();
1.400 + int res;
1.401 + try {
1.402 + res = process.waitFor();
1.403 + } catch (InterruptedException ex) {
1.404 + throw new IOException(ex);
1.405 + }
1.406 + LOG.log(Level.INFO, "Exit code: {0}", res);
1.407 +
1.408 + deleteTree((File)brwsr[1]);
1.409 + }
1.410 +
1.411 + private static void drain(String name, InputStream is) throws IOException {
1.412 + int av = is.available();
1.413 + if (av > 0) {
1.414 + StringBuilder sb = new StringBuilder();
1.415 + sb.append("v== ").append(name).append(" ==v\n");
1.416 + while (av-- > 0) {
1.417 + sb.append((char)is.read());
1.418 + }
1.419 + sb.append("\n^== ").append(name).append(" ==^");
1.420 + LOG.log(Level.INFO, sb.toString());
1.421 + }
1.422 + }
1.423 +
1.424 + private void deleteTree(File file) {
1.425 + if (file == null) {
1.426 + return;
1.427 + }
1.428 + File[] arr = file.listFiles();
1.429 + if (arr != null) {
1.430 + for (File s : arr) {
1.431 + deleteTree(s);
1.432 + }
1.433 + }
1.434 + file.delete();
1.435 + }
1.436 +
1.437 + @Override
1.438 + public void close() throws IOException {
1.439 + shutdown();
1.440 + }
1.441 +
1.442 + private class Res implements Bck2Brwsr.Resources {
1.443 + @Override
1.444 + public InputStream get(String resource) throws IOException {
1.445 + for (ClassLoader l : loaders) {
1.446 + URL u = null;
1.447 + Enumeration<URL> en = l.getResources(resource);
1.448 + while (en.hasMoreElements()) {
1.449 + u = en.nextElement();
1.450 + }
1.451 + if (u != null) {
1.452 + return u.openStream();
1.453 + }
1.454 + }
1.455 + throw new IOException("Can't find " + resource);
1.456 + }
1.457 + }
1.458 +
1.459 + private static class Page extends HttpHandler {
1.460 + private final String resource;
1.461 + private final String[] args;
1.462 + private final Res res;
1.463 +
1.464 + public Page(Res res, String resource, String... args) {
1.465 + this.res = res;
1.466 + this.resource = resource;
1.467 + this.args = args.length == 0 ? new String[] { "$0" } : args;
1.468 + }
1.469 +
1.470 + @Override
1.471 + public void service(Request request, Response response) throws Exception {
1.472 + String r = resource;
1.473 + if (r == null) {
1.474 + r = request.getHttpHandlerPath();
1.475 + }
1.476 + if (r.startsWith("/")) {
1.477 + r = r.substring(1);
1.478 + }
1.479 + String[] replace = {};
1.480 + if (r.endsWith(".html")) {
1.481 + response.setContentType("text/html");
1.482 + LOG.info("Content type text/html");
1.483 + replace = args;
1.484 + }
1.485 + if (r.endsWith(".xhtml")) {
1.486 + response.setContentType("application/xhtml+xml");
1.487 + LOG.info("Content type application/xhtml+xml");
1.488 + replace = args;
1.489 + }
1.490 + OutputStream os = response.getOutputStream();
1.491 + try (InputStream is = res.get(r)) {
1.492 + copyStream(is, os, request.getRequestURL().toString(), replace);
1.493 + } catch (IOException ex) {
1.494 + response.setDetailMessage(ex.getLocalizedMessage());
1.495 + response.setError();
1.496 + response.setStatus(404);
1.497 + }
1.498 + }
1.499 + }
1.500 +
1.501 + private static class VM extends HttpHandler {
1.502 + private final String bck2brwsr;
1.503 +
1.504 + public VM(Res loader) throws IOException {
1.505 + StringBuilder sb = new StringBuilder();
1.506 + Bck2Brwsr.generate(sb, loader);
1.507 + sb.append(
1.508 + "(function WrapperVM(global) {"
1.509 + + " function ldCls(res) {\n"
1.510 + + " var request = new XMLHttpRequest();\n"
1.511 + + " request.open('GET', '/classes/' + res, false);\n"
1.512 + + " request.send();\n"
1.513 + + " if (request.status !== 200) return null;\n"
1.514 + + " var arr = eval('(' + request.responseText + ')');\n"
1.515 + + " return arr;\n"
1.516 + + " }\n"
1.517 + + " var prevvm = global.bck2brwsr;\n"
1.518 + + " global.bck2brwsr = function() {\n"
1.519 + + " var args = Array.prototype.slice.apply(arguments);\n"
1.520 + + " args.unshift(ldCls);\n"
1.521 + + " return prevvm.apply(null, args);\n"
1.522 + + " };\n"
1.523 + + "})(this);\n"
1.524 + );
1.525 + this.bck2brwsr = sb.toString();
1.526 + }
1.527 +
1.528 + @Override
1.529 + public void service(Request request, Response response) throws Exception {
1.530 + response.setCharacterEncoding("UTF-8");
1.531 + response.setContentType("text/javascript");
1.532 + response.getWriter().write(bck2brwsr);
1.533 + }
1.534 + }
1.535 +
1.536 + private static class Classes extends HttpHandler {
1.537 + private final Res loader;
1.538 +
1.539 + public Classes(Res loader) {
1.540 + this.loader = loader;
1.541 + }
1.542 +
1.543 + @Override
1.544 + public void service(Request request, Response response) throws Exception {
1.545 + String res = request.getHttpHandlerPath();
1.546 + if (res.startsWith("/")) {
1.547 + res = res.substring(1);
1.548 + }
1.549 + try (InputStream is = loader.get(res)) {
1.550 + response.setContentType("text/javascript");
1.551 + Writer w = response.getWriter();
1.552 + w.append("[");
1.553 + for (int i = 0;; i++) {
1.554 + int b = is.read();
1.555 + if (b == -1) {
1.556 + break;
1.557 + }
1.558 + if (i > 0) {
1.559 + w.append(", ");
1.560 + }
1.561 + if (i % 20 == 0) {
1.562 + w.write("\n");
1.563 + }
1.564 + if (b > 127) {
1.565 + b = b - 256;
1.566 + }
1.567 + w.append(Integer.toString(b));
1.568 + }
1.569 + w.append("\n]");
1.570 + } catch (IOException ex) {
1.571 + response.setStatus(HttpStatus.NOT_FOUND_404);
1.572 + response.setError();
1.573 + response.setDetailMessage(ex.getMessage());
1.574 + }
1.575 + }
1.576 + }
1.577 +}