Merge pull request #3218 from eduar-hte/remove-dynamic-casts

Remove unnecessary dynamic casts
This commit is contained in:
Ervin Hegedus 2024-08-09 17:24:55 +02:00 committed by GitHub
commit 6f0e566f98
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 845 additions and 851 deletions

View File

@ -50,7 +50,7 @@ class RuleWithOperator : public RuleWithActions {
bool evaluate(Transaction *transaction, bool evaluate(Transaction *transaction,
std::shared_ptr<RuleMessage> rm) override; std::shared_ptr<RuleMessage> rm) override;
void getVariablesExceptions(Transaction *t, void getVariablesExceptions(Transaction &t,
variables::Variables *exclusion, variables::Variables *addition); variables::Variables *exclusion, variables::Variables *addition);
inline void getFinalVars(variables::Variables *vars, inline void getFinalVars(variables::Variables *vars,
variables::Variables *eclusion, Transaction *trans); variables::Variables *eclusion, Transaction *trans);

View File

@ -51,18 +51,12 @@ bool SetVar::evaluate(RuleWithActions *rule, Transaction *t) {
std::string m_variableNameExpanded; std::string m_variableNameExpanded;
auto *v = m_variable.get(); auto *v = m_variable.get();
variables::Tx_DynamicElement *tx = dynamic_cast< auto tx = dynamic_cast<variables::Tx_DynamicElement *> (v);
variables::Tx_DynamicElement *> (v); auto session = dynamic_cast<variables::Session_DynamicElement *> (v);
variables::Session_DynamicElement *session = dynamic_cast< auto ip = dynamic_cast<variables::Ip_DynamicElement *> (v);
variables::Session_DynamicElement *> (v); auto resource = dynamic_cast<variables::Resource_DynamicElement *> (v);
variables::Ip_DynamicElement *ip = dynamic_cast< auto global = dynamic_cast<variables::Global_DynamicElement *> (v);
variables::Ip_DynamicElement *> (v); auto user = dynamic_cast<variables::User_DynamicElement *> (v);
variables::Resource_DynamicElement *resource = dynamic_cast<
variables::Resource_DynamicElement *> (v);
variables::Global_DynamicElement *global = dynamic_cast<
variables::Global_DynamicElement *> (v);
variables::User_DynamicElement *user = dynamic_cast<
variables::User_DynamicElement *> (v);
if (tx) { if (tx) {
m_variableNameExpanded = tx->m_string->evaluate(t, rule); m_variableNameExpanded = tx->m_string->evaluate(t, rule);
} else if (session) { } else if (session) {

File diff suppressed because it is too large Load Diff

View File

@ -1086,8 +1086,9 @@ expression:
std::vector<actions::Action *> *a = new std::vector<actions::Action *>(); std::vector<actions::Action *> *a = new std::vector<actions::Action *>();
std::vector<actions::transformations::Transformation *> *t = new std::vector<actions::transformations::Transformation *>(); std::vector<actions::transformations::Transformation *> *t = new std::vector<actions::transformations::Transformation *>();
for (auto &i : *$4.get()) { for (auto &i : *$4.get()) {
if (dynamic_cast<actions::transformations::Transformation *>(i.get())) { if (auto pt = dynamic_cast<actions::transformations::Transformation *>(i.get())) {
t->push_back(dynamic_cast<actions::transformations::Transformation *>(i.release())); t->push_back(pt);
i.release();
} else { } else {
a->push_back(i.release()); a->push_back(i.release());
} }
@ -1135,8 +1136,9 @@ expression:
std::vector<actions::Action *> *a = new std::vector<actions::Action *>(); std::vector<actions::Action *> *a = new std::vector<actions::Action *>();
std::vector<actions::transformations::Transformation *> *t = new std::vector<actions::transformations::Transformation *>(); std::vector<actions::transformations::Transformation *> *t = new std::vector<actions::transformations::Transformation *>();
for (auto &i : *$2.get()) { for (auto &i : *$2.get()) {
if (dynamic_cast<actions::transformations::Transformation *>(i.get())) { if (auto pt = dynamic_cast<actions::transformations::Transformation *>(i.get())) {
t->push_back(dynamic_cast<actions::transformations::Transformation *>(i.release())); t->push_back(pt);
i.release();
} else { } else {
a->push_back(i.release()); a->push_back(i.release());
} }
@ -1155,8 +1157,9 @@ expression:
std::vector<actions::Action *> *a = new std::vector<actions::Action *>(); std::vector<actions::Action *> *a = new std::vector<actions::Action *>();
std::vector<actions::transformations::Transformation *> *t = new std::vector<actions::transformations::Transformation *>(); std::vector<actions::transformations::Transformation *> *t = new std::vector<actions::transformations::Transformation *>();
for (auto &i : *$2.get()) { for (auto &i : *$2.get()) {
if (dynamic_cast<actions::transformations::Transformation *>(i.get())) { if (auto pt = dynamic_cast<actions::transformations::Transformation *>(i.get())) {
t->push_back(dynamic_cast<actions::transformations::Transformation *>(i.release())); t->push_back(pt);
i.release();
} else { } else {
a->push_back(i.release()); a->push_back(i.release());
} }

View File

@ -99,17 +99,16 @@ RuleWithActions::RuleWithActions(
} else if (dynamic_cast<actions::MultiMatch *>(a)) { } else if (dynamic_cast<actions::MultiMatch *>(a)) {
m_containsMultiMatchAction = true; m_containsMultiMatchAction = true;
delete a; delete a;
} else if (dynamic_cast<actions::Severity *>(a)) { } else if (auto sa = dynamic_cast<actions::Severity *>(a)) {
m_severity = dynamic_cast<actions::Severity *>(a); m_severity = sa;
} else if (dynamic_cast<actions::LogData *>(a)) { } else if (auto lda = dynamic_cast<actions::LogData *>(a)) {
m_logData = dynamic_cast<actions::LogData*>(a); m_logData = lda;
} else if (dynamic_cast<actions::Msg *>(a)) { } else if (auto ma = dynamic_cast<actions::Msg *>(a)) {
m_msg = dynamic_cast<actions::Msg*>(a); m_msg = ma;
} else if (dynamic_cast<actions::SetVar *>(a)) { } else if (auto sva = dynamic_cast<actions::SetVar *>(a)) {
m_actionsSetVar.push_back( m_actionsSetVar.push_back(sva);
dynamic_cast<actions::SetVar *>(a)); } else if (auto ta = dynamic_cast<actions::Tag *>(a)) {
} else if (dynamic_cast<actions::Tag *>(a)) { m_actionsTag.push_back(ta);
m_actionsTag.push_back(dynamic_cast<actions::Tag *>(a));
} else if (dynamic_cast<actions::Block *>(a)) { } else if (dynamic_cast<actions::Block *>(a)) {
m_actionsRuntimePos.push_back(a); m_actionsRuntimePos.push_back(a);
m_containsStaticBlockAction = true; m_containsStaticBlockAction = true;
@ -213,7 +212,7 @@ void RuleWithActions::executeActionsIndependentOfChainedRuleResult(Transaction *
if (m_ruleId != b.first) { if (m_ruleId != b.first) {
continue; continue;
} }
actions::Action *a = dynamic_cast<actions::Action*>(b.second.get()); actions::Action *a = b.second.get();
if (a->isDisruptive() == true && *a->m_name.get() == "block") { if (a->isDisruptive() == true && *a->m_name.get() == "block") {
ms_dbg_a(trans, 9, "Rule contains a `block' action"); ms_dbg_a(trans, 9, "Rule contains a `block' action");
*containsBlock = true; *containsBlock = true;
@ -266,7 +265,7 @@ void RuleWithActions::executeActionsAfterFullMatch(Transaction *trans,
if (m_ruleId != b.first) { if (m_ruleId != b.first) {
continue; continue;
} }
actions::Action *a = dynamic_cast<actions::Action*>(b.second.get()); actions::Action *a = b.second.get();
executeAction(trans, containsBlock, ruleMessage, a, false); executeAction(trans, containsBlock, ruleMessage, a, false);
disruptiveAlreadyExecuted = true; disruptiveAlreadyExecuted = true;
} }
@ -394,8 +393,7 @@ void RuleWithActions::executeTransformations(
for (Transformation *a : m_transformations) { for (Transformation *a : m_transformations) {
if (none == 0) { if (none == 0) {
Transformation *t = dynamic_cast<Transformation *>(a); executeTransformation(a, &value, trans, &ret, &path,
executeTransformation(t, &value, trans, &ret, &path,
&transformations); &transformations);
} }
if (a->m_isNone) { if (a->m_isNone) {
@ -423,8 +421,7 @@ void RuleWithActions::executeTransformations(
} }
Transformation *a = dynamic_cast<Transformation*>(b.second.get()); Transformation *a = dynamic_cast<Transformation*>(b.second.get());
if (none == 0) { if (none == 0) {
Transformation *t = dynamic_cast<Transformation *>(a); executeTransformation(a, &value, trans, &ret, &path,
executeTransformation(t, &value, trans, &ret, &path,
&transformations); &transformations);
} }
if (a->m_isNone) { if (a->m_isNone) {
@ -479,7 +476,7 @@ std::vector<actions::Action *> RuleWithActions::getActionsByName(const std::stri
if (m_ruleId != b.first) { if (m_ruleId != b.first) {
continue; continue;
} }
actions::Action *z = dynamic_cast<actions::Action*>(b.second.get()); actions::Action *z = b.second.get();
if (*z->m_name.get() == name) { if (*z->m_name.get() == name) {
ret.push_back(z); ret.push_back(z);
} }
@ -489,7 +486,7 @@ std::vector<actions::Action *> RuleWithActions::getActionsByName(const std::stri
if (m_ruleId != b.first) { if (m_ruleId != b.first) {
continue; continue;
} }
actions::Action *z = dynamic_cast<actions::Action*>(b.second.get()); actions::Action *z = b.second.get();
if (*z->m_name.get() == name) { if (*z->m_name.get() == name) {
ret.push_back(z); ret.push_back(z);
} }

View File

@ -131,44 +131,43 @@ bool RuleWithOperator::executeOperatorAt(Transaction *trans, const std::string &
} }
void RuleWithOperator::getVariablesExceptions(Transaction *t, template<typename MapType, typename Operation>
void getVariablesExceptionsHelper(
variables::Variables *exclusion, variables::Variables *addition,
const MapType &map, Operation op) {
for (const auto &[x, v] : map) {
if (op(x)) {
auto b = v.get();
if (auto vme = dynamic_cast<variables::VariableModificatorExclusion*>(b)) {
exclusion->push_back(vme->m_base.get());
} else {
addition->push_back(b);
}
}
}
}
void RuleWithOperator::getVariablesExceptions(Transaction &t,
variables::Variables *exclusion, variables::Variables *addition) { variables::Variables *exclusion, variables::Variables *addition) {
for (const auto &[tag, v] : t->m_rules->m_exceptions.m_variable_update_target_by_tag) { // cppcheck-suppress ctunullpointer getVariablesExceptionsHelper(exclusion, addition,
if (containsTag(*tag.get(), t)) { t.m_rules->m_exceptions.m_variable_update_target_by_tag,
if (Variable *b{v.get()};dynamic_cast<variables::VariableModificatorExclusion*>(b)) { [this, &t](const auto &tag) { return containsTag(*tag.get(), &t); });
exclusion->push_back(dynamic_cast<variables::VariableModificatorExclusion*>(b)->m_base.get());
} else {
addition->push_back(b);
}
}
}
for (const auto &[msg, v] : t->m_rules->m_exceptions.m_variable_update_target_by_msg) { getVariablesExceptionsHelper(exclusion, addition,
if (containsMsg(*msg.get(), t)) { t.m_rules->m_exceptions.m_variable_update_target_by_msg,
if (Variable *b{v.get()}; dynamic_cast<variables::VariableModificatorExclusion *>(b)) { [this, &t](const auto &msg) { return containsMsg(*msg.get(), &t); });
exclusion->push_back(dynamic_cast<variables::VariableModificatorExclusion *>(b)->m_base.get());
} else {
addition->push_back(b);
}
}
}
for (const auto &[id, v] : t->m_rules->m_exceptions.m_variable_update_target_by_id) { // cppcheck-suppress unassignedVariable getVariablesExceptionsHelper(exclusion, addition,
if (m_ruleId == id) { t.m_rules->m_exceptions.m_variable_update_target_by_id,
if (Variable *b{v.get()};dynamic_cast<variables::VariableModificatorExclusion *>(b)) { [this](const auto &id) { return m_ruleId == id; });
exclusion->push_back(dynamic_cast<variables::VariableModificatorExclusion *>(b)->m_base.get());
} else {
addition->push_back(b);
}
}
}
} }
inline void RuleWithOperator::getFinalVars(variables::Variables *vars, inline void RuleWithOperator::getFinalVars(variables::Variables *vars,
variables::Variables *exclusion, Transaction *trans) { variables::Variables *exclusion, Transaction *trans) {
variables::Variables addition; variables::Variables addition;
getVariablesExceptions(trans, exclusion, &addition); getVariablesExceptions(*trans, exclusion, &addition); // cppcheck-suppress ctunullpointer
for (int i = 0; i < m_variables->size(); i++) { for (int i = 0; i < m_variables->size(); i++) {
Variable *variable = m_variables->at(i); Variable *variable = m_variables->at(i);

View File

@ -87,7 +87,7 @@ int main(int argc, char **argv) {
continue; continue;
} }
if (dynamic_cast<modsecurity::RuleUnconditional *>(z.get()) != nullptr) { if (dynamic_cast<modsecurity::RuleUnconditional *>(z.get())) {
std::string op = "Unconditional"; std::string op = "Unconditional";
if (operators.count(op) > 0) { if (operators.count(op) > 0) {
operators[op] = 1 + operators[op]; operators[op] = 1 + operators[op];
@ -96,9 +96,7 @@ int main(int argc, char **argv) {
} }
} }
if (dynamic_cast<modsecurity::RuleWithOperator *>(z.get()) != nullptr) { if (auto rwo = dynamic_cast<modsecurity::RuleWithOperator *>(z.get())) {
auto *rwo = dynamic_cast<modsecurity::RuleWithOperator *>(z.get());
std::string op = rwo->getOperatorName(); std::string op = rwo->getOperatorName();
if (operators.count(op) > 0) { if (operators.count(op) > 0) {
operators[op] = 1 + operators[op]; operators[op] = 1 + operators[op];