platform/platform-impl/src/com/intellij/ide/XmlRpcServerImpl.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.ide;
  17
  18 import com.intellij.openapi.application.ApplicationManager;
  19 import com.intellij.openapi.components.ApplicationComponent;
  20 import com.intellij.openapi.diagnostic.Logger;
  21 import org.apache.xmlrpc.IdeaAwareWebServer;
  22 import org.apache.xmlrpc.IdeaAwareXmlRpcServer;
  23 import org.apache.xmlrpc.WebServer;
  24 import org.jetbrains.annotations.NonNls;
  25 import org.jetbrains.annotations.NotNull;
  26
  27 import java.io.IOException;
  28 import java.net.BindException;
  29 import java.net.ServerSocket;
  30
  31 /**
  32  * @author mike
  33  */
  34 public class XmlRpcServerImpl implements XmlRpcServer, ApplicationComponent {
  35   private static final Logger LOG = Logger.getInstance("#com.intellij.ide.XmlRpcServerImpl");
  36   private static final int FIRST_PORT_NUMBER = 63342;
  37   private static final int PORTS_COUNT = 20;
  38   public static int detectedPortNumber = -1;
  39   private WebServer myWebServer;
  40   @NonNls private static final String PROPERTY_RPC_PORT = "rpc.port";
  41
  42   @NotNull
  43   @NonNls
  44   public String getComponentName() {
  45     return "XmlRpcServer";
  46   }
  47
  48   public void initComponent() {
  49     if (ApplicationManager.getApplication().isUnitTestMode() || !checkPort()) return;
  50     final Thread thread = Thread.currentThread();
  51     final int currentPriority = thread.getPriority();
  52     try {
  53       thread.setPriority(Thread.NORM_PRIORITY - 2);
  54       myWebServer = new IdeaAwareWebServer(getPortNumber(), null, new IdeaAwareXmlRpcServer());
  55       myWebServer.start();
  56     }
  57     catch (Exception e) {
  58       LOG.error(e);
  59       myWebServer = null;
  60     }
  61     finally {
  62       thread.setPriority(currentPriority);
  63     }
  64   }
  65
  66   public int getPortNumber() {
  67     return detectedPortNumber == -1 ? getDefaultPort() : detectedPortNumber;
  68   }
  69
  70   private static int getDefaultPort() {
  71     if (System.getProperty(PROPERTY_RPC_PORT) != null) return Integer.parseInt(System.getProperty(PROPERTY_RPC_PORT));
  72     return FIRST_PORT_NUMBER;
  73   }
  74
  75   private static boolean checkPort() {
  76     ServerSocket socket = null;
  77     try {
  78       final int firstPort = getDefaultPort();
  79       for (int i = 0; i < PORTS_COUNT; i++) {
  80         int port = firstPort + i;
  81         try {
  82           socket = new ServerSocket(port);
  83           detectedPortNumber = port;
  84           return true;
  85         }
  86         catch (BindException ignored) {
  87         }
  88       }
  89
  90       try {
  91         // try any port
  92         socket = new ServerSocket(0);
  93         detectedPortNumber = socket.getLocalPort();
  94         return true;
  95       } catch (BindException ignored) {
  96       }
  97     }
  98     catch (IOException e) {
  99       LOG.info(e);
 100     }
 101     finally {
 102       if (socket != null) {
 103         try {
 104           socket.close();
 105         }
 106         catch (IOException e) {
 107           LOG.error(e);
 108         }
 109       }
 110     }
 111     return false;
 112   }
 113
 114   public void disposeComponent() {
 115     if (myWebServer != null) {
 116       myWebServer.shutdown();
 117     }
 118   }
 119
 120   public void addHandler(String name, Object handler) {
 121     if (myWebServer != null) {
 122       myWebServer.addHandler(name, handler);
 123     }
 124     else {
 125       LOG.info("Handler not registered because XML-RPC server is not running");
 126     }
 127   }
 128
 129   public void removeHandler(String name) {
 130     if (myWebServer != null) {
 131       myWebServer.removeHandler(name);
 132     }
 133   }
 134 }