gfwleak/tango-tsg-service-chaining-engine
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TSG-13698 tsg-service-chaining-engine向SAPP发送控制报文

Browse Source
This commit is contained in:
luwenpeng
2023-03-06 15:09:01 +08:00
parent 75c2ba842f
commit c56bc6a824
3 changed files with 82 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
platform/src/packet_io.cpp
View File

@@ -4,6 +4,7 @@
#include <netinet/ether.h>
#include <marsio.h>
#include <cjson/cJSON.h>
#include <MESA/MESA_prof_load.h>
#include "log.h"
@@ -1167,6 +1168,75 @@ static int handle_session_opening(struct metadata *meta, struct ctrl_pkt_parser
return 0;
}
/*
{
"tsync": "1.0",
"session_id": "123456789",
"state": "active",
"method": "log_update",
"params": {
"sf_profile_ids": [
2,
3,
4,
5,
6,
7
]
}
}
*/
static void send_event_log(struct session_ctx *s_ctx, int thread_seq, void *ctx)
{
struct thread_ctx *thread = (struct thread_ctx *)ctx;
struct sce_ctx *sce_ctx = thread->ref_sce_ctx;
struct packet_io *packet_io = thread->ref_io;
struct selected_chaining *chaining = s_ctx->chaining;
char buffer[32] = {0};
sprintf(buffer, "%lu", s_ctx->session_id);
cJSON *root = cJSON_CreateObject();
cJSON_AddStringToObject(root, "tsync", "1.0");
cJSON_AddStringToObject(root, "session_id", buffer);
cJSON_AddStringToObject(root, "state", "closing");
cJSON_AddStringToObject(root, "method", "log_update");
cJSON *sf_profile_ids = cJSON_CreateArray();
for (int i = 0; i < chaining->chaining_used; i++)
{
struct selected_sf *node = &(chaining->chaining[i]);
if (node->sf_need_skip == 0 && node->sf_action == SESSION_ACTION_FORWARD)
{
cJSON *id = cJSON_CreateNumber(node->sf_profile_id);
cJSON_AddItemToArray(sf_profile_ids, id);
}
}
cJSON *params = cJSON_CreateObject();
cJSON_AddItemToObject(params, "sf_profile_ids", sf_profile_ids);
cJSON_AddItemToObject(root, "params", params);
char *json_str = cJSON_PrintUnformatted(root);
LOG_INFO("%s: session %lu %s event log: %s", LOG_TAG_METRICS, s_ctx->session_id, s_ctx->first_ctrl_pkt.addr_string, json_str);
marsio_buff_t *tx_buffs[1];
marsio_buff_malloc_device(packet_io->dev_nf_interface.mr_dev, tx_buffs, 1, 0, thread_seq);
char *dst = marsio_buff_append(tx_buffs[0], s_ctx->first_ctrl_pkt.header_len + strlen(json_str));
memcpy(dst, s_ctx->first_ctrl_pkt.header_data, s_ctx->first_ctrl_pkt.header_len);
memcpy(dst + s_ctx->first_ctrl_pkt.header_len, json_str, strlen(json_str));
struct metadata meta = {0};
meta.session_id = s_ctx->session_id;
meta.is_ctrl_pkt = 1;
meta.l7_offset = s_ctx->first_ctrl_pkt.header_len;
meta.sids.num = 1;
meta.sids.elems[0] = sce_ctx->firewall_sids;
packet_io_set_metadata(tx_buffs[0], &meta);
marsio_send_burst(packet_io->dev_nf_interface.mr_path, thread_seq, tx_buffs, 1);
free(json_str);
cJSON_Delete(root);
}
// return 0 : success
// return -1 : error
static int handle_session_closing(struct metadata *meta, struct ctrl_pkt_parser *parser, int thread_seq, void *ctx)
@@ -1187,7 +1257,8 @@ static int handle_session_closing(struct metadata *meta, struct ctrl_pkt_parser
LOG_INFO("%s: session %lu %s metrics log: policy %d sff_profile_id %d sf_profile_id %d sf_need_skip %d sf_action_reason %s rx_pkts %lu rx_bytes %lu tx_pkts %lu tx_bytes %lu", LOG_TAG_METRICS, s_ctx->session_id, s_ctx->first_ctrl_pkt.addr_string, node->policy_id, node->sff_profile_id, node->sf_profile_id, node->sf_need_skip, session_action_reason_to_string(node->sf_action_reason), node->rx.n_pkts, node->rx.n_bytes, node->tx.n_pkts, node->tx.n_bytes);
}
// TODO send log to firewall
send_event_log(s_ctx, thread_seq, ctx);
__atomic_fetch_add(&g_metrics->send_log, 1, __ATOMIC_RELAXED);
__atomic_fetch_sub(&g_metrics->session_nums, 1, __ATOMIC_RELAXED);
session_table_delete_by_id(thread->session_table, meta->session_id);