1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#pragma once
///@file

#include "lix/libutil/file-system.hh"
#include "lix/libutil/processes.hh"
#include "lix/libutil/sync.hh"
#include <cstdint>

namespace nix {

class SSH
{
private:

    const std::string host;
    const std::optional<uint16_t> port;
    bool fakeSSH;
    const std::string keyFile;
    const std::string sshPublicHostKey;
    const bool compress;
    const int logFD;

    struct State
    {
        std::unique_ptr<AutoDelete> tmpDir;
    };

    Sync<State> state_;

    void addCommonSSHOpts(Strings & args);

public:
    SSH(const std::string & host,
        const std::optional<uint16_t> port,
        const std::string & keyFile,
        const std::string & sshPublicHostKey,
        bool compress,
        int logFD);

    struct Connection
    {
        Pid sshPid;
        AutoCloseFD socket;
    };

    std::unique_ptr<Connection> startCommand(const std::string & command);
};

}