tvl-depot/src/libstore/remote-store.cc

174 lines
3.6 KiB
C++
Raw Normal View History

#include "serialise.hh"
#include "util.hh"
2006-11-30 19:35:50 +01:00
#include "remote-store.hh"
#include "worker-protocol.hh"
2006-11-30 21:45:20 +01:00
#include "archive.hh"
2006-11-30 19:35:50 +01:00
#include <iostream>
#include <unistd.h>
2006-11-30 19:35:50 +01:00
namespace nix {
RemoteStore::RemoteStore()
{
toChild.create();
fromChild.create();
/* Start the worker. */
string worker = "nix-worker";
child = fork();
switch (child) {
case -1:
throw SysError("unable to fork");
case 0:
try { /* child */
fromChild.readSide.close();
if (dup2(fromChild.writeSide, STDOUT_FILENO) == -1)
throw SysError("dupping write side");
toChild.writeSide.close();
if (dup2(toChild.readSide, STDIN_FILENO) == -1)
throw SysError("dupping read side");
execlp(worker.c_str(), worker.c_str(),
"--slave", NULL);
throw SysError(format("executing `%1%'") % worker);
} catch (std::exception & e) {
std::cerr << format("child error: %1%\n") % e.what();
}
quickExit(1);
}
fromChild.writeSide.close();
toChild.readSide.close();
from.fd = fromChild.readSide;
to.fd = toChild.writeSide;
/* Send the magic greeting, check for the reply. */
writeInt(WORKER_MAGIC_1, to);
unsigned int magic = readInt(from);
if (magic != WORKER_MAGIC_2) throw Error("protocol mismatch");
2006-11-30 19:35:50 +01:00
}
RemoteStore::~RemoteStore()
{
writeInt(wopQuit, to);
readInt(from);
child.wait(true);
2006-11-30 19:35:50 +01:00
}
bool RemoteStore::isValidPath(const Path & path)
{
writeInt(wopIsValidPath, to);
writeString(path, to);
unsigned int reply = readInt(from);
return reply != 0;
2006-11-30 19:35:50 +01:00
}
Substitutes RemoteStore::querySubstitutes(const Path & path)
2006-11-30 19:35:50 +01:00
{
throw Error("not implemented 2");
}
bool RemoteStore::hasSubstitutes(const Path & path)
{
writeInt(wopHasSubstitutes, to);
writeString(path, to);
unsigned int reply = readInt(from);
return reply != 0;
2006-11-30 19:35:50 +01:00
}
Hash RemoteStore::queryPathHash(const Path & path)
{
writeInt(wopQueryPathHash, to);
writeString(path, to);
string hash = readString(from);
return parseHash(htSHA256, hash);
2006-11-30 19:35:50 +01:00
}
void RemoteStore::queryReferences(const Path & path,
2006-11-30 19:35:50 +01:00
PathSet & references)
{
writeInt(wopQueryReferences, to);
writeString(path, to);
PathSet references2 = readStringSet(from);
references.insert(references2.begin(), references2.end());
2006-11-30 19:35:50 +01:00
}
void RemoteStore::queryReferrers(const Path & path,
2006-11-30 19:35:50 +01:00
PathSet & referrers)
{
writeInt(wopQueryReferrers, to);
writeString(path, to);
PathSet referrers2 = readStringSet(from);
referrers.insert(referrers2.begin(), referrers2.end());
2006-11-30 19:35:50 +01:00
}
Path RemoteStore::addToStore(const Path & _srcPath, bool fixed,
bool recursive, string hashAlgo)
2006-11-30 19:35:50 +01:00
{
Path srcPath(absPath(_srcPath));
2006-11-30 21:45:20 +01:00
writeInt(wopAddToStore, to);
writeString(baseNameOf(srcPath), to);
writeInt(fixed ? 1 : 0, to);
writeInt(recursive ? 1 : 0, to);
writeString(hashAlgo, to);
dumpPath(srcPath, to);
Path path = readString(from);
return path;
2006-11-30 19:35:50 +01:00
}
Path RemoteStore::addTextToStore(const string & suffix, const string & s,
const PathSet & references)
{
2006-11-30 21:45:20 +01:00
writeInt(wopAddTextToStore, to);
writeString(suffix, to);
writeString(s, to);
writeStringSet(references, to);
2006-11-30 21:45:20 +01:00
Path path = readString(from);
return path;
2006-11-30 19:35:50 +01:00
}
void RemoteStore::buildDerivations(const PathSet & drvPaths)
{
writeInt(wopBuildDerivations, to);
writeStringSet(drvPaths, to);
readInt(from);
2006-11-30 19:35:50 +01:00
}
void RemoteStore::ensurePath(const Path & path)
2006-11-30 19:35:50 +01:00
{
writeInt(wopEnsurePath, to);
writeString(path, to);
readInt(from);
2006-11-30 19:35:50 +01:00
}
}