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

✨ feat(plugin_manager_init): set max_msg_per_stage as parameter

Browse Source
This commit is contained in:
yangwei
2024-09-06 13:41:40 +08:00
parent cc542dc365
commit 9cb50f81fb
4 changed files with 30 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
infra/plugin_manager/test/plugin_manager_gtest_main.cpp
View File

@@ -49,7 +49,7 @@ void whitebox_test_plugin_manager_intrisic_metadata(struct stellar *st, struct p
TEST(plugin_manager_init, init_with_null_toml) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
plugin_manager_exit(plug_mgr);
}
@@ -64,7 +64,7 @@ static void test_mock_overwrite_packet_exdata_free(int idx, void *ex_ptr, void *
TEST(plugin_manager_init, packet_exdata_new_index_overwrite) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
const char *exdata_name="PACKET_EXDATA";
@@ -95,7 +95,7 @@ void test_mock_overwrite_packet_msg_free(void *msg, void *msg_free_arg){}
TEST(plugin_manager_init, stellar_mq_topic_create_and_update) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
const char *topic_name="PACKET_TOPIC";
@@ -159,7 +159,7 @@ void test_mock_overwrite_on_packet_msg(int topic_id, const void *msg, void *plug
TEST(plugin_manager_init, stellar_mq_subscribe) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
@@ -198,7 +198,7 @@ static void test_mock_overwrite_session_exdata_free(int idx, void *ex_ptr, void
TEST(plugin_manager_init, session_exdata_new_index_overwrite) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
const char *exdata_name="SESSION_EXDATA";
@@ -232,7 +232,7 @@ void test_mock_overwrite_on_session_msg(int topic_id, const void *msg, void *plu
TEST(plugin_manager_init, stellar_mq_subscribe_overwrite) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
const char *topic_name="SESSION_TOPIC";
@@ -276,7 +276,7 @@ int test_plugin_on_polling_func(void *plugin_env)
TEST(plugin_manager_init, polling_plugin_register) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
st.plug_mgr=plug_mgr;
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
@@ -334,7 +334,7 @@ static void test_basic_on_packet(struct packet *pkt, void *plugin_env)
TEST(plugin_manager, packet_plugin_illegal_exdata) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
unsigned char ip_proto=6;
@@ -374,7 +374,7 @@ static void test_proto_filter_on_packet(struct packet *pkt, void *plugin_env)
TEST(plugin_manager, DISABLED_packet_plugins_with_proto_filter) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
struct packet_plugin_env env;
@@ -481,7 +481,7 @@ static void test_packet_exdata_free(int idx, void *ex_ptr, void *arg)
TEST(plugin_manager, packet_plugins_share_exdata) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
unsigned char ip_proto=6;
@@ -576,7 +576,7 @@ static void test_mq_pub_on_packet(struct packet *pkt, void *plugin_env)
TEST(plugin_manager, packet_plugins_mq_pub_sub) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
unsigned char ip_proto=6;
@@ -665,16 +665,16 @@ static void overlimit_pub_on_packet(struct packet *pkt, void *plugin_env)
EXPECT_TRUE(env!=NULL);
//EXPECT_EQ(pkt->ip_proto, ip_protocol);
int topic_id_num=(int)(sizeof(env->packet_topic_id) / sizeof(env->packet_topic_id[0]));
int cnt=0;
unsigned int cnt=0;
int *msg;
for(int i=0; i<topic_id_num; i++)
{
for(int j=0; j < MAX_MSG_PER_DISPATCH; j++)
for(unsigned int j=0; j < env->plug_mgr->max_message_dispatch; j++)
{
msg=CALLOC(int, 1);
*msg=cnt;
int pub_ret=stellar_mq_publish_message(env->plug_mgr->st, env->packet_topic_id[i], msg);
if(cnt < MAX_MSG_PER_DISPATCH)
if(cnt < env->plug_mgr->max_message_dispatch)
{
ASSERT_EQ(pub_ret, 0);
env->msg_pub_cnt+=1;
@@ -693,7 +693,7 @@ static void overlimit_pub_on_packet(struct packet *pkt, void *plugin_env)
TEST(plugin_manager, packet_plugins_pub_overlimit) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
unsigned char ip_proto=6;
@@ -755,7 +755,7 @@ TEST(plugin_manager, packet_plugins_pub_overlimit) {
}
plugin_manager_exit(plug_mgr);
EXPECT_EQ(N_packet*MAX_MSG_PER_DISPATCH, env.msg_pub_cnt);
EXPECT_EQ(N_packet*MAX_MSG_PER_STAGE, env.msg_pub_cnt);
EXPECT_EQ(env.msg_free_cnt, env.msg_pub_cnt);
EXPECT_EQ(env.msg_sub_cnt, env.msg_pub_cnt*topic_sub_num);
}
@@ -801,7 +801,7 @@ static void test_exdata_free_pub_msg_on_packet_msg(int topic_id, const void *msg
TEST(plugin_manager, packet_plugin_exdata_free_pub_msg) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
unsigned char ip_proto=6;
@@ -937,7 +937,7 @@ TEST(plugin_manager, no_plugin_register_runtime) {
struct stellar st={0};
// init stage
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
@@ -996,7 +996,7 @@ static void test_basic_session_exdata_free(int idx, void *ex_ptr, void *arg)
TEST(plugin_manager, session_plugin_on_tcp) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL,MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
unsigned char ip_proto=6;
@@ -1133,7 +1133,7 @@ static void test_overlimit_session_msg_free(void *msg, void *msg_free_arg)
TEST(plugin_manager,DISABLED_session_plugin_pub_msg_overlimt) {
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
unsigned char ip_proto=6;
@@ -1258,7 +1258,7 @@ TEST(plugin_manager, DISABLED_session_plugin_pub_msg_on_closing) {
memset(&env, 0, sizeof(struct session_manager_plugin_env));
// pesudo init stage
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
// plugin manager register plugin
@@ -1403,7 +1403,7 @@ TEST(plugin_manager, test_session_mq_priority) {
memset(&env, 0, sizeof(struct session_plugin_env));
// pesudo init stage
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
// plugin manager register plugin
@@ -1504,7 +1504,7 @@ TEST(plugin_manager, session_exdata_free_pub_msg) {
struct session_manager_plugin_env env;
// pesudo init stage
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
whitebox_test_plugin_manager_intrisic_metadata(&st, plug_mgr);
// plugin manager register plugin
@@ -1596,7 +1596,7 @@ TEST(plugin_manager, basic_polling_plugins) {
// pesudo init stage
struct stellar st={0};
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, NULL, MAX_MSG_PER_STAGE);
st.plug_mgr=plug_mgr;
struct polling_plugin_env env;
memset(&env, 0, sizeof(struct polling_plugin_env));