2017-03-03 18:05:50 +00:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include "util.hh"
|
2017-03-03 18:28:27 +00:00
|
|
|
#include "sync.hh"
|
2017-03-03 18:05:50 +00:00
|
|
|
|
|
|
|
namespace nix {
|
|
|
|
|
|
|
|
class SSHMaster
|
|
|
|
{
|
|
|
|
private:
|
|
|
|
|
2017-03-03 18:28:27 +00:00
|
|
|
const std::string host;
|
|
|
|
const std::string keyFile;
|
|
|
|
const bool useMaster;
|
|
|
|
const bool compress;
|
|
|
|
|
|
|
|
struct State
|
|
|
|
{
|
|
|
|
Pid sshMaster;
|
|
|
|
std::unique_ptr<AutoDelete> tmpDir;
|
|
|
|
Path socketPath;
|
|
|
|
};
|
|
|
|
|
|
|
|
Sync<State> state_;
|
2017-03-03 18:05:50 +00:00
|
|
|
|
2017-03-21 13:35:50 +00:00
|
|
|
void addCommonSSHOpts(Strings & args);
|
|
|
|
|
2017-03-03 18:05:50 +00:00
|
|
|
public:
|
|
|
|
|
|
|
|
SSHMaster(const std::string & host, const std::string & keyFile, bool useMaster, bool compress)
|
|
|
|
: host(host)
|
|
|
|
, keyFile(keyFile)
|
|
|
|
, useMaster(useMaster)
|
|
|
|
, compress(compress)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
struct Connection
|
|
|
|
{
|
|
|
|
Pid sshPid;
|
|
|
|
AutoCloseFD out, in;
|
|
|
|
};
|
|
|
|
|
|
|
|
std::unique_ptr<Connection> startCommand(const std::string & command);
|
|
|
|
|
2017-03-03 18:28:27 +00:00
|
|
|
Path startMaster();
|
2017-03-03 18:05:50 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
}
|