lix/src/libstore/ssh-store.cc

105 lines
2.5 KiB
C++
Raw Normal View History

2016-09-12 12:03:29 +00:00
#include "store-api.hh"
#include "remote-store.hh"
2016-09-02 18:31:38 +00:00
#include "remote-fs-accessor.hh"
#include "archive.hh"
#include "worker-protocol.hh"
#include "pool.hh"
#include "ssh.hh"
2016-09-02 18:31:38 +00:00
namespace nix {
static std::string uriScheme = "ssh-ng://";
2017-02-07 18:20:15 +00:00
2016-09-12 12:03:29 +00:00
class SSHStore : public RemoteStore
{
public:
const Setting<Path> sshKey{(Store*) this, "", "ssh-key", "path to an SSH private key"};
const Setting<bool> compress{(Store*) this, false, "compress", "whether to compress the connection"};
SSHStore(const std::string & host, const Params & params)
: Store(params)
, RemoteStore(params)
, host(host)
, master(
host,
sshKey,
// Use SSH master only if using more than 1 connection.
connections->capacity() > 1,
compress)
{
}
2016-09-12 12:03:29 +00:00
std::string getUri() override
{
return uriScheme + host;
}
2016-09-12 12:03:29 +00:00
void narFromPath(const Path & path, Sink & sink) override;
ref<FSAccessor> getFSAccessor() override;
private:
struct Connection : RemoteStore::Connection
{
std::unique_ptr<SSHMaster::Connection> sshConn;
2016-09-12 12:03:29 +00:00
};
ref<RemoteStore::Connection> openConnection() override;
std::string host;
SSHMaster master;
2016-09-12 12:03:29 +00:00
};
2016-09-02 18:31:38 +00:00
class ForwardSource : public Source
{
Source & readSource;
Sink & writeSink;
public:
ForwardSource(Source & readSource, Sink & writeSink) : readSource(readSource), writeSink(writeSink) {}
size_t read(unsigned char * data, size_t len) override
{
auto n = readSource.read(data, len);
writeSink(data, n);
return n;
2016-09-02 18:31:38 +00:00
}
};
void SSHStore::narFromPath(const Path & path, Sink & sink)
{
auto conn(connections->get());
conn->to << wopNarFromPath << path;
conn->processStderr();
ParseSink ps;
auto fwd = ForwardSource(conn->from, sink);
parseDump(ps, fwd);
}
ref<FSAccessor> SSHStore::getFSAccessor()
{
return make_ref<RemoteFSAccessor>(ref<Store>(shared_from_this()));
}
ref<RemoteStore::Connection> SSHStore::openConnection()
{
auto conn = make_ref<Connection>();
conn->sshConn = master.startCommand("nix-daemon --stdio");
conn->to = FdSink(conn->sshConn->in.get());
conn->from = FdSource(conn->sshConn->out.get());
2016-09-02 18:31:38 +00:00
initConnection(*conn);
return conn;
}
static RegisterStoreImplementation regStore([](
const std::string & uri, const Store::Params & params)
-> std::shared_ptr<Store>
{
2017-02-07 18:20:15 +00:00
if (std::string(uri, 0, uriScheme.size()) != uriScheme) return 0;
return std::make_shared<SSHStore>(std::string(uri, uriScheme.size()), params);
2016-09-02 18:31:38 +00:00
});
}