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

export the logging API symbol

Browse Source
This commit is contained in:
luwenpeng
2024-08-26 19:01:10 +08:00
parent 83eb9d9c5c
commit 11bf852c15
4 changed files with 114 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/core/version.map
View File

@@ -62,6 +62,8 @@ global:
stellar_free; stellar_free;
stellar_loopbreak; stellar_loopbreak;
stellar_get_logger; stellar_get_logger;
log_print;
log_check_level;
http_message_*; http_message_*;
http_decoder_init; http_decoder_init;

4
src/packet/packet_parser.cpp
View File

@@ -472,7 +472,7 @@ static inline const char *parse_ipv4(struct packet *pkt, const char *data, uint1
// ip fragmented // ip fragmented
if (ip4_hdr_get_mf_flag(hdr) || ip4_hdr_get_frag_offset(hdr)) if (ip4_hdr_get_mf_flag(hdr) || ip4_hdr_get_frag_offset(hdr))
{ {
PACKET_PARSE_LOG_WARN("packet %p ip layer %p is fragmented", pkt, layer); PACKET_PARSE_LOG_DEBUG("packet %p ip layer %p is fragmented", pkt, layer);
pkt->frag_layer = layer; pkt->frag_layer = layer;
return layer->pld_ptr; return layer->pld_ptr;
} }
@@ -569,7 +569,7 @@ static inline const char *parse_ipv6(struct packet *pkt, const char *data, uint1
// ipv6 fragment // ipv6 fragment
if (next_proto == IPPROTO_FRAGMENT) if (next_proto == IPPROTO_FRAGMENT)
{ {
PACKET_PARSE_LOG_WARN("packet %p ipv6 layer %p is fragmented", pkt, layer); PACKET_PARSE_LOG_DEBUG("packet %p ipv6 layer %p is fragmented", pkt, layer);
pkt->frag_layer = layer; pkt->frag_layer = layer;
return layer->pld_ptr; return layer->pld_ptr;
} }

3
src/packet_io/dumpfile_io.cpp
View File

@@ -29,6 +29,7 @@ struct dumpfile_io
char dumpfile_path[256]; char dumpfile_path[256];
pcap_t *pcap; pcap_t *pcap;
struct logger *logger;
struct packet_queue *queue[MAX_THREAD_NUM]; struct packet_queue *queue[MAX_THREAD_NUM];
struct packet_io_stat stat[MAX_THREAD_NUM]; struct packet_io_stat stat[MAX_THREAD_NUM];
uint64_t io_thread_need_exit; uint64_t io_thread_need_exit;
@@ -210,6 +211,7 @@ static int all_packet_processed(struct dumpfile_io *handle)
static void *dumpfile_thread(void *arg) static void *dumpfile_thread(void *arg)
{ {
struct dumpfile_io *handle = (struct dumpfile_io *)arg; struct dumpfile_io *handle = (struct dumpfile_io *)arg;
__thread_local_logger = handle->logger;
ATOMIC_SET(&handle->io_thread_is_runing, 1); ATOMIC_SET(&handle->io_thread_is_runing, 1);
PACKET_IO_LOG_FATAL("dumpfile io thread is running"); PACKET_IO_LOG_FATAL("dumpfile io thread is running");
@@ -292,6 +294,7 @@ struct dumpfile_io *dumpfile_io_new(const char *dumpfile_path, enum packet_io_mo
handle->mode = mode; handle->mode = mode;
handle->nr_threads = nr_threads; handle->nr_threads = nr_threads;
handle->logger = __thread_local_logger;
strncpy(handle->dumpfile_path, dumpfile_path, MIN(strlen(dumpfile_path), sizeof(handle->dumpfile_path))); strncpy(handle->dumpfile_path, dumpfile_path, MIN(strlen(dumpfile_path), sizeof(handle->dumpfile_path)));
for (uint16_t i = 0; i < handle->nr_threads; i++) for (uint16_t i = 0; i < handle->nr_threads; i++)

207
test/packet_inject/packet_inject.cpp
View File

@@ -6,19 +6,16 @@
#include <arpa/inet.h> #include <arpa/inet.h>
#include "toml.h" #include "toml.h"
#include "stellar/stellar.h" #include "stellar/log.h"
#include "stellar/packet.h" #include "stellar/packet.h"
#include "stellar/session.h" #include "stellar/session.h"
#include "stellar/stellar.h"
#include "stellar/stellar_mq.h" #include "stellar/stellar_mq.h"
#pragma GCC diagnostic ignored "-Wunused-parameter" #pragma GCC diagnostic ignored "-Wunused-parameter"
#define LOG_ERR(fmt, ...) printf("ERROR [packet inject] " fmt, ##__VA_ARGS__) #define INJTECT_PLUGIN_LOG_ERROR(logger, format, ...) STELLAR_LOG_ERROR(logger, "inject plugin", format, ##__VA_ARGS__)
#define LOG_INFO(fmt, ...) printf("INFO [packet inject] " fmt, ##__VA_ARGS__) #define INJTECT_PLUGIN_LOG_INFO(logger, format, ...) STELLAR_LOG_INFO(logger, "inject plugin", format, ##__VA_ARGS__)
/******************************************************************************
* Config
******************************************************************************/
enum inject_type enum inject_type
{ {
@@ -44,7 +41,43 @@ struct config
enum flow_direction direction; enum flow_direction direction;
}; };
static const char *inject_type_to_str(enum inject_type type) struct inject_plugin_ctx
{
struct config config;
struct stellar *st;
struct logger *logger;
int sess_plug_id;
int tcp_topic_id;
int udp_topic_id;
};
struct packet_exdata
{
enum flow_direction flow_dir;
union
{
struct in_addr v4;
struct in6_addr v6;
} src_addr, dst_addr;
uint16_t src_port; // host byte order
uint16_t dst_port; // host byte order
uint16_t tcp_payload_len;
uint32_t tcp_seq; // host byte order
uint32_t tcp_ack; // host byte order
uint8_t tcp_flags;
uint32_t inc_seq;
uint32_t inc_ack;
};
/******************************************************************************
* Config
******************************************************************************/
static const char *type_to_str(enum inject_type type)
{ {
switch (type) switch (type)
{ {
@@ -65,7 +98,7 @@ static const char *inject_type_to_str(enum inject_type type)
} }
} }
static int load_config(struct config *config, const char *file) static int load_config(struct logger *logger, struct config *config, const char *file)
{ {
int ret = -1; int ret = -1;
char errbuf[200]; char errbuf[200];
@@ -78,28 +111,28 @@ static int load_config(struct config *config, const char *file)
fp = fopen(file, "r"); fp = fopen(file, "r");
if (fp == NULL) if (fp == NULL)
{ {
LOG_ERR("open config file %s failed, %s\n", file, strerror(errno)); INJTECT_PLUGIN_LOG_ERROR(logger, "open config file %s failed, %s", file, strerror(errno));
goto error_out; goto error_out;
} }
root = toml_parse_file(fp, errbuf, sizeof(errbuf)); root = toml_parse_file(fp, errbuf, sizeof(errbuf));
if (root == NULL) if (root == NULL)
{ {
LOG_ERR("parse config file %s failed, %s\n", file, errbuf); INJTECT_PLUGIN_LOG_ERROR(logger, "parse config file %s failed, %s", file, errbuf);
goto error_out; goto error_out;
} }
sub = toml_table_in(root, "packet_inject"); sub = toml_table_in(root, "packet_inject");
if (sub == NULL) if (sub == NULL)
{ {
LOG_ERR("config file missing packet_inject section\n"); INJTECT_PLUGIN_LOG_ERROR(logger, "config file missing packet_inject section");
goto error_out; goto error_out;
} }
ptr = toml_raw_in(sub, "filter_ip"); ptr = toml_raw_in(sub, "filter_ip");
if (ptr == NULL) if (ptr == NULL)
{ {
LOG_ERR("config file missing packet_inject->filter_ip\n"); INJTECT_PLUGIN_LOG_ERROR(logger, "config file missing packet_inject->filter_ip");
goto error_out; goto error_out;
} }
if (strcmp(ptr, "any") == 0) if (strcmp(ptr, "any") == 0)
@@ -116,14 +149,14 @@ static int load_config(struct config *config, const char *file)
} }
else else
{ {
LOG_ERR("parse packet_inject->filter_ip failed, invalid ip address: %s\n", ptr); INJTECT_PLUGIN_LOG_ERROR(logger, "parse packet_inject->filter_ip failed, invalid ip address: %s", ptr);
goto error_out; goto error_out;
} }
ptr = toml_raw_in(sub, "filter_port"); ptr = toml_raw_in(sub, "filter_port");
if (ptr == NULL) if (ptr == NULL)
{ {
LOG_ERR("config file missing packet_inject->filter_port\n"); INJTECT_PLUGIN_LOG_ERROR(logger, "config file missing packet_inject->filter_port");
goto error_out; goto error_out;
} }
config->port = atoi(ptr); config->port = atoi(ptr);
@@ -131,7 +164,7 @@ static int load_config(struct config *config, const char *file)
ptr = toml_raw_in(sub, "filter_dir"); ptr = toml_raw_in(sub, "filter_dir");
if (ptr == NULL) if (ptr == NULL)
{ {
LOG_ERR("config file missing packet_inject->filter_dir\n"); INJTECT_PLUGIN_LOG_ERROR(logger, "config file missing packet_inject->filter_dir");
goto error_out; goto error_out;
} }
if (strcmp(ptr, "C2S") == 0) if (strcmp(ptr, "C2S") == 0)
@@ -144,27 +177,27 @@ static int load_config(struct config *config, const char *file)
} }
else else
{ {
LOG_ERR("parse packet_inject->filter_dir failed, invalid direction: %s\n", ptr); INJTECT_PLUGIN_LOG_ERROR(logger, "parse packet_inject->filter_dir failed, invalid direction: %s", ptr);
goto error_out; goto error_out;
} }
ptr = toml_raw_in(sub, "filter_pkts"); ptr = toml_raw_in(sub, "filter_pkts");
if (ptr == NULL) if (ptr == NULL)
{ {
LOG_ERR("config file missing packet_inject->filter_pkts\n"); INJTECT_PLUGIN_LOG_ERROR(logger, "config file missing packet_inject->filter_pkts");
goto error_out; goto error_out;
} }
config->number = atoi(ptr); config->number = atoi(ptr);
if (config->number == 0) if (config->number == 0)
{ {
LOG_ERR("parse packet_inject->filter_pkts failed, invalid number: %s\n", ptr); INJTECT_PLUGIN_LOG_ERROR(logger, "parse packet_inject->filter_pkts failed, invalid number: %s", ptr);
goto error_out; goto error_out;
} }
ptr = toml_raw_in(sub, "inject_type"); ptr = toml_raw_in(sub, "inject_type");
if (ptr == NULL) if (ptr == NULL)
{ {
LOG_ERR("config file missing packet_inject->inject_type\n"); INJTECT_PLUGIN_LOG_ERROR(logger, "config file missing packet_inject->inject_type");
goto error_out; goto error_out;
} }
if (strcmp(ptr, "TCP-RST") == 0) if (strcmp(ptr, "TCP-RST") == 0)
@@ -193,7 +226,7 @@ static int load_config(struct config *config, const char *file)
} }
else else
{ {
LOG_ERR("parse packet_inject->inject_type failed, invalid inject type: %s\n", ptr); INJTECT_PLUGIN_LOG_ERROR(logger, "parse packet_inject->inject_type failed, invalid inject type: %s", ptr);
goto error_out; goto error_out;
} }
@@ -213,7 +246,7 @@ error_out:
return ret; return ret;
} }
static void print_config(const struct config *config) static void print_config(struct logger *logger, const struct config *config)
{ {
char addr_str[INET6_ADDRSTRLEN] = {0}; char addr_str[INET6_ADDRSTRLEN] = {0};
@@ -230,39 +263,17 @@ static void print_config(const struct config *config)
break; break;
} }
LOG_INFO("config->filter_ip : %s\n", addr_str); INJTECT_PLUGIN_LOG_INFO(logger, "config->filter_ip : %s", addr_str);
LOG_INFO("config->filter_port : %d\n", config->port); INJTECT_PLUGIN_LOG_INFO(logger, "config->filter_port : %d", config->port);
LOG_INFO("config->filter_dir : %s\n", config->direction == FLOW_DIRECTION_C2S ? "C2S" : "S2C"); INJTECT_PLUGIN_LOG_INFO(logger, "config->filter_dir : %s", config->direction == FLOW_DIRECTION_C2S ? "C2S" : "S2C");
LOG_INFO("config->filter_pkts : %lu\n", config->number); INJTECT_PLUGIN_LOG_INFO(logger, "config->filter_pkts : %lu", config->number);
LOG_INFO("config->inject_type : %s\n", inject_type_to_str(config->type)); INJTECT_PLUGIN_LOG_INFO(logger, "config->inject_type : %s", type_to_str(config->type));
} }
/****************************************************************************** /******************************************************************************
* Utils * Utils
******************************************************************************/ ******************************************************************************/
struct packet_exdata
{
enum flow_direction flow_dir;
union
{
struct in_addr v4;
struct in6_addr v6;
} src_addr, dst_addr;
uint16_t src_port; // host byte order
uint16_t dst_port; // host byte order
uint16_t tcp_payload_len;
uint32_t tcp_seq; // host byte order
uint32_t tcp_ack; // host byte order
uint8_t tcp_flags;
uint32_t inc_seq;
uint32_t inc_ack;
};
static inline void packet_exdata_init(const struct packet *pkt, enum flow_direction dir, struct packet_exdata *pkt_exdata) static inline void packet_exdata_init(const struct packet *pkt, enum flow_direction dir, struct packet_exdata *pkt_exdata)
{ {
memset(pkt_exdata, 0, sizeof(struct packet_exdata)); memset(pkt_exdata, 0, sizeof(struct packet_exdata));
@@ -323,27 +334,27 @@ static inline uint32_t uint32_add(uint32_t seq, uint32_t inc)
return seq; return seq;
} }
static void craft_and_send_udp_packet(struct stellar *st, struct session *sess, struct packet_exdata *pkt_exdata, static void build_and_send_udp_packet(struct inject_plugin_ctx *ctx, struct session *sess, struct packet_exdata *pkt_exdata,
enum flow_direction inject_dir, const char *udp_payload, uint16_t udp_payload_len) enum flow_direction inject_dir, const char *udp_payload, uint16_t udp_payload_len)
{ {
const struct packet *origin_pkt = session_get_first_packet(sess, inject_dir); const struct packet *origin_pkt = session_get_first_packet(sess, inject_dir);
if (origin_pkt == NULL) if (origin_pkt == NULL)
{ {
LOG_ERR("craft UDP packet failed, %s origin packet is NULL\n", inject_dir == FLOW_DIRECTION_C2S ? "C2S" : "S2C"); INJTECT_PLUGIN_LOG_ERROR(ctx->logger, "build UDP packet failed, %s origin packet is NULL", inject_dir == FLOW_DIRECTION_C2S ? "C2S" : "S2C");
return; return;
} }
struct packet *craft_pkt = packet_build_udp(origin_pkt, udp_payload, udp_payload_len); struct packet *build_pkt = packet_build_udp(origin_pkt, udp_payload, udp_payload_len);
if (craft_pkt == NULL) if (build_pkt == NULL)
{ {
LOG_ERR("craft UDP packet failed\n"); INJTECT_PLUGIN_LOG_ERROR(ctx->logger, "build UDP packet failed");
return; return;
} }
stellar_send_build_packet(st, craft_pkt); stellar_send_build_packet(ctx->st, build_pkt);
} }
static void craft_and_send_tcp_packet(struct stellar *st, struct session *sess, struct packet_exdata *pkt_exdata, static void build_and_send_tcp_packet(struct inject_plugin_ctx *ctx, struct session *sess, struct packet_exdata *pkt_exdata,
enum flow_direction inject_dir, uint8_t tcp_flags, const char *tcp_payload, uint16_t tcp_payload_len) enum flow_direction inject_dir, uint8_t tcp_flags, const char *tcp_payload, uint16_t tcp_payload_len)
{ {
uint32_t tcp_seq = 0; uint32_t tcp_seq = 0;
@@ -390,44 +401,35 @@ static void craft_and_send_tcp_packet(struct stellar *st, struct session *sess,
const struct packet *origin_pkt = session_get_first_packet(sess, inject_dir); const struct packet *origin_pkt = session_get_first_packet(sess, inject_dir);
if (origin_pkt == NULL) if (origin_pkt == NULL)
{ {
LOG_ERR("craft TCP packet failed, %s origin packet is NULL\n", inject_dir == FLOW_DIRECTION_C2S ? "C2S" : "S2C"); INJTECT_PLUGIN_LOG_ERROR(ctx->logger, "build TCP packet failed, %s origin packet is NULL", inject_dir == FLOW_DIRECTION_C2S ? "C2S" : "S2C");
return; return;
} }
struct packet *craft_pkt = packet_build_tcp(origin_pkt, tcp_seq, tcp_ack, tcp_flags, NULL, 0, tcp_payload, tcp_payload_len); struct packet *build_pkt = packet_build_tcp(origin_pkt, tcp_seq, tcp_ack, tcp_flags, NULL, 0, tcp_payload, tcp_payload_len);
if (craft_pkt == NULL) if (build_pkt == NULL)
{ {
LOG_ERR("craft TCP packet failed\n"); INJTECT_PLUGIN_LOG_ERROR(ctx->logger, "build TCP packet failed");
return; return;
} }
stellar_send_build_packet(st, craft_pkt); stellar_send_build_packet(ctx->st, build_pkt);
} }
/****************************************************************************** /******************************************************************************
* Core logic * Core logic
******************************************************************************/ ******************************************************************************/
struct plugin_ctx
{
struct config config;
struct stellar *st;
int sess_plug_id;
int tcp_topic_id;
int udp_topic_id;
};
static void *on_sess_new(struct session *sess, void *plugin_ctx) static void *on_sess_new(struct session *sess, void *plugin_ctx)
{ {
// struct plugin_ctx *ctx = (struct plugin_ctx *)plugin_ctx; struct inject_plugin_ctx *ctx = (struct inject_plugin_ctx *)plugin_ctx;
LOG_INFO("handle session new: %s\n", session_get0_readable_addr(sess)); INJTECT_PLUGIN_LOG_INFO(ctx->logger, "handle session new: %s", session_get0_readable_addr(sess));
return NULL; return NULL;
} }
static void on_sess_free(struct session *sess, void *sess_ctx, void *plugin_ctx) static void on_sess_free(struct session *sess, void *sess_ctx, void *plugin_ctx)
{ {
// struct plugin_ctx *ctx = (struct plugin_ctx *)plugin_ctx; struct inject_plugin_ctx *ctx = (struct inject_plugin_ctx *)plugin_ctx;
LOG_INFO("handle session free: %s\n", session_get0_readable_addr(sess)); INJTECT_PLUGIN_LOG_INFO(ctx->logger, "handle session free: %s", session_get0_readable_addr(sess));
} }
static void on_sess_msg(struct session *sess, int topic_id, const void *msg, void *sess_ctx, void *plugin_ctx) static void on_sess_msg(struct session *sess, int topic_id, const void *msg, void *sess_ctx, void *plugin_ctx)
@@ -439,11 +441,10 @@ static void on_sess_msg(struct session *sess, int topic_id, const void *msg, voi
char buffer[1024] = {0}; char buffer[1024] = {0};
struct packet *pkt = (struct packet *)msg; struct packet *pkt = (struct packet *)msg;
struct plugin_ctx *ctx = (struct plugin_ctx *)plugin_ctx; struct inject_plugin_ctx *ctx = (struct inject_plugin_ctx *)plugin_ctx;
struct stellar *st = ctx->st;
struct config *config = &ctx->config; struct config *config = &ctx->config;
enum flow_direction flow_dir = session_get_current_flow_direction(sess); enum flow_direction flow_dir = session_get_current_flow_direction(sess);
LOG_INFO("handle session msg: %s (C2S received packets: %lu, S2C received packets: %lu)\n", INJTECT_PLUGIN_LOG_INFO(ctx->logger, "handle session msg: %s (C2S received packets: %lu, S2C received packets: %lu)",
session_get0_readable_addr(sess), session_get0_readable_addr(sess),
session_get_stat(sess, FLOW_DIRECTION_C2S, STAT_RAW_PACKETS_RECEIVED), session_get_stat(sess, FLOW_DIRECTION_C2S, STAT_RAW_PACKETS_RECEIVED),
session_get_stat(sess, FLOW_DIRECTION_S2C, STAT_RAW_PACKETS_RECEIVED)); session_get_stat(sess, FLOW_DIRECTION_S2C, STAT_RAW_PACKETS_RECEIVED));
@@ -491,36 +492,36 @@ static void on_sess_msg(struct session *sess, int topic_id, const void *msg, voi
switch (config->type) switch (config->type)
{ {
case INJECT_TCP_RST: case INJECT_TCP_RST:
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_RST | TH_ACK, NULL, 0); build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_RST | TH_ACK, NULL, 0);
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_RST | TH_ACK, NULL, 0); build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_RST | TH_ACK, NULL, 0);
session_set_discard(sess); session_set_discard(sess);
break; break;
case INJECT_TCP_FIN: case INJECT_TCP_FIN:
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_FIN | TH_ACK, NULL, 0); build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_FIN | TH_ACK, NULL, 0);
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_FIN | TH_ACK, NULL, 0); build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_FIN | TH_ACK, NULL, 0);
session_set_discard(sess); session_set_discard(sess);
break; break;
case INJECT_TCP_PAYLOAD: case INJECT_TCP_PAYLOAD:
snprintf(buffer, sizeof(buffer), "HTTP/1.1 200 OK\r\nContent-Length: %d\r\n\r\n%s", 5 + 5 + 2, "Hello"); snprintf(buffer, sizeof(buffer), "HTTP/1.1 200 OK\r\nContent-Length: %d\r\n\r\n%s", 5 + 5 + 2, "Hello");
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_ACK, buffer, strlen(buffer)); // inject payload to client build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_ACK, buffer, strlen(buffer)); // inject payload to client
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_ACK, "World\r\n", 7); // inject payload to client build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_ACK, "World\r\n", 7); // inject payload to client
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_RST | TH_ACK, NULL, 0); // inject RST to client build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_RST | TH_ACK, NULL, 0); // inject RST to client
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_RST | TH_ACK, NULL, 0); // inject RST to server build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_RST | TH_ACK, NULL, 0); // inject RST to server
session_set_discard(sess); session_set_discard(sess);
break; break;
case INJECT_TCP_PAYLOAD_FIN_RST: case INJECT_TCP_PAYLOAD_FIN_RST:
snprintf(buffer, sizeof(buffer), "HTTP/1.1 200 OK\r\nContent-Length: %d\r\n\r\n%s", 5 + 5 + 2, "Hello"); snprintf(buffer, sizeof(buffer), "HTTP/1.1 200 OK\r\nContent-Length: %d\r\n\r\n%s", 5 + 5 + 2, "Hello");
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_ACK, buffer, strlen(buffer)); // inject payload to client build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_ACK, buffer, strlen(buffer)); // inject payload to client
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_ACK, "World\r\n", 7); // inject payload to client build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_ACK, "World\r\n", 7); // inject payload to client
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_FIN | TH_ACK, NULL, 0); // inject FIN to client build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_FIN | TH_ACK, NULL, 0); // inject FIN to client
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_RST | TH_ACK, NULL, 0); // inject RST to client build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, TH_RST | TH_ACK, NULL, 0); // inject RST to client
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_FIN | TH_ACK, NULL, 0); // inject FIN to server build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_FIN | TH_ACK, NULL, 0); // inject FIN to server
craft_and_send_tcp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_RST | TH_ACK, NULL, 0); // inject RST to server build_and_send_tcp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_C2S, TH_RST | TH_ACK, NULL, 0); // inject RST to server
session_set_discard(sess); session_set_discard(sess);
break; break;
case INJECT_UDP_PAYLOAD: case INJECT_UDP_PAYLOAD:
craft_and_send_udp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_C2S, "Hello Server", 12); build_and_send_udp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_C2S, "Hello Server", 12);
craft_and_send_udp_packet(st, sess, &pkt_exdata, FLOW_DIRECTION_S2C, "Hello Client", 12); build_and_send_udp_packet(ctx, sess, &pkt_exdata, FLOW_DIRECTION_S2C, "Hello Client", 12);
session_set_discard(sess); session_set_discard(sess);
break; break;
case INJECT_CTRL_MSG: case INJECT_CTRL_MSG:
@@ -537,24 +538,26 @@ static void on_sess_msg(struct session *sess, int topic_id, const void *msg, voi
extern "C" extern "C"
{ {
void *packet_inject_init(struct stellar *st) void *packet_inject_init(struct stellar *st)
{ {
struct plugin_ctx *ctx = (struct plugin_ctx *)calloc(1, sizeof(struct plugin_ctx)); struct inject_plugin_ctx *ctx = (struct inject_plugin_ctx *)calloc(1, sizeof(struct inject_plugin_ctx));
if (ctx == NULL) if (ctx == NULL)
{ {
return NULL; return NULL;
} }
if (load_config(&ctx->config, "./plugin/inject.toml") == -1) ctx->st = st;
ctx->logger = stellar_get_logger(st);
INJTECT_PLUGIN_LOG_INFO(ctx->logger, "packet inject plugin init");
if (load_config(ctx->logger, &ctx->config, "./plugin/inject.toml") == -1)
{ {
LOG_ERR("load config failed\n"); INJTECT_PLUGIN_LOG_ERROR(ctx->logger, "load config failed");
free(ctx); free(ctx);
return NULL; return NULL;
} }
print_config(&ctx->config); print_config(ctx->logger, &ctx->config);
ctx->st = st;
ctx->sess_plug_id = stellar_session_plugin_register(st, on_sess_new, on_sess_free, ctx); ctx->sess_plug_id = stellar_session_plugin_register(st, on_sess_new, on_sess_free, ctx);
ctx->tcp_topic_id = stellar_mq_get_topic_id(st, TOPIC_TCP); ctx->tcp_topic_id = stellar_mq_get_topic_id(st, TOPIC_TCP);
ctx->udp_topic_id = stellar_mq_get_topic_id(st, TOPIC_UDP); ctx->udp_topic_id = stellar_mq_get_topic_id(st, TOPIC_UDP);
@@ -562,17 +565,15 @@ extern "C"
stellar_session_mq_subscribe(st, ctx->tcp_topic_id, on_sess_msg, ctx->sess_plug_id); stellar_session_mq_subscribe(st, ctx->tcp_topic_id, on_sess_msg, ctx->sess_plug_id);
stellar_session_mq_subscribe(st, ctx->udp_topic_id, on_sess_msg, ctx->sess_plug_id); stellar_session_mq_subscribe(st, ctx->udp_topic_id, on_sess_msg, ctx->sess_plug_id);
LOG_INFO("init\n");
return ctx; return ctx;
} }
void packet_inject_exit(void *plugin_ctx) void packet_inject_exit(void *plugin_ctx)
{ {
struct plugin_ctx *ctx = (struct plugin_ctx *)plugin_ctx; struct inject_plugin_ctx *ctx = (struct inject_plugin_ctx *)plugin_ctx;
if (ctx) if (ctx)
{ {
LOG_INFO("exit\n"); INJTECT_PLUGIN_LOG_INFO(ctx->logger, "packet inject plugin exit");
free(ctx); free(ctx);
} }
} }