diff options
| author | liuwentan <[email protected]> | 2023-08-10 16:10:50 +0800 |
|---|---|---|
| committer | liuwentan <[email protected]> | 2023-08-10 16:10:50 +0800 |
| commit | 42f44802715b8d71ac02fd2363f5bafc7dba8581 (patch) | |
| tree | 45abb9380554f8227974b61f4e758242ce71ef14 /test/maat_framework_gtest.cpp | |
| parent | fb0cb5405d8236b23b5866952eda99e54d25aa5b (diff) | |
[FEATURE]expr_matcher support dual engine(hyperscan & rulescan) & benchmark
Diffstat (limited to 'test/maat_framework_gtest.cpp')
| -rw-r--r-- | test/maat_framework_gtest.cpp | 1255 |
1 files changed, 1138 insertions, 117 deletions
diff --git a/test/maat_framework_gtest.cpp b/test/maat_framework_gtest.cpp index 36f4f32..fc65c42 100644 --- a/test/maat_framework_gtest.cpp +++ b/test/maat_framework_gtest.cpp @@ -744,7 +744,8 @@ TEST_F(MaatFlagScan, FlagPlus) { state = NULL; } -class MaatStringScan : public testing::Test +//hyperscan engine +class MaatHsStringScan : public testing::Test { protected: static void SetUpTestCase() { @@ -766,6 +767,7 @@ protected: maat_options_set_stat_file(opts, "./stat.log"); maat_options_set_logger(opts, "./maat_framework_gtest.log", LOG_LEVEL_INFO); maat_options_set_accept_tags(opts, accept_tags); + //maat_options_set_expr_engine(opts, MAAT_EXPR_ENGINE_HS); //default _shared_maat_inst = maat_new(opts, table_info_path); maat_options_free(opts); @@ -785,12 +787,12 @@ protected: static struct maat *_shared_maat_inst; }; -struct maat *MaatStringScan::_shared_maat_inst; -struct log_handle *MaatStringScan::logger; +struct maat *MaatHsStringScan::_shared_maat_inst; +struct log_handle *MaatHsStringScan::logger; -TEST_F(MaatStringScan, ScanDataOnlyOneByte) { +TEST_F(MaatHsStringScan, ScanDataOnlyOneByte) { const char *table_name = "HTTP_URL"; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; int table_id = maat_get_table_id(maat_inst, table_name); ASSERT_GT(table_id, 0); @@ -809,9 +811,9 @@ TEST_F(MaatStringScan, ScanDataOnlyOneByte) { state = NULL; } -TEST_F(MaatStringScan, Full) { +TEST_F(MaatHsStringScan, Full) { const char *table_name = "HTTP_URL"; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; int table_id = maat_get_table_id(maat_inst, table_name); ASSERT_GT(table_id, 0); @@ -831,14 +833,14 @@ TEST_F(MaatStringScan, Full) { state = NULL; } -TEST_F(MaatStringScan, Regex) { +TEST_F(MaatHsStringScan, Regex) { int ret = 0; long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; const char *cookie = "Cookie: Txa123aheadBCAxd"; const char *table_name = "HTTP_URL"; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); int table_id = maat_get_table_id(maat_inst, table_name); @@ -850,7 +852,7 @@ TEST_F(MaatStringScan, Regex) { state = NULL; } -TEST_F(MaatStringScan, ExprPlus) { +TEST_F(MaatHsStringScan, ExprPlus) { long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; @@ -859,7 +861,7 @@ TEST_F(MaatStringScan, ExprPlus) { const char *scan_data1 = "http://www.cyberessays.com/search_results.php?action=search&query=abckkk,1234567"; const char *scan_data2 = "Addis Sapphire Hotel"; const char *table_name = "HTTP_SIGNATURE"; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); int table_id = maat_get_table_id(maat_inst, table_name); @@ -885,12 +887,12 @@ TEST_F(MaatStringScan, ExprPlus) { state = NULL; } -TEST_F(MaatStringScan, ExprPlusWithOffset) +TEST_F(MaatHsStringScan, ExprPlusWithOffset) { long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); const char *region_name = "Payload"; unsigned char udp_payload_not_hit[] = { /* Stun packet */ @@ -939,11 +941,11 @@ TEST_F(MaatStringScan, ExprPlusWithOffset) state = NULL; } -TEST_F(MaatStringScan, ExprPlusWithHex) { +TEST_F(MaatHsStringScan, ExprPlusWithHex) { long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); const char *scan_data1 = "text/html; charset=UTF-8"; const char *scan_data2 = "Batman\\:Take me Home.Superman/:Fine,stay with me."; @@ -975,11 +977,11 @@ TEST_F(MaatStringScan, ExprPlusWithHex) { state = NULL; } -TEST_F(MaatStringScan, ExprAndExprPlus) { +TEST_F(MaatHsStringScan, ExprAndExprPlus) { long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); const char *expr_table_name = "HTTP_URL"; const char *expr_plus_table_name = "HTTP_SIGNATURE"; @@ -1007,11 +1009,11 @@ TEST_F(MaatStringScan, ExprAndExprPlus) { state = NULL; } -TEST_F(MaatStringScan, ShouldNotHitExprPlus) { +TEST_F(MaatHsStringScan, ShouldNotHitExprPlus) { long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); const char *region_name = "tcp.payload"; unsigned char udp_payload_not_hit[] = { /* Stun packet */ @@ -1041,10 +1043,10 @@ TEST_F(MaatStringScan, ShouldNotHitExprPlus) { state = NULL; } -TEST_F(MaatStringScan, Expr8) { +TEST_F(MaatHsStringScan, Expr8) { const char *table_name = "KEYWORDS_TABLE"; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); int table_id = maat_get_table_id(maat_inst, table_name); char scan_data[128] = "string1, string2, string3, string4, string5, string6, string7, string8"; @@ -1065,11 +1067,11 @@ TEST_F(MaatStringScan, Expr8) { state = NULL; } -TEST_F(MaatStringScan, HexBinCaseSensitive) { +TEST_F(MaatHsStringScan, HexBinCaseSensitive) { const char *table_name = "KEYWORDS_TABLE"; const char *scan_data1 = "String TeST should not hit."; const char *scan_data2 = "String TEST should hit"; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; int thread_id = 0; int table_id = maat_get_table_id(maat_inst, table_name); @@ -1092,7 +1094,7 @@ TEST_F(MaatStringScan, HexBinCaseSensitive) { maat_state_free(state); } -TEST_F(MaatStringScan, BugReport20190325) { +TEST_F(MaatHsStringScan, BugReport20190325) { unsigned char scan_data[] = {/* Packet 1 */ 0x01, 0x00, 0x00, 0x00, 0x79, 0x00, 0x00, 0x00, 0x00, 0xf4, 0x01, 0x00, 0x00, 0x32, 0x00, 0x00, @@ -1112,7 +1114,7 @@ TEST_F(MaatStringScan, BugReport20190325) { 0x00, 0x31, 0x3a, 0x47, 0x32, 0x2e, 0x34, 0x30, 0x00}; const char *table_name = "TROJAN_PAYLOAD"; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; int thread_id = 0; int table_id = maat_get_table_id(maat_inst, table_name); @@ -1130,13 +1132,13 @@ TEST_F(MaatStringScan, BugReport20190325) { state = NULL; } -TEST_F(MaatStringScan, PrefixAndSuffix) { +TEST_F(MaatHsStringScan, PrefixAndSuffix) { const char *hit_twice = "[email protected]"; const char *hit_suffix = "[email protected]"; const char *hit_prefix = "[email protected]"; const char *cont_sz_table_name = "CONTENT_SIZE"; const char *mail_addr_table_name = "MAIL_ADDR"; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; int thread_id = 0; int cont_sz_table_id = maat_get_table_id(maat_inst, cont_sz_table_name); @@ -1176,10 +1178,10 @@ TEST_F(MaatStringScan, PrefixAndSuffix) { state = NULL; } -TEST_F(MaatStringScan, MaatUnescape) { +TEST_F(MaatHsStringScan, MaatUnescape) { const char *scan_data = "Batman\\:Take me Home.Superman/:Fine,stay with me."; const char *table_name = "KEYWORDS_TABLE"; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; int thread_id = 0; int table_id = maat_get_table_id(maat_inst, table_name); @@ -1197,13 +1199,13 @@ TEST_F(MaatStringScan, MaatUnescape) { state = NULL; } -TEST_F(MaatStringScan, OffsetChunk64) { +TEST_F(MaatHsStringScan, OffsetChunk64) { const char *table_name = "IMAGE_FP"; const char *file_name = "./testdata/mesa_logo.jpg"; long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); FILE *fp = fopen(file_name, "r"); @@ -1236,13 +1238,13 @@ TEST_F(MaatStringScan, OffsetChunk64) { state = NULL; } -TEST_F(MaatStringScan, OffsetChunk1460) { +TEST_F(MaatHsStringScan, OffsetChunk1460) { const char *table_name = "IMAGE_FP"; const char *file_name = "./testdata/mesa_logo.jpg"; long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); FILE *fp = fopen(file_name, "r"); @@ -1275,14 +1277,14 @@ TEST_F(MaatStringScan, OffsetChunk1460) { state = NULL; } -TEST_F(MaatStringScan, StreamScanUTF8) { +TEST_F(MaatHsStringScan, StreamScanUTF8) { const char *table_name = "TROJAN_PAYLOAD"; const char* file_name = "./testdata/jd.com.html"; long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; char scan_data[2048]; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); FILE *fp = fopen(file_name, "r"); @@ -1313,11 +1315,11 @@ TEST_F(MaatStringScan, StreamScanUTF8) { state = NULL; } -TEST_F(MaatStringScan, StreamInput) { +TEST_F(MaatHsStringScan, StreamInput) { long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); const char *scan_data = "http://www.cyberessays.com/search_results.php?action=search&query=yulingjing,abckkk,1234567"; const char *table_name = "HTTP_URL"; @@ -1342,13 +1344,13 @@ TEST_F(MaatStringScan, StreamInput) { state = NULL; } -TEST_F(MaatStringScan, dynamic_config) { +TEST_F(MaatHsStringScan, dynamic_config) { const char *table_name = "HTTP_URL"; char data[128] = "hello world, welcome to maat version4, it's funny."; long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStringScan::_shared_maat_inst; + struct maat *maat_inst = MaatHsStringScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); int table_id = maat_get_table_id(maat_inst, table_name); @@ -1379,10 +1381,680 @@ TEST_F(MaatStringScan, dynamic_config) { keywords, NULL, 1, 0, 0, 0); /* EXPR_TYPE_AND MATCH_METHOD_SUB */ EXPECT_EQ(ret, 1); + sleep(WAIT_FOR_EFFECTIVE_S * 3); + + ret = maat_scan_string(maat_inst, table_id, data, strlen(data), results, + ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 1); + EXPECT_EQ(results[0], compile_id); + maat_state_reset(state); + + /* expr table del line */ + ret = expr_table_set_line(maat_inst, table_name, MAAT_OP_DEL, item_id, group_id, + keywords, NULL, 1, 0, 0, 0); /* EXPR_TYPE_AND MATCH_METHOD_SUB */ + EXPECT_EQ(ret, 1); + + /* group2compile table del line */ + ret = group2compile_table_set_line(maat_inst, g2c_table_name, MAAT_OP_DEL, group_id, + compile_id, 0, "null", 1, 0); + EXPECT_EQ(ret, 1); + + /* compile table del line */ + ret = compile_table_set_line(maat_inst, compile_table_name, MAAT_OP_DEL, compile_id, "null", 1, 0); + EXPECT_EQ(ret, 1); + sleep(WAIT_FOR_EFFECTIVE_S); ret = maat_scan_string(maat_inst, table_id, data, strlen(data), results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HALF_HIT); + EXPECT_EQ(n_hit_result, 0); + maat_state_free(state); + state = NULL; +} + +class MaatRsStringScan : public testing::Test +{ +protected: + static void SetUpTestCase() { + const char *accept_tags = "{\"tags\":[{\"tag\":\"location\",\"value\":\"北京/朝阳/华严北里/甲22号\"}," + "{\"tag\":\"isp\",\"value\":\"移动\"},{\"tag\":\"location\",\"value\":\"Astana\"}]}"; + char redis_ip[64] = "127.0.0.1"; + int redis_port = 6379; + int redis_db = 0; + + logger = log_handle_create("./maat_framework_gtest.log", 0); + int ret = write_config_to_redis(redis_ip, redis_port, redis_db, logger); + if (ret < 0) { + log_error(logger, MODULE_FRAMEWORK_GTEST, + "[%s:%d] write config to redis failed.", __FUNCTION__, __LINE__); + } + + struct maat_options *opts = maat_options_new(); + maat_options_set_redis(opts, redis_ip, redis_port, redis_db); + maat_options_set_stat_file(opts, "./stat.log"); + maat_options_set_logger(opts, "./maat_framework_gtest.log", LOG_LEVEL_INFO); + maat_options_set_accept_tags(opts, accept_tags); + maat_options_set_expr_engine(opts, MAAT_EXPR_ENGINE_RS); + + _shared_maat_inst = maat_new(opts, table_info_path); + maat_options_free(opts); + if (NULL == _shared_maat_inst) { + log_error(logger, MODULE_FRAMEWORK_GTEST, + "[%s:%d] create maat instance in MaatStringScan failed.", + __FUNCTION__, __LINE__); + } + } + + static void TearDownTestCase() { + maat_free(_shared_maat_inst); + log_handle_destroy(logger); + } + + static struct log_handle *logger; + static struct maat *_shared_maat_inst; +}; + +struct maat *MaatRsStringScan::_shared_maat_inst; +struct log_handle *MaatRsStringScan::logger; + +TEST_F(MaatRsStringScan, ScanDataOnlyOneByte) { + const char *table_name = "HTTP_URL"; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + const char scan_data = 0x20; + + int ret = maat_scan_string(maat_inst, table_id, &scan_data, sizeof(scan_data), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_OK); + EXPECT_EQ(n_hit_result, 0); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, Full) { + const char *table_name = "HTTP_URL"; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + const char *scan_data = "http://www.cyberessays.com/search_results.php?action=search&query=username,abckkk,1234567"; + + int ret = maat_scan_string(maat_inst, table_id, scan_data, strlen(scan_data), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 1); + EXPECT_EQ(results[0], 125); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, Regex) { + int ret = 0; + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + const char *cookie = "Cookie: Txa123aheadBCAxd"; + const char *table_name = "HTTP_URL"; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + + int table_id = maat_get_table_id(maat_inst, table_name); + ret = maat_scan_string(maat_inst, table_id, cookie, strlen(cookie), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], 146); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, ExprPlus) { + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + const char *region_name1 ="HTTP URL"; + const char *region_name2 ="我的diStricT"; + const char *scan_data1 = "http://www.cyberessays.com/search_results.php?action=search&query=abckkk,1234567"; + const char *scan_data2 = "Addis Sapphire Hotel"; + const char *table_name = "HTTP_SIGNATURE"; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + + int table_id = maat_get_table_id(maat_inst, table_name); + int ret = maat_scan_string(maat_inst, table_id, scan_data1, strlen(scan_data1), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_ERR);//Should return error for district not setting. + + ret = maat_state_set_scan_district(state, table_id, region_name1, strlen(region_name1)); + ASSERT_EQ(ret, 0); + ret = maat_scan_string(maat_inst, table_id, scan_data1, strlen(scan_data1), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], 128); + maat_state_reset(state); + + ret = maat_state_set_scan_district(state, table_id, region_name2, strlen(region_name2)); + ASSERT_EQ(ret, 0); + ret = maat_scan_string(maat_inst, table_id, scan_data2, strlen(scan_data2), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], 190); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, ExprPlusWithOffset) +{ + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + const char *region_name = "Payload"; + unsigned char udp_payload_not_hit[] = { /* Stun packet */ + 0x00, 0x03, 0x00, 0x4a, 0x21, 0x12, 0xa4, 0x42, + 0x4f, 0xc2, 0xc2, 0x70, 0xb3, 0xa8, 0x4e, 0x22, + 0xf5, 0x22, 0x87, 0x4c, 0x40, 0x00, 0x00, 0x46, + 0x03, 0x02, 0xab, 0x39, 0xbb, 0x97, 0xe5, 0x01, + 0x3a, 0x46, 0x1c, 0x28, 0x5b, 0xab, 0xfa, 0x9a, + 0xab, 0x2e, 0x71, 0x39, 0x66, 0xa0, 0xd7, 0xb9, + 0xd8, 0x41, 0xa7, 0xa0, 0x84, 0xa9, 0xf3, 0x1b, + 0x03, 0x7f, 0xa8, 0x28, 0xa2, 0xd3, 0x64, 0xc2, + 0x3d, 0x20, 0xe0, 0xb1, 0x41, 0x12, 0x6c, 0x2f, + 0xc5, 0xbb, 0xc3, 0xba, 0x69, 0x73, 0x52, 0x64, + 0xf6, 0x30, 0x81, 0xf4, 0x3f, 0xc2, 0x19, 0x6a, + 0x68, 0x61, 0x93, 0x08, 0xc0, 0x0a }; + unsigned char udp_payload_hit[] = { /* Stun packet */ //rule:"1-1:03&9-10:2d&14-16:2d34&19-21:2d&24-25:2d" + 0x00, 0x03, 0x00, 0x4a, 0x21, 0x12, 0xa4, 0x42, //1-1:03 + 0x4f, 0xc2, 0x2d, 0x70, 0xb3, 0xa8, 0x4e, 0x2d, //10-10:2d + 0x34, 0x22, 0x87, 0x4c, 0x2d, 0x00, 0x00, 0x46, //15-16:2d34 + 0x2d, 0x34, 0xab, 0x39, 0xbb, 0x97, 0xe5, 0x01, //20-20:2d + 0x03, 0x46, 0x1c, 0x28, 0x5b, 0xab, 0xfa, 0x9a, //24-24:2d + 0xab, 0x2e, 0x71, 0x39, 0x66, 0xa0, 0xd7, 0xb9, + 0xd8, 0x41, 0xa7, 0xa0, 0x84, 0xa9, 0xf3, 0x1b, + 0x03, 0x7f, 0xa8, 0x28, 0xa2, 0xd3, 0x64, 0xc2, + 0x3d, 0x20, 0xe0, 0xb1, 0x41, 0x12, 0x6c, 0x2f, + 0xc5, 0xbb, 0xc3, 0xba, 0x69, 0x73, 0x52, 0x64, + 0xf6, 0x30, 0x81, 0xf4, 0x3f, 0xc2, 0x19, 0x6a, + 0x68, 0x61, 0x93, 0x08, 0xc0, 0x0a }; + + int table_id = maat_get_table_id(maat_inst, "APP_PAYLOAD"); + ASSERT_GT(table_id, 0); + + int ret = maat_state_set_scan_district(state, table_id, region_name, strlen(region_name)); + EXPECT_EQ(ret, 0); + + ret = maat_scan_string(maat_inst, table_id, (char*)udp_payload_not_hit, sizeof(udp_payload_not_hit), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_OK); + + ret = maat_scan_string(maat_inst, table_id, (char*)udp_payload_hit, sizeof(udp_payload_hit), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], 148); + + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, ExprPlusWithHex) { + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + const char *scan_data1 = "text/html; charset=UTF-8"; + const char *scan_data2 = "Batman\\:Take me Home.Superman/:Fine,stay with me."; + const char *region_name1 = "Content-Type"; + const char *region_name2 = "User-Agent"; + + int table_id = maat_get_table_id(maat_inst, "HTTP_SIGNATURE"); + ASSERT_GT(table_id, 0); + + int ret = maat_state_set_scan_district(state, table_id, region_name1, strlen(region_name1)); + ASSERT_EQ(ret, 0); + ret = maat_scan_string(maat_inst, table_id, scan_data1, strlen(scan_data1), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], 156); + + ret = maat_state_set_scan_district(state, table_id, region_name2, strlen(region_name2)); + ASSERT_EQ(ret, 0); + ret = maat_scan_string(maat_inst, table_id, scan_data1, strlen(scan_data1), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_OK); //maat-v3 consider as half hit, it's unreasonable + + table_id = maat_get_table_id(maat_inst, "KEYWORDS_TABLE"); + ret = maat_scan_string(maat_inst, table_id, scan_data2, strlen(scan_data2), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], 132); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, ExprAndExprPlus) { + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + const char *expr_table_name = "HTTP_URL"; + const char *expr_plus_table_name = "HTTP_SIGNATURE"; + const char *region_name = "I love China"; + const char *scan_data = "today is Monday and yesterday is Tuesday"; + + int expr_table_id = maat_get_table_id(maat_inst, expr_table_name); + int expr_plus_table_id = maat_get_table_id(maat_inst, expr_plus_table_name); + + int ret = maat_scan_string(maat_inst, expr_plus_table_id, scan_data, strlen(scan_data), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_ERR); + + ret = maat_state_set_scan_district(state, expr_plus_table_id, region_name, strlen(region_name)); + ASSERT_EQ(ret, 0); + ret = maat_scan_string(maat_inst, expr_plus_table_id, scan_data, strlen(scan_data), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HALF_HIT); + + ret = maat_scan_string(maat_inst, expr_table_id, scan_data, strlen(scan_data), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], 195); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, ShouldNotHitExprPlus) { + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + const char *region_name = "tcp.payload"; + unsigned char udp_payload_not_hit[] = { /* Stun packet */ + 0x00, 0x03, 0x00, 0x4a, 0x21, 0x12, 0xa4, 0x42, + 0x4f, 0xc2, 0xc2, 0x70, 0xb3, 0xa8, 0x4e, 0x22, + 0xf5, 0x22, 0x87, 0x4c, 0x40, 0x00, 0x00, 0x46, + 0x03, 0x02, 0xab, 0x39, 0xbb, 0x97, 0xe5, 0x01, + 0x3a, 0x46, 0x1c, 0x28, 0x5b, 0xab, 0xfa, 0x9a, + 0xab, 0x2e, 0x71, 0x39, 0x66, 0xa0, 0xd7, 0xb9, + 0xd8, 0x41, 0xa7, 0xa0, 0x84, 0xa9, 0xf3, 0x1b, + 0x03, 0x7f, 0xa8, 0x28, 0xa2, 0xd3, 0x64, 0xc2, + 0x3d, 0x20, 0xe0, 0xb1, 0x41, 0x12, 0x6c, 0x2f, + 0xc5, 0xbb, 0xc3, 0xba, 0x69, 0x73, 0x52, 0x64, + 0xf6, 0x30, 0x81, 0xf4, 0x3f, 0xc2, 0x19, 0x6a, + 0x68, 0x61, 0x93, 0x08, 0xc0, 0x0a, 0xab, 0x00 }; + + int table_id = maat_get_table_id(maat_inst, "APP_PAYLOAD"); + ASSERT_GT(table_id, 0); + + int ret = maat_state_set_scan_district(state, table_id, region_name, strlen(region_name)); + ASSERT_EQ(ret, 0); + + ret = maat_scan_string(maat_inst, table_id, (char *)udp_payload_not_hit, sizeof(udp_payload_not_hit), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_OK); //maat-v3 consider as half hit, it's unreasonable + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, Expr8) { + const char *table_name = "KEYWORDS_TABLE"; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + int table_id = maat_get_table_id(maat_inst, table_name); + char scan_data[128] = "string1, string2, string3, string4, string5, string6, string7, string8"; + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + + int ret = maat_scan_string(maat_inst, table_id, scan_data, strlen(scan_data), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 1); + EXPECT_EQ(results[0], 182); + + struct maat_hit_path hit_path[HIT_PATH_SIZE] = {0}; + int n_read = 0; + n_read = maat_state_get_hit_paths(state, hit_path, HIT_PATH_SIZE); + EXPECT_NE(n_read, 0); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, HexBinCaseSensitive) { + const char *table_name = "KEYWORDS_TABLE"; + const char *scan_data1 = "String TeST should not hit."; + const char *scan_data2 = "String TEST should hit"; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + int thread_id = 0; + + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + int ret = maat_scan_string(maat_inst, table_id, scan_data1, strlen(scan_data1), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_OK); + maat_state_reset(state); + + ret = maat_scan_string(maat_inst, table_id, scan_data2, strlen(scan_data2), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 2); + EXPECT_EQ(results[0], 206); + EXPECT_EQ(results[1], 191); + maat_state_free(state); +} + +TEST_F(MaatRsStringScan, BugReport20190325) { + unsigned char scan_data[] = {/* Packet 1 */ + 0x01, 0x00, 0x00, 0x00, 0x79, 0x00, 0x00, 0x00, + 0x00, 0xf4, 0x01, 0x00, 0x00, 0x32, 0x00, 0x00, + 0x00, 0xe8, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x01, 0x01, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0xff, 0xff, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x2d, 0x3d, 0x3d, 0x20, 0x48, 0x3d, 0x48, 0x20, + 0x3d, 0x3d, 0x2d, 0x3a, 0x00, 0x02, 0x00, 0x00, + 0x00, 0x07, 0x0e, 0x00, 0x00, 0xe8, 0x03, 0x00, + 0x00, 0x4c, 0x69, 0x6e, 0x75, 0x78, 0x20, 0x33, + 0x2e, 0x31, 0x39, 0x2e, 0x30, 0x2d, 0x31, 0x35, + 0x2d, 0x67, 0x65, 0x6e, 0x65, 0x72, 0x69, 0x63, + 0x00, 0x31, 0x3a, 0x47, 0x32, 0x2e, 0x34, 0x30, + 0x00}; + const char *table_name = "TROJAN_PAYLOAD"; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + int thread_id = 0; + + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + int ret = maat_scan_string(maat_inst, table_id, (char *)scan_data, sizeof(scan_data), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 1); + EXPECT_EQ(results[0], 150); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, PrefixAndSuffix) { + const char *hit_twice = "[email protected]"; + const char *hit_suffix = "[email protected]"; + const char *hit_prefix = "[email protected]"; + const char *cont_sz_table_name = "CONTENT_SIZE"; + const char *mail_addr_table_name = "MAIL_ADDR"; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + int thread_id = 0; + + int cont_sz_table_id = maat_get_table_id(maat_inst, cont_sz_table_name); + ASSERT_GT(cont_sz_table_id, 0); + + int mail_addr_table_id = maat_get_table_id(maat_inst, mail_addr_table_name); + ASSERT_GT(mail_addr_table_id, 0); + + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + int ret = maat_scan_integer(maat_inst, cont_sz_table_id, 2015, results, + ARRAY_SIZE, &n_hit_result, state); + + ret = maat_scan_string(maat_inst, mail_addr_table_id, hit_twice, strlen(hit_twice), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 2); + EXPECT_EQ(results[0], 151); + EXPECT_EQ(results[1], 152); + maat_state_reset(state); + + ret = maat_scan_string(maat_inst, mail_addr_table_id, hit_suffix, strlen(hit_suffix), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 1); + EXPECT_EQ(results[0], 151); + + ret = maat_scan_integer(maat_inst, cont_sz_table_id, 2015, results, + ARRAY_SIZE, &n_hit_result, state); + ret = maat_scan_string(maat_inst, mail_addr_table_id, hit_prefix, strlen(hit_prefix), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 1); + EXPECT_EQ(results[0], 152); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, MaatUnescape) { + const char *scan_data = "Batman\\:Take me Home.Superman/:Fine,stay with me."; + const char *table_name = "KEYWORDS_TABLE"; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + int thread_id = 0; + + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + int ret = maat_scan_string(maat_inst, table_id, scan_data, strlen(scan_data), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 1); + EXPECT_EQ(results[0], 132); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, OffsetChunk64) { + const char *table_name = "IMAGE_FP"; + const char *file_name = "./testdata/mesa_logo.jpg"; + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + + FILE *fp = fopen(file_name, "r"); + ASSERT_FALSE(fp==NULL); + + char scan_data[64]; + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + struct maat_stream *sp = maat_stream_new(maat_inst, table_id, state); + ASSERT_TRUE(sp != NULL); + + int ret = 0; + int read_size = 0; + int pass_flag = 0; + while (0 == feof(fp)) { + read_size = fread(scan_data, 1, sizeof(scan_data), fp); + ret = maat_stream_scan(sp, scan_data, read_size, + results, ARRAY_SIZE, &n_hit_result, state); + if (ret > 0) { + pass_flag = 1; + break; + } + } + EXPECT_EQ(pass_flag, 1); + EXPECT_EQ(results[0], 136); + maat_stream_free(sp); + fclose(fp); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, OffsetChunk1460) { + const char *table_name = "IMAGE_FP"; + const char *file_name = "./testdata/mesa_logo.jpg"; + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + + FILE *fp = fopen(file_name, "r"); + ASSERT_FALSE(fp==NULL); + + char scan_data[1460]; + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + struct maat_stream *sp = maat_stream_new(maat_inst, table_id, state); + ASSERT_TRUE(sp != NULL); + + int ret = 0; + int read_size = 0; + int pass_flag = 0; + while (0 == feof(fp)) { + read_size = fread(scan_data, 1, sizeof(scan_data), fp); + ret = maat_stream_scan(sp, scan_data, read_size, + results, ARRAY_SIZE, &n_hit_result, state); + if (ret > 0) { + pass_flag = 1; + break; + } + } + EXPECT_EQ(pass_flag, 1); + EXPECT_EQ(results[0], 136); + maat_stream_free(sp); + fclose(fp); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, StreamScanUTF8) { + const char *table_name = "TROJAN_PAYLOAD"; + const char* file_name = "./testdata/jd.com.html"; + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + char scan_data[1500]; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + + FILE *fp = fopen(file_name, "r"); + ASSERT_FALSE(fp == NULL); + + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + struct maat_stream *sp = maat_stream_new(maat_inst, table_id, state); + ASSERT_FALSE(sp == NULL); + + int pass_flag = 0; + while (0 == feof(fp)) { + size_t read_size = fread(scan_data, 1, sizeof(scan_data), fp); + //read_size can't exceed 1500 + int ret = maat_stream_scan(sp, scan_data, read_size, results, ARRAY_SIZE, + &n_hit_result, state); + if (ret == MAAT_SCAN_HIT) { + pass_flag = 1; + break; + } + } + + EXPECT_EQ(pass_flag, 1); + EXPECT_EQ(results[0], 157); + maat_stream_free(sp); + fclose(fp); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, StreamInput) { + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + const char *scan_data = "http://www.cyberessays.com/search_results.php?action=search&query=yulingjing,abckkk,1234567"; + const char *table_name = "HTTP_URL"; + + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + struct maat_stream *sp = maat_stream_new(maat_inst, table_id, state); + ASSERT_TRUE(sp != NULL); + + int ret = maat_stream_scan(sp, "www.cyberessays.com", strlen("www.cyberessays.com"), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_OK); + + ret = maat_stream_scan(sp, scan_data, strlen(scan_data), results, ARRAY_SIZE, + &n_hit_result, state); + maat_stream_free(sp); + + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], 125); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatRsStringScan, dynamic_config) { + const char *table_name = "HTTP_URL"; + char data[128] = "hello world, welcome to maat version4, it's funny."; + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatRsStringScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + + int table_id = maat_get_table_id(maat_inst, table_name); + int ret = maat_scan_string(maat_inst, table_id, data, strlen(data), results, + ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HALF_HIT); + EXPECT_EQ(n_hit_result, 0); + maat_state_reset(state); + + const char *compile_table_name = "COMPILE"; + const char *g2c_table_name = "GROUP2COMPILE"; + + /* compile table add line */ + long long compile_id = maat_cmd_incrby(maat_inst, "TEST_SEQ", 1); + ret = compile_table_set_line(maat_inst, compile_table_name, MAAT_OP_ADD, compile_id, "null", 1, 0); + EXPECT_EQ(ret, 1); + + /* group2compile table add line */ + long long group_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); + ret = group2compile_table_set_line(maat_inst, g2c_table_name, MAAT_OP_ADD, group_id, + compile_id, 0, "null", 1, 0); + EXPECT_EQ(ret, 1); + + /* expr table add line */ + long long item_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); + const char *keywords = "welcome to maat"; + ret = expr_table_set_line(maat_inst, table_name, MAAT_OP_ADD, item_id, group_id, + keywords, NULL, 1, 0, 0, 0); /* EXPR_TYPE_AND MATCH_METHOD_SUB */ + EXPECT_EQ(ret, 1); + + sleep(WAIT_FOR_EFFECTIVE_S * 3); + + ret = maat_scan_string(maat_inst, table_id, data, strlen(data), results, + ARRAY_SIZE, &n_hit_result, state); EXPECT_EQ(ret, MAAT_SCAN_HIT); EXPECT_EQ(n_hit_result, 1); EXPECT_EQ(results[0], compile_id); @@ -1412,7 +2084,102 @@ TEST_F(MaatStringScan, dynamic_config) { state = NULL; } -class MaatStreamScan : public testing::Test +class MaatHsStreamScan : public testing::Test +{ +protected: + static void SetUpTestCase() { + char redis_ip[64] = "127.0.0.1"; + int redis_port = 6379; + int redis_db = 0; + + struct maat_options *opts = maat_options_new(); + maat_options_set_redis(opts, redis_ip, redis_port, redis_db); + maat_options_set_logger(opts, "./maat_framework_gtest.log", LOG_LEVEL_INFO); + + _shared_maat_inst = maat_new(opts, table_info_path); + assert(_shared_maat_inst != NULL); + + maat_cmd_flushDB(_shared_maat_inst); + maat_free(_shared_maat_inst); + + maat_options_set_foreign_cont_dir(opts, "./foreign_files/"); + maat_options_set_rule_effect_interval_ms(opts, 0); + maat_options_set_gc_timeout_ms(opts, 0); // start GC immediately + maat_options_set_stat_file(opts, "./stat.log"); + _shared_maat_inst = maat_new(opts, table_info_path); + maat_options_free(opts); + } + + static void TearDownTestCase() { + maat_free(_shared_maat_inst); + } + + static struct maat *_shared_maat_inst; +}; + +struct maat *MaatHsStreamScan::_shared_maat_inst; + +TEST_F(MaatHsStreamScan, dynamic_config) { + const char *scan_data1 = "hello world cyberessays.com/search_results.php?action=search&query=yulingjing,abckkk,1234567"; + const char *table_name = "HTTP_URL"; + const char *keywords1 = "hello"; + char keyword_buf[128]; + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + int thread_id = 0; + struct maat *maat_inst = MaatHsStreamScan::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + + // STEP 1: add keywords1 and wait scan stream to hit + long long compile1_id = maat_cmd_incrby(maat_inst, "TEST_SEQ", 1); + int ret = test_add_expr_command(maat_inst, table_name, compile1_id, 0, keywords1); + EXPECT_EQ(ret, 1); + + sleep(WAIT_FOR_EFFECTIVE_S); + + int table_id = maat_get_table_id(maat_inst, table_name); + ASSERT_GT(table_id, 0); + + struct maat_stream *sp = maat_stream_new(maat_inst, table_id, state); + ASSERT_TRUE(sp != NULL); + + ret = maat_stream_scan(sp, "www.cyberessays.com", strlen("www.cyberessays.com"), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_OK); + + ret = maat_stream_scan(sp, scan_data1, strlen(scan_data1), results, ARRAY_SIZE, + &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], compile1_id); + maat_state_reset(state); + + // STEP 2: Inc config update, use same stream to scan and wait old expr_runtime invalid + random_keyword_generate(keyword_buf, sizeof(keyword_buf)); + long long compile_id = maat_cmd_incrby(maat_inst, "TEST_SEQ", 1); + ret = test_add_expr_command(maat_inst, table_name, compile_id, 0, keyword_buf); + EXPECT_EQ(ret, 1); + + // Inc config has not yet taken effect, stream scan can hit compile + ret = maat_stream_scan(sp, scan_data1, strlen(scan_data1), results, ARRAY_SIZE, + &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(results[0], compile1_id); + maat_state_reset(state); + + sleep(WAIT_FOR_EFFECTIVE_S); + + // Inc config has taken effect, stream reference old expr_runtime, should not hit compile + ret = maat_stream_scan(sp, scan_data1, strlen(scan_data1), results, ARRAY_SIZE, + &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_OK); + + maat_stream_free(sp); + maat_state_free(state); + sp = NULL; + state = NULL; +} + +class MaatRsStreamScan : public testing::Test { protected: static void SetUpTestCase() { @@ -1434,6 +2201,7 @@ protected: maat_options_set_rule_effect_interval_ms(opts, 0); maat_options_set_gc_timeout_ms(opts, 0); // start GC immediately maat_options_set_stat_file(opts, "./stat.log"); + maat_options_set_expr_engine(opts, MAAT_EXPR_ENGINE_RS); _shared_maat_inst = maat_new(opts, table_info_path); maat_options_free(opts); } @@ -1445,9 +2213,9 @@ protected: static struct maat *_shared_maat_inst; }; -struct maat *MaatStreamScan::_shared_maat_inst; +struct maat *MaatRsStreamScan::_shared_maat_inst; -TEST_F(MaatStreamScan, dynamic_config) { +TEST_F(MaatRsStreamScan, dynamic_config) { const char *scan_data1 = "hello world cyberessays.com/search_results.php?action=search&query=yulingjing,abckkk,1234567"; const char *table_name = "HTTP_URL"; const char *keywords1 = "hello"; @@ -1455,7 +2223,7 @@ TEST_F(MaatStreamScan, dynamic_config) { long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; int thread_id = 0; - struct maat *maat_inst = MaatStreamScan::_shared_maat_inst; + struct maat *maat_inst = MaatRsStreamScan::_shared_maat_inst; struct maat_state *state = maat_state_new(maat_inst, thread_id); // STEP 1: add keywords1 and wait scan stream to hit @@ -5554,7 +6322,7 @@ TEST_F(MaatCmdTest, GroupInMassCompiles) { state = NULL; } -TEST_F(MaatCmdTest, HitPath) { +TEST_F(MaatCmdTest, HitGroup) { const char *g2g_table_name = "GROUP2GROUP"; const char *g2c_table_name = "GROUP2COMPILE"; const char *compile_table_name = "COMPILE"; @@ -5580,7 +6348,7 @@ TEST_F(MaatCmdTest, HitPath) { //item1 -> group1 -> compile1 long long item1_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); ret = expr_table_set_line(maat_inst, http_sig_table_name, MAAT_OP_ADD, item1_id, group1_id, - "graph_theory", "URL", 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ + "hit group item first", "URL", 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ EXPECT_EQ(ret, 1); /* item1 -> group1 -> compile1 @@ -5607,7 +6375,7 @@ TEST_F(MaatCmdTest, HitPath) { */ long long item2_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); ret = expr_table_set_line(maat_inst, http_sig_table_name, MAAT_OP_ADD, item2_id, group2_id, - "time=2020-02-11", "Cookie", 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ + "hit group item second", "Cookie", 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ EXPECT_EQ(ret, 1); /* @@ -5626,7 +6394,7 @@ TEST_F(MaatCmdTest, HitPath) { long long item3_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); long long group3_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); ret = ip_table_set_line(maat_inst, ip_table_name, MAAT_OP_ADD, item3_id, group3_id, - IPv4, "220.181.38.148", "220.181.38.149", 0, 65535, 0); + IPv4, "220.181.38.150", "220.181.38.151", 0, 65535, 0); EXPECT_EQ(ret, 1); char temp[1024]={0}; @@ -5634,14 +6402,28 @@ TEST_F(MaatCmdTest, HitPath) { long long item4_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); long long group4_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); ret = expr_table_set_line(maat_inst, keywords_table_name, MAAT_OP_ADD, item4_id, group4_id, - str_escape(temp, sizeof(temp), "a finite or infinite"), + str_escape(temp, sizeof(temp), "hit group item forth"), NULL, 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ EXPECT_EQ(ret, 1); - sleep(WAIT_FOR_EFFECTIVE_S); + /* + item1 -> group1 -> group11 + / \ + item5 -> / \ -> compile1 + / + item2 -> group2 -> group21 _/ + */ + //item5 -> group1 which means group1 has multi items + long long item5_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); + ret = expr_table_set_line(maat_inst, keywords_table_name, MAAT_OP_ADD, item5_id, group1_id, + str_escape(temp, sizeof(temp), "hit group item fifth"), + NULL, 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ + EXPECT_EQ(ret, 1); - const char* http_url = "en.wikipedia.org/wiki/Path_(graph_theory)"; - const char* http_resp_hdr_cookie = "laptop=thinkpad X1 extrem;time=2020-02-11T15:34:00;" + sleep(WAIT_FOR_EFFECTIVE_S * 2); + + const char* http_url = "en.wikipedia.org hit group item first"; + const char* http_resp_hdr_cookie = "laptop=thinkpad X1 extrem;hit group item second" "main[XWJOKE]=hoho; Hm_lvt_bbac0322e6ee13093f98d5c4b5a10912=1578874808;"; int http_req_table_id = maat_get_table_id(maat_inst, "HTTP_REQUEST_HEADER"); @@ -5650,9 +6432,6 @@ TEST_F(MaatCmdTest, HitPath) { ret = maat_state_set_scan_district(state, http_req_table_id, "URL", strlen("URL")); EXPECT_EQ(ret, 0); - int Nth_scan = 0; - - Nth_scan++; long long results[ARRAY_SIZE] = {0}; size_t n_hit_result = 0; ret = maat_scan_string(maat_inst, http_req_table_id, http_url, strlen(http_url), @@ -5664,21 +6443,322 @@ TEST_F(MaatCmdTest, HitPath) { struct maat_hit_group hit_groups[128]; memset(hit_groups, 0, sizeof(hit_groups)); - int n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, sizeof(hit_groups)); + int n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, 128); EXPECT_EQ(n_hit_group, 2); + EXPECT_EQ(hit_groups[0].item_id, item1_id); EXPECT_EQ(hit_groups[0].group_id, group1_id); EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[1].item_id, item1_id); EXPECT_EQ(hit_groups[1].group_id, group11_id); EXPECT_EQ(hit_groups[1].vtable_id, http_req_table_id); memset(hit_groups, 0, sizeof(hit_groups)); - n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_INC, hit_groups, sizeof(hit_groups)); + n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_INC, hit_groups, 128); EXPECT_EQ(n_hit_group, 2); + EXPECT_EQ(hit_groups[0].item_id, item1_id); EXPECT_EQ(hit_groups[0].group_id, group1_id); EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[1].item_id, item1_id); EXPECT_EQ(hit_groups[1].group_id, group11_id); EXPECT_EQ(hit_groups[1].vtable_id, http_req_table_id); + int http_res_table_id = maat_get_table_id(maat_inst, "HTTP_RESPONSE_HEADER"); + ASSERT_GT(http_res_table_id, 0); + + ret = maat_state_set_scan_district(state, http_res_table_id, "Cookie", strlen("Cookie")); + EXPECT_EQ(ret, 0); + + ret = maat_scan_string(maat_inst, http_res_table_id, http_resp_hdr_cookie, strlen(http_resp_hdr_cookie), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HIT); + EXPECT_EQ(n_hit_result, 1); + EXPECT_EQ(results[0], compile1_id); + scan_count = maat_state_get_scan_count(state); + EXPECT_EQ(scan_count, 2); + + n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, 128); + EXPECT_EQ(n_hit_group, 4); + EXPECT_EQ(hit_groups[0].item_id, item1_id); + EXPECT_EQ(hit_groups[0].group_id, group1_id); + EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[1].item_id, item1_id); + EXPECT_EQ(hit_groups[1].group_id, group11_id); + EXPECT_EQ(hit_groups[1].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[2].item_id, item2_id); + EXPECT_EQ(hit_groups[2].group_id, group21_id); + EXPECT_EQ(hit_groups[2].vtable_id, http_res_table_id); + + EXPECT_EQ(hit_groups[3].item_id, item2_id); + EXPECT_EQ(hit_groups[3].group_id, group2_id); + EXPECT_EQ(hit_groups[3].vtable_id, http_res_table_id); + + memset(hit_groups, 0, sizeof(hit_groups)); + n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_INC, hit_groups, 128); + EXPECT_EQ(n_hit_group, 2); + EXPECT_EQ(hit_groups[0].item_id, item2_id); + EXPECT_EQ(hit_groups[0].group_id, group21_id); + EXPECT_EQ(hit_groups[0].vtable_id, http_res_table_id); + + EXPECT_EQ(hit_groups[1].item_id, item2_id); + EXPECT_EQ(hit_groups[1].group_id, group2_id); + EXPECT_EQ(hit_groups[1].vtable_id, http_res_table_id); + + + const char* keywords1="In graph theory, hit group item forth"; + const char *keywords2="To test one group hit group item fifth"; + + int keywords_table_id = maat_get_table_id(maat_inst, keywords_table_name); + ASSERT_GT(keywords_table_id, 0); + + struct maat_stream *stream = maat_stream_new(maat_inst, keywords_table_id, state); + ret = maat_stream_scan(stream, keywords1, strlen(keywords1), results, ARRAY_SIZE, + &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HALF_HIT); + scan_count = maat_state_get_scan_count(state); + EXPECT_EQ(scan_count, 3); + + n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, 128); + EXPECT_EQ(n_hit_group, 5); + EXPECT_EQ(hit_groups[0].item_id, item1_id); + EXPECT_EQ(hit_groups[0].group_id, group1_id); + EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[1].item_id, item1_id); + EXPECT_EQ(hit_groups[1].group_id, group11_id); + EXPECT_EQ(hit_groups[1].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[2].item_id, item2_id); + EXPECT_EQ(hit_groups[2].group_id, group21_id); + EXPECT_EQ(hit_groups[2].vtable_id, http_res_table_id); + + EXPECT_EQ(hit_groups[3].item_id, item2_id); + EXPECT_EQ(hit_groups[3].group_id, group2_id); + EXPECT_EQ(hit_groups[3].vtable_id, http_res_table_id); + + EXPECT_EQ(hit_groups[4].item_id, item4_id); + EXPECT_EQ(hit_groups[4].group_id, group4_id); + EXPECT_EQ(hit_groups[4].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 + + int ip_table_id = maat_get_table_id(maat_inst, ip_table_name); + ASSERT_GT(ip_table_id, 0); + + uint32_t ip_addr; + inet_pton(AF_INET, "220.181.38.150", &ip_addr); + uint16_t port = htons(17272); + ret = maat_scan_ipv4(maat_inst, ip_table_id, ip_addr, port, 6, results, ARRAY_SIZE, + &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HALF_HIT); + scan_count = maat_state_get_scan_count(state); + EXPECT_EQ(scan_count, 4); + + n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, 128); + EXPECT_EQ(n_hit_group, 6); + EXPECT_EQ(hit_groups[0].item_id, item1_id); + EXPECT_EQ(hit_groups[0].group_id, group1_id); + EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[1].item_id, item1_id); + EXPECT_EQ(hit_groups[1].group_id, group11_id); + EXPECT_EQ(hit_groups[1].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[2].item_id, item2_id); + EXPECT_EQ(hit_groups[2].group_id, group21_id); + EXPECT_EQ(hit_groups[2].vtable_id, http_res_table_id); + + EXPECT_EQ(hit_groups[3].item_id, item2_id); + EXPECT_EQ(hit_groups[3].group_id, group2_id); + EXPECT_EQ(hit_groups[3].vtable_id, http_res_table_id); + + EXPECT_EQ(hit_groups[4].item_id, item3_id); + EXPECT_EQ(hit_groups[4].group_id, group3_id); + EXPECT_EQ(hit_groups[4].vtable_id, 0); //physical table(ip_table) vtable_id is 0 + + EXPECT_EQ(hit_groups[5].item_id, item4_id); + EXPECT_EQ(hit_groups[5].group_id, group4_id); + EXPECT_EQ(hit_groups[5].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 + + memset(hit_groups, 0, sizeof(hit_groups)); + n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_INC, hit_groups, 128); + EXPECT_EQ(n_hit_group, 2); + EXPECT_EQ(hit_groups[0].item_id, item3_id); + EXPECT_EQ(hit_groups[0].group_id, group3_id); + EXPECT_EQ(hit_groups[0].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 + + EXPECT_EQ(hit_groups[1].item_id, item4_id); + EXPECT_EQ(hit_groups[1].group_id, group4_id); + EXPECT_EQ(hit_groups[1].vtable_id, 0); + + ret = maat_stream_scan(stream, keywords2, strlen(keywords2), results, ARRAY_SIZE, + &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HALF_HIT); + + scan_count = maat_state_get_scan_count(state); + EXPECT_EQ(scan_count, 5); + + memset(hit_groups, 0, sizeof(hit_groups)); + n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, 128); + EXPECT_EQ(n_hit_group, 8); + EXPECT_EQ(hit_groups[0].item_id, item1_id); + EXPECT_EQ(hit_groups[0].group_id, group1_id); + EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); + + EXPECT_EQ(hit_groups[1].item_id, item1_id); + EXPECT_EQ(hit_groups[1].group_id, group11_id); + EXPECT_EQ(hit_groups[1].vtable_id, http_req_table_id); + + + EXPECT_EQ(hit_groups[2].item_id, item2_id); + EXPECT_EQ(hit_groups[2].group_id, group21_id); + EXPECT_EQ(hit_groups[2].vtable_id, http_res_table_id); + + EXPECT_EQ(hit_groups[3].item_id, item2_id); + EXPECT_EQ(hit_groups[3].group_id, group2_id); + EXPECT_EQ(hit_groups[3].vtable_id, http_res_table_id); + + EXPECT_EQ(hit_groups[4].item_id, item3_id); + EXPECT_EQ(hit_groups[4].group_id, group3_id); + EXPECT_EQ(hit_groups[4].vtable_id, 0); //physical table(ip_table) vtable_id is 0 + + EXPECT_EQ(hit_groups[5].item_id, item4_id); + EXPECT_EQ(hit_groups[5].group_id, group4_id); + EXPECT_EQ(hit_groups[5].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 + + EXPECT_EQ(hit_groups[6].item_id, item5_id); + EXPECT_EQ(hit_groups[6].group_id, group1_id); + EXPECT_EQ(hit_groups[6].vtable_id, 0); + + EXPECT_EQ(hit_groups[7].item_id, item5_id); + EXPECT_EQ(hit_groups[7].group_id, group11_id); + EXPECT_EQ(hit_groups[7].vtable_id, 0); + + memset(hit_groups, 0, sizeof(hit_groups)); + n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_INC, hit_groups, 128); + EXPECT_EQ(n_hit_group, 2); + EXPECT_EQ(hit_groups[0].item_id, item5_id); + EXPECT_EQ(hit_groups[0].group_id, group1_id); + EXPECT_EQ(hit_groups[0].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 + + EXPECT_EQ(hit_groups[1].item_id, item5_id); + EXPECT_EQ(hit_groups[1].group_id, group11_id); + EXPECT_EQ(hit_groups[1].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 + + maat_stream_free(stream); + maat_state_free(state); + state = NULL; +} + +TEST_F(MaatCmdTest, HitPath) { + const char *g2g_table_name = "GROUP2GROUP"; + const char *g2c_table_name = "GROUP2COMPILE"; + const char *compile_table_name = "COMPILE"; + const char *http_sig_table_name = "HTTP_SIGNATURE"; + const char *ip_table_name = "IP_CONFIG"; + const char *keywords_table_name = "KEYWORDS_TABLE"; + int thread_id = 0; + struct maat *maat_inst = MaatCmdTest::_shared_maat_inst; + struct maat_state *state = maat_state_new(maat_inst, thread_id); + + /* compile1 */ + long long compile1_id = maat_cmd_incrby(maat_inst, "TEST_SEQ", 1); + int ret = compile_table_set_line(maat_inst, compile_table_name, MAAT_OP_ADD, compile1_id, + "null", 2, 0); + EXPECT_EQ(ret, 1); + + //group1 -> compile1 + long long group1_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); + ret = group2compile_table_set_line(maat_inst, g2c_table_name, MAAT_OP_ADD, group1_id, + compile1_id, 0, "HTTP_REQUEST_HEADER", 1, 0); + EXPECT_EQ(ret, 1); + + //item1 -> group1 -> compile1 + long long item1_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); + ret = expr_table_set_line(maat_inst, http_sig_table_name, MAAT_OP_ADD, item1_id, group1_id, + "graph_theory", "URL", 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ + EXPECT_EQ(ret, 1); + + /* item1 -> group1 -> compile1 + / + group21_/ + */ + long long group21_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); + ret = group2compile_table_set_line(maat_inst, g2c_table_name, MAAT_OP_ADD, group21_id, + compile1_id, 0, "HTTP_RESPONSE_HEADER", 2, 0); + EXPECT_EQ(ret, 1); + + /* item1 -> group1 -> compile1 + / + group2 -> group21 _/ + */ + long long group2_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); + ret = group2group_table_set_line(maat_inst, g2g_table_name, MAAT_OP_ADD, group2_id, + group21_id, 0, 0); + EXPECT_EQ(ret, 1); + + /* item1 -> group1 -> compile1 + / + item2 -> group2 -> group21 _/ + */ + long long item2_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); + ret = expr_table_set_line(maat_inst, http_sig_table_name, MAAT_OP_ADD, item2_id, group2_id, + "time=2020-02-11", "Cookie", 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ + EXPECT_EQ(ret, 1); + + /* + item1 -> group1 -> group11 + \ + \ -> compile1 + / + item2 -> group2 -> group21 _/ + */ + long long group11_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); + ret = group2group_table_set_line(maat_inst, g2g_table_name, MAAT_OP_ADD, group1_id, + group11_id, 0, 0); + EXPECT_EQ(ret, 1); + + //item3 -> group3, group3 is not referenced by any compile. + long long item3_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); + long long group3_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); + ret = ip_table_set_line(maat_inst, ip_table_name, MAAT_OP_ADD, item3_id, group3_id, + IPv4, "220.181.38.148", "220.181.38.149", 0, 65535, 0); + EXPECT_EQ(ret, 1); + + char temp[1024]={0}; + //item4 -> group4, group4 is not referenced by any compile. + long long item4_id = maat_cmd_incrby(maat_inst, "SEQUENCE_REGION", 1); + long long group4_id = maat_cmd_incrby(maat_inst, "SEQUENCE_GROUP", 1); + ret = expr_table_set_line(maat_inst, keywords_table_name, MAAT_OP_ADD, item4_id, group4_id, + str_escape(temp, sizeof(temp), "a finite or infinite"), + NULL, 0, 0, 0, 0); /*EXPR_TYPE_STRING MATCH_METHOD_SUB*/ + EXPECT_EQ(ret, 1); + + sleep(WAIT_FOR_EFFECTIVE_S * 2); + + const char* http_url = "en.wikipedia.org/wiki/Path_(graph_theory)"; + const char* http_resp_hdr_cookie = "laptop=thinkpad X1 extrem;time=2020-02-11T15:34:00;" + "main[XWJOKE]=hoho; Hm_lvt_bbac0322e6ee13093f98d5c4b5a10912=1578874808;"; + + int http_req_table_id = maat_get_table_id(maat_inst, "HTTP_REQUEST_HEADER"); + ASSERT_GT(http_req_table_id, 0); + + ret = maat_state_set_scan_district(state, http_req_table_id, "URL", strlen("URL")); + EXPECT_EQ(ret, 0); + + int Nth_scan = 0; + + Nth_scan++; + long long results[ARRAY_SIZE] = {0}; + size_t n_hit_result = 0; + ret = maat_scan_string(maat_inst, http_req_table_id, http_url, strlen(http_url), + results, ARRAY_SIZE, &n_hit_result, state); + EXPECT_EQ(ret, MAAT_SCAN_HALF_HIT); + + size_t scan_count = maat_state_get_scan_count(state); + EXPECT_EQ(scan_count, 1); + struct maat_hit_path hit_path[128]; memset(hit_path, 0, sizeof(hit_path)); int n_read = maat_state_get_hit_paths(state, hit_path, sizeof(hit_path)); @@ -5715,25 +6795,6 @@ TEST_F(MaatCmdTest, HitPath) { scan_count = maat_state_get_scan_count(state); EXPECT_EQ(scan_count, 2); - n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, sizeof(hit_groups)); - EXPECT_EQ(n_hit_group, 4); - EXPECT_EQ(hit_groups[0].group_id, group1_id); - EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); - EXPECT_EQ(hit_groups[1].group_id, group21_id); - EXPECT_EQ(hit_groups[1].vtable_id, http_res_table_id); - EXPECT_EQ(hit_groups[2].group_id, group2_id); - EXPECT_EQ(hit_groups[2].vtable_id, http_res_table_id); - EXPECT_EQ(hit_groups[3].group_id, group11_id); - EXPECT_EQ(hit_groups[3].vtable_id, http_req_table_id); - - memset(hit_groups, 0, sizeof(hit_groups)); - n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_INC, hit_groups, sizeof(hit_groups)); - EXPECT_EQ(n_hit_group, 2); - EXPECT_EQ(hit_groups[0].group_id, group21_id); - EXPECT_EQ(hit_groups[0].vtable_id, http_res_table_id); - EXPECT_EQ(hit_groups[1].group_id, group2_id); - EXPECT_EQ(hit_groups[1].vtable_id, http_res_table_id); - n_read = maat_state_get_hit_paths(state, hit_path, sizeof(hit_path)); EXPECT_EQ(n_read, 4); @@ -5787,25 +6848,6 @@ that the edges be all directed in the same direction."; scan_count = maat_state_get_scan_count(state); EXPECT_EQ(scan_count, 3); - n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, sizeof(hit_groups)); - EXPECT_EQ(n_hit_group, 5); - EXPECT_EQ(hit_groups[0].group_id, group1_id); - EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); - EXPECT_EQ(hit_groups[1].group_id, group21_id); - EXPECT_EQ(hit_groups[1].vtable_id, http_res_table_id); - EXPECT_EQ(hit_groups[2].group_id, group2_id); - EXPECT_EQ(hit_groups[2].vtable_id, http_res_table_id); - EXPECT_EQ(hit_groups[3].group_id, group11_id); - EXPECT_EQ(hit_groups[3].vtable_id, http_req_table_id); - EXPECT_EQ(hit_groups[4].group_id, group4_id); - EXPECT_EQ(hit_groups[4].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 - - memset(hit_groups, 0, sizeof(hit_groups)); - n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_INC, hit_groups, sizeof(hit_groups)); - EXPECT_EQ(n_hit_group, 1); - EXPECT_EQ(hit_groups[0].group_id, group4_id); - EXPECT_EQ(hit_groups[0].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 - n_read = maat_state_get_hit_paths(state, hit_path, sizeof(hit_path)); EXPECT_EQ(n_read, 5); @@ -5831,27 +6873,6 @@ that the edges be all directed in the same direction."; scan_count = maat_state_get_scan_count(state); EXPECT_EQ(scan_count, 4); - n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_FULL, hit_groups, sizeof(hit_groups)); - EXPECT_EQ(n_hit_group, 6); - EXPECT_EQ(hit_groups[0].group_id, group1_id); - EXPECT_EQ(hit_groups[0].vtable_id, http_req_table_id); - EXPECT_EQ(hit_groups[1].group_id, group21_id); - EXPECT_EQ(hit_groups[1].vtable_id, http_res_table_id); - EXPECT_EQ(hit_groups[2].group_id, group2_id); - EXPECT_EQ(hit_groups[2].vtable_id, http_res_table_id); - EXPECT_EQ(hit_groups[3].group_id, group11_id); - EXPECT_EQ(hit_groups[3].vtable_id, http_req_table_id); - EXPECT_EQ(hit_groups[4].group_id, group3_id); - EXPECT_EQ(hit_groups[4].vtable_id, 0); //physical table(ip_table) vtable_id is 0 - EXPECT_EQ(hit_groups[5].group_id, group4_id); - EXPECT_EQ(hit_groups[5].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 - - memset(hit_groups, 0, sizeof(hit_groups)); - n_hit_group = maat_state_get_hit_groups(state, MAAT_LIST_TYPE_INC, hit_groups, sizeof(hit_groups)); - EXPECT_EQ(n_hit_group, 1); - EXPECT_EQ(hit_groups[0].group_id, group3_id); - EXPECT_EQ(hit_groups[0].vtable_id, 0); //physical table(keywords_table) vtable_id is 0 - n_read = maat_state_get_hit_paths(state, hit_path, sizeof(hit_path)); EXPECT_EQ(n_read, 6); @@ -6217,7 +7238,7 @@ TEST_F(MaatCmdTest, CompileDelete_TSG6548) { IPv4, "192.168.73.163", "192.168.73.180", 0, 65535, 0); EXPECT_EQ(ret, 1); - sleep(WAIT_FOR_EFFECTIVE_S); + sleep(WAIT_FOR_EFFECTIVE_S * 2); uint32_t ip_addr; inet_pton(AF_INET, "192.168.73.169", &ip_addr); @@ -6445,7 +7466,7 @@ TEST_F(MaatCmdTest, StreamScanSegfaultWhenVersionRollBack_TSG6324) { //DON'T DO THIS!!! //Roll back version, trigger full update. //This operation generates FATAL logs in test_maat_redis.log.yyyy-mm-dd. - //For example: Add group 22 vt_id 0 to clause 2 of compile 979 failed, group is already exisited + //For example: Add group 22 vt_id 0 to clause 2 of compile 979 failed, group is already existed maat_cmd_incrby(maat_inst, "MAAT_VERSION", -100); //Wating for scanner garbage collect expiration. |
