forked from lix-project/lix
Probably fix a segfault in PathLocks
This commit is contained in:
parent
b30d1e7ada
commit
47f587700d
|
@ -1,5 +1,6 @@
|
||||||
#include "pathlocks.hh"
|
#include "pathlocks.hh"
|
||||||
#include "util.hh"
|
#include "util.hh"
|
||||||
|
#include "sync.hh"
|
||||||
|
|
||||||
#include <cerrno>
|
#include <cerrno>
|
||||||
#include <cstdlib>
|
#include <cstdlib>
|
||||||
|
@ -72,7 +73,7 @@ bool lockFile(int fd, LockType lockType, bool wait)
|
||||||
close a descriptor, the previous lock will be closed as well. And
|
close a descriptor, the previous lock will be closed as well. And
|
||||||
there is no way to query whether we already have a lock (F_GETLK
|
there is no way to query whether we already have a lock (F_GETLK
|
||||||
only works on locks held by other processes). */
|
only works on locks held by other processes). */
|
||||||
static StringSet lockedPaths; /* !!! not thread-safe */
|
static Sync<StringSet> lockedPaths_;
|
||||||
|
|
||||||
|
|
||||||
PathLocks::PathLocks()
|
PathLocks::PathLocks()
|
||||||
|
@ -108,8 +109,14 @@ bool PathLocks::lockPaths(const PathSet & _paths,
|
||||||
|
|
||||||
debug(format("locking path ‘%1%’") % path);
|
debug(format("locking path ‘%1%’") % path);
|
||||||
|
|
||||||
if (lockedPaths.find(lockPath) != lockedPaths.end())
|
{
|
||||||
throw Error("deadlock: trying to re-acquire self-held lock");
|
auto lockedPaths(lockedPaths_.lock());
|
||||||
|
if (lockedPaths->count(lockPath))
|
||||||
|
throw Error("deadlock: trying to re-acquire self-held lock ‘%s’", lockPath);
|
||||||
|
lockedPaths->insert(lockPath);
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
AutoCloseFD fd;
|
AutoCloseFD fd;
|
||||||
|
|
||||||
|
@ -150,7 +157,12 @@ bool PathLocks::lockPaths(const PathSet & _paths,
|
||||||
|
|
||||||
/* Use borrow so that the descriptor isn't closed. */
|
/* Use borrow so that the descriptor isn't closed. */
|
||||||
fds.push_back(FDPair(fd.release(), lockPath));
|
fds.push_back(FDPair(fd.release(), lockPath));
|
||||||
lockedPaths.insert(lockPath);
|
|
||||||
|
} catch (...) {
|
||||||
|
lockedPaths_.lock()->erase(lockPath);
|
||||||
|
throw;
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
|
@ -172,7 +184,8 @@ void PathLocks::unlock()
|
||||||
for (auto & i : fds) {
|
for (auto & i : fds) {
|
||||||
if (deletePaths) deleteLockFile(i.second, i.first);
|
if (deletePaths) deleteLockFile(i.second, i.first);
|
||||||
|
|
||||||
lockedPaths.erase(i.second);
|
lockedPaths_.lock()->erase(i.second);
|
||||||
|
|
||||||
if (close(i.first) == -1)
|
if (close(i.first) == -1)
|
||||||
printError(
|
printError(
|
||||||
format("error (ignored): cannot close lock file on ‘%1%’") % i.second);
|
format("error (ignored): cannot close lock file on ‘%1%’") % i.second);
|
||||||
|
@ -193,7 +206,7 @@ void PathLocks::setDeletion(bool deletePaths)
|
||||||
bool pathIsLockedByMe(const Path & path)
|
bool pathIsLockedByMe(const Path & path)
|
||||||
{
|
{
|
||||||
Path lockPath = path + ".lock";
|
Path lockPath = path + ".lock";
|
||||||
return lockedPaths.find(lockPath) != lockedPaths.end();
|
return lockedPaths_.lock()->count(lockPath);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue