eldritch horrors
ac74dc3a2b
this does nothing on classes that are never used as base classes and have only a single base class of their own. all of these classes are also made final to prove that they are indeed in this category. once we need to derive any of them we can erase their `final` specifiers. Change-Id: I4bd8e50a58815227392b5a5a762fda4542d15bae
112 lines
3.1 KiB
C++
112 lines
3.1 KiB
C++
#include "lix/libstore/ssh-store.hh"
|
|
#include "lix/libstore/store-api.hh"
|
|
#include "lix/libstore/remote-store.hh"
|
|
#include "lix/libstore/remote-store-connection.hh"
|
|
#include "lix/libstore/worker-protocol.hh"
|
|
#include "lix/libutil/pool.hh"
|
|
#include "lix/libstore/ssh.hh"
|
|
#include "lix/libutil/strings.hh"
|
|
|
|
namespace nix {
|
|
|
|
struct SSHStoreConfig : virtual RemoteStoreConfig, virtual CommonSSHStoreConfig
|
|
{
|
|
using RemoteStoreConfig::RemoteStoreConfig;
|
|
using CommonSSHStoreConfig::CommonSSHStoreConfig;
|
|
|
|
const Setting<Path> remoteProgram{this, "nix-daemon", "remote-program",
|
|
"Path to the `nix-daemon` executable on the remote machine."};
|
|
|
|
const std::string name() override { return "Experimental SSH Store"; }
|
|
|
|
std::string doc() override
|
|
{
|
|
return
|
|
#include "ssh-store.md"
|
|
;
|
|
}
|
|
};
|
|
|
|
class SSHStore final : public RemoteStore
|
|
{
|
|
SSHStoreConfig config_;
|
|
|
|
public:
|
|
SSHStore(const std::string & scheme, const std::string & host, SSHStoreConfig config)
|
|
: Store(config)
|
|
, RemoteStore(config)
|
|
, config_(std::move(config))
|
|
, host(host)
|
|
, master(
|
|
host,
|
|
config_.sshKey,
|
|
config_.sshPublicHostKey,
|
|
// Use SSH master only if using more than 1 connection.
|
|
connections->capacity() > 1,
|
|
config_.compress)
|
|
{
|
|
}
|
|
|
|
SSHStoreConfig & config() override { return config_; }
|
|
const SSHStoreConfig & config() const override { return config_; }
|
|
|
|
static std::set<std::string> uriSchemes() { return {"ssh-ng"}; }
|
|
|
|
std::string getUri() override
|
|
{
|
|
return *uriSchemes().begin() + "://" + host;
|
|
}
|
|
|
|
// FIXME extend daemon protocol, move implementation to RemoteStore
|
|
std::optional<std::string> getBuildLogExact(const StorePath & path) override
|
|
{ unsupported("getBuildLogExact"); }
|
|
|
|
protected:
|
|
|
|
struct Connection : RemoteStore::Connection
|
|
{
|
|
std::unique_ptr<SSHMaster::Connection> sshConn;
|
|
|
|
void closeWrite() override
|
|
{
|
|
sshConn->in.close();
|
|
}
|
|
};
|
|
|
|
ref<RemoteStore::Connection> openConnection() override;
|
|
|
|
std::string host;
|
|
|
|
SSHMaster master;
|
|
|
|
void setOptions(RemoteStore::Connection & conn) override
|
|
{
|
|
/* TODO Add a way to explicitly ask for some options to be
|
|
forwarded. One option: A way to query the daemon for its
|
|
settings, and then a series of params to SSHStore like
|
|
forward-cores or forward-overridden-cores that only
|
|
override the requested settings.
|
|
*/
|
|
};
|
|
};
|
|
|
|
ref<RemoteStore::Connection> SSHStore::openConnection()
|
|
{
|
|
auto conn = make_ref<Connection>();
|
|
|
|
std::string command = config_.remoteProgram + " --stdio";
|
|
if (config_.remoteStore.get() != "")
|
|
command += " --store " + shellEscape(config_.remoteStore.get());
|
|
|
|
conn->sshConn = master.startCommand(command);
|
|
conn->to = FdSink(conn->sshConn->in.get());
|
|
conn->from = FdSource(conn->sshConn->out.get());
|
|
return conn;
|
|
}
|
|
|
|
void registerSSHStore() {
|
|
StoreImplementations::add<SSHStore, SSHStoreConfig>();
|
|
}
|
|
|
|
}
|