github_mirrors/ModSecurity
3
0
Fork 1
mirror of https://github.com/owasp-modsecurity/ModSecurity.git synced 2025-11-15 17:12:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Having a class Rules

Browse Source
This commit is contained in:
Felipe Zimmerle
2018-11-19 13:57:47 -03:00
parent 6504f2664b
commit 57553f08e3
6 changed files with 80 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
headers/modsecurity/rules.h Normal file
View File

@@ -0,0 +1,48 @@
/*
* 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 <stdio.h>
#include <string.h>
#ifdef __cplusplus
#include <ctime>
#include <iostream>
#include <string>
#include <vector>
#include <list>
#endif
#ifndef HEADERS_MODSECURITY_RULES_H_
#define HEADERS_MODSECURITY_RULES_H_
#include "modsecurity/rules_set_properties.h"
#include "modsecurity/modsecurity.h"
#include "modsecurity/transaction.h"
#include "modsecurity/rule.h"
#ifdef __cplusplus
namespace modsecurity {
class Rules : public std::vector<Rule *> {
public:
};
} // namespace modsecurity
#endif
#endif // HEADERS_MODSECURITY_RULES_H_

27
headers/modsecurity/rules_set.h
View File

@@ -25,12 +25,14 @@
#endif
#ifndef HEADERS_MODSECURITY_RULES_H_
#define HEADERS_MODSECURITY_RULES_H_
#ifndef HEADERS_MODSECURITY_RULES_SET_H_
#define HEADERS_MODSECURITY_RULES_SET_H_
#include "modsecurity/rules_set_properties.h"
#include "modsecurity/modsecurity.h"
#include "modsecurity/transaction.h"
#include "modsecurity/rule.h"
#include "modsecurity/rules.h"
#ifdef __cplusplus
@@ -40,17 +42,16 @@ namespace Parser {
class Driver;
}
class RulesSetPhases {
public:
~RulesSetPhases() {
/** Cleanup the rules */
for (int i = 0; i < modsecurity::Phases::NUMBER_OF_PHASES; i++) {
std::vector<Rule *> rules = m_rules[i];
while (rules.empty() == false) {
Rule *rule = rules.back();
rules.pop_back();
Rules *rules = &m_rules[i];
while (rules->empty() == false) {
Rule *rule = rules->back();
rules->pop_back();
if (rule->refCountDecreaseAndCheck()) {
rule = NULL;
}
@@ -84,8 +85,8 @@ class RulesSetPhases {
std::sort (v.begin(), v.end());
for (int i = 0; i < modsecurity::Phases::NUMBER_OF_PHASES; i++) {
for (size_t j = 0; j < from->at(i).size(); j++) {
Rule *rule = from->at(i).at(j);
for (size_t j = 0; j < from->at(i)->size(); j++) {
Rule *rule = from->at(i)->at(j);
if (std::binary_search(v.begin(), v.end(), rule->m_ruleId)) {
if (err != NULL) {
*err << "Rule id: " << std::to_string(rule->m_ruleId) \
@@ -115,10 +116,10 @@ class RulesSetPhases {
}
}
std::vector<modsecurity::Rule *> operator[](int index) { return m_rules[index]; }
std::vector<modsecurity::Rule *> at(int index) { return m_rules[index]; }
Rules *operator[](int index) { return &m_rules[index]; }
Rules *at(int index) { return &m_rules[index]; }
std::vector<modsecurity::Rule *> m_rules[8];
Rules m_rules[8];
};
@@ -192,4 +193,4 @@ int msc_rules_cleanup(RulesSet *rules);
} // namespace modsecurity
#endif
#endif // HEADERS_MODSECURITY_RULES_H_
#endif // HEADERS_MODSECURITY_RULES_SET_H_