2017-03-03 18:05:50 +00:00
|
|
|
#pragma once
|
2023-04-01 03:18:41 +00:00
|
|
|
///@file
|
2017-03-03 18:05:50 +00:00
|
|
|
|
2024-05-28 09:52:13 +00:00
|
|
|
#include "file-system.hh"
|
2024-05-28 11:14:13 +00:00
|
|
|
#include "processes.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;
|
2018-08-03 16:12:28 +00:00
|
|
|
bool fakeSSH;
|
2017-03-03 18:28:27 +00:00
|
|
|
const std::string keyFile;
|
2021-02-25 01:52:22 +00:00
|
|
|
const std::string sshPublicHostKey;
|
2017-03-03 18:28:27 +00:00
|
|
|
const bool useMaster;
|
|
|
|
const bool compress;
|
2017-05-02 10:01:46 +00:00
|
|
|
const int logFD;
|
2017-03-03 18:28:27 +00:00
|
|
|
|
|
|
|
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);
|
2023-05-16 12:48:04 +00:00
|
|
|
bool isMasterRunning();
|
2017-03-21 13:35:50 +00:00
|
|
|
|
2017-03-03 18:05:50 +00:00
|
|
|
public:
|
|
|
|
|
2021-02-25 01:52:22 +00:00
|
|
|
SSHMaster(const std::string & host, const std::string & keyFile, const std::string & sshPublicHostKey, bool useMaster, bool compress, int logFD = -1);
|
2017-03-03 18:05:50 +00:00
|
|
|
|
|
|
|
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
|
|
|
};
|
|
|
|
|
|
|
|
}
|