| /*========================================================================= |
| * Copyright (c) 2010-2014 Pivotal Software, Inc. All Rights Reserved. |
| * This product is protected by U.S. and international copyright |
| * and intellectual property laws. Pivotal products are covered by |
| * one or more patents listed at http://www.pivotal.io/patents. |
| *========================================================================= |
| */ |
| package com.gemstone.gemfire.internal; |
| |
| import java.io.*; |
| import java.net.*; |
| |
| import com.gemstone.gemfire.distributed.internal.DistributionConfig; |
| import com.gemstone.gemfire.internal.i18n.LocalizedStrings; |
| |
| import java.util.Enumeration; |
| import java.util.Random; |
| |
| /** |
| * This class determines whether or not a given port is available and |
| * can also provide a randomly selected available port. |
| * |
| */ |
| public class AvailablePort { |
| |
| /** Is the port available for a Socket (TCP) connection? */ |
| public static final int SOCKET = 0; |
| |
| /** Is the port available for a JGroups (UDP) connection */ |
| public static final int JGROUPS = 1; |
| |
| /////////////////////// Static Methods /////////////////////// |
| |
| /** |
| * see if there is a gemfire system property that establishes a |
| * default address for the given protocol, and return it |
| */ |
| private static InetAddress getAddress(int protocol) { |
| String name = null; |
| try { |
| if (protocol == SOCKET) { |
| name = System.getProperty("gemfire.bind-address"); |
| } |
| else if (protocol == JGROUPS) { |
| name = System.getProperty("gemfire.mcast-address"); |
| } |
| if (name != null) { |
| return InetAddress.getByName(name); |
| } |
| } |
| catch (UnknownHostException e) { |
| throw new RuntimeException("Unable to resolve address " + name); |
| } |
| return null; |
| } |
| |
| |
| /** |
| * Returns whether or not the given port on the local host is |
| * available (that is, unused). |
| * |
| * @param port |
| * The port to check |
| * @param protocol |
| * The protocol to check (either {@link #SOCKET} or {@link |
| * #JGROUPS}). |
| * |
| * @throws IllegalArgumentException |
| * <code>protocol</code> is unknown |
| */ |
| public static boolean isPortAvailable(final int port, int protocol) { |
| return isPortAvailable(port, protocol, getAddress(protocol)); |
| } |
| |
| |
| /** |
| * Returns whether or not the given port on the local host is |
| * available (that is, unused). |
| * |
| * @param port |
| * The port to check |
| * @param protocol |
| * The protocol to check (either {@link #SOCKET} or {@link |
| * #JGROUPS}). |
| * @param addr the bind address (or mcast address) to use |
| * |
| * @throws IllegalArgumentException |
| * <code>protocol</code> is unknown |
| */ |
| public static boolean isPortAvailable(final int port, int protocol, InetAddress addr) { |
| if (protocol == SOCKET) { |
| // Try to create a ServerSocket |
| if(addr == null) { |
| return testAllInterfaces(port); |
| } else { |
| return testOneInterface(addr, port); |
| } |
| } |
| |
| else if (protocol == JGROUPS) { |
| DatagramSocket socket = null; |
| try { |
| socket = new MulticastSocket(); |
| socket.setSoTimeout(Integer.getInteger("AvailablePort.timeout", 2000).intValue()); |
| byte[] buffer = new byte[4]; |
| buffer[0] = (byte)'p'; |
| buffer[1] = (byte)'i'; |
| buffer[2] = (byte)'n'; |
| buffer[3] = (byte)'g'; |
| SocketAddress mcaddr = new InetSocketAddress( |
| addr==null? DistributionConfig.DEFAULT_MCAST_ADDRESS : addr, port); |
| DatagramPacket packet = new DatagramPacket(buffer, 0, buffer.length, mcaddr); |
| socket.send(packet); |
| try { |
| socket.receive(packet); |
| packet.getData(); // make sure there's data, but no need to process it |
| return false; |
| } |
| catch (SocketTimeoutException ste) { |
| //System.out.println("socket read timed out"); |
| return true; |
| } |
| catch (Exception e) { |
| e.printStackTrace(); |
| return false; |
| } |
| } |
| catch (java.io.IOException ioe) { |
| if (ioe.getMessage().equals("Network is unreachable")) { |
| throw new RuntimeException(LocalizedStrings.AvailablePort_NETWORK_IS_UNREACHABLE.toLocalizedString(), ioe); |
| } |
| ioe.printStackTrace(); |
| return false; |
| } |
| catch (Exception e) { |
| e.printStackTrace(); |
| return false; |
| } |
| finally { |
| if (socket != null) { |
| try { |
| socket.close(); |
| } |
| catch (Exception e) { |
| e.printStackTrace(); |
| } |
| } |
| } |
| } |
| |
| else { |
| throw new IllegalArgumentException(LocalizedStrings.AvailablePort_UNKNOWN_PROTOCOL_0.toLocalizedString(Integer.valueOf(protocol))); |
| } |
| } |
| |
| public static Keeper isPortKeepable(final int port, int protocol, InetAddress addr) { |
| if (protocol == SOCKET) { |
| // Try to create a ServerSocket |
| if(addr == null) { |
| return keepAllInterfaces(port); |
| } else { |
| return keepOneInterface(addr, port); |
| } |
| } else if (protocol == JGROUPS) { |
| throw new IllegalArgumentException("You can not keep the JGROUPS protocol"); |
| } else { |
| throw new IllegalArgumentException(LocalizedStrings.AvailablePort_UNKNOWN_PROTOCOL_0.toLocalizedString(Integer.valueOf(protocol))); |
| } |
| } |
| |
| private static boolean testOneInterface(InetAddress addr, int port) { |
| Keeper k = keepOneInterface(addr, port); |
| if (k != null) { |
| k.release(); |
| return true; |
| } else { |
| return false; |
| } |
| } |
| private static Keeper keepOneInterface(InetAddress addr, int port) { |
| ServerSocket server = null; |
| try { |
| //(new Exception("Opening server socket on " + port)).printStackTrace(); |
| server = new ServerSocket(); |
| String osName = System.getProperty("os.name"); |
| if(osName != null && !osName.startsWith("Windows")) { |
| server.setReuseAddress(true); |
| } |
| if (addr != null) { |
| server.bind(new InetSocketAddress(addr, port)); |
| } |
| else { |
| server.bind(new InetSocketAddress(port)); |
| } |
| Keeper result = new Keeper(server, port); |
| server = null; |
| return result; |
| } |
| catch (java.io.IOException ioe) { |
| if (ioe.getMessage().equals("Network is unreachable")) { |
| throw new RuntimeException("Network is unreachable"); |
| } |
| //ioe.printStackTrace(); |
| if(addr instanceof Inet6Address) { |
| byte[] addrBytes = addr.getAddress(); |
| if ((addrBytes[0] == (byte) 0xfe) && (addrBytes[1] == (byte) 0x80)) { |
| //Hack, early Sun 1.5 versions (like Hitachi's JVM) cannot handle IPv6 |
| //link local addresses. Cannot trust InetAddress.isLinkLocalAddress() |
| //see http://bugs.sun.com/bugdatabase/view_bug.do?bug_id=6558853 |
| //By returning true we ignore these interfaces and potentially say a |
| //port is not in use when it really is. |
| Keeper result = new Keeper(server, port); |
| server = null; |
| return result; |
| } |
| } |
| return null; |
| } |
| catch (Exception ex) { |
| return null; |
| } finally { |
| if (server != null) |
| try { |
| server.close(); |
| } catch (Exception ex) { |
| |
| } |
| } |
| } |
| |
| /** |
| * Test to see if a given port is available port on all interfaces on this host. |
| * @param port |
| * @return true of if the port is free on all interfaces |
| */ |
| private static boolean testAllInterfaces(int port) { |
| Keeper k = keepAllInterfaces(port); |
| if (k != null) { |
| k.release(); |
| return true; |
| } else { |
| return false; |
| } |
| } |
| private static Keeper keepAllInterfaces(int port) { |
| //First check to see if we can bind to the wildcard address. |
| if(!testOneInterface(null, port)) { |
| return null; |
| } |
| |
| //Now check all of the addresses for all of the addresses |
| //on this system. On some systems (solaris, aix) binding |
| //to the wildcard address will successfully bind to only some |
| //of the interfaces if other interfaces are in use. We want to |
| //make sure this port is completely free. |
| // |
| //Note that we still need the check of the wildcard address, above, |
| //because on some systems (aix) we can still bind to specific addresses |
| //if someone else has bound to the wildcard address. |
| Enumeration en; |
| try { |
| en = NetworkInterface.getNetworkInterfaces(); |
| } catch (SocketException e) { |
| throw new RuntimeException(e); |
| } |
| while(en.hasMoreElements()) { |
| NetworkInterface next = (NetworkInterface) en.nextElement(); |
| Enumeration en2 = next.getInetAddresses(); |
| while(en2.hasMoreElements()) { |
| InetAddress addr = (InetAddress) en2.nextElement(); |
| boolean available = testOneInterface(addr, port); |
| if(!available) { |
| return null; |
| } |
| } |
| } |
| // Now do it one more time but reserve the wildcard address |
| return keepOneInterface(null, port); |
| } |
| |
| |
| /** |
| * Returns a randomly selected available port in the range 5001 to |
| * 32767. |
| * |
| * @param protocol |
| * The protocol to check (either {@link #SOCKET} or {@link |
| * #JGROUPS}). |
| * |
| * @throws IllegalArgumentException |
| * <code>protocol</code> is unknown |
| */ |
| public static int getRandomAvailablePort(int protocol) { |
| return getRandomAvailablePort(protocol, getAddress(protocol)); |
| } |
| public static Keeper getRandomAvailablePortKeeper(int protocol) { |
| return getRandomAvailablePortKeeper(protocol, getAddress(protocol)); |
| } |
| |
| /** |
| * Returns a randomly selected available port in the provided range. |
| * |
| * @param protocol |
| * The protocol to check (either {@link #SOCKET} or {@link |
| * #JGROUPS}). |
| * |
| * @throws IllegalArgumentException |
| * <code>protocol</code> is unknown |
| */ |
| public static int getAvailablePortInRange(int rangeBase, int rangeTop, int protocol) { |
| return getAvailablePortInRange(protocol, getAddress(protocol), rangeBase, rangeTop); |
| } |
| |
| /** |
| * Returns a randomly selected available port in the range 5001 to |
| * 32767 that satisfies a modulus |
| * |
| * @param protocol |
| * The protocol to check (either {@link #SOCKET} or {@link |
| * #JGROUPS}). |
| * |
| * @throws IllegalArgumentException |
| * <code>protocol</code> is unknown |
| */ |
| public static int getRandomAvailablePortWithMod(int protocol,int mod) { |
| return getRandomAvailablePortWithMod(protocol, getAddress(protocol),mod); |
| } |
| |
| |
| /** |
| * Returns a randomly selected available port in the range 5001 to |
| * 32767. |
| * |
| * @param protocol |
| * The protocol to check (either {@link #SOCKET} or {@link |
| * #JGROUPS}). |
| * @param addr the bind-address or mcast address to use |
| * |
| * @throws IllegalArgumentException |
| * <code>protocol</code> is unknown |
| */ |
| public static int getRandomAvailablePort(int protocol, InetAddress addr) { |
| while (true) { |
| int port = getRandomWildcardBindPortNumber(); |
| if (isPortAvailable(port, protocol, addr)) { |
| // don't return the products default multicast port |
| if ( !(protocol == JGROUPS && port == DistributionConfig.DEFAULT_MCAST_PORT) ){ |
| return port; |
| } |
| } |
| } |
| } |
| public static Keeper getRandomAvailablePortKeeper(int protocol, InetAddress addr) { |
| while (true) { |
| int port = getRandomWildcardBindPortNumber(); |
| Keeper result = isPortKeepable(port, protocol, addr); |
| if (result != null) { |
| return result; |
| } |
| } |
| } |
| |
| /** |
| * Returns a randomly selected available port in the provided range. |
| * |
| * @param protocol |
| * The protocol to check (either {@link #SOCKET} or {@link |
| * #JGROUPS}). |
| * @param addr the bind-address or mcast address to use |
| * |
| * @throws IllegalArgumentException |
| * <code>protocol</code> is unknown |
| */ |
| public static int getAvailablePortInRange(int protocol, InetAddress addr, |
| int rangeBase, int rangeTop) { |
| for (int port = rangeBase; port <= rangeTop; port++) { |
| if (isPortAvailable(port, protocol, addr)) { |
| return port; |
| } |
| } |
| return -1; |
| } |
| |
| /** |
| * Returns a randomly selected available port in the range 5001 to |
| * 32767 that satisfies a modulus and the provided protocol |
| * |
| * @param protocol |
| * The protocol to check (either {@link #SOCKET} or {@link |
| * #JGROUPS}). |
| * @param addr the bind-address or mcast address to use |
| * |
| * @throws IllegalArgumentException |
| * <code>protocol</code> is unknown |
| */ |
| public static int getRandomAvailablePortWithMod(int protocol, InetAddress addr,int mod) { |
| while (true) { |
| int port = getRandomWildcardBindPortNumber(); |
| if (isPortAvailable(port, protocol, addr) && (port % mod)==0) { |
| return port; |
| } |
| } |
| } |
| |
| |
| public static java.util.Random rand; |
| |
| static { |
| boolean fast = Boolean.getBoolean("AvailablePort.fastRandom"); |
| if (fast) |
| rand = new Random(); |
| else |
| rand = new java.security.SecureRandom(); |
| } |
| |
| private static int getRandomWildcardBindPortNumber() { |
| int rangeBase; |
| int rangeTop; |
| // wcb port range on Windows is 1024..5000 (and Linux?) |
| // wcb port range on Solaris is 32768..65535 |
| // if (System.getProperty("os.name").equals("SunOS")) { |
| // rangeBase=32768; |
| // rangeTop=65535; |
| // } else { |
| // rangeBase=1024; |
| // rangeTop=5000; |
| // } |
| rangeBase = 20001; // 20000/udp is securid |
| rangeTop = 29999; // 30000/tcp is spoolfax |
| |
| return rand.nextInt(rangeTop-rangeBase) + rangeBase; |
| } |
| |
| private static int getRandomPortNumberInRange(int rangeBase, int rangeTop) { |
| return rand.nextInt(rangeTop-rangeBase) + rangeBase; |
| } |
| |
| public static int getRandomAvailablePortInRange(int rangeBase, int rangeTop, int protocol) { |
| int numberOfPorts = rangeTop - rangeBase; |
| //do "5 times the numberOfPorts" iterations to select a port number. This will ensure that |
| //each of the ports from given port range get a chance at least once |
| int numberOfRetrys = numberOfPorts * 5; |
| for (int i = 0; i < numberOfRetrys; i++) { |
| int port = rand.nextInt(numberOfPorts + 1) + rangeBase;//add 1 to numberOfPorts so that rangeTop also gets included |
| if (isPortAvailable(port, protocol, getAddress(protocol))) { |
| return port; |
| } |
| } |
| return -1; |
| } |
| |
| /** |
| * This class will keep an allocated port allocated until it is used. |
| * This makes the window smaller that can cause bug 46690 |
| * @author darrel |
| * |
| */ |
| public static class Keeper { |
| private final ServerSocket ss; |
| private final int port; |
| |
| public Keeper(ServerSocket ss, int port) { |
| this.ss = ss; |
| this.port = port; |
| } |
| public Keeper(ServerSocket ss, Integer port) { |
| this.ss = ss; |
| this.port = port != null ? port : 0; |
| } |
| |
| public int getPort() { |
| return this.port; |
| } |
| |
| /** |
| * Once you call this the socket will be freed and can then be reallocated by someone else. |
| */ |
| public void release() { |
| try { |
| if (this.ss != null) { |
| this.ss.close(); |
| } |
| } catch (IOException ignore) { |
| } |
| } |
| } |
| |
| /////////////////////// Main Program /////////////////////// |
| |
| private static final PrintStream out = System.out; |
| private static final PrintStream err = System.err; |
| |
| private static void usage(String s) { |
| err.println("\n** " + s + "\n"); |
| err.println("usage: java AvailablePort socket|jgroups [\"addr\" network-address] [port]"); |
| err.println(""); |
| err.println(LocalizedStrings.AvailablePort_THIS_PROGRAM_EITHER_PRINTS_WHETHER_OR_NOT_A_PORT_IS_AVAILABLE_FOR_A_GIVEN_PROTOCOL_OR_IT_PRINTS_OUT_AN_AVAILABLE_PORT_FOR_A_GIVEN_PROTOCOL.toLocalizedString()); |
| err.println(""); |
| System.exit(1); |
| } |
| |
| public static void main(String[] args) { |
| String protocolString = null; |
| String addrString = null; |
| String portString = null; |
| |
| for (int i = 0; i < args.length; i++) { |
| if (protocolString == null) { |
| protocolString = args[i]; |
| |
| } |
| else if (args[i].equals("addr")) { |
| addrString = args[++i]; |
| } |
| else if (portString == null) { |
| portString = args[i]; |
| } |
| else { |
| usage("Spurious command line: " + args[i]); |
| } |
| } |
| |
| int protocol; |
| |
| if (protocolString == null) { |
| usage("Missing protocol"); |
| return; |
| |
| } else if (protocolString.equalsIgnoreCase("socket")) { |
| protocol = SOCKET; |
| |
| } else if (protocolString.equalsIgnoreCase("javagroups") || |
| protocolString.equalsIgnoreCase("jgroups")) { |
| protocol = JGROUPS; |
| |
| } else { |
| usage("Unknown protocol: " + protocolString); |
| return; |
| } |
| |
| InetAddress addr = null; |
| if (addrString != null) { |
| try { |
| addr = InetAddress.getByName(addrString); |
| } |
| catch (Exception e) { |
| e.printStackTrace(); |
| System.exit(1); |
| } |
| } |
| |
| if (portString != null) { |
| int port; |
| try { |
| port = Integer.parseInt(portString); |
| |
| } catch (NumberFormatException ex) { |
| usage("Malformed port: " + portString); |
| return; |
| } |
| |
| out.println("\nPort " + port + " is " + |
| (isPortAvailable(port, protocol, addr) ? "" : "not ") + |
| "available for a " + protocolString + |
| " connection\n"); |
| |
| } else { |
| out.println("\nRandomly selected " + protocolString + " port: " |
| + getRandomAvailablePort(protocol, addr) + "\n"); |
| |
| } |
| |
| } |
| |
| } |