mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-11-18 02:10:36 +03:00
105 lines
3.4 KiB
C++
105 lines
3.4 KiB
C++
/*
|
|
* ModSecurity, http://www.modsecurity.org/
|
|
* Copyright (c) 2015 - 2020 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/rules_set_phases.h"
|
|
#include "src/rule_with_operator.h"
|
|
|
|
|
|
namespace modsecurity {
|
|
|
|
|
|
bool RulesSetPhases::insert(std::shared_ptr<Rule> rule) {
|
|
if (rule->getPhase() >= modsecurity::Phases::NUMBER_OF_PHASES) {
|
|
return false;
|
|
}
|
|
m_rulesAtPhase[rule->getPhase()].insert(rule);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
int RulesSetPhases::append(RulesSetPhases *from, std::ostringstream *err) {
|
|
int amount_of_rules = 0;
|
|
std::vector<int64_t> v;
|
|
|
|
for (int i = 0; i < modsecurity::Phases::NUMBER_OF_PHASES; i++) {
|
|
v.reserve(m_rulesAtPhase[i].size());
|
|
for (size_t z = 0; z < m_rulesAtPhase[i].size(); z++) {
|
|
RuleWithOperator *rule_ckc = dynamic_cast<RuleWithOperator *>(m_rulesAtPhase[i].at(z).get());
|
|
//RuleWithOperator *rule_ckc = dynamic_cast<RuleWithOperator *>(m_rulesAtPhase->at(i).get());
|
|
if (!rule_ckc) {
|
|
continue;
|
|
}
|
|
v.push_back(rule_ckc->getId());
|
|
}
|
|
}
|
|
std::sort (v.begin(), v.end());
|
|
|
|
for (int phase = 0; phase < modsecurity::Phases::NUMBER_OF_PHASES; phase++) {
|
|
int res = m_rulesAtPhase[phase].append(from->at(phase), v, err);
|
|
if (res < 0) {
|
|
return res;
|
|
}
|
|
amount_of_rules = amount_of_rules + res;
|
|
|
|
/**
|
|
* An action set in a child will overwrite an action set on a parent.
|
|
*
|
|
*/
|
|
std::vector<std::shared_ptr<actions::Action> > *actions_to = &at(phase)->m_defaultActions;
|
|
std::vector<std::shared_ptr<actions::transformations::Transformation> > *actions_t_to = &at(phase)->m_defaultTransformations;
|
|
if (actions_to->size() == 0 || actions_t_to->size() == 0) {
|
|
std::vector<std::shared_ptr<actions::Action> > *actions_from = &from->at(phase)->m_defaultActions;
|
|
|
|
actions_to->clear();
|
|
for (size_t j = 0; j < actions_from->size(); j++) {
|
|
actions_to->push_back(actions_from->at(j));
|
|
}
|
|
|
|
std::vector<std::shared_ptr<actions::transformations::Transformation> > *actions_t_from = &from->at(phase)->m_defaultTransformations;
|
|
actions_t_to->clear();
|
|
for (size_t j = 0; j < actions_t_from->size(); j++) {
|
|
actions_t_to->push_back(actions_t_from->at(j));
|
|
}
|
|
at(phase)->fixDefaultActions();
|
|
}
|
|
}
|
|
|
|
return amount_of_rules;
|
|
}
|
|
|
|
void RulesSetPhases::dump() const {
|
|
for (int i = 0; i <= modsecurity::Phases::NUMBER_OF_PHASES; i++) {
|
|
const Rules *rules = &m_rulesAtPhase[i];
|
|
std::cout << "Phase: " << std::to_string(i);
|
|
std::cout << " (" << std::to_string(m_rulesAtPhase[i].size());
|
|
std::cout << " rules)" << std::endl;
|
|
m_rulesAtPhase[i].dump();
|
|
}
|
|
}
|
|
|
|
|
|
Rules *RulesSetPhases::operator[](int index) {
|
|
return &m_rulesAtPhase[index];
|
|
}
|
|
|
|
|
|
Rules *RulesSetPhases::at(int index) {
|
|
return &m_rulesAtPhase[index];
|
|
}
|
|
|
|
|
|
} // namespace modsecurity
|