gfwleak/tango-tfe
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TFE适配marsio_dp_trace_measurements_can_emit()的变更

Browse Source
This commit is contained in:
luwenpeng
2024-04-26 17:37:22 +08:00
parent 0d17311373
commit 909f3b291d
2 changed files with 78 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
common/include/tfe_dp_trace.h
View File

@@ -8,18 +8,62 @@ extern "C"
#include <marsio.h>
#define DP_TRACE_INFO(mr_ins, mr_buff, module, fmt, ...) \
do \
{ \
if (marsio_dp_trace_measurements_can_emit(mr_ins, mr_buff)) \
{ \
marsio_dp_trace_measurement_emit_fmt(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TRACE, module, fmt, ##__VA_ARGS__); \
marsio_dp_trace_measurement_emit_fmt(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TELEMETRY, module, fmt, ##__VA_ARGS__); \
} \
} while (0)
static inline void record_msg_on_ctrl_pkt(char *buff, int size, uint64_t rule_id, uint64_t session_id, const char *state, const char *action, const char *reason)
{
int n = snprintf(buff, size, "intercept rule id=%lu, session=%lu, state=%s", rule_id, session_id, state);
if (action)
{
n += snprintf(buff + n, size - n, ", action=%s", action);
}
if (reason)
{
n += snprintf(buff + n, size - n, ", reason=%s", reason);
}
}
static inline void record_msg_on_raw_pkt(char *buff, int size, uint64_t rule_id, const char *traffic, const char *action, const char *reason)
{
int n = snprintf(buff, size, "intercept rule id=%lu, traffic=%s, action=%s", rule_id, traffic, action);
if (reason)
{
snprintf(buff + n, size - n, ", reason=%s", reason);
}
}
static inline void tfe_dp_telemetry_on_ctrl_pkt(mr_instance *mr_ins, marsio_buff_t *mr_buff, uint64_t rule_id, uint64_t session_id, const char *state, const char *action, const char *reason)
{
if (marsio_dp_trace_measurements_can_emit(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TRACE))
{
char buff[512] = {0};
record_msg_on_ctrl_pkt(buff, sizeof(buff), rule_id, session_id, state, action, reason);
marsio_dp_trace_measurement_emit_str(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TRACE, "Session Synchronization", buff);
}
if (marsio_dp_trace_measurements_can_emit(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TELEMETRY))
{
char buff[512] = {0};
record_msg_on_ctrl_pkt(buff, sizeof(buff), rule_id, session_id, state, action, reason);
marsio_dp_trace_measurement_emit_str(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TELEMETRY, "Session Synchronization", buff);
}
}
static inline void tfe_dp_on_raw_pkt(mr_instance *mr_ins, marsio_buff_t *mr_buff, uint64_t rule_id, const char *traffic, const char *action, const char *reason)
{
if (marsio_dp_trace_measurements_can_emit(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TRACE))
{
char buff[512] = {0};
record_msg_on_raw_pkt(buff, sizeof(buff), rule_id, traffic, action, reason);
marsio_dp_trace_measurement_emit_str(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TRACE, "Packet I/O", buff);
}
if (marsio_dp_trace_measurements_can_emit(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TELEMETRY))
{
char buff[512] = {0};
record_msg_on_raw_pkt(buff, sizeof(buff), rule_id, traffic, action, reason);
marsio_dp_trace_measurement_emit_str(mr_ins, mr_buff, DP_TRACE_MEASUREMENT_TYPE_TELEMETRY, "Packet I/O", buff);
}
}
#ifdef __cpluscplus
}
#endif
#endif
#endif