Cleanup on Action class

This commit is contained in:
Felipe Zimmerle 2019-03-15 01:56:40 -03:00
parent 4cd9fb39e6
commit 37c9e710e2
No known key found for this signature in database
GPG Key ID: E6DFB08CE8B11277
112 changed files with 299 additions and 308 deletions

View File

@ -39,35 +39,27 @@ namespace actions {
class Action { class Action {
public: public:
explicit Action(const std::string& _action) explicit Action(const std::string& _action)
: m_isNone(false), : m_actionKind(2),
temporaryAction(false),
action_kind(2),
m_name(nullptr), m_name(nullptr),
m_parser_payload("") { m_parser_payload("") {
set_name_and_payload(_action); set_name_and_payload(_action);
} }
Action(const std::string& _action, int kind) Action(const std::string& _action, int kind)
: m_isNone(false), : m_actionKind(kind),
temporaryAction(false),
action_kind(kind),
m_name(nullptr), m_name(nullptr),
m_parser_payload("") { m_parser_payload("") {
set_name_and_payload(_action); set_name_and_payload(_action);
} }
Action(const Action &a) Action(const Action &a)
: m_isNone(a.m_isNone), : m_actionKind(a.m_actionKind),
temporaryAction(a.temporaryAction),
action_kind(a.action_kind),
m_name(a.m_name), m_name(a.m_name),
m_parser_payload(a.m_parser_payload) { } m_parser_payload(a.m_parser_payload) { }
Action &operator=(const Action& a) { Action &operator=(const Action& a) {
m_isNone = a.m_isNone; m_actionKind = a.m_actionKind;
temporaryAction = a.temporaryAction;
action_kind = a.action_kind;
m_name = a.m_name; m_name = a.m_name;
m_parser_payload = a.m_parser_payload; m_parser_payload = a.m_parser_payload;
return *this; return *this;
@ -75,52 +67,24 @@ class Action {
virtual ~Action() { } virtual ~Action() { }
virtual bool init(std::string *error) { return true; }
virtual std::string execute(const std::string &exp, virtual std::string execute(const std::string &exp,
Transaction *transaction); Transaction *transaction);
virtual bool execute(RuleWithActions *rule, Transaction *transaction); virtual bool execute(RuleWithActions *rule,
Transaction *transaction);
/** /**
* This method is meant to be used by transformations a particular * This method is meant to be used by transformations a particular
* type of action. * type of action.
* *
*/ */
virtual void execute(Transaction *t, virtual void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
}; };
virtual bool init(std::string *error) { return true; }
virtual bool isDisruptive() { return false; } virtual bool isDisruptive() { return false; }
void set_name_and_payload(const std::string& data) {
size_t pos = data.find(":");
std::string t = "t:";
if (data.compare(0, t.length(), t) == 0) {
pos = data.find(":", 2);
}
if (pos == std::string::npos) {
m_name = std::shared_ptr<std::string>(new std::string(data));
return;
}
m_name = std::shared_ptr<std::string>(new std::string(data, 0, pos));
m_parser_payload = std::string(data, pos + 1, data.length());
if (m_parser_payload.at(0) == '\'' && m_parser_payload.size() > 2) {
m_parser_payload.erase(0, 1);
m_parser_payload.pop_back();
}
}
bool m_isNone;
bool temporaryAction;
int action_kind;
std::shared_ptr<std::string> m_name;
std::string m_parser_payload;
/** /**
* *
* Define the action kind regarding to the execution time. * Define the action kind regarding to the execution time.
@ -152,7 +116,35 @@ class Action {
*/ */
RunTimeOnlyIfMatchKind, RunTimeOnlyIfMatchKind,
}; };
};
int m_actionKind;
std::shared_ptr<std::string> m_name;
std::string m_parser_payload;
private:
void set_name_and_payload(const std::string& data) {
size_t pos = data.find(":");
std::string t = "t:";
if (data.compare(0, t.length(), t) == 0) {
pos = data.find(":", 2);
}
if (pos == std::string::npos) {
m_name = std::shared_ptr<std::string>(new std::string(data));
return;
}
m_name = std::shared_ptr<std::string>(new std::string(data, 0, pos));
m_parser_payload = std::string(data, pos + 1, data.length());
if (m_parser_payload.at(0) == '\'' && m_parser_payload.size() > 2) {
m_parser_payload.erase(0, 1);
m_parser_payload.pop_back();
}
}
};
} // namespace actions } // namespace actions

