public static void main(String[] arg) {
if (arg.length != 2) {
System.err.println("usage: java ScpTo file1 user@remotehost:file2");
System.exit(-1);
}
FileInputStream fis = null;
try {
String lfile = arg[0];
String user = arg[1].substring(0, arg[1].indexOf('@'));
arg[1] = arg[1].substring(arg[1].indexOf('@') + 1);
String host = arg[1].substring(0, arg[1].indexOf(':'));
String rfile = arg[1].substring(arg[1].indexOf(':') + 1);
JSch jsch = new JSch();
Session session = jsch.getSession(user, host, 22);
// username and password will be given via UserInfo interface.
UserInfo ui = new MyUserInfo();
session.setUserInfo(ui);
session.connect();
boolean ptimestamp = true;
// exec 'scp -t rfile' remotely
String command = "scp " + (ptimestamp ? "-p" : "") + " -t " + rfile;
Channel channel = session.openChannel("exec");
((ChannelExec) channel).setCommand(command);
// get I/O streams for remote scp
OutputStream out = channel.getOutputStream();
InputStream in = channel.getInputStream();
channel.connect();
if (checkAck(in) != 0) {
System.exit(0);
}
File _lfile = new File(lfile);
if (ptimestamp) {
command = "T " + (_lfile.lastModified() / 1000) + " 0";
// The access time should be sent here,
// but it is not accessible with JavaAPI ;-<
command += (" " + (_lfile.lastModified() / 1000) + " 0\n");
out.write(command.getBytes());
out.flush();
if (checkAck(in) != 0) {
System.exit(0);
}
}
// send "C0644 filesize filename", where filename should not include '/'
long filesize = _lfile.length();
command = "C0644 " + filesize + " ";
if (lfile.lastIndexOf('/') > 0) {
command += lfile.substring(lfile.lastIndexOf('/') + 1);
} else {
command += lfile;
}
command += "\n";
out.write(command.getBytes());
out.flush();
if (checkAck(in) != 0) {
System.exit(0);
}
// send a content of lfile
fis = new FileInputStream(lfile);
byte[] buf = new byte[1024];
while (true) {
int len = fis.read(buf, 0, buf.length);
if (len <= 0) break;
out.write(buf, 0, len); // out.flush();
}
fis.close();
fis = null;
// send '\0'
buf[0] = 0;
out.write(buf, 0, 1);
out.flush();
if (checkAck(in) != 0) {
System.exit(0);
}
out.close();
channel.disconnect();
session.disconnect();
System.exit(0);
} catch (Exception e) {
System.out.println(e);
try {
if (fis != null) fis.close();
} catch (Exception ee) {
}
}
}
/**
* execute the given command in the remote host
*
* @param command
* @return - command output in the remote host
*/
public String exec(String command) {
if (!connected) {
throw new ActionFailedException("There is no session!");
}
StringBuffer data = new StringBuffer();
OutputStream out = null;
InputStream in = null;
try {
Channel channel;
boolean channel_connected = false;
int count = 0;
while (!channel_connected) {
try {
channel = session.openChannel("exec");
((ChannelExec) channel).setCommand(command);
out = channel.getOutputStream();
in = channel.getInputStream();
channel.connect();
channel_connected = true;
} catch (Exception e) {
count++;
String msg = e.getMessage();
if (count < 5) {
AutomationLogger.getInstance()
.warn(
"Failed to connect to SSH server due to "
+ msg
+ ". Will try again in 1 second");
if (msg.startsWith("session is down")) {
AutomationLogger.getInstance().info("Try to re-connect session");
connect();
}
Timer.sleep(1000);
} else {
throw new ActionFailedException("Failed to connect to SSH server due to " + e);
}
}
}
byte[] buf = new byte[1024];
// read
count = 0;
while ((count = in.read(buf)) > 0) {
data.append(new String(buf, 0, count));
}
} catch (Exception e) {
AutomationLogger.getInstance().warn(e);
} finally {
try {
in.close();
} catch (Exception e) {
}
try {
out.close();
} catch (Exception e) {
}
if (channel != null) {
channel.disconnect();
}
}
return data.toString();
}