org.eclipse.che.plugin.jdb.server.util.JavaDebuggerUtils.java Source code

Introduction

Here is the source code for org.eclipse.che.plugin.jdb.server.util.JavaDebuggerUtils.java

Source

/*
 * Copyright (c) 2012-2017 Red Hat, Inc.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Red Hat, Inc. - initial API and implementation
 */
package org.eclipse.che.plugin.jdb.server.util;

import static java.lang.System.getProperty;
import static java.util.Collections.singletonList;

import com.google.common.collect.ImmutableMap;
import com.sun.jdi.ThreadReference;
import com.sun.jdi.VirtualMachine;
import java.lang.reflect.Field;
import java.util.Map;
import java.util.Optional;
import java.util.concurrent.BlockingQueue;
import org.eclipse.che.api.debug.shared.model.Breakpoint;
import org.eclipse.che.api.debug.shared.model.Location;
import org.eclipse.che.api.debug.shared.model.ThreadState;
import org.eclipse.che.api.debug.shared.model.event.DebuggerEvent;
import org.eclipse.che.api.debug.shared.model.event.SuspendEvent;
import org.eclipse.che.api.debug.shared.model.impl.action.StartActionImpl;
import org.eclipse.che.api.debugger.server.exceptions.DebuggerException;
import org.eclipse.che.plugin.jdb.server.JavaDebugger;
import org.eclipse.che.plugin.jdb.server.JavaDebuggerFactory;

/** @author Anatolii Bazko */
public class JavaDebuggerUtils {

    /** Connects to process and starts debug. */
    public static JavaDebugger startJavaDebugger(Breakpoint breakpoint, BlockingQueue<DebuggerEvent> debuggerEvents)
            throws Exception {
        Map<String, String> connectionProperties = ImmutableMap.of("host", "localhost", "port",
                getProperty("debug.port"));

        JavaDebuggerFactory factory = new JavaDebuggerFactory();
        JavaDebugger debugger = (JavaDebugger) factory.create(connectionProperties, debuggerEvents::add);

        debugger.start(new StartActionImpl(singletonList(breakpoint)));

        return debugger;
    }

    public static void ensureSuspendAtDesiredLocation(Location desiredLocation,
            BlockingQueue<DebuggerEvent> debuggerEvents) throws InterruptedException {
        for (;;) {
            DebuggerEvent event = debuggerEvents.take();
            if (event instanceof SuspendEvent) {
                SuspendEvent suspendEvent = (SuspendEvent) event;
                Location location = suspendEvent.getLocation();

                if (location.getTarget().equals(desiredLocation.getTarget())
                        && location.getLineNumber() == desiredLocation.getLineNumber()) {

                    return;
                }
            }
        }
    }

    /**
     * Terminates Virtual Machine.
     *
     * @see VirtualMachine#exit(int)
     */
    public static void terminateVirtualMachineQuietly(JavaDebugger javaDebugger) throws Exception {
        Field vmField = JavaDebugger.class.getDeclaredField("vm");
        vmField.setAccessible(true);
        VirtualMachine vm = (VirtualMachine) vmField.get(javaDebugger);

        try {
            vm.exit(0);
        } catch (Exception ignored) {
            // quietly ignore exception, if VM has been already terminated
        }
    }

    /**
     * Iterates threads until main is found and returns its id.
     *
     * @see ThreadReference#uniqueID()
     */
    public static long findMainThreadId(JavaDebugger javaDebugger) throws DebuggerException {
        Optional<ThreadState> main = javaDebugger.getThreadDump().stream().filter(t -> t.getName().equals("main"))
                .findAny();
        main.orElseThrow(() -> new DebuggerException("Main thread not found"));
        return main.get().getId();
    }
}