Open LMDB files with MDB_NOLOCK if no write access

BVLC · Sep 18, 2015 · 274c88a · 274c88a
1 parent 3d12b5d
commit 274c88a
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/caffe/util/db_lmdb.cpp b/src/caffe/util/db_lmdb.cpp
@@ -19,7 +19,18 @@ void LMDB::Open(const string& source, Mode mode) {
   if (mode == READ) {
     flags = MDB_RDONLY | MDB_NOTLS;
   }
-  MDB_CHECK(mdb_env_open(mdb_env_, source.c_str(), flags, 0664));
+  int rc = mdb_env_open(mdb_env_, source.c_str(), flags, 0664);
+  if (rc == EACCES) {
+      LOG(WARNING) << "Permission denied. Trying with MDB_NOLOCK ...";
+      // Close and re-open environment handle
+      mdb_env_close(mdb_env_);
+      MDB_CHECK(mdb_env_create(&mdb_env_));
+      flags |= MDB_NOLOCK;
+      // Try again
+      MDB_CHECK(mdb_env_open(mdb_env_, source.c_str(), flags, 0664));
+  } else {
+      MDB_CHECK(rc);
+  }
   LOG(INFO) << "Opened lmdb " << source;
 }