From e86b36105cf48683969ecef5d7794f40309ae4e3 Mon Sep 17 00:00:00 2001 From: hasherezade Date: Thu, 8 Feb 2024 08:36:18 -0800 Subject: [PATCH] [FEATURE] In pattern_tree: added wrapper for text patterns --- utils/pattern_tree.h | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/utils/pattern_tree.h b/utils/pattern_tree.h index 8fcab5c6c..b8d5e396d 100644 --- a/utils/pattern_tree.h +++ b/utils/pattern_tree.h @@ -116,8 +116,9 @@ namespace pattern_tree { static bool addPattern(Node* rootN, const char* _name, const BYTE* pattern, size_t pattern_size) { - if (!rootN || !pattern) return false; - + if (!rootN || !pattern || !pattern_size) { + return false; + } Node* next = rootN; for (size_t i = 0; i < pattern_size; i++) { next = next->addNext(pattern[i]); @@ -127,6 +128,11 @@ namespace pattern_tree { return true; } + static bool addTextPattern(Node* rootN, const char* pattern1) + { + return Node::addPattern(rootN, pattern1, (const BYTE*)pattern1, strlen(pattern1)); + } + //--- Node()