diff --git a/src/operators/pm.cc b/src/operators/pm.cc index 3b52e053..44d9a765 100644 --- a/src/operators/pm.cc +++ b/src/operators/pm.cc @@ -40,6 +40,7 @@ Pm::~Pm() { free(m_p); m_p = NULL; + pthread_mutex_destroy(&m_lock); } @@ -86,8 +87,9 @@ bool Pm::evaluate(Transaction *transaction, Rule *rule, pt.parser = m_p; pt.ptr = NULL; const char *match = NULL; - + pthread_mutex_lock(&m_lock); rc = acmp_process_quick(&pt, &match, input.c_str(), input.length()); + pthread_mutex_unlock(&m_lock); bool capture = rule && rule->getActionsByName("capture").size() > 0; if (rc > 0 && transaction) { @@ -114,6 +116,8 @@ bool Pm::init(const std::string &file, std::string *error) { std::istringstream *iss; const char *err = NULL; + pthread_mutex_init(&m_lock, NULL); + char *content = parse_pm_content(m_param.c_str(), m_param.length(), &err); if (content == NULL) { iss = new std::istringstream(m_param); @@ -129,7 +133,9 @@ bool Pm::init(const std::string &file, std::string *error) { acmp_add_pattern(m_p, a.c_str(), NULL, NULL, a.length()); } - acmp_prepare(m_p); + while (m_p->is_failtree_done == 0) { + acmp_prepare(m_p); + } if (content) { free(content); diff --git a/src/operators/pm.h b/src/operators/pm.h index dbdd3aa5..5d3f189d 100644 --- a/src/operators/pm.h +++ b/src/operators/pm.h @@ -55,6 +55,9 @@ class Pm : public Operator { protected: ACMP *m_p; + + private: + pthread_mutex_t m_lock; }; diff --git a/src/utils/acmp.cc b/src/utils/acmp.cc index b3b8e9aa..69618f92 100644 --- a/src/utils/acmp.cc +++ b/src/utils/acmp.cc @@ -297,6 +297,10 @@ static void acmp_add_btree_leaves(acmp_btree_node_t *node, acmp_node_t *nodes[], if ((pos - lb) > 1) { left = lb + (pos - lb) / 2; node->left = reinterpret_cast(calloc(1, sizeof(acmp_btree_node_t))); + node->left->node = NULL; + node->left->right = NULL; + node->left->left = NULL; + node->left->letter = 0; /* ENH: Check alloc succeded */ node->left->node = nodes[left]; node->left->letter = nodes[left]->letter; @@ -307,6 +311,10 @@ static void acmp_add_btree_leaves(acmp_btree_node_t *node, acmp_node_t *nodes[], if ((rb - pos) > 1) { right = pos + (rb - pos) / 2; node->right = reinterpret_cast(calloc(1, sizeof(acmp_btree_node_t))); + node->right->node = NULL; + node->right->right = NULL; + node->right->left = NULL; + node->right->letter = 0; /* ENH: Check alloc succeded */ node->right->node = nodes[right]; node->right->letter = nodes[right]->letter; @@ -355,8 +363,12 @@ static void acmp_build_binary_tree(ACMP *parser, acmp_node_t *node) { nodes[i] = nodes[j]; nodes[j] = tmp; } - if (node->btree) { free (node->btree); node->btree = NULL; } + if (node->btree != NULL) { + free(node->btree); + node->btree = NULL; + } node->btree = reinterpret_cast(calloc(1, sizeof(acmp_btree_node_t))); + /* ENH: Check alloc succeded */ pos = count / 2; node->btree->node = nodes[pos]; @@ -365,7 +377,9 @@ static void acmp_build_binary_tree(ACMP *parser, acmp_node_t *node) { for (i = 0; i < count; i++) { if (nodes[i]->child != NULL) acmp_build_binary_tree(parser, nodes[i]); } - free(nodes); + if (nodes != NULL) { + free(nodes); + } } /** @@ -532,9 +546,11 @@ int acmp_process_quick(ACMPT *acmpt, const char **match, const char *data, size_ const char *end; int offset = 0; + /* if (acmpt->parser->is_failtree_done == 0) { acmp_prepare(acmpt->parser); }; + */ parser = acmpt->parser; if (acmpt->ptr == NULL) acmpt->ptr = parser->root_node;