blob: aae9236bcd6eb61116e611ae0349a035677abba6 [file] [log] [blame]
// Copyright (C) 2013 The Android Open Source Project
//
// 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.
// See the License for the specific language governing permissions and
// limitations under the License.
package com.google.gerrit.acceptance;
import java.io.IOException;
import java.io.InputStream;
import java.util.Scanner;
import com.jcraft.jsch.ChannelExec;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
public class SshSession {
private final TestAccount account;
private Session session;
private String error;
public SshSession(TestAccount account) {
this.account = account;
}
public String exec(String command) throws JSchException, IOException {
ChannelExec channel = (ChannelExec) getSession().openChannel("exec");
try {
channel.setCommand(command);
channel.setInputStream(null);
InputStream in = channel.getInputStream();
channel.connect();
Scanner s = new Scanner(channel.getErrStream()).useDelimiter("\\A");
error = s.hasNext() ? s.next() : null;
s = new Scanner(in).useDelimiter("\\A");
return s.hasNext() ? s.next() : "";
} finally {
channel.disconnect();
}
}
public boolean hasError() {
return error != null;
}
public String getError() {
return error;
}
public void close() {
if (session != null) {
session.disconnect();
session = null;
}
}
private Session getSession() throws JSchException {
if (session == null) {
JSch jsch = new JSch();
jsch.addIdentity("KeyPair",
account.privateKey(), account.sshKey.getPublicKeyBlob(), null);
session = jsch.getSession(account.username, "localhost", 29418);
session.setConfig("StrictHostKeyChecking", "no");
session.connect();
}
return session;
}
public String getUrl() {
StringBuilder b = new StringBuilder();
b.append("ssh://");
b.append(session.getUserName());
b.append("@");
b.append(session.getHost());
b.append(":");
b.append(session.getPort());
return b.toString();
}
}