blob: 794f8327951077d760244fe56a20221bb47ea0fd [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 static com.google.common.base.Preconditions.checkState;
import com.jcraft.jsch.ChannelExec;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import java.io.IOException;
import java.io.InputStream;
import java.net.InetSocketAddress;
import java.util.Scanner;
public class SshSession {
private final InetSocketAddress addr;
private final TestAccount account;
private Session session;
private String error;
public SshSession(GerritServer server, TestAccount account) {
this.addr = server.getSshdAddress();
this.account = account;
}
public void open() throws JSchException {
getSession();
}
@SuppressWarnings("resource")
public String exec(String command, InputStream opt) throws JSchException,
IOException {
ChannelExec channel = (ChannelExec) getSession().openChannel("exec");
try {
channel.setCommand(command);
channel.setInputStream(opt);
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 InputStream exec2(String command, InputStream opt) throws JSchException,
IOException {
ChannelExec channel = (ChannelExec) getSession().openChannel("exec");
channel.setCommand(command);
channel.setInputStream(opt);
InputStream in = channel.getInputStream();
channel.connect();
return in;
}
public String exec(String command) throws JSchException, IOException {
return exec(command, null);
}
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,
addr.getAddress().getHostAddress(),
addr.getPort());
session.setConfig("StrictHostKeyChecking", "no");
session.connect();
}
return session;
}
public String getUrl() {
checkState(session != null, "session must be opened");
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();
}
}