yacy_search_server/htroot/Connections_p.java

//Connections_p.java
//-----------------------
//part of YaCy
//(C) by Michael Peter Christen; mc@yacy.net
//first published on http://www.anomic.de
//Frankfurt, Germany, 2004, 2005
//
//$LastChangedDate$
//$LastChangedRevision$
//$LastChangedBy$
//
//This program is free software; you can redistribute it and/or modify
//it under the terms of the GNU General Public License as published by
//the Free Software Foundation; either version 2 of the License, or
//(at your option) any later version.
//
//This program is distributed in the hope that it will be useful,
//but WITHOUT ANY WARRANTY; without even the implied warranty of
//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//GNU General Public License for more details.
//
//You should have received a copy of the GNU General Public License
//along with this program; if not, write to the Free Software
//Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

//You must compile this file with
//javac -classpath .:../classes Network.java
//if the shell's current path is HTROOT

import java.io.UnsupportedEncodingException;
import java.net.InetAddress;
import java.net.URLEncoder;
import java.util.Properties;
import java.util.Set;

import de.anomic.http.HttpConnectionInfo;
import de.anomic.http.JakartaCommonsHttpClient;
import de.anomic.http.httpRequestHeader;
import de.anomic.http.httpd;
import de.anomic.kelondro.kelondroDate;
import de.anomic.plasma.plasmaSwitchboard;
import de.anomic.server.serverCore;
import de.anomic.server.serverHandler;
import de.anomic.server.serverObjects;
import de.anomic.server.serverSwitch;
import de.anomic.server.serverThread;
import de.anomic.server.serverCore.Session;
import de.anomic.urlRedirector.urlRedirectord;
import de.anomic.yacy.yacySeed;

public final class Connections_p {    
    
