package org.apache.maven.wagon.tck.http.fixture;
/*
* 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 org.apache.log4j.Logger;
import org.mortbay.jetty.Connector;
import org.mortbay.jetty.Handler;
import org.mortbay.jetty.Server;
import org.mortbay.jetty.handler.DefaultHandler;
import org.mortbay.jetty.handler.HandlerCollection;
import org.mortbay.jetty.nio.SelectChannelConnector;
import org.mortbay.jetty.security.Constraint;
import org.mortbay.jetty.security.ConstraintMapping;
import org.mortbay.jetty.security.HashUserRealm;
import org.mortbay.jetty.security.SecurityHandler;
import org.mortbay.jetty.security.SslSocketConnector;
import org.mortbay.jetty.servlet.AbstractSessionManager;
import org.mortbay.jetty.servlet.FilterHolder;
import org.mortbay.jetty.servlet.FilterMapping;
import org.mortbay.jetty.servlet.ServletHolder;
import org.mortbay.jetty.servlet.SessionHandler;
import org.mortbay.jetty.webapp.WebAppContext;
import javax.servlet.Filter;
import javax.servlet.Servlet;
import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import static org.apache.maven.wagon.tck.http.util.TestUtil.getResource;
public class ServerFixture
{
private static Logger logger = Logger.getLogger( ServerFixture.class );
public static final String SERVER_ROOT_RESOURCE_PATH = "default-server-root";
// it seems that some JDKs have a problem if you use different key stores
// so we gonna reuse the keystore which is is used in the wagon implementations already
public static final String SERVER_SSL_KEYSTORE_RESOURCE_PATH = "ssl/keystore";
public static final String SERVER_SSL_KEYSTORE_PASSWORD = "wagonhttp";
public static final String SERVER_HOST = "localhost";
private final Server server;
private final WebAppContext webappContext;
private final HashUserRealm securityRealm;
private final SecurityHandler securityHandler;
private int filterCount = 0;
private int httpPort;
public ServerFixture( final boolean ssl )
throws URISyntaxException, IOException
{
server = new Server();
if ( ssl )
{
SslSocketConnector connector = new SslSocketConnector();
String keystore = getResource( SERVER_SSL_KEYSTORE_RESOURCE_PATH ).getAbsolutePath();
Logger.getLogger( ServerFixture.class ).info( "TCK Keystore path: " + keystore );
System.setProperty( "javax.net.ssl.keyStore", keystore );
System.setProperty( "javax.net.ssl.trustStore", keystore );
// connector.setHost( SERVER_HOST );
//connector.setPort( port );
connector.setKeystore( keystore );
connector.setPassword( SERVER_SSL_KEYSTORE_PASSWORD );
connector.setKeyPassword( SERVER_SSL_KEYSTORE_PASSWORD );
server.addConnector( connector );
}
else
{
Connector connector = new SelectChannelConnector();
connector.setHost( "localhost" );
//connector.setPort( port );
server.addConnector( connector );
}
Constraint constraint = new Constraint();
constraint.setName( Constraint.__BASIC_AUTH );
constraint.setRoles( new String[]{ "allowed" } );
constraint.setAuthenticate( true );
ConstraintMapping cm = new ConstraintMapping();
cm.setConstraint( constraint );
cm.setPathSpec( "/protected/*" );
securityHandler = new SecurityHandler();
securityRealm = new HashUserRealm( "Test Server" );
securityHandler.setUserRealm( securityRealm );
securityHandler.setConstraintMappings( new ConstraintMapping[]{ cm } );
webappContext = new WebAppContext();
webappContext.setContextPath( "/" );
File base = getResource( SERVER_ROOT_RESOURCE_PATH );
logger.info( "docroot: " + base );
webappContext.setWar( base.getAbsolutePath() );
webappContext.addHandler( securityHandler );
SessionHandler sessionHandler = webappContext.getSessionHandler();
( (AbstractSessionManager) sessionHandler.getSessionManager() ).setUsingCookies( false );
HandlerCollection handlers = new HandlerCollection();
handlers.setHandlers( new Handler[]{ webappContext, new DefaultHandler() } );
server.setHandler( handlers );
}
public void addFilter( final String pathSpec, final Filter filter )
{
String name = "filter" + filterCount++;
FilterMapping fm = new FilterMapping();
fm.setPathSpec( pathSpec );
fm.setFilterName( name );
FilterHolder fh = new FilterHolder( filter );
fh.setName( name );
webappContext.getServletHandler().addFilter( fh, fm );
}
public void addServlet( final String pathSpec, final Servlet servlet )
{
webappContext.getServletHandler().addServletWithMapping( new ServletHolder( servlet ), pathSpec );
}
public void addUser( final String user, final String password )
{
securityRealm.put( user, password );
securityRealm.addUserToRole( user, "allowed" );
}
public Server getServer()
{
return server;
}
public WebAppContext getWebappContext()
{
return webappContext;
}
public void stop()
throws Exception
{
if ( server != null )
{
server.stop();
}
}
public void start()
throws Exception
{
if ( server.isStarted() || server.isRunning() )
{
return;
}
server.start();
int total = 0;
while ( total < 3000 && !server.isStarted() )
{
server.wait( 10 );
total += 10;
}
if ( !server.isStarted() )
{
throw new IllegalStateException( "Server didn't start in: " + total + "ms." );
}
this.httpPort = server.getConnectors()[0].getLocalPort();
}
public int getHttpPort()
{
return httpPort;
}
}