TSG-13347 统一术语使用session替换stream

common/src/session_table.cpp

@@ -0,0 +1,195 @@
+#include <assert.h>
+
+#include "session_table.h"
+#include "log.h"
+
+struct session_table
+{
+    struct session_node *root_by_id;
+    struct session_node *root_by_addr;
+    uint64_t session_node_count;
+};
+
+// Note: session_addr must be initialized by memset(0) before use !!!
+
+struct session_table *session_table_create()
+{
+    struct session_table *table = (struct session_table *)calloc(1, sizeof(struct session_table));
+    assert(table);
+    table->session_node_count = 0;
+
+    return table;
+}
+
+void session_table_destory(struct session_table *table)
+{
+    if (table)
+    {
+        struct session_node *temp = NULL;
+        struct session_node *node = NULL;
+        HASH_ITER(hh1, table->root_by_id, node, temp)
+        {
+            HASH_DELETE(hh1, table->root_by_id, node);
+            HASH_DELETE(hh2, table->root_by_addr, node);
+
+            if (node->val_freecb && node->val_data)
+            {
+                node->val_freecb(node->val_data);
+            }
+
+            free(node);
+            node = NULL;
+        }
+
+        free(table);
+        table = NULL;
+    }
+}
+
+uint64_t session_table_count(struct session_table *table)
+{
+    if (table)
+    {
+        return table->session_node_count;
+    }
+    else
+    {
+        return 0;
+    }
+}
+
+// session_addr : deep copy
+// val_data     : shallow copy (malloc by user, free by val_freecb)
+int session_table_insert(struct session_table *table, uint64_t session_id, const struct addr_tuple4 *session_addr, void *val_data, const fn_free_cb *val_freecb)
+{
+    struct session_node *temp = NULL;
+    HASH_FIND(hh1, table->root_by_id, &session_id, sizeof(session_id), temp);
+    if (temp)
+    {
+        LOG_DEBUG("session table insert: key %lu exists", session_id);
+        return -1;
+    }
+
+    temp = (struct session_node *)calloc(1, sizeof(struct session_node));
+    assert(temp);
+
+    temp->session_id = session_id;
+    memcpy(&temp->session_addr, session_addr, sizeof(struct addr_tuple4));
+    temp->val_data = val_data;
+    temp->val_freecb = val_freecb;
+
+    HASH_ADD(hh1, table->root_by_id, session_id, sizeof(temp->session_id), temp);
+    HASH_ADD(hh2, table->root_by_addr, session_addr, sizeof(temp->session_addr), temp);
+
+    LOG_DEBUG("session table insert: key %lu success", session_id);
+    table->session_node_count++;
+
+    return 0;
+}
+
+int session_table_delete_by_id(struct session_table *table, uint64_t session_id)
+{
+    struct session_node *temp = NULL;
+    HASH_FIND(hh1, table->root_by_id, &session_id, sizeof(session_id), temp);
+    if (!temp)
+    {
+        LOG_DEBUG("session table delete: key %lu not exists", session_id);
+        return -1;
+    }
+
+    HASH_DELETE(hh1, table->root_by_id, temp);
+    HASH_DELETE(hh2, table->root_by_addr, temp);
+
+    if (temp->val_freecb && temp->val_data)
+    {
+        temp->val_freecb(temp->val_data);
+        temp->val_data = NULL;
+    }
+
+    free(temp);
+    temp = NULL;
+
+    LOG_DEBUG("session table delete: key %lu success", session_id);
+    table->session_node_count--;
+
+    return 0;
+}
+
+int session_table_delete_by_addr(struct session_table *table, const struct addr_tuple4 *session_addr)
+{
+    struct session_node *temp = NULL;
+    char *addr_str = addr_tuple4_to_str(session_addr);
+    HASH_FIND(hh2, table->root_by_addr, session_addr, sizeof(struct addr_tuple4), temp);
+    if (!temp)
+    {
+        struct addr_tuple4 reverse_addr;
+        addr_tuple4_reverse(session_addr, &reverse_addr);
+        HASH_FIND(hh2, table->root_by_addr, &reverse_addr, sizeof(struct addr_tuple4), temp);
+        if (!temp)
+        {
+            LOG_DEBUG("session table delete: key %s not exists", addr_str);
+            free(addr_str);
+            return -1;
+        }
+    }
+
+    HASH_DELETE(hh1, table->root_by_id, temp);
+    HASH_DELETE(hh2, table->root_by_addr, temp);
+
+    if (temp->val_freecb && temp->val_data)
+    {
+        temp->val_freecb(temp->val_data);
+        temp->val_data = NULL;
+    }
+
+    free(temp);
+    temp = NULL;
+
+    LOG_DEBUG("session table delete: key %s success", addr_str);
+    free(addr_str);
+    addr_str = NULL;
+    table->session_node_count--;
+
+    return 0;
+}
+
+struct session_node *session_table_search_by_id(struct session_table *table, uint64_t session_id)
+{
+    struct session_node *temp = NULL;
+    HASH_FIND(hh1, table->root_by_id, &session_id, sizeof(session_id), temp);
+    if (!temp)
+    {
+        LOG_DEBUG("session table search: key %lu not exists", session_id);
+        return NULL;
+    }
+
+    LOG_DEBUG("session table search: key %lu success", session_id);
+
+    return temp;
+}
+
+struct session_node *session_table_search_by_addr(struct session_table *table, const struct addr_tuple4 *session_addr)
+{
+    struct session_node *temp = NULL;
+    char *addr_str = addr_tuple4_to_str(session_addr);
+    HASH_FIND(hh2, table->root_by_addr, session_addr, sizeof(struct addr_tuple4), temp);
+    if (!temp)
+    {
+        struct addr_tuple4 reverse_addr;
+        addr_tuple4_reverse(session_addr, &reverse_addr);
+        HASH_FIND(hh2, table->root_by_addr, &reverse_addr, sizeof(struct addr_tuple4), temp);
+        if (!temp)
+        {
+            LOG_DEBUG("session table search: key %s not exists", addr_str);
+            free(addr_str);
+            addr_str = NULL;
+            return NULL;
+        }
+    }
+
+    LOG_DEBUG("session table search: key %s success", addr_str);
+    free(addr_str);
+    addr_str = NULL;
+
+    return temp;
+}