1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
#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_global_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);
shaper_global_stat_curr_session_inc(ctx->global_stat);
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);
shaper_global_stat_curr_session_dec(ctx->global_stat);
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);
shaping_flow_free(sf);
}
shaper_global_stat_curr_session_dec(ctx->global_stat);
return;
}
|
