package org.apache.maven.wagon.providers.ssh;
/*
* 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.mina.core.session.IoSession;
import org.apache.sshd.SshServer;
import org.apache.sshd.common.Session;
import org.apache.sshd.common.session.AbstractSession;
import org.apache.sshd.server.Command;
import org.apache.sshd.server.CommandFactory;
import org.apache.sshd.server.FileSystemFactory;
import org.apache.sshd.server.FileSystemView;
import org.apache.sshd.server.SshFile;
import org.apache.sshd.server.auth.UserAuthPassword;
import org.apache.sshd.server.auth.UserAuthPublicKey;
import org.apache.sshd.server.filesystem.NativeSshFile;
import org.apache.sshd.server.keyprovider.PEMGeneratorHostKeyProvider;
import org.apache.sshd.server.session.SessionFactory;
import org.apache.sshd.server.shell.ProcessShellFactory;
import org.codehaus.plexus.util.FileUtils;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
/**
* @author Olivier Lamy
*/
public class SshServerEmbedded
{
private String wagonProtocol;
private int port;
private SshServer sshd;
private List<String> sshKeysResources = new ArrayList<String>();
public TestPublickeyAuthenticator publickeyAuthenticator;
public TestPasswordAuthenticator passwordAuthenticator = new TestPasswordAuthenticator();
private boolean keyAuthz;
/**
* @param wagonProtocol scp scpexe
* @param sshKeysResources paths in the classlaoder with ssh keys
*/
public SshServerEmbedded( String wagonProtocol, List<String> sshKeysResources, boolean keyAuthz )
{
this.wagonProtocol = wagonProtocol;
this.sshKeysResources = sshKeysResources;
this.sshd = SshServer.setUpDefaultServer();
//this.sshd.setKeyExchangeFactories( );
this.keyAuthz = keyAuthz;
publickeyAuthenticator = new TestPublickeyAuthenticator( this.keyAuthz );
}
/**
* @return random port used
*/
public int start()
throws IOException
{
sshd.setPort( 0 );
sshd.setUserAuthFactories( Arrays.asList( new UserAuthPublicKey.Factory(), new UserAuthPassword.Factory() ) );
sshd.setPublickeyAuthenticator( this.publickeyAuthenticator );
sshd.setPasswordAuthenticator( this.passwordAuthenticator );
sshd.setUserAuthFactories( Arrays.asList( new UserAuthPublicKey.Factory(), new UserAuthPassword.Factory() ) );
//ResourceKeyPairProvider resourceKeyPairProvider =
// new ResourceKeyPairProvider( sshKeysResources.toArray( new String[sshKeysResources.size()] ) );
File path = new File( "target/keys" );
path.mkdirs();
path = new File( path, "simple.key" );
path.delete();
PEMGeneratorHostKeyProvider provider = new PEMGeneratorHostKeyProvider();
provider.setAlgorithm( "RSA" );
provider.setKeySize( 1024 );
provider.setPath( path.getPath() );
sshd.setKeyPairProvider( provider );
SessionFactory sessionFactory = new SessionFactory()
{
@Override
protected AbstractSession doCreateSession( IoSession ioSession )
throws Exception
{
return super.doCreateSession( ioSession );
}
};
sshd.setSessionFactory( sessionFactory );
//sshd.setFileSystemFactory( );
final ProcessShellFactory processShellFactory =
new ProcessShellFactory( new String[]{ "/bin/sh", "-i", "-l" } );
sshd.setShellFactory( processShellFactory );
CommandFactory delegateCommandFactory = new CommandFactory()
{
public Command createCommand( String command )
{
return new ShellCommand( command );
}
};
ScpCommandFactory commandFactory = new ScpCommandFactory( delegateCommandFactory );
sshd.setCommandFactory( commandFactory );
FileSystemFactory fileSystemFactory = new FileSystemFactory()
{
public FileSystemView createFileSystemView( Session session )
throws IOException
{
return new FileSystemView()
{
public SshFile getFile( String file )
{
file = file.replace( "\\", "" );
file = file.replace( "\"", "" );
File f = new File( FileUtils.normalize( file ) );
return new SshServerEmbedded.TestSshFile( f.getAbsolutePath(), f,
System.getProperty( "user.name" ) );
}
public SshFile getFile( SshFile baseDir, String file )
{
file = file.replace( "\\", "" );
file = file.replace( "\"", "" );
File f = new File( FileUtils.normalize( file ) );
return new SshServerEmbedded.TestSshFile( f.getAbsolutePath(), f,
System.getProperty( "user.name" ) );
}
};
}
};
sshd.setNioWorkers( 0 );
//sshd.setScheduledExecutorService( );
sshd.setFileSystemFactory( fileSystemFactory );
sshd.start();
this.port = sshd.getPort();
return this.port;
}
public void stop()
throws InterruptedException
{
sshd.stop( Boolean.getBoolean( "sshd.stopImmediatly" ) );
}
public int getPort()
{
return port;
}
public static class TestSshFile
extends NativeSshFile
{
public TestSshFile( String fileName, File file, String userName )
{
super( FileUtils.normalize( fileName ), file, userName );
}
}
}