199 lines
4.3 KiB
C
199 lines
4.3 KiB
C
#include "plugin_manager.h"
|
|
|
|
#include "stellar/stellar.h"
|
|
|
|
#include "stellar/session_exdata.h"
|
|
#include "stellar/session_mq.h"
|
|
|
|
|
|
#include "stellar/packet_exdata.h"
|
|
#include "stellar/packet_mq.h"
|
|
|
|
|
|
#include "bitmap.h"
|
|
#include "utarray.h"
|
|
|
|
struct per_thread_exdata_array
|
|
{
|
|
struct stellar_exdata *exdata_array;
|
|
};
|
|
|
|
struct stellar_message;
|
|
|
|
struct plugin_manger_per_thread_data
|
|
{
|
|
struct per_thread_exdata_array per_thread_pkt_exdata_array;
|
|
struct stellar_message *priority_mq[SESSION_MQ_PRIORITY_MAX];// message list
|
|
struct stellar_message *dealth_letter_queue;// dlq list
|
|
long long pub_packet_msg_cnt;
|
|
};
|
|
|
|
|
|
|
|
struct plugin_manager_schema
|
|
{
|
|
struct stellar *st;
|
|
UT_array *plugin_load_specs_array;
|
|
UT_array *packet_exdata_schema_array;
|
|
UT_array *session_exdata_schema_array;
|
|
UT_array *stellar_mq_schema_array;
|
|
UT_array *registered_session_plugin_array;
|
|
UT_array *registered_packet_plugin_array;
|
|
UT_array *registered_polling_plugin_array;
|
|
int packet_mq_topic_num;
|
|
int session_mq_topic_num;
|
|
int packet_topic_subscriber_num;
|
|
int session_topic_subscriber_num;
|
|
int tcp_topic_id;
|
|
int tcp_stream_topic_id;
|
|
int udp_topic_id;
|
|
int egress_topic_id;
|
|
int control_packet_topic_id;
|
|
int max_message_dispatch;
|
|
struct plugin_manger_per_thread_data *per_thread_data;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
enum plugin_exdata_state
|
|
{ INIT, ACTIVE, EXIT };
|
|
|
|
struct stellar_exdata
|
|
{
|
|
void *exdata;
|
|
enum plugin_exdata_state state;
|
|
};
|
|
|
|
|
|
|
|
struct stellar_exdata_schema
|
|
{
|
|
char *name;
|
|
union
|
|
{
|
|
void *free_func;
|
|
session_exdata_free *sess_free_func;
|
|
packet_exdata_free *pkt_free_func;
|
|
};
|
|
|
|
void *free_arg;
|
|
int idx;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
|
|
enum stellar_topic_type
|
|
{
|
|
ON_SESSION_TOPIC,
|
|
ON_PACKET_TOPIC,
|
|
};
|
|
|
|
struct stellar_message
|
|
{
|
|
struct
|
|
{
|
|
int topic_id;
|
|
enum stellar_topic_type type;
|
|
enum session_mq_priority priority;
|
|
} header;
|
|
void *body;
|
|
struct stellar_message *next, *prev;
|
|
} __attribute__((aligned(sizeof(void *))));
|
|
|
|
typedef struct stellar_mq_subscriber
|
|
{
|
|
int topic_subscriber_idx;
|
|
int plugin_idx;
|
|
union
|
|
{
|
|
on_session_msg_cb_func *sess_msg_cb;
|
|
on_packet_msg_cb_func *pkt_msg_cb;
|
|
void *msg_cb;
|
|
};
|
|
struct stellar_mq_subscriber *next, *prev;
|
|
}stellar_mq_subscriber __attribute__((aligned(sizeof(void*))));
|
|
|
|
|
|
struct stellar_mq_topic_schema
|
|
{
|
|
char *topic_name;
|
|
void *free_cb_arg;
|
|
int topic_id;
|
|
int subscriber_cnt;
|
|
int is_destroyed;
|
|
union
|
|
{
|
|
void *free_cb;
|
|
session_msg_free_cb_func *sess_msg_free_cb;
|
|
packet_msg_free_cb_func *pkt_msg_free_cb;
|
|
};
|
|
struct stellar_mq_subscriber *subscribers;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
|
|
|
|
struct session_plugin_ctx_runtime
|
|
{
|
|
enum plugin_exdata_state state;
|
|
int session_plugin_id;
|
|
void *plugin_ctx;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
|
|
|
|
struct plugin_manager_runtime
|
|
{
|
|
struct plugin_manager_schema *plug_mgr;
|
|
struct session *sess;
|
|
struct bitmap *session_mq_status; //N * M bits, N topic, M subscriber
|
|
struct bitmap *session_topic_status; //N bits, N topic
|
|
struct stellar_exdata *sess_exdata_array;
|
|
struct session_plugin_ctx_runtime *plugin_ctx_array;//N plugins TODO: call alloc and free
|
|
int current_session_plugin_id;
|
|
int pub_session_msg_cnt;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
struct registered_packet_plugin_schema
|
|
{
|
|
char ip_protocol;
|
|
plugin_on_packet_func *on_packet;
|
|
void *plugin_env;
|
|
UT_array *registed_packet_mq_subscriber_info;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
struct registered_polling_plugin_schema
|
|
{
|
|
plugin_on_polling_func *on_polling;
|
|
void *plugin_env;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
struct stellar_mq_subscriber_info
|
|
{
|
|
int topic_id;
|
|
int subscriber_idx;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
struct registered_session_plugin_schema
|
|
{
|
|
session_ctx_new_func *on_ctx_new;
|
|
session_ctx_free_func *on_ctx_free;
|
|
void *plugin_env;
|
|
UT_array *registed_session_mq_subscriber_info;
|
|
}__attribute__((aligned(sizeof(void*))));
|
|
|
|
#define SESSION_PULGIN_ID_BASE 0x00000
|
|
#define PACKET_PULGIN_ID_BASE 0x10000
|
|
#define POLLING_PULGIN_ID_BASE 0x20000
|
|
|
|
/*******************************
|
|
* PLUGIN MANAGER INIT & EXIT *
|
|
*******************************/
|
|
|
|
#define MAX_MSG_PER_DISPATCH 128
|
|
|
|
#include <dlfcn.h>
|
|
|
|
struct plugin_specific
|
|
{
|
|
char plugin_name[256];
|
|
plugin_on_load_func *load_cb;
|
|
plugin_on_unload_func *unload_cb;
|
|
void *plugin_ctx;
|
|
}__attribute__((aligned(sizeof(void*)))); |