github_mirrors/ModSecurity
3
0
Fork 1
mirror of https://github.com/owasp-modsecurity/ModSecurity.git synced 2025-09-29 19:24:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Creates the RulesSetPhases clas

Browse Source
This commit is contained in:
Felipe Zimmerle
2018-11-14 19:44:28 -03:00
parent 014adabda4
commit 072e4edc53
8 changed files with 304 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/parser/driver.cc
View File

@@ -45,7 +45,7 @@ int Driver::addSecMarker(std::string marker) {
for (int i = 0; i < modsecurity::Phases::NUMBER_OF_PHASES; i++) {
Rule *rule = new Rule(marker);
rule->m_phase = i;
m_rules[i].push_back(rule);
m_rulesSetPhases.insert(rule);
}
return 0;
}
@@ -58,14 +58,15 @@ int Driver::addSecAction(Rule *rule) {
return false;
}
m_rules[rule->m_phase].push_back(rule);
m_rulesSetPhases.insert(rule);
return true;
}
int Driver::addSecRuleScript(RuleScript *rule) {
m_rules[rule->m_phase].push_back(rule);
m_rulesSetPhases.insert(rule);
return true;
}
@@ -118,7 +119,7 @@ int Driver::addSecRule(Rule *rule) {
return false;
}
for (int i = 0; i < modsecurity::Phases::NUMBER_OF_PHASES; i++) {
std::vector<Rule *> rules = m_rules[i];
std::vector<Rule *> rules = m_rulesSetPhases[i];
for (int j = 0; j < rules.size(); j++) {
if (rules[j]->m_ruleId == rule->m_ruleId) {
m_parserError << "Rule id: " << std::to_string(rule->m_ruleId) \
@@ -129,7 +130,7 @@ int Driver::addSecRule(Rule *rule) {
}
lastRule = rule;
m_rules[rule->m_phase].push_back(rule);
m_rulesSetPhases.insert(rule);
return true;
}

2
src/parser/driver.h
View File

@@ -90,6 +90,8 @@ class Driver : public RulesSetProperties {
std::string buffer;
Rule *lastRule;
RulesSetPhases m_rulesSetPhases;
};

14
src/rules_set.cc
View File

@@ -110,7 +110,7 @@ int RulesSet::evaluate(int phase, Transaction *t) {
return 0;
}
std::vector<Rule *> rules = m_rules[phase];
std::vector<Rule *> rules = m_rulesSetPhases[phase];
ms_dbg_a(t, 9, "This phase consists of " \
+ std::to_string(rules.size()) + " rule(s).");
@@ -222,7 +222,10 @@ int RulesSet::evaluate(int phase, Transaction *t) {
int RulesSet::merge(Driver *from) {
int amount_of_rules = 0;
amount_of_rules = mergeProperties(
amount_of_rules = m_rulesSetPhases.append(&from->m_rulesSetPhases,
&m_parserError);
mergeProperties(
dynamic_cast<RulesSetProperties *>(from),
dynamic_cast<RulesSetProperties *>(this),
&m_parserError);
@@ -233,7 +236,10 @@ int RulesSet::merge(Driver *from) {
int RulesSet::merge(RulesSet *from) {
int amount_of_rules = 0;
amount_of_rules = mergeProperties(
amount_of_rules = m_rulesSetPhases.append(&from->m_rulesSetPhases,
&m_parserError);
mergeProperties(
dynamic_cast<RulesSetProperties *>(from),
dynamic_cast<RulesSetProperties *>(this),
&m_parserError);
@@ -254,7 +260,7 @@ void RulesSet::debug(int level, const std::string &id,
void RulesSet::dump() const {
std::cout << "Rules: " << std::endl;
for (int i = 0; i < modsecurity::Phases::NUMBER_OF_PHASES; i++) {
std::vector<Rule *> rules = m_rules[i];
std::vector<Rule *> rules = m_rulesSetPhases[i];
std::cout << "Phase: " << std::to_string(i);
std::cout << " (" << std::to_string(rules.size());
std::cout << " rules)" << std::endl;