View File

@ -89,6 +89,12 @@
typedef struct ModSecurity_t modsecurity; typedef struct ModSecurity_t modsecurity;
#else #else
namespace modsecurity { namespace modsecurity {
/**
* Further that will be changed to be a stack-based string,
* for the benefit of performance.
*/
using ModSecString = std::string;
/** /**
* *
* The Phases enumerator consists in mapping the different stages of a * The Phases enumerator consists in mapping the different stages of a

View File

@ -64,18 +64,17 @@ using MatchActionsPtr = std::vector<actions::Action *>;
using XmlNSs = std::vector<std::shared_ptr<actions::XmlNS> >; using XmlNSs = std::vector<std::shared_ptr<actions::XmlNS> >;
using XmlNSsPtr = std::vector<actions::XmlNS *>; using XmlNSsPtr = std::vector<actions::XmlNS *>;
using ModSecStackString = std::basic_string<char, std::char_traits<char>, std::allocator<char> >;
class TransformationResult { class TransformationResult {
public: public:
TransformationResult( TransformationResult(
ModSecStackString *after, ModSecString *after,
std::string *transformation) std::string *transformation)
: m_after(*after), : m_after(*after),
m_transformation(transformation) { }; m_transformation(transformation) { };
explicit TransformationResult( explicit TransformationResult(
ModSecStackString *after) ModSecString *after)
: m_after(*after), : m_after(*after),
m_transformation(nullptr) { }; m_transformation(nullptr) { };
@ -84,7 +83,7 @@ class TransformationResult {
m_transformation(t2.m_transformation) { }; m_transformation(t2.m_transformation) { };
ModSecStackString *getAfter() { ModSecString *getAfter() {
return &m_after; return &m_after;
} }
@ -95,7 +94,7 @@ class TransformationResult {
private: private:
ModSecStackString m_after; ModSecString m_after;
std::string *m_transformation; std::string *m_transformation;
}; };
@ -224,7 +223,7 @@ class RuleWithActions : public Rule {
static void executeTransformation( static void executeTransformation(
Transaction *transaction, Transaction *transaction,
ModSecStackString in, ModSecString in,
TransformationsResults *ret, TransformationsResults *ret,
Transformation *transformation); Transformation *transformation);

View File

@ -40,52 +40,20 @@ namespace modsecurity {
class Rules { class Rules {
public: public:
void dump() const { void dump() const;
for (int j = 0; j < m_rules.size(); j++) {
std::cout << " Rule ID: " << m_rules.at(j)->getReference();
std::cout << "--" << m_rules.at(j) << std::endl;
}
}
int append(Rules *from, const std::vector<int64_t> &ids, std::ostringstream *err) { int append(Rules *from, const std::vector<int64_t> &ids, std::ostringstream *err);
size_t j = 0;
for (; j < from->size(); j++) {
RuleWithOperator *rule = dynamic_cast<RuleWithOperator *>(from->at(j).get());
if (rule && std::binary_search(ids.begin(), ids.end(), rule->getId())) {
if (err != NULL) {
*err << "Rule id: " << std::to_string(rule->getId()) \
<< " is duplicated" << std::endl;
}
return -1;
}
}
m_rules.insert(m_rules.end(), from->m_rules.begin(), from->m_rules.end());
return j;
}
bool insert(const std::shared_ptr<Rule> &rule) { bool insert(const std::shared_ptr<Rule> &rule);
return insert(rule, nullptr, nullptr);
}
bool insert(std::shared_ptr<Rule> rule, const std::vector<int64_t> *ids, std::ostringstream *err) { bool insert(std::shared_ptr<Rule> rule, const std::vector<int64_t> *ids, std::ostringstream *err);
RuleWithOperator *r = dynamic_cast<RuleWithOperator *>(rule.get());
if (r && ids != nullptr && std::binary_search(ids->begin(), ids->end(), r->getId())) { size_t size() const;
if (err != nullptr) { std::shared_ptr<Rule> operator[](int index) const;
*err << "Rule id: " << std::to_string(r->getId()) \ std::shared_ptr<Rule> at(int index) const;
<< " is duplicated" << std::endl;
}
return false;
}
m_rules.push_back(rule);
return true;
}
void fixDefaultActions(); void fixDefaultActions();
size_t size() const { return m_rules.size(); }
std::shared_ptr<Rule> operator[](int index) const { return m_rules[index]; }
std::shared_ptr<Rule> at(int index) const { return m_rules[index]; }
std::vector<std::shared_ptr<actions::Action> > m_defaultActions; std::vector<std::shared_ptr<actions::Action> > m_defaultActions;
std::vector<std::shared_ptr<actions::transformations::Transformation> > m_defaultTransformations; std::vector<std::shared_ptr<actions::transformations::Transformation> > m_defaultTransformations;

View File

@ -49,7 +49,6 @@ typedef struct Rules_t RulesSet;
#include "modsecurity/collection/collection.h" #include "modsecurity/collection/collection.h"
#include "modsecurity/variable_origin.h" #include "modsecurity/variable_origin.h"
#ifndef NO_LOGS #ifndef NO_LOGS
#define ms_dbg(b, c) \ #define ms_dbg(b, c) \
do { \ do { \

View File

@ -42,10 +42,6 @@ pkginclude_HEADERS = \
../headers/modsecurity/intervention.h \ ../headers/modsecurity/intervention.h \
../headers/modsecurity/modsecurity.h \ ../headers/modsecurity/modsecurity.h \
../headers/modsecurity/rule.h \ ../headers/modsecurity/rule.h \
../headers/modsecurity/rule_marker.h \
../headers/modsecurity/rule_unconditional.h \
../headers/modsecurity/rule_with_actions.h \
../headers/modsecurity/rule_with_operator.h \
../headers/modsecurity/rules.h \ ../headers/modsecurity/rules.h \
../headers/modsecurity/rule_message.h \ ../headers/modsecurity/rule_message.h \
../headers/modsecurity/rules_set.h \ ../headers/modsecurity/rules_set.h \
@ -286,6 +282,7 @@ libmodsecurity_la_SOURCES = \
debug_log/debug_log_writer.cc \ debug_log/debug_log_writer.cc \
run_time_string.cc \ run_time_string.cc \
rule.cc \ rule.cc \
rules.cc \
rule_unconditional.cc \ rule_unconditional.cc \
rule_with_actions.cc \ rule_with_actions.cc \
rule_with_operator.cc \ rule_with_operator.cc \

View File

@ -21,7 +21,7 @@
#include "modsecurity/actions/action.h" #include "modsecurity/actions/action.h"
#include "modsecurity/transaction.h" #include "modsecurity/transaction.h"
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace actions { namespace actions {

View File

@ -24,6 +24,7 @@
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/intervention.h" #include "modsecurity/intervention.h"
#include "src/actions/data/status.h" #include "src/actions/data/status.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace actions { namespace actions {

View File

@ -20,6 +20,8 @@
#include "modsecurity/transaction.h" #include "modsecurity/transaction.h"
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace actions { namespace actions {

View File

@ -21,7 +21,7 @@
#include "modsecurity/actions/action.h" #include "modsecurity/actions/action.h"
#include "modsecurity/transaction.h" #include "modsecurity/transaction.h"
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace actions { namespace actions {

View File

@ -22,6 +22,7 @@
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/modsecurity.h" #include "modsecurity/modsecurity.h"
#include "src/utils/string.h" #include "src/utils/string.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {

View File

@ -21,6 +21,7 @@
#include "modsecurity/actions/action.h" #include "modsecurity/actions/action.h"
#include "modsecurity/transaction.h" #include "modsecurity/transaction.h"
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {

View File

@ -20,6 +20,8 @@
#include "modsecurity/transaction.h" #include "modsecurity/transaction.h"
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace actions { namespace actions {

View File

@ -21,6 +21,8 @@
#include "modsecurity/transaction.h" #include "modsecurity/transaction.h"
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "src/utils/string.h" #include "src/utils/string.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace actions { namespace actions {

View File

@ -30,6 +30,7 @@
#include "src/variables/tx.h" #include "src/variables/tx.h"
#include "src/variables/user.h" #include "src/variables/user.h"
#include "src/variables/variable.h" #include "src/variables/variable.h"
#include "modsecurity/rule_with_operator.h"
namespace modsecurity { namespace modsecurity {
namespace actions { namespace actions {

View File

@ -33,8 +33,8 @@ namespace transformations {
void Base64Decode::execute(Transaction *t, void Base64Decode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
std::string value(in.c_str(), in.size()); std::string value(in.c_str(), in.size());
std::string ret = Utils::Base64::decode(value); std::string ret = Utils::Base64::decode(value);
out.assign(ret.c_str(), ret.size()); out.assign(ret.c_str(), ret.size());

View File

@ -34,8 +34,8 @@ class Base64Decode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -33,8 +33,8 @@ namespace transformations {
void Base64DecodeExt::execute(Transaction *t, void Base64DecodeExt::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
std::string ret = Utils::Base64::decode_forgiven(in.c_str()); std::string ret = Utils::Base64::decode_forgiven(in.c_str());
out.assign(ret.c_str(), ret.size()); out.assign(ret.c_str(), ret.size());
} }

View File

@ -34,8 +34,8 @@ class Base64DecodeExt : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -33,8 +33,8 @@ namespace transformations {
void Base64Encode::execute(Transaction *t, void Base64Encode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
std::string ret = Utils::Base64::encode( std::string ret = Utils::Base64::encode(
std::string(in.c_str(), in.size())); std::string(in.c_str(), in.size()));
out.assign(ret.c_str(), ret.size()); out.assign(ret.c_str(), ret.size());

View File

@ -34,8 +34,8 @@ class Base64Encode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -32,8 +32,8 @@ namespace transformations {
void CmdLine::execute(Transaction *t, void CmdLine::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
int space = 0; int space = 0;
for (auto& a : in) { for (auto& a : in) {

View File

@ -34,8 +34,8 @@ class CmdLine : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -32,8 +32,8 @@ namespace transformations {
void CompressWhitespace::execute(Transaction *t, void CompressWhitespace::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
int inWhiteSpace = 0; int inWhiteSpace = 0;
size_t i = 0; size_t i = 0;
out.reserve(in.size()); out.reserve(in.size());

View File

@ -34,8 +34,8 @@ class CompressWhitespace : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -35,17 +35,19 @@ namespace transformations {
void CssDecode::execute(Transaction *t, void CssDecode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
size_t s = in.size();
char *tmp = reinterpret_cast<char *>( char *tmp = reinterpret_cast<char *>(
malloc(sizeof(char) * in.size() + 1)); malloc(sizeof(char) * s + 1));
memcpy(tmp, in.c_str(), in.size() + 1); memcpy(tmp, in.c_str(), s + 1);
tmp[in.size()] = '\0'; tmp[s] = '\0';
CssDecode::css_decode_inplace(reinterpret_cast<unsigned char *>(tmp), size_t r = CssDecode::css_decode_inplace(reinterpret_cast<unsigned char *>(tmp),
in.size()); s);
out.assign(tmp, 0, in.size()); out.assign(tmp, r);
free(tmp); free(tmp);
} }

View File

@ -35,8 +35,8 @@ class CssDecode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static int css_decode_inplace(unsigned char *input, int64_t input_len); static int css_decode_inplace(unsigned char *input, int64_t input_len);
}; };

View File

@ -136,8 +136,8 @@ int EscapeSeqDecode::ansi_c_sequences_decode_inplace(unsigned char *input,
void EscapeSeqDecode::execute(Transaction *t, void EscapeSeqDecode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *tmp = (unsigned char *) malloc(sizeof(char) unsigned char *tmp = (unsigned char *) malloc(sizeof(char)
* in.size() + 1); * in.size() + 1);
memcpy(tmp, in.c_str(), in.size() + 1); memcpy(tmp, in.c_str(), in.size() + 1);

View File

@ -34,8 +34,8 @@ class EscapeSeqDecode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static int ansi_c_sequences_decode_inplace(unsigned char *input, int input_len); static int ansi_c_sequences_decode_inplace(unsigned char *input, int input_len);
}; };

View File

@ -33,8 +33,8 @@ namespace transformations {
void HexDecode::execute(Transaction *t, void HexDecode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
int size = 0; int size = 0;

View File

@ -34,8 +34,8 @@ class HexDecode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static int inplace(unsigned char *data, int len); static int inplace(unsigned char *data, int len);
}; };

View File

@ -33,8 +33,8 @@ namespace transformations {
void HexEncode::execute(Transaction *t, void HexEncode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
std::stringstream result; std::stringstream result;
for (std::size_t i=0; i < in.length(); i++) { for (std::size_t i=0; i < in.length(); i++) {
int ii = reinterpret_cast<char>(in[i]); int ii = reinterpret_cast<char>(in[i]);

View File

@ -34,8 +34,8 @@ class HexEncode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };

View File

@ -34,8 +34,8 @@ namespace transformations {
void HtmlEntityDecode::execute(Transaction *t, void HtmlEntityDecode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
input = reinterpret_cast<unsigned char *> input = reinterpret_cast<unsigned char *>

View File

@ -37,8 +37,8 @@ class HtmlEntityDecode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static int inplace(unsigned char *input, uint64_t input_len); static int inplace(unsigned char *input, uint64_t input_len);
}; };

View File

@ -35,8 +35,8 @@ namespace transformations {
void JsDecode::execute(Transaction *t, void JsDecode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
input = reinterpret_cast<unsigned char *> input = reinterpret_cast<unsigned char *>

View File

@ -34,8 +34,8 @@ class JsDecode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static int inplace(unsigned char *input, uint64_t input_len); static int inplace(unsigned char *input, uint64_t input_len);
}; };

View File

@ -32,8 +32,8 @@ namespace transformations {
void Length::execute(Transaction *t, void Length::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
out.assign(std::to_string(in.size()).c_str()); out.assign(std::to_string(in.size()).c_str());
} }

View File

@ -34,8 +34,8 @@ class Length : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -28,8 +28,8 @@ namespace transformations {
void LowerCase::execute(Transaction *t, void LowerCase::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
std::locale loc; std::locale loc;
out.resize(in.size()); out.resize(in.size());
for (std::string::size_type i=0; i < in.size(); ++i) { for (std::string::size_type i=0; i < in.size(); ++i) {

View File

@ -36,8 +36,8 @@ class LowerCase : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -32,8 +32,8 @@ namespace transformations {
void Md5::execute(Transaction *t, void Md5::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
std::string ret = Utils::Md5::digest(std::string(in.c_str(), in.size())); std::string ret = Utils::Md5::digest(std::string(in.c_str(), in.size()));
out.assign(ret.c_str(), ret.size()); out.assign(ret.c_str(), ret.size());

View File

@ -34,8 +34,8 @@ class Md5 : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -32,8 +32,8 @@ namespace transformations {
void None::execute(Transaction *t, void None::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { } ModSecString &out) { }
} // namespace transformations } // namespace transformations

View File

@ -32,11 +32,15 @@ class None : public Transformation {
public: public:
explicit None(const std::string &action) explicit None(const std::string &action)
: Transformation(action) : Transformation(action)
{ m_isNone = true; } { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
bool isNone() override {
return true;
}
}; };
} // namespace transformations } // namespace transformations

View File

@ -34,8 +34,8 @@ namespace transformations {
void NormalisePath::execute(Transaction *t, void NormalisePath::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
int changed = 0; int changed = 0;
char *tmp = reinterpret_cast<char *>( char *tmp = reinterpret_cast<char *>(

View File

@ -34,8 +34,8 @@ class NormalisePath : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static int normalize_path_inplace(unsigned char *input, int input_len, static int normalize_path_inplace(unsigned char *input, int input_len,
int win, int *changed); int win, int *changed);

View File

@ -35,8 +35,8 @@ namespace transformations {
void NormalisePathWin::execute(Transaction *t, void NormalisePathWin::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
int changed; int changed;
char *tmp = reinterpret_cast<char *>( char *tmp = reinterpret_cast<char *>(

View File

@ -34,8 +34,8 @@ class NormalisePathWin : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -33,8 +33,8 @@ namespace transformations {
void ParityEven7bit::execute(Transaction *t, void ParityEven7bit::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
input = reinterpret_cast<unsigned char *> input = reinterpret_cast<unsigned char *>

View File

@ -34,8 +34,8 @@ class ParityEven7bit : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static bool inplace(unsigned char *input, uint64_t input_len); static bool inplace(unsigned char *input, uint64_t input_len);
}; };

View File

@ -33,8 +33,8 @@ namespace transformations {
void ParityOdd7bit::execute(Transaction *t, void ParityOdd7bit::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
input = reinterpret_cast<unsigned char *> input = reinterpret_cast<unsigned char *>

View File

@ -34,8 +34,8 @@ class ParityOdd7bit : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static bool inplace(unsigned char *input, uint64_t input_len); static bool inplace(unsigned char *input, uint64_t input_len);
}; };

View File

@ -33,8 +33,8 @@ namespace transformations {
void ParityZero7bit::execute(Transaction *t, void ParityZero7bit::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
input = reinterpret_cast<unsigned char *> input = reinterpret_cast<unsigned char *>

View File

@ -34,8 +34,8 @@ class ParityZero7bit : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static bool inplace(unsigned char *input, uint64_t input_len); static bool inplace(unsigned char *input, uint64_t input_len);
}; };

View File

@ -33,8 +33,8 @@ namespace transformations {
void RemoveComments::execute(Transaction *t, void RemoveComments::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
input = reinterpret_cast<unsigned char *> input = reinterpret_cast<unsigned char *>

View File

@ -35,8 +35,8 @@ class RemoveComments : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override;; ModSecString &out) override;;
}; };

View File

@ -32,8 +32,8 @@ namespace transformations {
void RemoveCommentsChar::execute(Transaction *t, void RemoveCommentsChar::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
int64_t i; int64_t i;
out = in; out = in;

View File

@ -34,8 +34,8 @@ class RemoveCommentsChar : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -34,8 +34,8 @@ namespace transformations {
void RemoveNulls::execute(Transaction *t, void RemoveNulls::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
int64_t i; int64_t i;
out = in; out = in;

View File

@ -34,8 +34,8 @@ class RemoveNulls : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -33,8 +33,8 @@ namespace transformations {
void RemoveWhitespace::execute(Transaction *t, void RemoveWhitespace::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
out = in; out = in;
int64_t i = 0; int64_t i = 0;

View File

@ -34,8 +34,8 @@ class RemoveWhitespace : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -33,8 +33,8 @@ namespace transformations {
void ReplaceComments::execute(Transaction *t, void ReplaceComments::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
uint64_t i, j, incomment; uint64_t i, j, incomment;
char *input = reinterpret_cast<char *>( char *input = reinterpret_cast<char *>(

View File

@ -34,8 +34,8 @@ class ReplaceComments : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -32,8 +32,8 @@ namespace transformations {
void ReplaceNulls::execute(Transaction *t, void ReplaceNulls::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
int64_t i; int64_t i;
out = in; out = in;

View File

@ -34,8 +34,8 @@ class ReplaceNulls : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -33,8 +33,8 @@ namespace transformations {
void Sha1::execute(Transaction *t, void Sha1::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
auto a = Utils::Sha1::digest( auto a = Utils::Sha1::digest(
std::string(in.c_str(), in.size()) std::string(in.c_str(), in.size())

View File

@ -34,8 +34,8 @@ class Sha1 : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -42,8 +42,8 @@ namespace transformations {
#endif #endif
void SqlHexDecode::execute(Transaction *t, void SqlHexDecode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
int size = 0; int size = 0;

View File

@ -34,8 +34,8 @@ class SqlHexDecode : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static int inplace(unsigned char *data, int len); static int inplace(unsigned char *data, int len);

View File

@ -114,6 +114,7 @@ Transformation* Transformation::instantiate(std::string a) {
return new Transformation(a); return new Transformation(a);
} }
} // namespace transformations } // namespace transformations
} // namespace actions } // namespace actions
} // namespace modsecurity } // namespace modsecurity

View File

@ -32,12 +32,7 @@ class Transformation : public Action {
explicit Transformation(const std::string& _action) explicit Transformation(const std::string& _action)
: Action(_action, RunTimeBeforeMatchAttemptKind) { } : Action(_action, RunTimeBeforeMatchAttemptKind) { }
void execute(Transaction *t, virtual bool isNone() { return false; }
ModSecStackString &in,
ModSecStackString &out) override {
// FIXME: this should be remove as soon as #1974 got fixed.
out.assign(in.c_str(), in.length());
}
static Transformation* instantiate(std::string a); static Transformation* instantiate(std::string a);
}; };

View File

@ -31,27 +31,27 @@ namespace actions {
namespace transformations { namespace transformations {
void Trim::ltrim(ModSecStackString *s) { void Trim::ltrim(ModSecString *s) {
s->erase(s->begin(), std::find_if(s->begin(), s->end(), s->erase(s->begin(), std::find_if(s->begin(), s->end(),
std::not1(std::ptr_fun<int, int>(std::isspace)))); std::not1(std::ptr_fun<int, int>(std::isspace))));
} }
void Trim::rtrim(ModSecStackString *s) { void Trim::rtrim(ModSecString *s) {
s->erase(std::find_if(s->rbegin(), s->rend(), s->erase(std::find_if(s->rbegin(), s->rend(),
std::not1(std::ptr_fun<int, int>(std::isspace))).base(), s->end()); std::not1(std::ptr_fun<int, int>(std::isspace))).base(), s->end());
} }
void Trim::trim(ModSecStackString *s) { void Trim::trim(ModSecString *s) {
rtrim(s); rtrim(s);
ltrim(s); ltrim(s);
} }
void Trim::execute(Transaction *t, void Trim::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
out = in; out = in;
trim(&out); trim(&out);
}; };

View File

@ -34,12 +34,12 @@ class Trim : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
void ltrim(ModSecStackString *s); void ltrim(ModSecString *s);
void rtrim(ModSecStackString *s); void rtrim(ModSecString *s);
void trim(ModSecStackString *s); void trim(ModSecString *s);
}; };
} // namespace transformations } // namespace transformations

View File

@ -33,8 +33,8 @@ namespace transformations {
void TrimLeft::execute(Transaction *t, void TrimLeft::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
out = in; out = in;
ltrim(&out); ltrim(&out);
}; };

View File

@ -35,8 +35,8 @@ class TrimLeft : public Trim {
: Trim(action) { }; : Trim(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -32,8 +32,8 @@ namespace transformations {
void TrimRight::execute(Transaction *t, void TrimRight::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
out = in; out = in;
rtrim(&out); rtrim(&out);
}; };

View File

@ -35,8 +35,8 @@ class TrimRight : public Trim {
: Trim(action) { }; : Trim(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -28,8 +28,8 @@ namespace transformations {
void UpperCase::execute(Transaction *t, void UpperCase::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
std::locale loc; std::locale loc;
out.reserve(in.size()); out.reserve(in.size());
for (std::string::size_type i=0; i < in.size(); ++i) { for (std::string::size_type i=0; i < in.size(); ++i) {

View File

@ -36,8 +36,8 @@ class UpperCase : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -33,8 +33,8 @@ namespace transformations {
void UrlDecode::execute(Transaction *t, void UrlDecode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *val(NULL); unsigned char *val(NULL);
int invalid_count = 0; int invalid_count = 0;
int changed; int changed;

View File

@ -36,8 +36,8 @@ class UrlDecode : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
}; };
} // namespace transformations } // namespace transformations

View File

@ -39,8 +39,8 @@ namespace transformations {
void UrlDecodeUni::execute(Transaction *t, void UrlDecodeUni::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
input = reinterpret_cast<unsigned char *> input = reinterpret_cast<unsigned char *>

View File

@ -35,8 +35,8 @@ class UrlDecodeUni : public Transformation {
: Transformation(action) { } : Transformation(action) { }
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static int inplace(unsigned char *input, uint64_t input_len, static int inplace(unsigned char *input, uint64_t input_len,
Transaction *transaction); Transaction *transaction);

View File

@ -82,8 +82,8 @@ std::string UrlEncode::url_enc(const char *input,
void UrlEncode::execute(Transaction *t, void UrlEncode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
int changed; int changed;
std::string ret = url_enc(in.c_str(), in.size(), &changed); std::string ret = url_enc(in.c_str(), in.size(), &changed);

View File

@ -34,8 +34,8 @@ class UrlEncode : public Transformation {
: Transformation(action) { }; : Transformation(action) { };
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static std::string url_enc(const char *input, static std::string url_enc(const char *input,
unsigned int input_len, int *changed); unsigned int input_len, int *changed);

View File

@ -34,8 +34,8 @@ namespace transformations {
void Utf8ToUnicode::execute(Transaction *t, void Utf8ToUnicode::execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) { ModSecString &out) {
unsigned char *input; unsigned char *input;
int changed = 0; int changed = 0;

View File

@ -40,8 +40,8 @@ class Utf8ToUnicode : public Transformation {
void execute(Transaction *t, void execute(Transaction *t,
ModSecStackString &in, ModSecString &in,
ModSecStackString &out) override; ModSecString &out) override;
static char *inplace(unsigned char *input, uint64_t input_len, static char *inplace(unsigned char *input, uint64_t input_len,
int *changed); int *changed);

View File

@ -21,6 +21,7 @@
#include "modsecurity/actions/action.h" #include "modsecurity/actions/action.h"
#include "modsecurity/transaction.h" #include "modsecurity/transaction.h"
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {

View File

@ -439,8 +439,8 @@ std::string Lua::applyTransformations(lua_State *L, Transaction *t,
"t:" + std::string(name)); "t:" + std::string(name));
// FIXME: transformation is not yet returning null. // FIXME: transformation is not yet returning null.
if (tfn) { if (tfn) {
ModSecStackString in; ModSecString in;
ModSecStackString out; ModSecString out;
in.assign(newVar.c_str(), newVar.size()); in.assign(newVar.c_str(), newVar.size());
tfn->execute(t, in, out); tfn->execute(t, in, out);
newVar.assign(out.c_str(), out.size()); newVar.assign(out.c_str(), out.size());
@ -465,8 +465,8 @@ std::string Lua::applyTransformations(lua_State *L, Transaction *t,
// FIXME: transformation is not yet returning null. // FIXME: transformation is not yet returning null.
if (tfn) { if (tfn) {
ModSecStackString in; ModSecString in;
ModSecStackString out; ModSecString out;
in.assign(newVar.c_str(), newVar.size()); in.assign(newVar.c_str(), newVar.size());
tfn->execute(t, in, out); tfn->execute(t, in, out);
newVar.assign(out.c_str(), out.size()); newVar.assign(out.c_str(), out.size());

View File

@ -305,8 +305,8 @@ int ModSecurity::processContentOffset(const char *content, size_t len,
while (!trans.empty()) { while (!trans.empty()) {
modsecurity::actions::transformations::Transformation *t; modsecurity::actions::transformations::Transformation *t;
ModSecStackString in; ModSecString in;
ModSecStackString out; ModSecString out;
yajl_gen_map_open(g); yajl_gen_map_open(g);
yajl_gen_string(g, yajl_gen_string(g,

View File

@ -20,6 +20,8 @@
#include "src/operators/operator.h" #include "src/operators/operator.h"
#include "others/libinjection/src/libinjection.h" #include "others/libinjection/src/libinjection.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace operators { namespace operators {

View File

@ -19,6 +19,7 @@
#include "src/operators/operator.h" #include "src/operators/operator.h"
#include "others/libinjection/src/libinjection.h" #include "others/libinjection/src/libinjection.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {

View File

@ -28,6 +28,8 @@
#include "src/operators/operator.h" #include "src/operators/operator.h"
#include "src/utils/acmp.h" #include "src/utils/acmp.h"
#include "src/utils/string.h" #include "src/utils/string.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace operators { namespace operators {

View File

@ -25,6 +25,8 @@
#include "modsecurity/rules_set.h" #include "modsecurity/rules_set.h"
#include "src/operators/operator.h" #include "src/operators/operator.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace operators { namespace operators {

View File

@ -22,6 +22,8 @@
#include "src/operators/operator.h" #include "src/operators/operator.h"
#include "modsecurity/rule.h" #include "modsecurity/rule.h"
#include "modsecurity/rule_message.h" #include "modsecurity/rule_message.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace operators { namespace operators {

View File

@ -21,6 +21,8 @@
#include <vector> #include <vector>
#include "src/operators/operator.h" #include "src/operators/operator.h"
#include "modsecurity/rule_with_actions.h"
#if PCRE_HAVE_JIT #if PCRE_HAVE_JIT
#define pcre_study_opt PCRE_STUDY_JIT_COMPILE #define pcre_study_opt PCRE_STUDY_JIT_COMPILE

View File

@ -19,6 +19,8 @@
#include <list> #include <list>
#include "src/operators/operator.h" #include "src/operators/operator.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace operators { namespace operators {

View File

@ -20,6 +20,8 @@
#include <list> #include <list>
#include "src/operators/operator.h" #include "src/operators/operator.h"
#include "modsecurity/rule_with_actions.h"
namespace modsecurity { namespace modsecurity {
namespace operators { namespace operators {

Some files were not shown because too many files have changed in this diff Show More