use of com.jcraft.jsch.JSch in project azure-sdk-for-java by Azure.
the class TestVirtualMachineCustomData method createResource.
@Override
public VirtualMachine createResource(VirtualMachines virtualMachines) throws Exception {
final String vmName = "vm" + this.testId;
final String publicIpDnsLabel = SdkContext.randomResourceName("abc", 16);
// Prepare the custom data
//
String cloudInitFilePath = getClass().getClassLoader().getResource("cloud-init").getPath();
// In Windows remove leading slash
cloudInitFilePath = cloudInitFilePath.replaceFirst("^/(.:/)", "$1");
byte[] cloudInitAsBytes = Files.readAllBytes(Paths.get(cloudInitFilePath));
byte[] cloudInitEncoded = Base64.encodeBase64(cloudInitAsBytes);
String cloudInitEncodedString = new String(cloudInitEncoded);
PublicIPAddress pip = pips.define(publicIpDnsLabel).withRegion(Region.US_EAST).withNewResourceGroup().withLeafDomainLabel(publicIpDnsLabel).create();
VirtualMachine vm = virtualMachines.define(vmName).withRegion(pip.regionName()).withExistingResourceGroup(pip.resourceGroupName()).withNewPrimaryNetwork("10.0.0.0/28").withPrimaryPrivateIPAddressDynamic().withExistingPrimaryPublicIPAddress(pip).withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS).withRootUsername("testuser").withRootPassword("12NewPA$$w0rd!").withCustomData(cloudInitEncodedString).withSize(VirtualMachineSizeTypes.STANDARD_D3_V2).create();
pip.refresh();
Assert.assertTrue(pip.hasAssignedNetworkInterface());
if (!MockIntegrationTestBase.IS_MOCKED) {
JSch jsch = new JSch();
Session session = null;
ChannelExec channel = null;
try {
java.util.Properties config = new java.util.Properties();
config.put("StrictHostKeyChecking", "no");
session = jsch.getSession("testuser", publicIpDnsLabel + "." + "eastus.cloudapp.azure.com", 22);
session.setPassword("12NewPA$$w0rd!");
session.setConfig(config);
session.connect();
// Try running the package installed via init script
//
channel = (ChannelExec) session.openChannel("exec");
BufferedReader in = new BufferedReader(new InputStreamReader(channel.getInputStream()));
channel.setCommand("pwgen;");
channel.connect();
String msg;
while ((msg = in.readLine()) != null) {
Assert.assertFalse(msg.startsWith("The program 'pwgen' is currently not installed"));
}
} catch (Exception e) {
Assert.fail("SSH connection failed" + e.getMessage());
} finally {
if (channel != null) {
channel.disconnect();
}
if (session != null) {
session.disconnect();
}
}
}
return vm;
}
use of com.jcraft.jsch.JSch in project GNS by MobilityFirst.
the class SSHClient method scpTo.
/**
*
* @param user
* @param host
* @param keyFile
* @param lfile
* @param rfile
*/
public static void scpTo(String user, String host, File keyFile, String lfile, String rfile) {
if (verbose) {
System.out.println("Remote copy file from " + lfile + " to " + host + "@" + user + ":" + rfile);
}
FileInputStream fis = null;
try {
JSch jsch = new JSch();
Session session = authenticateWithKey(jsch, user, host, keyFile);
// username and password will be given via UserInfo interface.
UserInfo ui = new UserInfoPrompted();
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();
int ack = checkAck(in);
if (ack != 0) {
System.out.println("ACK was " + ack);
return;
}
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();
ack = checkAck(in);
if (ack != 0) {
System.out.println("ACK was " + ack);
return;
}
}
// 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();
ack = checkAck(in);
if (ack != 0) {
System.out.println("ACK was " + ack);
return;
}
// 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.flush();
out.write(buf, 0, len);
}
fis.close();
fis = null;
// send '\0'
buf[0] = 0;
out.write(buf, 0, 1);
out.flush();
ack = checkAck(in);
if (ack != 0) {
System.out.println("ACK was " + ack);
return;
}
out.close();
channel.disconnect();
session.disconnect();
} catch (JSchException | IOException e) {
GNSConfig.getLogger().severe(e.toString());
try {
if (fis != null) {
fis.close();
}
} catch (Exception ee) {
}
}
}
use of com.jcraft.jsch.JSch in project GNS by MobilityFirst.
the class SSHClient method exec.
/**
*
* @param user
* @param host
* @param keyFile
* @param command
* @param useSudo
* @param sudoPasswd
*/
public static void exec(String user, String host, File keyFile, String command, boolean useSudo, String sudoPasswd) {
if (verbose) {
System.out.println("Remote execute command on " + host + (useSudo ? " as root user: " : " as user " + user + ": ") + command);
}
try {
JSch jsch = new JSch();
Session session = authenticateWithKey(jsch, user, host, keyFile);
// username and password will be given via UserInfo interface.
UserInfo ui = new UserInfoPrompted();
session.setUserInfo(ui);
session.connect();
if (command == null) {
command = JOptionPane.showInputDialog("Enter command", "set|grep SSH");
}
Channel channel = session.openChannel("exec");
if (useSudo && sudoPasswd != null) {
((ChannelExec) channel).setCommand("sudo -S -p '' " + command);
} else if (useSudo) {
((ChannelExec) channel).setCommand("sudo -p '' " + command);
} else {
((ChannelExec) channel).setCommand(command);
}
// if (useSudo) {
// ((ChannelExec) channel).setCommand("sudo -S -p '' " + command);
// } else {
// ((ChannelExec) channel).setCommand(command);
// }
// ??? NOT SURE WHY THIS IS HERE
channel.setInputStream(null);
InputStream in = channel.getInputStream();
// prep the out channel so we can give a password if it is needed
OutputStream out = null;
if (useSudo && sudoPasswd != null) {
out = channel.getOutputStream();
}
((ChannelExec) channel).setErrStream(System.err);
// WESTY ADDED THIS *****
if (useSudo) {
// fixes "you must have a tty to sudo" problem
((ChannelExec) channel).setPty(true);
}
channel.connect();
// suppply the password for sudo
if (out != null) {
out.write((sudoPasswd + "\n").getBytes());
out.flush();
}
byte[] tmp = new byte[MAXCOMMANDBYTES];
while (true) {
while (in.available() > 0) {
int i = in.read(tmp, 0, MAXCOMMANDBYTES);
if (i < 0) {
break;
}
if (!verbose) {
System.out.print("o");
} else {
System.out.print(new String(tmp, 0, i));
}
}
if (channel.isClosed()) {
GNSConfig.getLogger().log(Level.FINE, "exit status: {0}", channel.getExitStatus());
break;
}
try {
Thread.sleep(1000);
} catch (Exception ee) {
}
}
channel.disconnect();
session.disconnect();
} catch (JSchException | IOException e) {
e.printStackTrace();
GNSConfig.getLogger().severe(e.toString());
}
}
use of com.jcraft.jsch.JSch in project azure-tools-for-java by Microsoft.
the class SparkBatchDebugSession method factory.
/**
* Create a SparkBatchDebugSession instance for specified host and user
*
* @param host The SSH host
* @param user The SSH user
* @return an SparkBatchDebugSession instance
* @throws JSchException JSch operation exceptions
*/
public static SparkBatchDebugSession factory(String host, String user) throws JSchException {
JSch jsch = new JSch();
Session session = jsch.getSession(user, host);
java.util.Properties config = new java.util.Properties() {
{
put("StrictHostKeyChecking", "no");
}
};
session.setConfig(config);
return new SparkBatchDebugSession(jsch, session);
}
use of com.jcraft.jsch.JSch in project azure-tools-for-java by Microsoft.
the class AzureDockerCertVaultOps method generateSSHKeys.
public static AzureDockerCertVault generateSSHKeys(String passPhrase, String comment) throws AzureDockerException {
try {
AzureDockerCertVault result = new AzureDockerCertVault();
JSch jsch = new JSch();
KeyPair keyPair = KeyPair.genKeyPair(jsch, KeyPair.RSA);
ByteArrayOutputStream privateKeyBuff = new ByteArrayOutputStream(2048);
ByteArrayOutputStream publicKeyBuff = new ByteArrayOutputStream(2048);
keyPair.writePublicKey(publicKeyBuff, (comment != null) ? comment : "DockerSSHCerts");
if (passPhrase == null || passPhrase.isEmpty()) {
keyPair.writePrivateKey(privateKeyBuff);
} else {
keyPair.writePrivateKey(privateKeyBuff, passPhrase.getBytes());
}
result.sshKey = privateKeyBuff.toString();
result.sshPubKey = publicKeyBuff.toString();
return result;
} catch (Exception e) {
throw new AzureDockerException(e.getMessage());
}
}
Aggregations