add project code

1 files changed, 88 insertions, 0 deletions

diff --git a/shaping/src/shaper_session.cpp b/shaping/src/shaper_session.cpp
new file mode 100644
index 0000000..badcea9
--- /dev/null
+++ b/shaping/src/shaper_session.cpp
@@ -0,0 +1,88 @@
+#include "session_table.h"
+#include "raw_packet.h"
+#include "utils.h"
+#include "log.h"
+#include "shaper_marsio.h"
+#include "shaper_session.h"
+#include "shaper_maat.h"
+#include "shaper_stat.h"
+#include "shaper.h"
+
+struct shaping_flow* shaper_session_opening(struct shaping_thread_ctx *ctx, struct metadata *meta, struct ctrl_pkt_data *ctrl_data, struct raw_pkt_parser *raw_parser)
+{
+    struct shaping_flow *sf = NULL;
+    struct session_node *node = NULL;
+
+    node  = session_table_search_by_id(ctx->session_table, meta->session_id);
+    if (node) {
+        sf = (struct shaping_flow *)node->val_data;
+        LOG_ERROR("%s: session id %lu for %s has already exist", LOG_TAG_SHAPING, meta->session_id, addr_tuple4_to_str(&sf->tuple4));
+        return NULL;
+    }
+
+    sf = shaping_flow_new();
+    raw_packet_parser_get_most_inner_tuple4(raw_parser, &sf->tuple4);
+    shaper_rules_update(ctx, sf, ctrl_data->shaping_rule_ids, ctrl_data->shaping_rule_num);
+
+    session_table_insert(ctx->session_table, meta->session_id, &sf->tuple4, sf, NULL);
+
+    return sf;
+}
+
+struct shaping_flow* shaper_session_close(struct shaping_thread_ctx *ctx, struct metadata *meta)
+{
+    struct session_node *session_node = NULL;
+    struct shaping_flow *sf = NULL;
+
+    session_node = session_table_search_by_id(ctx->session_table, meta->session_id);
+    if (!session_node) {
+        return NULL;
+    }
+
+    sf = (struct shaping_flow *)session_node->val_data;
+    sf->flag |= STREAM_CLOSE;
+    session_table_delete_by_id(ctx->session_table, meta->session_id);
+
+    return sf;
+}
+
+struct shaping_flow* shaper_session_active(struct shaping_thread_ctx *ctx, struct metadata *meta, struct ctrl_pkt_data *ctrl_data)
+{
+    struct shaping_flow *sf = NULL;
+    struct session_node *node = NULL;
+
+    node = session_table_search_by_id(ctx->session_table, meta->session_id);
+    if (!node) {
+        return NULL;
+    }
+
+    sf = (struct shaping_flow *)node->val_data;
+    shaper_rules_update(ctx, sf, ctrl_data->shaping_rule_ids, ctrl_data->shaping_rule_num);
+
+    return sf;
+}
+
+struct shaping_flow* shaper_session_reset_all(struct shaping_thread_ctx *ctx, struct metadata *meta)
+{
+    struct shaping_ctx *shaping_ctx = ctx->ref_ctx;
+
+    LOG_ERROR("%s: session %lu resetall: notification clears all session tables !!!", LOG_TAG_SHAPING, meta->session_id);
+    for (int i = 0; i < shaping_ctx->thread_num; i++) {
+        __atomic_add_fetch(&shaping_ctx->thread_ctx[i].session_need_reset, 1, __ATOMIC_SEQ_CST);
+    }
+
+    return NULL;
+}
+
+void shaper_session_data_free_cb(void *session_data, void *data)
+{
+    struct shaping_flow *sf = (struct shaping_flow *)session_data;
+    struct shaping_thread_ctx *ctx = (struct shaping_thread_ctx *)data;
+
+    if (sf) {
+        shaper_queue_clear(sf, &ctx->stat->stat_hashtbl, ctx);
+        shaping_flow_free(sf);
+    }
+
+    return;
+}
\ No newline at end of file