summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorniubinghui <[email protected]>2024-08-19 14:53:42 +0800
committerniubinghui <[email protected]>2024-08-19 14:53:42 +0800
commit4898a47663a838a490ec6ff282cf72076745537d (patch)
treef19e0eb011da8b1c78f26d7089508a0b2d3a1187
parentafe428938de4a3d4f10283147897e14d45b8c5d2 (diff)
【修改】修改example,从pluginmanage的样例中摘出其中一部分
-rw-r--r--example/Makefile16
-rw-r--r--example/example_plugin_manage.c44
-rw-r--r--example/plugin/example_plugin-2.lua29
-rw-r--r--example/simple_example_plugin.c457
-rw-r--r--include/lua_plugin_manage.h4
5 files changed, 539 insertions, 11 deletions
diff --git a/example/Makefile b/example/Makefile
index 2582e10..4fffc2b 100644
--- a/example/Makefile
+++ b/example/Makefile
@@ -2,25 +2,35 @@ TOPDIR = ./..
CC=gcc
MAKE=make
TARGET=example
+SIMPLE=simple_stellar_plugin.so
EXAMPLE_FLAG = -DLUAPLUGIN_EXAMPLE
SRC := example_plugin_manage.c
-
OBJECTS := example_plugin_manage.o
+SIMPLE_SRC := simple_example_plugin.c
+SIMPLE_OBJECTS := simple_example_plugin.o
+
INCLUDE = -I$(TOPDIR)/output/include -I$(TOPDIR)/dependence/include -I$(TOPDIR)/example/include
CFLAGS = -g -Wextra -Wall -O0 -fPIC
# CFLAGS += -pedantic -fsanitize=address
# LDLIBS = -L$(TOPDIR)/output/lib -llua -ldl -lm
LDLIBS += -L$(TOPDIR)/output/libs -lluaplugin -L$(TOPDIR)/dependence/lib -ltoml -lbitmap -lplugin_manager
-all:$(OBJECTS)
+TARGET:$(OBJECTS)
$(CC) $(CFLAGS) -o $(TARGET) $(OBJECTS) $(LDLIBS)
$(OBJECTS):$(SRC)
$(CC) $(TEST_FLAG) $(INCLUDE) $(CFLAGS) $(SRC) -c $^
+SIMPLE:$(SIMPLE_OBJECTS)
+ $(CC) $(CFLAGS) --shared -o $(SIMPLE) $(SIMPLE_OBJECTS) $(LDLIBS)
+ cp $(SIMPLE) plugin
+
+$(SIMPLE_OBJECTS):$(SIMPLE_SRC)
+ $(CC) $(TEST_FLAG) $(INCLUDE) $(CFLAGS) $(SIMPLE_SRC) -c $^
+
clean:
- rm -rf $(OBJECTS) $(TARGET)
+ rm -rf $(OBJECTS) $(TARGET) $(SIMPLE_OBJECTS) $(SIMPLE)
rm -rf $(TOPDIR)/output/libs/$(TARGET) \ No newline at end of file
diff --git a/example/example_plugin_manage.c b/example/example_plugin_manage.c
index 3e8667e..0e560d3 100644
--- a/example/example_plugin_manage.c
+++ b/example/example_plugin_manage.c
@@ -11,8 +11,15 @@
#define PLUGIN_CONFIG_PATH "./conf/plugin_manage.toml"
#define LUA_CONFIG_PATH "./conf/lua_plugin_manage.toml"
+// #define DEBUG_PLUGIN_SCHEMA
+
static struct lua_config_specific *config_load(const char *config_file_name, int *specific_num);
+#ifdef DEBUG_PLUGIN_SCHEMA
static void debug_plugin_manage_schema(struct plugin_manager_schema *schema);
+#endif
+
+#define PACKET_COUNT 10
+#define SESSION_COUNT 10
int main()
{
@@ -22,10 +29,45 @@ int main()
int num = 0;
struct lua_config_specific *specific = config_load(LUA_CONFIG_PATH, &num);
struct plugin_manager_schema *plug_mgr = plugin_manager_init(&st, PLUGIN_CONFIG_PATH);
+ /* 初始化lua插件 */
struct lua_plugin_manage_schema *lua_schema = lua_plugin_manage_init(&st, num, specific);
st.lua_plug_mgr = lua_schema;
+#ifdef DEBUG_PLUGIN_SCHEMA
debug_plugin_manage_schema(plug_mgr);
+#endif
+
+ unsigned char ip_proto=6;
+ struct packet pkt={&st, TCP, ip_proto};
+ struct session sess[SESSION_COUNT];
+ memset(&sess, 0, sizeof(sess));
+
+ for(int i=0; i < SESSION_COUNT; i++)
+ {
+ sess[i].plug_mgr_rt=plugin_manager_session_runtime_new(plug_mgr, &sess[i]);
+ sess[i].type=SESSION_TYPE_TCP;
+ }
+
+ for (int j = 0; j < PACKET_COUNT; j++)
+ {
+ // plugin_manager_on_packet_ingress(plug_mgr, &pkt);
+
+ for (int i = 0; i < SESSION_COUNT; i++)
+ {
+ sess[i].sess_pkt_cnt+=1;
+ plugin_manager_on_session_ingress(&sess[i], &pkt);
+ plugin_manager_on_session_egress(&sess[i], &pkt);
+ }
+
+ // plugin_manager_on_packet_egress(plug_mgr, &pkt);
+ }
+
+ for(int i=0; i < SESSION_COUNT; i++)
+ {
+ plugin_manager_on_session_closing(&sess[i]);
+ plugin_manager_session_runtime_free(sess[i].plug_mgr_rt);
+ }
+ plugin_manager_exit(plug_mgr);
lua_plugin_manage_exit(lua_schema);
return 0;
}
@@ -83,6 +125,7 @@ static struct lua_config_specific *config_load(const char *config_file_name, int
return new_spec;
}
+#ifdef DEBUG_PLUGIN_SCHEMA
static void debug_plugin_manage_schema(struct plugin_manager_schema *schema)
{
struct registered_session_plugin_schema * plugin = NULL;
@@ -92,4 +135,5 @@ static void debug_plugin_manage_schema(struct plugin_manager_schema *schema)
}
return;
}
+#endif
diff --git a/example/plugin/example_plugin-2.lua b/example/plugin/example_plugin-2.lua
index faffe5b..62c13ed 100644
--- a/example/plugin/example_plugin-2.lua
+++ b/example/plugin/example_plugin-2.lua
@@ -1,16 +1,24 @@
function plugin_ctx_new(sess, plug_env, sess_context)
print("now create new ctx example-2, plugin id ", plug_env.id)
- local sessid = session.getid(sess)
+ local sesstype = session.gettype(sess)
sess_context.id = 200
- print("session id is ", sessid)
- session.setid(sess, 50000)
+ sess_context.called = 0
+ print("session type is ", sesstype)
+ -- session.setid(sess, 50000)
end
function plugin_ctx_free(sess, sess_context, plug_env)
- print(sess_context.id)
+ print(plug_env.id, sess_context.id, sess_context.called, plug_env.called)
print("now begin to free ctx context example-2")
end
+function on_message(sess, topic_id, msg, sess_context, env)
+ -- print(sess, topic_id, msg, sess_context, env)
+ sess_context.called = sess_context.called + 1
+ env.calledcount = env.calledcount + 1
+ print("call on message", env.id, env.calledcount, sess_context.called)
+end
+
function free_message()
print("free message")
end
@@ -18,8 +26,17 @@ end
function plugin_load(stellar, plug_env)
print("now begin to load plugin example-2")
plug_env.id = plugin_manage.register(stellar, plugin_ctx_new, plugin_ctx_free, plug_env)
- topic_id = message.gettopicid(stellar, "TOPIC_SESSION_STAT")
+ topic_id = message.gettopicid(stellar, "TCP")
print("get topic id is ", topic_id)
+ message.subscribetopic(stellar, topic_id, on_message, plug_env.id)
+ -- print("subscribetopic result", bool)
+ -- mq_topic_id = message.gettopicid(stellar, "SESSION_MQ_TOPIC")
+ -- print("get session mq topic is", mq_topic_id)
+ -- message.subscribetopic(stellar, mq_topic_id, on_message, plug_env.id)
+
+ plug_env.calledcount = 0
+
+ --[[
create_id = message.gettopicid(stellar, "TOPIC_LUA_SESSION_TEST")
if (create_id < 0)
then
@@ -30,9 +47,9 @@ function plugin_load(stellar, plug_env)
else
print("has created, id is ", create_id)
end
+ --]]
end
function plugin_unload(plug_env)
print("now running unload plugin example-2 function, plugin id is ", plug_env.id)
-
end \ No newline at end of file
diff --git a/example/simple_example_plugin.c b/example/simple_example_plugin.c
new file mode 100644
index 0000000..a0aaace
--- /dev/null
+++ b/example/simple_example_plugin.c
@@ -0,0 +1,457 @@
+#pragma GCC diagnostic ignored "-Wunused-parameter"
+
+#include "stellar/stellar.h"
+#include "stellar/session.h"
+#include "stellar/utils.h"
+#include "stellar/session_exdata.h"
+#include "stellar/session_mq.h"
+#include "stellar/packet_exdata.h"
+#include "stellar/packet_mq.h"
+
+#include <stdio.h>
+#include <string.h>
+#include <assert.h>
+
+
+struct simple_stellar_plugin_env
+{
+ struct stellar *st;
+ int session_plugin_id;
+ int session_exdata_idx;
+ int packet_exdata_idx;
+ int packet_topic_id;
+ int stat_topic_id;
+ int egress_topic_id;
+ int tcp_topic_id;
+ int udp_topic_id;
+ int tcp_stream_topic_id;
+ long long tcp_packet_count;
+ long long udp_packet_count;
+ long long icmp_packet_count;
+ long long icmp6_packet_count;
+ long long polling_times;
+};
+
+struct mq_message_stat
+{
+ uint32_t c2s_pkts;
+ uint32_t c2s_bytes;
+ uint32_t s2c_pkts;
+ uint32_t s2c_bytes;
+ uint32_t c2s_stream_pkts;
+ uint32_t c2s_stream_bytes;
+ uint32_t s2c_stream_pkts;
+ uint32_t s2c_stream_bytes;
+};
+
+
+static void print_session_stat(struct session *sess, struct mq_message_stat *stat, int plugin_id, const char *banner)
+{
+ if (stat)
+ {
+ printf("%s(plug:%d)-----------%20s: ", banner, plugin_id, session_get0_readable_addr(sess));
+ printf("server-pkt=%u, server-count=%u, client-pkt=%u, client-count=%u, ", stat->c2s_pkts, stat->c2s_bytes,
+ stat->s2c_pkts, stat->s2c_bytes);
+ printf("total-pkt=%u, ", stat->c2s_pkts + stat->s2c_pkts);
+ printf("total-count=%u\n", stat->c2s_bytes + stat->s2c_bytes);
+ if(stat->c2s_stream_pkts+stat->s2c_stream_pkts > 0)
+ {
+ printf("----------------server-stream_pkt=%u, server-stream_count=%u, client-stream_pkt=%u, client-stream_count=%u\n", stat->c2s_stream_pkts, stat->c2s_stream_bytes,
+ stat->s2c_stream_pkts, stat->s2c_stream_bytes);
+ }
+ }
+ return;
+}
+
+/*******************************
+ * simple session & packet plugin *
+ *******************************/
+
+static void *simple_plugin_session_ctx_new(struct session *sess, void *plugin_env)
+{
+ struct mq_message_stat * stat= CALLOC(struct mq_message_stat, 1);
+ struct simple_stellar_plugin_env *env=(struct simple_stellar_plugin_env *)plugin_env;
+ session_exdata_set(sess, env->session_exdata_idx, stat);
+ return stat;
+}
+
+static void simple_plugin_session_ctx_free(struct session *sess, void *session_ctx, void *plugin_env)
+{
+ struct simple_stellar_plugin_env *env=(struct simple_stellar_plugin_env *)plugin_env;
+ struct mq_message_stat *stat = (struct mq_message_stat *)session_ctx;
+ print_session_stat(sess, stat, env->session_plugin_id, __FUNCTION__);
+ session_exdata_set(sess, env->session_exdata_idx, NULL);
+ if(session_ctx)FREE(session_ctx);
+ return;
+}
+
+static void simple_plugin_on_session_msg(struct session *sess, int topic_id, const void *data, void *plugin_ctx, void *plugin_env)
+{
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)plugin_env;
+ struct mq_message_stat *stat = (struct mq_message_stat *)plugin_ctx;
+ struct packet * pkt=(struct packet *)data;
+ if (pkt)
+ {
+ // TEST: try stellar_session_plugin_dettach_current_session when pkt > 3
+ if(stat->c2s_pkts+stat->s2c_pkts >= 3 && session_get_type(sess)== SESSION_TYPE_UDP)
+ {
+ stellar_session_plugin_dettach_current_session(sess);
+ return;
+ }
+
+ size_t payload_len = 0;
+ session_get0_current_payload(sess, &payload_len);
+ int dir = packet_get_direction(pkt);
+ if (dir==PACKET_DIRECTION_C2S)
+ {
+ if(topic_id==env->tcp_stream_topic_id)
+ {
+ stat->c2s_stream_bytes += payload_len;
+ stat->c2s_stream_pkts += 1;
+ }
+ else
+ {
+ stat->c2s_bytes += payload_len;
+ stat->c2s_pkts += 1;
+ }
+ }
+ if (dir==PACKET_DIRECTION_S2C)
+ {
+ if(topic_id==env->tcp_stream_topic_id)
+ {
+ stat->s2c_stream_bytes += payload_len;
+ stat->s2c_stream_pkts += 1;
+ }
+ else
+ {
+ stat->s2c_bytes += payload_len;
+ stat->s2c_pkts += 1;
+ }
+ }
+ session_mq_publish_message(sess, ((struct simple_stellar_plugin_env *)plugin_env)->stat_topic_id, stat);
+ }
+ return;
+}
+
+void simple_plugin_on_packet(struct packet *pkt, unsigned char ip_protocol, void *plugin_env)
+{
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)plugin_env;
+ packet_exdata_set(pkt, env->packet_exdata_idx, env);
+ packet_mq_publish_message(pkt, env->packet_topic_id, env);
+ switch (ip_protocol)
+ {
+ case IPPROTO_TCP:
+ env->tcp_packet_count++;
+ break;
+ case IPPROTO_UDP:
+ env->udp_packet_count++;
+ break;
+ case IPPROTO_ICMP:
+ env->icmp_packet_count++;
+ break;
+ case IPPROTO_ICMPV6:
+ env->icmp6_packet_count++;
+ break;
+ default:
+ perror("invalid ip_protocol\n");
+ exit(-1);
+ break;
+ }
+ return;
+}
+
+void simple_plugin_packet_get_exdata(struct packet *pkt, unsigned char ip_protocol , void *plugin_env)
+{
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)plugin_env;
+ struct simple_stellar_plugin_env *exdata = (struct simple_stellar_plugin_env *)packet_exdata_get(pkt, env->packet_exdata_idx);
+ if(memcmp(env, exdata, sizeof(struct simple_stellar_plugin_env)) != 0)
+ {
+ abort();
+ }
+ return;
+}
+
+
+int simple_plugin_on_polling(void *plugin_env)
+{
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)plugin_env;
+ env->polling_times++;
+ return 0;
+}
+
+static void simple_plugin_packet_exdata_free(struct packet *pkt , int idx , void *ex_ptr, void *arg)
+{
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)arg;
+ assert(env);
+ struct simple_stellar_plugin_env *exdata = (struct simple_stellar_plugin_env *)ex_ptr;
+ if(memcmp(env, exdata, sizeof(struct simple_stellar_plugin_env)) != 0)
+ {
+ abort();
+ }
+}
+
+static void simple_plugin_packet_msg_free(struct packet *pkt , void *msg, void *msg_free_arg)
+{
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)msg_free_arg;
+ assert(env);
+ struct simple_stellar_plugin_env *exdata = (struct simple_stellar_plugin_env *)msg;
+ if(memcmp(env, exdata, sizeof(struct simple_stellar_plugin_env)) != 0)
+ {
+ abort();
+ }
+
+}
+
+static void simple_plugin_on_packet_msg_cb(struct packet *pkt , int topic_id , const void *msg, void *plugin_env)
+{
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)plugin_env;
+ assert(env);
+ struct simple_stellar_plugin_env *exdata = (struct simple_stellar_plugin_env *)msg;
+ if(memcmp(env, exdata, sizeof(struct simple_stellar_plugin_env)) != 0)
+ {
+ abort();
+ }
+}
+
+void *simple_session_packet_plugin_init(struct stellar *st)
+{
+ struct simple_stellar_plugin_env *env = CALLOC(struct simple_stellar_plugin_env, 1);
+ env->st = st;
+ env->session_exdata_idx = stellar_session_exdata_new_index(st, "EXDATA_SESSION_STAT", NULL, NULL);
+ env->session_plugin_id = stellar_session_plugin_register(st,
+ simple_plugin_session_ctx_new,
+ simple_plugin_session_ctx_free,
+ env);
+ int tcp_plugin_id=stellar_packet_plugin_register(st, IPPROTO_TCP, simple_plugin_on_packet, env);
+ int udp_plugin_id=stellar_packet_plugin_register(st, IPPROTO_UDP, simple_plugin_on_packet, env);
+ int icmp_plugin_id=stellar_packet_plugin_register(st, IPPROTO_ICMP, simple_plugin_on_packet, env);
+ int icmp6_plugin_id=stellar_packet_plugin_register(st, IPPROTO_ICMPV6, simple_plugin_on_packet, env);
+
+ if(tcp_plugin_id < 0 || udp_plugin_id < 0 || icmp_plugin_id < 0 || icmp6_plugin_id < 0)
+ {
+ perror("register packet plugin return invalid plugin id\n");
+ exit(-1);
+ }
+
+ env->packet_topic_id=stellar_packet_mq_get_topic_id(st, "TOPIC_PACKET_ENV");
+ if(env->packet_topic_id < 0)
+ {
+ env->packet_topic_id=stellar_packet_mq_create_topic(st, "TOPIC_PACKET_ENV", simple_plugin_packet_msg_free, env);
+ }
+
+ tcp_plugin_id=stellar_packet_plugin_register(st, IPPROTO_TCP, simple_plugin_packet_get_exdata, env);
+ udp_plugin_id=stellar_packet_plugin_register(st, IPPROTO_UDP, simple_plugin_packet_get_exdata, env);
+ icmp_plugin_id=stellar_packet_plugin_register(st, IPPROTO_ICMP, simple_plugin_packet_get_exdata, env);
+ icmp6_plugin_id=stellar_packet_plugin_register(st, IPPROTO_ICMPV6, simple_plugin_packet_get_exdata, env);
+
+ if(tcp_plugin_id < 0 || udp_plugin_id < 0 || icmp_plugin_id < 0 || icmp6_plugin_id < 0)
+ {
+ perror("register packet plugin get exdata return invalid plugin id\n");
+ exit(-1);
+ }
+
+ stellar_packet_mq_subscribe(st, env->packet_topic_id, simple_plugin_on_packet_msg_cb, tcp_plugin_id);
+ stellar_packet_mq_subscribe(st, env->packet_topic_id, simple_plugin_on_packet_msg_cb, udp_plugin_id);
+ stellar_packet_mq_subscribe(st, env->packet_topic_id, simple_plugin_on_packet_msg_cb, icmp_plugin_id);
+ stellar_packet_mq_subscribe(st, env->packet_topic_id, simple_plugin_on_packet_msg_cb, icmp6_plugin_id);
+
+ env->packet_exdata_idx=stellar_packet_exdata_new_index(st, "EXDATA_PACKET_ENV", simple_plugin_packet_exdata_free, env);
+
+ int polling_plugin_id=stellar_polling_plugin_register(st, simple_plugin_on_polling, env);
+ if(polling_plugin_id < 0)
+ {
+ perror("register polling plugin return invalid plugin id \n");
+ exit(-1);
+ }
+
+ env->tcp_stream_topic_id=stellar_session_mq_get_topic_id(st, TOPIC_TCP_STREAM);
+ env->tcp_topic_id=stellar_session_mq_get_topic_id(st, TOPIC_TCP);
+ env->udp_topic_id=stellar_session_mq_get_topic_id(st, TOPIC_UDP);
+ if(env->tcp_topic_id < 0 || env->udp_topic_id < 0 || env->tcp_stream_topic_id < 0)
+ {
+ perror("get tcp or udp topic id failed\n");
+ exit(-1);
+ }
+
+ stellar_session_mq_subscribe(st, env->tcp_stream_topic_id, simple_plugin_on_session_msg, env->session_plugin_id);
+ stellar_session_mq_subscribe(st, env->tcp_topic_id, simple_plugin_on_session_msg, env->session_plugin_id);
+ stellar_session_mq_subscribe(st, env->udp_topic_id, simple_plugin_on_session_msg, env->session_plugin_id);
+
+ int stat_topic_id=stellar_session_mq_get_topic_id(st, "TOPIC_SESSION_STAT");
+ if(stat_topic_id < 0)
+ {
+ stat_topic_id=stellar_session_mq_create_topic(st, "TOPIC_SESSION_STAT", NULL, NULL);
+ }
+ env->stat_topic_id = stat_topic_id;
+ return env;
+}
+
+void simple_session_packet_plugin_exit(void *plugin_env)
+{
+ if(plugin_env)
+ {
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)plugin_env;
+ printf("(%s):tcp_packet_num:%lld, udp_packet_num:%lld, icmp_packet_num:%lld, icmp6_packet_num:%lld, polling_times:%lld\n", __FUNCTION__, env->tcp_packet_count, env->udp_packet_count, env->icmp_packet_count, env->icmp6_packet_count, env->polling_times);
+ FREE(plugin_env);
+ }
+ return;
+}
+
+/*******************************
+ * simple plugin sub session stat *
+ *******************************/
+
+static void simple_plugin_sub_session_stat_on_msg(struct session *sess, int topic_id, const void *data, void *plugin_ctx , void *plugin_env)
+{
+
+ struct simple_stellar_plugin_env *env = (struct simple_stellar_plugin_env *)plugin_env;
+ if(topic_id == env->egress_topic_id)
+ {
+ session_mq_ignore_message(sess, topic_id, env->session_plugin_id);
+ }
+ if (topic_id == env->stat_topic_id)
+ {
+ struct mq_message_stat *stat = (struct mq_message_stat *)data;
+ struct mq_message_stat *exdata_stat = (struct mq_message_stat *)session_exdata_get(sess, env->session_exdata_idx);
+ if (memcmp(exdata_stat, stat, sizeof(struct mq_message_stat)) != 0)
+ {
+ perror("exdata and mq data not equal\n");
+ exit(-1);
+ }
+ // print_session_stat(sess, stat, env->plugin_id, __FUNCTION__);
+ session_mq_unignore_message(sess, env->egress_topic_id, env->session_plugin_id);
+ }
+ return;
+}
+
+void *simple_plugin_sub_session_stat_init(struct stellar *st)
+{
+ printf("init new plugin, regist\n");
+ struct simple_stellar_plugin_env *env = CALLOC(struct simple_stellar_plugin_env, 1);
+ env->st = st;
+ env->session_exdata_idx = stellar_session_exdata_new_index(st, "EXDATA_SESSION_STAT", NULL, NULL);
+ int topic_id=stellar_session_mq_get_topic_id(st, "TOPIC_SESSION_STAT");
+ if(topic_id < 0)
+ {
+ topic_id=stellar_session_mq_create_topic(st, "TOPIC_SESSION_STAT", NULL, NULL);
+ }
+ env->stat_topic_id = topic_id;
+ env->session_plugin_id = stellar_session_plugin_register(st,
+ NULL,
+ NULL,
+ env);
+ stellar_session_mq_subscribe(st, topic_id, simple_plugin_sub_session_stat_on_msg, env->session_plugin_id);
+
+ // TEST: subscirbe egress message then ignore
+ env->egress_topic_id=stellar_session_mq_get_topic_id(st, TOPIC_EGRESS);
+ if(env->egress_topic_id < 0)
+ {
+ perror("get egress topic id failed\n");
+ exit(-1);
+ }
+ stellar_session_mq_subscribe(st, env->egress_topic_id, simple_plugin_sub_session_stat_on_msg, env->session_plugin_id);
+ return env;
+}
+
+void simple_plugin_sub_session_stat_exit(void *plugin_env)
+{
+ if(plugin_env)FREE(plugin_env);
+ return;
+}
+
+
+
+
+struct session_plugin_env
+{
+ struct plugin_manager_schema *plug_mgr;
+ int N_session;
+ int N_per_session_pkt_cnt;
+ int intrinsc_tcp_topic_id;
+ int intrinsc_egress_topic_id;
+ int basic_exdata_idx;
+ int basic_exdata_free_called;
+ int basic_on_session_ingress_called;
+ int basic_on_session_egress_called;
+ int basic_ctx_new_called;
+ int basic_ctx_free_called;
+ int test_mq_pub_plugin_id;
+ int test_mq_sub_plugin_id;
+ int test_mq_pub_called;
+ int test_mq_sub_called;
+ int test_mq_free_called;
+ int test_mq_topic_id;
+ int plugin_id_1;
+ int plugin_id_2;
+ int plugin_id_1_called;
+ int plugin_id_2_called;
+};
+
+struct test_basic_ctx
+{
+ int called;
+};
+
+static void *test_dettach_session_ctx_new(struct session *sess, void *plugin_env)
+{
+ printf("now running ctx new\n");
+ struct test_basic_ctx *ctx=(struct test_basic_ctx *)calloc(1, sizeof(struct test_basic_ctx));
+ struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
+ session_mq_publish_message(sess, env->test_mq_topic_id, plugin_env);// publish success, but won't be delivered to currnet plugin
+ stellar_session_plugin_dettach_current_session(sess);
+ ctx->called+=1;
+ session_mq_publish_message(sess, env->test_mq_topic_id, plugin_env);// publish success, but won't be delivered to currnet plugin
+ return ctx;
+}
+
+static void test_dettach_session_ctx_free(struct session *sess, void *session_ctx, void *plugin_env)
+{
+ struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
+ env->basic_ctx_free_called+=1;
+ struct test_basic_ctx *ctx=(struct test_basic_ctx *)session_ctx;
+ printf("now free ctx, called is %d\n", ctx->called);
+ session_mq_publish_message(sess, env->test_mq_topic_id, plugin_env);// publish success, but won't be delivered to currnet plugin
+ free(ctx);
+}
+
+static void test_dettach_on_session(struct session *sess, int topic_id, const void *msg , void *per_session_ctx, void *plugin_env)
+{
+ struct test_basic_ctx *ctx=(struct test_basic_ctx *)per_session_ctx;
+ // struct session_plugin_env *env = (struct session_plugin_env *)plugin_env;
+ ctx->called+=1;
+}
+
+static void test_dettach_msg_free(struct session *sess, void *msg, void *msg_free_arg)
+{
+ struct session_plugin_env *env = (struct session_plugin_env *)msg_free_arg;
+ env->test_mq_free_called+=1;
+ return;
+}
+
+void * test_plugin_init(struct stellar * st)
+{
+ struct session_plugin_env * env = CALLOC(struct session_plugin_env, 1);
+ // env.plug_mgr=plug_mgr;
+ // env->N_per_session_pkt_cnt=10;
+ // env->N_session=10
+
+ /* 先插入一个C插件 */
+ int plugin_id=stellar_session_plugin_register(st, test_dettach_session_ctx_new, test_dettach_session_ctx_free, env);
+ printf("plugin regist id is %d\n", plugin_id);
+
+ env->intrinsc_tcp_topic_id=stellar_session_mq_get_topic_id(st, TOPIC_TCP);
+ printf("TCP topic id is %d\n", env->intrinsc_egress_topic_id);
+ stellar_session_mq_subscribe(st, env->intrinsc_tcp_topic_id, test_dettach_on_session, plugin_id);
+
+ env->test_mq_topic_id=stellar_session_mq_create_topic(st, "SESSION_MQ_TOPIC", test_dettach_msg_free, &env);
+ stellar_session_mq_subscribe(st, env->test_mq_topic_id, test_dettach_on_session, plugin_id);
+
+ return env;
+}
+
+void test_plugin_free(void * plug_env)
+{
+ struct session_plugin_env * env = (struct session_plugin_env *)plug_env;
+ if (env) free(env);
+ return;
+} \ No newline at end of file
diff --git a/include/lua_plugin_manage.h b/include/lua_plugin_manage.h
index 0bbf19c..6259a3d 100644
--- a/include/lua_plugin_manage.h
+++ b/include/lua_plugin_manage.h
@@ -14,7 +14,7 @@
* lua_plugin_manage_exit
* 2. 声明数据结构
* struct lua_plugin_manage_schema
- *
+ *
* 08-09
* 1. 修改参数函数原型, 传入参数修改为已经加载的配置信息
************************************************************************/
@@ -34,5 +34,5 @@ struct lua_config_specific
struct lua_plugin_manage_schema;
-struct lua_plugin_manage_schema *lua_plugin_manage_init(struct stellar *st, int specific_count, struct lua_config_specific *specific);
+struct lua_plugin_manage_schema *lua_plugin_manage_init(struct stellar *st, int specific_count, struct lua_config_specific *specifics);
void lua_plugin_manage_exit(struct lua_plugin_manage_schema *lua_plug_mgr); \ No newline at end of file