Avoid creating a new std::string on the heap to create VariableValue

- Introduced helper method addVariableOrigin to reduce code duplication.
This commit is contained in:
eduar-hte 2024-05-18 16:55:41 +00:00 committed by Eduardo Arias
parent bb07de9ad7
commit f8dd09f7c9
3 changed files with 28 additions and 47 deletions

View File

@ -35,7 +35,7 @@ namespace variables {
class Rule_DictElement : public VariableDictElement { \ class Rule_DictElement : public VariableDictElement { \
public: public:
explicit Rule_DictElement(const std::string &dictElement) explicit Rule_DictElement(const std::string &dictElement)
: VariableDictElement(std::string("RULE"), dictElement) { } : VariableDictElement(m_rule, dictElement) { }
static void id(Transaction *t, static void id(Transaction *t,
RuleWithActions *rule, RuleWithActions *rule,
@ -49,13 +49,8 @@ class Rule_DictElement : public VariableDictElement { \
if (!r || r->m_ruleId == 0) { if (!r || r->m_ruleId == 0) {
return; return;
} }
std::string *a = new std::string(std::to_string(r->m_ruleId));
VariableValue *var = new VariableValue(&m_rule, &m_rule_id, addVariableOrigin(m_rule_id, std::to_string(r->m_ruleId), l);
a
);
delete a;
var->addOrigin();
l->push_back(var);
} }
@ -72,13 +67,7 @@ class Rule_DictElement : public VariableDictElement { \
return; return;
} }
std::string *a = new std::string(r->m_rev); addVariableOrigin(m_rule_rev, r->m_rev, l);
VariableValue *var = new VariableValue(&m_rule, &m_rule_rev,
a
);
delete a;
var->addOrigin();
l->push_back(var);
} }
@ -92,13 +81,7 @@ class Rule_DictElement : public VariableDictElement { \
} }
if (r && r->hasSeverity()) { if (r && r->hasSeverity()) {
std::string *a = new std::string(std::to_string(r->severity())); addVariableOrigin(m_rule_severity, std::to_string(r->severity()), l);
VariableValue *var = new VariableValue(&m_rule, &m_rule_severity,
a
);
delete a;
var->addOrigin();
l->push_back(var);
} }
} }
@ -113,13 +96,7 @@ class Rule_DictElement : public VariableDictElement { \
} }
if (r && r->hasLogData()) { if (r && r->hasLogData()) {
std::string *a = new std::string(r->logData(t)); addVariableOrigin(m_rule_logdata, r->logData(t), l);
VariableValue *var = new VariableValue(&m_rule, &m_rule_logdata,
a
);
delete a;
var->addOrigin();
l->push_back(var);
} }
} }
@ -133,36 +110,30 @@ class Rule_DictElement : public VariableDictElement { \
} }
if (r && r->hasMsg()) { if (r && r->hasMsg()) {
std::string *a = new std::string(r->msg(t)); addVariableOrigin(m_rule_msg, r->msg(t), l);
VariableValue *var = new VariableValue(&m_rule, &m_rule_msg,
a
);
delete a;
var->addOrigin();
l->push_back(var);
} }
} }
void evaluate(Transaction *t, void evaluate(Transaction *t,
RuleWithActions *rule, RuleWithActions *rule,
std::vector<const VariableValue *> *l) override { std::vector<const VariableValue *> *l) override {
if (m_dictElement == "id") { if (m_dictElement == m_rule_id) {
id(t, rule, l); id(t, rule, l);
return; return;
} }
if (rule && m_dictElement == "rev") { if (rule && m_dictElement == m_rule_rev) {
rev(t, rule, l); rev(t, rule, l);
return; return;
} }
if (rule && m_dictElement == "severity") { if (rule && m_dictElement == m_rule_severity) {
severity(t, rule, l); severity(t, rule, l);
return; return;
} }
if (m_dictElement == "logdata") { if (m_dictElement == m_rule_logdata) {
logData(t, rule, l); logData(t, rule, l);
return; return;
} }
if (m_dictElement == "msg") { if (m_dictElement == m_rule_msg) {
msg(t, rule, l); msg(t, rule, l);
return; return;
} }
@ -174,6 +145,18 @@ class Rule_DictElement : public VariableDictElement { \
static const std::string m_rule_severity; static const std::string m_rule_severity;
static const std::string m_rule_logdata; static const std::string m_rule_logdata;
static const std::string m_rule_msg; static const std::string m_rule_msg;
private:
static inline void addVariableOrigin(const std::string &key,
const std::string &value,
std::vector<const VariableValue *> *l) {
auto var = new VariableValue(&m_rule, &key,
&value
);
var->addOrigin();
l->push_back(var);
}
}; };

View File

@ -707,9 +707,8 @@ class VariableModificatorCount : public Variable {
} }
reslIn.clear(); reslIn.clear();
std::string *res = new std::string(std::to_string(count)); auto res = std::to_string(count);
val = new VariableValue(m_fullName.get(), res); val = new VariableValue(m_fullName.get(), &res);
delete res;
l->push_back(val); l->push_back(val);
return; return;

View File

@ -124,13 +124,12 @@ void XML::evaluate(Transaction *t,
content = reinterpret_cast<char *>( content = reinterpret_cast<char *>(
xmlNodeGetContent(nodes->nodeTab[i])); xmlNodeGetContent(nodes->nodeTab[i]));
if (content != NULL) { if (content != NULL) {
std::string *a = new std::string(content); auto a = std::string(content);
VariableValue *var = new VariableValue(m_fullName.get(), VariableValue *var = new VariableValue(m_fullName.get(),
a); &a);
if (!m_keyExclusion.toOmit(*m_fullName)) { if (!m_keyExclusion.toOmit(*m_fullName)) {
l->push_back(var); l->push_back(var);
} }
delete a;
xmlFree(content); xmlFree(content);
} }
} }