feature： TSG-21853 Refactoring TFE Kafka infrastructure

1 files changed, 7 insertions, 14 deletions

diff --git a/platform/src/ssl_service_cache.cpp b/platform/src/ssl_service_cache.cpp
index 0561f43..13c7adc 100644
--- a/platform/src/ssl_service_cache.cpp
+++ b/platform/src/ssl_service_cache.cpp
@@ -3,12 +3,6 @@
 #include <tfe_resource.h>
 #include <ssl_stream.h>
 
-struct ssl_ja3_enforcer
-{
-    struct maat *maat;
-    int table_id;
-};
-
 struct ssl_svc_ja3
 {
     char ja3_hash[33];
@@ -26,7 +20,7 @@ struct ssl_svc_addr
 	const char *dport;
 };
 
-static struct ssl_ja3_enforcer g_static_enforcer = {0};
+static int table_id = 0;
 
 static void ssl_svc_ja3_param_dup_cb(int table_id, void **to, void **from, long argl, void *argp)
 {
@@ -93,24 +87,23 @@ static void ssl_svc_ja3_param_free(struct ssl_svc_ja3 *param)
 
 static int ssl_svc_ja3_init(const char *table_name)
 {
-    g_static_enforcer.maat = (struct maat *)tfe_bussiness_resouce_get(STATIC_MAAT);
-    g_static_enforcer.table_id = maat_get_table_id(g_static_enforcer.maat, table_name);
-    if (g_static_enforcer.table_id < 0)
+    table_id = maat_get_table_id(tfe_get_maat_handle(), table_name);
+    if (table_id < 0)
     {
         TFE_LOG_ERROR(g_default_logger, "Maat table %s register failed.", table_name);
         return 0;
     }
-    int ret = maat_plugin_table_ex_schema_register(g_static_enforcer.maat,
+    int ret = maat_plugin_table_ex_schema_register(tfe_get_maat_handle(),
                                      	 		table_name,
                                       			ssl_svc_ja3_param_new_cb,
                                       			ssl_svc_ja3_param_free_cb,
                                       			ssl_svc_ja3_param_dup_cb,
                                       			0,
-                                      			&g_static_enforcer);
+                                      			NULL);
     if (ret < 0)
     {
         TFE_LOG_ERROR(g_default_logger, "failed at Maat_plugin_EX_register(%s), table_id = %d, ret = %d",
-                      table_name, g_static_enforcer.table_id, ret);
+                      table_name, table_id, ret);
         return 0;
     }
 
@@ -122,7 +115,7 @@ enum ssl_ja3_pinning_status ssl_svc_ja3_scan(char *ja3_hash, const char *addr_st
     enum ssl_ja3_pinning_status ret = JA3_PINNING_STATUS_UNKNOWN;
     struct ssl_svc_ja3 *param = NULL;
 
-    param = (struct ssl_svc_ja3 *)maat_plugin_table_get_ex_data(g_static_enforcer.maat, g_static_enforcer.table_id, ja3_hash, strlen(ja3_hash));
+    param = (struct ssl_svc_ja3 *)maat_plugin_table_get_ex_data(tfe_get_maat_handle(), table_id, ja3_hash, strlen(ja3_hash));
     if (param == NULL)
     {
         ret = JA3_PINNING_STATUS_UNKNOWN;