1 files changed, 169 insertions, 0 deletions
diff --git a/test/src/gtest_fieldstat3.cpp b/test/src/gtest_fieldstat3.cpp
new file mode 100644
index 0000000..0e9df5c
--- /dev/null
+++ b/test/src/gtest_fieldstat3.cpp
@@ -0,0 +1,169 @@
+#include <stdio.h>
+#include <string.h>
+#include <unistd.h>
+#include <MESA/fieldstat.h>
+
+#include "tsg_entry.h"
+#include "gtest_common.h"
+#include "tsg_variable.h"
+#include <MESA/MESA_prof_load.h>
+#include <MESA/MESA_handle_logger.h>
+
+#include <gtest/gtest.h>
+
+const char *tsg_gtest_conffile = "tsgconf/main.conf";
+extern struct tsg_statistic g_tsg_statis_para;
+pthread_t g_pid[8];
+
+TEST(FIELDSATA3, InterceptIllegalParameter)
+{
+    struct _traffic_info _info;
+    struct maat_rule p_result;
+
+    int ret = tsg_set_intercept_flow(NULL, &_info, 0);
+    EXPECT_EQ(ret, -1);
+
+    ret = tsg_set_intercept_flow(&p_result, NULL, 0);
+    EXPECT_EQ(ret, -1);
+
+    ret = tsg_set_intercept_flow(&p_result, &_info, -1);
+    EXPECT_EQ(ret, -1);
+}
+
+TEST(FIELDSATA3, PolicyIllegalParameter)
+{
+    struct maat_rule p_result;
+    struct streaminfo a_stream;
+
+    int ret = tsg_set_policy_flow(NULL, &p_result, 0);
+    EXPECT_EQ(ret, -1);
+
+    ret = tsg_set_policy_flow(&a_stream, NULL, 0);
+    EXPECT_EQ(ret, -1);
+
+    ret = tsg_set_policy_flow(&a_stream, &p_result, -1);
+    EXPECT_EQ(ret, -1);
+}
+
+TEST(FIELDSATA3, Intercept)
+{
+    struct _traffic_info _info;
+    struct maat_rule p_result;
+
+    _info.con_num = 10;
+    _info.in_bytes = 1000;
+    _info.in_packets = 25;
+    _info.out_bytes = 1001;
+    _info.out_packets = 24;
+
+    p_result.action = TSG_ACTION_INTERCEPT;
+    p_result.rule_id = 95536;
+    p_result.do_log = 1;
+    p_result.service_id = 10;
+
+    for (int i = 0; i < 50000; i++)
+    {
+        // p_result.rule_id += i;
+        // _info.in_bytes += i;
+        int ret = tsg_set_intercept_flow(&p_result, &_info, 0);
+        EXPECT_EQ(ret, 0);
+        // usleep(500);
+    }
+}
+
+TEST(FIELDSATA3, PolicyFlow)
+{
+    struct maat_rule  p_result;
+    struct streaminfo a_stream;
+
+    p_result.action = TSG_ACTION_MONITOR;
+    p_result.rule_id = 95500;
+
+    for (int i = 0; i < 50000; i++)
+    {
+        // p_result.rule_id += i;
+        // _info.in_bytes += i;
+        int ret = tsg_set_policy_flow(&a_stream, &p_result, 0);
+        EXPECT_EQ(ret, 0);
+        // usleep(500);
+    }
+}
+
+#if 0
+void *run_time_funtion(void *arg)
+{
+    uint8_t pid = *(uint8_t *)arg;
+
+    struct maat_rule p_result;
+    struct streaminfo a_stream;
+    struct _traffic_info _info;
+
+    while (1)
+    {
+        srand(time(NULL));
+        p_result.action = TSG_ACTION_MONITOR;
+        p_result.rule_id = rand() % 95500;
+
+        usleep(rand() % 500);
+        int ret = tsg_set_policy_flow(&a_stream, &p_result, pid);
+        EXPECT_EQ(ret, 0);
+
+        srand(time(NULL));
+        usleep(rand() % 500);
+
+        p_result.action = TSG_ACTION_DENY;
+        p_result.rule_id = rand() % 95500;
+        ret = tsg_set_policy_flow(&a_stream, &p_result, pid);
+        EXPECT_EQ(ret, 0);
+
+        srand(time(NULL));
+        usleep(rand() % 500);
+
+        p_result.action = TSG_ACTION_BYPASS;
+        p_result.rule_id = rand() % 95500;
+        ret = tsg_set_policy_flow(&a_stream, &p_result, pid);
+        EXPECT_EQ(ret, 0);
+
+        _info.con_num = rand() % 10;
+        _info.in_bytes = rand() % 1000;
+        _info.in_packets = rand() % 25;
+        _info.out_bytes = rand() % 1001;
+        _info.out_packets = rand() % 24;
+
+        p_result.action = TSG_ACTION_INTERCEPT;
+        p_result.rule_id = rand() % 95536;
+
+        ret = tsg_set_intercept_flow(&p_result, &_info, pid);
+        EXPECT_EQ(ret, 0);
+    }
+
+    return NULL;
+}
+
+TEST(FIELDSATA3, MultiThreading)
+{
+    for (uint8_t i = 0; i < 8; i++)
+    {
+        uint8_t i_pid = i;
+        pthread_create(g_pid + i, NULL, run_time_funtion, (void *)&i_pid);
+        EXPECT_NE(g_pid[i], 0);
+    }
+}
+#endif
+
+int main(int argc, char *argv[])
+{
+    void *logger = MESA_create_runtime_log_handle("log/gtest_fieldstat3.log", RLOG_LV_FATAL);
+    tsg_statistic_init(tsg_gtest_conffile, logger);
+    testing::InitGoogleTest(&argc, argv);
+    int ret = RUN_ALL_TESTS();
+//    sleep(30);
+//    for (int i = 0; i < 8; i++)
+//    {
+//        pthread_cancel(g_pid[i]);
+//    }
+
+    tsg_statistic_destroy();
+
+    return ret;
+}