Adds support to the action `accuracy'

This commit is contained in:
Felipe Zimmerle
2016-02-10 13:34:41 -03:00
parent 5a2a81a568
commit 714df8db20
6 changed files with 100 additions and 3 deletions

47
src/actions/accuracy.cc Normal file
View File

@@ -0,0 +1,47 @@
/*
* 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 "actions/accuracy.h"
#include <iostream>
#include <string>
#include "actions/action.h"
#include "modsecurity/transaction.h"
#include "src/utils.h"
#include "modsecurity/rule.h"
#include "src/macro_expansion.h"
namespace modsecurity {
namespace actions {
Accuracy::Accuracy(std::string action)
: Action(action, ConfigurationKind),
m_accuracy_str(action) {
if (m_accuracy_str.at(0) == '\'') {
m_accuracy_str.erase(0, 1);
m_accuracy_str.pop_back();
}
m_accuracy = std::stoi(m_accuracy_str);
}
bool Accuracy::evaluate(Rule *rule, Transaction *transaction) {
rule->m_accuracy = m_accuracy;
return true;
}
} // namespace actions
} // namespace modsecurity

45
src/actions/accuracy.h Normal file
View File

@@ -0,0 +1,45 @@
/*
* 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 <string>
#include "actions/action.h"
#ifndef SRC_ACTIONS_ACCURACY_H_
#define SRC_ACTIONS_ACCURACY_H_
class Transaction;
namespace modsecurity {
class Transaction;
namespace actions {
class Accuracy : public Action {
public:
explicit Accuracy(std::string action);
bool evaluate(Rule *rule, Transaction *transaction) override;
private:
std::string m_accuracy_str;
int m_accuracy;
};
} // namespace actions
} // namespace modsecurity
#endif // SRC_ACTIONS_ACCURACY_H_