2017-03-03 18:05:50 +00:00
|
|
|
|
#include "ssh.hh"
|
|
|
|
|
|
|
|
|
|
namespace nix {
|
|
|
|
|
|
2017-03-21 13:35:50 +00:00
|
|
|
|
void SSHMaster::addCommonSSHOpts(Strings & args)
|
|
|
|
|
{
|
|
|
|
|
for (auto & i : tokenizeString<Strings>(getEnv("NIX_SSHOPTS")))
|
|
|
|
|
args.push_back(i);
|
|
|
|
|
if (!keyFile.empty())
|
|
|
|
|
args.insert(args.end(), {"-i", keyFile});
|
|
|
|
|
if (compress)
|
|
|
|
|
args.push_back("-C");
|
|
|
|
|
}
|
|
|
|
|
|
2017-03-03 18:05:50 +00:00
|
|
|
|
std::unique_ptr<SSHMaster::Connection> SSHMaster::startCommand(const std::string & command)
|
|
|
|
|
{
|
2017-03-03 18:28:27 +00:00
|
|
|
|
Path socketPath = startMaster();
|
2017-03-03 18:05:50 +00:00
|
|
|
|
|
|
|
|
|
Pipe in, out;
|
|
|
|
|
in.create();
|
|
|
|
|
out.create();
|
|
|
|
|
|
|
|
|
|
auto conn = std::make_unique<Connection>();
|
|
|
|
|
conn->sshPid = startProcess([&]() {
|
|
|
|
|
restoreSignals();
|
|
|
|
|
|
|
|
|
|
close(in.writeSide.get());
|
|
|
|
|
close(out.readSide.get());
|
|
|
|
|
|
|
|
|
|
if (dup2(in.readSide.get(), STDIN_FILENO) == -1)
|
|
|
|
|
throw SysError("duping over stdin");
|
|
|
|
|
if (dup2(out.writeSide.get(), STDOUT_FILENO) == -1)
|
|
|
|
|
throw SysError("duping over stdout");
|
|
|
|
|
|
|
|
|
|
Strings args = { "ssh", host.c_str(), "-x", "-a" };
|
2017-03-21 13:35:50 +00:00
|
|
|
|
addCommonSSHOpts(args);
|
2017-03-03 18:28:27 +00:00
|
|
|
|
if (socketPath != "")
|
2017-03-03 18:05:50 +00:00
|
|
|
|
args.insert(args.end(), {"-S", socketPath});
|
|
|
|
|
args.push_back(command);
|
|
|
|
|
execvp(args.begin()->c_str(), stringsToCharPtrs(args).data());
|
|
|
|
|
|
|
|
|
|
throw SysError("executing ‘%s’ on ‘%s’", command, host);
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
in.readSide = -1;
|
|
|
|
|
out.writeSide = -1;
|
|
|
|
|
|
|
|
|
|
conn->out = std::move(out.readSide);
|
|
|
|
|
conn->in = std::move(in.writeSide);
|
|
|
|
|
|
|
|
|
|
return conn;
|
|
|
|
|
}
|
|
|
|
|
|
2017-03-03 18:28:27 +00:00
|
|
|
|
Path SSHMaster::startMaster()
|
2017-03-03 18:05:50 +00:00
|
|
|
|
{
|
2017-03-03 18:28:27 +00:00
|
|
|
|
if (!useMaster) return "";
|
2017-03-03 18:05:50 +00:00
|
|
|
|
|
2017-03-03 18:28:27 +00:00
|
|
|
|
auto state(state_.lock());
|
2017-03-03 18:05:50 +00:00
|
|
|
|
|
2017-03-03 18:28:27 +00:00
|
|
|
|
if (state->sshMaster != -1) return state->socketPath;
|
|
|
|
|
|
|
|
|
|
state->tmpDir = std::make_unique<AutoDelete>(createTempDir("", "nix", true, true, 0700));
|
|
|
|
|
|
|
|
|
|
state->socketPath = (Path) *state->tmpDir + "/ssh.sock";
|
2017-03-03 18:05:50 +00:00
|
|
|
|
|
|
|
|
|
Pipe out;
|
|
|
|
|
out.create();
|
|
|
|
|
|
2017-03-03 18:28:27 +00:00
|
|
|
|
state->sshMaster = startProcess([&]() {
|
2017-03-03 18:05:50 +00:00
|
|
|
|
restoreSignals();
|
|
|
|
|
|
|
|
|
|
close(out.readSide.get());
|
|
|
|
|
|
|
|
|
|
if (dup2(out.writeSide.get(), STDOUT_FILENO) == -1)
|
|
|
|
|
throw SysError("duping over stdout");
|
|
|
|
|
|
|
|
|
|
Strings args =
|
2017-03-03 18:28:27 +00:00
|
|
|
|
{ "ssh", host.c_str(), "-M", "-N", "-S", state->socketPath
|
2017-03-03 18:05:50 +00:00
|
|
|
|
, "-o", "LocalCommand=echo started"
|
|
|
|
|
, "-o", "PermitLocalCommand=yes"
|
|
|
|
|
};
|
2017-03-21 13:35:50 +00:00
|
|
|
|
addCommonSSHOpts(args);
|
2017-03-03 18:05:50 +00:00
|
|
|
|
execvp(args.begin()->c_str(), stringsToCharPtrs(args).data());
|
|
|
|
|
|
|
|
|
|
throw SysError("starting SSH master");
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
out.writeSide = -1;
|
|
|
|
|
|
|
|
|
|
std::string reply;
|
|
|
|
|
try {
|
|
|
|
|
reply = readLine(out.readSide.get());
|
|
|
|
|
} catch (EndOfFile & e) { }
|
|
|
|
|
|
|
|
|
|
if (reply != "started")
|
|
|
|
|
throw Error("failed to start SSH master connection to ‘%s’", host);
|
2017-03-03 18:28:27 +00:00
|
|
|
|
|
|
|
|
|
return state->socketPath;
|
2017-03-03 18:05:50 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|