tunnelresourcetype.java

这是linux下ssl vpn的实现程序

/*
 *  SSL-Explorer
 *
 *  Copyright (C) 2003-2006 3SP LTD. All Rights Reserved
 *
 *  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., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
			
package com.sslexplorer.tunnels;

import java.util.Collection;
import java.util.Iterator;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.struts.util.LabelValueBean;

import com.sslexplorer.core.CoreAttributeConstants;
import com.sslexplorer.core.CoreEvent;
import com.sslexplorer.core.CoreEventConstants;
import com.sslexplorer.core.CoreServlet;
import com.sslexplorer.forwarding.VPNListeningSocket;
import com.sslexplorer.navigation.FavoriteResourceType;
import com.sslexplorer.navigation.WrappedFavoriteItem;
import com.sslexplorer.policyframework.DefaultResourceType;
import com.sslexplorer.policyframework.PolicyConstants;
import com.sslexplorer.policyframework.Resource;
import com.sslexplorer.policyframework.ResourceChangeEvent;
import com.sslexplorer.policyframework.ResourceDeleteEvent;
import com.sslexplorer.security.SessionInfo;
import com.sslexplorer.security.VPNSession;
import com.sslexplorer.tunnels.forms.TunnelItem;

/**
 * Implementation of a {@link com.sslexplorer.policyframework.ResourceType}
 * for <i>SSL Tunnel</i> resources.
 * 
 * @author Brett Smith <a href="mailto: [e-mail]">&lt;[e-mail]&gt;</a>
 * @version $Revision: 1.5 $
 */
public class TunnelResourceType extends DefaultResourceType implements FavoriteResourceType {

    final static Log log = LogFactory.getLog(TunnelResourceType.class);

    /**
     * Constructor
     */
    public TunnelResourceType() {
        super(PolicyConstants.SSL_TUNNEL_RESOURCE_TYPE_ID, "policyframework", PolicyConstants.DELEGATION_CLASS);
    }
    
    /* (non-Javadoc)
     * @see com.sslexplorer.navigation.FavoriteResourceType#createWrappedFavoriteItem(int, javax.servlet.http.HttpServletRequest, java.lang.String)
     */
    public WrappedFavoriteItem createWrappedFavoriteItem(int resourceId, HttpServletRequest request, String type) throws Exception {
        VPNSession session = CoreServlet.getServlet().getLogonController().getPrimaryVPNSession(
            CoreServlet.getServlet().getLogonController().getVPNSessionsByLogon(request));
        Collection listeningSockets = session == null ? null : session.getListeningSockets();
        Tunnel tunnel = CoreServlet.getServlet().getSystemDatabase().getTunnel(resourceId);
        if(tunnel == null) {
            return null;
        }
        VPNListeningSocket socket = null;
        if (tunnel.getResourceId() != -1) {
            if (listeningSockets != null) {
                for (Iterator it2 = listeningSockets.iterator(); it2.hasNext() && socket == null;) {
                    VPNListeningSocket vpns = (VPNListeningSocket) it2.next();
                    if (vpns.getTunnel().getResourceId() == tunnel.getResourceId()) {
                        socket = vpns;
                    }
                }
            }
        }
        if (tunnel != null) {
            if (socket != null) {
                return new WrappedFavoriteItem(new TunnelItem(socket, session, null), type);
            } else {
                return new WrappedFavoriteItem(new TunnelItem(tunnel, session, null), type);
            }
        } 
        return null;
    }

    /* (non-Javadoc)
     * @see com.sslexplorer.navigation.FavoriteResourceType#getResourceById(int)
     */
    public Resource getResourceById(int resourceId) throws Exception {
        return CoreServlet.getServlet().getSystemDatabase().getTunnel(resourceId);
    }

    /* (non-Javadoc)
     * @see com.sslexplorer.navigation.FavoriteResourceType#getResourceByName(java.lang.String)
     */
    public Resource getResourceByName(String resourceName) throws Exception {
        return CoreServlet.getServlet().getSystemDatabase().getTunnel(resourceName);
    }


    /* (non-Javadoc)
     * @see com.sslexplorer.boot.policyframework.ResourceType#removeResource(int, com.sslexplorer.security.SessionInfo)
     */
    public Resource removeResource(int resourceId, SessionInfo session) throws Exception {
        throw new Exception("The remove is implemented in the UpdateTunnelAction");
//        Tunnel resource = null;
//        try {
//            resource = CoreServlet.getServlet().getSystemDatabase().removeTunnel(resourceId);
//            // #ifdef XTRA
//            CoreServlet.getServlet().fireCoreEvent(addTunnelAttributes(
//                new ResourceDeleteEvent(this, CoreEventConstants.REMOVE_TUNNEL, resource, session,
//                    CoreEvent.STATE_SUCCESSFUL), resource));
//            // #endif
//            return resource;
//        } catch (Exception e) {
//            // #ifdef XTRA
//            CoreServlet.getServlet().fireCoreEvent(addTunnelAttributes(
//                new ResourceDeleteEvent(this, CoreEventConstants.REMOVE_TUNNEL, null, session,
//                    CoreEvent.STATE_UNSUCCESSFUL), resource));
//            // #endif
//            throw e;
//        }
    }

    /* (non-Javadoc)
     * @see com.sslexplorer.boot.policyframework.ResourceType#updateResource(com.sslexplorer.boot.policyframework.Resource, com.sslexplorer.security.SessionInfo)
     */
    public void updateResource(Resource resource, SessionInfo session) throws Exception {
        Tunnel t = null;
        try {
            t = (Tunnel)resource;
            CoreServlet.getServlet().getSystemDatabase().updateTunnel(
                t.getResourceId(), t.getResourceName(), t.getResourceDescription(),
                t.getType(), t.isAutoStart(), t.getTransport(), t.getUsername(),
                t.getSourcePort(), t.getDestination(), t.isAllowExternalHosts(), t.getParentResourcePermission());
        } catch (Exception e) {
            throw e;
        }
    }
    
    CoreEvent addTunnelAttributes(CoreEvent evt, Tunnel tunnel) {
        return evt;
    }

}