launcher/fx/src/main/java/org/apidesign/bck2brwsr/launcher/BaseHTTPLauncher.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Sat, 03 May 2014 20:41:35 +0200
branchclosure
changeset 1521 6c709f05afa5
parent 1513 ba912ef24b27
child 1526 4d8e295a2c2e
permissions -rw-r--r--
Log I/O exceptions before throwing them into Grizzly
jaroslav@1033
     1
/**
jaroslav@1033
     2
 * Back 2 Browser Bytecode Translator
jaroslav@1033
     3
 * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
jaroslav@1033
     4
 *
jaroslav@1033
     5
 * This program is free software: you can redistribute it and/or modify
jaroslav@1033
     6
 * it under the terms of the GNU General Public License as published by
jaroslav@1033
     7
 * the Free Software Foundation, version 2 of the License.
jaroslav@1033
     8
 *
jaroslav@1033
     9
 * This program is distributed in the hope that it will be useful,
jaroslav@1033
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
jaroslav@1033
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
jaroslav@1033
    12
 * GNU General Public License for more details.
jaroslav@1033
    13
 *
jaroslav@1033
    14
 * You should have received a copy of the GNU General Public License
jaroslav@1033
    15
 * along with this program. Look for COPYING file in the top folder.
jaroslav@1033
    16
 * If not, see http://opensource.org/licenses/GPL-2.0.
jaroslav@1033
    17
 */
jaroslav@1033
    18
package org.apidesign.bck2brwsr.launcher;
jaroslav@1033
    19
jaroslav@1188
    20
import java.io.ByteArrayInputStream;
jaroslav@1249
    21
import java.io.ByteArrayOutputStream;
jaroslav@1033
    22
import java.io.Closeable;
jaroslav@1033
    23
import java.io.File;
jaroslav@1033
    24
import java.io.IOException;
jaroslav@1033
    25
import java.io.InputStream;
jaroslav@1033
    26
import java.io.InterruptedIOException;
jaroslav@1033
    27
import java.io.OutputStream;
jaroslav@1088
    28
import java.io.Reader;
jaroslav@1033
    29
import java.io.UnsupportedEncodingException;
jaroslav@1033
    30
import java.io.Writer;
jaroslav@1489
    31
import java.net.JarURLConnection;
jaroslav@1033
    32
import java.net.URI;
jaroslav@1033
    33
import java.net.URISyntaxException;
jaroslav@1033
    34
import java.net.URL;
jaroslav@1033
    35
import java.util.ArrayList;
jaroslav@1033
    36
import java.util.Arrays;
jaroslav@1033
    37
import java.util.Enumeration;
jaroslav@1033
    38
import java.util.LinkedHashSet;
jaroslav@1033
    39
import java.util.List;
jaroslav@1033
    40
import java.util.Set;
jaroslav@1033
    41
import java.util.concurrent.BlockingQueue;
jaroslav@1076
    42
import java.util.concurrent.Callable;
jaroslav@1033
    43
import java.util.concurrent.CountDownLatch;
jaroslav@1033
    44
import java.util.concurrent.LinkedBlockingQueue;
jaroslav@1033
    45
import java.util.concurrent.TimeUnit;
jaroslav@1489
    46
import java.util.jar.JarFile;
jaroslav@1033
    47
import java.util.logging.Level;
jaroslav@1033
    48
import java.util.logging.Logger;
jaroslav@1033
    49
import org.apidesign.bck2brwsr.launcher.InvocationContext.Resource;
jaroslav@1033
    50
import org.glassfish.grizzly.PortRange;
jaroslav@1033
    51
import org.glassfish.grizzly.http.server.HttpHandler;
jaroslav@1033
    52
import org.glassfish.grizzly.http.server.HttpServer;
jaroslav@1033
    53
import org.glassfish.grizzly.http.server.NetworkListener;
jaroslav@1033
    54
import org.glassfish.grizzly.http.server.Request;
jaroslav@1033
    55
import org.glassfish.grizzly.http.server.Response;
jaroslav@1033
    56
import org.glassfish.grizzly.http.server.ServerConfiguration;
jaroslav@1271
    57
import org.glassfish.grizzly.http.server.StaticHttpHandler;
jaroslav@1033
    58
import org.glassfish.grizzly.http.util.HttpStatus;
jaroslav@1092
    59
import org.glassfish.grizzly.threadpool.ThreadPoolConfig;
jaroslav@1249
    60
import org.glassfish.grizzly.websockets.WebSocket;
jaroslav@1249
    61
import org.glassfish.grizzly.websockets.WebSocketAddOn;
jaroslav@1249
    62
import org.glassfish.grizzly.websockets.WebSocketApplication;
jaroslav@1249
    63
import org.glassfish.grizzly.websockets.WebSocketEngine;
jaroslav@1033
    64
jaroslav@1033
    65
/**
jaroslav@1033
    66
 * Lightweight server to launch Bck2Brwsr applications and tests.
jaroslav@1033
    67
 * Supports execution in native browser as well as Java's internal 
jaroslav@1033
    68
 * execution engine.
jaroslav@1033
    69
 */
jaroslav@1076
    70
