detect remote debugger connections (internal mode only)

diff --git a/platform/platform-impl/src/com/intellij/internal/DebugAttachDetector.java b/platform/platform-impl/src/com/intellij/internal/DebugAttachDetector.java
new file mode 100644 (file)
index 0000000..97e5419
--- /dev/null
+++ b/platform/platform-impl/src/com/intellij/internal/DebugAttachDetector.java
@@ -0,0 +1,103 @@
+/*
+ * Copyright 2000-2015 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.intellij.internal;
+
+import com.intellij.notification.*;
+import com.intellij.openapi.application.ex.ApplicationManagerEx;
+import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.util.Alarm;
+import com.intellij.util.SingleAlarm;
+
+import java.io.IOException;
+import java.lang.management.ManagementFactory;
+import java.net.Socket;
+
+/**
+ * @author egor
+ */
+public class DebugAttachDetector {
+  private static final Logger LOG = Logger.getInstance(DebugAttachDetector.class);
+
+  private String myHost;
+  private int myPort = -1;
+  private SingleAlarm myAlarm;
+  private boolean myAttached;
+  private boolean myReady = false;
+
+  public DebugAttachDetector() {
+    if (!ApplicationManagerEx.getApplicationEx().isInternal() || "true".equals(System.getProperty("idea.debug.mode"))) return;
+
+    for (String argument : ManagementFactory.getRuntimeMXBean().getInputArguments()) {
+      if (argument.startsWith("-agentlib:jdwp")) {
+        String[] params = argument.split(",");
+        for (String param : params) {
+          if (param.startsWith("address")) {
+            try {
+              String[] address = param.split("=")[1].split(":");
+              if (address.length == 1) {
+                myHost = "127.0.0.1";
+                myPort = Integer.parseInt(address[0]);
+              }
+              else {
+                myHost = address[0];
+                myPort = Integer.parseInt(address[1]);
+              }
+            }
+            catch (Exception e) {
+              LOG.error(e);
+              return;
+            }
+            break;
+          }
+        }
+        break;
+      }
+    }
+
+    if (myPort < 0) return;
+
+    myAlarm = new SingleAlarm(new Runnable() {
+      @Override
+      public void run() {
+        boolean attached = checkAttached();
+        if (!myReady) {
+          myAttached = attached;
+          myReady = true;
+        }
+        else if (attached != myAttached) {
+          myAttached = attached;
+          Notifications.Bus.notify(new Notification(Notifications.SYSTEM_MESSAGES_GROUP_ID,
+                                                    "Remote debugger",
+                                                    myAttached ? "attached" : "detached",
+                                                    NotificationType.WARNING));
+        }
+        myAlarm.request();
+      }
+    }, 5000, Alarm.ThreadToUse.POOLED_THREAD, null);
+    myAlarm.request();
+  }
+
+  private boolean checkAttached() {
+    try {
+      Socket socket = new Socket(myHost, myPort);
+      socket.close();
+      return false;
+    }
+    catch (IOException e) {
+      return true;
+    }
+  }
+}

diff --git a/platform/platform-resources/src/componentSets/Platform.xml b/platform/platform-resources/src/componentSets/Platform.xml
index c58019c209078c57ae552b7f8dc42e9af4612670..48f5c1d8112b35f3b5f86f971938049e239593dd 100644 (file)
--- a/platform/platform-resources/src/componentSets/Platform.xml
+++ b/platform/platform-resources/src/componentSets/Platform.xml
@@ -127,6 +127,10 @@
       <headless-implementation-class/>
     </component>
 
+    <component>
+      <implementation-class>com.intellij.internal.DebugAttachDetector</implementation-class>
+    </component>
+
     <component>
       <implementation-class>com.intellij.diagnostic.DebugLogManager</implementation-class>
     </component>