package net.timbusproject.extractors.modules.tavernaextractor.utils;
import com.jcraft.jsch.*;
import java.io.File;
import java.io.IOException;
import java.lang.String;
import java.nio.file.Files;
import java.nio.file.Path;
/**
* Created by marco unterberger on 28.08.2014.
*
* Copy input files from a remote location to use them for the extraction.
*
* * munterberger@sba-research.org
*/
public class SSHManager {
private final String username;
private final String password;
private final String fqdn;
private final int port;
public SSHManager(String fqdn, int port, String username, String password){
this.username = username;
this.password = password;
this.fqdn = fqdn;
this.port = port;
}
public Session createSession(int timeout) throws SSHManagerException{
JSch jsch = new JSch();
Session session = null;
try {
session = jsch.getSession(username, fqdn, port);
session.setPassword(password);
/*
From jsch README:
StrictHostKeyChecking: ask | yes | no
default: ask
If this property is set to ``yes'', jsch will never automatically add
host keys to the $HOME/.ssh/known_hosts file, and refuses to connect
to hosts whose host key has changed. This property forces the user
to manually add all new hosts. If this property is set to ``no'',
jsch will automatically add new host keys to the user known hosts
files. If this property is set to ``ask'', new host keys will be
added to the user known host files only after the user has confirmed
that is what they really want to do, and jsch will refuse to connect
to hosts whose host key has changed.
*/
session.setConfig("StrictHostKeyChecking", "no");
session.connect(timeout);
} catch (JSchException e) {
e.printStackTrace();
}
return session;
}
public File readFile(Session session, String path) throws SSHManagerException{
if(!session.isConnected()){
try{
session.connect();
}
catch (JSchException e){
e.printStackTrace();
throw new SSHManagerException(e.getLocalizedMessage());
}
}
Channel channel = null;
try {
channel = session.openChannel("sftp");
channel.connect();
} catch (JSchException e) {
e.printStackTrace();
throw new SSHManagerException(e.getLocalizedMessage());
}
ChannelSftp sftpChannel = (ChannelSftp) channel;
File file = null;
try {
file = File.createTempFile("tavernaExtractor-", ".input");
} catch (IOException e) {
e.printStackTrace();
throw new SSHManagerException(e.getLocalizedMessage());
}
try {
sftpChannel.get(path, file.getAbsolutePath());
} catch (SftpException e) {
e.printStackTrace();
throw new SSHManagerException(e.getLocalizedMessage());
}
sftpChannel.disconnect();
session.disconnect();
return file;
}
public void sendFile(Session session, String from, String to) throws SSHManagerException{
if(!session.isConnected()){
try{
session.connect();
}
catch (JSchException e){
e.printStackTrace();
throw new SSHManagerException(e.getLocalizedMessage());
}
}
Channel channel = null;
try {
channel = session.openChannel("sftp");
channel.connect();
} catch (JSchException e) {
e.printStackTrace();
throw new SSHManagerException(e.getLocalizedMessage());
}
ChannelSftp sftpChannel = (ChannelSftp) channel;
try {
sftpChannel.put(from, to);
} catch (SftpException e) {
e.printStackTrace();
throw new SSHManagerException(e.getLocalizedMessage());
}
session.disconnect();
}
public void directoryExists(Session session, Path path) throws SSHManagerException{
if(!session.isConnected()){
try{
session.connect();
}
catch (JSchException e){
throw new SSHManagerException(e.getLocalizedMessage());
}
}
Channel channel = null;
try {
channel = session.openChannel("sftp");
channel.connect();
} catch (JSchException e) {
throw new SSHManagerException(e.getLocalizedMessage());
}
ChannelSftp sftpChannel = (ChannelSftp) channel;
try {
sftpChannel.lstat(path.toString()); // if dir not exists an exception is thrown
}catch (SftpException e) {
throw new SSHManagerException(path.toString()+" does not exists!" +e.getLocalizedMessage());
}
}
}