    public static serverObjects respond(final httpRequestHeader header, final serverObjects post, final serverSwitch<?> env) {
        // return variable that accumulates replacements
        final plasmaSwitchboard sb = (plasmaSwitchboard) env;
        final serverObjects prop = new serverObjects();
         
        
        // get the virtualHost string
        final String virtualHost = sb.getConfig("fileHost","localhost");
        
        // get the serverCore thread
        final serverThread httpd = sb.getThread("10_httpd");
        
        /* waiting for all threads to finish */
        int threadCount  = serverCore.sessionThreadGroup.activeCount();    
        final Thread[] threadList = new Thread[((serverCore) httpd).getJobCount()];     
        threadCount = serverCore.sessionThreadGroup.enumerate(threadList);              
        
        // determines if name lookup should be done or not 
        boolean doNameLookup = false;
        if (post != null) {  
            if (post.containsKey("nameLookup") && post.get("nameLookup","true").equals("true")) {
                doNameLookup = true;
            }
            if (post.containsKey("closeSession")) {
                final String sessionName = post.get("closeSession",null);
                if (sessionName != null) {
                    for ( int currentThreadIdx = 0; currentThreadIdx < threadCount; currentThreadIdx++ )  {                        
                        final Thread currentThread = threadList[currentThreadIdx];
                        if (
                                (currentThread != null) && 
                                (currentThread instanceof serverCore.Session) && 
                                (currentThread.isAlive()) &&
                                (currentThread.getName().equals(sessionName))
                        ){
                            // trying to gracefull stop session
                            ((Session)currentThread).setStopped(true);
                            try { Thread.sleep(100); } catch (final InterruptedException ex) {}
                            
                            // trying to interrupt session
                            if (currentThread.isAlive()) {
                                currentThread.interrupt();
                                try { Thread.sleep(100); } catch (final InterruptedException ex) {}
                            } 
                            
                            // trying to close socket
                            if (currentThread.isAlive()) {
                                ((Session)currentThread).close();
                            }
                            
                            // waiting for session to finish
                            if (currentThread.isAlive()) {
                                try { currentThread.join(500); } catch (final InterruptedException ex) {}
                            }
                        }
                    }
                }
                prop.put("LOCATION","");
                return prop;                
            }
        }  
        
        int idx = 0, numActiveRunning = 0, numActivePending = 0;
        boolean dark = true;
        for ( int currentThreadIdx = 0; currentThreadIdx < threadCount; currentThreadIdx++ )  {
            final Thread currentThread = threadList[currentThreadIdx];
            if ((currentThread != null) && (currentThread instanceof serverCore.Session) && (currentThread.isAlive())) {
                // getting the session object
                final Session currentSession = ((Session)currentThread);
                
                // getting the session runtime
                final long sessionTime = currentSession.getTime();
                
                // getting the request command line
                boolean blockingRequest = false;
                String commandLine = currentSession.getCommandLine();
                if (commandLine == null) blockingRequest = true;                
                final int commandCount = currentSession.getCommandCount();
                
                // getting the source ip address and port
                final InetAddress userAddress = currentSession.getUserAddress();
                final int userPort = currentSession.getUserPort();
                if (userAddress == null) continue;
                
                final boolean isSSL = currentSession.isSSL();
                
                String dest = null;
                String prot = null;
                final serverHandler cmdObj = currentSession.getCommandObj();
                if (cmdObj instanceof httpd) {
                    prot = isSSL ? "https":"http";
                    
                    // getting the http command object
                    final httpd currentHttpd =  (httpd)cmdObj;
                    
                    // getting the connection properties of this session
                    final Properties conProp = (Properties) currentHttpd.getConProp().clone();
                    
                    // getting the destination host
                    dest = conProp.getProperty(httpRequestHeader.CONNECTION_PROP_HOST);
                    if (dest==null)continue;
                } else if (cmdObj instanceof urlRedirectord) {
                    prot = "urlRedirector";
                    
                    final urlRedirectord urlRedir = (urlRedirectord)cmdObj;
                    commandLine = urlRedir.getURL();
                }                
                
                if ((dest != null) && (dest.equals(virtualHost))) dest = sb.webIndex.seedDB.mySeed().getName() + ".yacy";
                
                // determining if the source is a yacy host
                yacySeed seed = null;
                if (doNameLookup) {
                    seed = sb.webIndex.seedDB.lookupByIP(userAddress,true,false,false);
                    if (seed != null) {
                        if ((seed.hash.equals(sb.webIndex.seedDB.mySeed().hash)) && 
                                (!seed.get(yacySeed.PORT,"").equals(Integer.toString(userPort)))) {
                            seed = null;
                        }
                    }
                }
                
                prop.put("list_" + idx + "_dark", dark ? "1" : "0");
                dark=!dark;
                try {
                    prop.put("list_" + idx + "_sessionID",URLEncoder.encode(currentSession.getName(),"UTF8"));
                } catch (final UnsupportedEncodingException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                prop.putHTML("list_" + idx + "_sessionName", currentSession.getName());
                prop.put("list_" + idx + "_proto", prot);
                if (sessionTime > 1000*60) {
                    prop.put("list_" + idx + "_ms", "0");
                    prop.put("list_" + idx + "_ms_duration",kelondroDate.formatInterval(sessionTime));
                } else {
                    prop.put("list_" + idx + "_ms", "1");
                    prop.putNum("list_" + idx + "_ms_duration", sessionTime);
                }
                prop.putHTML("list_" + idx + "_source",(seed!=null)?seed.getName()+".yacy":userAddress.getHostAddress()+":"+userPort);
                prop.putHTML("list_" + idx + "_dest",(dest==null)?"-":dest);
                if (blockingRequest) {
                    prop.put("list_" + idx + "_running", "0");
                    prop.putNum("list_" + idx + "_running_reqNr", commandCount+1);
                    numActivePending++;
                } else {
                    prop.put("list_" + idx + "_running", "1");
                    prop.put("list_" + idx + "_running_command", commandLine==null ? "" :commandLine);
                    numActiveRunning++;
                }
                prop.putNum("list_" + idx + "_used", commandCount);
                idx++;                
            }
        }     
        prop.put("list", idx);
        
        prop.putNum("numMax", ((serverCore)httpd).getMaxSessionCount());
        prop.putNum("numActiveRunning", numActiveRunning);
        prop.putNum("numActivePending", numActivePending);
        
        // client sessions
        final Set<HttpConnectionInfo> allConnections = HttpConnectionInfo.getAllConnections();
        // TODO sorting
//        Arrays.sort(a, httpc.connectionTimeComparatorInstance);
        int c = 0;
        synchronized (allConnections) {
        for (final HttpConnectionInfo conInfo: allConnections) {
            prop.put("clientList_" + c + "_clientProtocol", conInfo.getProtocol());
            prop.putNum("clientList_" + c + "_clientLifetime", conInfo.getLifetime());
            prop.putNum("clientList_" + c + "_clientIdletime", conInfo.getIdletime());
            prop.put("clientList_" + c + "_clientTargetHost", conInfo.getTargetHost());
            prop.putHTML("clientList_" + c + "_clientCommand", conInfo.getCommand());
            prop.put("clientList_" + c + "_clientID", conInfo.getID());
            c++;
        }
        }
        prop.put("clientList", c);
        prop.put("clientActive", JakartaCommonsHttpClient.connectionCount());
        
        // return rewrite values for templates
        return prop;
    }
}