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