Take a shared lock of global state during regular operation

Fixes issue where proxies might be replaced whilst in-use.
Splits file/dir proxy map lock.

4 files changed, 11 insertions, 5 deletions

diff --git a/netfs/fuse/fuseApp.cpp b/netfs/fuse/fuseApp.cpp
index a5dd0b1..d4835b9 100644
--- a/netfs/fuse/fuseApp.cpp
+++ b/netfs/fuse/fuseApp.cpp
@@ -150,7 +150,6 @@ NetFS::FuseApp::connectToService()
 {
 	if (!service) {
 		Lock(_lock);
-
 		auto proxyAddr = fcr->ServiceIdentity;
 		for (const auto & ep : fcr->Endpoints) {
 			proxyAddr += ":" + ep;
@@ -166,6 +165,7 @@ void
 NetFS::FuseApp::connectToVolume()
 {
 	if (!volume) {
+		Lock(_lock);
 		volume = service->connect(fcr->ExportName, fcr->AuthToken);
 		if (!volume) {
 			throw std::runtime_error("Invalid filesystem proxy");
diff --git a/netfs/fuse/fuseApp.h b/netfs/fuse/fuseApp.h
index 35c56dd..9653a55 100644
--- a/netfs/fuse/fuseApp.h
+++ b/netfs/fuse/fuseApp.h
@@ -124,7 +124,7 @@ namespace NetFS {
 			Ice::StringSeq args;
 			Ice::CommunicatorPtr ic;
 			Client::ResourcePtr fcr;
-			mutable std::shared_mutex _lock;
+			mutable std::shared_mutex _proxymaplock;
 
 			NetFS::VolumePrxPtr volume;
 			NetFS::ServicePrxPtr service;
diff --git a/netfs/fuse/fuseApp.impl.h b/netfs/fuse/fuseApp.impl.h
index 9d73de0..8dd9f1e 100644
--- a/netfs/fuse/fuseApp.impl.h
+++ b/netfs/fuse/fuseApp.impl.h
@@ -11,7 +11,7 @@ namespace NetFS {
 	FuseApp::setProxy(uint64_t & fh, const Params & ... params)
 	{
 		auto & map = getMap<Handle>();
-		Lock(_lock);
+		Lock(_proxymaplock);
 		while (map.find(fh = ++openHandleId) != map.end()) ;
 		map.emplace(fh, std::make_shared<typename Handle::element_type>(params...));
 	}
@@ -21,7 +21,7 @@ namespace NetFS {
 	FuseApp::getProxy(uint64_t localID)
 	{
 		const auto & map = getMap<Handle>();
-		SharedLock(_lock);
+		SharedLock(_proxymaplock);
 		auto i = map.find(localID);
 		if (i != map.end()) {
 			return i->second;
@@ -34,7 +34,7 @@ namespace NetFS {
 	FuseApp::clearProxy(uint64_t localID)
 	{
 		auto & map = getMap<Handle>();
-		Lock(_lock);
+		Lock(_proxymaplock);
 		map.erase(localID);
 	}
 }
diff --git a/netfs/fuse/fuseAppBase.h b/netfs/fuse/fuseAppBase.h
index 5cc8f7f..f0866bb 100644
--- a/netfs/fuse/fuseAppBase.h
+++ b/netfs/fuse/fuseAppBase.h
@@ -10,6 +10,8 @@
 #include <syslog.h>
 #include <visibility.h>
 #include <buffer.h>
+#include <shared_mutex>
+#include <lockHelpers.h>
 
 class DLL_PUBLIC FuseAppBase {
 	public:
@@ -145,6 +147,7 @@ class DLL_PUBLIC FuseAppBase {
 					for (int t = 0; ; ++t) {
 						try {
 							fuseApp->beforeOperation();
+							SharedLock(fuseApp->_lock);
 							return (fuseApp->*bfunc)(a...);
 						}
 						catch (const std::exception & ex) {
@@ -174,6 +177,9 @@ class DLL_PUBLIC FuseAppBase {
 		}
 
 		static FuseAppBase * fuseApp;
+
+	protected:
+		mutable std::shared_mutex _lock;
 };
 
 #endif