diff --git a/Makefile b/Makefile index 5ddbd5429..d937bce50 100644 --- a/Makefile +++ b/Makefile @@ -52,7 +52,8 @@ TESTS = \ table_test \ version_edit_test \ version_set_test \ - write_batch_test + write_batch_test \ + filelock_test TOOLS = \ manifest_dump @@ -174,6 +175,10 @@ leveldb_server_test: thrift/test/simpletest.o $(LIBRARY) manifest_dump: tools/manifest_dump.o $(LIBOBJECTS) $(CXX) tools/manifest_dump.o $(LIBOBJECTS) -o $@ $(LDFLAGS) +filelock_test: util/filelock_test.o $(LIBOBJECTS) $(TESTHARNESS) + $(CXX) util/filelock_test.o $(LIBOBJECTS) $(TESTHARNESS) -o $@ $(LDFLAGS) + + ifeq ($(PLATFORM), IOS) # For iOS, create universal object files to be used on both the simulator and # a device. diff --git a/db/db_test.cc b/db/db_test.cc index e2abdf9b0..674667c4d 100644 --- a/db/db_test.cc +++ b/db/db_test.cc @@ -213,6 +213,10 @@ class DBTest { ASSERT_OK(TryReopen(options)); } + Status PureReopen(Options* options, DB** db) { + return DB::Open(*options, dbname_, db); + } + Status TryReopen(Options* options) { delete db_; db_ = NULL; @@ -828,6 +832,13 @@ TEST(DBTest, WAL) { ASSERT_EQ("v3", Get("foo")); } +TEST(DBTest, CheckLock) { + DB* localdb; + Options options = CurrentOptions(); + ASSERT_TRUE(TryReopen(&options).ok()); + ASSERT_TRUE(!(PureReopen(&options, &localdb).ok())); // second open should fail +} + TEST(DBTest, FLUSH) { Options options = CurrentOptions(); WriteOptions writeOpt = WriteOptions(); diff --git a/util/env_posix.cc b/util/env_posix.cc index 95d324602..9d668f5e0 100644 --- a/util/env_posix.cc +++ b/util/env_posix.cc @@ -3,6 +3,7 @@ // found in the LICENSE file. See the AUTHORS file for names of contributors. #include +#include #include #include #include @@ -31,6 +32,10 @@ namespace leveldb { namespace { +// list of pathnames that are locked +static std::set lockedFiles; +static port::Mutex mutex_lockedFiles; + static Status IOError(const std::string& context, int err_number) { return Status::IOError(context, strerror(err_number)); } @@ -291,7 +296,28 @@ class PosixMmapFile : public WritableFile { } }; -static int LockOrUnlock(int fd, bool lock) { +static int LockOrUnlock(const std::string& fname, int fd, bool lock) { + mutex_lockedFiles.Lock(); + if (lock) { + // If it already exists in the lockedFiles set, then it is already locked, + // and fail this lock attempt. Otherwise, insert it into lockedFiles. + // This check is needed because fcntl() does not detect lock conflict + // if the fcntl is issued by the same thread that earlier acquired + // this lock. + if (lockedFiles.insert(fname).second == false) { + mutex_lockedFiles.Unlock(); + errno = ENOLCK; + return -1; + } + } else { + // If we are unlocking, then verify that we had locked it earlier, + // it should already exist in lockedFiles. Remove it from lockedFiles. + if (lockedFiles.erase(fname) != 1) { + mutex_lockedFiles.Unlock(); + errno = ENOLCK; + return -1; + } + } errno = 0; struct flock f; memset(&f, 0, sizeof(f)); @@ -299,12 +325,19 @@ static int LockOrUnlock(int fd, bool lock) { f.l_whence = SEEK_SET; f.l_start = 0; f.l_len = 0; // Lock/unlock entire file - return fcntl(fd, F_SETLK, &f); + int value = fcntl(fd, F_SETLK, &f); + if (value == -1 && lock) { + // if there is an error in locking, then remove the pathname from lockedfiles + lockedFiles.erase(fname); + } + mutex_lockedFiles.Unlock(); + return value; } class PosixFileLock : public FileLock { public: int fd_; + std::string filename; }; class PosixEnv : public Env { @@ -435,12 +468,13 @@ class PosixEnv : public Env { int fd = open(fname.c_str(), O_RDWR | O_CREAT, 0644); if (fd < 0) { result = IOError(fname, errno); - } else if (LockOrUnlock(fd, true) == -1) { + } else if (LockOrUnlock(fname, fd, true) == -1) { result = IOError("lock " + fname, errno); close(fd); } else { PosixFileLock* my_lock = new PosixFileLock; my_lock->fd_ = fd; + my_lock->filename = fname; *lock = my_lock; } return result; @@ -449,7 +483,7 @@ class PosixEnv : public Env { virtual Status UnlockFile(FileLock* lock) { PosixFileLock* my_lock = reinterpret_cast(lock); Status result; - if (LockOrUnlock(my_lock->fd_, false) == -1) { + if (LockOrUnlock(my_lock->filename, my_lock->fd_, false) == -1) { result = IOError("unlock", errno); } close(my_lock->fd_); diff --git a/util/filelock_test.cc b/util/filelock_test.cc new file mode 100644 index 000000000..ea24527c3 --- /dev/null +++ b/util/filelock_test.cc @@ -0,0 +1,57 @@ +// Copyright (c) 2012 Facebook. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +#include "leveldb/status.h" +#include "leveldb/env.h" + +#include +#include "util/coding.h" +#include "util/testharness.h" + +namespace leveldb { + +class LockTest { + public: + static LockTest* current_; + std::string file_; + leveldb::Env* env_; + + LockTest() : file_(test::TmpDir() + "/db_testlock_file"), + env_(leveldb::Env::Default()) { + current_ = this; + } + + ~LockTest() { + } + + Status LockFile(FileLock** db_lock) { + return env_->LockFile(file_, db_lock); + } + + Status UnlockFile(FileLock* db_lock) { + return env_->UnlockFile(db_lock); + } +}; +LockTest* LockTest::current_; + +TEST(LockTest, LockBySameThread) { + FileLock* lock1; + FileLock* lock2; + + // acquire a lock on a file + ASSERT_OK(LockFile(&lock1)); + + // re-acquire the lock on the same file. This should fail. + ASSERT_TRUE(LockFile(&lock2).IsIOError()); + + // release the lock + ASSERT_OK(UnlockFile(lock1)); + +} + +} // namespace leveldb + +int main(int argc, char** argv) { + return leveldb::test::RunAllTests(); +}