package org.eclipse.jdt.internal.launching;
import java.io.IOException;
import java.net.ConnectException;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.debug.core.ILaunch;
import org.eclipse.debug.core.ILaunchConfiguration;
import org.eclipse.debug.core.model.IDebugTarget;
import org.eclipse.jdi.Bootstrap;
import org.eclipse.jdi.TimeoutException;
import org.eclipse.jdt.debug.core.JDIDebugModel;
import org.eclipse.jdt.launching.IJavaLaunchConfigurationConstants;
import org.eclipse.jdt.launching.IVMConnector;
import org.eclipse.osgi.util.NLS;
import com.sun.jdi.VMDisconnectedException;
import com.sun.jdi.VirtualMachine;
import com.sun.jdi.connect.AttachingConnector;
import com.sun.jdi.connect.Connector;
import com.sun.jdi.connect.IllegalConnectorArgumentsException;
public class SocketAttachConnector implements IVMConnector {
protected static AttachingConnector getAttachingConnector() throws CoreException {
AttachingConnector connector= null;
Iterator<AttachingConnector> iter= Bootstrap.virtualMachineManager().attachingConnectors().iterator();
while (iter.hasNext()) {
AttachingConnector lc= iter.next();
if (lc.name().equals("com.sun.jdi.SocketAttach")) {
connector= lc;
break;
}
}
if (connector == null) {
abort(LaunchingMessages.SocketAttachConnector_Socket_attaching_connector_not_available_3, null, IJavaLaunchConfigurationConstants.ERR_SHARED_MEMORY_CONNECTOR_UNAVAILABLE);
}
return connector;
}
@Override
public String getIdentifier() {
return IJavaLaunchConfigurationConstants.ID_SOCKET_ATTACH_VM_CONNECTOR;
}
@Override
public String getName() {
return LaunchingMessages.SocketAttachConnector_Standard__Socket_Attach__4;
}
protected static void abort(String message, Throwable exception, int code) throws CoreException {
throw new CoreException(new Status(IStatus.ERROR, LaunchingPlugin.getUniqueIdentifier(), code, message, exception));
}
@Override
public void connect(Map<String, String> arguments, IProgressMonitor monitor, ILaunch launch) throws CoreException {
if (monitor == null) {
monitor = new NullProgressMonitor();
}
IProgressMonitor subMonitor = new SubProgressMonitor(monitor, 1);
subMonitor.beginTask(LaunchingMessages.SocketAttachConnector_Connecting____1, 2);
subMonitor.subTask(LaunchingMessages.SocketAttachConnector_Configuring_connection____1);
AttachingConnector connector= getAttachingConnector();
String portNumberString = arguments.get("port");
if (portNumberString == null) {
abort(LaunchingMessages.SocketAttachConnector_Port_unspecified_for_remote_connection__2, null, IJavaLaunchConfigurationConstants.ERR_UNSPECIFIED_PORT);
}
String host = arguments.get("hostname");
if (host == null) {
abort(LaunchingMessages.SocketAttachConnector_Hostname_unspecified_for_remote_connection__4, null, IJavaLaunchConfigurationConstants.ERR_UNSPECIFIED_HOSTNAME);
}
Map<String, Connector.Argument> map= connector.defaultArguments();
Connector.Argument param= map.get("hostname");
param.setValue(host);
param= map.get("port");
param.setValue(portNumberString);
String timeoutString = arguments.get("timeout");
if (timeoutString != null) {
param= map.get("timeout");
param.setValue(timeoutString);
}
ILaunchConfiguration configuration = launch.getLaunchConfiguration();
boolean allowTerminate = false;
if (configuration != null) {
allowTerminate = configuration.getAttribute(IJavaLaunchConfigurationConstants.ATTR_ALLOW_TERMINATE, false);
}
subMonitor.worked(1);
subMonitor.subTask(LaunchingMessages.SocketAttachConnector_Establishing_connection____2);
try {
VirtualMachine vm = connector.attach(map);
String vmLabel = constructVMLabel(vm, host, portNumberString, configuration);
IDebugTarget debugTarget= JDIDebugModel.newDebugTarget(launch, vm, vmLabel, null, allowTerminate, true);
launch.addDebugTarget(debugTarget);
subMonitor.worked(1);
subMonitor.done();
} catch (TimeoutException e) {
abort(LaunchingMessages.SocketAttachConnector_0, e, IJavaLaunchConfigurationConstants.ERR_REMOTE_VM_CONNECTION_FAILED);
} catch (UnknownHostException e) {
abort(NLS.bind(LaunchingMessages.SocketAttachConnector_Failed_to_connect_to_remote_VM_because_of_unknown_host____0___1, new String[]{host}), e, IJavaLaunchConfigurationConstants.ERR_REMOTE_VM_CONNECTION_FAILED);
} catch (ConnectException e) {
abort(LaunchingMessages.SocketAttachConnector_Failed_to_connect_to_remote_VM_as_connection_was_refused_2, e, IJavaLaunchConfigurationConstants.ERR_REMOTE_VM_CONNECTION_FAILED);
} catch (IOException e) {
abort(LaunchingMessages.SocketAttachConnector_Failed_to_connect_to_remote_VM_1, e, IJavaLaunchConfigurationConstants.ERR_REMOTE_VM_CONNECTION_FAILED);
} catch (IllegalConnectorArgumentsException e) {
abort(LaunchingMessages.SocketAttachConnector_Failed_to_connect_to_remote_VM_1, e, IJavaLaunchConfigurationConstants.ERR_REMOTE_VM_CONNECTION_FAILED);
}
}
protected String constructVMLabel(VirtualMachine vm, String host, String port, ILaunchConfiguration configuration) {
String name = null;
try {
name = vm.name();
} catch (TimeoutException e) {
} catch (VMDisconnectedException e) {
}
if (name == null) {
if (configuration == null) {
name = "";
} else {
name = configuration.getName();
}
}
StringBuilder buffer = new StringBuilder(name);
if(!"".equals(name)) {
buffer.append(' ');
}
buffer.append('[');
buffer.append(host);
buffer.append(':');
buffer.append(port);
buffer.append(']');
return buffer.toString();
}
@Override
public Map<String, Connector.Argument> getDefaultArguments() throws CoreException {
Map<String, Connector.Argument> def = getAttachingConnector().defaultArguments();
Connector.IntegerArgument arg = (Connector.IntegerArgument)def.get("port");
arg.setValue(8000);
return def;
}
@Override
public List<String> getArgumentOrder() {
List<String> list = new ArrayList<>(2);
list.add("hostname");
list.add("port");
return list;
}
}