mirror of
https://github.com/privatevoid-net/nix-super.git
synced 2024-11-16 03:06:17 +02:00
f71b4da0b3
This increases test coverage, and gets the worker protocol ready to be used by Hydra. Why don't we just try to use the store interface in Hydra? Well, the problem is that the store interface works on connection pools, with each opreation getting potentially a different connection, but the way temp roots work requires that we keep one logical "transaction" (temp root session) using the same connection. The longer-term solution probably is making connections themselves implement the store interface, but that is something that builds on this, so I feel OK that this is not churn in the wrong direction. Fixes #9584
106 lines
3.3 KiB
C++
106 lines
3.3 KiB
C++
#include "serve-protocol-connection.hh"
|
|
#include "serve-protocol-impl.hh"
|
|
#include "build-result.hh"
|
|
#include "derivations.hh"
|
|
|
|
namespace nix {
|
|
|
|
ServeProto::Version ServeProto::BasicClientConnection::handshake(
|
|
BufferedSink & to, Source & from, ServeProto::Version localVersion, std::string_view host)
|
|
{
|
|
to << SERVE_MAGIC_1 << localVersion;
|
|
to.flush();
|
|
|
|
unsigned int magic = readInt(from);
|
|
if (magic != SERVE_MAGIC_2)
|
|
throw Error("'nix-store --serve' protocol mismatch from '%s'", host);
|
|
auto remoteVersion = readInt(from);
|
|
if (GET_PROTOCOL_MAJOR(remoteVersion) != 0x200)
|
|
throw Error("unsupported 'nix-store --serve' protocol version on '%s'", host);
|
|
return std::min(remoteVersion, localVersion);
|
|
}
|
|
|
|
ServeProto::Version
|
|
ServeProto::BasicServerConnection::handshake(BufferedSink & to, Source & from, ServeProto::Version localVersion)
|
|
{
|
|
unsigned int magic = readInt(from);
|
|
if (magic != SERVE_MAGIC_1)
|
|
throw Error("protocol mismatch");
|
|
to << SERVE_MAGIC_2 << localVersion;
|
|
to.flush();
|
|
auto remoteVersion = readInt(from);
|
|
return std::min(remoteVersion, localVersion);
|
|
}
|
|
|
|
StorePathSet ServeProto::BasicClientConnection::queryValidPaths(
|
|
const StoreDirConfig & store, bool lock, const StorePathSet & paths, SubstituteFlag maybeSubstitute)
|
|
{
|
|
to << ServeProto::Command::QueryValidPaths << lock << maybeSubstitute;
|
|
write(store, *this, paths);
|
|
to.flush();
|
|
|
|
return Serialise<StorePathSet>::read(store, *this);
|
|
}
|
|
|
|
std::map<StorePath, UnkeyedValidPathInfo>
|
|
ServeProto::BasicClientConnection::queryPathInfos(const StoreDirConfig & store, const StorePathSet & paths)
|
|
{
|
|
std::map<StorePath, UnkeyedValidPathInfo> infos;
|
|
|
|
to << ServeProto::Command::QueryPathInfos;
|
|
ServeProto::write(store, *this, paths);
|
|
to.flush();
|
|
|
|
while (true) {
|
|
auto storePathS = readString(from);
|
|
if (storePathS == "")
|
|
break;
|
|
|
|
auto storePath = store.parseStorePath(storePathS);
|
|
assert(paths.count(storePath) == 1);
|
|
auto info = ServeProto::Serialise<UnkeyedValidPathInfo>::read(store, *this);
|
|
infos.insert_or_assign(std::move(storePath), std::move(info));
|
|
}
|
|
|
|
return infos;
|
|
}
|
|
|
|
void ServeProto::BasicClientConnection::putBuildDerivationRequest(
|
|
const StoreDirConfig & store,
|
|
const StorePath & drvPath,
|
|
const BasicDerivation & drv,
|
|
const ServeProto::BuildOptions & options)
|
|
{
|
|
to << ServeProto::Command::BuildDerivation << store.printStorePath(drvPath);
|
|
writeDerivation(to, store, drv);
|
|
|
|
ServeProto::write(store, *this, options);
|
|
|
|
to.flush();
|
|
}
|
|
|
|
BuildResult ServeProto::BasicClientConnection::getBuildDerivationResponse(const StoreDirConfig & store)
|
|
{
|
|
return ServeProto::Serialise<BuildResult>::read(store, *this);
|
|
}
|
|
|
|
void ServeProto::BasicClientConnection::narFromPath(
|
|
const StoreDirConfig & store, const StorePath & path, std::function<void(Source &)> fun)
|
|
{
|
|
to << ServeProto::Command::DumpStorePath << store.printStorePath(path);
|
|
to.flush();
|
|
|
|
fun(from);
|
|
}
|
|
|
|
void ServeProto::BasicClientConnection::importPaths(const StoreDirConfig & store, std::function<void(Sink &)> fun)
|
|
{
|
|
to << ServeProto::Command::ImportPaths;
|
|
fun(to);
|
|
to.flush();
|
|
|
|
if (readInt(from) != 1)
|
|
throw Error("remote machine failed to import closure");
|
|
}
|
|
|
|
}
|