/**
* Copyright (c) 2013/2014 Verein zur Foerderung der IT-Sicherheit in Oesterreich (SBA).
* The work has been developed in the TIMBUS Project and the above-mentioned are Members of the TIMBUS Consortium.
* TIMBUS is supported by the European Union under the 7th Framework Programme for research and technological
* development and demonstration activities (FP7/2007-2013) under grant agreement no. 269940.
*
* Licensed 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, including without
* limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTIBITLY, or FITNESS FOR A PARTICULAR
* PURPOSE. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise,
* unless required by applicable law or agreed to in writing, shall any Contributor be liable for damages, including
* any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this
* License or out of the use or inability to use the Work.
* See the License for the specific language governing permissions and limitation under the License.
*/
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();
throw new SSHManagerException("Can not connect to machine "+fqdn+":"+port+". "+e.getLocalizedMessage());
}
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");
sftpChannel.get(path, file.getAbsolutePath());
} catch (SftpException | IOException e) {
e.printStackTrace();
throw new SSHManagerException("Can not read file '"+path.toString()+"' from remote machine. "+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();
ChannelSftp sftpChannel = (ChannelSftp) channel;
sftpChannel.put(from, to);
} catch (SftpException | JSchException e) {
e.printStackTrace();
throw new SSHManagerException("Can not save file from '"+from+"' to '"+to+"' on remote machine. "+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();
ChannelSftp sftpChannel = (ChannelSftp) channel;
sftpChannel.lstat(path.toString()); // if dir not exists an exception is thrown
}catch (SftpException | JSchException e) {
throw new SSHManagerException("Directory '"+path.toString()+"' does not exist on remote machine. " +e.getLocalizedMessage()+"!");
}
}
}