com.github.fonimus.ssh.shell.SshShellConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-ssh-shell-starter Show documentation
The newest version!
package com.github.fonimus.ssh.shell;

import java.io.IOException;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;

import org.apache.sshd.server.SshServer;
import org.apache.sshd.server.auth.password.PasswordAuthenticator;
import org.apache.sshd.server.auth.pubkey.RejectAllPublickeyAuthenticator;
import org.apache.sshd.server.keyprovider.SimpleGeneratorHostKeyProvider;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * Ssh shell configuration
 */
@Configuration
public class SshShellConfiguration {

	private static final Logger LOGGER = LoggerFactory.getLogger(SshShellConfiguration.class);

	private SshShellProperties properties;

	private SshShellCommandFactory shellCommandFactory;

	private PasswordAuthenticator passwordAuthenticator;

	public SshShellConfiguration(SshShellProperties properties,
			SshShellCommandFactory shellCommandFactory,
			PasswordAuthenticator passwordAuthenticator) {
		this.properties = properties;
		this.shellCommandFactory = shellCommandFactory;
		this.passwordAuthenticator = passwordAuthenticator;
	}

	/**
	 * Start ssh server
	 *
	 * @throws IOException in case of error
	 */
	@PostConstruct
	public void startServer() throws IOException {
		sshServer().start();
		LOGGER.info("Ssh server started [{}:{}]", properties.getHost(), properties.getPort());
	}

	/**
	 * Stop ssh server
	 *
	 * @throws IOException in case of error
	 */
	@PreDestroy
	public void stopServer() throws IOException {
		sshServer().stop();
	}

	/**
	 * Construct ssh server thanks to ssh shell properties
	 *
	 * @return ssh server
	 */
	@Bean
	public SshServer sshServer() {
		SshServer server = SshServer.setUpDefaultServer();
		server.setKeyPairProvider(new SimpleGeneratorHostKeyProvider(properties.getHostKeyFile()));
		server.setPublickeyAuthenticator(RejectAllPublickeyAuthenticator.INSTANCE);
		server.setHost(properties.getHost());
		server.setPasswordAuthenticator(passwordAuthenticator);
		server.setPort(properties.getPort());
		server.setShellFactory(() -> shellCommandFactory);
		server.setCommandFactory(command -> shellCommandFactory);
		return server;
	}

}