| package org.apache.jcs.auxiliary.remote; |
| |
| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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. |
| */ |
| |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.net.InetSocketAddress; |
| import java.net.ServerSocket; |
| import java.net.Socket; |
| import java.rmi.RMISecurityManager; |
| import java.rmi.RemoteException; |
| import java.rmi.registry.LocateRegistry; |
| import java.rmi.registry.Registry; |
| import java.rmi.server.RMISocketFactory; |
| import java.util.Enumeration; |
| import java.util.Properties; |
| |
| import org.apache.commons.logging.Log; |
| import org.apache.commons.logging.LogFactory; |
| |
| /** |
| * This class provides some basic utilities for doing things such as starting the registry properly. |
| */ |
| public class RemoteUtils |
| { |
| /** The logger. */ |
| private final static Log log = LogFactory.getLog( RemoteUtils.class ); |
| |
| /** No instances please. */ |
| private RemoteUtils() |
| { |
| super(); |
| } |
| |
| /** |
| * Creates and exports a registry on the specified port of the local host. |
| * <p> |
| * @param port |
| * @return the port the registry was started on |
| * @throws RemoteException |
| */ |
| public static int createRegistry( int port ) |
| throws RemoteException |
| { |
| if ( log.isInfoEnabled() ) |
| { |
| log.info( "createRegistry> setting security manager" ); |
| } |
| System.setSecurityManager( new RMISecurityManager() ); |
| |
| if ( port < 1024 ) |
| { |
| if ( log.isInfoEnabled() ) |
| { |
| log.info( "Port chosen was less than 1024, will use default [" + Registry.REGISTRY_PORT + "] instead." ); |
| } |
| port = Registry.REGISTRY_PORT; |
| } |
| |
| if ( log.isInfoEnabled() ) |
| { |
| log.info( "createRegistry> creating registry on port [" + port + "]" ); |
| } |
| LocateRegistry.createRegistry( port ); |
| return port; |
| } |
| |
| /** |
| * Loads properties for the named props file. |
| * <p> |
| * @param propFile |
| * @return The properties object for the file |
| * @throws IOException |
| */ |
| public static Properties loadProps( String propFile ) |
| throws IOException |
| { |
| InputStream is = RemoteUtils.class.getResourceAsStream( propFile ); |
| Properties props = new Properties(); |
| try |
| { |
| props.load( is ); |
| if ( log.isDebugEnabled() ) |
| { |
| log.debug( "props.size=" + props.size() ); |
| } |
| |
| if ( log.isDebugEnabled() ) |
| { |
| if ( props != null ) |
| { |
| Enumeration en = props.keys(); |
| StringBuffer buf = new StringBuffer(); |
| while ( en.hasMoreElements() ) |
| { |
| String key = (String) en.nextElement(); |
| buf.append( "\n" + key + " = " + props.getProperty( key ) ); |
| } |
| log.debug( buf.toString() ); |
| } |
| else |
| { |
| log.debug( "props is null" ); |
| } |
| } |
| |
| } |
| catch ( Exception ex ) |
| { |
| log.error( "Error loading remote properties, for file name [" + propFile + "]", ex ); |
| } |
| finally |
| { |
| if ( is != null ) |
| { |
| is.close(); |
| } |
| } |
| return props; |
| } |
| |
| /** |
| * Configure a custom socket factory to set the timeout value. This sets the global socket |
| * factory. It's used only if a custom factory is not configured for the specific object. |
| * <p> |
| * @param timeoutMillis |
| */ |
| public static void configureGlobalCustomSocketFactory( final int timeoutMillis ) |
| { |
| try |
| { |
| // Don't set a socket factory if the setting is -1 |
| if ( timeoutMillis > 0 ) |
| { |
| if ( log.isInfoEnabled() ) |
| { |
| log.info( "RmiSocketFactoryTimeoutMillis [" + timeoutMillis + "]. " |
| + " Configuring a custom socket factory." ); |
| } |
| |
| // use this socket factory to add a timeout. |
| RMISocketFactory.setSocketFactory( new RMISocketFactory() |
| { |
| public Socket createSocket( String host, int port ) |
| throws IOException |
| { |
| Socket socket = new Socket(); |
| socket.setSoTimeout( timeoutMillis ); |
| socket.setSoLinger( false, 0 ); |
| socket.connect( new InetSocketAddress( host, port ), timeoutMillis ); |
| return socket; |
| } |
| |
| public ServerSocket createServerSocket( int port ) |
| throws IOException |
| { |
| return new ServerSocket( port ); |
| } |
| } ); |
| } |
| } |
| catch ( Exception e ) |
| { |
| // Only try to do it once. Otherwise we |
| // Generate errors for each region on construction. |
| RMISocketFactory factoryInUse = RMISocketFactory.getSocketFactory(); |
| if ( factoryInUse != null && !factoryInUse.getClass().getName().startsWith( "org.apache.jcs" ) ) |
| { |
| log.info( "Could not create new custom socket factory. " + e.getMessage() + " Factory in use = " |
| + RMISocketFactory.getSocketFactory() ); |
| } |
| } |
| } |
| } |