fix(invitation): add locks to protect concurrent rules edition

The purpose is to avoid simultaneous modification of rules in some specific cases

Signed-off-by: Thierry Bugier <[email protected]>

Author: btry

inc/fusioninventory.class.php

@@ -39,6 +39,9 @@ class PluginFlyvemdmFusionInventory {
 
    const RULE_NAME = 'Flyve MDM invitation to entity';
 
+   const LOCK_NAME = 'flyvemdm_entityRule';
+
+
    /**
     * Creates or updates an entity rule
     *
@@ -52,6 +55,10 @@ public function addInvitationRule(PluginFlyvemdmInvitation $invitation) {
          Session::addMessageAfterRedirect(__('Unable to get rule for entity', 'flyvemdm'),
             true, ERROR);
          return;
+      } catch (EntityRuleEditionException $exception) {
+         Session::addMessageAfterRedirect(__('Cannot get lock for entity rules edition', 'flyvemdm'),
+            true, ERROR);
+         return;
       }
 
       $ruleCriteria = new RuleCriteria();
@@ -117,6 +124,19 @@ public function deleteInvitationRuleCriteria(PluginFlyvemdmInvitation $invitatio
       ]);
 
       $ruleId = $row[$ruleFk];
+
+      // get a lock
+      $attempts = 0;
+      $locked = 0;
+      do {
+         $locked = $DB->getLock(self::LOCK_NAME);
+         usleep(50000); // 50 milliseconds
+         $attempts++;
+      } while ($locked !== 1 && $attempts < 10);
+      if ($locked !== 1) {
+         return; // No lock, then give up disabling
+      }
+
       $rows = $ruleCriteria->find("`$ruleFk` = '$ruleId' AND `criteria` = 'tag' AND `condition` = '0'");
       if (count($rows) === 0) {
          $rule = new PluginFusioninventoryInventoryRuleEntity();
@@ -125,6 +145,7 @@ public function deleteInvitationRuleCriteria(PluginFlyvemdmInvitation $invitatio
             'is_active' => '0',
          ]);
       }
+      $DB->releaseLock(self::LOCK_NAME);
    }
 
    /**
@@ -144,10 +165,23 @@ private function getRuleCriteriaValue(PluginFlyvemdmInvitation $invitation) {
     *
     * @return PluginFusioninventoryInventoryRuleEntity|null
     * @throws FusionInventoryRuleInconsistency
+    * @throws EntityRuleEditionException
     */
    private function getRule($entityId, $create = true) {
       global $DB;
 
+      // get a lock
+      $attempts = 0;
+      $locked = 0;
+      do {
+         $locked = $DB->getLock(self::LOCK_NAME);
+         usleep(50000); // 50 milliseconds
+         $attempts++;
+      } while ($locked !== 1 && $attempts < 10);
+      if ($locked !== 1) {
+         throw new EntityRuleEditionException(__('Cannot get lock for entity rules edition'));
+      }
+
       $ruleEntityTable = PluginFusioninventoryInventoryRuleEntity::getTable();
       $ruleActionTable = RuleAction::getTable();
       $request = [
@@ -175,13 +209,16 @@ private function getRule($entityId, $create = true) {
          $rule = new PluginFusioninventoryInventoryRuleEntity();
          $row = $result->next();
          $rule->getFromDB($row['id']);
+         $DB->releaseLock(self::LOCK_NAME);
          return $rule;
       }
       if ($result->count() > 1) {
+         $DB->releaseLock(self::LOCK_NAME);
          throw new FusionInventoryRuleInconsistency(__('Import rule is not unique'));
       }
 
       if (!$create) {
+         $DB->releaseLock(self::LOCK_NAME);
          return null;
       }
 
@@ -203,6 +240,7 @@ private function getRule($entityId, $create = true) {
          'field'                    => Entity::getForeignKeyField(),
          'value'                    => $entityId,
       ]);
+      $DB->releaseLock(self::LOCK_NAME);
       return $rule;
    }
 }