/* * 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. */ package org.googlecode.perftrace.log4j.net; import java.io.File; import java.net.InetAddress; import java.net.ServerSocket; import java.net.Socket; import java.util.Hashtable; import org.googlecode.perftrace.log4j.Hierarchy; import org.googlecode.perftrace.log4j.Level; import org.googlecode.perftrace.log4j.LogManager; import org.googlecode.perftrace.log4j.Logger; import org.googlecode.perftrace.log4j.PropertyConfigurator; import org.googlecode.perftrace.log4j.spi.LoggerRepository; import org.googlecode.perftrace.log4j.spi.RootLogger; /** A {@link SocketNode} based server that uses a different hierarchy for each client. <pre> <b>Usage:</b> java org.apache.log4j.net.SocketServer port configFile configDir where <b>port</b> is a part number where the server listens, <b>configFile</b> is a configuration file fed to the {@link PropertyConfigurator} and <b>configDir</b> is a path to a directory containing configuration files, possibly one for each client host. </pre> <p>The <code>configFile</code> is used to configure the log4j default hierarchy that the <code>SocketServer</code> will use to report on its actions. <p>When a new connection is opened from a previously unknown host, say <code>foo.bar.net</code>, then the <code>SocketServer</code> will search for a configuration file called <code>foo.bar.net.lcf</code> under the directory <code>configDir</code> that was passed as the third argument. If the file can be found, then a new hierarchy is instantiated and configured using the configuration file <code>foo.bar.net.lcf</code>. If and when the host <code>foo.bar.net</code> opens another connection to the server, then the previously configured hierarchy is used. <p>In case there is no file called <code>foo.bar.net.lcf</code> under the directory <code>configDir</code>, then the <em>generic</em> hierarchy is used. The generic hierarchy is configured using a configuration file called <code>generic.lcf</code> under the <code>configDir</code> directory. If no such file exists, then the generic hierarchy will be identical to the log4j default hierarchy. <p>Having different client hosts log using different hierarchies ensures the total independence of the clients with respect to their logging settings. <p>Currently, the hierarchy that will be used for a given request depends on the IP address of the client host. For example, two separate applicatons running on the same host and logging to the same server will share the same hierarchy. This is perfectly safe except that it might not provide the right amount of independence between applications. The <code>SocketServer</code> is intended as an example to be enhanced in order to implement more elaborate policies. @author Ceki Gülcü @since 1.0 */ public class SocketServer { static String GENERIC = "generic"; static String CONFIG_FILE_EXT = ".lcf"; static Logger cat = Logger.getLogger(SocketServer.class); static SocketServer server; static int port; // key=inetAddress, value=hierarchy Hashtable hierarchyMap; LoggerRepository genericHierarchy; File dir; public static void main(String argv[]) { if(argv.length == 3) init(argv[0], argv[1], argv[2]); else usage("Wrong number of arguments."); try { cat.info("Listening on port " + port); ServerSocket serverSocket = new ServerSocket(port); while(true) { cat.info("Waiting to accept a new client."); Socket socket = serverSocket.accept(); InetAddress inetAddress = socket.getInetAddress(); cat.info("Connected to client at " + inetAddress); LoggerRepository h = (LoggerRepository) server.hierarchyMap.get(inetAddress); if(h == null) { h = server.configureHierarchy(inetAddress); } cat.info("Starting new socket node."); new Thread(new SocketNode(socket, h)).start(); } } catch(Exception e) { e.printStackTrace(); } } static void usage(String msg) { System.err.println(msg); System.err.println( "Usage: java " +SocketServer.class.getName() + " port configFile directory"); System.exit(1); } static void init(String portStr, String configFile, String dirStr) { try { port = Integer.parseInt(portStr); } catch(java.lang.NumberFormatException e) { e.printStackTrace(); usage("Could not interpret port number ["+ portStr +"]."); } PropertyConfigurator.configure(configFile); File dir = new File(dirStr); if(!dir.isDirectory()) { usage("["+dirStr+"] is not a directory."); } server = new SocketServer(dir); } public SocketServer(File directory) { this.dir = directory; hierarchyMap = new Hashtable(11); } // This method assumes that there is no hiearchy for inetAddress // yet. It will configure one and return it. LoggerRepository configureHierarchy(InetAddress inetAddress) { cat.info("Locating configuration file for "+inetAddress); // We assume that the toSting method of InetAddress returns is in // the format hostname/d1.d2.d3.d4 e.g. torino/192.168.1.1 String s = inetAddress.toString(); int i = s.indexOf("/"); if(i == -1) { cat.warn("Could not parse the inetAddress ["+inetAddress+ "]. Using default hierarchy."); return genericHierarchy(); } else { String key = s.substring(0, i); File configFile = new File(dir, key+CONFIG_FILE_EXT); if(configFile.exists()) { Hierarchy h = new Hierarchy(new RootLogger(Level.DEBUG)); hierarchyMap.put(inetAddress, h); new PropertyConfigurator().doConfigure(configFile.getAbsolutePath(), h); return h; } else { cat.warn("Could not find config file ["+configFile+"]."); return genericHierarchy(); } } } LoggerRepository genericHierarchy() { if(genericHierarchy == null) { File f = new File(dir, GENERIC+CONFIG_FILE_EXT); if(f.exists()) { genericHierarchy = new Hierarchy(new RootLogger(Level.DEBUG)); new PropertyConfigurator().doConfigure(f.getAbsolutePath(), genericHierarchy); } else { cat.warn("Could not find config file ["+f+ "]. Will use the default hierarchy."); genericHierarchy = LogManager.getLoggerRepository(); } } return genericHierarchy; } }