stellar-stellar-2022/src/main.cpp

/*
**********************************************************************************************
*	File: main.cpp
*   Description: stellar main entry
*
*   Copyright: (c) 2018-2022 Geedge Networks, Inc. All rights reserved.
***********************************************************************************************
*/

#include <pthread.h>
#include <unistd.h>

#include "global_var.h"
#include "logger.h"
#include "packet_io.h"
#include "session_manager.h"
#include "plugin_manager.h"
#include "http.h"

struct stellar_event_base_loop_arg
{
	struct packet_io_device *dev;
	struct session_manager *session_mgr;
	struct plugin_manager *plug_mgr;
	int tid;
};

void *stellar_event_base_loop(void *arg)
{
	struct stellar_packet *rx_pkt;
	struct stellar_event *event;
	struct stellar_event_base_loop_arg *thread_arg = (struct stellar_event_base_loop_arg *)arg;
	while(1)
	{
		int fetch_num = packet_io_device_rx(thread_arg->dev, thread_arg->tid, &rx_pkt, 1);
		if(fetch_num > 0)
		{
			event = session_manager_commit(thread_arg->session_mgr, rx_pkt);
			while(event)
			{
				plugin_manager_dispatch(thread_arg->plug_mgr ,event);
				event = session_manager_fetch_event(thread_arg->session_mgr);
			}

			//clean session_manager event queue
			packet_io_device_tx(thread_arg->dev, thread_arg->tid, &rx_pkt, 1);
		}
		else
		{
			printf("no fetch num\n");
			//dispatch to time event

			//dispatch to trigger polling event 
		}
	}
	return nullptr;
}


struct packet_io_device *
packet_io_init(const char *instance_name, const enum packet_io_run_mode mode, const int wrk_thread_num)
{
	struct packet_io_instance *ppio_inst = packet_io_instance_create(instance_name, mode);
	if (nullptr == ppio_inst) {
		log_error(ST_ERR_PIO_INSTANCE, "packet_io instance init failed.");
		return nullptr;
	}

	struct packet_io_device *ppio_dev = packet_io_device_open(ppio_inst, "eth1", wrk_thread_num, wrk_thread_num);
	if (nullptr == ppio_dev) {
		log_error(ST_ERR_PIO_DEVICE, "packet_io device open failed.");
		
	}
	return ppio_dev;
}

int main(int argc, char ** argv)
{
	//packet_io_init
	struct packet_io_device *dev = packet_io_init("stellar", PACKET_IO_RUN_MODE_PCAP_LIVE, 2);

	//manager_init
	struct session_manager *session_mgr = session_manager_init();
	struct plugin_manager *plug_mgr = plugin_manager_create();

	// register build-in plugin
	plugin_manager_register(plug_mgr, "HTTP", SESSION_EVENT_ALL, http_entry);

	// load external plugins
	char file_path[] = "./plugs/plugins.inf";
	plugin_manager_load(plug_mgr, file_path);

	//create_worker_thread
	stellar_event_base_loop_arg arg = {dev, session_mgr, plug_mgr, 0};
	pthread_t worker_pid;
	pthread_create(&worker_pid, nullptr, stellar_event_base_loop, (void *)&arg);
	//main_loop
	while (1)
	{
		/* main loop code */
		usleep(1);
	}

	plugin_manager_unload(plug_mgr);
	plugin_manager_destory(plug_mgr);
	
	return 0;
}