New upstream version 8.1.0

storage/source/net/message/nodes/RemoveBuddyGroupMsgEx.cpp

@@ -0,0 +1,124 @@
+#include "RemoveBuddyGroupMsgEx.h"
+
+#include <common/net/message/nodes/RemoveBuddyGroupRespMsg.h>
+#include <net/message/storage/listing/ListChunkDirIncrementalMsgEx.h>
+#include <program/Program.h>
+
+static FhgfsOpsErr checkChunkDirRemovable(const int dirFD)
+{
+   DIR* dir = fdopendir(dirFD);
+   std::unique_ptr<DIR, StorageTk::CloseDirDeleter> _dir(dir);
+
+   while (true)
+   {
+      struct dirent* result;
+
+#if USE_READDIR_R
+      struct dirent buffer;
+      if (readdir_r(dir, &buffer, &result) != 0)
+         break;
+#else
+      errno = 0;
+      result = readdir(dir);
+      if (!result && errno)
+         break;
+#endif
+
+      if (!result)
+         return FhgfsOpsErr_SUCCESS;
+
+      if (strcmp(result->d_name, ".") == 0 || strcmp(result->d_name, "..") == 0)
+         continue;
+
+      struct stat statData;
+
+      const int statRes = ::fstatat(dirfd(dir), result->d_name, &statData, AT_SYMLINK_NOFOLLOW);
+      if (statRes != 0)
+      {
+         LOG(MIRRORING, ERR, "Could not stat something in chunk directory.");
+         return FhgfsOpsErr_INTERNAL;
+      }
+
+      if (!S_ISDIR(statData.st_mode))
+         return FhgfsOpsErr_NOTEMPTY;
+
+      const int subdir = ::openat(dirfd(dir), result->d_name, O_RDONLY);
+      if (subdir < 0)
+      {
+         LOG(MIRRORING, ERR, "Could not open directory in chunk path.");
+         return FhgfsOpsErr_INTERNAL;
+      }
+
+      const FhgfsOpsErr checkRes = checkChunkDirRemovable(subdir);
+      if (checkRes != FhgfsOpsErr_SUCCESS)
+         return checkRes;
+   }
+
+   return FhgfsOpsErr_INTERNAL;
+}
+
+bool RemoveBuddyGroupMsgEx::processIncoming(ResponseContext& ctx)
+{
+   App* app = Program::getApp();
+
+   if (type != NODETYPE_Storage)
+   {
+      ctx.sendResponse(RemoveBuddyGroupRespMsg(FhgfsOpsErr_INTERNAL));
+      return true;
+   }
+
+   uint16_t targetID = app->getMirrorBuddyGroupMapper()->getPrimaryTargetID(groupID);
+   if (app->getTargetMapper()->getNodeID(targetID) != app->getLocalNode().getNumID())
+      targetID = app->getMirrorBuddyGroupMapper()->getSecondaryTargetID(groupID);
+   if (app->getTargetMapper()->getNodeID(targetID) != app->getLocalNode().getNumID())
+   {
+      LOG(MIRRORING, ERR, "Group is not mapped on this target.", groupID);
+      ctx.sendResponse(RemoveBuddyGroupRespMsg(FhgfsOpsErr_INTERNAL));
+      return true;
+   }
+
+   auto* const target = app->getStorageTargets()->getTarget(targetID);
+   if (!target)
+   {
+      LOG(MIRRORING, ERR, "Could not open directory file descriptor.", groupID);
+      ctx.sendResponse(RemoveBuddyGroupRespMsg(FhgfsOpsErr_INTERNAL));
+      return true;
+   }
+
+   const int dirFD = openat(*target->getMirrorFD(), ".", O_RDONLY);
+   if (dirFD < 0)
+   {
+      LOG(MIRRORING, ERR, "Could not open directory file descriptor.", groupID);
+      ctx.sendResponse(RemoveBuddyGroupRespMsg(FhgfsOpsErr_INTERNAL));
+      return true;
+   }
+
+   const FhgfsOpsErr checkRes = checkChunkDirRemovable(dirFD);
+
+   const bool forceAndNotEmpty = checkRes == FhgfsOpsErr_NOTEMPTY && force;
+
+   if (checkRes == FhgfsOpsErr_SUCCESS || forceAndNotEmpty)
+   {
+      if (!checkOnly)
+      {
+         auto* const bgm = Program::getApp()->getMirrorBuddyGroupMapper();
+         const NumNodeID localID = Program::getApp()->getLocalNode().getNumID();
+
+         if (!bgm->unmapMirrorBuddyGroup(groupID, localID))
+         {
+            ctx.sendResponse(RemoveBuddyGroupRespMsg(FhgfsOpsErr_INTERNAL));
+            return true;
+         }
+      }
+
+      ctx.sendResponse(RemoveBuddyGroupRespMsg(FhgfsOpsErr_SUCCESS));
+      return true;
+   }
+   else
+   {
+      ctx.sendResponse(RemoveBuddyGroupRespMsg(checkRes));
+      return true;
+      
+   }
+
+}