154 lines
4.4 KiB
C++
154 lines
4.4 KiB
C++
#pragma once
|
|
|
|
namespace nix {
|
|
|
|
|
|
#define WORKER_MAGIC_1 0x6e697863
|
|
#define WORKER_MAGIC_2 0x6478696f
|
|
|
|
#define PROTOCOL_VERSION 0x119
|
|
#define GET_PROTOCOL_MAJOR(x) ((x) & 0xff00)
|
|
#define GET_PROTOCOL_MINOR(x) ((x) & 0x00ff)
|
|
|
|
|
|
typedef enum {
|
|
wopIsValidPath = 1,
|
|
wopHasSubstitutes = 3,
|
|
wopQueryPathHash = 4, // obsolete
|
|
wopQueryReferences = 5, // obsolete
|
|
wopQueryReferrers = 6,
|
|
wopAddToStore = 7,
|
|
wopAddTextToStore = 8, // obsolete since 1.25, Nix 3.0. Use wopAddToStore
|
|
wopBuildPaths = 9,
|
|
wopEnsurePath = 10,
|
|
wopAddTempRoot = 11,
|
|
wopAddIndirectRoot = 12,
|
|
wopSyncWithGC = 13,
|
|
wopFindRoots = 14,
|
|
wopExportPath = 16, // obsolete
|
|
wopQueryDeriver = 18, // obsolete
|
|
wopSetOptions = 19,
|
|
wopCollectGarbage = 20,
|
|
wopQuerySubstitutablePathInfo = 21,
|
|
wopQueryDerivationOutputs = 22, // obsolete
|
|
wopQueryAllValidPaths = 23,
|
|
wopQueryFailedPaths = 24,
|
|
wopClearFailedPaths = 25,
|
|
wopQueryPathInfo = 26,
|
|
wopImportPaths = 27, // obsolete
|
|
wopQueryDerivationOutputNames = 28, // obsolete
|
|
wopQueryPathFromHashPart = 29,
|
|
wopQuerySubstitutablePathInfos = 30,
|
|
wopQueryValidPaths = 31,
|
|
wopQuerySubstitutablePaths = 32,
|
|
wopQueryValidDerivers = 33,
|
|
wopOptimiseStore = 34,
|
|
wopVerifyStore = 35,
|
|
wopBuildDerivation = 36,
|
|
wopAddSignatures = 37,
|
|
wopNarFromPath = 38,
|
|
wopAddToStoreNar = 39,
|
|
wopQueryMissing = 40,
|
|
wopQueryDerivationOutputMap = 41,
|
|
} WorkerOp;
|
|
|
|
|
|
#define STDERR_NEXT 0x6f6c6d67
|
|
#define STDERR_READ 0x64617461 // data needed from source
|
|
#define STDERR_WRITE 0x64617416 // data for sink
|
|
#define STDERR_LAST 0x616c7473
|
|
#define STDERR_ERROR 0x63787470
|
|
#define STDERR_START_ACTIVITY 0x53545254
|
|
#define STDERR_STOP_ACTIVITY 0x53544f50
|
|
#define STDERR_RESULT 0x52534c54
|
|
|
|
|
|
class Store;
|
|
struct Source;
|
|
|
|
template<class T> T readStorePaths(const Store & store, Source & from);
|
|
|
|
void writeStorePaths(const Store & store, Sink & out, const StorePathSet & paths);
|
|
|
|
/* To guide overloading */
|
|
template<typename T>
|
|
struct Phantom {};
|
|
|
|
|
|
namespace worker_proto {
|
|
/* FIXME maybe move more stuff inside here */
|
|
|
|
StorePath read(const Store & store, Source & from, Phantom<StorePath> _);
|
|
void write(const Store & store, Sink & out, const StorePath & storePath);
|
|
|
|
template<typename T>
|
|
std::map<std::string, T> read(const Store & store, Source & from, Phantom<std::map<std::string, T>> _);
|
|
template<typename T>
|
|
void write(const Store & store, Sink & out, const std::map<string, T> & resMap);
|
|
template<typename T>
|
|
std::optional<T> read(const Store & store, Source & from, Phantom<std::optional<T>> _);
|
|
template<typename T>
|
|
void write(const Store & store, Sink & out, const std::optional<T> & optVal);
|
|
|
|
/* Specialization which uses and empty string for the empty case, taking
|
|
advantage of the fact StorePaths always serialize to a non-empty string.
|
|
This is done primarily for backwards compatability, so that StorePath <=
|
|
std::optional<StorePath>, where <= is the compatability partial order.
|
|
*/
|
|
template<>
|
|
void write(const Store & store, Sink & out, const std::optional<StorePath> & optVal);
|
|
|
|
template<typename T>
|
|
std::map<std::string, T> read(const Store & store, Source & from, Phantom<std::map<std::string, T>> _)
|
|
{
|
|
std::map<string, T> resMap;
|
|
auto size = (size_t)readInt(from);
|
|
while (size--) {
|
|
auto thisKey = readString(from);
|
|
resMap.insert_or_assign(std::move(thisKey), nix::worker_proto::read(store, from, Phantom<T> {}));
|
|
}
|
|
return resMap;
|
|
}
|
|
|
|
template<typename T>
|
|
void write(const Store & store, Sink & out, const std::map<string, T> & resMap)
|
|
{
|
|
out << resMap.size();
|
|
for (auto & i : resMap) {
|
|
out << i.first;
|
|
nix::worker_proto::write(store, out, i.second);
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
std::optional<T> read(const Store & store, Source & from, Phantom<std::optional<T>> _)
|
|
{
|
|
auto tag = readNum<uint8_t>(from);
|
|
switch (tag) {
|
|
case 0:
|
|
return std::nullopt;
|
|
case 1:
|
|
return nix::worker_proto::read(store, from, Phantom<T> {});
|
|
default:
|
|
throw Error("got an invalid tag bit for std::optional: %#04x", tag);
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
void write(const Store & store, Sink & out, const std::optional<T> & optVal)
|
|
{
|
|
out << (optVal ? 1 : 0);
|
|
if (optVal)
|
|
nix::worker_proto::write(store, out, *optVal);
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
StorePathCAMap readStorePathCAMap(const Store & store, Source & from);
|
|
|
|
void writeStorePathCAMap(const Store & store, Sink & out, const StorePathCAMap & paths);
|
|
|
|
}
|