Refactoring on Action - having RuleWithAction and RuleWithActionsProperties

2026-01-01 22:25:45 +03:00 · 2020-09-23 13:48:15 -03:00
parent bd59555206
commit 79d574baa3
44 changed files with 2172 additions and 1533 deletions
--- a/src/rule_with_actions_properties.cc
+++ b/src/rule_with_actions_properties.cc
@@ -0,0 +1,150 @@
+/*
+ * ModSecurity, http://www.modsecurity.org/
+ * Copyright (c) 2015 Trustwave Holdings, Inc. (http://www.trustwave.com/)
+ *
+ * You may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * If any of the files related to licensing are missing or if you have any
+ * other questions related to licensing please contact Trustwave Holdings, Inc.
+ * directly using the email address security@modsecurity.org.
+ *
+ */
+
+#include "modsecurity/rule.h"
+
+#include <stdio.h>
+
+#include <algorithm>
+#include <cstring>
+#include <iostream>
+#include <list>
+#include <memory>
+#include <string>
+#include <utility>
+
+#include "src/actions/set_var.h"
+#include "src/actions/tag.h"
+#include "src/actions/transformations/transformation.h"
+
+
+namespace modsecurity {
+
+
+RuleWithActionsProperties::RuleWithActionsProperties(Transformations *transformations) :
+    m_hasAuditLog(false),
+    m_hasBlock(false),
+    m_hasLog(false),
+    m_hasMultiMatch(false),
+    m_hasNoAuditLog(false),
+    m_hasNoLog(false),
+    m_executeIfMatchActions(),
+    m_setVars(),
+    m_disruptiveAction(nullptr),
+    m_tags(),
+    m_transformations(transformations != nullptr ? *transformations : Transformations())
+{ }
+
+
+
+RuleWithActionsProperties::RuleWithActionsProperties(const RuleWithActionsProperties &o) :
+    m_hasAuditLog(o.m_hasAuditLog),
+    m_hasBlock(o.m_hasBlock),
+    m_hasLog(o.m_hasLog),
+    m_hasMultiMatch(o.m_hasMultiMatch),
+    m_hasNoAuditLog(o.m_hasNoAuditLog),
+    m_hasNoLog(o.m_hasNoLog),
+    m_executeIfMatchActions(),
+    m_setVars(),
+    m_disruptiveAction(o.m_disruptiveAction),
+    m_tags(),
+    m_transformations(o.m_transformations)
+{
+    copyActionsWithRunTimeStrings(o);
+}
+
+
+RuleWithActionsProperties &RuleWithActionsProperties::operator=(const RuleWithActionsProperties &o) {
+    m_hasAuditLog = o.m_hasAuditLog;
+    m_hasBlock = o.m_hasBlock;
+    m_hasLog = o.m_hasLog;
+    m_hasMultiMatch = o.m_hasMultiMatch;
+    m_hasNoAuditLog = o.m_hasNoAuditLog;
+    m_hasNoLog = o.m_hasNoAuditLog;
+    m_disruptiveAction = o.m_disruptiveAction;
+    m_transformations = o.m_transformations;
+    copyActionsWithRunTimeStrings(o);
+
+    return *this;
+}
+
+
+void inline RuleWithActionsProperties::copyActionsWithRunTimeStrings(const RuleWithActionsProperties &o) {
+    for (auto &i : o.m_executeIfMatchActions) {
+        actions::ActionWithRunTimeString *arts = dynamic_cast<actions::ActionWithRunTimeString *>(i.get());
+        if (!arts) {
+            m_executeIfMatchActions.push_back(i);
+            continue;
+        }
+        std::shared_ptr<actions::ActionWithExecution> z(dynamic_cast<actions::ActionWithExecution *>(arts->clone()));
+        actions::ActionWithRunTimeString *aa = dynamic_cast<actions::ActionWithRunTimeString *>(z.get());
+        aa->populate(nullptr);
+        m_executeIfMatchActions.push_back(z);
+    }
+    for (auto &i : o.m_setVars) {
+        actions::ActionWithRunTimeString *arts = dynamic_cast<actions::ActionWithRunTimeString *>(i.get());
+        if (!arts) {
+            m_setVars.push_back(i);
+            continue;
+        }
+        std::shared_ptr<actions::SetVar> z(dynamic_cast<actions::SetVar *>(arts->clone()));
+        actions::ActionWithRunTimeString *aa = dynamic_cast<actions::ActionWithRunTimeString *>(z.get());
+        aa->populate(nullptr);
+        m_setVars.push_back(z);
+    }
+    for (auto &i : o.m_tags) {
+        actions::ActionWithRunTimeString *arts = dynamic_cast<actions::ActionWithRunTimeString *>(i.get());
+        if (!arts) {
+            m_tags.push_back(i);
+            continue;
+        }
+        std::shared_ptr<actions::Tag> z(dynamic_cast<actions::Tag *>(arts->clone()));
+        actions::Tag *aa = dynamic_cast<actions::Tag *>(z.get());
+        aa->populate(nullptr);
+        m_tags.push_back(z);
+    }
+}
+
+
+void RuleWithActionsProperties::populate(const RuleWithActions *r) const {
+    /**
+     *
+     * ActionWithRunTimeString needs to be aware of the Rule that it
+     * belongs to. It is necessary to resolve some variables
+     * (e.g. Rule); Clone and associate are mandatory.
+     *
+     */
+    for (auto &i : m_executeIfMatchActions) {
+        actions::ActionWithRunTimeString *arts = dynamic_cast<actions::ActionWithRunTimeString *>(i.get());
+        if (arts != nullptr) {
+            arts->populate(r);
+        }
+    }
+    for (auto &i : m_setVars) {
+        actions::ActionWithRunTimeString *arts = dynamic_cast<actions::ActionWithRunTimeString *>(i.get());
+        if (arts != nullptr) {
+            arts->populate(r);
+        }
+    }
+    for (auto &i : m_tags) {
+        actions::ActionWithRunTimeString *arts = dynamic_cast<actions::ActionWithRunTimeString *>(i.get());
+        if (arts != nullptr) {
+            arts->populate(r);
+        }
+    }
+}
+
+
+}  // namespace modsecurity