mirror of
https://github.com/privatevoid-net/nix-super.git
synced 2024-09-21 08:58:04 +03:00
cd4d2705ec
Opening an SSHStore or LegacySSHStore does not actually establish a connection, so the try/catch block here did nothing. Added a Store::connect() method to test whether a connection can be established.
145 lines
3.6 KiB
C++
145 lines
3.6 KiB
C++
#pragma once
|
|
|
|
#include <limits>
|
|
#include <string>
|
|
|
|
#include "store-api.hh"
|
|
|
|
|
|
namespace nix {
|
|
|
|
|
|
class Pipe;
|
|
class Pid;
|
|
struct FdSink;
|
|
struct FdSource;
|
|
template<typename T> class Pool;
|
|
|
|
|
|
/* FIXME: RemoteStore is a misnomer - should be something like
|
|
DaemonStore. */
|
|
class RemoteStore : public virtual Store
|
|
{
|
|
public:
|
|
|
|
const Setting<int> maxConnections{(Store*) this, 1,
|
|
"max-connections", "maximum number of concurrent connections to the Nix daemon"};
|
|
|
|
RemoteStore(const Params & params);
|
|
|
|
/* Implementations of abstract store API methods. */
|
|
|
|
bool isValidPathUncached(const Path & path) override;
|
|
|
|
PathSet queryValidPaths(const PathSet & paths, bool maybeSubstitute = false) override;
|
|
|
|
PathSet queryAllValidPaths() override;
|
|
|
|
void queryPathInfoUncached(const Path & path,
|
|
std::function<void(std::shared_ptr<ValidPathInfo>)> success,
|
|
std::function<void(std::exception_ptr exc)> failure) override;
|
|
|
|
void queryReferrers(const Path & path, PathSet & referrers) override;
|
|
|
|
PathSet queryValidDerivers(const Path & path) override;
|
|
|
|
PathSet queryDerivationOutputs(const Path & path) override;
|
|
|
|
StringSet queryDerivationOutputNames(const Path & path) override;
|
|
|
|
Path queryPathFromHashPart(const string & hashPart) override;
|
|
|
|
PathSet querySubstitutablePaths(const PathSet & paths) override;
|
|
|
|
void querySubstitutablePathInfos(const PathSet & paths,
|
|
SubstitutablePathInfos & infos) override;
|
|
|
|
void addToStore(const ValidPathInfo & info, const ref<std::string> & nar,
|
|
bool repair, bool dontCheckSigs,
|
|
std::shared_ptr<FSAccessor> accessor) override;
|
|
|
|
Path addToStore(const string & name, const Path & srcPath,
|
|
bool recursive = true, HashType hashAlgo = htSHA256,
|
|
PathFilter & filter = defaultPathFilter, bool repair = false) override;
|
|
|
|
Path addTextToStore(const string & name, const string & s,
|
|
const PathSet & references, bool repair = false) override;
|
|
|
|
void buildPaths(const PathSet & paths, BuildMode buildMode) override;
|
|
|
|
BuildResult buildDerivation(const Path & drvPath, const BasicDerivation & drv,
|
|
BuildMode buildMode) override;
|
|
|
|
void ensurePath(const Path & path) override;
|
|
|
|
void addTempRoot(const Path & path) override;
|
|
|
|
void addIndirectRoot(const Path & path) override;
|
|
|
|
void syncWithGC() override;
|
|
|
|
Roots findRoots() override;
|
|
|
|
void collectGarbage(const GCOptions & options, GCResults & results) override;
|
|
|
|
void optimiseStore() override;
|
|
|
|
bool verifyStore(bool checkContents, bool repair) override;
|
|
|
|
void addSignatures(const Path & storePath, const StringSet & sigs) override;
|
|
|
|
void queryMissing(const PathSet & targets,
|
|
PathSet & willBuild, PathSet & willSubstitute, PathSet & unknown,
|
|
unsigned long long & downloadSize, unsigned long long & narSize) override;
|
|
|
|
void connect() override;
|
|
|
|
protected:
|
|
|
|
struct Connection
|
|
{
|
|
FdSink to;
|
|
FdSource from;
|
|
unsigned int daemonVersion;
|
|
|
|
virtual ~Connection();
|
|
|
|
void processStderr(Sink * sink = 0, Source * source = 0);
|
|
};
|
|
|
|
ref<Connection> openConnectionWrapper();
|
|
|
|
virtual ref<Connection> openConnection() = 0;
|
|
|
|
void initConnection(Connection & conn);
|
|
|
|
ref<Pool<Connection>> connections;
|
|
|
|
private:
|
|
|
|
std::atomic_bool failed{false};
|
|
|
|
void setOptions(Connection & conn);
|
|
};
|
|
|
|
class UDSRemoteStore : public LocalFSStore, public RemoteStore
|
|
{
|
|
public:
|
|
|
|
UDSRemoteStore(const Params & params);
|
|
|
|
std::string getUri() override;
|
|
|
|
private:
|
|
|
|
struct Connection : RemoteStore::Connection
|
|
{
|
|
AutoCloseFD fd;
|
|
};
|
|
|
|
ref<RemoteStore::Connection> openConnection() override;
|
|
};
|
|
|
|
|
|
}
|