NixOS/nix
NixOS/nix
1
0
Fork 0
mirror of https://github.com/NixOS/nix synced 2024-09-19 23:03:53 -04:00
Code Activity

SSHStore: uri -> host

Browse source
This commit is contained in:
Eelco Dolstra 2017-02-07 19:20:15 +01:00
parent 612aeb2df5
commit 7a58ad0ef5
No known key found for this signature in database
GPG key ID: 8170B4726D7198DE
1 changed files with 12 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
src/libstore/ssh-store.cc
View file

@ -7,11 +7,13 @@
namespace nix { namespace nix {
static std::string uriScheme = "ssh://";
class SSHStore : public RemoteStore class SSHStore : public RemoteStore
{ {
public: public:
SSHStore(string uri, const Params & params, size_t maxConnections = std::numeric_limits<size_t>::max()); SSHStore(string host, const Params & params, size_t maxConnections = std::numeric_limits<size_t>::max());
std::string getUri() override; std::string getUri() override;
@ -36,17 +38,17 @@ private:
Pid sshMaster; Pid sshMaster;
string uri; string host;
Path key; Path key;
}; };
SSHStore::SSHStore(string uri, const Params & params, size_t maxConnections) SSHStore::SSHStore(string host, const Params & params, size_t maxConnections)
: Store(params) : Store(params)
, RemoteStore(params, maxConnections) , RemoteStore(params, maxConnections)
, tmpDir(createTempDir("", "nix", true, true, 0700)) , tmpDir(createTempDir("", "nix", true, true, 0700))
, socketPath((Path) tmpDir + "/ssh.sock") , socketPath((Path) tmpDir + "/ssh.sock")
, uri(std::move(uri)) , host(std::move(host))
, key(get(params, "ssh-key", "")) , key(get(params, "ssh-key", ""))
{ {
/* open a connection and perform the handshake to verify all is well */ /* open a connection and perform the handshake to verify all is well */
@ -55,7 +57,7 @@ SSHStore::SSHStore(string uri, const Params & params, size_t maxConnections)
string SSHStore::getUri() string SSHStore::getUri()
{ {
return "ssh://" + uri; return uriScheme + host;
} }
class ForwardSource : public Source class ForwardSource : public Source
@ -93,9 +95,9 @@ ref<RemoteStore::Connection> SSHStore::openConnection()
sshMaster = startProcess([&]() { sshMaster = startProcess([&]() {
restoreSignals(); restoreSignals();
if (key.empty()) if (key.empty())
execlp("ssh", "ssh", "-N", "-M", "-S", socketPath.c_str(), uri.c_str(), NULL); execlp("ssh", "ssh", "-N", "-M", "-S", socketPath.c_str(), host.c_str(), NULL);
else else
execlp("ssh", "ssh", "-N", "-M", "-S", socketPath.c_str(), "-i", key.c_str(), uri.c_str(), NULL); execlp("ssh", "ssh", "-N", "-M", "-S", socketPath.c_str(), "-i", key.c_str(), host.c_str(), NULL);
throw SysError("starting ssh master"); throw SysError("starting ssh master");
}); });
} }
@ -109,7 +111,7 @@ ref<RemoteStore::Connection> SSHStore::openConnection()
throw SysError("duping over STDIN"); throw SysError("duping over STDIN");
if (dup2(out.writeSide.get(), STDOUT_FILENO) == -1) if (dup2(out.writeSide.get(), STDOUT_FILENO) == -1)
throw SysError("duping over STDOUT"); throw SysError("duping over STDOUT");
execlp("ssh", "ssh", "-S", socketPath.c_str(), uri.c_str(), "nix-daemon", "--stdio", NULL); execlp("ssh", "ssh", "-S", socketPath.c_str(), host.c_str(), "nix-daemon", "--stdio", NULL);
throw SysError("executing nix-daemon --stdio over ssh"); throw SysError("executing nix-daemon --stdio over ssh");
}); });
in.readSide = -1; in.readSide = -1;
@ -126,8 +128,8 @@ static RegisterStoreImplementation regStore([](
const std::string & uri, const Store::Params & params) const std::string & uri, const Store::Params & params)
-> std::shared_ptr<Store> -> std::shared_ptr<Store>
{ {
if (std::string(uri, 0, 6) != "ssh://") return 0; if (std::string(uri, 0, uriScheme.size()) != uriScheme) return 0;
return std::make_shared<SSHStore>(uri.substr(6), params); return std::make_shared<SSHStore>(std::string(uri, uriScheme.size()), params);
}); });
} }