abstract class BaseHTTPLauncher extends Launcher implements Closeable, Callable<HttpServer> {
jaroslav@1088
    71
    static final Logger LOG = Logger.getLogger(BaseHTTPLauncher.class.getName());
jaroslav@1033
    72
    private static final InvocationContext END = new InvocationContext(null, null, null);
jaroslav@1165
    73
    private final Set<ClassLoader> loaders = new LinkedHashSet<ClassLoader>();
jaroslav@1165
    74
    private final BlockingQueue<InvocationContext> methods = new LinkedBlockingQueue<InvocationContext>();
jaroslav@1033
    75
    private long timeOut;
jaroslav@1033
    76
    private final Res resources = new Res();
jaroslav@1033
    77
    private final String cmd;
jaroslav@1033
    78
    private Object[] brwsr;
jaroslav@1033
    79
    private HttpServer server;
jaroslav@1033
    80
    private CountDownLatch wait;
jaroslav@1033
    81
    
jaroslav@1041
    82
    public BaseHTTPLauncher(String cmd) {
jaroslav@1033
    83
        this.cmd = cmd;
jaroslav@1043
    84
        addClassLoader(BaseHTTPLauncher.class.getClassLoader());
jaroslav@1033
    85
        setTimeout(180000);
jaroslav@1033
    86
    }
jaroslav@1033
    87
    
jaroslav@1033
    88
    @Override
jaroslav@1033
    89
    InvocationContext runMethod(InvocationContext c) throws IOException {
jaroslav@1033
    90
        loaders.add(c.clazz.getClassLoader());
jaroslav@1033
    91
        methods.add(c);
jaroslav@1033
    92
        try {
jaroslav@1033
    93
            c.await(timeOut);
jaroslav@1033
    94
        } catch (InterruptedException ex) {
jaroslav@1033
    95
            throw new IOException(ex);
jaroslav@1033
    96
        }
jaroslav@1033
    97
        return c;
jaroslav@1033
    98
    }
jaroslav@1033
    99
    
jaroslav@1033
   100
    public void setTimeout(long ms) {
jaroslav@1033
   101
        timeOut = ms;
jaroslav@1033
   102
    }
jaroslav@1033
   103
    
jaroslav@1033
   104
    public void addClassLoader(ClassLoader url) {
jaroslav@1033
   105
        this.loaders.add(url);
jaroslav@1033
   106
    }
jaroslav@1041
   107
    
jaroslav@1041
   108
    ClassLoader[] loaders() {
jaroslav@1041
   109
        return loaders.toArray(new ClassLoader[loaders.size()]);
jaroslav@1041
   110
    }
jaroslav@1033
   111
jaroslav@1033
   112
    public void showURL(String startpage) throws IOException {
jaroslav@1033
   113
        if (!startpage.startsWith("/")) {
jaroslav@1033
   114
            startpage = "/" + startpage;
jaroslav@1033
   115
        }
jaroslav@1271
   116
        HttpServer s = initServer(".", true, "");
jaroslav@1033
   117
        int last = startpage.lastIndexOf('/');
jaroslav@1033
   118
        String prefix = startpage.substring(0, last);
jaroslav@1033
   119
        String simpleName = startpage.substring(last);
jaroslav@1033
   120
        s.getServerConfiguration().addHttpHandler(new SubTree(resources, prefix), "/");
jaroslav@1076
   121
        server = s;
jaroslav@1033
   122
        try {
jaroslav@1033
   123
            launchServerAndBrwsr(s, simpleName);
jaroslav@1165
   124
        } catch (Exception ex) {
jaroslav@1033
   125
            throw new IOException(ex);
jaroslav@1033
   126
        }
jaroslav@1033
   127
    }
jaroslav@1033
   128
jaroslav@1273
   129
    void showDirectory(File dir, String startpage, boolean addClasses) throws IOException {
jaroslav@1033
   130
        if (!startpage.startsWith("/")) {
jaroslav@1033
   131
            startpage = "/" + startpage;
jaroslav@1033
   132
        }
jaroslav@1271
   133
        String prefix = "";
jaroslav@1271
   134
        int last = startpage.lastIndexOf('/');
jaroslav@1271
   135
        if (last >= 0) {
jaroslav@1271
   136
            prefix = startpage.substring(0, last);
jaroslav@1271
   137
        }
jaroslav@1273
   138
        HttpServer s = initServer(dir.getPath(), addClasses, prefix);
jaroslav@1033
   139
        try {
jaroslav@1033
   140
            launchServerAndBrwsr(s, startpage);
jaroslav@1165
   141
        } catch (Exception ex) {
jaroslav@1033
   142
            throw new IOException(ex);
jaroslav@1033
   143
        }
jaroslav@1033
   144
    }
jaroslav@1033
   145
jaroslav@1033
   146
    @Override
jaroslav@1033
   147
    public void initialize() throws IOException {
jaroslav@1033
   148
        try {
jaroslav@1033
   149
            executeInBrowser();
jaroslav@1033
   150
        } catch (InterruptedException ex) {
jaroslav@1033
   151
            final InterruptedIOException iio = new InterruptedIOException(ex.getMessage());
jaroslav@1033
   152
            iio.initCause(ex);
jaroslav@1033
   153
            throw iio;
jaroslav@1033
   154
        } catch (Exception ex) {
jaroslav@1033
   155
            if (ex instanceof IOException) {
jaroslav@1033
   156
                throw (IOException)ex;
jaroslav@1033
   157
            }
jaroslav@1033
   158
            if (ex instanceof RuntimeException) {
jaroslav@1033
   159
                throw (RuntimeException)ex;
jaroslav@1033
   160
            }
jaroslav@1033
   161
            throw new IOException(ex);
jaroslav@1033
   162
        }
jaroslav@1033
   163
    }
jaroslav@1033
   164
    
jaroslav@1271
   165
    private HttpServer initServer(String path, boolean addClasses, String vmPrefix) throws IOException {
jaroslav@1271
   166
        HttpServer s = HttpServer.createSimpleServer(null, new PortRange(8080, 65535));
jaroslav@1253
   167
        /*
jaroslav@1092
   168
        ThreadPoolConfig fewThreads = ThreadPoolConfig.defaultConfig().copy().
jaroslav@1092
   169
            setPoolName("Fx/Bck2 Brwsr").
jaroslav@1253
   170
            setCorePoolSize(3).
jaroslav@1092
   171
            setMaxPoolSize(5);
jaroslav@1092
   172
        ThreadPoolConfig oneKernel = ThreadPoolConfig.defaultConfig().copy().
jaroslav@1092
   173
            setPoolName("Kernel Fx/Bck2").
jaroslav@1253
   174
            setCorePoolSize(3).
jaroslav@1092
   175
            setMaxPoolSize(3);
jaroslav@1092
   176
        for (NetworkListener nl : s.getListeners()) {
jaroslav@1092
   177
            nl.getTransport().setWorkerThreadPoolConfig(fewThreads);
jaroslav@1092
   178
            nl.getTransport().setKernelThreadPoolConfig(oneKernel);
jaroslav@1092
   179
        }
jaroslav@1500
   180
*/        
jaroslav@1033
   181
        final ServerConfiguration conf = s.getServerConfiguration();
jaroslav@1271
   182
        VMAndPages vm = new VMAndPages();
jaroslav@1271
   183
        conf.addHttpHandler(vm, "/");
jaroslav@1271
   184
        if (vmPrefix != null) {
jaroslav@1271
   185
            vm.registerVM(vmPrefix + "/bck2brwsr.js");
jaroslav@1271
   186
        }
jaroslav@1271
   187
        if (path != null) {
jaroslav@1271
   188
            vm.addDocRoot(path);
jaroslav@1271
   189
        }
jaroslav@1033
   190
        if (addClasses) {
jaroslav@1033
   191
            conf.addHttpHandler(new Classes(resources), "/classes/");
jaroslav@1033
   192
        }
jaroslav@1253
   193
        final WebSocketAddOn addon = new WebSocketAddOn();
jaroslav@1253
   194
        for (NetworkListener listener : s.getListeners()) {
jaroslav@1253
   195
            listener.registerAddOn(addon);
jaroslav@1253
   196
        }
jaroslav@1033
   197
        return s;
jaroslav@1033
   198
    }
jaroslav@1033
   199
    
jaroslav@1033
   200
    private void executeInBrowser() throws InterruptedException, URISyntaxException, IOException {
jaroslav@1033
   201
        wait = new CountDownLatch(1);
jaroslav@1271
   202
        server = initServer(".", true, "");
jaroslav@1033
   203
        final ServerConfiguration conf = server.getServerConfiguration();
jaroslav@1033
   204
        
jaroslav@1188
   205
        class DynamicResourceHandler extends HttpHandler {
jaroslav@1033
   206
            private final InvocationContext ic;
jaroslav@1188
   207
            private int resourcesCount;
jaroslav@1235
   208
            DynamicResourceHandler delegate;
jaroslav@1033
   209
            public DynamicResourceHandler(InvocationContext ic) {
jaroslav@1033
   210
                this.ic = ic;
jaroslav@1033
   211
                for (Resource r : ic.resources) {
jaroslav@1033
   212
                    conf.addHttpHandler(this, r.httpPath);
jaroslav@1033
   213
                }
jaroslav@1033
   214
            }
jaroslav@1033
   215
jaroslav@1235
   216
            public void close(DynamicResourceHandler del) {
jaroslav@1033
   217
                conf.removeHttpHandler(this);
jaroslav@1235
   218
                delegate = del;
jaroslav@1033
   219
            }
jaroslav@1033
   220
            
jaroslav@1033
   221
            @Override
jaroslav@1033
   222
            public void service(Request request, Response response) throws Exception {
jaroslav@1235
   223
                if (delegate != null) {
jaroslav@1235
   224
                    delegate.service(request, response);
jaroslav@1235
   225
                    return;
jaroslav@1235
   226
                }
jaroslav@1235
   227
                
jaroslav@1188
   228
                if ("/dynamic".equals(request.getRequestURI())) {
jaroslav@1249
   229
                    boolean webSocket = false;
jaroslav@1188
   230
                    String mimeType = request.getParameter("mimeType");
jaroslav@1188
   231
                    List<String> params = new ArrayList<String>();
jaroslav@1188
   232
                    for (int i = 0; ; i++) {
jaroslav@1188
   233
                        String p = request.getParameter("param" + i);
jaroslav@1188
   234
                        if (p == null) {
jaroslav@1188
   235
                            break;
jaroslav@1188
   236
                        }
jaroslav@1188
   237
                        params.add(p);
jaroslav@1249
   238
                        if ("protocol:ws".equals(p)) {
jaroslav@1249
   239
                            webSocket = true;
jaroslav@1249
   240
                            continue;
jaroslav@1249
   241
                        }                    }
jaroslav@1188
   242
                    final String cnt = request.getParameter("content");
jaroslav@1188
   243
                    String mangle = cnt.replace("%20", " ").replace("%0A", "\n");
jaroslav@1188
   244
                    ByteArrayInputStream is = new ByteArrayInputStream(mangle.getBytes("UTF-8"));
jaroslav@1249
   245
                    URI url;
jaroslav@1249
   246
                    final Resource res = new Resource(is, mimeType, "/dynamic/res" + ++resourcesCount, params.toArray(new String[params.size()]));
jaroslav@1249
   247
                    if (webSocket) {
jaroslav@1249
   248
                        url = registerWebSocket(res);
jaroslav@1249
   249
                    } else {
jaroslav@1249
   250
                        url = registerResource(res);
jaroslav@1249
   251
                    }
jaroslav@1188
   252
                    response.getWriter().write(url.toString());
jaroslav@1188
   253
                    response.getWriter().write("\n");
jaroslav@1188
   254
                    return;
jaroslav@1188
   255
                }
jaroslav@1188
   256
                
jaroslav@1033
   257
                for (Resource r : ic.resources) {
jaroslav@1033
   258
                    if (r.httpPath.equals(request.getRequestURI())) {
jaroslav@1033
   259
                        LOG.log(Level.INFO, "Serving HttpResource for {0}", request.getRequestURI());
jaroslav@1033
   260
                        response.setContentType(r.httpType);
jaroslav@1033
   261
                        r.httpContent.reset();
jaroslav@1033
   262
                        String[] params = null;
jaroslav@1033
   263
                        if (r.parameters.length != 0) {
jaroslav@1033
   264
                            params = new String[r.parameters.length];
jaroslav@1033
   265
                            for (int i = 0; i < r.parameters.length; i++) {
jaroslav@1033
   266
                                params[i] = request.getParameter(r.parameters[i]);
jaroslav@1088
   267
                                if (params[i] == null) {
jaroslav@1088
   268
                                    if ("http.method".equals(r.parameters[i])) {
jaroslav@1088
   269
                                        params[i] = request.getMethod().toString();
jaroslav@1088
   270
                                    } else if ("http.requestBody".equals(r.parameters[i])) {
jaroslav@1088
   271
                                        Reader rdr = request.getReader();
jaroslav@1088
   272
                                        StringBuilder sb = new StringBuilder();
jaroslav@1088
   273
                                        for (;;) {
jaroslav@1088
   274
                                            int ch = rdr.read();
jaroslav@1088
   275
                                            if (ch == -1) {
jaroslav@1088
   276
                                                break;
jaroslav@1088
   277
                                            }
jaroslav@1088
   278
                                            sb.append((char)ch);
jaroslav@1088
   279
                                        }
jaroslav@1088
   280
                                        params[i] = sb.toString();
jaroslav@1088
   281
                                    }
jaroslav@1088
   282
                                }
jaroslav@1088
   283
                                if (params[i] == null) {
jaroslav@1088
   284
                                    params[i] = "null";
jaroslav@1088
   285
                                }
jaroslav@1033
   286
                            }
jaroslav@1033
   287
                        }
jaroslav@1033
   288
                        
jaroslav@1033
   289
                        copyStream(r.httpContent, response.getOutputStream(), null, params);
jaroslav@1033
   290
                    }
jaroslav@1033
   291
                }
jaroslav@1033
   292
            }
jaroslav@1249
   293
            
jaroslav@1249
   294
            private URI registerWebSocket(Resource r) {
jaroslav@1249
   295
                WebSocketEngine.getEngine().register("", r.httpPath, new WS(r));
jaroslav@1249
   296
                return pageURL("ws", server, r.httpPath);
jaroslav@1249
   297
            }
jaroslav@1186
   298
jaroslav@1188
   299
            private URI registerResource(Resource r) {
jaroslav@1186
   300
                if (!ic.resources.contains(r)) {
jaroslav@1186
   301
                    ic.resources.add(r);
jaroslav@1186
   302
                    conf.addHttpHandler(this, r.httpPath);
jaroslav@1186
   303
                }
jaroslav@1249
   304
                return pageURL("http", server, r.httpPath);
jaroslav@1186
   305
            }
jaroslav@1033
   306
        }
jaroslav@1033
   307
        
jaroslav@1088
   308
        conf.addHttpHandler(new Page(resources, harnessResource()), "/execute");
jaroslav@1033
   309
        
jaroslav@1033
   310
        conf.addHttpHandler(new HttpHandler() {
jaroslav@1033
   311
            int cnt;
jaroslav@1165
   312
            List<InvocationContext> cases = new ArrayList<InvocationContext>();
jaroslav@1033
   313
            DynamicResourceHandler prev;
jaroslav@1033
   314
            @Override
jaroslav@1033
   315
            public void service(Request request, Response response) throws Exception {
jaroslav@1033
   316
                String id = request.getParameter("request");
jaroslav@1033
   317
                String value = request.getParameter("result");
jaroslav@1033
   318
                if (value != null && value.indexOf((char)0xC5) != -1) {
jaroslav@1033
   319
                    value = toUTF8(value);
jaroslav@1033
   320
                }
jaroslav@1033
   321
                
jaroslav@1033
   322
                
jaroslav@1033
   323
                InvocationContext mi = null;
jaroslav@1033
   324
                int caseNmbr = -1;
jaroslav@1033
   325
                
jaroslav@1033
   326
                if (id != null && value != null) {
jaroslav@1033
   327
                    LOG.log(Level.INFO, "Received result for case {0} = {1}", new Object[]{id, value});
jaroslav@1033
   328
                    value = decodeURL(value);
jaroslav@1033
   329
                    int indx = Integer.parseInt(id);
jaroslav@1033
   330
                    cases.get(indx).result(value, null);
jaroslav@1033
   331
                    if (++indx < cases.size()) {
jaroslav@1033
   332
                        mi = cases.get(indx);
jaroslav@1033
   333
                        LOG.log(Level.INFO, "Re-executing case {0}", indx);
jaroslav@1033
   334
                        caseNmbr = indx;
jaroslav@1033
   335
                    }
jaroslav@1033
   336
                } else {
jaroslav@1033
   337
                    if (!cases.isEmpty()) {
jaroslav@1033
   338
                        LOG.info("Re-executing test cases");
jaroslav@1033
   339
                        mi = cases.get(0);
jaroslav@1033
   340
                        caseNmbr = 0;
jaroslav@1033
   341
                    }
jaroslav@1033
   342
                }
jaroslav@1033
   343
                
jaroslav@1033
   344
                if (mi == null) {
jaroslav@1033
   345
                    mi = methods.take();
jaroslav@1033
   346
                    caseNmbr = cnt++;
jaroslav@1033
   347
                }
jaroslav@1033
   348
                if (mi == END) {
jaroslav@1033
   349
                    response.getWriter().write("");
jaroslav@1033
   350
                    wait.countDown();
jaroslav@1033
   351
                    cnt = 0;
jaroslav@1033
   352
                    LOG.log(Level.INFO, "End of data reached. Exiting.");
jaroslav@1033
   353
                    return;
jaroslav@1033
   354
                }
jaroslav@1235
   355
                final DynamicResourceHandler newRH = new DynamicResourceHandler(mi);
jaroslav@1235
   356
                if (prev != null) {
jaroslav@1235
   357
                    prev.close(newRH);
jaroslav@1235
   358
                }
jaroslav@1235
   359
                prev = newRH;
jaroslav@1188
   360
                conf.addHttpHandler(prev, "/dynamic");
jaroslav@1033
   361
                
jaroslav@1033
   362
                cases.add(mi);
jaroslav@1033
   363
                final String cn = mi.clazz.getName();
jaroslav@1033
   364
                final String mn = mi.methodName;
jaroslav@1033
   365
                LOG.log(Level.INFO, "Request for {0} case. Sending {1}.{2}", new Object[]{caseNmbr, cn, mn});
jaroslav@1033
   366
                response.getWriter().write("{"
jaroslav@1033
   367
                    + "className: '" + cn + "', "
jaroslav@1033
   368
                    + "methodName: '" + mn + "', "
jaroslav@1033
   369
                    + "request: " + caseNmbr
jaroslav@1033
   370
                );
jaroslav@1033
   371
                if (mi.html != null) {
jaroslav@1033
   372
                    response.getWriter().write(", html: '");
jaroslav@1033
   373
                    response.getWriter().write(encodeJSON(mi.html));
jaroslav@1033
   374
                    response.getWriter().write("'");
jaroslav@1033
   375
                }
jaroslav@1033
   376
                response.getWriter().write("}");
jaroslav@1033
   377
            }
jaroslav@1033
   378
        }, "/data");
jaroslav@1033
   379
jaroslav@1033
   380
        this.brwsr = launchServerAndBrwsr(server, "/execute");
jaroslav@1033
   381
    }
jaroslav@1033
   382
    
jaroslav@1033
   383
    private static String encodeJSON(String in) {
jaroslav@1033
   384
        StringBuilder sb = new StringBuilder();
jaroslav@1033
   385
        for (int i = 0; i < in.length(); i++) {
jaroslav@1033
   386
            char ch = in.charAt(i);
jaroslav@1033
   387
            if (ch < 32 || ch == '\'' || ch == '"') {
jaroslav@1033
   388
                sb.append("\\u");
jaroslav@1033
   389
                String hs = "0000" + Integer.toHexString(ch);
jaroslav@1033
   390
                hs = hs.substring(hs.length() - 4);
jaroslav@1033
   391
                sb.append(hs);
jaroslav@1033
   392
            } else {
jaroslav@1033
   393
                sb.append(ch);
jaroslav@1033
   394
            }
jaroslav@1033
   395
        }
jaroslav@1033
   396
        return sb.toString();
jaroslav@1033
   397
    }
jaroslav@1033
   398
    
jaroslav@1033
   399
    @Override
jaroslav@1033
   400
    public void shutdown() throws IOException {
jaroslav@1033
   401
        methods.offer(END);
jaroslav@1033
   402
        for (;;) {
jaroslav@1033
   403
            int prev = methods.size();
jaroslav@1033
   404
            try {
jaroslav@1033
   405
                if (wait != null && wait.await(timeOut, TimeUnit.MILLISECONDS)) {
jaroslav@1033
   406
                    break;
jaroslav@1033
   407
                }
jaroslav@1033
   408
            } catch (InterruptedException ex) {
jaroslav@1033
   409
                throw new IOException(ex);
jaroslav@1033
   410
            }
jaroslav@1033
   411
            if (prev == methods.size()) {
jaroslav@1033
   412
                LOG.log(
jaroslav@1033
   413
                    Level.WARNING, 
jaroslav@1033
   414
                    "Timeout and no test has been executed meanwhile (at {0}). Giving up.", 
jaroslav@1033
   415
                    methods.size()
jaroslav@1033
   416
                );
jaroslav@1033
   417
                break;
jaroslav@1033
   418
            }
jaroslav@1033
   419
            LOG.log(Level.INFO, 
jaroslav@1033
   420
                "Timeout, but tests got from {0} to {1}. Trying again.", 
jaroslav@1033
   421
                new Object[]{prev, methods.size()}
jaroslav@1033
   422
            );
jaroslav@1033
   423
        }
jaroslav@1033
   424
        stopServerAndBrwsr(server, brwsr);
jaroslav@1033
   425
    }
jaroslav@1033
   426
    
jaroslav@1033
   427
    static void copyStream(InputStream is, OutputStream os, String baseURL, String... params) throws IOException {
jaroslav@1033
   428
        for (;;) {
jaroslav@1033
   429
            int ch = is.read();
jaroslav@1033
   430
            if (ch == -1) {
jaroslav@1033
   431
                break;
jaroslav@1033
   432
            }
jaroslav@1033
   433
            if (ch == '$' && params.length > 0) {
jaroslav@1033
   434
                int cnt = is.read() - '0';
jaroslav@1033
   435
                if (baseURL != null && cnt == 'U' - '0') {
jaroslav@1033
   436
                    os.write(baseURL.getBytes("UTF-8"));
jaroslav@1033
   437
                } else {
jaroslav@1033
   438
                    if (cnt >= 0 && cnt < params.length) {
jaroslav@1033
   439
                        os.write(params[cnt].getBytes("UTF-8"));
jaroslav@1033
   440
                    } else {
jaroslav@1033
   441
                        os.write('$');
jaroslav@1033
   442
                        os.write(cnt + '0');
jaroslav@1033
   443
                    }
jaroslav@1033
   444
                }
jaroslav@1033
   445
            } else {
jaroslav@1033
   446
                os.write(ch);
jaroslav@1033
   447
            }
jaroslav@1033
   448
        }
jaroslav@1033
   449
    }
jaroslav@1033
   450
jaroslav@1033
   451
    private Object[] launchServerAndBrwsr(HttpServer server, final String page) throws IOException, URISyntaxException, InterruptedException {
jaroslav@1033
   452
        server.start();
jaroslav@1249
   453
        URI uri = pageURL("http", server, page);
jaroslav@1041
   454
        return showBrwsr(uri);
jaroslav@1033
   455
    }
jaroslav@1033
   456
    private static String toUTF8(String value) throws UnsupportedEncodingException {
jaroslav@1033
   457
        byte[] arr = new byte[value.length()];
jaroslav@1033
   458
        for (int i = 0; i < arr.length; i++) {
jaroslav@1033
   459
            arr[i] = (byte)value.charAt(i);
jaroslav@1033
   460
        }
jaroslav@1033
   461
        return new String(arr, "UTF-8");
jaroslav@1033
   462
    }
jaroslav@1033
   463
    
jaroslav@1033
   464
    private static String decodeURL(String s) {
jaroslav@1033
   465
        for (;;) {
jaroslav@1033
   466
            int pos = s.indexOf('%');
jaroslav@1033
   467
            if (pos == -1) {
jaroslav@1033
   468
                return s;
jaroslav@1033
   469
            }
jaroslav@1033
   470
            int i = Integer.parseInt(s.substring(pos + 1, pos + 2), 16);
jaroslav@1033
   471
            s = s.substring(0, pos) + (char)i + s.substring(pos + 2);
jaroslav@1033
   472
        }
jaroslav@1033
   473
    }
jaroslav@1033
   474
    
jaroslav@1033
   475
    private void stopServerAndBrwsr(HttpServer server, Object[] brwsr) throws IOException {
jaroslav@1033
   476
        if (brwsr == null) {
jaroslav@1033
   477
            return;
jaroslav@1033
   478
        }
jaroslav@1033
   479
        Process process = (Process)brwsr[0];
jaroslav@1033
   480
        
jaroslav@1033
   481
        server.stop();
jaroslav@1033
   482
        InputStream stdout = process.getInputStream();
jaroslav@1033
   483
        InputStream stderr = process.getErrorStream();
jaroslav@1033
   484
        drain("StdOut", stdout);
jaroslav@1033
   485
        drain("StdErr", stderr);
jaroslav@1033
   486
        process.destroy();
jaroslav@1033
   487
        int res;
jaroslav@1033
   488
        try {
jaroslav@1033
   489
            res = process.waitFor();
jaroslav@1033
   490
        } catch (InterruptedException ex) {
jaroslav@1033
   491
            throw new IOException(ex);
jaroslav@1033
   492
        }
jaroslav@1033
   493
        LOG.log(Level.INFO, "Exit code: {0}", res);
jaroslav@1033
   494
jaroslav@1033
   495
        deleteTree((File)brwsr[1]);
jaroslav@1033
   496
    }
jaroslav@1033
   497
    
jaroslav@1033
   498
    private static void drain(String name, InputStream is) throws IOException {
jaroslav@1033
   499
        int av = is.available();
jaroslav@1033
   500
        if (av > 0) {
jaroslav@1033
   501
            StringBuilder sb = new StringBuilder();
jaroslav@1033
   502
            sb.append("v== ").append(name).append(" ==v\n");
jaroslav@1033
   503
            while (av-- > 0) {
jaroslav@1033
   504
                sb.append((char)is.read());
jaroslav@1033
   505
            }
jaroslav@1033
   506
            sb.append("\n^== ").append(name).append(" ==^");
jaroslav@1033
   507
            LOG.log(Level.INFO, sb.toString());
jaroslav@1033
   508
        }
jaroslav@1033
   509
    }
jaroslav@1033
   510
jaroslav@1033
   511
    private void deleteTree(File file) {
jaroslav@1033
   512
        if (file == null) {
jaroslav@1033
   513
            return;
jaroslav@1033
   514
        }
jaroslav@1033
   515
        File[] arr = file.listFiles();
jaroslav@1033
   516
        if (arr != null) {
jaroslav@1033
   517
            for (File s : arr) {
jaroslav@1033
   518
                deleteTree(s);
jaroslav@1033
   519
            }
jaroslav@1033
   520
        }
jaroslav@1033
   521
        file.delete();
jaroslav@1033
   522
    }
jaroslav@1033
   523
jaroslav@1033
   524
    @Override
jaroslav@1076
   525
    public HttpServer call() throws Exception {
jaroslav@1076
   526
        return server;
jaroslav@1076
   527
    }
jaroslav@1076
   528
    
jaroslav@1076
   529
    @Override
jaroslav@1033
   530
    public void close() throws IOException {
jaroslav@1033
   531
        shutdown();
jaroslav@1033
   532
    }
jaroslav@1033
   533
jaroslav@1041
   534
    protected Object[] showBrwsr(URI uri) throws IOException {
jaroslav@1041
   535
        LOG.log(Level.INFO, "Showing {0}", uri);
jaroslav@1041
   536
        if (cmd == null) {
jaroslav@1041
   537
            try {
jaroslav@1041
   538
                LOG.log(Level.INFO, "Trying Desktop.browse on {0} {2} by {1}", new Object[] {
jaroslav@1041
   539
                    System.getProperty("java.vm.name"),
jaroslav@1041
   540
                    System.getProperty("java.vm.vendor"),
jaroslav@1041
   541
                    System.getProperty("java.vm.version"),
jaroslav@1041
   542
                });
jaroslav@1041
   543
                java.awt.Desktop.getDesktop().browse(uri);
jaroslav@1041
   544
                LOG.log(Level.INFO, "Desktop.browse successfully finished");
jaroslav@1041
   545
                return null;
jaroslav@1041
   546
            } catch (UnsupportedOperationException ex) {
jaroslav@1041
   547
                LOG.log(Level.INFO, "Desktop.browse not supported: {0}", ex.getMessage());
jaroslav@1041
   548
                LOG.log(Level.FINE, null, ex);
jaroslav@1041
   549
            }
jaroslav@1041
   550
        }
jaroslav@1041
   551
        {
jaroslav@1041
   552
            String cmdName = cmd == null ? "xdg-open" : cmd;
jaroslav@1041
   553
            String[] cmdArr = { 
jaroslav@1041
   554
                cmdName, uri.toString()
jaroslav@1041
   555
            };
jaroslav@1041
   556
            LOG.log(Level.INFO, "Launching {0}", Arrays.toString(cmdArr));
jaroslav@1041
   557
            final Process process = Runtime.getRuntime().exec(cmdArr);
jaroslav@1041
   558
            return new Object[] { process, null };
jaroslav@1041
   559
        }
jaroslav@1041
   560
    }
jaroslav@1041
   561
jaroslav@1088
   562
    abstract void generateBck2BrwsrJS(StringBuilder sb, Res loader) throws IOException;
jaroslav@1088
   563
    abstract String harnessResource();
jaroslav@1489
   564
    String compileJar(JarFile jar) throws IOException {
jaroslav@1489
   565
        return null;
jaroslav@1489
   566
    }
jaroslav@1505
   567
    String compileFromClassPath(URL f, Res loader) throws IOException {
jaroslav@1489
   568
        return null;
jaroslav@1489
   569
    }
jaroslav@1041
   570
jaroslav@1249
   571
    private static URI pageURL(String protocol, HttpServer server, final String page) {
jaroslav@1186
   572
        NetworkListener listener = server.getListeners().iterator().next();
jaroslav@1186
   573
        int port = listener.getPort();
jaroslav@1186
   574
        try {
jaroslav@1249
   575
            return new URI(protocol + "://localhost:" + port + page);
jaroslav@1186
   576
        } catch (URISyntaxException ex) {
jaroslav@1186
   577
            throw new IllegalStateException(ex);
jaroslav@1186
   578
        }
jaroslav@1186
   579
    }
jaroslav@1186
   580
jaroslav@1489
   581
    final class Res {
jaroslav@1489
   582
        String compileJar(JarFile jar) throws IOException {
jaroslav@1489
   583
            return BaseHTTPLauncher.this.compileJar(jar);
jaroslav@1489
   584
        }
jaroslav@1505
   585
        String compileFromClassPath(URL f) throws IOException {
jaroslav@1504
   586
            return BaseHTTPLauncher.this.compileFromClassPath(f, this);
jaroslav@1489
   587
        }
jaroslav@1513
   588
        public URL get(String resource, int skip) throws IOException {
jaroslav@1375
   589
            if (!resource.endsWith(".class")) {
jaroslav@1375
   590
                return getResource(resource, skip);
jaroslav@1375
   591
            }
jaroslav@1098
   592
            URL u = null;
jaroslav@1033
   593
            for (ClassLoader l : loaders) {
jaroslav@1033
   594
                Enumeration<URL> en = l.getResources(resource);
jaroslav@1033
   595
                while (en.hasMoreElements()) {
jaroslav@1033
   596
                    u = en.nextElement();
jaroslav@1098
   597
                    if (u.toExternalForm().matches("^.*emul.*rt\\.jar.*$")) {
jaroslav@1489
   598
                        return u;
jaroslav@1098
   599
                    }
jaroslav@1033
   600
                }
jaroslav@1098
   601
            }
jaroslav@1098
   602
            if (u != null) {
jaroslav@1098
   603
                if (u.toExternalForm().contains("rt.jar")) {
jaroslav@1347
   604
                    LOG.log(Level.WARNING, "No fallback to bootclasspath for {0}", u);
jaroslav@1347
   605
                    return null;
jaroslav@1033
   606
                }
jaroslav@1489
   607
                return u;
jaroslav@1033
   608
            }
jaroslav@1033
   609
            throw new IOException("Can't find " + resource);
jaroslav@1033
   610
        }
jaroslav@1513
   611
        private URL getResource(String resource, int skip) throws IOException {
jaroslav@1375
   612
            for (ClassLoader l : loaders) {
jaroslav@1375
   613
                Enumeration<URL> en = l.getResources(resource);
jaroslav@1375
   614
                while (en.hasMoreElements()) {
jaroslav@1375
   615
                    final URL now = en.nextElement();
jaroslav@1383
   616
                    if (now.toExternalForm().contains("sisu-inject-bean")) {
jaroslav@1383
   617
                        // certainly we don't want this resource, as that
jaroslav@1383
   618
                        // module is not compiled with target 1.6, currently
jaroslav@1383
   619
                        continue;
jaroslav@1383
   620
                    }
jaroslav@1375
   621
                    if (--skip < 0) {
jaroslav@1513
   622
                        return now;
jaroslav@1375
   623
                    }
jaroslav@1375
   624
                }
jaroslav@1375
   625
            }
jaroslav@1383
   626
            throw new IOException("Not found (anymore of) " + resource);
jaroslav@1375
   627
        }
jaroslav@1033
   628
    }
jaroslav@1033
   629
jaroslav@1033
   630
    private static class Page extends HttpHandler {
jaroslav@1033
   631
        final String resource;
jaroslav@1033
   632
        private final String[] args;
jaroslav@1033
   633
        private final Res res;
jaroslav@1033
   634
        
jaroslav@1033
   635
        public Page(Res res, String resource, String... args) {
jaroslav@1033
   636
            this.res = res;
jaroslav@1033
   637
            this.resource = resource;
jaroslav@1033
   638
            this.args = args.length == 0 ? new String[] { "$0" } : args;
jaroslav@1033
   639
        }
jaroslav@1033
   640
jaroslav@1033
   641
        @Override
jaroslav@1033
   642
        public void service(Request request, Response response) throws Exception {
jaroslav@1033
   643
            String r = computePage(request);
jaroslav@1033
   644
            if (r.startsWith("/")) {
jaroslav@1033
   645
                r = r.substring(1);
jaroslav@1033
   646
            }
jaroslav@1033
   647
            String[] replace = {};
jaroslav@1033
   648
            if (r.endsWith(".html")) {
jaroslav@1033
   649
                response.setContentType("text/html");
jaroslav@1033
   650
                LOG.info("Content type text/html");
jaroslav@1033
   651
                replace = args;
jaroslav@1033
   652
            }
jaroslav@1033
   653
            if (r.endsWith(".xhtml")) {
jaroslav@1033
   654
                response.setContentType("application/xhtml+xml");
jaroslav@1033
   655
                LOG.info("Content type application/xhtml+xml");
jaroslav@1033
   656
                replace = args;
jaroslav@1033
   657
            }
jaroslav@1033
   658
            OutputStream os = response.getOutputStream();
jaroslav@1165
   659
            try { 
jaroslav@1513
   660
                InputStream is = res.get(r, 0).openStream();
jaroslav@1033
   661
                copyStream(is, os, request.getRequestURL().toString(), replace);
jaroslav@1033
   662
            } catch (IOException ex) {
jaroslav@1033
   663
                response.setDetailMessage(ex.getLocalizedMessage());
jaroslav@1033
   664
                response.setError();
jaroslav@1033
   665
                response.setStatus(404);
jaroslav@1033
   666
            }
jaroslav@1033
   667
        }
jaroslav@1033
   668
jaroslav@1033
   669
        protected String computePage(Request request) {
jaroslav@1033
   670
            String r = resource;
jaroslav@1033
   671
            if (r == null) {
jaroslav@1033
   672
                r = request.getHttpHandlerPath();
jaroslav@1033
   673
            }
jaroslav@1033
   674
            return r;
jaroslav@1033
   675
        }
jaroslav@1033
   676
    }
jaroslav@1033
   677
    
jaroslav@1033
   678
    private static class SubTree extends Page {
jaroslav@1033
   679
jaroslav@1033
   680
        public SubTree(Res res, String resource, String... args) {
jaroslav@1033
   681
            super(res, resource, args);
jaroslav@1033
   682
        }
jaroslav@1033
   683
jaroslav@1033
   684
        @Override
jaroslav@1033
   685
        protected String computePage(Request request) {
jaroslav@1033
   686
            return resource + request.getHttpHandlerPath();
jaroslav@1033
   687
        }
jaroslav@1033
   688
        
jaroslav@1033
   689
        
jaroslav@1033
   690
    }
jaroslav@1033
   691
jaroslav@1271
   692
    private class VMAndPages extends StaticHttpHandler {
jaroslav@1271
   693
        private String vmResource;
jaroslav@1271
   694
        
jaroslav@1271
   695
        public VMAndPages() {
jaroslav@1271
   696
            super((String[]) null);
jaroslav@1271
   697
        }
jaroslav@1271
   698
        
jaroslav@1033
   699
        @Override
jaroslav@1033
   700
        public void service(Request request, Response response) throws Exception {
jaroslav@1271
   701
            if (request.getRequestURI().equals(vmResource)) {
jaroslav@1271
   702
                response.setCharacterEncoding("UTF-8");
jaroslav@1271
   703
                response.setContentType("text/javascript");
jaroslav@1271
   704
                StringBuilder sb = new StringBuilder();
jaroslav@1271
   705
                generateBck2BrwsrJS(sb, BaseHTTPLauncher.this.resources);
jaroslav@1271
   706
                response.getWriter().write(sb.toString());
jaroslav@1271
   707
            } else {
jaroslav@1271
   708
                super.service(request, response);
jaroslav@1271
   709
            }
jaroslav@1271
   710
        }
jaroslav@1271
   711
jaroslav@1271
   712
        private void registerVM(String vmResource) {
jaroslav@1271
   713
            this.vmResource = vmResource;
jaroslav@1033
   714
        }
jaroslav@1033
   715
    }
jaroslav@1033
   716
jaroslav@1033
   717
    private static class Classes extends HttpHandler {
jaroslav@1033
   718
        private final Res loader;
jaroslav@1033
   719
jaroslav@1033
   720
        public Classes(Res loader) {
jaroslav@1033
   721
            this.loader = loader;
jaroslav@1033
   722
        }
jaroslav@1033
   723
jaroslav@1033
   724
        @Override
jaroslav@1033
   725
        public void service(Request request, Response response) throws Exception {
jaroslav@1033
   726
            String res = request.getHttpHandlerPath();
jaroslav@1033
   727
            if (res.startsWith("/")) {
jaroslav@1033
   728
                res = res.substring(1);
jaroslav@1033
   729
            }
jaroslav@1513
   730
            String skip = request.getParameter("skip");
jaroslav@1513
   731
            int skipCnt = skip == null ? 0 : Integer.parseInt(skip);
jaroslav@1513
   732
            URL url = loader.get(res, skipCnt);
jaroslav@1521
   733
            try {
jaroslav@1521
   734
                if (url.getProtocol().equals("jar")) {
jaroslav@1521
   735
                    JarURLConnection juc = (JarURLConnection) url.openConnection();
jaroslav@1521
   736
                    String s = loader.compileJar(juc.getJarFile());
jaroslav@1521
   737
                    if (s != null) {
jaroslav@1521
   738
                        Writer w = response.getWriter();
jaroslav@1521
   739
                        w.append(s);
jaroslav@1521
   740
                        w.close();
jaroslav@1521
   741
                        return;
jaroslav@1521
   742
                    }
jaroslav@1489
   743
                }
jaroslav@1521
   744
                if (url.getProtocol().equals("file")) {
jaroslav@1521
   745
                    String s = loader.compileFromClassPath(url);
jaroslav@1521
   746
                    if (s != null) {
jaroslav@1521
   747
                        Writer w = response.getWriter();
jaroslav@1521
   748
                        w.append(s);
jaroslav@1521
   749
                        w.close();
jaroslav@1521
   750
                        return;
jaroslav@1521
   751
                    }
jaroslav@1489
   752
                }
jaroslav@1521
   753
            } catch (IOException ex) {
jaroslav@1521
   754
                LOG.log(Level.SEVERE, "Cannot handle " + res, ex);
jaroslav@1521
   755
                throw ex;
jaroslav@1489
   756
            }
jaroslav@1489
   757
            Exception ex = new Exception("Won't server bytes of " + url);
jaroslav@1489
   758
            /*
jaroslav@1489
   759
            try (InputStream is = url.openStream()) {
jaroslav@1513
   760
=======
jaroslav@1165
   761
            InputStream is = null;
jaroslav@1165
   762
            try {
jaroslav@1375
   763
                String skip = request.getParameter("skip");
jaroslav@1375
   764
                int skipCnt = skip == null ? 0 : Integer.parseInt(skip);
jaroslav@1375
   765
                is = loader.get(res, skipCnt);
jaroslav@1513
   766
>>>>>>> other
jaroslav@1033
   767
                response.setContentType("text/javascript");
jaroslav@1033
   768
                Writer w = response.getWriter();
jaroslav@1489
   769
                w.append("([");
jaroslav@1033
   770
                for (int i = 0;; i++) {
jaroslav@1033
   771
                    int b = is.read();
jaroslav@1033
   772
                    if (b == -1) {
jaroslav@1033
   773
                        break;
jaroslav@1033
   774
                    }
jaroslav@1033
   775
                    if (i > 0) {
jaroslav@1033
   776
                        w.append(", ");
jaroslav@1033
   777
                    }
jaroslav@1033
   778
                    if (i % 20 == 0) {
jaroslav@1033
   779
                        w.write("\n");
jaroslav@1033
   780
                    }
jaroslav@1033
   781
                    if (b > 127) {
jaroslav@1033
   782
                        b = b - 256;
jaroslav@1033
   783
                    }
jaroslav@1033
   784
                    w.append(Integer.toString(b));
jaroslav@1033
   785
                }
jaroslav@1489
   786
                w.append("\n])");
jaroslav@1489
   787
            } catch (IOException ex) 
jaroslav@1489
   788
            */ {
jaroslav@1033
   789
                response.setStatus(HttpStatus.NOT_FOUND_404);
jaroslav@1033
   790
                response.setError();
jaroslav@1033
   791
                response.setDetailMessage(ex.getMessage());
jaroslav@1513
   792
            } /*finally {
jaroslav@1165
   793
                if (is != null) {
jaroslav@1165
   794
                    is.close();
jaroslav@1165
   795
                }
jaroslav@1513
   796
            }*/
jaroslav@1033
   797
        }
jaroslav@1513
   798
jaroslav@1033
   799
    }
jaroslav@1249
   800
    private static class WS extends WebSocketApplication {
jaroslav@1249
   801
jaroslav@1249
   802
        private final Resource r;
jaroslav@1249
   803
jaroslav@1249
   804
        private WS(Resource r) {
jaroslav@1249
   805
            this.r = r;
jaroslav@1249
   806
        }
jaroslav@1249
   807
jaroslav@1249
   808
        @Override
jaroslav@1249
   809
        public void onMessage(WebSocket socket, String text) {
jaroslav@1249
   810
            try {
jaroslav@1249
   811
                r.httpContent.reset();
jaroslav@1249
   812
                ByteArrayOutputStream out = new ByteArrayOutputStream();
jaroslav@1249
   813
                copyStream(r.httpContent, out, null, text);
jaroslav@1249
   814
                String s = new String(out.toByteArray(), "UTF-8");
jaroslav@1249
   815
                socket.send(s);
jaroslav@1249
   816
            } catch (IOException ex) {
jaroslav@1513
   817
                LOG.log(Level.WARNING, null, ex);
jaroslav@1249
   818
            }
jaroslav@1249
   819
        }
jaroslav@1249
   820
jaroslav@1249
   821
    }}